chore: comment clean up

This commit is contained in:
Timothy J. Baek 2024-03-20 19:32:25 -07:00
parent 1a240e103e
commit ed0d4b5a2a

View file

@ -67,20 +67,15 @@ class ModifyModelsResponseMiddleware(BaseHTTPMiddleware):
response = await call_next(request)
user = request.state.user
# Check if the request is for the `/models` route
if "/models" in request.url.path:
# Ensure the response is a StreamingResponse
if isinstance(response, StreamingResponse):
# Read the content of the streaming response
body = b""
async for chunk in response.body_iterator:
body += chunk
# Modify the content as needed
data = json.loads(body.decode("utf-8"))
print(data)
if app.state.MODEL_FILTER_ENABLED:
if user and user.role == "user":
data["data"] = list(
@ -91,14 +86,11 @@ class ModifyModelsResponseMiddleware(BaseHTTPMiddleware):
)
)
# Example modification: Add a new key-value pair
# Modified Flag
data["modified"] = True
# Return a new JSON response with the modified content
return JSONResponse(content=data)
return response
# Add the middleware to the app
app.add_middleware(ModifyModelsResponseMiddleware)