polished up the mcp client script

mcp_client.py

@@ -5,8 +5,11 @@ from smolagents import InferenceClientModel, CodeAgent, ToolCollection, MCPClien
 
 
 if __name__ == "__main__":
+
+    only_list_tools = True
     
     try:
+
         mcp_client = MCPClient(
             {"url": "https://agents-mcp-hackathon-f1-mcp-server.hf.space/gradio_api/mcp/sse", "transport": "sse"})
         tools = mcp_client.get_tools()
@@ -14,6 +17,10 @@ if __name__ == "__main__":
         print("### MCP tools ### ")
         print("\n".join(f"{t.name}: {t.description}" for t in tools))
 
+        if only_list_tools:
+            mcp_client.close()
+            exit(0)
+
         # Define model
         model = InferenceClientModel(
             model_id="Qwen/Qwen2.5-32B-Instruct",
@@ -22,17 +29,16 @@ if __name__ == "__main__":
 
         agent = CodeAgent(tools=[*tools], model=model)
 
-        run_inference = False # TEMP
-        if run_inference:
 
-            chat_interface = gr.ChatInterface(
-                fn=lambda message, history: str(agent.run(message)),
-                type="messages",
-                examples=["Prime factorization of 68"],
-                title="Agent with MCP Tools",
-                description="This is a simple agent that uses MCP tools to answer questions."
-            )
+        chat_interface = gr.ChatInterface(
+            fn=lambda message, history: str(agent.run(message)),
+            type="messages",
+            examples=["Prime factorization of 68"],
+            title="Agent with MCP Tools",
+            description="This is a simple agent that uses MCP tools to answer questions."
+        )
+
+        chat_interface.launch()
 
-            chat_interface.launch()
     finally:
         mcp_client.close()

mcp_client_template.py

@@ -1,15 +0,0 @@
-from smolagents.mcp_client import MCPClient
-
-
-mcp_client = MCPClient(
-    {"url": "https://abidlabs-mcp-tools2.hf.space/gradio_api/mcp/sse", "transport": "sse"}
-)
-
-
-
-
-if __name__ == "__main__":
-
-    # List all available tools from the MCP server
-    with mcp_client as tools:
-        print("\n".join(f"{t.name}: {t.description}" for t in tools))