返回提交历史
Modified
g4f/Provider/needs_auth/OpenaiChat.py
+8
-12
XFEstudio/gpt4free
Fix generate image in OpenaiChat
db47d71f
代码差异
1 个文件
+8
-12
@@ -256,23 +256,19 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
256
256
@classmethod
257
257
async def get_generated_images(cls, session: StreamSession, auth_result: AuthResult, parts: list, prompt: str, conversation_id: str) -> AsyncIterator:
258
258
download_urls = []
259
for element in [parts] if isinstance(parts, str) else parts:
260
if isinstance(element, dict) and element.get("content_type") == "image_asset_pointer":
261
if not prompt:
262
prompt = element["metadata"]["dalle"]["prompt"]
263
element = element["asset_pointer"]
264
element = element.split("sediment://")[-1]
265
url = f"{cls.url}/backend-api/conversation/{conversation_id}/attachment/{element}/download"
266
debug.log(f"OpenaiChat: Downloading image: {url}")
259
element = element.split("sediment://")[-1]
260
url = f"{cls.url}/backend-api/conversation/{conversation_id}/attachment/{element}/download"
261
debug.log(f"OpenaiChat: Downloading image: {url}")
262
try:
267
263
async with session.get(url, headers=auth_result.headers) as response:
268
264
cls._update_request_args(auth_result, session)
269
265
await raise_for_status(response)
270
266
data = await response.json()
271
267
download_url = data.get("download_url")
272
if download_url is None:
273
print(data)
274
else:
275
download_urls.append(download_url)
268
download_urls.append(download_url)
269
except Exception as e:
270
debug.error("OpenaiChat: Download image failed")
271
debug.error(e)
276
272
return ImagePreview(download_urls, prompt)
277
273
278
274
@classmethod