Update app.py

app.py

@@ -15,37 +15,32 @@ from langchain.tools import Tool
 from langgraph.graph import START, StateGraph
 from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode, tools_condition
-from langchain_mistralai import ChatMistralAI
+from langchain_deepseek import ChatDeepSeek
 import cv2
 
 
 import getpass
 import os
 
-if "Mistral" not in os.environ:
-    os.environ["Mistral"] = getpass.getpass("Enter your Mistral API key: ")
+if "DS_AGENT_API" not in os.environ:
+    os.environ["DS_AGENT_API"] = getpass.getpass("Enter your DS API key: ")
 
-print("Loading LLM...")
-chat3 = ChatMistralAI(
-    model="mistral-large-2411",
-    temperature=0.01,
-    max_retries=6,
-    mistral_api_key = os.getenv("Mistral")
-)
-chat2 = ChatMistralAI(
-    model="pixtral-large-2411",
+
+chat1 = ChatDeepSeek(
+    model="deepseek-chat",
     temperature=0.01,
     max_retries=6,
-    mistral_api_key = os.getenv("Mistral")
+    api_key = os.getenv("DS_AGENT_API")
 )
-print(f"Model {chat3.model} downloaded!")
-chat1 = ChatMistralAI(
-    model="mistral-medium-2505",
+print(f"Model {chat1.model} downloaded!")
+
+chat2 = ChatDeepSeek(
+    model="deepseek-reasoner",
     temperature=0.01,
     max_retries=6,
-    mistral_api_key = os.getenv("Mistral")
+    api_key = os.getenv("DS_AGENT_API")
 )
-print(f"Model {chat1.model} downloaded!")
+print(f"Model {chat2.model} downloaded!")
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -208,10 +203,10 @@ class BasicAgent:
         wrong_answers = ["Requests rate limit exceeded", "", " ", "  ", "insufficient information"]
         while not answer or answer in wrong_answers or answer.lower().startswith("error"):   
             try:
-                alfred = build_agent(chat2)
+                alfred = build_agent(chat1)
                 answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content   
             except:
-                alfred = build_agent(chat1)
+                alfred = build_agent(chat2)
                 answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content 
      
         if answer: