Skip to content

Commit b693b28

Browse files
authored
fix view tool (#1092)
1 parent c9b02b0 commit b693b28

File tree

2 files changed

+17
-10
lines changed

2 files changed

+17
-10
lines changed

patchwork/common/tools/code_edit_tools.py

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -133,15 +133,22 @@ def __view(self, abs_path: Path, view_range):
133133
content = "\n".join(lines[start - 1 : end])
134134
return content
135135
elif abs_path.is_dir():
136-
result = []
137-
for root, dirs, files in os.walk(abs_path):
138-
level = root[len(str(abs_path)) :].count(os.sep)
139-
if level <= 2:
140-
for d in dirs:
141-
result.append(d)
142-
for f in files:
143-
result.append(f)
144-
return "\n".join(result)
136+
directories = []
137+
files = []
138+
for file in abs_path.iterdir():
139+
directories.append(file.name) if file.is_dir() else files.append(file.name)
140+
141+
rv = ""
142+
if len(directories) > 0:
143+
rv += "Directories: \n"
144+
rv += '\n'.join(directories)
145+
rv += "\n"
146+
147+
if len(files) > 0:
148+
rv += "Files: \n"
149+
rv += '\n'.join(files)
150+
151+
return rv
145152

146153
def __create(self, file_text, abs_path):
147154
if abs_path.exists():

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "patchwork-cli"
3-
version = "0.0.82"
3+
version = "0.0.83"
44
description = ""
55
authors = ["patched.codes"]
66
license = "AGPL"

0 commit comments

Comments
 (0)