Update main.py

main.py

@@ -74,7 +74,7 @@ async def get_completion(payload: Payload, request: Request):
     # Define a generator to stream the response
     def stream_generator():
         try:
-            with session.post(url, json=payload_dict, stream=True, timeout=5) as response:
+            with session.post(url, json=payload_dict, stream=True, timeout=15) as response:
                 response.raise_for_status()
                 for chunk in response.iter_content(chunk_size=1024):
                     if chunk:  # Only yield non-empty chunks