Collection

create_collection(request, requesting_user=Depends(get_current_user)) async

Create a private collection and provision its backing vector index (Qdrant).

The new collection is private to its creator. The MongoDB collection ID is used as the Qdrant collection name.

Parameters:

Name	Type	Description	Default
request	CollectionRequest	New collection parameters.	required
requesting_user	User	Authenticated user injected by dependency.	Depends(get_current_user)

Returns:

Type	Description
Collection	Created collection.

Raises:

Type	Description
HTTPException	500 if vector collection creation fails or on server errors.

Source code in routers/collection.py

@router.post("/collections", response_model=Collection)
async def create_collection(
    request: CollectionRequest,
    requesting_user: User = Depends(get_current_user),
) -> Collection:
    """
    Create a private collection and provision its backing vector index (Qdrant).

    The new collection is private to its creator. The MongoDB collection ID is used as the Qdrant collection name.

    Args:
        request (CollectionRequest): New collection parameters.
        requesting_user (User): Authenticated user injected by dependency.

    Returns:
        Created collection.

    Raises:
        HTTPException: 500 if vector collection creation fails or on server errors.
    """
    collection = Collection(
        name=request.name,
        user_id=requesting_user.id,
        description=request.description,
        embeddings_model=request.embeddings_model,
    )
    await collection.save()

    try:
        VectorStoreManager(embeddings_model=request.embeddings_model).create_collection(
            collection.id
        )
    except HTTPException as e:
        raise e
    except Exception as e:
        logger.warning(
            f"Warning: failed to create Qdrant collection {collection.id}: {e}"
        )
        await collection.delete()
        raise HTTPException(
            status_code=500,
            detail=f"Failed to create vector collection: {str(e)}",
        )

    return collection

delete_collection(collection_id, requesting_user=Depends(get_current_user)) async

Delete a collection and its related resources.

Deletes documents in the collection, attempts to remove the vector index, and finally deletes the collection record.

Parameters:

Name	Type	Description	Default
collection_id	str	Collection identifier.	required
requesting_user	User	Authenticated user injected by dependency.	Depends(get_current_user)

Returns:

Type	Description
dict	Confirmation message.

Raises:

Type	Description
HTTPException	404 if not found; 403 if deletion is forbidden; 500 for server errors.

Source code in routers/collection.py

@router.delete("/collections/{collection_id}")
async def delete_collection(
    collection_id: str,
    requesting_user: User = Depends(get_current_user),
) -> dict:
    """
    Delete a collection and its related resources.

    Deletes documents in the collection, attempts to remove the vector index, and finally deletes the collection record.

    Args:
        collection_id (str): Collection identifier.
        requesting_user (User): Authenticated user injected by dependency.

    Returns:
        Confirmation message.

    Raises:
        HTTPException: 404 if not found; 403 if deletion is forbidden; 500 for server errors.
    """
    try:
        collection = await Collection.find_by_id(collection_id)
        if not collection:
            raise HTTPException(status_code=404, detail="Collection not found")
        if collection.user_id != requesting_user.id:
            raise HTTPException(
                status_code=403,
                detail="You are not allowed to delete this collection",
            )

        await Document.delete_many({"collection_id": collection_id})

        try:
            await anyio.to_thread.run_sync(
                vector_store.delete_collection, collection_id
            )
        except Exception as e:
            logger.warning(
                f"Warning: failed to delete Qdrant collection {collection_id}: {e}"
            )

        await collection.delete()
        return {"message": "Collection deleted successfully"}

    except HTTPException as e:
        raise e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Server error: {str(e)}")

get_collection(collection_id) async

Get a collection by id with document and vector counts.

Parameters:

Name	Type	Description	Default
collection_id	str	Target collection identifier.	required

Returns:

Type	Description
dict	Collection data including documents_count and points_count.

Raises:

Type	Description
HTTPException	404 if collection is not found.

Source code in routers/collection.py

@router.get("/collections/{collection_id}")
async def get_collection(collection_id: str) -> dict:
    """
    Get a collection by id with document and vector counts.

    Args:
        collection_id (str): Target collection identifier.

    Returns:
        Collection data including documents_count and points_count.

    Raises:
        HTTPException: 404 if collection is not found.
    """
    collection = await Collection.find_by_id(collection_id)
    if not collection:
        raise HTTPException(status_code=404, detail="Collection not found")

    documents_count, points_count = await _get_counts_for_id(collection_id)

    return {
        **collection.dict(),
        "documents_count": documents_count,
        "points_count": points_count,
    }

