mle-kit-mcp 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

mle_kit_mcp/tools/bash.py

@@ -9,7 +9,7 @@ from mle_kit_mcp.files import get_workspace_dir
 _container = None
 _client = None
 
-BASE_IMAGE = "python:3.9-slim"
+BASE_IMAGE = "python:3.12-slim"
 DOCKER_WORKSPACE_DIR_PATH = "/workdir"
 
 

mle_kit_mcp/tools/remote_gpu.py

@@ -37,10 +37,10 @@ _instance_info: Optional[InstanceInfo] = None
 
 
 def cleanup_machine(signum: Optional[Any] = None, frame: Optional[Any] = None) -> None:
-    print("Cleaning up...")
     global _instance_info
     signal.alarm(0)
     if _instance_info and _sdk:
+        print("Cleaning up...")
         try:
             _sdk.destroy_instance(id=_instance_info.instance_id)
             _instance_info = None
@@ -94,7 +94,7 @@ def get_offers(vast_sdk: VastAI, gpu_name: str) -> List[int]:
 
 
 def run_command(
-    instance: InstanceInfo, command: str, timeout: int = 60
+    instance: InstanceInfo, command: str, timeout: int = 60, raise_exc: bool = True
 ) -> subprocess.CompletedProcess[str]:
     cmd = [
         "ssh",
@@ -117,7 +117,7 @@ def run_command(
     ]
     try:
         result = subprocess.run(cmd, capture_output=True, text=True, timeout=timeout)
-        if result.returncode != 0:
+        if result.returncode != 0 and raise_exc:
             raise Exception(
                 f"Error running command: {command}; "
                 f"Output: {result.stdout}; "
@@ -178,6 +178,8 @@ def launch_instance(vast_sdk: VastAI, gpu_name: str) -> Optional[InstanceInfo]:
     offer_ids = get_offers(vast_sdk, gpu_name)
 
     instance_id = None
+    info: Optional[InstanceInfo] = None
+
     for offer_id in offer_ids:
         print(f"Launching offer {offer_id}...")
         instance = vast_sdk.create_instance(id=offer_id, image=BASE_IMAGE, disk=50.0)
@@ -301,7 +303,7 @@ def remote_bash(command: str, timeout: Optional[int] = 60) -> str:
     init_all()
     assert _instance_info
     assert timeout
-    result = run_command(_instance_info, command, timeout=timeout)
+    result = run_command(_instance_info, command, timeout=timeout, raise_exc=False)
     output = ("STDOUT: " + result.stdout + "\n") if result.stdout else ""
     output += ("STDERR: " + result.stderr) if result.stderr else ""
     return output.replace(VAST_AI_GREETING, "")

mle_kit_mcp/tools/text_editor.py

@@ -136,7 +136,6 @@ def text_editor(
     If `path` is a file and `show_lines` is True, `view` displays the result of applying `cat -n`.
     If `path` is a file and `show_lines` is False, `view` displays the result of applying `cat`.
     If `path` is a directory, `view` lists non-hidden files and directories up to 2 levels deep.
-    The `write` command cannot be used if the specified `path` already exists as a file.
     If a `command` generates a long output, it will be truncated and marked with `<response clipped>`.
     The `undo_edit` command will revert the last edit made to the file at `path`.
     Always write arguments with keys, do not rely on positions.

{mle_kit_mcp-0.0.3.dist-info → mle_kit_mcp-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.0.3
+Version: 0.0.5
 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
@@ -11,13 +11,9 @@ Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: mcp>=1.9.2
-Requires-Dist: black==25.1.0
-Requires-Dist: mypy==1.16.0
-Requires-Dist: flake8==7.2.0
 Requires-Dist: fire>=0.7.0
 Requires-Dist: docker>=7.1.0
 Requires-Dist: vastai-sdk==0.1.16
-Requires-Dist: pytest>=8.4.1
 Dynamic: license-file
 
 # MLE KIT MCP

{mle_kit_mcp-0.0.3.dist-info → mle_kit_mcp-0.0.5.dist-info}/RECORD

@@ -5,12 +5,12 @@ mle_kit_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mle_kit_mcp/server.py,sha256=E9cXdKwVBASBzKyrZGHCTvj6BIMN-EbVSQZDFNg0YnE,1062
 mle_kit_mcp/utils.py,sha256=wd7wSyddHRHOYdxmXw8uoAOBxVZOL2_vjNomss07inc,1654
 mle_kit_mcp/tools/__init__.py,sha256=0aLl0gD-JteSvOs2PgVhbv0Wnh6fodFySgQWQvoI1xI,215
-mle_kit_mcp/tools/bash.py,sha256=9IIcely_J_fvNPLdJvHFOL72OG3rVIpNC3MB5S36pJI,2460
-mle_kit_mcp/tools/remote_gpu.py,sha256=7bhkwlB-qnzb7CExKuCS7YNPg19K-SLjamHTxVzi2uU,11374
-mle_kit_mcp/tools/text_editor.py,sha256=s9lPzqQWpZ2277f99TgMyKeOSTrv-PhX4HVtdXVeQ24,9228
-mle_kit_mcp-0.0.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mle_kit_mcp-0.0.3.dist-info/METADATA,sha256=9xP8U4vC840biALBdSzPqA8UI8zyr-HloOW0V8ps1wg,1126
-mle_kit_mcp-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mle_kit_mcp-0.0.3.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
-mle_kit_mcp-0.0.3.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
-mle_kit_mcp-0.0.3.dist-info/RECORD,,
+mle_kit_mcp/tools/bash.py,sha256=10hYo_hIBvQh8wbjssixb89SXG05H6nHtndShr6V31E,2461
+mle_kit_mcp/tools/remote_gpu.py,sha256=paYXwbHm7-ROddcDayRKIkqacUPVxTcvvj-CLs26i6E,11474
+mle_kit_mcp/tools/text_editor.py,sha256=EJ832vgdFdycrLOf3eOYMXnlMgB2_iH_u4ANvZVrMsE,9139
+mle_kit_mcp-0.0.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mle_kit_mcp-0.0.5.dist-info/METADATA,sha256=5HZ8PgimNnqIVHBKyJUUU6RwbYi1QjeIRSW_a1Z6yOA,1011
+mle_kit_mcp-0.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mle_kit_mcp-0.0.5.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
+mle_kit_mcp-0.0.5.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
+mle_kit_mcp-0.0.5.dist-info/RECORD,,