rag-gateway
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
zhaoqingang
2025-01-14 536a8bbbe42c527ef9eaee9a8a8d2ee3615f5a11 
 app/api/chat.py


@@ -685,7 +685,7 @@


                        complete_response = ""


                        async for rag_response in dify_service.chat(token, current_user.id, question, files,


                                                                    conversation_id, {}):


                            # print(rag_response)


                            print(rag_response)


                            try:


                                if rag_response[:5] == "data:":


                                    # 如果是，则截取掉前5个字符，并去除首尾空白符


@@ -712,9 +712,11 @@


                                            elif isinstance(answer, dict):




                                                message = answer.get("title", "")


                                                if answer.get("status") == "failed":


                                                    message = answer.get("error")




                                            result = {"message": message, "type": "system"}


                                            continue


                                            # continue


                                    elif data.get("event") == "message":  # "event": "message_end"


                                         # 正常输出


                                        answer = data.get("answer", "")


@@ -724,7 +726,7 @@


                                        if isinstance(answer, str):


                                            logger.error("----------------未知数据--------------------")


                                            logger.error(data)


                                            result = {"message": "", "type": "close", "download_url": ""}


                                            # result = {"message": "", "type": "close", "download_url": ""}


                                        elif isinstance(answer, dict):


                                            download_url = ""


                                            outputs = answer.get("outputs", {})


@@ -734,8 +736,8 @@


                                            else:


                                                message = answer.get("error", "")




                                            # result = {"message": message, "type": "message",


                                            #           "download_url": download_url}


                                            result = {"message": message, "type": "system",


                                                      "download_url": download_url}


                                            try:


                                                SessionService(db).update_session(chat_id,


                                                                                  message={"role": "assistant",


@@ -748,7 +750,7 @@


                                                logger.error("保存dify的会话异常！")


                                                logger.error(e)


                                            # await websocket.send_json(result)


                                        continue


                                        # continue


                                    elif data.get("event") == "message_end":


                                        result = {"message": "", "type": "close"}