This commit is contained in:
Timothy Jaeryang Baek
2024-11-20 06:09:53 -08:00
2 changed files with 79 additions and 74 deletions

View File

@@ -855,6 +855,11 @@ class PipelineMiddleware(BaseHTTPMiddleware):
status_code=status.HTTP_401_UNAUTHORIZED,
content={"detail": "Not authenticated"},
)
except HTTPException as e:
return JSONResponse(
status_code=e.status_code,
content={"detail": e.detail},
)
model_list = await get_all_models()
models = {model["id"]: model for model in model_list}