Commit f9283bc3 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

enh: pipeline user email support

parent d5b76b5e
...@@ -326,7 +326,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware): ...@@ -326,7 +326,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
user = get_current_user( user = get_current_user(
request, request,
get_http_authorization_cred(request.headers.get("Authorization")) get_http_authorization_cred(request.headers.get("Authorization")),
) )
# Remove the citations from the body # Remove the citations from the body
...@@ -471,7 +471,7 @@ app.add_middleware(ChatCompletionMiddleware) ...@@ -471,7 +471,7 @@ app.add_middleware(ChatCompletionMiddleware)
def filter_pipeline(payload, user): 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"] model_id = payload["model"]
filters = [ filters = [
model model
...@@ -560,7 +560,7 @@ class PipelineMiddleware(BaseHTTPMiddleware): ...@@ -560,7 +560,7 @@ class PipelineMiddleware(BaseHTTPMiddleware):
user = get_current_user( user = get_current_user(
request, request,
get_http_authorization_cred(request.headers.get("Authorization")) get_http_authorization_cred(request.headers.get("Authorization")),
) )
try: try:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment