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: +pic2txt

Browse source
This commit is contained in:
莘权 马 2024-06-27 14:35:05 +08:00
parent 9dc8d7307b
commit 742ff0e80a
11 changed files with 137 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

0
tests/metagpt/actions/requirement_analysis/requirement/__init__.py Normal file
View file

28
tests/metagpt/actions/requirement_analysis/requirement/test_pic2txt.py Normal file
View file

@ -0,0 +1,28 @@
import pytest
from metagpt.actions.requirement_analysis.requirement.pic2txt import Pic2Txt
from metagpt.const import TEST_DATA_PATH
from metagpt.utils.common import aread
@pytest.mark.asyncio
async def test_pic2txt(context):
images = [
TEST_DATA_PATH / "requirements/pic/1.png",
TEST_DATA_PATH / "requirements/pic/2.png",
TEST_DATA_PATH / "requirements/pic/3.png",
TEST_DATA_PATH / "requirements/pic/4.png",
TEST_DATA_PATH / "requirements/pic/5.png",
]
textual_user_requirements = await aread(filename=TEST_DATA_PATH / "requirements/1.original_requirement.txt")
acknowledge = await aread(filename=TEST_DATA_PATH / "requirements/1.acknowledge.md")
action = Pic2Txt(context=context)
rsp = await action.run(
image_paths=images, textual_user_requirement=textual_user_requirements, acknowledge=acknowledge
)
assert rsp
if __name__ == "__main__":
pytest.main([__file__, "-s"])