apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Editor.read + file size检查

Browse source
This commit is contained in:
莘权 马 2024-09-20 14:24:17 +08:00
parent d37954d43a
commit dccd199249
2 changed files with 13 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
metagpt/tools/libs/editor.py
View file

@ -104,14 +104,18 @@ class Editor(BaseModel):
async def read(self, path: str) -> FileBlock:
"""Read the whole content of a file. Using absolute paths as the argument for specifying the file location."""
error = FileBlock(
file_path=str(path),
block_content="The file is too large to read. Use `Editor.similarity_search` to read the file instead.",
)
path = Path(path)
if path.stat().st_size > 5 * DEFAULT_MIN_TOKEN_COUNT:
return error
content = await File.read_text_file(path)
if not content:
return FileBlock(file_path=str(path), block_content="")
if self.is_large_file(content=content):
return FileBlock(
file_path=str(path),
block_content="The file is too large to read. Use `Editor.similarity_search` to read the file instead.",
)
return error
self.resource.report(str(path), "path")
lines = content.splitlines(keepends=True)