Update app.py

app.py

@@ -16,29 +16,38 @@ from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, ApiModel
 
 from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, ApiModel
 
+from smolagents import ToolCallingAgent, DuckDuckGoSearchTool
+import requests
+import os
+
+class HFModel:
+    def __init__(self):
+        self.API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2"
+        self.headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+
+    def generate(self, prompt):
+        response = requests.post(
+            self.API_URL,
+            headers=self.headers,
+            json={"inputs": prompt, "parameters": {"max_new_tokens": 100}}
+        )
+        return response.json()[0]["generated_text"]
+
+
 class BasicAgent:
     def __init__(self):
-        print("🔥 Smart Agent initialized")
+        print("🔥 Custom HF Agent initialized")
 
-        # ✅ FIXED MODEL (IMPORTANT)
-        self.model = ApiModel(
-            model_id="mistralai/Mistral-7B-Instruct-v0.2",
-            provider="hf"   # ⭐ THIS LINE FIXES YOUR ERROR
-        )
+        self.model = HFModel()
 
         self.agent = ToolCallingAgent(
             tools=[DuckDuckGoSearchTool()],
             model=self.model,
-            max_steps=10,
+            max_steps=8,
             name="gaia_agent",
             description="Smart QA agent"
         )
 
-        self.agent.system_prompt = """
-        Return ONLY the final answer.
-        No explanation.
-        """
-
     def clean(self, text):
         text = str(text).strip()
         text = text.replace("Final Answer:", "")