mle-kit-mcp 0.2.2__tar.gz → 0.2.4__tar.gz

{mle_kit_mcp-0.2.2 → mle_kit_mcp-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.2.2
+Version: 0.2.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.2.2 → mle_kit_mcp-0.2.4}/mle_kit_mcp/llm_proxy.py

@@ -6,7 +6,6 @@ from openai import AsyncOpenAI
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import JSONResponse
 
-ACCESS_TOKEN = os.getenv("ACCESS_TOKEN", "")
 OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY", "")
 OPENROUTER_BASE_URL = os.getenv("OPENROUTER_BASE_URL", "https://openrouter.ai/api/v1")
 
@@ -14,17 +13,8 @@ app = FastAPI()
 client = AsyncOpenAI(base_url=OPENROUTER_BASE_URL, api_key=OPENROUTER_API_KEY)
 
 
-def _check_auth(request: Request) -> None:
-    auth = request.headers.get("authorization", "")
-    if not ACCESS_TOKEN or auth != f"Bearer {ACCESS_TOKEN}":
-        raise HTTPException(status_code=401, detail="Unauthorized")
-    if not OPENROUTER_API_KEY:
-        raise HTTPException(status_code=500, detail="OpenRouter key not configured")
-
-
 @app.post("/v1/chat/completions")
 async def chat_completions(request: Request) -> JSONResponse:
-    _check_auth(request)
     payload = await request.json()
     if isinstance(payload, dict) and payload.get("stream"):
         payload.pop("stream", None)

{mle_kit_mcp-0.2.2 → mle_kit_mcp-0.2.4}/mle_kit_mcp/tools/llm_proxy.py

@@ -2,15 +2,12 @@ import os
 import json
 import time
 import random
-import secrets
 from pathlib import Path
-from typing import Optional
 
 from dotenv import load_dotenv
 
 from mle_kit_mcp.tools.bash import get_container
 from mle_kit_mcp.files import get_workspace_dir
