From f9283bc3118a3478f2b2701430226218d0a0eb69 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Wed, 19 Jun 2024 17:19:35 -0700 Subject: [PATCH] enh: pipeline user email support --- backend/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/main.py b/backend/main.py index 38c5e07b7..0a0587159 100644 --- a/backend/main.py +++ b/backend/main.py @@ -326,7 +326,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware): user = get_current_user( request, - get_http_authorization_cred(request.headers.get("Authorization")) + get_http_authorization_cred(request.headers.get("Authorization")), ) # Remove the citations from the body @@ -471,7 +471,7 @@ app.add_middleware(ChatCompletionMiddleware) def filter_pipeline(payload, user): - user = {"id": user.id, "name": user.name, "role": user.role} + user = {"id": user.id, "email": user.email, "name": user.name, "role": user.role} model_id = payload["model"] filters = [ model @@ -560,7 +560,7 @@ class PipelineMiddleware(BaseHTTPMiddleware): user = get_current_user( request, - get_http_authorization_cred(request.headers.get("Authorization")) + get_http_authorization_cred(request.headers.get("Authorization")), ) try: