Skip to content

Added parameter to grep_files to return the content of matching lines. #4

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
certo-gco wants to merge 2 commits into
base: master
Choose a base branch
from
Open

Added parameter to grep_files to return the content of matching lines. #4

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
74d36bb
Added parameter to `grep_files` to return the content of matching lines.
certo-gco Dec 22, 2025
b5e8f96
Mypy munging.
certo-gco Dec 22, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 24 additions & 6 deletions tools/vfs.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -261,14 +261,20 @@ class GrepFileSchema(BaseModel):
Search for a specific string in the files on the VFS. Returns a list of
file names which contain the query somewhere in their contents. Matching
file names are output one per line. Empty lines should be ignored.
When invoked with the argument `matching_line_only` set to `True`, returns
pairs of the form `file_name: matching_line_content`, i.e.
the file names are paired with the content of the entire line matching
the query after the filename.
"""

search_string: str = Field(description="The query string to search for provided as a python regex. Thus, you must escape any special characters (like [, |, etc.)")
matching_line_only: bool = Field(description="Specify whether the content of the line matching the query should be returned with the filename")

@tool(args_schema=GrepFileSchema)
def grep_files(
state: Annotated[InputType, InjectedState],
search_string: str
search_string: str,
matching_line_only: bool
) -> str:
comp: re.Pattern
try:
Expand All @@ -281,8 +287,14 @@ def grep_files(
for (k, v) in state["vfs"].items():
if not get_filter(k):
continue
if comp.search(v) is not None:
matches.append(k)

if matching_line_only:
for line in v.splitlines():
if comp.search(line) is not None:
matches.append(f"{k}: {line}")
else:
if comp.search(v) is not None:
matches.append(k)

if (layer := conf.get("fs_layer", None)) is not None:
p = pathlib.Path(layer)
Expand All @@ -294,9 +306,15 @@ def grep_files(
continue
if rel_name in state["vfs"]:
continue
if not comp.search(f.read_text()):
continue
matches.append(rel_name)

content = f.read_text()
if matching_line_only:
for line in content.splitlines():
if comp.search(line) is not None:
matches.append(f"{rel_name}: {line}")
else:
if comp.search(content):
matches.append(rel_name)

return "\n".join(matches)

Expand Down