lean-lsp-mcp 0.12.0__py3-none-any.whl → 0.13.0__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 (use declaration_name to focus on one proof).
 - 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.
 
@@ -280,8 +284,11 @@ def diagnostic_messages(
 
     Args:
         file_path (str): Abs path to Lean file
-        start_line (int, optional): Start line (1-indexed) for filtering diagnostics. If provided, only diagnostics from this line onwards are returned.
-        end_line (int, optional): End line (1-indexed) for filtering diagnostics. If provided with start_line, only diagnostics in this range are returned.
+        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.
 
     Returns:
         List[str] | str: Diagnostic msgs or error msg
@@ -292,19 +299,23 @@ def diagnostic_messages(
 
     client: LeanLSPClient = ctx.request_context.lifespan_context.client
 
-    # leanclient requires both start_line and end_line to filter
+    # 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
-    if start_line is not None or end_line is not None:
-        start_line_0 = (start_line - 1) if start_line is not None else 0
-        end_line_0 = (end_line - 1) if end_line is not None else 999999
-        diagnostics = client.get_diagnostics(
-            rel_path,
-            start_line=start_line_0,
-            end_line=end_line_0,
-            inactivity_timeout=10.0,
-        )
-    else:
-        diagnostics = client.get_diagnostics(rel_path, inactivity_timeout=10.0)
+    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(
+        rel_path,
+        start_line=start_line_0,
+        end_line=end_line_0,
+        inactivity_timeout=8.0,
+    )
 
     return format_diagnostics(diagnostics)
 
@@ -660,7 +671,9 @@ def run_code(ctx: Context, code: str) -> List[str] | str:
         assert client is not None  # startup_client guarantees an initialized client
         client.open_file(rel_path)
         opened_file = True
-        diagnostics = format_diagnostics(client.get_diagnostics(rel_path))
+        diagnostics = format_diagnostics(
+            client.get_diagnostics(rel_path, inactivity_timeout=8.0)
+        )
     finally:
         if opened_file:
             try:

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.0.dist-info → lean_lsp_mcp-0.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lean-lsp-mcp
-Version: 0.12.0
+Version: 0.13.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.5.0
+Requires-Dist: leanclient==0.5.2
 Requires-Dist: mcp[cli]==1.21.0
 Requires-Dist: orjson>=3.11.1
 Provides-Extra: lint

lean_lsp_mcp-0.13.0.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=hJuWqRBOEAtzkQT83W2YgjgOJss9GZIGk2M1Y45X6bI,898
+lean_lsp_mcp/search_utils.py,sha256=X2LPynDNLi767UDxbxHpMccOkbnfKJKv_HxvRNxIXM4,3984
+lean_lsp_mcp/server.py,sha256=K-m_a8CiR8cUlyyZEq9t_R3Y07KZ-mZ_Ljws42nPCLU,36200
+lean_lsp_mcp/utils.py,sha256=YE6o6eswOi47AYojITQ2RcR-DspqKtgACeV-O7xgOKM,10554
+lean_lsp_mcp-0.13.0.dist-info/licenses/LICENSE,sha256=CQlxnf0tQyoVrBE93JYvAUYxv6Z5Yg6sX0pwogOkFvo,1071
+lean_lsp_mcp-0.13.0.dist-info/METADATA,sha256=OMClNzemdx8HJSJ5QvJ4ZYRz4ZqtyFvSDDztn2gH1Lk,19626
+lean_lsp_mcp-0.13.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lean_lsp_mcp-0.13.0.dist-info/entry_points.txt,sha256=nQbvwctWkWD7I-2f4VrdVQBZYGUw8CnUnFC6QjXxOSE,51
+lean_lsp_mcp-0.13.0.dist-info/top_level.txt,sha256=LGEK0lgMSNPIQ6mG8EO-adaZEGPi_0daDs004epOTF0,13
+lean_lsp_mcp-0.13.0.dist-info/RECORD,,

lean_lsp_mcp-0.12.0.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=oxuytxr58QLS-lCVRxYgtlLGEW0uHPhw2gKC5oRChqk,35799
-lean_lsp_mcp/utils.py,sha256=zLu2VIhaX4yocY07F3Z94LB2jRGrkH1ID9SjR3poE9A,8255
-lean_lsp_mcp-0.12.0.dist-info/licenses/LICENSE,sha256=CQlxnf0tQyoVrBE93JYvAUYxv6Z5Yg6sX0pwogOkFvo,1071
-lean_lsp_mcp-0.12.0.dist-info/METADATA,sha256=wMGQwNNQxVIt_hxZZgOZDIJ1E5wQmq2h4x9hRJrBgjw,19626
-lean_lsp_mcp-0.12.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lean_lsp_mcp-0.12.0.dist-info/entry_points.txt,sha256=nQbvwctWkWD7I-2f4VrdVQBZYGUw8CnUnFC6QjXxOSE,51
-lean_lsp_mcp-0.12.0.dist-info/top_level.txt,sha256=LGEK0lgMSNPIQ6mG8EO-adaZEGPi_0daDs004epOTF0,13
-lean_lsp_mcp-0.12.0.dist-info/RECORD,,