oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-12-06 07:12:10 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Image: Add a revised_prompt field to API results for OpenAI compatibility

Browse source
This commit is contained in:
oobabooga 2025-12-04 17:41:09 -08:00
parent 56f2a9512f
commit 15c6e43597
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
extensions/openai/images.py
View file

@ -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