Update app.py

app.py

@@ -38,14 +38,35 @@ 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(),
-            system_prompt=SYSTEM_PROMPT
         )
         super().__init__(model=model, tools=tools)
+        self.system_prompt = SYSTEM_PROMPT
+
+    def __call__(self, task):
+        # Inject the system prompt manually
+        if isinstance(task, dict) and "question" in task:
+            task["messages"] = [
+                {"role": "system", "content": self.system_prompt},
+                {"role": "user", "content": task["question"]}
+            ]
+        return super().__call__(task)
+
+
 
 # Evaluation + Submission function
 def run_and_submit_all(profile: gr.OAuthProfile | None):