From 30311881800e4840a13f13dd702b093543b2082e Mon Sep 17 00:00:00 2001 From: zhangxiao <898441624@qq.com> Date: 星期三, 16 十月 2024 15:00:07 +0800 Subject: [PATCH] Merge remote-tracking branch 'origin/master' --- app/service/ragflow.py | 19 +++++++++++++++++++ 1 files changed, 19 insertions(+), 0 deletions(-) diff --git a/app/service/ragflow.py b/app/service/ragflow.py index 934af1a..7d1e6ae 100644 --- a/app/service/ragflow.py +++ b/app/service/ragflow.py @@ -80,3 +80,22 @@ for item in data ] return result + + async def set_session(self, token: str, dialog_id: str, name: str, chat_id: str, is_new: bool) -> bool: + url = f"{self.base_url}/v1/conversation/set?dialog_id={dialog_id}" + headers = { + "Authorization": token + } + + 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 False + return True + -- Gitblit v1.8.0