apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-07-02 16:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixbug: 修复通用智能体role及其相关的TalkAction和SkillAction

Browse source
This commit is contained in:
莘权 马 2023-12-25 22:39:03 +08:00
parent 6261251279
commit 0fdb552468
12 changed files with 541 additions and 512 deletions
Download patch file Download diff file Expand all files Collapse all files

21
metagpt/tools/openai_text_to_image.py
View file

@ -11,23 +11,23 @@ import base64
import aiohttp
import requests
from openai import AsyncOpenAI
from metagpt.config import CONFIG, Config
from metagpt.config import Config
from metagpt.llm import LLM
from metagpt.logs import logger
class OpenAIText2Image:
def __init__(self, openai_api_key):
def __init__(self):
"""
:param openai_api_key: OpenAI API key, For more details, checkout: `https://platform.openai.com/account/api-keys`
"""
self.openai_api_key = openai_api_key if openai_api_key else CONFIG.OPENAI_API_KEY
self._client = AsyncOpenAI(api_key=self.openai_api_key, base_url=CONFIG.openai_api_base)
self._llm = LLM()
self._client = self._llm.async_client
def __del__(self):
if self._client:
self._client.close()
if self._llm:
self._llm.close()
async def text_2_image(self, text, size_type="1024x1024"):
"""Text to image
@ -66,19 +66,16 @@ class OpenAIText2Image:
# Export
async def oas3_openai_text_to_image(text, size_type: str = "1024x1024", openai_api_key=""):
async def oas3_openai_text_to_image(text, size_type: str = "1024x1024"):
"""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: One of ['256x256', '512x512', '1024x1024']
:return: The image data is returned in Base64 encoding.
"""
if not text:
return ""
if not openai_api_key:
openai_api_key = CONFIG.OPENAI_API_KEY
return await OpenAIText2Image(openai_api_key).text_2_image(text, size_type=size_type)
return await OpenAIText2Image().text_2_image(text, size_type=size_type)
if __name__ == "__main__":