mle-kit-mcp 1.0.2__py3-none-any.whl → 1.1.0__py3-none-any.whl

mle_kit_mcp/tools/__init__.py

@@ -8,6 +8,7 @@ from .llm_proxy import (
     llm_proxy_local,
     llm_proxy_remote,
 )
+from .file_system import glob, grep
 
 
 __all__ = [
@@ -17,4 +18,6 @@ __all__ = [
     "remote_download",
     "llm_proxy_local",
     "llm_proxy_remote",
+    "glob",
+    "grep",
 ]

mle_kit_mcp/tools/bash.py

@@ -1,6 +1,7 @@
 import atexit
 import signal
 import shlex
+import os
 from typing import Optional, Any
 
 from docker import from_env as docker_from_env  # type: ignore
@@ -26,6 +27,8 @@ def get_docker_client() -> DockerClient:
 
 def create_container() -> Container:
     client = get_docker_client()
+    uid = os.getuid()
+    gid = os.getgid()
     container = client.containers.run(
         BASE_IMAGE,
         "tail -f /dev/null",
@@ -33,6 +36,11 @@ def create_container() -> Container:
         remove=True,
         tty=True,
         stdin_open=True,
+        user=f"{uid}:{gid}",
+        environment={
+            "HOME": DOCKER_WORKSPACE_DIR_PATH,
+            "XDG_CACHE_HOME": f"{DOCKER_WORKSPACE_DIR_PATH}/.cache",
+        },
         volumes={
             get_workspace_dir(): {
                 "bind": DOCKER_WORKSPACE_DIR_PATH,

mle_kit_mcp/tools/file_system.py

@@ -0,0 +1,155 @@
+from typing import List, Optional
+from pathlib import Path
+import subprocess
+
+from mle_kit_mcp.files import get_workspace_dir
+
+
+def glob(pattern: str, path: Optional[str] = None) -> List[str]:
+    """
+    - Fast file pattern matching tool that works with any codebase size
+    - Supports glob patterns like "**/*.js" or "src/**/*.ts"
+    - Returns matching file paths sorted by modification time
+    - Use this tool when you need to find files by name patterns
+
+    Args:
+        pattern: The glob pattern to match files against, required.
+        path: The directory to search in. If not specified, the current working directory will be used.
+            IMPORTANT: Omit this field to use the default directory. DO NOT enter "undefined" or "null".
+            Simply omit it for the default behavior.
+            Must be a valid directory path if provided.
+
+    Returns:
+        A list of matching file paths.
+    """
+    full_path: Path = get_workspace_dir()
+    if path is not None:
+        full_path = get_workspace_dir() / path
+    files = [p for p in full_path.glob(pattern)]
+    files.sort(key=lambda p: p.stat().st_mtime, reverse=True)
+    resolved_files = [str(p.resolve()) for p in files]
+    resolved_files = [f.replace(str(get_workspace_dir()) + "/", "") for f in resolved_files]
+    return resolved_files
+
+
+def grep(
+    pattern: str,
+    path: Optional[str] = None,
+    glob: Optional[str] = None,
+    output_mode: str = "files_with_matches",
+    before_context: Optional[int] = None,
+    after_context: Optional[int] = None,
+    center_context: Optional[int] = None,
+    insensitive: bool = False,
+    type: Optional[str] = None,
+    head_limit: Optional[int] = None,
+    multiline: bool = False,
+) -> str:
+    r"""
+    A powerful search tool built on ripgrep
+
+    Usage:
+    - ALWAYS use Grep for search tasks. NEVER invoke `grep` or `rg` as a Bash command.
+    - The Grep tool has been optimized for correct permissions and access.
+    - Supports full regex syntax (e.g., "log.*Error")
+    - Filter files with glob parameter (e.g., "*.js", "**/*.tsx") or type parameter (e.g., "js", "py", "rust")
+    - Output modes: "content" shows matching lines, "files_with_matches" shows only file paths (default), "count" shows match counts
+    - Pattern syntax: Uses ripgrep (not grep) - literal braces need escaping (use `interface\{\}` to find `interface{}` in Go code)
+    - Multiline matching: By default patterns match within single lines only. For cross-line patterns like `struct \{[\s\S]*?field`, use `multiline: true`
+
+    Args:
+        pattern: The regular expression pattern to search for in file contents
+        path: File or directory to search. Defaults to current working directory.
+        glob: Glob pattern to filter files (e.g. "*.js", "*.{ts,tsx}")
+        output_mode: The output mode to use. Possible values are "content", "files_with_matches", "count".
+            "content" shows matching lines (supports -A/-B/-C context, -n line numbers, head_limit)
+            "files_with_matches" shows file paths (supports head_limit)
+            "count" shows match counts (supports head_limit).
+            Defaults to "files_with_matches"
+        before_context: Number of lines to show before each match (rg -B). Requires output_mode: "content", ignored otherwise.
+        after_context: Number of lines to show after each match (rg -A). Requires output_mode: "content", ignored otherwise.
+        center_context: Number of lines to show before and after each match (rg -C). Requires output_mode: "content", ignored otherwise.
+        insensitive: Case insensitive search (rg -i). Defaults to False.
+        type: File type to search (rg --type). Common types: js, py, rust, go, java, etc. More efficient than include for standard file types.
+        head_limit: Limit output to first N lines/entries, equivalent to "| head -N". Works across all output modes: content (limits output lines), files_with_matches (limits file paths), count (limits count entries). When unspecified, shows all results from ripgrep.
+        multiline: Enable multiline mode where . matches newlines and patterns can span lines (rg -U --multiline-dotall). Default: false.
+
+    Returns:
+        A string with the search results.
+    """
+    assert pattern, "'pattern' must be a non-empty string"
+
+    valid_output_modes = {"content", "files_with_matches", "count"}
+    assert (
+        output_mode in valid_output_modes
+    ), f"Invalid output_mode: {output_mode}. Expected one of {sorted(valid_output_modes)}"
+
+    if center_context is not None:
+        assert (
+            before_context is None and after_context is None
+        ), "Use either 'center_context' or ('before_context'/'after_context'), not both"
+
+    cmd: List[str] = [
+        "rg",
+        "--color=never",
+        "--no-heading",
+    ]
+
+    if output_mode == "files_with_matches":
+        cmd.append("-l")
+    elif output_mode == "count":
+        cmd.append("-c")
+    else:
+        cmd.append("-n")
+
+    if insensitive:
+        cmd.append("-i")
+
+    if multiline:
+        cmd.extend(["--multiline", "--multiline-dotall"])
+
+    if type:
+        cmd.extend(["--type", type])
+
+    if glob:
+        cmd.extend(["--glob", glob])
+
+    if output_mode == "content":
+        if center_context is not None:
+            cmd.extend(["-C", str(center_context)])
+        else:
+            if before_context is not None:
+                cmd.extend(["-B", str(before_context)])
+            if after_context is not None:
+                cmd.extend(["-A", str(after_context)])
+
+    cmd.append("--")
+    cmd.append(pattern)
+
+    search_root: Path = get_workspace_dir()
+    if path is not None:
+        search_root = search_root / path
+    cmd.append(str(search_root))
+
+    try:
+        result = subprocess.run(
+            cmd,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+            check=False,
+        )
+    except FileNotFoundError:
+        return "ripgrep (rg) not found. Please install ripgrep in the environment where this tool runs."
+
+    stdout = (result.stdout or "").rstrip("\n")
+    stderr = (result.stderr or "").strip()
+
+    if head_limit is not None and head_limit >= 0 and stdout:
+        stdout_lines = stdout.splitlines()
+        stdout = "\n".join(stdout_lines[:head_limit])
+
+    if result.returncode not in (0, 1) and stderr:
+        return stderr
+
+    return stdout

{mle_kit_mcp-1.0.2.dist-info → mle_kit_mcp-1.1.0.dist-info}/METADATA

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

@@ -5,14 +5,15 @@ 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=6KGjhcjjqP9PxKw17X7ZejeRrGfCB8Mk4SmnoTklCMs,1587
 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=IDzR3dkGOKGuwEE0FooQMI-Ztx99t8YJKoMu4FjxNGw,3922
+mle_kit_mcp/tools/__init__.py,sha256=o5Cy8gk67QiqHd40z0LF-nXrezSq4reuvhkWiMlCLpc,392
+mle_kit_mcp/tools/bash.py,sha256=blL3SWmrqnVdSsWvLu3H3WWjBGdVR1EwylIPLtAJ8co,4154
+mle_kit_mcp/tools/file_system.py,sha256=TvgNNma8fdFDtYdBvdnHAAgUlbpecyQ1TSJYklQOlI4,6391
 mle_kit_mcp/tools/llm_proxy.py,sha256=uEPZETqJWGOwLESWfvIjJidL6LSRry5j-qq6rpjtxLM,4623
 mle_kit_mcp/tools/remote_gpu.py,sha256=fqDWiGnCR6h34LSInw_vMqdU-AfL8Z12969kQ4-cShI,12616
 mle_kit_mcp/tools/text_editor.py,sha256=hkobiyYB5um6bs5sWYDQ1S2Y5n31i7I6fOBUDFkNhmM,9531
-mle_kit_mcp-1.0.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mle_kit_mcp-1.0.2.dist-info/METADATA,sha256=eYWcC94DpWIT7VtD7gk7qw_AKAYYyFsvViWcNCCsUL8,1074
-mle_kit_mcp-1.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mle_kit_mcp-1.0.2.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
-mle_kit_mcp-1.0.2.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
-mle_kit_mcp-1.0.2.dist-info/RECORD,,
+mle_kit_mcp-1.1.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mle_kit_mcp-1.1.0.dist-info/METADATA,sha256=We9HrrSEG96vC82qIigDPXazpZBktLA24ySU1PMuJd4,1074
+mle_kit_mcp-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mle_kit_mcp-1.1.0.dist-info/entry_points.txt,sha256=-iHSUVPN49jkBj1ySpc-P0rVF5-IPHw-KWNayNIiEsk,49
+mle_kit_mcp-1.1.0.dist-info/top_level.txt,sha256=XeBtCq_CnVI0gh0Z_daZOLmGl5XPlkA8RgHaj5s5VQY,12
+mle_kit_mcp-1.1.0.dist-info/RECORD,,