update

app.py

@@ -4,7 +4,6 @@ import os
 import gradio as gr
 import pandas as pd
 import requests
-from mcp import StdioServerParameters
 from smolagents import (
     CodeAgent,
     DuckDuckGoSearchTool,
@@ -27,31 +26,21 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-
-        server_parameters = StdioServerParameters(
-            command="uvx",
-            args=["--quiet", "mediawiki-mcp-server@0.0.1"],
-            env={"UV_PYTHON": "3.12", **os.environ},
+        self.agent = CodeAgent(
+            tools=[
+                VisitWebpageTool(),
+                DuckDuckGoSearchTool(),
+            ],
+            model=InferenceClientModel(),
+            additional_authorized_imports=["requests", "bs4", "wikipedia"],
+            max_steps=20,
         )
 
-        with ToolCollection.from_mcp(
-            server_parameters, trust_remote_code=True
-        ) as tool_collection:
-            self.agent = CodeAgent(
-                tools=[
-                    DuckDuckGoSearchTool(),
-                    VisitWebpageTool(),
-                    *tool_collection.tools,
-                ],
-                model=InferenceClientModel(),
-                additional_authorized_imports=["requests", "bs4"],
-                max_steps=20,
-            )
-
     def __call__(self, question: str) -> str:
         print(f"Agent received question: {question}")
         prompt = f"""
-            Answer the following question: `{question}`.
+            Answer the following question: `{question}`. Please follow following rules:
+            1. wikipedia Python package is provided to extract data from Wikipedia pages, please use it when possible.
         """
         result = self.agent.run(prompt)
         print(f"Agent responded with: {result}")

requirements.txt

@@ -4,5 +4,4 @@ smolagents
 duckduckgo_search
 bs4
 markdownify
-smolagents[mcp]
-mediawiki-mcp-server
+wikipedia