lisone/ds2api
1
0
Fork 0
mirror of https://github.com/CJackHwang/ds2api.git synced 2026-05-15 05:35:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add support for parsing dictionary-based SSE fragments and prevent duplicate stream termination messages.

Browse Source
This commit is contained in:
CJACK
2026-02-13 22:44:39 +08:00
parent ee0b7f08a0
commit 648b80bb7b
3 changed files with 29 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
core/sse_parser.py
View File

@@ -255,6 +255,26 @@ def parse_sse_chunk_for_content(
return ([], True, new_fragment_type)
contents.extend(result)
# 处理字典值（初始响应 chunk，包含 response.fragments）
elif isinstance(v_value, dict):
response_obj = v_value.get("response", v_value)
fragments = response_obj.get("fragments", [])
if isinstance(fragments, list):
for frag in fragments:
if isinstance(frag, dict):
frag_type = frag.get("type", "").upper()
frag_content = frag.get("content", "")
if frag_type == "THINK" or frag_type == "THINKING":
new_fragment_type = "thinking"
if frag_content:
contents.append((frag_content, "thinking"))
elif frag_type == "RESPONSE":
new_fragment_type = "text"
if frag_content:
contents.append((frag_content, "text"))
elif frag_content:
contents.append((frag_content, ptype))
return (contents, False, new_fragment_type)