Alexi Canales commited on
Commit
0169438
·
1 Parent(s): e9d3bd2

Fixing MCP initialization

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,6 +4,7 @@ from mcp.client.stdio import StdioServerParameters
4
  from smolagents import InferenceClientModel, CodeAgent, ToolCollection
5
  from smolagents.mcp_client import MCPClient
6
 
 
7
  try:
8
  mcp_client = MCPClient(
9
  # Try this working example on the hub:
@@ -25,4 +26,5 @@ try:
25
 
26
  demo.launch()
27
  finally:
28
- mcp_client.disconnect()
 
 
4
  from smolagents import InferenceClientModel, CodeAgent, ToolCollection
5
  from smolagents.mcp_client import MCPClient
6
 
7
+ mcp_client = None
8
  try:
9
  mcp_client = MCPClient(
10
  # Try this working example on the hub:
 
26
 
27
  demo.launch()
28
  finally:
29
+ if mcp_client:
30
+ mcp_client.disconnect()