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.
2025-07-14 12:38:41 +00:00 · 2025-07-14 12:38:41 +00:00 · f701410055
parent a7952ff024
commit f701410055
1 changed files with 1 additions and 1 deletions
--- 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()