letta-nightly 0.1.7.dev20240924104148__py3-none-any.whl

letta/server/rest_api/auth/index.py

@@ -0,0 +1,43 @@
+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

letta/server/rest_api/auth_token.py

@@ -0,0 +1,22 @@
+import uuid
+
+from fastapi import Depends, HTTPException
+from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
+
+from letta.server.server import SyncServer
+
+security = HTTPBearer()
+
+
+def get_current_user(server: SyncServer, password: str, auth: HTTPAuthorizationCredentials = Depends(security)) -> uuid.UUID:
+    try:
+        api_key_or_password = auth.credentials
+        if api_key_or_password == password:
+            # user is admin so we just return the default uuid
+            return server.authenticate_user()
+        user_id = server.api_key_to_user(api_key=api_key_or_password)
+        return user_id
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(status_code=403, detail=f"Authentication error: {e}")