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

mle_kit_mcp/tools/remote_gpu.py

@@ -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}; "
@@ -301,7 +301,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-0.0.3.dist-info → mle_kit_mcp-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.0.3
+Version: 0.0.4
 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.0.3.dist-info → mle_kit_mcp-0.0.4.dist-info}/RECORD

@@ -6,11 +6,11 @@ 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/remote_gpu.py,sha256=Rwy0vuPgBqZDonsJjhapbrBnw1dKViKUJonTC8AiPJE,11429
 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-0.0.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mle_kit_mcp-0.0.4.dist-info/METADATA,sha256=-kG6R55hu0Ta3nyNMY87DLWETscN7rMnSelcIISh4SA,1126
+mle_kit_mcp-0.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mle_kit_mcp-0.0.4.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
+mle_kit_mcp-0.0.4.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
+mle_kit_mcp-0.0.4.dist-info/RECORD,,