Update main.py

main.py

@@ -202,9 +202,13 @@ async def chat_completions(request: ChatRequest):
         else:
             response_content += chunk.content
 
-    # Clean up the response to get only the plain text content
-    start = response_content.find('"content": "') + len('"content": "')
-    end = response_content.find('"', start)
-    clean_content = response_content[start:end].replace('\\n', '\n')  # Handle newline characters
-
-    return Response(content=clean_content.strip(), media_type="text/plain")  # Return plain text
+    # Clean up the response to extract the plain text content
+    clean_content = response_content.replace('\\n', '\n').strip()
+
+    return {
+        "id": str(uuid.uuid4()),
+        "object": "chat.completion",
+        "created": int(datetime.now().timestamp()),
+        "model": request.model,
+        "choices": [{"index": 0, "message": {"role": "assistant", "content": clean_content}, "finish_reason": "stop"}]
+    }