Update main.py

main.py

@@ -162,45 +162,45 @@ async def chat_completions(
     original_uuid = uuid.uuid4()
     uuid_str = str(original_uuid).replace("-", "")
 
-    # Using the OpenAI API
-    json_data = {
-        'prompt': "\n".join(
-            [
-                f"{'User' if msg.role == 'user' else 'Assistant'}: {msg.content}"
-                for msg in request.messages
-            ]
-        ),
-        'stream': True,
-        'app_name': 'ChitChat_Edge_Ext',
-        'app_version': '4.28.0',
-        'tz_name': 'Asia/Karachi',
-        'cid': '',
-        'model': request.model,
-        'search': False,
-        'auto_search': False,
-        'filter_search_history': False,
-        'from': 'chat',
-        'group_id': 'default',
-        'chat_models': [],
-        'files': [],
-        'prompt_template': {
-            'key': '',
-            'attributes': {
-                'lang': 'original',
-            },
-        },
-        'tools': {
-            'auto': [
-                'search',
-                'text_to_image',
-                'data_analysis',
-            ],
-        },
-        'extra_info': {
-            'origin_url': '',
-            'origin_title': '',
-        },
-    }
+# Using the OpenAI API  
+json_data = {  
+    'prompt': "\n".join(  
+        [  
+            f"{'User' if msg.role == 'user' else 'Assistant'}: {msg.content}"  
+            for msg in request.messages  
+        ]  
+    ),  
+    'stream': True,  
+    'app_name': 'ChitChat_Chrome_Ext',  
+    'app_version': '4.28.0',  
+    'tz_name': 'Asia/Karachi',  
+    'cid': '',  
+    'model': request.model, 
+    'search': False,  
+    'auto_search': False,  
+    'filter_search_history': False,  
+    'from': 'chat',  
+    'group_id': 'default',  
+    'chat_models': [],  
+    'files': [],  
+    'prompt_template': {  
+        'key': 'artifacts',  
+        'attributes': {  
+            'lang': 'original',  
+        },  
+    },  
+    'tools': {  
+        'auto': [  
+            'search',  
+            'text_to_image',  
+            'data_analysis',  
+        ],  
+    },  
+    'extra_info': {  
+        'origin_url': 'chrome-extension://difoiogjjojoaoomphldepapgpbgkhkb/standalone.html?from=sidebar',  
+        'origin_title': 'Sider',  
+    },  
+}
 
     async def generate():
         async with httpx.AsyncClient() as client: