also install dependencies

Author: zhjch05

src/mcp_server_metatool/server.py

@@ -9,6 +9,8 @@
 import os
 import re
 import tempfile
+import subprocess
+import ast
 
 
 def sanitize_name(name: str) -> str:
@@ -48,6 +50,34 @@ async def get_mcp_servers() -> list[StdioServerParameters]:
         return []
 
 
+def extract_imports(code: str) -> list[str]:
+    """Extract top-level import statements from the Python code."""
+    try:
+        tree = ast.parse(code)
+        imports = set()
+
+        for node in ast.walk(tree):
+            if isinstance(node, ast.Import):
+                for alias in node.names:
+                    imports.add(alias.name.split(".")[0])
+            elif isinstance(node, ast.ImportFrom) and node.module:
+                imports.add(node.module.split(".")[0])
+
+        return list(imports)
+    except Exception as e:
+        print(f"Error parsing imports: {e}")
+        return []
+
+
+def install_dependencies(dependencies: list[str]):
+    """Install required dependencies using uv pip."""
+    try:
+        print(f"Checking dependencies: {dependencies}")
+        subprocess.run(["uv", "pip", "install"] + dependencies, check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Failed to install dependencies: {e}")
+
+
 async def get_custom_mcp_servers() -> list[StdioServerParameters]:
     try:
         async with httpx.AsyncClient() as client:
@@ -58,23 +88,35 @@ async def get_custom_mcp_servers() -> list[StdioServerParameters]:
             response.raise_for_status()
             data = response.json()
             server_params = []
+
             for params in data:
                 if "code" not in params or "code_uuid" not in params:
                     continue
+
                 code_uuid = params["code_uuid"]
+
+                # Create temp file for the script
                 with tempfile.NamedTemporaryFile(
                     mode="w", suffix=f"_{code_uuid}.py", delete=False
                 ) as temp_file:
                     temp_file.write(params["code"])
+                    script_path = temp_file.name
+
+                # Extract dependencies from the code
+                dependencies = extract_imports(params["code"])
+                if dependencies:
+                    install_dependencies(dependencies)
+
                 params["command"] = "uv"
-                params["args"] = ["run", temp_file.name] + params.get(
-                    "additionalArgs", []
-                )
+                params["args"] = ["run", script_path] + params.get("additionalArgs", [])
+
                 if "env" in params and not params["env"]:
                     params["env"] = None
+
                 server_params.append(StdioServerParameters(**params))
             return server_params
-    except Exception:
+    except Exception as e:
+        print(f"Error fetching MCP servers: {e}")
         return []