fix text_to_image ut error

This commit is contained in:
shenchucheng 2024-02-08 14:43:09 +08:00
parent 1563e9a583
commit 2094bab781

View file

@ -26,7 +26,8 @@ async def test_text_to_image(mocker):
# Prerequisites
assert CONFIG.METAGPT_TEXT_TO_IMAGE_MODEL_URL
assert CONFIG.OPENAI_API_KEY
CONFIG.OPENAI_TEXT_TO_IMAGE_API_KEY = CONFIG.OPENAI_API_KEY
assert CONFIG.OPENAI_TEXT_TO_IMAGE_API_KEY
data = await text_to_image("Panda emoji", size_type="512x512")
assert "base64" in data or "http" in data