MCP client disconnet

app.py

@@ -4,27 +4,28 @@ import os
 from mcp import StdioServerParameters
 from smolagents import InferenceClientModel, CodeAgent, ToolCollection, MCPClient
 
-mcp_client = MCPClient(
-    {
-        "url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse"
-    }
-)
+try:
+    mcp_client = MCPClient(
+        {
+            "url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse"
+        }
+    )
 
-tools = mcp_client.get_tools()
+    tools = mcp_client.get_tools()
 
-model = InferenceClientModel(token=os.getenv("HF_TOKEN"))
+    model = InferenceClientModel(token=os.getenv("HF_TOKEN"))
 
-agent = CodeAgent(tools=[*tools], model=model)
+    agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64")
 
-demo = gr.ChatInterface(
-    fn= lambda message, history: str(agent.run(message)),
-    type="messages",
-    examples=["Prime factorization of 68"],
-    title="Agent with MCP tool",
-    description="This agent can use the MCP tool to answer questions.",
-)
+    demo = gr.ChatInterface(
+        fn= lambda message, history: str(agent.run(message)),
+        type="messages",
+        examples=["Prime factorization of 68"],
+        title="Agent with MCP tool",
+        description="This agent can use the MCP tool to answer questions.",
+    )
 
-demo.launch()
+    demo.launch()
 
 finally:
-mcp_client.disconnect()
+    mcp_client.disconnect()