-from mle_kit_mcp.utils import find_free_port
 
 from mle_kit_mcp.tools.remote_gpu import (
     get_instance as _remote_get_instance,
@@ -30,13 +27,12 @@ def _write_proxy_script(script_path: Path) -> None:
     script_path.write_text(script)
 
 
-def llm_proxy_local(port: Optional[int] = None) -> str:
+def llm_proxy_local() -> str:
     """
     Start a lightweight OpenRouter proxy inside the same Docker container used by the "bash" tool.
 
-    Returns a JSON string with fields: url, token, scope.
+    Returns a JSON string with url and scope.
     The url is reachable from inside the "bash" container as localhost.
-    Use the token in the Authorization header: "Bearer <token>" when calling the proxy.
     It runs a standard OpenAI compatible server, so you can use it with any OpenAI compatible client.
     You can use all models available on OpenRouter, for instance:
     - openai/gpt-5-mini
@@ -57,10 +53,9 @@ def llm_proxy_local(port: Optional[int] = None) -> str:
     dependencies_cmd = f"python -m pip install --quiet --no-input {DEPENDENCIES}"
     container.exec_run(["bash", "-lc", dependencies_cmd])
 
-    chosen_port = port or find_free_port()
-    token = secrets.token_urlsafe(24)
+    chosen_port = random.randint(5000, 6000)
     launch_cmd = (
-        f"OPENROUTER_API_KEY='{api_key}' ACCESS_TOKEN='{token}' "
+        f"OPENROUTER_API_KEY='{api_key}' "
         f"nohup python {OUTPUT_SCRIPT_FILE_NAME} "
         f"--host 127.0.0.1 --port {chosen_port} "
         f"> llm_proxy.log 2>&1 "
@@ -81,27 +76,22 @@ def llm_proxy_local(port: Optional[int] = None) -> str:
     return json.dumps(
         {
             "url": f"http://127.0.0.1:{chosen_port}/v1/chat/completions",
-            "token": token,
             "scope": "bash-container",
         }
     )
 
 
-def llm_proxy_remote(port: Optional[int] = None) -> str:
+def llm_proxy_remote() -> str:
     """
     Start a lightweight OpenRouter proxy on the remote GPU machine.
 
-    Returns a JSON string with fields: url, token, scope.
+    Returns a JSON string with url and scope.
     The url is reachable from inside the remote machine as localhost.
-    Use the token in the Authorization header: "Bearer <token>" when calling the proxy.
     It runs a standard OpenAI compatible server, so you can use it with any OpenAI compatible client.
     You can use all models available on OpenRouter, for instance:
     - openai/gpt-5-mini
     - google/gemini-2.5-pro
     - anthropic/claude-sonnet-4
-
-    Args:
-        port: Optional fixed port to bind on the remote. Random if omitted.
     """
 
     load_dotenv()
@@ -113,13 +103,12 @@ def llm_proxy_remote(port: Optional[int] = None) -> str:
     _write_proxy_script(script_path)
     _remote_send_rsync(instance, f"{script_path}", "/root")
 
-    chosen_port = port or random.randint(5000, 6000)
-    token = secrets.token_urlsafe(24)
+    chosen_port = random.randint(5000, 6000)
     dependencies_cmd = f"python3 -m pip install -q --no-input {DEPENDENCIES}"
     _remote_run_command(instance, dependencies_cmd, timeout=300)
 
     launch_cmd = (
-        f"OPENROUTER_API_KEY='{api_key}' ACCESS_TOKEN='{token}' "
+        f"OPENROUTER_API_KEY='{api_key}' "
         f"nohup python {OUTPUT_SCRIPT_FILE_NAME} "
         f"--host 127.0.0.1 --port {chosen_port} "
         f"> openrouter_proxy.log 2>&1 "
@@ -140,7 +129,6 @@ def llm_proxy_remote(port: Optional[int] = None) -> str:
     return json.dumps(
         {
             "url": f"http://127.0.0.1:{chosen_port}/v1/chat/completions",
-            "token": token,
             "scope": "remote-gpu",
         }
     )

{mle_kit_mcp-0.2.2 → mle_kit_mcp-0.2.4}/mle_kit_mcp/tools/remote_gpu.py

@@ -354,11 +354,17 @@ def create_remote_text_editor(
         command = args_dict["command"]
 
         if command != "write":
+            dir_path = "/".join(path.split("/")[:-1])
+            if dir_path:
+                recieve_rsync(instance, f"/root/{path}", f"{get_workspace_dir()}/{dir_path}")
             recieve_rsync(instance, f"/root/{path}", f"{get_workspace_dir()}")
 
         result: str = text_editor_func(*args, **kwargs)
 
         if command != "view":
+            dir_path = "/".join(path.split("/")[:-1])
+            if dir_path:
+                send_rsync(instance, f"{get_workspace_dir()}/{path}", f"/root/{dir_path}")
             send_rsync(instance, f"{get_workspace_dir()}/{path}", "/root")
 
         return result

{mle_kit_mcp-0.2.2 → mle_kit_mcp-0.2.4}/mle_kit_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.2.2
+Version: 0.2.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.2.2 → mle_kit_mcp-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mle-kit-mcp"
-version = "0.2.2"
+version = "0.2.4"
 description = "MCP server that provides different tools for MLE"
 readme = "README.md"
 authors = [

{mle_kit_mcp-0.2.2 → mle_kit_mcp-0.2.4}/tests/test_llm_proxy.py

@@ -4,21 +4,18 @@ from mle_kit_mcp.tools import llm_proxy_local, bash
 
 QUERY_SNIPPET = """
 import httpx
-headers = {{'Authorization': 'Bearer {token}'}}
 json_payload = {{'model': 'gpt-4o', 'messages': [{{'role': 'user', 'content': 'Hello, how are you?'}}]}}
-response = httpx.post("{url}", headers=headers, json=json_payload)
+response = httpx.post("{url}", json=json_payload)
 print(response.json())
 """
 
 
 def test_llm_proxy_local():
-    result = json.loads(llm_proxy_local(port=8001))
-    token = result["token"]
+    result = json.loads(llm_proxy_local())
     url = result["url"]
     assert url
-    assert token
 
-    snippet = QUERY_SNIPPET.format(url=url, token=token)
+    snippet = QUERY_SNIPPET.format(url=url)
     result = bash(f'cat > test_query.py << "EOF"\n{snippet}\nEOF')
     result = bash("python test_query.py")
     assert "content" in result