| | |
| | | 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", ""), |
| | |
| | | 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", "") |
| | |
| | | "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", |
| | |
| | | |
| | | 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", ""), |
| | |
| | | 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: |
| | |
| | | # print(e) |
| | | # continue |
| | | result = {"message": answer_str, "type": "message", |
| | | "download_url": ""} |
| | | "download_url": "", "workflow": {"node_data": workflow_list}} |
| | | |
| | | # try: |
| | | # await websocket.send_json(result) |