Update app.py

app.py

@@ -10,10 +10,6 @@ from audio_transcriber import AudioTranscriptionTool
 from image_analyzer import ImageAnalysisTool
 from wikipedia_searcher import WikipediaSearcher
 
-#wikipedia_tool = WikipediaSearcher()
-
-
-
 # GAIA scoring endpoint
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -37,25 +33,7 @@ Only return the exact answer."""
 # Define agent tools
 audio_tool = AudioTranscriptionTool()
 image_tool = ImageAnalysisTool()
-#wikipedia_tool = Tool.from_function(
-#    name="wikipedia_search",
- #   description="Search for facts using Wikipedia.",
- #   input_schema={"query": {"type": "string", "description": "Search query"}},
- #   output_type="string",
- #   forward=lambda query: WikipediaSearcher().search(query)
-#)
-
-
-
-# Wrap WikipediaSearcher search method as a Tool
-wikipedia_tool = Tool.from_function(
-    name="wikipedia_search",
-    description="Search for facts using Wikipedia.",
-    input_schema={"query": {"type": "string", "description": "Search query"}},
-    output_type="string",
-    forward=lambda query: WikipediaSearcher().search(query)
-)
-
+wikipedia_tool = WikipediaSearcher()
 
 tools = [audio_tool, image_tool, wikipedia_tool]
 
@@ -66,7 +44,7 @@ class MyAgent(CodeAgent):
             model="cognitivecomputations/dolphin-2.6-mixtral-8x7b",
             api_key=os.getenv("HF_API_TOKEN", "").strip()
         )
-        super().__init__(model=model, tools=tools)
+        super().__init__(model=model, tools=tools, system_prompt=SYSTEM_PROMPT)
 
 # Evaluation + Submission function
 def run_and_submit_all(profile: gr.OAuthProfile | None):