返回提交历史
Modified
g4f/Provider/needs_auth/OpenaiChat.py
+3
-3
XFEstudio/gpt4free
OpenaiChat: add auth_result.headers to ImageResponse (#3211)
* Update OpenaiChat.py --------- Co-authored-by: H Lohaus <hlohaus@users.noreply.github.com>
d3220833
代码差异
1 个文件
+3
-3
@@ -283,7 +283,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
283
283
return messages
284
284
285
285
@classmethod
286
async def get_generated_image(cls, session: StreamSession, auth_result: AuthResult, element: Union[dict, str], prompt: str = None, conversation_id: str = None) -> AsyncIterator:
286
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:
287
287
download_urls = []
288
288
is_sediment = False
289
289
if prompt is None:
@@ -319,7 +319,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
319
319
debug.error("OpenaiChat: Download image failed")
320
320
debug.error(e)
321
321
if download_urls:
322
return ImagePreview(download_urls, prompt) if is_sediment else ImageResponse(download_urls, prompt)
322
return ImagePreview(download_urls, prompt, {"headers":auth_result.headers}) if is_sediment else ImageResponse(download_urls, prompt, {"headers":auth_result.headers})
323
323
324
324
@classmethod
325
325
async def create_authed(
@@ -643,7 +643,7 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
643
643
if sources.list:
644
644
yield sources
645
645
if conversation.generated_images:
646
yield ImageResponse(conversation.generated_images.urls, conversation.prompt)
646
yield ImageResponse(conversation.generated_images.urls, conversation.prompt, {"headers":auth_result.headers})
647
647
conversation.generated_images = None
648
648
conversation.prompt = None
649
649
if return_conversation: