lean-lsp-mcp 0.17.1__py3-none-any.whl → 0.18.0__py3-none-any.whl

lean_lsp_mcp/models.py

@@ -134,11 +134,18 @@ class DeclarationInfo(BaseModel):
 
 
 class DiagnosticsResult(BaseModel):
-    """Wrapper for diagnostic messages list."""
+    """Wrapper for diagnostic messages list with build status."""
 
+    success: bool = Field(
+        True, description="True if the queried file/range has no errors"
+    )
     items: List[DiagnosticMessage] = Field(
         default_factory=list, description="List of diagnostic messages"
     )
+    failed_dependencies: List[str] = Field(
+        default_factory=list,
+        description="File paths of dependencies that failed to build",
+    )
 
 
 class CompletionsResult(BaseModel):

lean_lsp_mcp/outline_utils.py

@@ -1,11 +1,11 @@
 import re
 from typing import Dict, List, Optional, Tuple
+
 from leanclient import LeanLSPClient
 from leanclient.utils import DocumentContentChange
 
 from lean_lsp_mcp.models import FileOutline, OutlineEntry
 
-
 METHOD_KIND = {6, "method"}
 KIND_TAGS = {"namespace": "Ns"}
 
@@ -41,6 +41,10 @@ def _get_info_trees(
             for line in sorted(symbol_by_line.keys(), reverse=True)
         ],
     )
+
+    # Force file reload to reset diagnostics after the insert/revert cycle.
+    client.open_file(path, force_reopen=True)
+
     return info_trees
 
 

lean_lsp_mcp/server.py

@@ -64,17 +64,38 @@ from lean_lsp_mcp.utils import (
     OutputCapture,
     check_lsp_response,
     deprecated,
+    extract_failed_dependency_paths,
     extract_goals_list,
     extract_range,
     filter_diagnostics_by_position,
     find_start_position,
     get_declaration_range,
+    is_build_stderr,
 )
 
 # LSP Diagnostic severity: 1=error, 2=warning, 3=info, 4=hint
 DIAGNOSTIC_SEVERITY: Dict[int, str] = {1: "error", 2: "warning", 3: "info", 4: "hint"}
 
 
+async def _urlopen_json(req: urllib.request.Request, timeout: float):
+    """Run urllib.request.urlopen in a worker thread to avoid blocking the event loop."""
+
+    def _do_request():
+        with urllib.request.urlopen(req, timeout=timeout) as response:
+            return orjson.loads(response.read())
+
+    return await asyncio.to_thread(_do_request)
+
+
+async def _safe_report_progress(
+    ctx: Context, *, progress: int, total: int, message: str
+) -> None:
+    try:
+        await ctx.report_progress(progress=progress, total=total, message=message)
+    except Exception:
+        return
+
+
 _LOG_LEVEL = os.environ.get("LEAN_LOG_LEVEL", "INFO")
 configure_logging("CRITICAL" if _LOG_LEVEL == "NONE" else _LOG_LEVEL)
 logger = get_logger(__name__)
@@ -164,8 +185,7 @@ mcp = FastMCP(**mcp_kwargs)
 
 def rate_limited(category: str, max_requests: int, per_seconds: int):
     def decorator(func):
-        @functools.wraps(func)
-        def wrapper(*args, **kwargs):
+        def _apply_rate_limit(args, kwargs):
             ctx = kwargs.get("ctx")
             if ctx is None:
                 if not args:
@@ -181,11 +201,33 @@ def rate_limited(category: str, max_requests: int, per_seconds: int):
                 if timestamp > current_time - per_seconds
             ]
             if len(rate_limit[category]) >= max_requests:
-                return f"Tool limit exceeded: {max_requests} requests per {per_seconds} s. Try again later."
+                return (
+                    False,
+                    f"Tool limit exceeded: {max_requests} requests per {per_seconds} s. Try again later.",
+                )
             rate_limit[category].append(current_time)
-            return func(*args, **kwargs)
+            return True, None
+
+        if asyncio.iscoroutinefunction(func):
+
+            @functools.wraps(func)
+            async def wrapper(*args, **kwargs):
+                allowed, msg = _apply_rate_limit(args, kwargs)
+                if not allowed:
+                    return msg
+                return await func(*args, **kwargs)
+
+        else:
 
