Spaces:
Running
on
Zero
Running
on
Zero
Y Phung Nguyen
commited on
Commit
·
2f3ac98
1
Parent(s):
78ace46
Disable session check mcp
Browse files
client.py
CHANGED
|
@@ -42,32 +42,9 @@ async def get_mcp_session():
|
|
| 42 |
logger.warning("MCP not available - SDK not installed")
|
| 43 |
return None
|
| 44 |
|
| 45 |
-
#
|
| 46 |
if config.global_mcp_session is not None:
|
| 47 |
-
|
| 48 |
-
# Try to list tools as a quick health check (with timeout)
|
| 49 |
-
await asyncio.wait_for(config.global_mcp_session.list_tools(), timeout=2.0)
|
| 50 |
-
return config.global_mcp_session
|
| 51 |
-
except asyncio.TimeoutError:
|
| 52 |
-
logger.warning("MCP session health check timed out, recreating session")
|
| 53 |
-
# Clean up invalid session
|
| 54 |
-
try:
|
| 55 |
-
if config.global_mcp_stdio_ctx is not None:
|
| 56 |
-
await config.global_mcp_stdio_ctx.__aexit__(None, None, None)
|
| 57 |
-
except Exception:
|
| 58 |
-
pass
|
| 59 |
-
config.global_mcp_session = None
|
| 60 |
-
config.global_mcp_stdio_ctx = None
|
| 61 |
-
except Exception as e:
|
| 62 |
-
logger.warning(f"Existing MCP session appears invalid, recreating: {type(e).__name__}: {e}")
|
| 63 |
-
# Clean up invalid session
|
| 64 |
-
try:
|
| 65 |
-
if config.global_mcp_stdio_ctx is not None:
|
| 66 |
-
await config.global_mcp_stdio_ctx.__aexit__(None, None, None)
|
| 67 |
-
except Exception:
|
| 68 |
-
pass
|
| 69 |
-
config.global_mcp_session = None
|
| 70 |
-
config.global_mcp_stdio_ctx = None
|
| 71 |
|
| 72 |
try:
|
| 73 |
mcp_env = os.environ.copy()
|
|
|
|
| 42 |
logger.warning("MCP not available - SDK not installed")
|
| 43 |
return None
|
| 44 |
|
| 45 |
+
# Reuse existing session if available
|
| 46 |
if config.global_mcp_session is not None:
|
| 47 |
+
return config.global_mcp_session
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
|
| 49 |
try:
|
| 50 |
mcp_env = os.environ.copy()
|