Update agent.py

agent.py

@@ -148,11 +148,16 @@ def build_graph(provider: str = "huggingface"):
         chat = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
     elif provider == "huggingface":
         # Add huggingface endpoint
+        repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
         chat = ChatHuggingFace(
+            #llm=HuggingFaceEndpoint(
+            #    endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
+            #    temperature=0,
+            #),
             llm=HuggingFaceEndpoint(
-                endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
-                temperature=0,
-            ),
+                repo_id=repo_id,
+                temperature=0.1
+            )
         )
     else:
         raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")