Update api/utils.py

api/utils.py

@@ -39,6 +39,8 @@ async def get_ddg_vqd():
                     vqd = response.headers.get("x-vqd-4")
                     if not vqd:
                         logger.error("VQD token not found in response headers.")
+                    else:
+                        logger.debug(f"VQD token retrieved: {vqd}")
                     return vqd
                 else:
                     logger.error(f"Error: Status code {response.status} when fetching VQD")
@@ -109,8 +111,28 @@ async def process_ddg_streaming_response(request: ChatRequest):
     json_data = {
         "model": MODEL_MAPPING.get(request.model, request.model),
         "messages": message_history,
+        "previewToken": None,
+        "userId": None,
+        "codeModelMode": True,
+        "agentMode": agent_mode,
+        "trendingAgentMode": trending_agent_mode,
+        "isMicMode": False,
+        "userSystemPrompt": None,
+        "maxTokens": request.max_tokens,
+        "playgroundTopP": request.top_p,
+        "playgroundTemperature": request.temperature,
+        "isChromeExt": False,
+        "githubToken": None,
+        "clickedAnswer2": False,
+        "clickedAnswer3": False,
+        "clickedForceWebSearch": False,
+        "visitFromDelta": False,
+        "mobileClient": False,
+        "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
     }
 
+    logger.debug(f"Sending JSON payload to DDG API: {json.dumps(json_data)}")
+
     async with httpx.AsyncClient() as client:
         try:
             async with client.stream(
@@ -127,7 +149,7 @@ async def process_ddg_streaming_response(request: ChatRequest):
                         content = line
                         if content.startswith("$@$v=undefined-rv1$@$"):
                             content = content[21:]
-                        # No model_prefix handling as messages are correctly formatted
+                        # Strip the model prefix from the response content
                         cleaned_content = strip_model_prefix(content, model_prefix)
                         yield f"data: {json.dumps(create_chat_completion_data(cleaned_content, request.model, timestamp))}\n\n"
 
@@ -144,7 +166,6 @@ async def process_ddg_non_streaming_response(request: ChatRequest):
     agent_mode = AGENT_MODE.get(request.model, {})
     trending_agent_mode = TRENDING_AGENT_MODE.get(request.model, {})
     model_prefix = MODEL_PREFIXES.get(request.model, "")
-    # referer_path and referer_url are not required unless specified by DDG API
 
     # Obtain VQD token
     vqd = await get_ddg_vqd()
@@ -169,8 +190,28 @@ async def process_ddg_non_streaming_response(request: ChatRequest):
     json_data = {
         "model": MODEL_MAPPING.get(request.model, request.model),
         "messages": message_history,
+        "previewToken": None,
+        "userId": None,
+        "codeModelMode": True,
+        "agentMode": agent_mode,
+        "trendingAgentMode": trending_agent_mode,
+        "isMicMode": False,
+        "userSystemPrompt": None,
+        "maxTokens": request.max_tokens,
+        "playgroundTopP": request.top_p,
+        "playgroundTemperature": request.temperature,
+        "isChromeExt": False,
+        "githubToken": None,
+        "clickedAnswer2": False,
+        "clickedAnswer3": False,
+        "clickedForceWebSearch": False,
+        "visitFromDelta": False,
+        "mobileClient": False,
+        "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),
     }
 
+    logger.debug(f"Sending JSON payload to DDG API: {json.dumps(json_data)}")
+
     async with httpx.AsyncClient() as client:
         try:
             response = await client.post(