list_collections(request=Depends(), request_user=Depends(get_current_user)) async

List collections owned by the current user.

Parameters:

Name	Type	Description	Default
request	Pagination	Pagination parameters.	Depends()
request_user	User	Authenticated user injected by dependency.	Depends(get_current_user)

Returns:

Type	Description
PaginatedResponse[Collection]	Paginated list of user collections.

Source code in routers/collection.py

@router.get("/collections", response_model=PaginatedResponse[Collection])
async def list_collections(
    request: Pagination = Depends(), request_user: User = Depends(get_current_user)
) -> PaginatedResponse[Collection]:
    """
    List collections owned by the current user.

    Args:
        request (Pagination): Pagination parameters.
        request_user (User): Authenticated user injected by dependency.

    Returns:
        Paginated list of user collections.
    """
    return await Collection.find_all_with_pagination(
        limit=request.limit,
        page=request.page,
        filter_dict={"user_id": request_user.id},
        sort=[("timestamp", -1)],
    )

list_public_collections(pagination=Depends()) async

List public collections with pagination.

Combines platform-curated public collections and environment-specific public collections, then paginates the combined list.

Parameters:

Name	Type	Description	Default
pagination	Pagination	Pagination parameters.	Depends()

Returns:

Type	Description
PaginatedResponse[Collection]	Paginated list of public collections.

Source code in routers/collection.py

@router.get("/collections/public", response_model=PaginatedResponse[Collection])
async def list_public_collections(pagination: Pagination = Depends()) -> PaginatedResponse[Collection]:
    """
    List public collections with pagination.

    Combines platform-curated public collections and environment-specific public collections, then paginates the combined list.

    Args:
        pagination (Pagination): Pagination parameters.

    Returns:
        Paginated list of public collections.
    """
    public_collections, total_count = await vector_store.list_public_collections(
        page=pagination.page, limit=pagination.limit
    )

    if IS_PROD:
        public_collections = WILEY_PUBLIC_COLLECTIONS + public_collections
    else:
        public_collections = WILEY_PUBLIC_COLLECTIONS + STAGING_PUBLIC_COLLECTIONS
    total_count = len(public_collections)
    # Pagination must be done manually since Qdrant doesn't support collection pagination
    return PaginatedResponse(
        data=[
            Collection(
                id=collection["name"],
                name=collection.get("alias") or collection["name"],
                description=collection["description"],
                user_id=None,
                embeddings_model=DEFAULT_EMBEDDING_MODEL,
            )
            for collection in public_collections
        ],
        meta=get_pagination_metadata(total_count, pagination.page, pagination.limit),
    )

update_collection(request, collection_id, requesting_user=Depends(get_current_user)) async

Update a collection's mutable fields.

Parameters:

Name	Type	Description	Default
request	CollectionUpdate	Update payload (e.g., name).	required
collection_id	str	Collection identifier.	required
requesting_user	User	Authenticated user injected by dependency.	Depends(get_current_user)

Returns:

Type	Description
Collection	Updated collection.

Raises:

Type	Description
HTTPException	404 if not found; 403 if update is forbidden; 500 for server errors.

Source code in routers/collection.py

@router.patch("/collections/{collection_id}")
async def update_collection(
    request: CollectionUpdate,
    collection_id: str,
    requesting_user: User = Depends(get_current_user),
) -> Collection:
    """
    Update a collection's mutable fields.

    Args:
        request (CollectionUpdate): Update payload (e.g., name).
        collection_id (str): Collection identifier.
        requesting_user (User): Authenticated user injected by dependency.

    Returns:
        Updated collection.

    Raises:
        HTTPException: 404 if not found; 403 if update is forbidden; 500 for server errors.
    """
    try:
        collection = await Collection.find_by_id(collection_id)
        if not collection:
            raise HTTPException(status_code=404, detail="Collection not found")

        if collection.user_id != requesting_user.id:
            raise HTTPException(
                status_code=403,
                detail="You are not allowed to update this collection",
            )

        collection.name = request.name
        updated_collection = await collection.save()
        return updated_collection

    except HTTPException as e:
        raise e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Server error: {str(e)}")