diff options
author | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2024-05-20 01:28:08 +0200 |
---|---|---|
committer | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2024-05-20 01:28:08 +0200 |
commit | 80b2e9b80774c558cb863841e7e8214fbc0eca67 (patch) | |
tree | 449863877c368037c5951697d37e9bd997cf679a /g4f/Provider/needs_auth/Gemini.py | |
parent | Create windows release, improve styling (diff) | |
download | gpt4free-80b2e9b80774c558cb863841e7e8214fbc0eca67.tar gpt4free-80b2e9b80774c558cb863841e7e8214fbc0eca67.tar.gz gpt4free-80b2e9b80774c558cb863841e7e8214fbc0eca67.tar.bz2 gpt4free-80b2e9b80774c558cb863841e7e8214fbc0eca67.tar.lz gpt4free-80b2e9b80774c558cb863841e7e8214fbc0eca67.tar.xz gpt4free-80b2e9b80774c558cb863841e7e8214fbc0eca67.tar.zst gpt4free-80b2e9b80774c558cb863841e7e8214fbc0eca67.zip |
Diffstat (limited to 'g4f/Provider/needs_auth/Gemini.py')
-rw-r--r-- | g4f/Provider/needs_auth/Gemini.py | 9 |
1 files changed, 2 insertions, 7 deletions
diff --git a/g4f/Provider/needs_auth/Gemini.py b/g4f/Provider/needs_auth/Gemini.py index f9b1c4a5..71cc8d81 100644 --- a/g4f/Provider/needs_auth/Gemini.py +++ b/g4f/Provider/needs_auth/Gemini.py @@ -4,7 +4,6 @@ import os import json import random import re -import base64 from aiohttp import ClientSession, BaseConnector @@ -193,14 +192,10 @@ class Gemini(AsyncGeneratorProvider): yield content if image_prompt: images = [image[0][3][3] for image in response_part[4][0][12][7][0]] - resolved_images = [] if response_format == "b64_json": - for image in images: - async with client.get(image) as response: - data = base64.b64encode(await response.content.read()).decode() - resolved_images.append(data) - yield ImageDataResponse(resolved_images, image_prompt) + yield ImageResponse(images, image_prompt, {"cookies": cls._cookies}) else: + resolved_images = [] preview = [] for image in images: async with client.get(image, allow_redirects=False) as fetch: |