Commit d24c7e18 authored by Your Name's avatar Your Name

Fix http_request bug in image generation and add http_request parameter to save_image_response

parent 16ce04c5
...@@ -3463,7 +3463,7 @@ def get_cfg_scale(): ...@@ -3463,7 +3463,7 @@ def get_cfg_scale():
return cfg_scale return cfg_scale
# Helper function to save generated images and return response dict # Helper function to save generated images and return response dict
def save_image_response(img, request_format="base64"): def save_image_response(img, request_format="base64", http_request=None):
""" """
Save image to file path if configured, return response dict. Save image to file path if configured, return response dict.
...@@ -3691,7 +3691,7 @@ async def create_image_generation(request: ImageGenerationRequest, http_request: ...@@ -3691,7 +3691,7 @@ async def create_image_generation(request: ImageGenerationRequest, http_request:
import io import io
# Use helper function to save and get response # Use helper function to save and get response
img_data = save_image_response(img, request.response_format) img_data = save_image_response(img, request.response_format, http_request)
images.append(img_data) images.append(img_data)
return { return {
...@@ -3780,7 +3780,7 @@ async def create_image_generation(request: ImageGenerationRequest, http_request: ...@@ -3780,7 +3780,7 @@ async def create_image_generation(request: ImageGenerationRequest, http_request:
for img in result: for img in result:
# Use helper function to save and get response # Use helper function to save and get response
img_data = save_image_response(img) img_data = save_image_response(img, http_request=http_request)
images.append(img_data) images.append(img_data)
return { return {
...@@ -3878,7 +3878,7 @@ async def create_image_generation(request: ImageGenerationRequest, http_request: ...@@ -3878,7 +3878,7 @@ async def create_image_generation(request: ImageGenerationRequest, http_request:
for img in result: for img in result:
# Use helper function to save and get response # Use helper function to save and get response
img_data = save_image_response(img) img_data = save_image_response(img, http_request=http_request)
images.append(img_data) images.append(img_data)
return { return {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment