debugging

main.py

@@ -89,7 +89,6 @@ async def proxy(request: Request, model_path: str, api_key: str = Security(get_a
         except json.JSONDecodeError:
             pass # Not a json body, proxy as is
 
-    print(request_body_to_send)
 
     target_url = f"https://aiplatform.googleapis.com/v1/projects/{project_id}/locations/global/publishers/google/models/{model_path}?key={express_key}"
 
@@ -121,11 +120,19 @@ async def proxy(request: Request, model_path: str, api_key: str = Security(get_a
     else:
         try:
             response_data = await response.aread()
-            print(response_data)
+            response_json = json.loads(response_data)
+
+            if 'candidates' in response_json:
+                for candidate in response_json.get('candidates', []):
+                    if 'content' in candidate and 'parts' in candidate.get('content', {}):
+                        candidate['content']['parts'] = [part for part in candidate['content']['parts'] if part]
+
+            modified_response_data = json.dumps(response_json).encode('utf-8')
+
             return Response(
-                content=response_data,
+                content=modified_response_data,
                 status_code=response.status_code,
-                headers={"content-type":response.headers.get("content-type")},
+                headers=dict(response.headers),
             )
         finally:
             await response.aclose()