From f192a7a302373ae931104d6cb44076fe694da470 Mon Sep 17 00:00:00 2001 From: zhangxiao <898441624@qq.com> Date: 星期三, 16 十月 2024 14:53:32 +0800 Subject: [PATCH] 报表合并 --- app/service/ragflow.py | 21 +++++++++++++++++++++ 1 files changed, 21 insertions(+), 0 deletions(-) diff --git a/app/service/ragflow.py b/app/service/ragflow.py index df131f1..934af1a 100644 --- a/app/service/ragflow.py +++ b/app/service/ragflow.py @@ -59,3 +59,24 @@ return else: yield f"Error: {response.status_code}" + + async def get_chat_sessions(self, token: str, dialog_id: str) -> list: + url = f"{self.base_url}/v1/conversation/list?dialog_id={dialog_id}" + headers = { + "Authorization": token + } + async with httpx.AsyncClient() as client: + response = await client.get(url, headers=headers) + if response.status_code != 200: + raise Exception(f"Failed to fetch data from Ragflow API: {response.text}") + + data = response.json().get("data", []) + result = [ + { + "id": item["id"], + "name": item["name"], + "updated_time": item["update_time"] + } + for item in data + ] + return result -- Gitblit v1.8.0