Update app.py

app.py

@@ -17,9 +17,8 @@ import mimetypes
 import uuid
 import traceback
 from datetime import datetime
-
-# Pydantic
 from pydantic import ValidationError
+from dotenv import load_dotenv
 
 # Project imports
 from websocket_handler import websocket_endpoint
@@ -35,14 +34,14 @@ from logger import log_error, log_info, log_warning
 # Exception imports
 from exceptions import (
     DuplicateResourceError, 
-    ValidationError, 
-    ResourceNotFoundError,    
-    FlareException, 
     RaceConditionError, 
-    format_error_response, 
+    ValidationError, 
+    ResourceNotFoundError,
+    AuthenticationError,
+    AuthorizationError,
+    ConfigurationError,
     get_http_status_code
 )
-from dotenv import load_dotenv
 
 # Load .env file if exists
 load_dotenv()
@@ -152,7 +151,8 @@ app.include_router(chat_router, prefix="/api")
 # ---------------- Admin API routes ----------------------------------
 app.include_router(admin_router, prefix="/api/admin")
 
-# Global exception handler
+# ---------------- Exception Handlers ----------------------------------
+# Add global exception handler
 @app.exception_handler(Exception)
 async def global_exception_handler(request: Request, exc: Exception):
     """Handle all unhandled exceptions"""
@@ -194,6 +194,7 @@ async def global_exception_handler(request: Request, exc: Exception):
         })
     )
 
+# Add custom exception handlers
 @app.exception_handler(DuplicateResourceError)
 async def duplicate_resource_handler(request: Request, exc: DuplicateResourceError):
     """Handle duplicate resource errors"""
@@ -240,6 +241,40 @@ async def resource_not_found_handler(request: Request, exc: ResourceNotFoundErro
         }
     )
 
+@app.exception_handler(AuthenticationError)
+async def authentication_error_handler(request: Request, exc: AuthenticationError):
+    """Handle authentication errors"""
+    return JSONResponse(
+        status_code=401,
+        content={
+            "detail": str(exc),
+            "error_type": "authentication_error"
+        }
+    )
+
+@app.exception_handler(AuthorizationError)
+async def authorization_error_handler(request: Request, exc: AuthorizationError):
+    """Handle authorization errors"""
+    return JSONResponse(
+        status_code=403,
+        content={
+            "detail": str(exc),
+            "error_type": "authorization_error"
+        }
+    )
+
+@app.exception_handler(ConfigurationError)
+async def configuration_error_handler(request: Request, exc: ConfigurationError):
+    """Handle configuration errors"""
+    return JSONResponse(
+        status_code=500,
+        content={
+            "detail": str(exc),
+            "error_type": "configuration_error",
+            "config_key": exc.details.get("config_key")
+        }
+    )
+
 # ---------------- Metrics endpoint -----------------
 @app.get("/metrics")
 async def get_metrics():