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

2024-04-18 16:54:16 +08:00 · 2024-04-18 16:54:16 +08:00 · aa6d2e3035
commit aa6d2e3035
parent 4365843c20
1 changed files with 2 additions and 1 deletions
--- a/api/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py
+++ b/api/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py
@ -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(':'):