mle-kit-mcp 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

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/tools/llm_proxy.py

@@ -18,7 +18,7 @@ from mle_kit_mcp.tools.remote_gpu import (
     send_rsync as _remote_send_rsync,
 )
 
-INPUT_SCRIPT_FILE_NAME = "llm_proxy_source.py"
+INPUT_SCRIPT_FILE_NAME = "llm_proxy.py"
 OUTPUT_SCRIPT_FILE_NAME = "llm_proxy.py"
 DEPENDENCIES = "fastapi uvicorn httpx openai fire"
 START_TIMEOUT = 30

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.dist-info → mle_kit_mcp-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.1.0
+Version: 0.1.2
 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
@@ -14,6 +14,8 @@ Requires-Dist: mcp>=1.9.2
 Requires-Dist: fire>=0.7.0
 Requires-Dist: docker>=7.1.0
 Requires-Dist: vastai-sdk==0.1.16
+Requires-Dist: openai>=1.102.0
+Requires-Dist: fastapi>=0.116.1
 Dynamic: license-file
 
 # MLE KIT MCP

{mle_kit_mcp-0.1.0.dist-info → mle_kit_mcp-0.1.2.dist-info}/RECORD

@@ -1,18 +1,18 @@
 mle_kit_mcp/__init__.py,sha256=2Ru2I5u4cE7DrkkAsibDUEF1K6sYtqppb9VyFrRoQKI,94
 mle_kit_mcp/__main__.py,sha256=rcmsOtJd3SA82exjrcGBuxuptcoxF8AXI7jNjiVq2BY,59
 mle_kit_mcp/files.py,sha256=ux53kWw7hBAcOmS9qNI4gpQX8XcQPT2LICC--S5-TGI,635
-mle_kit_mcp/llm_proxy_source.py,sha256=zDfNDvG1nAkft-irj0KhdB-1z-UUtKweRJYG0fCveVE,1637
+mle_kit_mcp/llm_proxy.py,sha256=zDfNDvG1nAkft-irj0KhdB-1z-UUtKweRJYG0fCveVE,1637
 mle_kit_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mle_kit_mcp/server.py,sha256=W4YJ3m1-NKheJ5QlGggfMQyXPghzKgg5zXqb-TLYH1U,1271
 mle_kit_mcp/utils.py,sha256=iHNcEZZzPD37bEYE18SzJ3WUjLP3Ym-kc91SwcW1vlI,1984
 mle_kit_mcp/tools/__init__.py,sha256=r2fIg2mZ6zaeq0CzEKCEdeUTjV0pcA9NZaaOfBNVTnE,332
-mle_kit_mcp/tools/bash.py,sha256=9YmwRCR2HHTJ6Jhnn-fHuLRkDjsHo9m26tpdXoGCFtQ,2608
-mle_kit_mcp/tools/llm_proxy.py,sha256=9tm3k1R5y9o6Z4nkZXwILkYt6Ludc_TnQSwRw0KcNHA,5312
+mle_kit_mcp/tools/bash.py,sha256=kunYHc3dyPGOooT-KY9L7eI_N22lBrcDbTlcp_yTTws,2820
+mle_kit_mcp/tools/llm_proxy.py,sha256=eV8QwDBgsyfBzD454GzWsNjH6976E-e_fmO6CYpdLGc,5305
 mle_kit_mcp/tools/remote_gpu.py,sha256=Fv8SJC8bE3Oo8JHOKXoYpISXFMzUMFjEcCQOTyW7V1I,11544
-mle_kit_mcp/tools/text_editor.py,sha256=EJ832vgdFdycrLOf3eOYMXnlMgB2_iH_u4ANvZVrMsE,9139
-mle_kit_mcp-0.1.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mle_kit_mcp-0.1.0.dist-info/METADATA,sha256=aNQJCc6kZW4IaOvKi7KrxPJ0NqQgnFCmQqVk1uJ3yTg,1011
-mle_kit_mcp-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mle_kit_mcp-0.1.0.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
-mle_kit_mcp-0.1.0.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
-mle_kit_mcp-0.1.0.dist-info/RECORD,,
+mle_kit_mcp/tools/text_editor.py,sha256=0uGcSjcBjdPlqjYZYuWo29SmAMzBcWgKt4KRumFI3WE,9529
+mle_kit_mcp-0.1.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mle_kit_mcp-0.1.2.dist-info/METADATA,sha256=pGh_SbX6SFp_Ku1aOIUmPQjpDFXvBeXFebcyVzQcJVM,1074
+mle_kit_mcp-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mle_kit_mcp-0.1.2.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
+mle_kit_mcp-0.1.2.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
+mle_kit_mcp-0.1.2.dist-info/RECORD,,