From f7014100557a5d7cbe0bb4a8db450114be328628 Mon Sep 17 00:00:00 2001 From: Huang Yuhao Date: Mon, 14 Jul 2025 12:38:41 +0000 Subject: [PATCH] fix(mcp): correct exception handling during agent lookup Replaced `get_agent_by_name()` with `get_available_agents()[agent_identifier]` when fixing `/mcp add`, which changes the raised exception from `ValueError` to `KeyError`. Adjusted error handling accordingly. --- src/cai/repl/commands/mcp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cai/repl/commands/mcp.py b/src/cai/repl/commands/mcp.py index 5f78e0e8..a2dc6884 100644 --- a/src/cai/repl/commands/mcp.py +++ b/src/cai/repl/commands/mcp.py @@ -812,7 +812,7 @@ Example: `/mcp add burp 13` try: agent = get_available_agents()[agent_identifier] agent_display_name = getattr(agent, "name", agent_identifier) - except ValueError: + except KeyError: # Try by index try: agents = get_available_agents()