From 7cd2b8cd14965cc9b03478f77c3e6f111cb0f769 Mon Sep 17 00:00:00 2001 From: kqlio67 Date: Sun, 20 Oct 2024 12:44:17 +0300 Subject: refactor(g4f/Provider/Airforce.py): update image generation prompt and models --- g4f/Provider/Airforce.py | 4 +--- g4f/models.py | 8 -------- 2 files changed, 1 insertion(+), 11 deletions(-) (limited to 'g4f') diff --git a/g4f/Provider/Airforce.py b/g4f/Provider/Airforce.py index ac2b48fa..015766f4 100644 --- a/g4f/Provider/Airforce.py +++ b/g4f/Provider/Airforce.py @@ -81,7 +81,6 @@ class Airforce(AsyncGeneratorProvider, ProviderModelMixin): 'flux-pixel', 'flux-4o', 'any-dark', - 'dall-e-3', ] models = [ @@ -153,7 +152,7 @@ class Airforce(AsyncGeneratorProvider, ProviderModelMixin): if seed is None: seed = random.randint(0, 100000) - prompt = messages[0]['content'] + prompt = messages[-1]['content'] async with ClientSession(headers=headers) as session: params = { @@ -226,7 +225,6 @@ class Airforce(AsyncGeneratorProvider, ProviderModelMixin): content = json_data['choices'][0]['message']['content'] part_response = content - # Видаляємо повідомлення про перевищення ліміту символів part_response = re.sub( r"One message exceeds the \d+chars per message limit\..+https:\/\/discord\.com\/invite\/\S+", '', diff --git a/g4f/models.py b/g4f/models.py index e84f9103..3998027e 100644 --- a/g4f/models.py +++ b/g4f/models.py @@ -1,4 +1,3 @@ -# g4f/models.py from __future__ import annotations from dataclasses import dataclass @@ -828,12 +827,6 @@ dalle_2 = Model( base_provider = 'OpenAI', best_provider = NexraDallE2 -) -dalle_3 = Model( - name = 'dalle-3', - base_provider = 'OpenAI', - best_provider = Airforce - ) dalle = Model( @@ -1124,7 +1117,6 @@ class ModelUtils: ### OpenAI ### 'dalle': dalle, 'dalle-2': dalle_2, -'dalle-3': dalle_3, 'dalle-mini': dalle_mini, -- cgit v1.2.3