bug fixes

main.py

@@ -2,7 +2,7 @@ import os
 import re
 import httpx
 from fastapi import FastAPI, Request, HTTPException, Security
-from fastapi.responses import StreamingResponse
+from fastapi.responses import StreamingResponse, Response
 from fastapi.security import APIKeyHeader, APIKeyQuery
 from itertools import cycle
 import asyncio
@@ -110,10 +110,13 @@ async def proxy(request: Request, model_path: str, api_key: str = Security(get_a
     if "streamGenerateContent" in model_path:
         return StreamingResponse(stream_response(response), media_type=response.headers.get("content-type"))
     else:
-        # For non-streaming, read the response and return it.
-        # httpx.Response can be returned directly in FastAPI
-        await response.aread()
-        return response
+        # For non-streaming, read the response and return a new response object.
+        response_data = await response.aread()
+        return Response(
+            content=response_data,
+            status_code=response.status_code,
+            headers=dict(response.headers),
+        )
 
 if __name__ == "__main__":
     import uvicorn