fixbug: unit test

metagpt/learn/text_to_embedding.py

@@ -6,16 +6,19 @@
 @File    : text_to_embedding.py
 @Desc    : Text-to-Embedding skill, which provides text-to-embedding functionality.
 """
-
+import metagpt.config2
+from metagpt.config2 import Config
 from metagpt.tools.openai_text_to_embedding import oas3_openai_text_to_embedding
 
 
-async def text_to_embedding(text, model="text-embedding-ada-002", openai_api_key="", **kwargs):
+async def text_to_embedding(text, model="text-embedding-ada-002", config: Config = metagpt.config2.config):
     """Text to embedding
 
     :param text: The text used for embedding.
     :param model: One of ['text-embedding-ada-002'], ID of the model to use. For more details, checkout: `https://api.openai.com/v1/models`.
-    :param openai_api_key: OpenAI API key, For more details, checkout: `https://platform.openai.com/account/api-keys`
+    :param config: OpenAI config with API key, For more details, checkout: `https://platform.openai.com/account/api-keys`
     :return: A json object of :class:`ResultEmbedding` class if successful, otherwise `{}`.
     """
-    return await oas3_openai_text_to_embedding(text, model=model, openai_api_key=openai_api_key)
+    openai_api_key = config.get_openai_llm().api_key
+    proxy = config.get_openai_llm().proxy
+    return await oas3_openai_text_to_embedding(text, model=model, openai_api_key=openai_api_key, proxy=proxy)

metagpt/learn/text_to_image.py

@@ -8,6 +8,7 @@
 """
 import base64
 
+import metagpt.config2
 from metagpt.config2 import Config
 from metagpt.const import BASE64_FORMAT
 from metagpt.llm import LLM
@@ -16,27 +17,26 @@ from metagpt.tools.openai_text_to_image import oas3_openai_text_to_image
 from metagpt.utils.s3 import S3
 
 
-async def text_to_image(text, size_type: str = "512x512", model_url="", config: Config = None):
+async def text_to_image(text, size_type: str = "512x512", config: Config = metagpt.config2.config):
     """Text to image
 
     :param text: The text used for image conversion.
-    :param openai_api_key: OpenAI API key, For more details, checkout: `https://platform.openai.com/account/api-keys`
     :param size_type: If using OPENAI, the available size options are ['256x256', '512x512', '1024x1024'], while for MetaGPT, the options are ['512x512', '512x768'].
-    :param model_url: MetaGPT model url
     :param config: Config
     :return: The image data is returned in Base64 encoding.
     """
     image_declaration = "data:image/png;base64,"
 
+    model_url = config.METAGPT_TEXT_TO_IMAGE_MODEL_URL
     if model_url:
         binary_data = await oas3_metagpt_text_to_image(text, size_type, model_url)
     elif config.get_openai_llm():
-        binary_data = await oas3_openai_text_to_image(text, size_type, LLM())
+        llm = LLM(llm_config=config.get_openai_llm())
+        binary_data = await oas3_openai_text_to_image(text, size_type, llm=llm)
     else:
         raise ValueError("Missing necessary parameters.")
     base64_data = base64.b64encode(binary_data).decode("utf-8")
 
-    assert config.s3, "S3 config is required."
     s3 = S3(config.s3)
     url = await s3.cache(data=base64_data, file_ext=".png", format=BASE64_FORMAT)
     if url:

metagpt/learn/text_to_speech.py

@@ -6,8 +6,8 @@
 @File    : text_to_speech.py
 @Desc    : Text-to-Speech skill, which provides text-to-speech functionality
 """
-
-from metagpt.config2 import config
+import metagpt.config2
+from metagpt.config2 import Config
 from metagpt.const import BASE64_FORMAT
 from metagpt.tools.azure_tts import oas3_azsure_tts
 from metagpt.tools.iflytek_tts import oas3_iflytek_tts
@@ -20,12 +20,7 @@ async def text_to_speech(
     voice="zh-CN-XiaomoNeural",
     style="affectionate",
     role="Girl",
-    subscription_key="",
-    region="",
-    iflytek_app_id="",
-    iflytek_api_key="",
-    iflytek_api_secret="",
-    **kwargs,
+    config: Config = metagpt.config2.config,
 ):
     """Text to speech
     For more details, check out:`https://learn.microsoft.com/en-us/azure/ai-services/speech-service/language-support?tabs=tts`
@@ -44,6 +39,8 @@ async def text_to_speech(
 
     """
 
+    subscription_key = config.AZURE_TTS_SUBSCRIPTION_KEY
+    region = config.AZURE_TTS_REGION
     if subscription_key and region:
         audio_declaration = "data:audio/wav;base64,"
         base64_data = await oas3_azsure_tts(text, lang, voice, style, role, subscription_key, region)
@@ -52,6 +49,10 @@ async def text_to_speech(
         if url:
             return f"[{text}]({url})"
         return audio_declaration + base64_data if base64_data else base64_data
+
+    iflytek_app_id = config.IFLYTEK_APP_ID
+    iflytek_api_key = config.IFLYTEK_API_KEY
+    iflytek_api_secret = config.IFLYTEK_API_SECRET
     if iflytek_app_id and iflytek_api_key and iflytek_api_secret:
         audio_declaration = "data:audio/mp3;base64,"
         base64_data = await oas3_iflytek_tts(