22 lines
989 B
Python
22 lines
989 B
Python
from typing import Optional
|
|
|
|
from fastapi import APIRouter, Depends, Header
|
|
|
|
from letta.server.rest_api.utils import get_letta_server
|
|
from letta.server.server import SyncServer
|
|
|
|
router = APIRouter(prefix="/embeddings", tags=["embeddings"])
|
|
|
|
|
|
@router.get("/total_storage_size", response_model=float, operation_id="get_total_storage_size")
|
|
async def get_embeddings_total_storage_size(
|
|
server: SyncServer = Depends(get_letta_server),
|
|
actor_id: Optional[str] = Header(None, alias="user_id"), # Extract user_id from header, default to None if not present
|
|
storage_unit: Optional[str] = Header("GB", alias="storage_unit"), # Extract storage unit from header, default to GB
|
|
):
|
|
"""
|
|
Get the total size of all embeddings in the database for a user in the storage unit given.
|
|
"""
|
|
actor = await server.user_manager.get_actor_or_default_async(actor_id=actor_id)
|
|
return await server.passage_manager.estimate_embeddings_size_async(actor=actor, storage_unit=storage_unit)
|