xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-09 23:36:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: webchat stream is ok

Browse Source
This commit is contained in:
Dong_master
2025-08-01 11:33:16 +08:00
committed by Junyan Qin
parent b45cc59322
commit c6deed4e6e
4 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/api/http/controller/groups/pipelines/webchat.py
View File

@@ -14,8 +14,8 @@ class WebChatDebugRouterGroup(group.RouterGroup):
async def stream_generator(generator):
async for message in generator:
yield rf"data:{json.dumps({'message': message})}\n\n"
yield "data:{type: end}\n\n''"
yield f"data: {json.dumps({'message': message})}\n\n"
yield "data: {\"type\": \"end\"}\n\n"
try:
data = await quart.request.get_json()
session_type = data.get('session_type', 'person')

5
pkg/platform/sources/webchat.py
View File

@@ -242,11 +242,12 @@ class WebChatAdapter(msadapter.MessagePlatformAdapter):
if is_stream:
queue = use_session.resp_queues[message_id]
msg_id = len(use_session.get_message_list(pipeline_uuid)) + 1
while True:
resp_message = await queue.get()
print(resp_message)
resp_message.id = msg_id
if resp_message.is_final:
resp_message.id = len(use_session.get_message_list(pipeline_uuid)) + 1
resp_message.id = msg_id
use_session.get_message_list(pipeline_uuid).append(resp_message)
yield resp_message.model_dump()
break