feat(ui): initial auth plugin (#763)

Author: dazy-ds

examples/chat/authenticated_chat.py

@@ -195,6 +195,7 @@ def get_auth_backend() -> ListAuthenticationBackend:
     """Factory function to create the preferred authentication backend."""
     users = [
         {
+            "user_id": "8e6c5871-3817-4d62-828f-ef6789de31b9",
             "username": "admin",
             "password": "admin123",
             "email": "[email protected]",
@@ -203,6 +204,7 @@ def get_auth_backend() -> ListAuthenticationBackend:
             "metadata": {"department": "IT", "clearance_level": "high"},
         },
         {
+            "user_id": "d3afde97-35fb-41d0-9aa6-6a1c822db63e",
             "username": "moderator",
             "password": "mod123",
             "email": "[email protected]",
@@ -211,6 +213,7 @@ def get_auth_backend() -> ListAuthenticationBackend:
             "metadata": {"department": "Community", "shift": "day"},
         },
         {
+            "user_id": "7ef3d9d5-cdad-405a-a919-3d3ee5e1c34d",
             "username": "alice",
             "password": "alice123",
             "email": "[email protected]",
@@ -219,6 +222,7 @@ def get_auth_backend() -> ListAuthenticationBackend:
             "metadata": {"department": "Marketing", "join_date": "2024-01-15"},
         },
         {
+            "user_id": "acac16db-37f0-43cb-b18f-b005c3c3de88",
             "username": "bob",
             "password": "bob123",
             "email": "[email protected]",

packages/ragbits-chat/CHANGELOG.md

@@ -1,6 +1,7 @@
 # CHANGELOG
 
 ## Unreleased
+- Add authentication handling in the UI (#763)
 - Add backend authentication (#761)
 
 - Autogenerate typescript types based on backend typing (#727)

packages/ragbits-chat/src/ragbits/chat/api.py

@@ -478,7 +478,7 @@ async def _handle_login(self, request: LoginRequest) -> JSONResponse:
                 return JSONResponse(
                     content=LoginResponse(
                         success=True,
-                        user=auth_result.user.model_dump() if auth_result.user else None,
+                        user=auth_result.user if auth_result.user else None,
                         error_message=None,
                         jwt_token=auth_result.jwt_token,
                     ).model_dump()

packages/ragbits-chat/src/ragbits/chat/auth/types.py

@@ -60,7 +60,7 @@ class LoginResponse(BaseModel):
     """
 
     success: bool = Field(..., description="Whether login was successful")
-    user: dict | None = Field(None, description="User information")
+    user: User | None = Field(None, description="User information")
     error_message: str | None = Field(None, description="Error message if login failed")
     jwt_token: JWTToken | None = Field(..., description="Access jwt_token")
 

packages/ragbits-chat/src/ragbits/chat/providers/model_provider.py

@@ -10,6 +10,8 @@
 
 from pydantic import BaseModel
 
+from ragbits.chat.interface.types import AuthType
+
 
 class RagbitsChatModelProvider:
     """
@@ -37,8 +39,17 @@ def get_models(self) -> dict[str, type[BaseModel | Enum]]:
             return self._models_cache
 
         try:
+            from ragbits.chat.auth.types import (
+                CredentialsLoginRequest,
+                JWTToken,
+                LoginRequest,
+                LoginResponse,
+                LogoutRequest,
+                User,
+            )
             from ragbits.chat.interface.forms import UserSettings
             from ragbits.chat.interface.types import (
+                AuthenticationConfig,
                 ChatContext,
                 ChatMessageRequest,
                 ChatResponseType,
@@ -90,6 +101,15 @@ def get_models(self) -> dict[str, type[BaseModel | Enum]]:
                 # API request models
                 "ChatRequest": ChatMessageRequest,
                 "FeedbackRequest": FeedbackRequest,
+                # Auth
+                "AuthType": AuthType,
+                "AuthenticationConfig": AuthenticationConfig,
+                "CredentialsLoginRequest": CredentialsLoginRequest,
+                "JWTToken": JWTToken,
+                "LoginRequest": LoginRequest,
+                "LoginResponse": LoginResponse,
+                "LogoutRequest": LogoutRequest,
+                "User": User,
             }
 
             return self._models_cache
@@ -111,7 +131,9 @@ def get_categories(self) -> dict[str, list[str]]:
             return self._categories_cache
 
         self._categories_cache = {
-            "enums": ["ChatResponseType", "FeedbackType", "LiveUpdateType", "MessageRole"],
+            "enums": [model_name for model_name, model in self._models_cache.items() if issubclass(model, Enum)]
+            if self._models_cache
+            else [],
             "core_data": [
                 "ChatContext",
                 "LiveUpdate",
@@ -121,10 +143,28 @@ def get_categories(self) -> dict[str, list[str]]:
                 "ServerState",
                 "FeedbackItem",
                 "Image",
+                "JWTToken",
+                "User",
+            ],
+            "configuration": [
+                "HeaderCustomization",
+                "UICustomization",
+                "UserSettings",
+                "FeedbackConfig",
+                "AuthenticationConfig",
+            ],
+            "responses": [
+                "FeedbackResponse",
+                "ConfigResponse",
+                "LoginResponse",
+            ],
+            "requests": [
+                "ChatRequest",
+                "FeedbackRequest",
+                "CredentialsLoginRequest",
+                "LoginRequest",
+                "LogoutRequest",
             ],
-            "configuration": ["HeaderCustomization", "UICustomization", "UserSettings", "FeedbackConfig"],
-            "responses": ["FeedbackResponse", "ConfigResponse"],
-            "requests": ["ChatRequest", "FeedbackRequest"],
         }
 
         return self._categories_cache