Update main.py

main.py

@@ -9,7 +9,6 @@ from typing import AsyncGenerator, Dict, List, Any
 
 import aiohttp
 import brotli
-import importlib
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
@@ -67,17 +66,24 @@ HTTP_SESSION: aiohttp.ClientSession = None
 RETRYABLE_STATUSES = {400, 429, 500, 502, 503, 504}
 _ascii = string.ascii_letters + string.digits
 
-# ─── Utility ───
-def _rand(n, pool=_ascii): return "".join(random.choice(pool) for _ in range(n))
+# ─── Helpers ───
+def _rand(n, pool=_ascii): return ''.join(random.choice(pool) for _ in range(n))
 def random_email(): return _rand(12) + "@gmail.com"
 def random_id(): return _rand(21, string.digits)
 def random_customer_id(): return "cus_" + _rand(12)
+def generate_7char_id(): return _rand(7)
 
+# ─── Payload Builder ───
 def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
+    msg_id = generate_7char_id()
+    if messages:
+        messages[-1]["id"] = msg_id
+
+    now = int(time.time())
     return {
         "messages": messages,
         "agentMode": {},
-        "id": None,
+        "id": msg_id,
         "previewToken": None,
         "userId": None,
         "codeModelMode": True,
@@ -124,8 +130,8 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         "subscriptionCache": {
             "status": "PREMIUM",
             "customerId": random_customer_id(),
-            "expiryTimestamp": 1749108685,
-            "lastChecked": 1746822333827,
+            "expiryTimestamp": now + 60 * 86400,  # 60 days
+            "lastChecked": int(time.time() * 1000),  # milliseconds
             "isTrialSubscription": False
         },
         "beastMode": False,
@@ -133,7 +139,7 @@ def build_payload(messages: List[Dict[str, Any]]) -> Dict[str, Any]:
         "designerMode": False
     }
 
-# ─── Retry ───
+# ─── Retry Logic ───
 class RetryableStatusError(Exception):
     def __init__(self, status: int, text: str):
         super().__init__(f"status={status} body={text[:100]}...")
@@ -153,7 +159,6 @@ def log_retry(retry_state):
 async def get_blackbox_response(*, data, stream: bool, request_id: str) -> AsyncGenerator[str, None]:
     global HTTP_SESSION
     if not HTTP_SESSION:
-        importlib.import_module("brotli")
         HTTP_SESSION = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=REQUEST_TIMEOUT))
 
     async with HTTP_SESSION.post(BLACKBOX_URL, json=data, headers=HEADERS, timeout=REQUEST_TIMEOUT) as resp: