mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-16 22:48:09 +01:00
initial working with ollama
This commit is contained in:
21
run_ollama.py
Normal file
21
run_ollama.py
Normal file
@@ -0,0 +1,21 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user