chore: format

2026-01-08 01:55:56 +04:00
parent c417fdd94d
commit 700349064d
97 changed files with 4650 additions and 986 deletions
@@ -117,7 +117,9 @@ async def get_channels(
        last_message = Messages.get_last_message_by_channel_id(channel.id, db=db)
        last_message_at = last_message.created_at if last_message else None

-        channel_member = Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
+        channel_member = Channels.get_member_by_channel_and_user_id(
+            channel.id, user.id, db=db
+        )
        unread_count = (
            Messages.get_unread_message_count(
                channel.id, user.id, channel_member.last_read_at, db=db
@@ -135,7 +137,10 @@ async def get_channels(
            ]
            users = [
                UserIdNameStatusResponse(
-                    **{**user.model_dump(), "is_active": Users.is_user_active(user.id, db=db)}
+                    **{
+                        **user.model_dump(),
+                        "is_active": Users.is_user_active(user.id, db=db),
+                    }
                )
                for user in Users.get_users_by_user_ids(user_ids, db=db)
            ]
@@ -187,11 +192,15 @@ async def get_dm_channel_by_user_id(
        )

    try:
-        existing_channel = Channels.get_dm_channel_by_user_ids([user.id, user_id], db=db)
+        existing_channel = Channels.get_dm_channel_by_user_ids(
+            [user.id, user_id], db=db
+        )
        if existing_channel:
            participant_ids = [
                member.user_id
-                for member in Channels.get_members_by_channel_id(existing_channel.id, db=db)
+                for member in Channels.get_members_by_channel_id(
+                    existing_channel.id, db=db
+                )
            ]

            await emit_to_users(
@@ -203,7 +212,9 @@ async def get_dm_channel_by_user_id(
                f"channel:{existing_channel.id}", participant_ids
            )

-            Channels.update_member_active_status(existing_channel.id, user.id, True, db=db)
+            Channels.update_member_active_status(
+                existing_channel.id, user.id, True, db=db
+            )
            return ChannelModel(**existing_channel.model_dump())

        channel = Channels.insert_new_channel(
@@ -288,7 +299,9 @@ async def create_new_channel(
                    f"channel:{existing_channel.id}", participant_ids
                )

-                Channels.update_member_active_status(existing_channel.id, user.id, True, db=db)
+                Channels.update_member_active_status(
+                    existing_channel.id, user.id, True, db=db
+                )
                return ChannelModel(**existing_channel.model_dump())

        channel = Channels.insert_new_channel(form_data, user.id, db=db)
@@ -353,17 +366,23 @@ async def get_channel_by_id(
            )

        user_ids = [
-            member.user_id for member in Channels.get_members_by_channel_id(channel.id, db=db)
+            member.user_id
+            for member in Channels.get_members_by_channel_id(channel.id, db=db)
        ]

        users = [
            UserIdNameStatusResponse(
-                **{**user.model_dump(), "is_active": Users.is_user_active(user.id, db=db)}
+                **{
+                    **user.model_dump(),
+                    "is_active": Users.is_user_active(user.id, db=db),
+                }
            )
            for user in Users.get_users_by_user_ids(user_ids, db=db)
        ]

-        channel_member = Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
+        channel_member = Channels.get_member_by_channel_and_user_id(
+            channel.id, user.id, db=db
+        )
        unread_count = Messages.get_unread_message_count(
            channel.id, user.id, channel_member.last_read_at if channel_member else None
        )
@@ -373,7 +392,9 @@ async def get_channel_by_id(
                **channel.model_dump(),
                "user_ids": user_ids,
                "users": users,
-                "is_manager": Channels.is_user_channel_manager(channel.id, user.id, db=db),
+                "is_manager": Channels.is_user_channel_manager(
+                    channel.id, user.id, db=db
+                ),
                "write_access": True,
                "user_count": len(user_ids),
                "last_read_at": channel_member.last_read_at if channel_member else None,
@@ -389,12 +410,18 @@ async def get_channel_by_id(
            )

        write_access = has_access(
-            user.id, type="write", access_control=channel.access_control, strict=False, db=db
+            user.id,
+            type="write",
+            access_control=channel.access_control,
+            strict=False,
+            db=db,
        )

        user_count = len(get_users_with_access("read", channel.access_control))

-        channel_member = Channels.get_member_by_channel_and_user_id(channel.id, user.id, db=db)
+        channel_member = Channels.get_member_by_channel_and_user_id(
+            channel.id, user.id, db=db
+        )
        unread_count = Messages.get_unread_message_count(
            channel.id, user.id, channel_member.last_read_at if channel_member else None
        )
@@ -404,7 +431,9 @@ async def get_channel_by_id(
                **channel.model_dump(),
                "user_ids": user_ids,
                "users": users,
-                "is_manager": Channels.is_user_channel_manager(channel.id, user.id, db=db),
+                "is_manager": Channels.is_user_channel_manager(
+                    channel.id, user.id, db=db
+                ),
                "write_access": write_access or user.role == "admin",
                "user_count": user_count,
                "last_read_at": channel_member.last_read_at if channel_member else None,
@@ -453,7 +482,8 @@ async def get_channel_members_by_id(

    if channel.type == "dm":
        user_ids = [
-            member.user_id for member in Channels.get_members_by_channel_id(channel.id, db=db)
+            member.user_id
+            for member in Channels.get_members_by_channel_id(channel.id, db=db)
        ]
        users = Users.get_users_by_user_ids(user_ids, db=db)
        total = len(users)
@@ -533,7 +563,9 @@ async def update_is_active_member_by_id_and_user_id(
            status_code=status.HTTP_404_NOT_FOUND, detail=ERROR_MESSAGES.NOT_FOUND
        )

-    Channels.update_member_active_status(channel.id, user.id, form_data.is_active, db=db)
+    Channels.update_member_active_status(
+        channel.id, user.id, form_data.is_active, db=db
+    )
    return True


@@ -626,7 +658,9 @@ async def remove_members_by_id(
        )

    try:
-        deleted = Channels.remove_members_from_channel(channel.id, form_data.user_ids, db=db)
+        deleted = Channels.remove_members_from_channel(
+            channel.id, form_data.user_ids, db=db
+        )

        return deleted
    except Exception as e:
@@ -794,7 +828,9 @@ async def get_channel_messages(
                    **message.model_dump(),
                    "reply_count": len(thread_replies),
                    "latest_reply_at": latest_thread_reply_at,
-                    "reactions": Messages.get_reactions_by_message_id(message.id, db=db),
+                    "reactions": Messages.get_reactions_by_message_id(
+                        message.id, db=db
+                    ),
                    "user": UserNameResponse(**users[message.user_id].model_dump()),
                }
            )
@@ -857,7 +893,9 @@ async def get_pinned_channel_messages(
            MessageWithReactionsResponse(
                **{
                    **message.model_dump(),
-                    "reactions": Messages.get_reactions_by_message_id(message.id, db=db),
+                    "reactions": Messages.get_reactions_by_message_id(
+                        message.id, db=db
+                    ),
                    "user": UserNameResponse(**users[message.user_id].model_dump()),
                }
            )
@@ -871,7 +909,9 @@ async def get_pinned_channel_messages(
 ############################


-async def send_notification(name, webui_url, channel, message, active_user_ids, db=None):
+async def send_notification(
+    name, webui_url, channel, message, active_user_ids, db=None
+):
    users = get_users_with_access("read", channel.access_control)

    for user in users:
@@ -966,7 +1006,9 @@ async def model_response_handler(request, channel, message, user, db=None):
                for thread_message in thread_messages:
                    message_user = None
                    if thread_message.user_id not in message_users:
-                        message_user = Users.get_user_by_id(thread_message.user_id, db=db)
+                        message_user = Users.get_user_by_id(
+                            thread_message.user_id, db=db
+                        )
                        message_users[thread_message.user_id] = message_user
                    else:
                        message_user = message_users[thread_message.user_id]
@@ -1098,7 +1140,11 @@ async def new_message_handler(
            )
    else:
        if user.role != "admin" and not has_access(
-            user.id, type="write", access_control=channel.access_control, strict=False, db=db
+            user.id,
+            type="write",
+            access_control=channel.access_control,
+            strict=False,
+            db=db,
        ):
            raise HTTPException(
                status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT()
@@ -1417,7 +1463,9 @@ async def get_channel_thread_messages(
                status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT()
            )

-    message_list = Messages.get_messages_by_parent_id(id, message_id, skip, limit, db=db)
+    message_list = Messages.get_messages_by_parent_id(
+        id, message_id, skip, limit, db=db
+    )

    if not message_list:
        return []
@@ -1434,7 +1482,9 @@ async def get_channel_thread_messages(
                    **message.model_dump(),
                    "reply_count": 0,
                    "latest_reply_at": None,
-                    "reactions": Messages.get_reactions_by_message_id(message.id, db=db),
+                    "reactions": Messages.get_reactions_by_message_id(
+                        message.id, db=db
+                    ),
                    "user": UserNameResponse(**users[message.user_id].model_dump()),
                }
            )
@@ -1554,7 +1604,11 @@ async def add_reaction_to_message(
            )
    else:
        if user.role != "admin" and not has_access(
-            user.id, type="write", access_control=channel.access_control, strict=False, db=db
+            user.id,
+            type="write",
+            access_control=channel.access_control,
+            strict=False,
+            db=db,
        ):
            raise HTTPException(
                status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT()
@@ -1629,7 +1683,11 @@ async def remove_reaction_by_id_and_user_id_and_name(
            )
    else:
        if user.role != "admin" and not has_access(
-            user.id, type="write", access_control=channel.access_control, strict=False, db=db
+            user.id,
+            type="write",
+            access_control=channel.access_control,
+            strict=False,
+            db=db,
        ):
            raise HTTPException(
                status_code=status.HTTP_403_FORBIDDEN, detail=ERROR_MESSAGES.DEFAULT()
@@ -383,7 +383,7 @@ def calculate_chat_stats(
 def generate_chat_stats_jsonl_generator(user_id, filter):
    """
    Synchronous generator for streaming chat stats export.
-    
+
    NOTE: We intentionally do NOT pass a shared db session here. Instead, we let
    each batch create its own short-lived session via get_db_context(None).
    This is critical for SQLite in low-resource environments because:
@@ -18,8 +18,6 @@ log = logging.getLogger(__name__)
 router = APIRouter()


-
-
@router.get("/ef")
 async def get_embeddings(request: Request):
    return {"result": await request.app.state.EMBEDDING_FUNCTION("hello world")}
@@ -32,7 +30,9 @@ async def get_embeddings(request: Request):

@router.get("/", response_model=list[MemoryModel])
 async def get_memories(
-    request: Request, user=Depends(get_verified_user), db: Session = Depends(get_session)
+    request: Request,
+    user=Depends(get_verified_user),
+    db: Session = Depends(get_session),
 ):
    if not request.app.state.config.ENABLE_MEMORIES:
        raise HTTPException(
@@ -40,7 +40,9 @@ async def get_memories(
            detail=ERROR_MESSAGES.NOT_FOUND,
        )

-    if not has_permission(user.id, "features.memories", request.app.state.config.USER_PERMISSIONS):
+    if not has_permission(
+        user.id, "features.memories", request.app.state.config.USER_PERMISSIONS
+    ):
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -75,7 +77,9 @@ async def add_memory(
            detail=ERROR_MESSAGES.NOT_FOUND,
        )

-    if not has_permission(user.id, "features.memories", request.app.state.config.USER_PERMISSIONS):
+    if not has_permission(
+        user.id, "features.memories", request.app.state.config.USER_PERMISSIONS
+    ):
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -112,7 +116,10 @@ class QueryMemoryForm(BaseModel):

@router.post("/query")
 async def query_memory(
-    request: Request, form_data: QueryMemoryForm, user=Depends(get_verified_user), db: Session = Depends(get_session)
+    request: Request,
+    form_data: QueryMemoryForm,
+    user=Depends(get_verified_user),
+    db: Session = Depends(get_session),
 ):
    if not request.app.state.config.ENABLE_MEMORIES:
        raise HTTPException(
@@ -120,7 +127,9 @@ async def query_memory(
            detail=ERROR_MESSAGES.NOT_FOUND,
        )

-    if not has_permission(user.id, "features.memories", request.app.state.config.USER_PERMISSIONS):
+    if not has_permission(
+        user.id, "features.memories", request.app.state.config.USER_PERMISSIONS
+    ):
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -146,7 +155,9 @@ async def query_memory(
 ############################
@router.post("/reset", response_model=bool)
 async def reset_memory_from_vector_db(
-    request: Request, user=Depends(get_verified_user), db: Session = Depends(get_session)
+    request: Request,
+    user=Depends(get_verified_user),
+    db: Session = Depends(get_session),
 ):
    if not request.app.state.config.ENABLE_MEMORIES:
        raise HTTPException(
@@ -154,12 +165,14 @@ async def reset_memory_from_vector_db(
            detail=ERROR_MESSAGES.NOT_FOUND,
        )

-    if not has_permission(user.id, "features.memories", request.app.state.config.USER_PERMISSIONS):
+    if not has_permission(
+        user.id, "features.memories", request.app.state.config.USER_PERMISSIONS
+    ):
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
        )
-        
+
    VECTOR_DB_CLIENT.delete_collection(f"user-memory-{user.id}")

    memories = Memories.get_memories_by_user_id(user.id, db=db)
@@ -198,7 +211,9 @@ async def reset_memory_from_vector_db(

@router.delete("/delete/user", response_model=bool)
 async def delete_memory_by_user_id(
-    request: Request, user=Depends(get_verified_user), db: Session = Depends(get_session)
+    request: Request,
+    user=Depends(get_verified_user),
+    db: Session = Depends(get_session),
 ):
    if not request.app.state.config.ENABLE_MEMORIES:
        raise HTTPException(
@@ -206,7 +221,9 @@ async def delete_memory_by_user_id(
            detail=ERROR_MESSAGES.NOT_FOUND,
        )

-    if not has_permission(user.id, "features.memories", request.app.state.config.USER_PERMISSIONS):
+    if not has_permission(
+        user.id, "features.memories", request.app.state.config.USER_PERMISSIONS
+    ):
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -243,7 +260,9 @@ async def update_memory_by_id(
            detail=ERROR_MESSAGES.NOT_FOUND,
        )

-    if not has_permission(user.id, "features.memories", request.app.state.config.USER_PERMISSIONS):
+    if not has_permission(
+        user.id, "features.memories", request.app.state.config.USER_PERMISSIONS
+    ):
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -283,7 +302,10 @@ async def update_memory_by_id(

@router.delete("/{memory_id}", response_model=bool)
 async def delete_memory_by_id(
-    memory_id: str, request: Request, user=Depends(get_verified_user), db: Session = Depends(get_session)
+    memory_id: str,
+    request: Request,
+    user=Depends(get_verified_user),
+    db: Session = Depends(get_session),
 ):
    if not request.app.state.config.ENABLE_MEMORIES:
        raise HTTPException(
@@ -291,7 +313,9 @@ async def delete_memory_by_id(
            detail=ERROR_MESSAGES.NOT_FOUND,
        )

-    if not has_permission(user.id, "features.memories", request.app.state.config.USER_PERMISSIONS):
+    if not has_permission(
+        user.id, "features.memories", request.app.state.config.USER_PERMISSIONS
+    ):
        raise HTTPException(
            status_code=status.HTTP_403_FORBIDDEN,
            detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -1302,7 +1302,10 @@ async def generate_chat_completion(
            if not (
                user.id == model_info.user_id
                or has_access(
-                    user.id, type="read", access_control=model_info.access_control, db=db
+                    user.id,
+                    type="read",
+                    access_control=model_info.access_control,
+                    db=db,
                )
            ):
                raise HTTPException(
@@ -1409,7 +1412,10 @@ async def generate_openai_completion(
            if not (
                user.id == model_info.user_id
                or has_access(
-                    user.id, type="read", access_control=model_info.access_control, db=db
+                    user.id,
+                    type="read",
+                    access_control=model_info.access_control,
+                    db=db,
                )
            ):
                raise HTTPException(
@@ -1493,7 +1499,10 @@ async def generate_openai_chat_completion(
            if not (
                user.id == model_info.user_id
                or has_access(
-                    user.id, type="read", access_control=model_info.access_control, db=db
+                    user.id,
+                    type="read",
+                    access_control=model_info.access_control,
+                    db=db,
                )
            ):
                raise HTTPException(
@@ -1592,7 +1601,10 @@ async def get_openai_models(
            model_info = Models.get_model_by_id(model["id"], db=db)
            if model_info:
                if user.id == model_info.user_id or has_access(
-                    user.id, type="read", access_control=model_info.access_control, db=db
+                    user.id,
+                    type="read",
+                    access_control=model_info.access_control,
+                    db=db,
                ):
                    filtered_models.append(model)
        models = filtered_models
@@ -837,7 +837,10 @@ async def generate_chat_completion(
            if not (
                user.id == model_info.user_id
                or has_access(
-                    user.id, type="read", access_control=model_info.access_control, db=db
+                    user.id,
+                    type="read",
+                    access_control=model_info.access_control,
+                    db=db,
                )
            ):
                raise HTTPException(
@@ -23,7 +23,9 @@ router = APIRouter()


@router.get("/", response_model=list[PromptModel])
-async def get_prompts(user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_prompts(
+    user=Depends(get_verified_user), db: Session = Depends(get_session)
+):
    if user.role == "admin" and BYPASS_ADMIN_ACCESS_CONTROL:
        prompts = Prompts.get_prompts(db=db)
    else:
@@ -33,7 +35,9 @@ async def get_prompts(user=Depends(get_verified_user), db: Session = Depends(get


@router.get("/list", response_model=list[PromptAccessResponse])
-async def get_prompt_list(user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_prompt_list(
+    user=Depends(get_verified_user), db: Session = Depends(get_session)
+):
    if user.role == "admin" and BYPASS_ADMIN_ACCESS_CONTROL:
        prompts = Prompts.get_prompts(db=db)
    else:
@@ -59,11 +63,17 @@ async def get_prompt_list(user=Depends(get_verified_user), db: Session = Depends

@router.post("/create", response_model=Optional[PromptModel])
 async def create_new_prompt(
-    request: Request, form_data: PromptForm, user=Depends(get_verified_user), db: Session = Depends(get_session)
+    request: Request,
+    form_data: PromptForm,
+    user=Depends(get_verified_user),
+    db: Session = Depends(get_session),
 ):
    if user.role != "admin" and not (
        has_permission(
-            user.id, "workspace.prompts", request.app.state.config.USER_PERMISSIONS, db=db
+            user.id,
+            "workspace.prompts",
+            request.app.state.config.USER_PERMISSIONS,
+            db=db,
        )
        or has_permission(
            user.id,
@@ -99,7 +109,9 @@ async def create_new_prompt(


@router.get("/command/{command}", response_model=Optional[PromptAccessResponse])
-async def get_prompt_by_command(command: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def get_prompt_by_command(
+    command: str, user=Depends(get_verified_user), db: Session = Depends(get_session)
+):
    prompt = Prompts.get_prompt_by_command(f"/{command}", db=db)

    if prompt:
@@ -169,7 +181,9 @@ async def update_prompt_by_command(


@router.delete("/command/{command}/delete", response_model=bool)
-async def delete_prompt_by_command(command: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
+async def delete_prompt_by_command(
+    command: str, user=Depends(get_verified_user), db: Session = Depends(get_session)
+):
    prompt = Prompts.get_prompt_by_command(f"/{command}", db=db)
    if not prompt:
        raise HTTPException(