feat: Introduce `MCPClient` for multi-server communication and integrate with FastAPI lifespan events.

app.py

@@ -10,22 +10,25 @@ sys.path.append(os.path.join(os.path.dirname(__file__), "src"))
 
 from credentialwatch_agent.main import demo, mcp_client
 
-# Create FastAPI app
-app = FastAPI()
+from contextlib import asynccontextmanager
 
-@app.on_event("startup")
-async def startup_event():
-    """Connect to MCP servers on startup."""
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Manage application lifespan."""
     print("Connecting to MCP servers...")
-    # We can't use run_until_complete here because the loop is already running
-    await mcp_client.connect()
-
-@app.on_event("shutdown")
-async def shutdown_event():
-    """Close connections on shutdown."""
+    try:
+        await mcp_client.connect()
+    except Exception as e:
+        print(f"Error connecting to MCP servers: {e}")
+    
+    yield
+    
     print("Closing MCP connections...")
     await mcp_client.close()
 
+# Create FastAPI app with lifespan
+app = FastAPI(lifespan=lifespan)
+
 # Mount Gradio app
 # path="/" mounts it at the root
 app = gr.mount_gradio_app(app, demo, path="/")

src/credentialwatch_agent/mcp_client.py

@@ -36,7 +36,7 @@ class MCPClient:
             )
             await self._sessions["npi"].initialize()
         except Exception as e:
-            print(f"Failed to connect to NPI MCP at {self.npi_url}: {e}")
+            print(f"Warning: Failed to connect to NPI MCP at {self.npi_url}. Using mock data. Error: {e}")
 
         # Connect to Cred DB MCP
         try:
@@ -46,7 +46,7 @@ class MCPClient:
             )
             await self._sessions["cred_db"].initialize()
         except Exception as e:
-            print(f"Failed to connect to Cred DB MCP at {self.cred_db_url}: {e}")
+            print(f"Warning: Failed to connect to Cred DB MCP at {self.cred_db_url}. Using mock data. Error: {e}")
 
         # Connect to Alert MCP
         try:
@@ -56,7 +56,7 @@ class MCPClient:
             )
             await self._sessions["alert"].initialize()
         except Exception as e:
-            print(f"Failed to connect to Alert MCP at {self.alert_url}: {e}")
+            print(f"Warning: Failed to connect to Alert MCP at {self.alert_url}. Using mock data. Error: {e}")
 
     async def close(self):
         """Closes all connections."""