new model

app.py

@@ -3,21 +3,20 @@ import streamlit as st
 from pydantic_ai import Agent
 from pydantic_ai.models.groq import GroqModel
 from pydantic_ai.models.anthropic import AnthropicModel
+from pydantic_ai.models.mistral import MistralModel
 import nest_asyncio
 import random
 import planet_model as customModel
 from typing import List, Dict
 import constants
 import os
-from ollama import chat
-from ollama import ChatResponse
 
 api_key = os.getenv("api_key")
 
 
 random_number = random.randint(3000, 10000)
 
-model = AnthropicModel('claude-3-5-sonnet-latest', api_key = api_key)
+model = MistralModel('mistral-small-latest', api_key = 'ZGs1ZTvw6csiboQnBC9mi1cmsBjAPRu7')
 
 
 async def onClick(vacationType,familyType,duration,purpose,interests):
@@ -88,15 +87,6 @@ async def main():
         duration = ""
         purpose = ""
         interests = ""
-    response: ChatResponse = chat(model='llama3.2', messages=[
-        {
-            'role': 'user',
-            'content': 'Why is the sky blue?',
-        },
-            ])
-    print(response['message']['content'])
-    # or access fields directly from the response object
-    print(response.message.content)
         
 
 if __name__ == '__main__':

requirements.txt

@@ -4,5 +4,4 @@ pdfplumber
 streamlit_pdf_viewer
 pydantic_ai
 nest_asyncio
-ollama
 #pydantic_ai.models.groq