毕昇报告生成放出message,并去掉重复的step_message

zhangqian

2024-10-29 a625a4731c0f8d1ca57b164580bc44c37d98d533

毕昇报告生成放出message,并去掉重复的step_message

 app/api/report.py

@@ -65,8 +65,8 @@
                    files = data.get("files", [])
                    steps = data.get("intermediate_steps", "")
                    msg = data.get("message", "")

                    if len(files) != 0 or (steps and last_message == "step") or msg or data["type"] == "close":
                        last_message = "step" if steps else "message"
                        if data["type"] == "close":
                            t = "close"
                        else:
@@ -76,6 +76,8 @@
                        await websocket.send_json(result)
                        print(f"Forwarded to client, {chat_id}: {result}")

                    last_message = "step" if steps else "message"

            # 启动两个任务，分别处理客户端和服务端的消息
            tasks = [
                asyncio.create_task(forward_to_service()),