From 282a631b9ceee9a634ee1d93751a5254ed37ccef Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 18 三月 2025 10:10:48 +0800
Subject: [PATCH] 首页知识库对话-rg
---
app/models/dialog_model.py | 5 +++++
1 files changed, 5 insertions(+), 0 deletions(-)
diff --git a/app/models/dialog_model.py b/app/models/dialog_model.py
index 50a8f8e..a9aaf89 100644
--- a/app/models/dialog_model.py
+++ b/app/models/dialog_model.py
@@ -1,3 +1,4 @@
+import json
from datetime import datetime
from typing import Optional
@@ -24,6 +25,7 @@
# agent_id = Column(String(36))
mode = Column(String(36))
parameters = Column(Text)
+ kb_ids = Column(String(128))
def get_id(self):
return str(self.id)
@@ -43,6 +45,9 @@
'mode': self.mode,
}
+ def get_kb_ids(self):
+ return json.loads(self.kb_ids) if self.kb_ids else []
+
class ConversationModel(Base):
__tablename__ = 'conversation'
--
Gitblit v1.8.0