Parcourir la source

fix(openai_api_compatible): fixing the error when converting chunk to json (#3570)

tags/0.6.4
aniaan il y a 1 an
Parent
révision
aa6d2e3035
Aucun compte lié à l'adresse e-mail de l'auteur

+ 2
- 1
api/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py Voir le fichier

@@ -154,7 +154,7 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
json_result['object'] = 'chat.completion'
elif (completion_type is LLMMode.COMPLETION and json_result['object'] == ''):
json_result['object'] = 'text_completion'
if (completion_type is LLMMode.CHAT
and ('object' not in json_result or json_result['object'] != 'chat.completion')):
raise CredentialsValidateFailedError(
@@ -425,6 +425,7 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
finish_reason = 'Unknown'

for chunk in response.iter_lines(decode_unicode=True, delimiter=delimiter):
chunk = chunk.strip()
if chunk:
# ignore sse comments
if chunk.startswith(':'):

Chargement…
Annuler
Enregistrer