From 0650b889a36d9b9fd42415b9b9819676f839ae9b Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 08 四月 2025 09:54:09 +0800
Subject: [PATCH] 首页会话-模型修改
---
app/models/dialog_model.py | 10 +++++++++-
1 files changed, 9 insertions(+), 1 deletions(-)
diff --git a/app/models/dialog_model.py b/app/models/dialog_model.py
index 4a6a29e..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)
@@ -42,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'
@@ -49,7 +55,7 @@
create_date = Column(DateTime) # 鍒涘缓鏃堕棿
create_time = Column(BigInteger)
update_date = Column(DateTime) # 鏇存柊鏃堕棿
- update_time = Column(BigInteger)
+ update_time = Column(BigInteger, index=True)
tenant_id = Column(Integer) # 鍒涘缓浜�
dialog_id = Column(String(32))
name = Column(String(255)) # 鍚嶇О
@@ -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