github_mirrors/wiseflow
1
0
Fork 0
mirror of https://github.com/TeamWiseFlow/wiseflow.git synced 2025-02-02 18:28:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

repair openai wrapper

Browse Source
This commit is contained in:
bigbrother666sh 2024-10-10 20:23:58 +08:00
parent 60df751129
commit 61e3c042b4
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/llms/openai_wrapper.py
View File

@ -29,8 +29,10 @@ def openai_llm(messages: list, model: str, logger=None, **kwargs) -> str:
logger.warning(f'{e}\nRetrying in 60 second...') logger.warning(f'{e}\nRetrying in 60 second...')
time.sleep(60) time.sleep(60)
response = client.chat.completions.create(messages=messages, model=model, **kwargs) response = client.chat.completions.create(messages=messages, model=model, **kwargs)
if 'choices' not in response: if response and response.choices:
logger.warning(f'openai_llm warning: {response}') return response.choices[0].message.content
else:
logger.error(f'after many try, llm error: {response}')
return "" return ""
except Exception as e: except Exception as e:
if logger: if logger: