lean-lsp-mcp 0.12.1__py3-none-any.whl → 0.13.1__py3-none-any.whl

lean_lsp_mcp/instructions.py

@@ -1,5 +1,5 @@
 INSTRUCTIONS = """## General Rules
-- All line and column numbers are 1-indexed (use lean_file_contents if unsure).
+- All line and column numbers are 1-indexed.
 - Always analyze/search context before each file edit.
 - This MCP does NOT make permanent file changes. Use other tools for editing.
 - Work iteratively: Small steps, intermediate sorries, frequent checks.
@@ -7,7 +7,7 @@ INSTRUCTIONS = """## General Rules
 ## Key Tools
 - lean_local_search: Confirm declarations (theorems/lemmas/defs/etc.) exist. VERY USEFUL AND FAST!
 - lean_goal: Check proof state. USE OFTEN!
-- lean_diagnostic_messages: Understand the current proof situation.
+- lean_diagnostic_messages: Understand current proof situation.
 - lean_hover_info: Documentation about terms and lean syntax.
 - lean_leansearch: Search theorems using natural language or Lean terms.
 - lean_loogle: Search definitions and theorems by name, type, or subexpression.

lean_lsp_mcp/server.py

@@ -34,6 +34,7 @@ from lean_lsp_mcp.utils import (
     format_diagnostics,
     format_goal,
     format_line,
+    get_declaration_range,
     OptionalTokenVerifier,
 )
 
