From abb91124a4372b0efe5ab1b7aa25859c635d30eb Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期四, 10 四月 2025 13:49:37 +0800 Subject: [PATCH] merge --- app/models/dialog_model.py | 9 ++++++++- 1 files changed, 8 insertions(+), 1 deletions(-) diff --git a/app/models/dialog_model.py b/app/models/dialog_model.py index 1ab181f..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' @@ -50,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)) # 鍚嶇О @@ -83,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