diff --git a/src/codegen/extensions/langchain/tools.py b/src/codegen/extensions/langchain/tools.py index ee17aa691..867ade8e0 100644 --- a/src/codegen/extensions/langchain/tools.py +++ b/src/codegen/extensions/langchain/tools.py @@ -1094,6 +1094,7 @@ class SearchFilesByNameInput(BaseModel): page: int = Field(default=1, description="Page number to return (1-based)") files_per_page: int | float = Field(default=10, description="Number of files per page to return, use math.inf to return all files") + class SearchFilesByNameTool(BaseTool): """Tool for searching files by filename across a codebase.""" @@ -1107,11 +1108,9 @@ class SearchFilesByNameTool(BaseTool): args_schema: ClassVar[type[BaseModel]] = SearchFilesByNameInput codebase: Codebase = Field(exclude=True) - - def __init__(self, codebase: Codebase): super().__init__(codebase=codebase) - def _run(self, pattern: str) -> str: + def _run(self, pattern: str, page: int, files_per_page: int | float) -> str: """Execute the glob pattern search using fd.""" - return search_files_by_name(self.codebase, pattern, page=self.page, files_per_page=self.files_per_page).render() + return search_files_by_name(self.codebase, pattern, page=page, files_per_page=files_per_page).render()