Update main.py

main.py

@@ -36,17 +36,30 @@ CONNECTION_LIMIT_PER_HOST = 50
 REQUEST_TIMEOUT = 300
 WORKER_COUNT = 10
 
-# ─── Static Headers ───
+# ─── Required Headers ───
 HEADERS = {
+    "authority": "www.blackbox.ai",
+    "method": "POST",
+    "path": "/api/chat",
+    "scheme": "https",
     "accept": "*/*",
-    "content-type": "application/json",
+    "accept-encoding": "gzip, deflate, br, zstd",
+    "accept-language": "en-US,en;q=0.9",
     "origin": "https://www.blackbox.ai",
+    "priority": "u=1, i",
     "referer": "https://www.blackbox.ai/",
+    "sec-ch-ua": '"Chromium";v="136", "Google Chrome";v="136", "Not.A/Brand";v="99"',
+    "sec-ch-ua-mobile": "?0",
+    "sec-ch-ua-platform": '"Windows"',
+    "sec-fetch-dest": "empty",
+    "sec-fetch-mode": "cors",
+    "sec-fetch-site": "same-origin",
     "user-agent": (
         "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
         "AppleWebKit/537.36 (KHTML, like Gecko) "
         "Chrome/136.0.0.0 Safari/537.36"
     ),
+    "content-type": "application/json",
 }
 
 # ─── FastAPI Setup ───
@@ -72,7 +85,6 @@ class RetryableStatusError(Exception):
 
 RETRYABLE_STATUSES = {400, 429, 500, 502, 503, 504}
 
-# ─── Random Data ───
 _ascii = string.ascii_letters + string.digits
 def _rand(n, pool=_ascii): return "".join(random.choice(pool) for _ in range(n))
 def random_email(): return _rand(12) + "@gmail.com"
@@ -81,10 +93,13 @@ def random_customer_id(): return "cus_" + _rand(12)
 
 # ─── Payload Generator ───
 def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
+    unique_id = str(uuid.uuid4())
     return {
         "messages": messages,
-        "id": _rand(8),
         "agentMode": {},
+        "id": messages[-1]["id"] if messages else _rand(8),
+        "previewToken": None,
+        "userId": None,
         "codeModelMode": True,
         "trendingAgentMode": {},
         "isMicMode": False,
@@ -101,7 +116,7 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         "isMemoryEnabled": False,
         "mobileClient": False,
         "userSelectedModel": None,
-        "validated": str(uuid.uuid4()),
+        "validated": unique_id,
         "imageGenerationMode": False,
         "webSearchModePrompt": False,
         "deepSearchMode": True,
@@ -117,7 +132,7 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         },
         "session": {
             "user": {
-                "name": "S.C gaming",
+                "name": _rand(10),
                 "email": random_email(),
                 "image": "https://lh3.googleusercontent.com/a/default",
                 "id": random_id()
@@ -214,7 +229,6 @@ async def chat_completions(request: Request):
         if not messages:
             raise HTTPException(status_code=400, detail="Missing 'messages'")
         stream = body.get("stream", False)
-
         payload = build_payload(messages)
 
         if not stream: