Fix Request Timeout issue

app.py

@@ -139,10 +139,11 @@
 #     return response.json()
 
 
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 import httpx
 from pydantic import BaseModel
 
+
 class RequestBody(BaseModel):
     data: dict
 
@@ -158,7 +159,17 @@ async def proxy_get(path: str):
 
 @app.post("/proxy/{path:path}")
 async def proxy_post(path: str, request_body: RequestBody):
+    print(request_body)
+    print(request_body.model_dump())
     """ Forwards POST requests to the target server """
-    async with httpx.AsyncClient() as client:
-        response = await client.post(f"{TARGET_SERVER}/{path}", json = request_body.model_dump())
-    return response.json()
+    try:
+        data = request_body.model_dump()["data"]
+
+        async with httpx.AsyncClient() as client:
+            response = await client.post(f"{TARGET_SERVER}/{path}", json=data, timeout=120)
+        return response.json()
+    except Exception as e:
+        print(e)
+        return {"error": str(e)}
+    
+

requirements.txt

@@ -1,7 +1,5 @@
 fastapi
 uvicorn
 ollama
-openai
 pydantic
-requests
 httpx