From d278e940273493d1e0a5407c008e9c3d4d00af38 Mon Sep 17 00:00:00 2001 From: Rushil Patel Date: Tue, 18 Mar 2025 13:39:51 -0700 Subject: [PATCH 1/2] fix: params reference --- src/codegen/extensions/langchain/tools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/codegen/extensions/langchain/tools.py b/src/codegen/extensions/langchain/tools.py index ee17aa691..36ee1102f 100644 --- a/src/codegen/extensions/langchain/tools.py +++ b/src/codegen/extensions/langchain/tools.py @@ -1112,6 +1112,6 @@ class SearchFilesByNameTool(BaseTool): 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() From 559c5e94338ccc22068d57cca3c663c8b3b9ff37 Mon Sep 17 00:00:00 2001 From: rushilpatel0 <171610820+rushilpatel0@users.noreply.github.com> Date: Tue, 18 Mar 2025 20:40:51 +0000 Subject: [PATCH 2/2] Automated pre-commit update --- src/codegen/extensions/langchain/tools.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/codegen/extensions/langchain/tools.py b/src/codegen/extensions/langchain/tools.py index 36ee1102f..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,8 +1108,6 @@ class SearchFilesByNameTool(BaseTool): args_schema: ClassVar[type[BaseModel]] = SearchFilesByNameInput codebase: Codebase = Field(exclude=True) - - def __init__(self, codebase: Codebase): super().__init__(codebase=codebase)