black formatting

This commit is contained in:
Steve Androulakis
2024-12-31 12:28:02 -08:00
parent c6ea92bfe5
commit 396b748b7d
3 changed files with 20 additions and 20 deletions

View File

@@ -1,21 +1,23 @@
from ollama import chat, ChatResponse
def main():
model_name = 'mistral'
model_name = "mistral"
# The messages to pass to the model
messages = [
{
'role': 'user',
'content': 'Why is the sky blue?',
"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()
if __name__ == "__main__":
main()