From 2094bab781b520704055e1d445035bee265a8e88 Mon Sep 17 00:00:00 2001 From: shenchucheng Date: Thu, 8 Feb 2024 14:43:09 +0800 Subject: [PATCH] fix text_to_image ut error --- tests/metagpt/learn/test_text_to_image.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/metagpt/learn/test_text_to_image.py b/tests/metagpt/learn/test_text_to_image.py index 1485df5c6..f7be52842 100644 --- a/tests/metagpt/learn/test_text_to_image.py +++ b/tests/metagpt/learn/test_text_to_image.py @@ -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