Update app.py

app.py

@@ -38,37 +38,21 @@ wikipedia_tool = WikipediaSearcher()
 tools = [audio_tool, image_tool, wikipedia_tool]
 
 # Define the custom agent using Dolphin model (free Mixtral)
-#class MyAgent(CodeAgent):
- #   def __init__(self):
-    #    model = HfApiModel(
-     #       model="cognitivecomputations/dolphin-2.6-mixtral-8x7b",
-       #     api_key=os.getenv("HF_API_TOKEN", "").strip(),
-       #     system_prompt=SYSTEM_PROMPT
-       # )
-       # super().__init__(model=model, tools=tools)
-
-
 class MyAgent(CodeAgent):
     def __init__(self):
         model = HfApiModel(
             model="cognitivecomputations/dolphin-2.6-mixtral-8x7b",
-            api_key=os.getenv("HF_API_TOKEN", "").strip()
+            api_key=os.getenv("HF_API_TOKEN", "").strip(),
+            # No system_prompt here
         )
         super().__init__(model=model, tools=tools)
-        self.system_prompt = SYSTEM_PROMPT
-
-    def __call__(self, task):
-        # Inject system prompt manually
-        if isinstance(task, dict):
-            if "question" in task:
-                task["messages"] = [
-                    {"role": "system", "content": self.system_prompt},
-                    {"role": "user", "content": task["question"]}
-                ]
-            elif "messages" in task:
-                task["messages"].insert(0, {"role": "system", "content": self.system_prompt})
-        return super().__call__(task)
 
+    def __call__(self, question_dict):
+        system_message = {"role": "system", "content": SYSTEM_PROMPT}
+        user_message = {"role": "user", "content": question_dict.get("question", "")}
+        messages = [system_message, user_message]
+        # Pass messages directly in the call
+        return self.model(messages)
 
 # Evaluation + Submission function
 def run_and_submit_all(profile: gr.OAuthProfile | None):