8ae1e64987
Co-authored-by: Charles Packer <packercharles@gmail.com> Co-authored-by: Shubham Naik <shubham.naik10@gmail.com> Co-authored-by: Shubham Naik <shub@memgpt.ai>
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel, Field
|
|
|
|
from letta.log import get_logger
|
|
from letta.server.rest_api.interface import QueuingInterface
|
|
from letta.server.server import SyncServer
|
|
|
|
logger = get_logger(__name__)
|
|
router = APIRouter()
|
|
|
|
|
|
class AuthResponse(BaseModel):
|
|
uuid: UUID = Field(..., description="UUID of the user")
|
|
is_admin: Optional[bool] = Field(None, description="Whether the user is an admin")
|
|
|
|
|
|
class AuthRequest(BaseModel):
|
|
password: str = Field(None, description="Admin password provided when starting the Letta server")
|
|
|
|
|
|
def setup_auth_router(server: SyncServer, interface: QueuingInterface, password: str) -> APIRouter:
|
|
|
|
@router.post("/auth", tags=["auth"], response_model=AuthResponse)
|
|
def authenticate_user(request: AuthRequest) -> AuthResponse:
|
|
"""
|
|
Authenticates the user and sends response with User related data.
|
|
|
|
Currently, this is a placeholder that simply returns a UUID placeholder
|
|
"""
|
|
interface.clear()
|
|
|
|
is_admin = False
|
|
if request.password != password:
|
|
response = server.api_key_to_user(api_key=request.password)
|
|
else:
|
|
is_admin = True
|
|
response = server.authenticate_user()
|
|
return AuthResponse(uuid=response, is_admin=is_admin)
|
|
|
|
return router
|