Skip to content

Commit 94dbedf

Browse files
author
jzhu
committed
fix: restore imports removed by autoflake and add tool class exports
Critical fixes after Phase 2 cleanup: - Restore MCP client imports in mcp/clients.py (autoflake removed re-exports) - Add missing CrushMCPClient and CursorAgentMCPClient to mcp/__init__.py - Export all tool classes from tools/__init__.py for backward compatibility - Fix IfLowTool class name (was incorrectly IflowTool) Impact: - All imports now working correctly - Tests can import tool classes as before - No breaking changes to public API - Unit tests passing (39/39 in factory and value_objects)
1 parent 4be95e1 commit 94dbedf

3 files changed

Lines changed: 47 additions & 0 deletions

File tree

code_assistant_manager/mcp/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@
1111
CodeBuddyMCPClient,
1212
CodexMCPClient,
1313
CopilotMCPClient,
14+
CrushMCPClient,
15+
CursorAgentMCPClient,
1416
DroidMCPClient,
1517
GeminiMCPClient,
1618
IflowMCPClient,
@@ -31,6 +33,8 @@
3133
"QwenMCPClient",
3234
"CopilotMCPClient",
3335
"CodeBuddyMCPClient",
36+
"CrushMCPClient",
37+
"CursorAgentMCPClient",
3438
"DroidMCPClient",
3539
"IflowMCPClient",
3640
"ZedMCPClient",
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,31 @@
11
"""Imports for all MCP client implementations."""
2+
3+
from .claude import ClaudeMCPClient
4+
from .codebuddy import CodeBuddyMCPClient
5+
from .codex import CodexMCPClient
6+
from .copilot import CopilotMCPClient
7+
from .crush import CrushMCPClient
8+
from .cursor import CursorAgentMCPClient
9+
from .droid import DroidMCPClient
10+
from .gemini import GeminiMCPClient
11+
from .iflow import IflowMCPClient
12+
from .neovate import NeovateMCPClient
13+
from .qodercli import QoderCLIMCPClient
14+
from .qwen import QwenMCPClient
15+
from .zed import ZedMCPClient
16+
17+
__all__ = [
18+
"ClaudeMCPClient",
19+
"CodeBuddyMCPClient",
20+
"CodexMCPClient",
21+
"CopilotMCPClient",
22+
"CrushMCPClient",
23+
"CursorAgentMCPClient",
24+
"DroidMCPClient",
25+
"GeminiMCPClient",
26+
"IflowMCPClient",
27+
"NeovateMCPClient",
28+
"QoderCLIMCPClient",
29+
"QwenMCPClient",
30+
"ZedMCPClient",
31+
]

code_assistant_manager/tools/__init__.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,19 @@
3838
# Expose model selector functions
3939

4040
# Backwards-compatible exports: expose tool classes at package level
41+
from .claude import ClaudeTool # noqa: F401
42+
from .codebuddy import CodeBuddyTool # noqa: F401
43+
from .codex import CodexTool # noqa: F401
44+
from .copilot import CopilotTool # noqa: F401
45+
from .crush import CrushTool # noqa: F401
46+
from .cursor import CursorTool # noqa: F401
47+
from .droid import DroidTool # noqa: F401
48+
from .gemini import GeminiTool # noqa: F401
49+
from .iflow import IfLowTool # noqa: F401
50+
from .neovate import NeovateTool # noqa: F401
51+
from .qodercli import QoderCLITool # noqa: F401
52+
from .qwen import QwenTool # noqa: F401
53+
from .zed import ZedTool # noqa: F401
4154

4255
# Expose endpoint display functions
4356
from .endpoint_display import ( # noqa: F401

0 commit comments

Comments
 (0)