mle-kit-mcp 0.1.0__tar.gz → 0.1.1__tar.gz

{mle_kit_mcp-0.1.0 → mle_kit_mcp-0.1.1}/PKG-INFO

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

@@ -67,7 +67,7 @@ signal.signal(signal.SIGINT, cleanup_container)
 signal.signal(signal.SIGTERM, cleanup_container)
 
 
-def bash(command: str) -> str:
+def bash(command: str, cwd: Optional[str] = None) -> str:
     """
     Run commands in a bash shell.
     When invoking this tool, the contents of the "command" parameter does NOT need to be XML-escaped.
@@ -80,12 +80,16 @@ def bash(command: str) -> str:
 
     Args:
         command: The bash command to run.
+        cwd: The working directory to run the command in. Relative to the workspace directory.
     """
 
     container = get_container()
+    workdir = DOCKER_WORKSPACE_DIR_PATH
+    if cwd:
+        workdir = DOCKER_WORKSPACE_DIR_PATH + "/" + cwd
     result = container.exec_run(
         ["bash", "-c", command],
-        workdir=DOCKER_WORKSPACE_DIR_PATH,
+        workdir=workdir,
         stdout=True,
         stderr=True,
     )

{mle_kit_mcp-0.1.0 → mle_kit_mcp-0.1.1}/mle_kit_mcp/tools/text_editor.py

@@ -50,17 +50,23 @@ def _insert(path: Path, insert_line: int, new_str: str) -> str:
     return truncate_content(new_content, WRITE_MAX_OUTPUT_LENGTH, target_line=insert_line)
 
 
-def _str_replace(path: Path, old_str: str, new_str: str) -> str:
+def _str_replace(path: Path, old_str: str, new_str: str, dry_run: bool = False) -> str:
     assert path.is_file(), f"File not found: {path}"
     content = path.open().read()
     count = content.count(old_str)
     assert count != 0, "old_str not found in file"
     assert count == 1, "old_str is not unique in file"
     target_line = content[: content.find(old_str) + len(old_str)].count("\n")
-    _save_file_state(path, content.splitlines(True))
     new_content = content.replace(old_str, new_str)
-    path.write_text(new_content)
-    return truncate_content(new_content, WRITE_MAX_OUTPUT_LENGTH, target_line=target_line)
+    if not dry_run:
+        _save_file_state(path, content.splitlines(True))
+        path.write_text(new_content)
+    display_content = truncate_content(
+        new_content, WRITE_MAX_OUTPUT_LENGTH, target_line=target_line
+    )
+    if dry_run:
+        display_content = f"Dry run:\n{display_content}"
+    return display_content
 
 
 def _undo_edit(path: Path) -> str:
@@ -129,6 +135,7 @@ def text_editor(
     view_start_line: Optional[int] = None,
     view_end_line: Optional[int] = None,
     show_lines: Optional[bool] = False,
+    dry_run: Optional[bool] = False,
 ) -> str:
     """
     Custom editing tool for viewing, creating and editing files.
@@ -167,6 +174,7 @@ def text_editor(
         new_str: Required for `str_replace`, `insert` and `append`.
         old_str: Required for `str_replace` containing the string in `path` to replace.
         show_lines: Optional for view command. If True, the command will also output line numbers.
+        dry_run: Optional for `str_replace` command. If True, the command won't modify the file but will display the result.
     """
     assert not path.startswith(
         "/"
@@ -191,7 +199,8 @@ def text_editor(
     if command == "str_replace":
         assert old_str is not None, "'old_str' is required for 'str_replace' command"
         assert new_str is not None, "'new_str' is required for 'str_replace' command"
-        return _str_replace(path_obj, old_str, new_str)
+        assert dry_run is not None
+        return _str_replace(path_obj, old_str, new_str, dry_run=dry_run)
     if command == "undo_edit":
         return _undo_edit(path_obj)
     assert False, f"Not a valid command! List of commands: {valid_commands}"

{mle_kit_mcp-0.1.0 → mle_kit_mcp-0.1.1}/mle_kit_mcp.egg-info/PKG-INFO

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

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

{mle_kit_mcp-0.1.0 → mle_kit_mcp-0.1.1}/tests/test_bash.py

@@ -16,3 +16,9 @@ def test_bash() -> None:
 
     result = bash("fddafad")
     assert "fddafad: command not found" in result
+
+
+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")

{mle_kit_mcp-0.1.0 → mle_kit_mcp-0.1.1}/tests/test_text_editor.py

@@ -87,6 +87,18 @@ def test_text_editor_str_replace() -> None:
         assert "41.9" in new_content and "41.8" not in new_content
 
 
+def test_text_editor_str_replace_dry_run() -> None:
+    with tempfile.NamedTemporaryFile(dir=get_workspace_dir(), mode="w+") as f:
+        name = os.path.basename(f.name)
+        test_file = get_workspace_dir() / name
+        test_file.write_text(DOCUMENT1)
+
+        result = text_editor("str_replace", name, old_str="41.8", new_str="41.9", dry_run=True)
+        new_content = test_file.open().read()
+        assert "41.9" not in new_content and "41.8" in new_content
+        assert "Dry run" in result
+
+
 def test_text_editor_undo_edit() -> None:
     with tempfile.NamedTemporaryFile(dir=get_workspace_dir(), mode="w+") as f:
         name = os.path.basename(f.name)