Update main.py

main.py

@@ -54,37 +54,38 @@ async def chat_completions(request: ChatRequest):
         )
 
     # 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"
-    }
+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': [request.model],  # Try setting model here
+    'cid': "C082SLBVNR9J",
+    '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,  # Keep original model field
+    'prompt_template': {
+        'key': "artifacts",
+        'attributes': {
+            'lang': "original"
+        }
+    },
+    'search': False,
+    'tools': {
+        'auto': ["search", "text_to_image", "data_analysis"]
+    },
+    'tz_name': "Asia/Karachi"
+}
+
 
     logger.info(f"Outgoing API request payload: {json.dumps(json_data, indent=2)}")