From afc68af3a67b2f936429feb3a68c1f09e03ce930 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 11 四月 2025 10:07:24 +0800
Subject: [PATCH] 智能体中心不显示自研智能体
---
app/models/dialog_model.py | 39 ++++++++++++++++++++++++++++++---------
1 files changed, 30 insertions(+), 9 deletions(-)
diff --git a/app/models/dialog_model.py b/app/models/dialog_model.py
index 1934388..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
@@ -12,16 +13,19 @@
__mapper_args__ = {
# "order_by": 'SEQ'
}
- id = Column(String(32), primary_key=True) # id
+ id = Column(String(36), primary_key=True) # id
create_date = Column(DateTime, default=datetime.now()) # 鍒涘缓鏃堕棿
update_date = Column(DateTime, default=datetime.now(), onupdate=datetime.now()) # 鏇存柊鏃堕棿
- tenant_id = Column(String(32)) # 鍒涘缓浜�
+ tenant_id = Column(String(36)) # 鍒涘缓浜�
name = Column(String(255)) # 鍚嶇О
description = Column(Text) # 璇存槑
- icon = Column(Text) # 鍥炬爣
- status = Column(String(1), default="1") # 鐘舵��
- dialog_type = Column(String(1)) # # 骞冲彴
- agent_id = Column(String(32))
+ icon = Column(Text, default="intelligentFrame1") # 鍥炬爣
+ status = Column(String(1), default="0") # 鐘舵��
+ 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)
@@ -37,8 +41,12 @@
'icon': self.icon,
'status': self.status,
'agentType': self.dialog_type,
- 'agentId': self.agent_id,
+ # '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):
@@ -47,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)) # 鍚嶇О
@@ -73,8 +81,21 @@
description: Optional[str] = ""
dialogType: str
icon: str
+ mode: str
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