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

fix bug of save file and update prompt for gpt_v_generator tool

Browse source
This commit is contained in:
mannaandpoem 2024-01-19 10:00:56 +08:00
parent c7d46d0b0f
commit 06b35a34a6
4 changed files with 21 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/metagpt/tools/libs/test_gpt_v_generator.py
View file

@ -3,7 +3,7 @@
"""
@Time : 2024/01/15
@Author : mannaandpoem
@File : test_vision.py
@File : test_gpt_v_generator.py
"""
import pytest
@ -17,14 +17,14 @@ def mock_webpages(mocker):
<link rel="stylesheet" href="styles.css(">\n</html>\n```\n
```css\n.class { ... }\n```\n
```javascript\nfunction() { ... }\n```\n"""
mocker.patch("metagpt.tools.libs.gpt_v_generator.GPTvGenerator.generate_web_pages", return_value=mock_data)
mocker.patch("metagpt.tools.libs.gpt_v_generator.GPTvGenerator.generate_webpages", return_value=mock_data)
return mocker
def test_vision_generate_webpages(mock_webpages):
image_path = "image.png"
generator = GPTvGenerator()
rsp = generator.generate_web_pages(image_path=image_path)
rsp = generator.generate_webpages(image_path=image_path)
logs.logger.info(rsp)
assert "html" in rsp
assert "css" in rsp
@ -34,7 +34,7 @@ def test_vision_generate_webpages(mock_webpages):
def test_save_webpages(mock_webpages):
image_path = "image.png"
generator = GPTvGenerator()
webpages = generator.generate_web_pages(image_path)
webpages = generator.generate_webpages(image_path)
webpages_dir = generator.save_webpages(image_path=image_path, webpages=webpages)
logs.logger.info(webpages_dir)
assert webpages_dir.exists()