3 files changed, 8 insertions, 16 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)"""
diff --git a/g4f/models.py b/g4f/models.py
index 3b82270e..87a076a8 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -463,15 +463,6 @@ openhermes_2_5 = Model(
     best_provider = Airforce
 )
 
-
-### Pawan ### 
-cosmosrp = Model(
-    name = 'cosmosrp',
-    base_provider = 'Pawan',
-    best_provider = Airforce
-)
-
-
 ### Liquid ### 
 lfm_40b = Model(
     name = 'lfm-40b',
@@ -666,6 +657,7 @@ class ModelUtils:
 
                 
 ### Microsoft ###
+'phi-2': phi_2,
 'phi-3.5-mini': phi_3_5_mini,
 
 
@@ -764,10 +756,6 @@ class ModelUtils:
 ### Teknium ###   
 'openhermes-2.5': openhermes_2_5,
         
-        
-### Pawan ###   
-'cosmosrp': cosmosrp,
-
 
 ### Liquid ###   
 'lfm-40b': lfm_40b,