apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +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

7
tests/metagpt/tools/libs/test_editor.py
View file

@ -9,6 +9,7 @@ from metagpt.tools.libs.editor import Editor
from metagpt.tools.libs.index_repo import (
CHATS_INDEX_ROOT,
CHATS_ROOT,
DEFAULT_MIN_TOKEN_COUNT,
UPLOAD_ROOT,
IndexRepo,
)
@ -780,8 +781,10 @@ async def test_similarity_search(query, filename):
@pytest.mark.asyncio
async def test_read():
editor = Editor()
content = await editor.read(str(TEST_DATA_PATH / "pdf/9112674.pdf"))
assert "similarity_search" in content.block_content
filename = TEST_DATA_PATH / "pdf/9112674.pdf"
content = await editor.read(str(filename))
size = filename.stat().st_size
assert "similarity_search" in content.block_content and size > 5 * DEFAULT_MIN_TOKEN_COUNT
if __name__ == "__main__":