-

pyproject.toml

@@ -7,5 +7,7 @@ requires-python = ">=3.12"
 dependencies = [
     "gradio[mcp]>=5.31.0",
     "mcp>=1.9.0",
-    "smolagents[mcp]>=1.16.1",
+    "ollama>=0.4.8",
+    "smolagents[litellm,mcp]>=1.16.1",
+    "sympy>=1.14.0",
 ]

server.py

@@ -9,12 +9,17 @@ try:
     mcp_client = MCPClient(
         ## Try this working example on the hub:
         # {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
-        {"url": "http://localhost:7860/gradio_api/mcp/sse"}
+        {"url": "http://127.0.0.1:7861/gradio_api/mcp/sse"}
     )
     tools = mcp_client.get_tools()
 
     model = InferenceClientModel()
-    agent = CodeAgent(tools=[*tools], model=model)
+    authorized_agent_imports = ["math", "textblob", "sympy"]
+    
+    agent = CodeAgent(tools=[*tools], 
+                      model=model,
+                      additional_authorized_imports = authorized_agent_imports,
+    )
 
     demo = gr.ChatInterface(
         fn=lambda message, history: str(agent.run(message)),