Update main.py

main.py

@@ -629,8 +629,8 @@ async def chat_completions(request: ChatRequest, req: Request, api_key: str = De
         generator = Blackbox.create_async_generator(
             model=request.model,
             messages=[{"role": msg.role, "content": msg.content} for msg in request.messages],
-            proxy=request.headers.get('Proxy'),  # Assuming proxy info is passed via headers
-            websearch=request.query_params.get('websearch', 'false').lower() == 'true'
+            proxy=req.headers.get('Proxy'),  # Corrected usage
+            websearch=req.query_params.get('websearch', 'false').lower() == 'true'
         )
 
         logger.info(f"Started streaming response for API key: {api_key} | IP: {client_ip}")