fix: inference mcp servers

Author: damassi

agent-chat-cli.config.yaml

@@ -10,7 +10,7 @@ include_partial_messages: true
 
 # Enable dynamic/lazy MCP server inference. Useful if one has many MCP servers or
 # many tools, or is cost conscious about loading everything up front.
-mcp_server_inference: false
+mcp_server_inference: true
 
 # Global tool restrictions
 disallowed_tools: ["Bash"]

src/agent_chat_cli/core/agent_loop.py

@@ -25,7 +25,6 @@
 from agent_chat_cli.utils.enums import AgentMessageType, ContentType, ControlCommand
 from agent_chat_cli.core.mcp_inference import infer_mcp_servers
 from agent_chat_cli.utils.logger import log_json
-from agent_chat_cli.utils.mcp_server_status import MCPServerStatus
 
 if TYPE_CHECKING:
     from agent_chat_cli.app import AgentChatCLIApp
@@ -166,7 +165,7 @@ async def _handle_message(self, message: Any) -> None:
                 self.session_id = message.data["session_id"]
 
                 # Report status back to UI
-                MCPServerStatus.update(message.data["mcp_servers"])
+                # MCPServerStatus.update(message.data["mcp_servers"])
 
         # Handle streaming messages
         if hasattr(message, "event"):

src/agent_chat_cli/utils/mcp_server_status.py

@@ -1,5 +1,7 @@
 from typing import Any, Callable
 
+from agent_chat_cli.utils.config import load_config
+
 
 class MCPServerStatus:
     # After the first query is sent, claude agent sdk sends back an init payload which
@@ -18,6 +20,13 @@ def update(cls, mcp_servers: list[dict[str, Any]]) -> None:
 
     @classmethod
     def is_connected(cls, server_name: str) -> bool:
+        config = load_config()
+
+        if config.mcp_server_inference is True and not cls._mcp_servers:
+            # If we're inferring servers based on input, we assume things can connect
+            # until we actually fetch the server, at which point status is updated.
+            return True
+
         for server in cls._mcp_servers:
             if server.get("name") == server_name:
                 return server.get("status") == "connected"