uv adapt

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "vendor/sdk"]
+	path = vendor/sdk
+	url = https://github.com/Pamir-AI/distiller-cm5-sdk.git

distiller_cm5_python/client/cli.py

@@ -24,7 +24,7 @@
     MODEL_NAME,
     TIMEOUT,
     LOGGING_LEVEL,
-    MCP_SERVER_SCRIPT_PATH,
+    ACTIVE_MCP_SERVER,
     API_KEY,
 )
 from distiller_cm5_python.utils.distiller_exception import (
@@ -242,11 +242,7 @@ def get_transcription_sync(data):
 
 def parse_arguments():
     parser = argparse.ArgumentParser(description="MCP Client CLI")
-    parser.add_argument(
-        "--server-script",
-        default=MCP_SERVER_SCRIPT_PATH,
-        help="Path to the MCP server script (e.g., ../server/server.py)",
-    )
+
     parser.add_argument(
         "--stream",
         action=argparse.BooleanOptionalAction,
@@ -289,16 +285,7 @@ async def main():
     # Log the effective level
     logger.info(f"Log level set to: {args.log_level}")  # This logger is now configured
 
-    # Make server script path absolute, default to MCP_SERVER_SCRIPT_PATH
-    server_script_path = os.path.abspath(args.server_script)
-    if not os.path.exists(server_script_path):
-        logger.error(f"Server script not found at: {server_script_path}")
-        print(
-            f"{Fore.RED}Error: Server script not found at '{server_script_path}'. Please provide a valid path.{Style.RESET_ALL}"
-        )
-        sys.exit(1)
-
-    logger.info(f"Using server script: {server_script_path}")
+    # Make mcp-server server dir exist
 
     # Initialize event dispatcher and handler
     event_handler = CLIEventHandler()
@@ -352,7 +339,7 @@ async def main():
 
     try:
         logger.info("Connecting to MCP server...")
-        connected = await client.connect_to_server(server_script_path)
+        connected = await client.connect_to_server(ACTIVE_MCP_SERVER)
         if not connected:
             logger.error("Failed to connect to the MCP server.")
             print(

distiller_cm5_python/client/mid_layer/llm_client.py

@@ -690,6 +690,8 @@ async def get_chat_completion_response(
             if "tool_calls" in message and message["tool_calls"]:
                 logger.debug("Found tool calls directly in response message object.")
                 tool_calls = message["tool_calls"]
+                if isinstance(tool_calls[0]['function']['arguments'], str):
+                    tool_calls[0]['function']['arguments'] = json.loads(tool_calls[0]['function']['arguments'])
             elif (
                 isinstance(full_response_content, str)
                 and "<tool_call>" in full_response_content
@@ -995,11 +997,16 @@ async def get_chat_completion_streaming_response(
                 else:
                     self._emit_success(dispatcher, str(uuid.uuid4()), EventType.MESSAGE, "tool call parsing failed, please retry")
 
-            logger.info(
-                f"LLMClient.get_chat_completion_streaming_response: Processed result. Content length: {len(full_response_content)}, Tool calls: {len(final_tool_calls)}"
-            )
+                logger.info(
+                    f"LLMClient.get_chat_completion_streaming_response: Processed result. Content length: {len(full_response_content)}, Tool calls: {len(final_tool_calls)}"
+                )
+            else:
+
+                logger.debug("Found tool calls directly in response message object.")
+
+                if len(final_tool_calls)>0 and isinstance(final_tool_calls[0]['function']['arguments'], str):
+                    final_tool_calls[0]['function']['arguments'] = json.loads(final_tool_calls[0]['function']['arguments'])
 
-
             return {
                 "message": {
                     "content": full_response_content,

distiller_cm5_python/client/mid_layer/mcp_client.py

@@ -22,6 +22,7 @@
     PROVIDER_TYPE,
     MODEL_NAME,
     TIMEOUT,
+    MCP_SERVERS
 )
 from contextlib import AsyncExitStack
 
@@ -105,15 +106,15 @@ def __init__(
 
         self.dispatcher = dispatcher
 
-    async def connect_to_server(self, server_script_path: str) -> bool:
+    async def connect_to_server(self, active_mcp_server: str) -> bool:
         """Connect to an MCP server"""
-
-        if not server_script_path.endswith(".py"):
-            raise UserVisibleError("Server script must be a .py file")
+        logger.debug(f"Connecting to server: {active_mcp_server}")
 
         # use current python interpreter by default
         server_params = StdioServerParameters(
-            command=sys.executable, args=[server_script_path], env=None
+            command=MCP_SERVERS.get(active_mcp_server).get("command", None),
+            args=MCP_SERVERS.get(active_mcp_server).get("args", None),
+            env=MCP_SERVERS.get(active_mcp_server).get("env", None),
         )
 
         try:
@@ -139,7 +140,7 @@ async def connect_to_server(self, server_script_path: str) -> bool:
             if self.server_name == "cli":
                 from distiller_cm5_python.utils.server_utils import extract_server_name
 
-                self.server_name = extract_server_name(server_script_path)
+                self.server_name = extract_server_name(active_mcp_server)
                 logger.debug(f"Using extracted server name: {self.server_name}")
 
             end_time = time.time()

distiller_cm5_python/mcp_server/led_use/.python-version

@@ -0,0 +1 @@
+3.12

distiller_cm5_python/mcp_server/led_use/pyproject.toml

@@ -0,0 +1,10 @@
+[project]
+name = "led-use"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "mcp>=1.9.1",
+    "nest-asyncio>=1.6.0",
+]