ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add stream chat stop flag [Response END]

Browse Source
This commit is contained in:
headbigsile
2025-06-27 16:58:08 +08:00
committed by GitHub
parent bb9905ed3e
commit 595dd5f75e
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/blackbox/chat.py
View File

@ -311,6 +311,7 @@ class Chat(Blackbox):
except json.JSONDecodeError: except json.JSONDecodeError:
# print("---- Error in JSON parsing ----") # 打印错误信息 # print("---- Error in JSON parsing ----") # 打印错误信息
continue # 继续处理下一个chunk，直到解析成功 continue # 继续处理下一个chunk，直到解析成功
yield "[Response END]"
else: else:
print("*"*90) print("*"*90)
@ -346,4 +347,4 @@ class Chat(Blackbox):
return EventSourceResponse(self.processing(prompt, context, setting)) return EventSourceResponse(self.processing(prompt, context, setting))
else: else:
response_content = "".join(self.processing(prompt, context, setting)) response_content = "".join(self.processing(prompt, context, setting))
return JSONResponse(content={"response": response_content}, status_code=status.HTTP_200_OK) return JSONResponse(content={"response": response_content}, status_code=status.HTTP_200_OK)