backend/apps/litellm/main.py

@@ -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)