mle-kit-mcp 0.2.3__py3-none-any.whl → 0.2.5__py3-none-any.whl

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/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/tools/text_editor.py

@@ -5,8 +5,8 @@ from pathlib import Path
 from mle_kit_mcp.files import get_workspace_dir
 from mle_kit_mcp.utils import truncate_content
 
-WRITE_MAX_OUTPUT_LENGTH = 500
-READ_MAX_OUTPUT_LENGTH = 3000
+WRITE_MAX_OUTPUT_LENGTH = 1000
+READ_MAX_OUTPUT_LENGTH = 10000
 
 # Global state for undo operations
 FILE_HISTORY: Dict[str, List[List[str]]] = defaultdict(list)

{mle_kit_mcp-0.2.3.dist-info → mle_kit_mcp-0.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mle-kit-mcp
-Version: 0.2.3
+Version: 0.2.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

{mle_kit_mcp-0.2.3.dist-info → mle_kit_mcp-0.2.5.dist-info}/RECORD

@@ -1,18 +1,18 @@
 mle_kit_mcp/__init__.py,sha256=2Ru2I5u4cE7DrkkAsibDUEF1K6sYtqppb9VyFrRoQKI,94
 mle_kit_mcp/__main__.py,sha256=rcmsOtJd3SA82exjrcGBuxuptcoxF8AXI7jNjiVq2BY,59
 mle_kit_mcp/files.py,sha256=ux53kWw7hBAcOmS9qNI4gpQX8XcQPT2LICC--S5-TGI,635
-mle_kit_mcp/llm_proxy.py,sha256=zDfNDvG1nAkft-irj0KhdB-1z-UUtKweRJYG0fCveVE,1637
+mle_kit_mcp/llm_proxy.py,sha256=01BG6OA8husOQXxgJQ7RnTNEE_1HDczlCNoAVnYWURQ,1225
 mle_kit_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mle_kit_mcp/server.py,sha256=R59DC7HVdpPcGHVXghPGUhrRKfn9Uqmy2MPkaYNfyzw,1320
 mle_kit_mcp/utils.py,sha256=iHNcEZZzPD37bEYE18SzJ3WUjLP3Ym-kc91SwcW1vlI,1984
 mle_kit_mcp/tools/__init__.py,sha256=r2fIg2mZ6zaeq0CzEKCEdeUTjV0pcA9NZaaOfBNVTnE,332
 mle_kit_mcp/tools/bash.py,sha256=kunYHc3dyPGOooT-KY9L7eI_N22lBrcDbTlcp_yTTws,2820
-mle_kit_mcp/tools/llm_proxy.py,sha256=RHW-PgxNnq2rs3sLJCA_CHNGt2ar1DYN2U0OlRjvDZ0,5234
+mle_kit_mcp/tools/llm_proxy.py,sha256=uEPZETqJWGOwLESWfvIjJidL6LSRry5j-qq6rpjtxLM,4623
 mle_kit_mcp/tools/remote_gpu.py,sha256=2Wjp6fYTGX9i6bZUhSWM5WMKh0eMc73wCdezes04bDg,12546
-mle_kit_mcp/tools/text_editor.py,sha256=0uGcSjcBjdPlqjYZYuWo29SmAMzBcWgKt4KRumFI3WE,9529
-mle_kit_mcp-0.2.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mle_kit_mcp-0.2.3.dist-info/METADATA,sha256=fzWaPwNQaj3zdsaoLyaU_ng4OajRllWXRD_qZ-Eu_84,1074
-mle_kit_mcp-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mle_kit_mcp-0.2.3.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
-mle_kit_mcp-0.2.3.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
-mle_kit_mcp-0.2.3.dist-info/RECORD,,
+mle_kit_mcp/tools/text_editor.py,sha256=hkobiyYB5um6bs5sWYDQ1S2Y5n31i7I6fOBUDFkNhmM,9531
+mle_kit_mcp-0.2.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mle_kit_mcp-0.2.5.dist-info/METADATA,sha256=pKWNkp9pLxtGAofpzamJA4BuTkzxwf0hqq8CNl1s0ts,1074
+mle_kit_mcp-0.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mle_kit_mcp-0.2.5.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
+mle_kit_mcp-0.2.5.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
+mle_kit_mcp-0.2.5.dist-info/RECORD,,