fix: show line numbers in semantic_edit

src/codegen/extensions/langchain/agent.py

@@ -13,14 +13,9 @@
 from codegen import Codebase
 
 from .tools import (
-    CommitTool,
     CreateFileTool,
     DeleteFileTool,
     EditFileTool,
-    GithubCreatePRCommentTool,
-    GithubCreatePRReviewCommentTool,
-    GithubCreatePRTool,
-    GithubViewPRTool,
     ListDirectoryTool,
     MoveSymbolTool,
     RenameFileTool,
@@ -69,11 +64,11 @@ def create_codebase_agent(
         RevealSymbolTool(codebase),
         SemanticEditTool(codebase),
         SemanticSearchTool(codebase),
-        CommitTool(codebase),
-        GithubCreatePRTool(codebase),
-        GithubViewPRTool(codebase),
-        GithubCreatePRCommentTool(codebase),
-        GithubCreatePRReviewCommentTool(codebase),
+        # CommitTool(codebase),
+        # GithubCreatePRTool(codebase),
+        # GithubViewPRTool(codebase),
+        # GithubCreatePRCommentTool(codebase),
+        # GithubCreatePRReviewCommentTool(codebase),
     ]
 
     prompt = ChatPromptTemplate.from_messages(
@@ -103,7 +98,6 @@ def create_codebase_agent(
         - Preview changes before applying
         - Ensure code quality with linting
 
-
         4. Code Search:
         - Text-based and semantic search
         - Search within specific directories

src/codegen/extensions/langchain/tools.py

@@ -49,9 +49,9 @@
 class ViewFileTool(BaseTool):
     """Tool for viewing file contents and metadata."""
 
     name: ClassVar[str] = "view_file"
     description: ClassVar[str] = "View the contents and metadata of a file in the codebase"
     args_schema: ClassVar[type[BaseModel]] = ViewFileInput
     codebase: Codebase = Field(exclude=True)
 
     def __init__(self, codebase: Codebase) -> None:
@@ -72,9 +72,9 @@
 class ListDirectoryTool(BaseTool):
     """Tool for listing directory contents."""
 
     name: ClassVar[str] = "list_directory"
     description: ClassVar[str] = "List contents of a directory in the codebase"
     args_schema: ClassVar[type[BaseModel]] = ListDirectoryInput
     codebase: Codebase = Field(exclude=True)
 
     def __init__(self, codebase: Codebase) -> None:
@@ -95,9 +95,9 @@
 class SearchTool(BaseTool):
     """Tool for searching the codebase."""
 
     name: ClassVar[str] = "search"
     description: ClassVar[str] = "Search the codebase using text search"
     args_schema: ClassVar[type[BaseModel]] = SearchInput
     codebase: Codebase = Field(exclude=True)
 
     def __init__(self, codebase: Codebase) -> None:
@@ -234,10 +234,23 @@
         return json.dumps(result, indent=2)
 
 
-_SEMANTIC_EDIT_BRIEF = """Tool for semantic editing of files.
-* Allows editing files by providing a draft of the new content
-* For large files, specify line ranges to edit
-* Will intelligently handle unchanged sections of code. Also supports appending to the end of a file."""
+_SEMANTIC_EDIT_BRIEF = """Tool for file editing via an LLM delegate. Describe the changes you want to make and an expert will apply them to the file.
+
+Specify the changes you want to make in the edit_content field, with helpful comments, like so:
+```
+# ... existing code ...
+
+# edit: change function name and body
+def function_redefinition():
+    return 'new_function_body'
+
+# ... existing code ...
+```
+
+Another agent will be responsible for applying your edit and will only see the `edit_content` and the range you specify, so be clear what you are looking to change.
+
+For large files, specify a range slightly larger than the edit you want to make, and be clear in your description what should and shouldn't change.
+"""
 
 
 class SemanticEditInput(BaseModel):

src/codegen/extensions/tools/semantic_edit.py

@@ -47,9 +47,9 @@
     Raises:
         ValueError: If response is not properly formatted with code blocks
     """
-    # Find content between ``` markers
-    pattern = r"```(?:\w*\n)?(.*?)```"
-    matches = re.findall(pattern, llm_response, re.DOTALL)
+    # Find content between ``` markers, allowing for any language identifier
+    pattern = r"```[^`\n]*\n?(.*?)```"
+    matches = re.findall(pattern, llm_response.strip(), re.DOTALL)
 
     if not matches:
         raise ValueError("LLM response must contain code wrapped in ``` blocks. Got response: " + llm_response[:200] + "...")
@@ -154,7 +154,7 @@
 
     # Extract code from markdown code block
     try:
         modified_segment = _extract_code_block(response.content)
     except ValueError as e:
         msg = f"Failed to parse LLM response: {e!s}"
         raise ValueError(msg)