From 3fc9f4f33cf90610c71a1de7b00db0f82b988e98 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期三, 16 十月 2024 23:22:53 +0800 Subject: [PATCH] 文档上传&获取报告生成变量&报告生成接口 --- app/service/ragflow.py | 21 ++++++++++++++------- 1 files changed, 14 insertions(+), 7 deletions(-) diff --git a/app/service/ragflow.py b/app/service/ragflow.py index e4d5657..0fe0c39 100644 --- a/app/service/ragflow.py +++ b/app/service/ragflow.py @@ -87,22 +87,29 @@ "Authorization": token } - data = {"dialog_id": dialog_id, + data = { + "dialog_id": dialog_id, "name": name, "is_new": is_new, "conversation_id": chat_id, - } + } async with httpx.AsyncClient() as client: response = await client.post(url, headers=headers, json=data) if response.status_code != 200: return [] - return [{ - "content": "浣犲ソ锛� 鎴戞槸浣犵殑鍔╃悊锛屾湁浠�涔堝彲浠ュ府鍒颁綘鐨勫悧锛�", - "role": "assistant" - }, + ret_code = response.json().get("retcode") + if ret_code != 0: + return [] + + return [ + { + "content": "浣犲ソ锛� 鎴戞槸浣犵殑鍔╃悊锛屾湁浠�涔堝彲浠ュ府鍒颁綘鐨勫悧锛�", + "role": "assistant" + }, { "content": name, "doc_ids": [], "role": "user" - }] + } + ] -- Gitblit v1.8.0