mle-kit-mcp 0.2.6__tar.gz → 1.0.1__tar.gz

{mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.2.6
+Version: 1.0.1
 Summary: MCP server that provides different tools for MLE
 Author-email: Ilya Gusev <phoenixilya@gmail.com>
 Project-URL: Homepage, https://github.com/IlyaGusev/mle_kit_mcp

{mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.1}/mle_kit_mcp/server.py

@@ -1,9 +1,8 @@
 import os
 from pathlib import Path
-from typing import Optional
+from typing import Optional, Literal
 
 import fire  # type: ignore
-import uvicorn
 from mcp.server.fastmcp import FastMCP
 from dotenv import load_dotenv
 
@@ -21,14 +20,26 @@ from .tools.llm_proxy import (
 from .files import get_workspace_dir, WorkspaceDirectory
 
 
-def run(host: str = "0.0.0.0", port: int = 5050, workspace: Optional[str] = None) -> None:
+def run(
+    host: str = "0.0.0.0",
+    port: int = 5050,
+    mount_path: str = "/",
+    streamable_http_path: str = "/mcp",
+    workspace: Optional[str] = None,
+    transport: Literal["stdio", "sse", "streamable-http"] = "streamable-http",
+) -> None:
     load_dotenv()
     if workspace:
         WorkspaceDirectory.set_dir(Path(workspace))
     workspace_path = get_workspace_dir()
     workspace_path.mkdir(parents=True, exist_ok=True)
 
-    server = FastMCP("MLE kit MCP", stateless_http=True)
+    server = FastMCP(
+        "MLE kit MCP",
+        stateless_http=True,
+        streamable_http_path=streamable_http_path,
+        mount_path=mount_path,
+    )
 
     remote_text_editor = create_remote_text_editor(text_editor)
 
@@ -41,9 +52,9 @@ def run(host: str = "0.0.0.0", port: int = 5050, workspace: Optional[str] = None
         server.add_tool(llm_proxy_local)
         server.add_tool(llm_proxy_remote)
 
-    http_app = server.streamable_http_app()
-
-    uvicorn.run(http_app, host=host, port=port)
+    server.settings.port = port
+    server.settings.host = host
+    server.run(transport=transport)
 
 
 if __name__ == "__main__":

{mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.1}/mle_kit_mcp/tools/remote_gpu.py

@@ -357,7 +357,8 @@ def create_remote_text_editor(
             dir_path = "/".join(path.split("/")[:-1])
             if dir_path:
                 recieve_rsync(instance, f"/root/{path}", f"{get_workspace_dir()}/{dir_path}")
-            recieve_rsync(instance, f"/root/{path}", f"{get_workspace_dir()}")
+            else:
+                recieve_rsync(instance, f"/root/{path}", f"{get_workspace_dir()}")
 
         result: str = text_editor_func(*args, **kwargs)
 
@@ -365,7 +366,8 @@ def create_remote_text_editor(
             dir_path = "/".join(path.split("/")[:-1])
             if dir_path:
                 send_rsync(instance, f"{get_workspace_dir()}/{path}", f"/root/{dir_path}")
-            send_rsync(instance, f"{get_workspace_dir()}/{path}", "/root")
+            else:
+                send_rsync(instance, f"{get_workspace_dir()}/{path}", "/root")
 
         return result
 

{mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.1}/mle_kit_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.2.6
+Version: 1.0.1
 Summary: MCP server that provides different tools for MLE
 Author-email: Ilya Gusev <phoenixilya@gmail.com>
 Project-URL: Homepage, https://github.com/IlyaGusev/mle_kit_mcp

{mle_kit_mcp-0.2.6 → mle_kit_mcp-1.0.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mle-kit-mcp"
-version = "0.2.6"
+version = "1.0.1"
 description = "MCP server that provides different tools for MLE"
 readme = "README.md"
 authors = [