From f2c43713b16e821f4a1ae97804c60f18734d5f6a Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期五, 13 十二月 2024 17:25:12 +0800
Subject: [PATCH] 问题优化

---
 app/models/group_model.py |   14 ++++++--------
 1 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/app/models/group_model.py b/app/models/group_model.py
index dda330a..529db57 100644
--- a/app/models/group_model.py
+++ b/app/models/group_model.py
@@ -17,9 +17,9 @@
                            , 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
+group_agent_table = Table('group_canvas', Base.metadata
                            , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE'))
-                           , Column('agent_id', String(36), ForeignKey('canvas.id', ondelete='CASCADE')))
+                           , Column('canvas_id', String(36), ForeignKey('canvas.id', ondelete='CASCADE')))
 
 group_llm_table = Table('group_llm', Base.metadata
                            , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE'))
@@ -39,6 +39,7 @@
     created_at = Column(DateTime, default=datetime.now())
     updated_at = Column(DateTime, default=datetime.now(), onupdate=datetime.now())
     creator = Column(Integer)
+    group_type = Column(Integer, default=1)
 
 
     knowledges = relationship('KnowledgeModel',
@@ -53,18 +54,15 @@
 
     agents = relationship('CanvasModel',
                              secondary=group_agent_table,
-                             backref=backref('groups', lazy='dynamic'),
-                             lazy="dynamic")
+                             backref=backref('groups', lazy='dynamic'))
 
     llms = relationship('CommonLlmModel',
                            secondary=group_llm_table,
-                           backref=backref('groups', lazy='dynamic'),
-                           lazy="dynamic")
+                           backref=backref('groups', lazy='dynamic'))
 
     unified_agents = relationship('UnifiedAgentModel',
                                      secondary=group_unified_agent_table,
-                                     backref=backref('roles', lazy='dynamic'),
-                                     lazy="dynamic")
+                                     backref=backref('groups', lazy='dynamic'))
     def to_dict(self):
         return {
             'groupId': self.id,

--
Gitblit v1.8.0