Update main.py

main.py

@@ -42,23 +42,6 @@ class ChatRequest(BaseModel):
     stream: Optional[bool] = False
 
 
-def create_chat_completion_data(content: str, model: str, finish_reason: Optional[str] = None) -> Dict[str, Any]:
-    return {
-        "id": f"chatcmpl-{uuid.uuid4()}",
-        "object": "chat.completion.chunk",
-        "created": int(datetime.now().timestamp()),
-        "model": model,
-        "choices": [
-            {
-                "index": 0,
-                "delta": {"content": content, "role": "assistant"},
-                "finish_reason": finish_reason,
-            }
-        ],
-        "usage": None,
-    }
-
-
 @app.post("/hf/v1/chat/completions")
 async def chat_completions(request: ChatRequest):
     logger.info(f"Received chat completion request for model: {request.model}")
@@ -70,16 +53,40 @@ async def chat_completions(request: ChatRequest):
             detail=f"Model {request.model} is not allowed. Allowed models are: {', '.join(ALLOWED_MODELS)}",
         )
 
+    # Build the payload as per the required structure
     json_data = {
         'prompt': "\n".join(
             [f"{'User' if msg.role == 'user' else 'Assistant'}: {msg.content}" for msg in request.messages]
         ),
         'stream': request.stream,
+        'app_name': "ChitChat_Chrome_Ext",
+        'app_version': "4.28.0",
+        'auto_search': False,
+        'chat_models': [],
+        'cid': "C082SLBVNR9J",  # Adjust based on specific requirements if needed
+        'extra_info': {
+            'origin_url': "chrome-extension://difoiogjjojoaoomphldepapgpbgkhkb/standalone.html?from=sidebar",
+            'origin_title': "Sider"
+        },
+        'files': [],
+        'filter_search_history': False,
+        'from': "chat",
+        'group_id': "default",
         'model': request.model,
+        'prompt_template': {
+            'key': "artifacts",
+            'attributes': {
+                'lang': "original"
+            }
+        },
+        'search': False,
+        'tools': {
+            'auto': ["search", "text_to_image", "data_analysis"]
+        },
+        'tz_name': "Asia/Karachi"
     }
-    
-    logger.info(f"Sending request to external API with model: {request.model}")
-    logger.debug(f"Request payload: {json.dumps(json_data, indent=2)}")
+
+    logger.info(f"Outgoing API request payload: {json.dumps(json_data, indent=2)}")
 
     async def generate():
         async with httpx.AsyncClient() as client: