From f95f801f35aa201cbaffd7d881c07edc9398b570 Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期一, 03 三月 2025 16:03:51 +0800 Subject: [PATCH] 增加外接知识库中转接口 --- app/models/agent_model.py | 4 +++- 1 files changed, 3 insertions(+), 1 deletions(-) diff --git a/app/models/agent_model.py b/app/models/agent_model.py index cc9120e..2bcc003 100644 --- a/app/models/agent_model.py +++ b/app/models/agent_model.py @@ -9,6 +9,8 @@ RAGFLOW = 1 BISHENG = 2 BASIC = 3 + DIFY = 4 + OTHER = 5 class AgentModel(Base): @@ -128,5 +130,5 @@ @staticmethod def is_type(record_id, t): - record = UnifiedAgent.get_by_id(record_id) + record = UnifiedAgentModel.get_by_id(record_id) return record and record.type == t \ No newline at end of file -- Gitblit v1.8.0