Y Phung Nguyen commited on
Commit
2f3ac98
·
1 Parent(s): 78ace46

Disable session check mcp

Browse files
Files changed (1) hide show
  1. client.py +2 -25
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
- # Check if existing session is still valid
46
  if config.global_mcp_session is not None:
47
- try:
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()