jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: the object field is empty string in some openAI api compatible model (#3506)

Browse source
This commit is contained in:
liuzhenghua 2024-04-16 12:13:10 +08:00 • committed by GitHub
commit 5e02a83b53
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py
View file

@ -150,6 +150,8 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
except json.JSONDecodeError as e: except json.JSONDecodeError as e:
raise CredentialsValidateFailedError('Credentials validation failed: JSON decode error') raise CredentialsValidateFailedError('Credentials validation failed: JSON decode error')
if (completion_type is LLMMode.CHAT and json_result['object'] == ''):
json_result['object'] = 'chat.completion'
if (completion_type is LLMMode.CHAT if (completion_type is LLMMode.CHAT
and ('object' not in json_result or json_result['object'] != 'chat.completion')): and ('object' not in json_result or json_result['object'] != 'chat.completion')):
raise CredentialsValidateFailedError( raise CredentialsValidateFailedError(
@ -807,4 +809,4 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
function=function function=function
) )
return tool_call return tool_call