added error handling

main.py

@@ -140,7 +140,6 @@ async def proxy(request: Request, model_path: str, api_key: str = Security(get_a
         if k.lower() not in ['host', 'authorization', 'x-goog-api-key', 'content-length']
     }
 
-    print(request_body_to_send)
 
     if "streamGenerateContent" in model_path:
         target_url = target_url + "&alt=sse"
@@ -153,11 +152,22 @@ async def proxy(request: Request, model_path: str, api_key: str = Security(get_a
     )
     response = await client.send(req, stream=True)
 
+    if response.status_code != 200:
+        try:
+            response_data = await response.aread()
+            return Response(
+                content=response_data,
+                status_code=response.status_code,
+                headers=dict(response.headers),
+            )
+        finally:
+            await response.aclose()
+            await client.aclose()
+
     if "streamGenerateContent" in model_path:
         async def stream_generator():
             try:
                 async for line in response.aiter_lines():
-                    print(line)
                     yield f"{line}\n"
             finally:
                 await response.aclose()
@@ -169,10 +179,10 @@ async def proxy(request: Request, model_path: str, api_key: str = Security(get_a
             response_data = await response.aread()
             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]
+            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')