From 28f41fceef54144cf87eaedd18d09a5a8b9cd5e1 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期三, 27 十一月 2024 09:40:58 +0800 Subject: [PATCH] dify 多次对话问题 --- app/api/agent.py | 8 ++++++++ 1 files changed, 8 insertions(+), 0 deletions(-) diff --git a/app/api/agent.py b/app/api/agent.py index 5e72c15..8b8add1 100644 --- a/app/api/agent.py +++ b/app/api/agent.py @@ -64,6 +64,12 @@ result = [item.to_dict() for item in records] return ResponseList(code=200, msg="", data=result) + elif agent.agent_type == AgentType.DIFY: + offset = (page - 1) * limit + records = db.query(SessionModel).filter(SessionModel.agent_id == agent_id, SessionModel.tenant_id==current_user.id).order_by(SessionModel.create_date.desc()).offset(offset).limit(limit).all() + result = [item.to_dict() for item in records] + return ResponseList(code=200, msg="", data=result) + else: return ResponseList(code=200, msg="Unsupported agent type") @@ -175,6 +181,8 @@ if "file_name" in i.get("content", {}): tmp_data["files"] = [{"file_name": i.get("content", {}).get("file_name"), "file_url": i.get("content", {}).get("file_url")}] + if "images" in i.get("content", {}): + tmp_data["images"] = i.get("content", {}).get("images") else: tmp_data["answer"] = i.get("content") data.append(tmp_data) -- Gitblit v1.8.0