feat: merge fixbug/rfc135_merge_geekan_cli_etc_1445

This commit is contained in:
莘权 马 2023-12-15 20:32:16 +08:00
parent 8636026c55
commit a9479843f6
8 changed files with 33 additions and 16 deletions

View file

@ -6,15 +6,17 @@
@File : test_text_to_image.py
@Desc : Unit tests.
"""
import asyncio
import base64
import pytest
from pydantic import BaseModel
from metagpt.learn.text_to_image import text_to_image
async def mock_text_to_image():
@pytest.mark.asyncio
async def test():
class Input(BaseModel):
input: str
size_type: str
@ -36,11 +38,5 @@ async def mock_text_to_image():
assert base64.b64decode(data, validate=True)
def test_suite():
loop = asyncio.get_event_loop()
task = loop.create_task(mock_text_to_image())
loop.run_until_complete(task)
if __name__ == "__main__":
test_suite()
pytest.main([__file__, "-s"])

View file

@ -11,6 +11,7 @@
import pytest
from metagpt.actions import UserRequirement
from metagpt.config import CONFIG
from metagpt.environment import Environment
from metagpt.logs import logger
from metagpt.roles import Architect, ProductManager, Role
@ -41,6 +42,10 @@ def test_get_roles(env: Environment):
@pytest.mark.asyncio
async def test_publish_and_process_message(env: Environment):
if CONFIG.git_repo:
CONFIG.git_repo.delete_repository()
CONFIG.git_repo = None
product_manager = ProductManager(name="Alice", profile="Product Manager", goal="做AI Native产品", constraints="资源有限")
architect = Architect(
name="Bob", profile="Architect", goal="设计一个可用、高效、较低成本的系统,包括数据结构与接口", constraints="资源有限,需要节省成本"