From 804a80bc7cb788a77af169242b84c892ec4fc638 Mon Sep 17 00:00:00 2001 From: H Lohaus Date: Sun, 24 Nov 2024 17:43:45 +0100 Subject: Arm2 (#2414) * Fix arm v7 build / improve api * Update stubs.py * Fix unit tests --- g4f/Provider/PollinationsAI.py | 3 +-- g4f/Provider/needs_auth/Gemini.py | 1 + 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'g4f/Provider') diff --git a/g4f/Provider/PollinationsAI.py b/g4f/Provider/PollinationsAI.py index a30f896d..e82222b1 100644 --- a/g4f/Provider/PollinationsAI.py +++ b/g4f/Provider/PollinationsAI.py @@ -46,8 +46,7 @@ class PollinationsAI(OpenaiAPI): seed: str = None, **kwargs ) -> AsyncResult: - if model: - model = cls.get_model(model) + model = cls.get_model(model) if model in cls.image_models: if prompt is None: prompt = messages[-1]["content"] diff --git a/g4f/Provider/needs_auth/Gemini.py b/g4f/Provider/needs_auth/Gemini.py index 7b84f284..e7c9de23 100644 --- a/g4f/Provider/needs_auth/Gemini.py +++ b/g4f/Provider/needs_auth/Gemini.py @@ -313,6 +313,7 @@ class Conversation(BaseConversation): self.conversation_id = conversation_id self.response_id = response_id self.choice_id = choice_id + async def iter_filter_base64(response_iter: AsyncIterator[bytes]) -> AsyncIterator[bytes]: search_for = b'[["wrb.fr","XqA3Ic","[\\"' end_with = b'\\' -- cgit v1.2.3