From 15c6e43597203a0468501621f0e9ea443d22e8fa Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 4 Dec 2025 17:41:09 -0800
Subject: [PATCH] Image: Add a revised_prompt field to API results for OpenAI
 compatibility

---
 extensions/openai/images.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/extensions/openai/images.py b/extensions/openai/images.py
index ef3f4169..1337771a 100644
--- a/extensions/openai/images.py
+++ b/extensions/openai/images.py
@@ -47,10 +47,15 @@ def generations(request):
     resp = {'created': int(time.time()), 'data': []}
     for img in images:
         b64 = _image_to_base64(img)
+
+        image_obj = {'revised_prompt': request.prompt}
+
         if request.response_format == 'b64_json':
-            resp['data'].append({'b64_json': b64})
+            image_obj['b64_json'] = b64
         else:
-            resp['data'].append({'url': f'data:image/png;base64,{b64}'})
+            image_obj['url'] = f'data:image/png;base64,{b64}'
+
+        resp['data'].append(image_obj)
 
     return resp