diff --git a/src/blackbox/chat.py b/src/blackbox/chat.py index 991e415..1b8e1e6 100644 --- a/src/blackbox/chat.py +++ b/src/blackbox/chat.py @@ -311,6 +311,7 @@ class Chat(Blackbox): except json.JSONDecodeError: # print("---- Error in JSON parsing ----") # 打印错误信息 continue # 继续处理下一个chunk,直到解析成功 + yield "[Response END]" else: print("*"*90) @@ -346,4 +347,4 @@ class Chat(Blackbox): return EventSourceResponse(self.processing(prompt, context, setting)) else: response_content = "".join(self.processing(prompt, context, setting)) - return JSONResponse(content={"response": response_content}, status_code=status.HTTP_200_OK) \ No newline at end of file + return JSONResponse(content={"response": response_content}, status_code=status.HTTP_200_OK)