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 | 22 ++++++++++++++++++++--
1 files changed, 20 insertions(+), 2 deletions(-)
diff --git a/app/models/dialog_model.py b/app/models/dialog_model.py
index 058581c..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)) # 鍚嶇О
@@ -80,4 +86,16 @@
class dialogDataUpdate(BaseModel):
id: str
- status: Optional[str] = "1"
\ No newline at end of file
+ status: Optional[str] = "1"
+ icon: Optional[str] = ""
+ name: Optional[str] = ""
+ description: Optional[str] = None
+
+
+class dialogList(BaseModel):
+ current: Optional[int] = 1
+ pageSize: Optional[int] = 10
+ keyword: Optional[str] = ""
+ label: Optional[list] = []
+ status: Optional[str] = ""
+ mode: Optional[str] = ""
\ No newline at end of file
--
Gitblit v1.8.0