mle-kit-mcp 0.0.4__tar.gz → 0.0.6__tar.gz

{mle_kit_mcp-0.0.4 → mle_kit_mcp-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.0.4
+Version: 0.0.6
 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.4 → mle_kit_mcp-0.0.6}/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-0.0.4 → mle_kit_mcp-0.0.6}/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
@@ -125,8 +125,11 @@ def run_command(
             )
     except subprocess.TimeoutExpired:
         raise Exception(
-            f"Command timed out after {timeout} seconds: {command}; "
-            f"Host: {instance.username}@{instance.ip}:{instance.port}"
+            f"Command timed out after {timeout} seconds: {command};\n"
+            f"Host: {instance.username}@{instance.ip}:{instance.port}\n"
+            f"Captured output: {result.stdout}\n"
+            f"Captured error: {result.stderr}\n"
+            f"You can increase the timeout by setting the timeout parameter to a higher value."
         )
     return result
 
@@ -178,6 +181,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)

{mle_kit_mcp-0.0.4 → mle_kit_mcp-0.0.6}/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.4 → mle_kit_mcp-0.0.6}/mle_kit_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.0.4
+Version: 0.0.6
 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.4 → mle_kit_mcp-0.0.6}/mle_kit_mcp.egg-info/requires.txt

@@ -1,8 +1,4 @@
 mcp>=1.9.2
-black==25.1.0
-mypy==1.16.0
-flake8==7.2.0
 fire>=0.7.0
 docker>=7.1.0
 vastai-sdk==0.1.16
-pytest>=8.4.1

{mle_kit_mcp-0.0.4 → mle_kit_mcp-0.0.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mle-kit-mcp"
-version = "0.0.4"
+version = "0.0.6"
 description = "MCP server that provides different tools for MLE"
 readme = "README.md"
 authors = [
@@ -18,12 +18,16 @@ classifiers = [
 ]
 dependencies = [
     "mcp>=1.9.2",
-    "black==25.1.0",
-    "mypy==1.16.0",
-    "flake8==7.2.0",
     "fire>=0.7.0",
     "docker>=7.1.0",
     "vastai-sdk==0.1.16",
+]
+
+[dependency-groups]
+dev = [
+    "black==25.1.0",
+    "mypy==1.16.0",
+    "flake8==7.2.0",
     "pytest>=8.4.1",
 ]