Paillat-dev/Botator
1
0
Fork 0
mirror of https://github.com/Paillat-dev/Botator.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix(openaicaller.py): fix error in generate_response method where error_call argument is not properly handled

Browse Source 
✨ feat(openaicaller.py): add default value for error_call argument in generate_response method to prevent potential errors
This commit is contained in:
Paillat
2023-08-23 15:39:19 +02:00
parent c7071ce7af
commit cc28f3d4a1
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/utils/openaicaller.py
View File

@@ -83,14 +83,20 @@ class openai_caller:
# async def generate_response(self, error_call=None, **kwargs): # async def generate_response(self, error_call=None, **kwargs):
async def generate_response(*args, **kwargs): async def generate_response(*args, **kwargs):
self = args[0] self = args[0]
if args.get("error_call", None) != None: if len(args) > 1:
error_call = args[1] error_call = args[1] or nothing
else: else:
async def nothing(x): async def nothing(x):
return x return x
error_call = nothing error_call = nothing
if error_call == None:
async def nothing(x):
return x
error_call = nothing
if kwargs.get("model", "") in chat_models: if kwargs.get("model", "") in chat_models:
return await self.chat_generate(error_call, **kwargs) return await self.chat_generate(error_call, **kwargs)
elif kwargs.get("engine", "") in text_models: elif kwargs.get("engine", "") in text_models: