diff options
Diffstat (limited to 'g4f/Provider')
| -rw-r--r-- | g4f/Provider/Airforce.py | 2 | ||||
| -rw-r--r-- | g4f/Provider/airforce/AirforceChat.py | 8 |
2 files changed, 7 insertions, 3 deletions
diff --git a/g4f/Provider/Airforce.py b/g4f/Provider/Airforce.py index 8ea0a174..c7ae44c0 100644 --- a/g4f/Provider/Airforce.py +++ b/g4f/Provider/Airforce.py @@ -20,7 +20,7 @@ class Airforce(AsyncGeneratorProvider, ProviderModelMixin): supports_message_history = AirforceChat.supports_message_history default_model = AirforceChat.default_model - models = [*AirforceChat.text_models, *AirforceImage.image_models] + models = [*AirforceChat.models, *AirforceImage.models] model_aliases = { **AirforceChat.model_aliases, diff --git a/g4f/Provider/airforce/AirforceChat.py b/g4f/Provider/airforce/AirforceChat.py index fc375270..cec911a3 100644 --- a/g4f/Provider/airforce/AirforceChat.py +++ b/g4f/Provider/airforce/AirforceChat.py @@ -1,8 +1,8 @@ from __future__ import annotations import re import json -from aiohttp import ClientSession import requests +from aiohttp import ClientSession from typing import List from ...typing import AsyncResult, Messages @@ -21,7 +21,11 @@ def clean_response(text: str) -> str: ] for pattern in patterns: text = re.sub(pattern, '', text) - return text.strip() + + # Remove the <|im_end|> token if present + text = text.replace("<|im_end|>", "").strip() + + return text def split_message(message: str, max_length: int = 1000) -> List[str]: """Splits the message into chunks of a given length (max_length)""" |