From ee8df9424fc4e3fb30a8b8e39436b26c902defb7 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 24 十二月 2024 13:37:15 +0800
Subject: [PATCH] 角色的资源过滤掉删除的

---
 app/models/group_model.py |   39 ++++++++++++++++++++-------------------
 1 files changed, 20 insertions(+), 19 deletions(-)

diff --git a/app/models/group_model.py b/app/models/group_model.py
index bb49da7..d5135b2 100644
--- a/app/models/group_model.py
+++ b/app/models/group_model.py
@@ -9,9 +9,9 @@
 from app.models.base_model import Base
 
 
-group_knowledge_table = Table('group_Knowledge', Base.metadata
+group_knowledge_table = Table('group_knowledge', Base.metadata
                            , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE'))
-                           , Column('knowledge_id', String(32), ForeignKey('knowledgebase.id', ondelete='CASCADE')))
+                           , Column('knowledge_id', String(32), ForeignKey('knowledgebases.id', ondelete='CASCADE')))
 
 group_dialog_table = Table('group_dialogs', 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',
@@ -82,29 +83,29 @@
 
     def to_json(self):
         json = {
-            'roleId': self.id,
+            'groupId': self.id,
             'createTime': self.created_at.strftime('%Y-%m-%d %H:%M:%S'),
             'updateTime': self.updated_at.strftime('%Y-%m-%d %H:%M:%S'),
             'groupName': self.group_name,
-            'remark': self.group_description,
+            'description': self.group_description,
             'status': self.status,
         }
-        if len(self.knowledges.all()) > 0:
-            json['knowledges'] = [knowledge.to_json() for knowledge in self.knowledges]
+        # if len(self.knowledges.all()) > 0:
+        json['knowledges'] = [knowledge.to_json() for knowledge in self.knowledges]
 
-        if len(self.dialogs.all()) > 0:
-            json['dialogs'] = [dialog.to_json() for dialog in self.dialogs]
+        # if len(self.dialogs.all()) > 0:
+        json['dialogs'] = [dialog.to_json() for dialog in self.dialogs]
 
-        if len(self.agents.all()) > 0:
-            json['agents'] = [agent.to_json() for agent in self.agents]
+        # if len(self.agents.all()) > 0:
+        #     json['agents'] = [agent.to_json() for agent in self.agents]
 
-        if len(self.llms.all()) > 0:
-            json['llms'] = [llm.to_json() for llm in self.llms]
+        # if len(self.llms.all()) > 0:
+        json['llms'] = [llm.to_json() for llm in self.llms]
 
         json['users'] = [user.to_dict() for user in self.users]
 
-        if hasattr(self, 'flag'):
-            json['flag'] = self.flag
+        # if hasattr(self, 'flag'):
+        #     json['flag'] = self.flag
 
         return json
 
@@ -119,18 +120,18 @@
 
 class GroupData(BaseModel):
     id: Optional[int] = None
-    group_name: Optional[str] = ""
-    group_description: Optional[str] = ""
-    group_status: Optional[str] = ""
+    groupName: Optional[str] = ""
+    description: Optional[str] = ""
+    status: Optional[str] = ""
 
 class GroupUsers(BaseModel):
     id: int
-    user_list: list
+    userList: list
 
 
 class GroupResource(BaseModel):
     id: int
-    agents: list
+    llms: list
     dialogs: list
     knowledges: list
 

--
Gitblit v1.8.0