feat(g4f/api/__init__.py): Update image generation response format

kqlio67 · Nov 9, 2024 · b2b978c · b2b978c
1 parent c1e6276
commit b2b978c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/g4f/api/__init__.py b/g4f/api/__init__.py
@@ -223,7 +223,7 @@ async def generate_image(config: ImageGenerationConfig):
                     response_format=config.response_format
                 )
                 # Convert Image objects to dictionaries
-                response_data = [image.to_dict() for image in response.data]
+                response_data = [{"url": image.url, "b64_json": image.b64_json} for image in response.data]
                 return JSONResponse({"data": response_data})
             except Exception as e:
                 logging.exception(e)