From 9ea07e00fc8b92e9b75849859c1ecce77c7096d1 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 10 四月 2025 13:31:23 +0800
Subject: [PATCH] 优化
---
app/models/dialog_model.py | 8 ++++++++
1 files changed, 8 insertions(+), 0 deletions(-)
diff --git a/app/models/dialog_model.py b/app/models/dialog_model.py
index f9340ef..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
@@ -23,6 +24,8 @@
dialog_type = Column(String(1)) # 骞冲彴
# 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)
@@ -41,6 +44,9 @@
# 'agentId': self.agent_id,
'mode': self.mode,
}
+
+ def get_kb_ids(self):
+ return json.loads(self.kb_ids) if self.kb_ids else []
class ConversationModel(Base):
@@ -82,6 +88,8 @@
id: str
status: Optional[str] = "1"
icon: Optional[str] = ""
+ name: Optional[str] = ""
+ description: Optional[str] = None
class dialogList(BaseModel):
--
Gitblit v1.8.0