34eed72150
* add user id validation * relax conversation id check to allow default while I'm here * fix annotation validation * -api changes
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
from typing import TYPE_CHECKING, List, Optional
|
|
|
|
from fastapi import APIRouter, Body, Depends, Query
|
|
|
|
from letta.schemas.user import User, UserCreate, UserUpdate
|
|
from letta.server.rest_api.dependencies import get_letta_server
|
|
from letta.validators import UserIdQueryRequired
|
|
|
|
if TYPE_CHECKING:
|
|
from letta.schemas.user import User
|
|
from letta.server.server import SyncServer
|
|
|
|
|
|
router = APIRouter(prefix="/users", tags=["users", "admin"])
|
|
|
|
|
|
@router.get("/", tags=["admin"], response_model=List[User], operation_id="list_users")
|
|
async def list_users(
|
|
after: Optional[str] = Query(None),
|
|
limit: Optional[int] = Query(50),
|
|
server: "SyncServer" = Depends(get_letta_server),
|
|
):
|
|
"""
|
|
Get a list of all users in the database
|
|
"""
|
|
return await server.user_manager.list_actors_async(after=after, limit=limit)
|
|
|
|
|
|
@router.post("/", tags=["admin"], response_model=User, operation_id="create_user")
|
|
async def create_user(
|
|
request: UserCreate = Body(...),
|
|
server: "SyncServer" = Depends(get_letta_server),
|
|
):
|
|
"""
|
|
Create a new user in the database
|
|
"""
|
|
user = User(**request.model_dump())
|
|
user = await server.user_manager.create_actor_async(user)
|
|
return user
|
|
|
|
|
|
@router.put("/", tags=["admin"], response_model=User, operation_id="update_user")
|
|
async def update_user(
|
|
user: UserUpdate = Body(...),
|
|
server: "SyncServer" = Depends(get_letta_server),
|
|
):
|
|
"""
|
|
Update a user in the database
|
|
"""
|
|
user = await server.user_manager.update_actor_async(user)
|
|
return user
|
|
|
|
|
|
@router.delete("/", tags=["admin"], response_model=User, operation_id="delete_user")
|
|
async def delete_user(
|
|
user_id: UserIdQueryRequired,
|
|
server: "SyncServer" = Depends(get_letta_server),
|
|
):
|
|
# TODO make a soft deletion, instead of a hard deletion
|
|
# Get the user first so we can return it after deletion
|
|
user = await server.user_manager.get_actor_by_id_async(actor_id=user_id)
|
|
await server.user_manager.delete_actor_by_id_async(user_id=user_id)
|
|
return user
|