Update main.py

main.py

@@ -197,7 +197,14 @@ async def chat_completions(request: ChatRequest):
 
     response_content = ""
     async for chunk in async_generator:
-        response_content += chunk if isinstance(chunk, str) else chunk.content
+        if isinstance(chunk, str):
+            response_content += chunk
+        else:
+            response_content += chunk.content
 
-    return Response(content=response_content.strip(), media_type="text/plain")  # Return plain text
+    # 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