-        wrapper.__doc__ = f"Limit: {max_requests}req/{per_seconds}s. " + wrapper.__doc__
+            @functools.wraps(func)
+            def wrapper(*args, **kwargs):
+                allowed, msg = _apply_rate_limit(args, kwargs)
+                if not allowed:
+                    return msg
+                return func(*args, **kwargs)
+
+        doc = wrapper.__doc__ or ""
+        wrapper.__doc__ = f"Limit: {max_requests}req/{per_seconds}s. {doc}"
         return wrapper
 
     return decorator
@@ -375,6 +417,7 @@ def file_outline(
 
 
 def _to_diagnostic_messages(diagnostics: List[Dict]) -> List[DiagnosticMessage]:
+    """Convert LSP diagnostics to DiagnosticMessage models."""
     result = []
     for diag in diagnostics:
         r = diag.get("fullRange", diag.get("range"))
@@ -394,6 +437,52 @@ def _to_diagnostic_messages(diagnostics: List[Dict]) -> List[DiagnosticMessage]:
     return result
 
 
+def _process_diagnostics(
+    diagnostics: List[Dict], build_success: bool
+) -> DiagnosticsResult:
+    """Process diagnostics, extracting dependency paths from build stderr.
+
+    Args:
+        diagnostics: List of diagnostic dicts from leanclient
+        build_success: Whether the build succeeded (from leanclient.DiagnosticsResult.success)
+    """
+    items = []
+    failed_deps: List[str] = []
+
+    for diag in diagnostics:
+        r = diag.get("fullRange", diag.get("range"))
+        if r is None:
+            continue
+
+        severity_int = diag.get("severity", 1)
+        message = diag.get("message", "")
+        line = r["start"]["line"] + 1
+        column = r["start"]["character"] + 1
+
+        # Check if this is a build failure at (1,1) - extract dependency paths, skip the item
+        if line == 1 and column == 1 and is_build_stderr(message):
+            failed_deps = extract_failed_dependency_paths(message)
+            continue  # Don't include the build stderr blob as a diagnostic item
+
+        # Normal diagnostic from the queried file
+        items.append(
+            DiagnosticMessage(
+                severity=DIAGNOSTIC_SEVERITY.get(
+                    severity_int, f"unknown({severity_int})"
+                ),
+                message=message,
+                line=line,
+                column=column,
+            )
+        )
+
+    return DiagnosticsResult(
+        success=build_success,
+        items=items,
+        failed_dependencies=failed_deps,
+    )
+
+
 @mcp.tool(
     "lean_diagnostic_messages",
     annotations=ToolAnnotations(
@@ -437,15 +526,14 @@ def diagnostic_messages(
     start_line_0 = (start_line - 1) if start_line is not None else None
     end_line_0 = (end_line - 1) if end_line is not None else None
 
-    diagnostics = client.get_diagnostics(
+    result = client.get_diagnostics(
         rel_path,
         start_line=start_line_0,
         end_line=end_line_0,
         inactivity_timeout=15.0,
     )
-    check_lsp_response(diagnostics, "get_diagnostics")
 
-    return DiagnosticsResult(items=_to_diagnostic_messages(diagnostics))
+    return _process_diagnostics(result.diagnostics, result.success)
 
 
 @mcp.tool(
@@ -880,7 +968,7 @@ class LocalSearchError(Exception):
         openWorldHint=False,
     ),
 )
-def local_search(
+async def local_search(
     ctx: Context,
     query: Annotated[str, Field(description="Declaration name or prefix")],
     limit: Annotated[int, Field(description="Max matches", ge=1)] = 10,
@@ -912,8 +1000,11 @@ def local_search(
         )
 
     try:
-        raw_results = lean_local_search(
-            query=query.strip(), limit=limit, project_root=resolved_root
+        raw_results = await asyncio.to_thread(
+            lean_local_search,
+            query=query.strip(),
+            limit=limit,
+            project_root=resolved_root,
         )
         results = [
             LocalSearchResult(name=r["name"], kind=r["kind"], file=r["file"])
@@ -934,7 +1025,7 @@ def local_search(
     ),
 )
 @rate_limited("leansearch", max_requests=3, per_seconds=30)
-def leansearch(
+async def leansearch(
     ctx: Context,
     query: Annotated[str, Field(description="Natural language or Lean term query")],
     num_results: Annotated[int, Field(description="Max results", ge=1)] = 5,
@@ -954,8 +1045,10 @@ def leansearch(
         method="POST",
     )
 
-    with urllib.request.urlopen(req, timeout=10) as response:
-        results = orjson.loads(response.read())
+    await _safe_report_progress(
+        ctx, progress=1, total=10, message="Awaiting response from leansearch.net"
+    )
+    results = await _urlopen_json(req, timeout=10)
 
     if not results or not results[0]:
         return LeanSearchResults(items=[])
@@ -1029,7 +1122,13 @@ async def loogle(
         )
     rate_limit.append(now)
 
-    result = loogle_remote(query, num_results)
+    await _safe_report_progress(
+        ctx,
+        progress=1,
+        total=10,
+        message="Awaiting response from loogle.lean-lang.org",
+    )
+    result = await asyncio.to_thread(loogle_remote, query, num_results)
     if isinstance(result, str):
         raise LeanToolError(result)  # Error message from remote
     return LoogleResults(items=result)
@@ -1045,7 +1144,7 @@ async def loogle(
     ),
 )
 @rate_limited("leanfinder", max_requests=10, per_seconds=30)
-def leanfinder(
+async def leanfinder(
     ctx: Context,
     query: Annotated[str, Field(description="Mathematical concept or proof state")],
     num_results: Annotated[int, Field(description="Max results", ge=1)] = 5,
@@ -1063,23 +1162,27 @@ def leanfinder(
     )
 
     results: List[LeanFinderResult] = []
-    with urllib.request.urlopen(req, timeout=10) as response:
-        data = orjson.loads(response.read())
-        for result in data["results"]:
-            if (
-                "https://leanprover-community.github.io/mathlib4_docs"
-                not in result["url"]
-            ):  # Only include mathlib4 results
-                continue
-            match = re.search(r"pattern=(.*?)#doc", result["url"])
-            if match:
-                results.append(
-                    LeanFinderResult(
-                        full_name=match.group(1),
-                        formal_statement=result["formal_statement"],
-                        informal_statement=result["informal_statement"],
-                    )
+    await _safe_report_progress(
+        ctx,
+        progress=1,
+        total=10,
+        message="Awaiting response from Lean Finder (Hugging Face)",
+    )
+    data = await _urlopen_json(req, timeout=10)
+    for result in data["results"]:
+        if (
+            "https://leanprover-community.github.io/mathlib4_docs" not in result["url"]
+        ):  # Only include mathlib4 results
+            continue
+        match = re.search(r"pattern=(.*?)#doc", result["url"])
+        if match:
+            results.append(
+                LeanFinderResult(
+                    full_name=match.group(1),
+                    formal_statement=result["formal_statement"],
+                    informal_statement=result["informal_statement"],
                 )
+            )
 
     return LeanFinderResults(items=results)
 
@@ -1094,7 +1197,7 @@ def leanfinder(
     ),
 )
 @rate_limited("lean_state_search", max_requests=3, per_seconds=30)
-def state_search(
+async def state_search(
     ctx: Context,
     file_path: Annotated[str, Field(description="Absolute path to Lean file")],
     line: Annotated[int, Field(description="Line number (1-indexed)", ge=1)],
@@ -1126,8 +1229,10 @@ def state_search(
         method="GET",
     )
 
-    with urllib.request.urlopen(req, timeout=10) as response:
-        results = orjson.loads(response.read())
+    await _safe_report_progress(
+        ctx, progress=1, total=10, message=f"Awaiting response from {url}"
+    )
+    results = await _urlopen_json(req, timeout=10)
 
     items = [StateSearchResult(name=r["name"]) for r in results]
     return StateSearchResults(items=items)
@@ -1143,7 +1248,7 @@ def state_search(
     ),
 )
 @rate_limited("hammer_premise", max_requests=3, per_seconds=30)
-def hammer_premise(
+async def hammer_premise(
     ctx: Context,
     file_path: Annotated[str, Field(description="Absolute path to Lean file")],
     line: Annotated[int, Field(description="Line number (1-indexed)", ge=1)],
@@ -1186,8 +1291,10 @@ def hammer_premise(
         data=orjson.dumps(data),
     )
 
-    with urllib.request.urlopen(req, timeout=10) as response:
-        results = orjson.loads(response.read())
+    await _safe_report_progress(
+        ctx, progress=1, total=10, message=f"Awaiting response from {url}"
+    )
+    results = await _urlopen_json(req, timeout=10)
 
     items = [PremiseResult(name=r["name"]) for r in results]
     return PremiseResults(items=items)

lean_lsp_mcp/utils.py

@@ -1,4 +1,5 @@
 import os
+import re
 import secrets
 import sys
 import tempfile
@@ -7,6 +8,30 @@ from typing import Any, List, Dict, Optional, Callable
 from mcp.server.auth.provider import AccessToken, TokenVerifier
 
 
+# Pattern to extract file paths from build stderr: "error: path/file.lean:line:col: message"
+BUILD_ERROR_FILE_PATTERN = re.compile(
+    r"^(?:error|warning):\s*([^\s:]+\.lean):\d+:\d+:",
+    re.MULTILINE | re.IGNORECASE,
+)
+
+
+def extract_failed_dependency_paths(message: str) -> List[str]:
+    """Extract unique file paths from lake build stderr output.
+
+    Returns sorted list of .lean file paths that had errors/warnings.
+    """
+    paths = set(BUILD_ERROR_FILE_PATTERN.findall(message))
+    return sorted(paths)
+
+
+def is_build_stderr(message: str) -> bool:
+    """Check if message looks like lake build stderr output."""
+    return (
+        "lake setup-file" in message
+        or BUILD_ERROR_FILE_PATTERN.search(message) is not None
+    )
+
+
 class LeanToolError(Exception):
     """Exception raised when a Lean MCP tool operation fails."""
 

{lean_lsp_mcp-0.17.1.dist-info → lean_lsp_mcp-0.18.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lean-lsp-mcp
-Version: 0.17.1
+Version: 0.18.0
 Summary: Lean Theorem Prover MCP
 Author-email: Oliver Dressler <hey@oli.show>
 License-Expression: MIT
@@ -8,7 +8,7 @@ Project-URL: Repository, https://github.com/oOo0oOo/lean-lsp-mcp
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: leanclient==0.8.0
+Requires-Dist: leanclient==0.9.0
 Requires-Dist: mcp[cli]==1.25.0
 Requires-Dist: orjson>=3.11.1
 Provides-Extra: lint
@@ -18,6 +18,7 @@ Requires-Dist: ruff>=0.2.0; extra == "dev"
 Requires-Dist: pytest>=8.3; extra == "dev"
 Requires-Dist: anyio>=4.4; extra == "dev"
 Requires-Dist: pytest-asyncio>=0.23; extra == "dev"
+Requires-Dist: pytest-timeout>=2.3; extra == "dev"
 Dynamic: license-file
 
 <h1 align="center">

lean_lsp_mcp-0.18.0.dist-info/RECORD

@@ -0,0 +1,17 @@
+lean_lsp_mcp/__init__.py,sha256=MN_bNFyb5-p33JWWGbrlUYBd1UUMQKtZYGC9KCh2mtM,1403
+lean_lsp_mcp/__main__.py,sha256=XnpTzfJc0T-j9tHtdkA8ovTr1c139ffTewcJGhxYDaM,49
+lean_lsp_mcp/client_utils.py,sha256=HgPuB35rMitn2Xm8SCAErsFLq15trB6VMz3FDFgmPd8,4897
+lean_lsp_mcp/file_utils.py,sha256=kCTYQSfmV-R2cm_NCi_L8W5Dcsm0_rTOPpTtpyAin78,1365
+lean_lsp_mcp/instructions.py,sha256=iJk_oD67tqNaC8K5OXEuXafULKSbbiHjZAsSRebOwdw,1904
+lean_lsp_mcp/loogle.py,sha256=zUgnDWoTIqa4G6GXStAIxxJUR545YbU8Z-8KMjddKV0,15500
+lean_lsp_mcp/models.py,sha256=WVW5K27m54W2THoAUMKvBC--qCSD6eSYjd9bcaiMk2s,6879
+lean_lsp_mcp/outline_utils.py,sha256=i7xL27UO2rTT48IdKXkoMq5FVJNxyA3tPuQREOBf_gU,11105
+lean_lsp_mcp/search_utils.py,sha256=MLqKGe4bhEvyfFLIBCmiDxkbcH4O5J3vl9mWnRSb_v0,6801
+lean_lsp_mcp/server.py,sha256=UDTuHeRX_RUxsfLrR3PqGWnAYLvmEb89iZtq0dNqymU,43859
+lean_lsp_mcp/utils.py,sha256=dGv84a4E-szOkQVYtSE-q9GbawpiVk47qvrkTN-Clts,13478
+lean_lsp_mcp-0.18.0.dist-info/licenses/LICENSE,sha256=CQlxnf0tQyoVrBE93JYvAUYxv6Z5Yg6sX0pwogOkFvo,1071
+lean_lsp_mcp-0.18.0.dist-info/METADATA,sha256=xXkTL1pV9lCCopASzat8OYGppv_GRT95GTVdMVccAAw,20838
+lean_lsp_mcp-0.18.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lean_lsp_mcp-0.18.0.dist-info/entry_points.txt,sha256=nQbvwctWkWD7I-2f4VrdVQBZYGUw8CnUnFC6QjXxOSE,51
+lean_lsp_mcp-0.18.0.dist-info/top_level.txt,sha256=LGEK0lgMSNPIQ6mG8EO-adaZEGPi_0daDs004epOTF0,13
+lean_lsp_mcp-0.18.0.dist-info/RECORD,,

lean_lsp_mcp-0.17.1.dist-info/RECORD

@@ -1,17 +0,0 @@
-lean_lsp_mcp/__init__.py,sha256=MN_bNFyb5-p33JWWGbrlUYBd1UUMQKtZYGC9KCh2mtM,1403
-lean_lsp_mcp/__main__.py,sha256=XnpTzfJc0T-j9tHtdkA8ovTr1c139ffTewcJGhxYDaM,49
-lean_lsp_mcp/client_utils.py,sha256=HgPuB35rMitn2Xm8SCAErsFLq15trB6VMz3FDFgmPd8,4897
-lean_lsp_mcp/file_utils.py,sha256=kCTYQSfmV-R2cm_NCi_L8W5Dcsm0_rTOPpTtpyAin78,1365
-lean_lsp_mcp/instructions.py,sha256=iJk_oD67tqNaC8K5OXEuXafULKSbbiHjZAsSRebOwdw,1904
-lean_lsp_mcp/loogle.py,sha256=zUgnDWoTIqa4G6GXStAIxxJUR545YbU8Z-8KMjddKV0,15500
-lean_lsp_mcp/models.py,sha256=2pLmvNsrMdn4vO1k119Jw8gqYeaeGKewWW0q1TabBCY,6604
-lean_lsp_mcp/outline_utils.py,sha256=-eoZNbx2eaKaYmuyFJnwUMWP8I9YXNWusue_2OYpDBM,10981
-lean_lsp_mcp/search_utils.py,sha256=MLqKGe4bhEvyfFLIBCmiDxkbcH4O5J3vl9mWnRSb_v0,6801
-lean_lsp_mcp/server.py,sha256=1G7-FZz1TNJ7uU36eSTm9yponWyMO3QhQiyOhhTNOH8,40626
-lean_lsp_mcp/utils.py,sha256=MmGgdhrLEvCtRRVNgN_vflO9_A25h76QbJXhBD-OKt0,12721
-lean_lsp_mcp-0.17.1.dist-info/licenses/LICENSE,sha256=CQlxnf0tQyoVrBE93JYvAUYxv6Z5Yg6sX0pwogOkFvo,1071
-lean_lsp_mcp-0.17.1.dist-info/METADATA,sha256=K0Ygues0l2TNEY6nXV4uyEds2QZfMV2DQ0kwId1uBHQ,20787
-lean_lsp_mcp-0.17.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lean_lsp_mcp-0.17.1.dist-info/entry_points.txt,sha256=nQbvwctWkWD7I-2f4VrdVQBZYGUw8CnUnFC6QjXxOSE,51
-lean_lsp_mcp-0.17.1.dist-info/top_level.txt,sha256=LGEK0lgMSNPIQ6mG8EO-adaZEGPi_0daDs004epOTF0,13
-lean_lsp_mcp-0.17.1.dist-info/RECORD,,