From de45d8dc19cd16b85344a3c3bbb243a12e8811a0 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 13 十一月 2024 15:16:59 +0800
Subject: [PATCH] 增加功能

---
 app/models/group_model.py |   25 +++++++++++++------------
 1 files changed, 13 insertions(+), 12 deletions(-)

diff --git a/app/models/group_model.py b/app/models/group_model.py
index 52e05a8..b9c30e9 100644
--- a/app/models/group_model.py
+++ b/app/models/group_model.py
@@ -10,24 +10,24 @@
 
 
 group_knowledge_table = Table('group_Knowledge', Base.metadata
-                           , Column('group_id', Integer, ForeignKey('group.id'))
-                           , Column('knowledge_id', String(32), ForeignKey('knowledgebase.id')))
+                           , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE'))
+                           , Column('knowledge_id', String(32), ForeignKey('knowledgebase.id', ondelete='CASCADE')))
 
 group_dialog_table = Table('group_dialogs', Base.metadata
-                           , Column('group_id', Integer, ForeignKey('group.id'))
-                           , Column('dialog_id', String(36), ForeignKey('dialogs.id')))
+                           , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE'))
+                           , Column('dialog_id', String(36), ForeignKey('dialogs.id', ondelete='CASCADE')))
 
 group_agent_table = Table('group_agent', Base.metadata
-                           , Column('group_id', Integer, ForeignKey('group.id'))
-                           , Column('agent_id', String(36), ForeignKey('user_canvas.id')))
+                           , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE'))
+                           , Column('agent_id', String(36), ForeignKey('user_canvas.id', ondelete='CASCADE')))
 
 group_llm_table = Table('group_llm', Base.metadata
-                           , Column('group_id', Integer, ForeignKey('group.id'))
-                           , Column('llm_id', String(36), ForeignKey('common_llm.id')))
+                           , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE'))
+                           , Column('llm_id', String(36), ForeignKey('common_llm.id', ondelete='CASCADE')))
 
 group_unified_agent_table = Table('group_unified_agent', Base.metadata
-                                    , Column('group_id', Integer, ForeignKey('group.id'))
-                                    ,Column('unified_agent_id', String(36), ForeignKey('unified_agent.id')))
+                                    , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE'))
+                                    ,Column('unified_agent_id', String(36), ForeignKey('unified_agent.id', ondelete='CASCADE')))
 
 
 class GroupModel(Base):
@@ -38,6 +38,7 @@
     status = Column(String(10), nullable=False, default="1")
     created_at = Column(DateTime, default=datetime.now())
     updated_at = Column(DateTime, default=datetime.now(), onupdate=datetime.now())
+    creator = Column(Integer)
 
 
     knowledges = relationship('KnowledgeModel',
@@ -79,8 +80,8 @@
 
 
     def __repr__(self):
-        return '<Role name:%r description:%r iconCls:%r seq:%r>\n' \
-            % (self.NAME, self.DESCRIPTION, self.ICONCLS, self.SEQ)
+        return '<Role name:%r description:%r>\n' \
+            % (self.group_name, self.group_description)
 
     def to_json(self):
         json = {

--
Gitblit v1.8.0