update

app.py

@@ -4,6 +4,7 @@ import os
 import gradio as gr
 import pandas as pd
 import requests
+from mcp import StdioServerParameters
 from smolagents import (
     CodeAgent,
     DuckDuckGoSearchTool,
@@ -11,6 +12,7 @@ from smolagents import (
     load_tool,
     tool,
     Tool,
+    ToolCollection,
     VisitWebpageTool,
 )
 
@@ -26,13 +28,26 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
 
-        self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
-            model=InferenceClientModel(),
-            additional_authorized_imports=["requests", "bs4"],
-            max_steps=20,
+        server_parameters = StdioServerParameters(
+            command="uvx",
+            args=["--quiet", "mediawiki-mcp-server@0.0.1"],
+            env={"UV_PYTHON": "3.12", **os.environ},
         )
 
+        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"""

requirements.txt

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