mirror of
https://github.com/ollama/ollama-python.git
synced 2026-01-13 21:57:16 +08:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from ollama import chat
|
|
|
|
messages = [
|
|
{
|
|
'role': 'user',
|
|
'content': 'Why is the sky blue?',
|
|
},
|
|
{
|
|
'role': 'assistant',
|
|
'content': "The sky is blue because of the way the Earth's atmosphere scatters sunlight.",
|
|
},
|
|
{
|
|
'role': 'user',
|
|
'content': 'What is the weather in Tokyo?',
|
|
},
|
|
{
|
|
'role': 'assistant',
|
|
'content': 'The weather in Tokyo is typically warm and humid during the summer months, with temperatures often exceeding 30°C (86°F). The city experiences a rainy season from June to September, with heavy rainfall and occasional typhoons. Winter is mild, with temperatures rarely dropping below freezing. The city is known for its high-tech and vibrant culture, with many popular tourist attractions such as the Tokyo Tower, Senso-ji Temple, and the bustling Shibuya district.',
|
|
},
|
|
]
|
|
|
|
while True:
|
|
user_input = input('Chat with history: ')
|
|
response = chat(
|
|
'llama3.2',
|
|
messages=[*messages, {'role': 'user', 'content': user_input}],
|
|
)
|
|
|
|
# Add the response to the messages to maintain the history
|
|
messages += [
|
|
{'role': 'user', 'content': user_input},
|
|
{'role': 'assistant', 'content': response.message.content},
|
|
]
|
|
print(response.message.content + '\n')
|