Update main.py

main.py

@@ -352,26 +352,27 @@ async def chat_completions(request: ChatRequest, req: Request, api_key: str = De
             "id": f"chatcmpl-{uuid.uuid4()}",
             "object": "chat.logger.info(f"Completed response generation for API key: {api_key} | IP: {client_ip}")"
         return {
-            "id": f"chatcmpl-{uuid.uuid4()}",
-            "object": "chat.completion",
-            "created": int(datetime.now().timestamp()),
-            "model": request.model,
-            "choices": [
-                {
-                    "message": {
-                        "role": "assistant",
-                        "content": response_content
-                    },
-                    "finish_reason": "stop",
-                    "index": 0
-                }
-            ],
-            "usage": {
-                "prompt_tokens": sum(len(msg.content.split()) for msg in request.messages),
-                "completion_tokens": len(response_content.split()),
-                "total_tokens": sum(len(msg.content.split()) for msg in request.messages) + len(response_content.split())
+    "id": f"chatcmpl-{uuid.uuid4()}",
+    "object": "chat.completion",
+    "created": int(datetime.now().timestamp()),
+    "model": request.model,
+    "choices": [
+        {
+            "message": {
+                "role": "assistant",
+                "content": response_content
             },
+            "finish_reason": "stop",
+            "index": 0
         }
+    ],
+    "usage": {
+        "prompt_tokens": sum(len(msg.content.split()) for msg in request.messages),
+        "completion_tokens": len(response_content.split()),
+        "total_tokens": sum(len(msg.content.split()) for msg in request.messages) + len(response_content.split())
+    }
+}  # This closing curly brace was missing
+
     except ModelNotWorkingException as e:
         logger.warning(f"Model not working: {e} | IP: {client_ip}")
         raise HTTPException(status_code=503, detail=str(e))