fix: output serialization for replacementEditTool

src/codegen/extensions/langchain/tools.py

@@ -1,6 +1,5 @@
 """Langchain tools for workspace operations."""
 
-import json
 from typing import Callable, ClassVar, Literal, Optional
 
 from langchain_core.tools.base import BaseTool
@@ -54,11 +53,11 @@
 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.
 For large files (>250 lines), content will be paginated. Use start_line and end_line to navigate through the file.
 The response will indicate if there are more lines available to view."""
     args_schema: ClassVar[type[BaseModel]] = ViewFileInput
     codebase: Codebase = Field(exclude=True)
 
     def __init__(self, codebase: Codebase) -> None:
@@ -93,9 +92,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:
@@ -116,9 +115,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:
@@ -139,7 +138,7 @@
 class EditFileTool(BaseTool):
     """Tool for editing files."""
 
     name: ClassVar[str] = "edit_file"
     description: ClassVar[str] = "Edit a file by replacing its entire content. This tool should only be used for replacing entire file contents."
     args_schema: ClassVar[type[BaseModel]] = EditFileInput
     codebase: Codebase = Field(exclude=True)
@@ -462,7 +461,6 @@
     def _run(self, pr_id: int) -> str:
         result = view_pr(self.codebase, pr_id)
         return result.render()
-        return json.dumps(result, indent=2)
 
 
 class GithubCreatePRCommentInput(BaseModel):
@@ -791,4 +789,4 @@
             end=end,
             count=count,
         )
-        return json.dumps(result, indent=2)
+        return result.render()