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()