mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-25 16:56:22 +02:00
0e723a5b8b
- improved search_knowledgebase_tool - Added new endpoint to batch-fetch comments for multiple messages, reducing the number of API calls. - Introduced CommentBatchRequest and CommentBatchResponse schemas for handling batch requests and responses. - Updated chat_comments_service to validate message existence and permissions before fetching comments. - Enhanced frontend with useBatchCommentsPreload hook to optimize comment loading for assistant messages.
108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
"""
|
|
Routes for chat comments and mentions.
|
|
"""
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.db import User, get_async_session
|
|
from app.schemas.chat_comments import (
|
|
CommentBatchRequest,
|
|
CommentBatchResponse,
|
|
CommentCreateRequest,
|
|
CommentListResponse,
|
|
CommentReplyResponse,
|
|
CommentResponse,
|
|
CommentUpdateRequest,
|
|
MentionListResponse,
|
|
)
|
|
from app.services.chat_comments_service import (
|
|
create_comment,
|
|
create_reply,
|
|
delete_comment,
|
|
get_comments_for_message,
|
|
get_comments_for_messages_batch,
|
|
get_user_mentions,
|
|
update_comment,
|
|
)
|
|
from app.users import current_active_user
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/messages/comments/batch", response_model=CommentBatchResponse)
|
|
async def batch_list_comments(
|
|
request: CommentBatchRequest,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
"""Batch-fetch comments for multiple messages in one request."""
|
|
return await get_comments_for_messages_batch(session, request.message_ids, user)
|
|
|
|
|
|
@router.get("/messages/{message_id}/comments", response_model=CommentListResponse)
|
|
async def list_comments(
|
|
message_id: int,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
"""List all comments for a message with their replies."""
|
|
return await get_comments_for_message(session, message_id, user)
|
|
|
|
|
|
@router.post("/messages/{message_id}/comments", response_model=CommentResponse)
|
|
async def add_comment(
|
|
message_id: int,
|
|
request: CommentCreateRequest,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
"""Create a top-level comment on an AI response."""
|
|
return await create_comment(session, message_id, request.content, user)
|
|
|
|
|
|
@router.post("/comments/{comment_id}/replies", response_model=CommentReplyResponse)
|
|
async def add_reply(
|
|
comment_id: int,
|
|
request: CommentCreateRequest,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
"""Reply to an existing comment."""
|
|
return await create_reply(session, comment_id, request.content, user)
|
|
|
|
|
|
@router.put("/comments/{comment_id}", response_model=CommentReplyResponse)
|
|
async def edit_comment(
|
|
comment_id: int,
|
|
request: CommentUpdateRequest,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
"""Update a comment's content (author only)."""
|
|
return await update_comment(session, comment_id, request.content, user)
|
|
|
|
|
|
@router.delete("/comments/{comment_id}")
|
|
async def remove_comment(
|
|
comment_id: int,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
"""Delete a comment (author or user with COMMENTS_DELETE permission)."""
|
|
return await delete_comment(session, comment_id, user)
|
|
|
|
|
|
# =============================================================================
|
|
# Mention Routes
|
|
# =============================================================================
|
|
|
|
|
|
@router.get("/mentions", response_model=MentionListResponse)
|
|
async def list_mentions(
|
|
search_space_id: int | None = None,
|
|
session: AsyncSession = Depends(get_async_session),
|
|
user: User = Depends(current_active_user),
|
|
):
|
|
"""List mentions for the current user."""
|
|
return await get_user_mentions(session, user, search_space_id)
|