simplify code

This commit is contained in:
better629 2024-01-30 14:32:57 +08:00
parent e7cd90f7f8
commit df728a034e

View file

@ -262,9 +262,6 @@ class OpenAILLM(BaseLLM):
)
imgs = []
for item in res.data:
if resp_format == "url":
img_url_or_b64 = item.url
else:
img_url_or_b64 = item.b64_json
img_url_or_b64 = item.url if resp_format == "url" else item.b64_json
imgs.append(decode_image(img_url_or_b64))
return imgs