remove sync api in openai

This commit is contained in:
geekan 2023-12-26 15:59:11 +08:00
parent e15de55368
commit bb1b9823d0
14 changed files with 52 additions and 207 deletions

View file

@ -25,10 +25,6 @@ class OpenAIText2Image:
self._llm = LLM()
self._client = self._llm.async_client
def __del__(self):
if self._llm:
self._llm.close()
async def text_2_image(self, text, size_type="1024x1024"):
"""Text to image

View file

@ -278,11 +278,11 @@ class UTGenerator:
question += self.build_api_doc(node, path, method)
self.ask_gpt_and_save(question, tag, summary)
def gpt_msgs_to_code(self, messages: list) -> str:
async def gpt_msgs_to_code(self, messages: list) -> str:
"""Choose based on different calling methods"""
result = ""
if self.chatgpt_method == "API":
result = GPTAPI().ask_code(msgs=messages)
result = await GPTAPI().aask_code(msgs=messages)
return result