mle-kit-mcp 0.2.5__tar.gz → 1.0.0__tar.gz

{mle_kit_mcp-0.2.5 → mle_kit_mcp-1.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.2.5
+Version: 1.0.0
 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.5 → mle_kit_mcp-1.0.0}/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.5 → mle_kit_mcp-1.0.0}/mle_kit_mcp/tools/bash.py

@@ -1,5 +1,6 @@
 import atexit
 import signal
+import shlex
 from typing import Optional, Any
 
 from docker import from_env as docker_from_env  # type: ignore
@@ -67,7 +68,11 @@ signal.signal(signal.SIGINT, cleanup_container)
 signal.signal(signal.SIGTERM, cleanup_container)
 
 
-def bash(command: str, cwd: Optional[str] = None) -> str:
+def bash(
+    command: str,
+    cwd: Optional[str] = None,
+    timeout: int = 60,
+) -> str:
     """
     Run commands in a bash shell.
     When invoking this tool, the contents of the "command" parameter does NOT need to be XML-escaped.
@@ -76,22 +81,48 @@ def bash(command: str, cwd: Optional[str] = None) -> str:
     State is persistent across command calls and discussions with the user.
     To inspect a particular line range of a file, e.g. lines 10-25, try 'sed -n 10,25p /path/to/the/file'.
     Please avoid commands that may produce a very large amount of output.
-    Please run long lived commands in the background, e.g. 'sleep 10 &' or start a server in the background.
 
     Args:
         command: The bash command to run.
         cwd: The working directory to run the command in. Relative to the workspace directory.
+        timeout: Timeout for the command execution in seconds. Kills after this. 60 seconds by default.
     """
-
+    assert timeout and timeout > 0, "Timeout must be set and greater than 0"
     container = get_container()
     workdir = DOCKER_WORKSPACE_DIR_PATH
     if cwd:
-        workdir = DOCKER_WORKSPACE_DIR_PATH + "/" + cwd
+        workdir += "/" + cwd
+
+    wrapped = f"bash -lc {shlex.quote(command)}"
+    final_command = f"timeout --signal=TERM --kill-after=5s {int(timeout)}s {wrapped}"
+
     result = container.exec_run(
-        ["bash", "-c", command],
+        ["bash", "-lc", final_command],
         workdir=workdir,
         stdout=True,
         stderr=True,
+        demux=True,
     )
-    output: str = result.output.decode("utf-8").strip()
+    stdout_bytes, stderr_bytes = (
+        result.output if isinstance(result.output, tuple) else (result.output, b"")
+    )
+
+    if result.exit_code in (124, 137):
+        timeout_msg = (
+            f"Command timed out after {int(timeout)} seconds: {command};\n"
+            f"You can increase the timeout by changing the parameter of the tool call."
+        )
+        stderr_bytes = (stderr_bytes or b"") + timeout_msg.encode("utf-8")
+
+    stdout_text = (stdout_bytes or b"").decode("utf-8", errors="replace").strip()
+    stderr_text = (stderr_bytes or b"").decode("utf-8", errors="replace").strip()
+
+    output_parts = []
+    if stdout_text:
+        output_parts.append("Command stdout: " + stdout_text)
+    if stderr_text:
+        output_parts.append("Command stderr: " + stderr_text)
+    output = ("\n".join(output_parts)).strip()
+    if not output:
+        output = "No output from the command"
     return output

{mle_kit_mcp-0.2.5 → mle_kit_mcp-1.0.0}/mle_kit_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.2.5
+Version: 1.0.0
 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.5 → mle_kit_mcp-1.0.0}/pyproject.toml

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

{mle_kit_mcp-0.2.5 → mle_kit_mcp-1.0.0}/tests/test_bash.py

@@ -6,10 +6,10 @@ from mle_kit_mcp.files import get_workspace_dir
 
 def test_bash() -> None:
     result = bash('echo "Hello World"')
-    assert result == "Hello World"
+    assert result == "Command stdout: Hello World"
 
     result = bash("pwd")
-    assert result == "/workdir"
+    assert result == "Command stdout: /workdir"
 
     result = bash("touch dummy")
     assert os.path.exists(get_workspace_dir() / "dummy")
@@ -22,3 +22,14 @@ def test_bash_cwd() -> None:
     bash("mkdir -p dummy_dir")
     bash("touch dummy", cwd="dummy_dir")
     assert os.path.exists(get_workspace_dir() / "dummy_dir" / "dummy")
+
+
+def test_bash_timeout_base() -> None:
+    result = bash("sleep 10", timeout=5)
+    assert "Command timed out" in result
+
+
+def test_bash_timeout_with_output() -> None:
+    result = bash("echo 'hello' && sleep 100", timeout=5)
+    assert "hello" in result
+    assert "Command timed out" in result

{mle_kit_mcp-0.2.5 → mle_kit_mcp-1.0.0}/tests/test_text_editor.py

@@ -5,6 +5,7 @@ import pytest
 
 from mle_kit_mcp.tools import text_editor
 from mle_kit_mcp.files import get_workspace_dir
+from mle_kit_mcp.tools.text_editor import READ_MAX_OUTPUT_LENGTH
 
 DOCUMENT1 = """
 The dominant sequence transduction models are based on complex recurrent or convolutional
@@ -211,7 +212,7 @@ def test_text_editor_large_file_handling() -> None:
 
         result = text_editor("view", name)
         assert "This content has been truncated" in result
-        assert len(result) <= 3100
+        assert len(result) <= READ_MAX_OUTPUT_LENGTH + 100
 
         result = text_editor("view", name, view_end_line=5)
         assert "<response clipped>" not in result