Skip to content

Commit dd83a61

Browse files
committed
fix: enabled infinite page size for global edit
1 parent 81d727c commit dd83a61

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

src/codegen/extensions/tools/global_replacement_edit.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import difflib
44
import logging
5+
import math
56
import re
67
from typing import ClassVar
78

@@ -103,7 +104,7 @@ def replacement_edit_global(
103104
)
104105

105106
diffs = []
106-
for file in search_files_by_name(codebase, file_pattern).files:
107+
for file in search_files_by_name(codebase, file_pattern, page=1, files_per_page=math.inf).files:
107108
if count is not None and count <= 0:
108109
break
109110
try:

src/codegen/extensions/tools/search_files_by_name.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1+
import math
12
import shutil
23
import subprocess
3-
from typing import ClassVar
4+
from typing import ClassVar, Optional
45

56
from pydantic import Field
67

@@ -44,7 +45,7 @@ def search_files_by_name(
4445
codebase: Codebase,
4546
pattern: str,
4647
page: int = 1,
47-
files_per_page: int = 10,
48+
files_per_page: int | float = 10,
4849
) -> SearchFilesByNameResultObservation:
4950
"""Search for files by name pattern in the codebase.
5051
@@ -58,8 +59,8 @@ def search_files_by_name(
5859
# Validate pagination parameters
5960
if page < 1:
6061
page = 1
61-
if files_per_page < 1:
62-
files_per_page = 10
62+
if files_per_page is not None and files_per_page < 1:
63+
files_per_page = 20
6364

6465
if shutil.which("fd") is None:
6566
logger.warning("fd is not installed, falling back to find")

0 commit comments

Comments
 (0)