mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-15 22:18:09 +01:00
24 lines
458 B
Python
24 lines
458 B
Python
from ollama import chat, ChatResponse
|
|
|
|
|
|
def main():
|
|
model_name = "mistral"
|
|
|
|
# The messages to pass to the model
|
|
messages = [
|
|
{
|
|
"role": "user",
|
|
"content": "Why is the sky blue?",
|
|
}
|
|
]
|
|
|
|
# Call ollama's chat function
|
|
response: ChatResponse = chat(model=model_name, messages=messages)
|
|
|
|
# Print the full message content
|
|
print(response.message.content)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|