From a27d23e9d7dde3a220795828971f480850c22b8f Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 14 十一月 2024 23:32:09 +0800 Subject: [PATCH] 文档智能改成毕昇的,去除毕昇文档回答的重复消息 --- app/service/bisheng.py | 22 ++++++++++++++++++++++ 1 files changed, 22 insertions(+), 0 deletions(-) diff --git a/app/service/bisheng.py b/app/service/bisheng.py index ad5d084..d4781da 100644 --- a/app/service/bisheng.py +++ b/app/service/bisheng.py @@ -73,6 +73,28 @@ ] return result + async def get_session_log(self, token: str, agent_id: str, conversation_id: str): + url = ( + f"{self.base_url}/api/v1/chat/history?" + f"flow_id={agent_id}&" + f"chat_id={conversation_id}&page_size=30&id=" + ) + headers = {'cookie': f"access_token_cookie={token};"} + async with httpx.AsyncClient() as client: + response = await client.get(url, headers=headers) + response.raise_for_status() + data = self._check_response(response) + session_log = { + "session_log": [ + { + "message": message.get("message"), + "role": message.get("category"), + } + for message in data + ], + } + return session_log + async def variable_list(self, token: str, agent_id: str) -> list: url = f"{self.base_url}/api/v1/variable/list?flow_id={agent_id}" headers = {'cookie': f"access_token_cookie={token};"} -- Gitblit v1.8.0