mle-kit-mcp 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- mle_kit_mcp/tools/remote_gpu.py +3 -3
- {mle_kit_mcp-0.0.7.dist-info → mle_kit_mcp-0.0.9.dist-info}/METADATA +1 -1
- {mle_kit_mcp-0.0.7.dist-info → mle_kit_mcp-0.0.9.dist-info}/RECORD +7 -7
- {mle_kit_mcp-0.0.7.dist-info → mle_kit_mcp-0.0.9.dist-info}/WHEEL +0 -0
- {mle_kit_mcp-0.0.7.dist-info → mle_kit_mcp-0.0.9.dist-info}/entry_points.txt +0 -0
- {mle_kit_mcp-0.0.7.dist-info → mle_kit_mcp-0.0.9.dist-info}/licenses/LICENSE +0 -0
- {mle_kit_mcp-0.0.7.dist-info → mle_kit_mcp-0.0.9.dist-info}/top_level.txt +0 -0
mle_kit_mcp/tools/remote_gpu.py
CHANGED
@@ -126,7 +126,7 @@ def run_command(
|
|
126
126
|
except subprocess.TimeoutExpired:
|
127
127
|
raise Exception(
|
128
128
|
f"Command timed out after {timeout} seconds: {command};\n"
|
129
|
-
f"You can increase the timeout by
|
129
|
+
f"You can increase the timeout by changing the parameter of the tool call."
|
130
130
|
)
|
131
131
|
return result
|
132
132
|
|
@@ -182,7 +182,7 @@ def launch_instance(vast_sdk: VastAI, gpu_name: str) -> Optional[InstanceInfo]:
|
|
182
182
|
|
183
183
|
for offer_id in offer_ids:
|
184
184
|
print(f"Launching offer {offer_id}...")
|
185
|
-
instance = vast_sdk.create_instance(id=offer_id, image=BASE_IMAGE, disk=
|
185
|
+
instance = vast_sdk.create_instance(id=offer_id, image=BASE_IMAGE, disk=300.0)
|
186
186
|
if not instance["success"]:
|
187
187
|
continue
|
188
188
|
instance_id = instance["new_contract"]
|
@@ -283,7 +283,7 @@ def init_all() -> None:
|
|
283
283
|
assert _instance_info, "Failed to connect to a remote instance! Try again"
|
284
284
|
|
285
285
|
|
286
|
-
def remote_bash(command: str, timeout:
|
286
|
+
def remote_bash(command: str, timeout: int = 60) -> str:
|
287
287
|
"""
|
288
288
|
Run commands in a bash shell on a remote machine with GPU cards.
|
289
289
|
When invoking this tool, the contents of the "command" parameter does NOT need to be XML-escaped.
|
@@ -6,11 +6,11 @@ mle_kit_mcp/server.py,sha256=E9cXdKwVBASBzKyrZGHCTvj6BIMN-EbVSQZDFNg0YnE,1062
|
|
6
6
|
mle_kit_mcp/utils.py,sha256=wd7wSyddHRHOYdxmXw8uoAOBxVZOL2_vjNomss07inc,1654
|
7
7
|
mle_kit_mcp/tools/__init__.py,sha256=0aLl0gD-JteSvOs2PgVhbv0Wnh6fodFySgQWQvoI1xI,215
|
8
8
|
mle_kit_mcp/tools/bash.py,sha256=10hYo_hIBvQh8wbjssixb89SXG05H6nHtndShr6V31E,2461
|
9
|
-
mle_kit_mcp/tools/remote_gpu.py,sha256
|
9
|
+
mle_kit_mcp/tools/remote_gpu.py,sha256=ik4Za5eiCAN6sk0l1kcc_AgX7feUx7rWEBAnIXmkdNc,11483
|
10
10
|
mle_kit_mcp/tools/text_editor.py,sha256=EJ832vgdFdycrLOf3eOYMXnlMgB2_iH_u4ANvZVrMsE,9139
|
11
|
-
mle_kit_mcp-0.0.
|
12
|
-
mle_kit_mcp-0.0.
|
13
|
-
mle_kit_mcp-0.0.
|
14
|
-
mle_kit_mcp-0.0.
|
15
|
-
mle_kit_mcp-0.0.
|
16
|
-
mle_kit_mcp-0.0.
|
11
|
+
mle_kit_mcp-0.0.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
12
|
+
mle_kit_mcp-0.0.9.dist-info/METADATA,sha256=wqPSyJ3nBZQlAFR4ChHRVzijdBp58Zk9Fla0ixfnxTg,1011
|
13
|
+
mle_kit_mcp-0.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
14
|
+
mle_kit_mcp-0.0.9.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
|
15
|
+
mle_kit_mcp-0.0.9.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
|
16
|
+
mle_kit_mcp-0.0.9.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|