zhaoqingang
2025-01-10 f0a780fecb0e5799f6427cc89da8e5600edc71a2
app/api/chat.py
@@ -578,7 +578,7 @@
                                                        message = answer.get("title", "")
                                                    result = {"message": message, "type": "system"}
                                                    result = {"message": message, "type": "system", "workflow":{"node_data": workflow_list}}
                                            elif data.get("event") == "node_finished":
                                                workflow_list.append({
                                                    "title": data.get("data", {}).get("title", ""),
@@ -599,7 +599,7 @@
                                                    message = answer.get("title", "")
                                                    if answer.get("status") == "failed":
                                                        message = answer.get("error", "")
                                                        result = {"message": message, "type": "system"}
                                                        result = {"message": message, "type": "system", "workflow":{"node_data": workflow_list}}
                                            elif data.get("event") == "workflow_finished":
                                                answer = data.get("data", "")
@@ -631,7 +631,7 @@
                                                        "error": answer.get("error", ""),
                                                        "elapsed_time": answer.get("elapsed_time", 0)
                                                    }
                                                    result = {"message": message, "type": "message", "download_url": download_url}
                                                    result = {"message": message, "type": "message", "download_url": download_url, "workflow":workflow_dict}
                                                    try:
                                                        SessionService(db).update_session(chat_id,
                                                                                          message={"role": "assistant",
@@ -713,7 +713,7 @@
                                                        message = answer.get("title", "")
                                                    result = {"message": message, "type": "system"}
                                                    result = {"message": message, "type": "system", "workflow":{"node_data": workflow_list}}
                                            elif data.get("event") == "node_finished":
                                                workflow_list.append({
                                                    "title": data.get("data", {}).get("title", ""),
@@ -735,7 +735,7 @@
                                                    message = answer.get("title", "")
                                                    if answer.get("status") == "failed":
                                                        message = answer.get("error", "")
                                                    result = {"message": message, "type": "system"}
                                                    result = {"message": message, "type": "system", "workflow":{"node_data": workflow_list}}
                                            elif data.get("event") == "message":
                                                answer_str = data.get("answer", "")
                                                # try:
@@ -745,7 +745,7 @@
                                                #     print(e)
                                                #     continue
                                                result = {"message": answer_str, "type": "message",
                                                          "download_url": ""}
                                                          "download_url": "", "workflow": {"node_data": workflow_list}}
                                                # try:
                                                #     await websocket.send_json(result)