返回提交历史
Modified
etc/unittest/models.py
+3
-2
Modified
g4f/Provider/Kimi.py
+1
-0
Modified
g4f/Provider/PollinationsAI.py
+2
-1
Modified
g4f/models.py
+3
-3
XFEstudio/gpt4free
Fix unittests
2aa40bb8
代码差异
4 个文件
+9
-6
@@ -19,7 +19,7 @@ class TestProviderHasModel(unittest.TestCase):
19
19
if model.name in provider.model_aliases:
20
20
model_name = provider.model_aliases[model.name]
21
21
else:
22
model_name = model.name
22
model_name = model.get_long_name()
23
23
self.provider_has_model(provider, model_name)
24
24
25
25
def provider_has_model(self, provider: Type[BaseProvider], model: str):
@@ -29,7 +29,8 @@ class TestProviderHasModel(unittest.TestCase):
29
29
except (MissingRequirementsError, MissingAuthError):
30
30
return
31
31
if self.cache[provider.__name__]:
32
self.assertIn(model, self.cache[provider.__name__], provider.__name__)
32
if not model in provider.model_aliases:
33
self.assertIn(model, self.cache[provider.__name__], provider.__name__)
33
34
34
35
def test_all_providers_working(self):
35
36
for model, providers in __models__.values():
@@ -16,6 +16,7 @@ class Kimi(AsyncAuthedProvider, ProviderModelMixin):
16
16
active_by_default = True
17
17
default_model = "kimi-k2"
18
18
models = [default_model]
19
model_aliases = {"moonshotai/Kimi-K2-Instruct": default_model}
19
20
20
21
@classmethod
21
22
async def on_auth_async(cls, proxy: str = None, **kwargs) -> AsyncIterator:
@@ -88,7 +88,8 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
88
88
vision_models = [default_vision_model]
89
89
_models_loaded = False
90
90
model_aliases = {
91
"gpt-4.1-nano": "openai",
91
"gpt-4.1-mini": "openai",
92
"gpt-4.1-nano": "openai-fast",
92
93
"gpt-4.1": "openai-large",
93
94
"o4-mini": "openai-reasoning",
94
95
"qwen-2.5-coder-32b": "qwen-coder",
@@ -197,20 +197,20 @@ default_vision = VisionModel(
197
197
gpt_4 = Model(
198
198
name = 'gpt-4',
199
199
base_provider = 'OpenAI',
200
best_provider = IterListProvider([Blackbox, PollinationsAI, Copilot, Yqcloud, WeWordle, OpenaiChat])
200
best_provider = IterListProvider([Blackbox, Copilot, Yqcloud, WeWordle, OpenaiChat])
201
201
)
202
202
203
203
# gpt-4o
204
204
gpt_4o = VisionModel(
205
205
name = 'gpt-4o',
206
206
base_provider = 'OpenAI',
207
best_provider = IterListProvider([Blackbox, PollinationsAI, OpenaiChat])
207
best_provider = IterListProvider([Blackbox, OpenaiChat])
208
208
)
209
209
210
210
gpt_4o_mini = Model(
211
211
name = 'gpt-4o-mini',
212
212
base_provider = 'OpenAI',
213
best_provider = IterListProvider([Blackbox, PollinationsAI, Chatai, OIVSCodeSer2, Startnest, OpenaiChat])
213
best_provider = IterListProvider([Blackbox, Chatai, OIVSCodeSer2, Startnest, OpenaiChat])
214
214
)
215
215
216
216
gpt_4o_mini_audio = AudioModel(