Update main.py

main.py

@@ -214,7 +214,6 @@ async def chat_completions(
     request: ChatRequest, 
     app_secret: str = Depends(lambda: os.getenv("APP_SECRET"))
 ):
-
     # Validate model
     if request.model not in Blackbox.models:
         raise HTTPException(
@@ -225,6 +224,13 @@ async def chat_completions(
     # Generate a UUID for the conversation
     conversation_id = str(uuid.uuid4()).replace("-", "")
 
+    # Define headers
+    headers = {
+        "Authorization": f"Bearer {app_secret}",
+        "Content-Type": "application/json",
+        "uniqueid": conversation_id,
+    }
+
     json_data = {
         "attachments": [],
         "conversationId": conversation_id,
@@ -233,12 +239,10 @@ async def chat_completions(
         ),
     }
 
-    headers["uniqueid"] = conversation_id
-
     async def generate():
         async with httpx.AsyncClient() as client:
             try:
-                async with client.stream('POST', f'{Blackbox.api_endpoint}', headers=headers, json=json_data, timeout=120.0) as response:
+                async with client.stream('POST', Blackbox.api_endpoint, headers=headers, json=json_data) as response:
                     response.raise_for_status()
                     async for line in response.aiter_lines():
                         if line and line != "[DONE]":