@@ -238,6 +239,8 @@ async def lsp_build(
 def file_contents(ctx: Context, file_path: str, annotate_lines: bool = True) -> str:
     """Get the text contents of a Lean file, optionally with line numbers.
 
+    Use sparingly (bloats context). Mainly when unsure about line numbers.
+
     Args:
         file_path (str): Abs path to Lean file
         annotate_lines (bool, optional): Annotate lines with line numbers. Defaults to True.
@@ -273,6 +276,7 @@ def diagnostic_messages(
     file_path: str,
     start_line: Optional[int] = None,
     end_line: Optional[int] = None,
+    declaration_name: Optional[str] = None,
 ) -> List[str] | str:
     """Get all diagnostic msgs (errors, warnings, infos) for a Lean file.
 
@@ -282,6 +286,10 @@ def diagnostic_messages(
         file_path (str): Abs path to Lean file
         start_line (int, optional): Start line (1-indexed). Filters from this line.
         end_line (int, optional): End line (1-indexed). Filters to this line.
+        declaration_name (str, optional): Name of a specific theorem/lemma/definition.
+            If provided, only returns diagnostics within that declaration.
+            Takes precedence over start_line/end_line.
+            Slow, requires waiting for full file analysis.
 
     Returns:
         List[str] | str: Diagnostic msgs or error msg
@@ -292,6 +300,13 @@ def diagnostic_messages(
 
     client: LeanLSPClient = ctx.request_context.lifespan_context.client
 
+    # If declaration_name is provided, get its range and use that for filtering
+    if declaration_name:
+        decl_range = get_declaration_range(client, rel_path, declaration_name)
+        if decl_range is None:
+            return f"Declaration '{declaration_name}' not found in file. Check the name (case-sensitive) and try again."
+        start_line, end_line = decl_range
+
     # Convert 1-indexed to 0-indexed for leanclient
     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
@@ -300,7 +315,7 @@ def diagnostic_messages(
         rel_path,
         start_line=start_line_0,
         end_line=end_line_0,
-        inactivity_timeout=8.0,
+        inactivity_timeout=15.0,
     )
 
     return format_diagnostics(diagnostics)
@@ -658,7 +673,7 @@ def run_code(ctx: Context, code: str) -> List[str] | str:
         client.open_file(rel_path)
         opened_file = True
         diagnostics = format_diagnostics(
-            client.get_diagnostics(rel_path, inactivity_timeout=8.0)
+            client.get_diagnostics(rel_path, inactivity_timeout=15.0)
         )
     finally:
         if opened_file:

lean_lsp_mcp/utils.py

@@ -259,3 +259,78 @@ def filter_diagnostics_by_position(
         matches.append(diagnostic)
 
     return matches
+
+
+def search_symbols(symbols: List[Dict], target_name: str) -> Dict | None:
+    """Recursively search through symbols and their children.
+
+    Args:
+        symbols: List of LSP document symbols
+        target_name: Name of the symbol to find (case-sensitive)
+
+    Returns:
+        The matching symbol dict, or None if not found
+    """
+    for symbol in symbols:
+        if symbol.get("name") == target_name:
+            return symbol
+        # Search nested declarations (children)
+        children = symbol.get("children", [])
+        if children:
+            result = search_symbols(children, target_name)
+            if result:
+                return result
+    return None
+
+
+def get_declaration_range(
+    client, file_path: str, declaration_name: str
+) -> tuple[int, int] | None:
+    """Get the line range (1-indexed) of a declaration by name using LSP document symbols.
+
+    Args:
+        client: The Lean LSP client instance (LeanLSPClient)
+        file_path: Relative path to the Lean file
+        declaration_name: Name of the declaration to find (case-sensitive)
+
+    Returns:
+        Tuple of (start_line, end_line) as 1-indexed integers, or None if not found
+    """
+    from lean_lsp_mcp.server import logger
+
+    try:
+        # Ensure file is opened (LSP needs this to analyze the file)
+        client.open_file(file_path)
+
+        # Get document symbols from LSP
+        symbols = client.get_document_symbols(file_path)
+
+        if not symbols:
+            logger.debug(
+                "No document symbols returned for '%s' - file may not be processed yet",
+                file_path,
+            )
+            return None
+
+        matching_symbol = search_symbols(symbols, declaration_name)
+        if not matching_symbol:
+            return None
+
+        # Extract range - LSP returns 0-indexed, convert to 1-indexed
+        range_info = matching_symbol.get("range")
+        if not range_info:
+            return None
+
+        start_line = range_info["start"]["line"] + 1
+        end_line = range_info["end"]["line"] + 1
+
+        return (start_line, end_line)
+
+    except Exception as e:
+        logger.warning(
+            "Failed to get declaration range for '%s' in '%s': %s",
+            declaration_name,
+            file_path,
+            e,
+        )
+        return None

{lean_lsp_mcp-0.12.1.dist-info → lean_lsp_mcp-0.13.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lean-lsp-mcp
-Version: 0.12.1
+Version: 0.13.1
 Summary: Lean Theorem Prover MCP
 Author-email: Oliver Dressler <hey@oli.show>
 License-Expression: MIT
@@ -8,8 +8,8 @@ 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.5.1
-Requires-Dist: mcp[cli]==1.21.0
+Requires-Dist: leanclient==0.5.3
+Requires-Dist: mcp[cli]==1.21.1
 Requires-Dist: orjson>=3.11.1
 Provides-Extra: lint
 Requires-Dist: ruff>=0.2.0; extra == "lint"

lean_lsp_mcp-0.13.1.dist-info/RECORD

@@ -0,0 +1,14 @@
+lean_lsp_mcp/__init__.py,sha256=lxqDq0G_sI2iu2Nniy-pTW7BE9Ux7ZXeDoGf0OAWIDc,763
+lean_lsp_mcp/__main__.py,sha256=XnpTzfJc0T-j9tHtdkA8ovTr1c139ffTewcJGhxYDaM,49
+lean_lsp_mcp/client_utils.py,sha256=hF941DEeRE3ICMgMhv9J4vv6bO4hZPJOAcFU03yIDXs,4988
+lean_lsp_mcp/file_utils.py,sha256=kCTYQSfmV-R2cm_NCi_L8W5Dcsm0_rTOPpTtpyAin78,1365
+lean_lsp_mcp/instructions.py,sha256=1Xgh_fkdRpz-lqdl6kpETdwEu-IGRQGIdSdU7o9t5hc,853
+lean_lsp_mcp/search_utils.py,sha256=X2LPynDNLi767UDxbxHpMccOkbnfKJKv_HxvRNxIXM4,3984
+lean_lsp_mcp/server.py,sha256=g_CcgMl4aOE0MqcNv5O0DvSR7mN_fSdmJQAddDSclfA,36261
+lean_lsp_mcp/utils.py,sha256=YE6o6eswOi47AYojITQ2RcR-DspqKtgACeV-O7xgOKM,10554
+lean_lsp_mcp-0.13.1.dist-info/licenses/LICENSE,sha256=CQlxnf0tQyoVrBE93JYvAUYxv6Z5Yg6sX0pwogOkFvo,1071
+lean_lsp_mcp-0.13.1.dist-info/METADATA,sha256=Y7CElIvCA6scvnk2PJyMANjW1dyg9VFk0-SBwpk4wU8,19626
+lean_lsp_mcp-0.13.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lean_lsp_mcp-0.13.1.dist-info/entry_points.txt,sha256=nQbvwctWkWD7I-2f4VrdVQBZYGUw8CnUnFC6QjXxOSE,51
+lean_lsp_mcp-0.13.1.dist-info/top_level.txt,sha256=LGEK0lgMSNPIQ6mG8EO-adaZEGPi_0daDs004epOTF0,13
+lean_lsp_mcp-0.13.1.dist-info/RECORD,,

lean_lsp_mcp-0.12.1.dist-info/RECORD

@@ -1,14 +0,0 @@
-lean_lsp_mcp/__init__.py,sha256=lxqDq0G_sI2iu2Nniy-pTW7BE9Ux7ZXeDoGf0OAWIDc,763
-lean_lsp_mcp/__main__.py,sha256=XnpTzfJc0T-j9tHtdkA8ovTr1c139ffTewcJGhxYDaM,49
-lean_lsp_mcp/client_utils.py,sha256=hF941DEeRE3ICMgMhv9J4vv6bO4hZPJOAcFU03yIDXs,4988
-lean_lsp_mcp/file_utils.py,sha256=kCTYQSfmV-R2cm_NCi_L8W5Dcsm0_rTOPpTtpyAin78,1365
-lean_lsp_mcp/instructions.py,sha256=y_gHlbeJoKnPohmcSVrQQds6mbBO1en-lxnXAfEypZE,892
-lean_lsp_mcp/search_utils.py,sha256=X2LPynDNLi767UDxbxHpMccOkbnfKJKv_HxvRNxIXM4,3984
-lean_lsp_mcp/server.py,sha256=AiPu6yw_m1wQdooXyhgKUGuL-Kmin4GNoz79iZNnTps,35457
-lean_lsp_mcp/utils.py,sha256=zLu2VIhaX4yocY07F3Z94LB2jRGrkH1ID9SjR3poE9A,8255
-lean_lsp_mcp-0.12.1.dist-info/licenses/LICENSE,sha256=CQlxnf0tQyoVrBE93JYvAUYxv6Z5Yg6sX0pwogOkFvo,1071
-lean_lsp_mcp-0.12.1.dist-info/METADATA,sha256=fwr43eEL0JDzu2wv9JJPoInsJyRL1dlGcz8RUhpPXs8,19626
-lean_lsp_mcp-0.12.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lean_lsp_mcp-0.12.1.dist-info/entry_points.txt,sha256=nQbvwctWkWD7I-2f4VrdVQBZYGUw8CnUnFC6QjXxOSE,51
-lean_lsp_mcp-0.12.1.dist-info/top_level.txt,sha256=LGEK0lgMSNPIQ6mG8EO-adaZEGPi_0daDs004epOTF0,13
-lean_lsp_mcp-0.12.1.dist-info/RECORD,,