Update main.py

main.py

@@ -188,7 +188,6 @@ class ChatRequest(BaseModel):
 async def chat_completions(request: ChatRequest):
     messages = [{"role": msg.role, "content": msg.content} for msg in request.messages]
 
-    # Use an async generator to get the response
     async_generator = Blackbox.create_async_generator(
         model=request.model,
         messages=messages
@@ -196,21 +195,9 @@ async def chat_completions(request: ChatRequest):
 
     response_content = ""
     async for chunk in async_generator:
-        response_content += chunk if isinstance(chunk, str) else chunk.content  # Concatenate response
+        response_content += chunk if isinstance(chunk, str) else chunk.content
 
     return {
-        "id": f"chatcmpl-{uuid.uuid4()}",  # Generate a unique ID
-        "object": "chat.completion",
-        "created": int(datetime.now().timestamp()),  # Current timestamp
-        "model": request.model,
-        "choices": [
-            {
-                "message": {
-                    "role": "assistant",
-                    "content": response_content
-                },
-                "finish_reason": "stop",
-                "index": 0
-            }
-        ]
+        "content": response_content  # Updated return
     }
+