hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 19:36:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: #12143 support streaming mode content start with "data:" (#12171)

Browse Source
This commit is contained in:
jifei
2025-01-03 16:33:37 +08:00
committed by GitHub
parent b873e6349c
commit 3c2e30f348
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/model_runtime/model_providers/moonshot/llm/llm.py
View File

@@ -252,7 +252,7 @@ class MoonshotLargeLanguageModel(OAIAPICompatLargeLanguageModel):
# ignore sse comments # ignore sse comments
if chunk.startswith(":"): if chunk.startswith(":"):
continue continue
decoded_chunk = chunk.strip().removeprefix("data: ") decoded_chunk = chunk.strip().removeprefix("data:").lstrip()
chunk_json = None chunk_json = None
try: try:
chunk_json = json.loads(decoded_chunk) chunk_json = json.loads(decoded_chunk)

2
api/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py
View File

@@ -479,7 +479,7 @@ class OAIAPICompatLargeLanguageModel(_CommonOaiApiCompat, LargeLanguageModel):
# ignore sse comments # ignore sse comments
if chunk.startswith(":"): if chunk.startswith(":"):
continue continue
decoded_chunk = chunk.strip().removeprefix("data: ") decoded_chunk = chunk.strip().removeprefix("data:").lstrip()
if decoded_chunk == "[DONE]": # Some provider returns "data: [DONE]" if decoded_chunk == "[DONE]": # Some provider returns "data: [DONE]"
continue continue

2
api/core/model_runtime/model_providers/stepfun/llm/llm.py
View File

@@ -250,7 +250,7 @@ class StepfunLargeLanguageModel(OAIAPICompatLargeLanguageModel):
# ignore sse comments # ignore sse comments
if chunk.startswith(":"): if chunk.startswith(":"):
continue continue
decoded_chunk = chunk.strip().removeprefix("data: ") decoded_chunk = chunk.strip().removeprefix("data:").lstrip()
chunk_json = None chunk_json = None
try: try:
chunk_json = json.loads(decoded_chunk) chunk_json = json.loads(decoded_chunk)