| | |
| | | from app.config.config import settings |
| | | from app.models import CanvasModel, DialogModel, KnowledgeModel |
| | | from app.models.group_model import GroupModel |
| | | from app.models.user_model import UserModel |
| | | from app.service.ragflow import RagflowService |
| | |
| | | |
| | | async def group_list(db, page_size: int, page_index: int, keyword: str, role_key:str, user_id): |
| | | logger.info("-------------------------group_list----------------------------------") |
| | | query = db.query(GroupModel) |
| | | if role_key != "admin": |
| | | query.filter(GroupModel.creator==user_id) |
| | | query = db.query(GroupModel).filter(GroupModel.creator==user_id) |
| | | # if role_key != "admin": |
| | | # query.filter(GroupModel.creator==user_id) |
| | | if keyword: |
| | | query = query.filter(GroupModel.group_name.like('%{}%'.format(keyword))) |
| | | items = query.order_by(GroupModel.id.desc()).limit(page_size).offset( |
| | |
| | | except Exception as e: |
| | | logger.error(e) |
| | | return False |
| | | return True |
| | | |
| | | async def save_group_resource(db, user_id, group_id, agents, dialogs, knowledges): |
| | | try: |
| | | group = db.query(GroupModel).filter(GroupModel.id.__eq__(group_id)).first() |
| | | group.agents = [db.get(CanvasModel, user) for user in agents] |
| | | group.dialogs = [db.get(DialogModel, user) for user in dialogs] |
| | | group.knowledges = [db.get(KnowledgeModel, user) for user in knowledges] |
| | | db.commit() |
| | | except Exception as e: |
| | | logger.error(e) |
| | | return False |
| | | return True |