From f64f37ba5f1c4702c73bd5f825f6cdf17cd9e493 Mon Sep 17 00:00:00 2001 From: zhangqian <zhangqian@123.com> Date: 星期四, 05 十二月 2024 15:13:41 +0800 Subject: [PATCH] 配置ip取环境变量的 --- app/models/group_model.py | 25 +++++++++++++++---------- 1 files changed, 15 insertions(+), 10 deletions(-) diff --git a/app/models/group_model.py b/app/models/group_model.py index a1ac287..bb49da7 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('user_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')) @@ -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', @@ -52,18 +53,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, @@ -79,8 +77,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 = { @@ -129,3 +127,10 @@ id: int user_list: list + +class GroupResource(BaseModel): + id: int + agents: list + dialogs: list + knowledges: list + -- Gitblit v1.8.0