mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-15 14:51:19 -08:00
Code Issues Projects Releases Packages Wiki Activity

OpenaiChat: add auth_result.headers to ImageResponse (#3211)

Browse source 
* Update OpenaiChat.py

---------

Co-authored-by: H Lohaus <hlohaus@users.noreply.github.com>
This commit is contained in:
Ammar 2025-10-18 18:33:17 +03:00 • committed by GitHub
parent 5c49074b4f
commit d322083305
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
g4f/Provider/needs_auth/OpenaiChat.py
View file

@ -283,7 +283,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
return messages
@classmethod
async def get_generated_image(cls, session: StreamSession, auth_result: AuthResult, element: Union[dict, str], prompt: str = None, conversation_id: str = None) -> AsyncIterator:
async def get_generated_image(cls, session: StreamSession, auth_result: AuthResult, element: Union[dict, str], prompt: str = None, conversation_id: str = None) -> ImagePreview|ImageResponse|None:
download_urls = []
is_sediment = False
if prompt is None:
@ -319,7 +319,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
debug.error("OpenaiChat: Download image failed")
debug.error(e)
if download_urls:
return ImagePreview(download_urls, prompt) if is_sediment else ImageResponse(download_urls, prompt)
return ImagePreview(download_urls, prompt, {"headers":auth_result.headers}) if is_sediment else ImageResponse(download_urls, prompt, {"headers":auth_result.headers})
@classmethod
async def create_authed(
@ -643,7 +643,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
if sources.list:
yield sources
if conversation.generated_images:
yield ImageResponse(conversation.generated_images.urls, conversation.prompt)
yield ImageResponse(conversation.generated_images.urls, conversation.prompt, {"headers":auth_result.headers})
conversation.generated_images = None
conversation.prompt = None
if return_conversation: