Update main.py

main.py

@@ -164,39 +164,40 @@ 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
-        ]
+    # 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',
-    'model': request.model,
-    'search': False,          # Explicitly turn off search
-    'auto_search': False,      # Turn off auto-search
-    'from': 'chat',
-    'group_id': 'default',
-    'prompt_template': {
+        'stream': True,
+        'app_name': 'ChitChat_Edge_Ext',
+        'app_version': '4.28.0',
+        'tz_name': 'Asia/Karachi',
+        'model': request.model,
+        'search': False,          # Explicitly turn off search
+        'auto_search': False,      # Turn off auto-search
+        'from': 'chat',
+        'group_id': 'default',
+        'prompt_template': {
         'key': 'artifacts',
         'attributes': {
             'lang': 'original',
         },
     },
-    'tools': {
-        'auto': ['text_to_image', 'data_analysis'],  # Remove 'search' from tools if not needed
+        'tools': {
+            'auto': ['text_to_image', 'data_analysis'],  # Remove 'search' from tools if not needed
     },
-    'extra_info': {
-        'origin_url': 'chrome-extension://difoiogjjojoaoomphldepapgpbgkhkb/standalone.html?from=sidebar',
-        'origin_title': 'Sider',
+        'extra_info': {
+            'origin_url': 'chrome-extension://difoiogjjojoaoomphldepapgpbgkhkb/standalone.html?from=sidebar',
+            'origin_title': 'Sider',
     },
 }
 
 
+
     async def generate():
         async with httpx.AsyncClient() as client:
             try: