mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-10-05 08:16:58 +08:00
0a070bdf10
* feat: introduce AnyProvider & LM Arena, overhaul model/provider logic
- **Provider additions & removals**
- Added `Provider/LMArenaProvider.py` with full async stream implementation and vision model support
- Registered `LMArenaProvider` in `Provider/__init__.py`; removed old `hf_space/LMArenaProvider.py`
- Created `providers/any_provider.py`; registers `AnyProvider` dynamically in `Provider`
- **Provider framework enhancements**
- `providers/base_provider.py`
- Added `video_models` and `audio_models` attributes
- `providers/retry_provider.py`
- Introduced `is_content()` helper; now treats `AudioResponse` as stream content
- **Cloudflare provider refactor**
- `Provider/Cloudflare.py`
- Re‑implemented `get_models()` with `read_models()` helper, `fallback_models`, robust nodriver/curl handling and model‑name cleaning
- **Other provider tweaks**
- `Provider/Copilot.py` – removed `"reasoning"` alias and initial `setOptions` WS message
- `Provider/PollinationsAI.py` & `PollinationsImage.py`
- Converted `audio_models` from list to dict, adjusted usage checks and labels
- `Provider/hf/__init__.py` – applies `model_aliases` remap before dispatch
- `Provider/hf_space/DeepseekAI_JanusPro7b.py` – now merges media before upload
- `needs_auth/Gemini.py` – dropped obsolete Gemini model entries
- `needs_auth/GigaChat.py` – added lowercase `"gigachat"` alias
- **API & client updates**
- Replaced `ProviderUtils` with new `Provider` map usage throughout API and GUI server
- Integrated `AnyProvider` as default fallback in `g4f/client` sync & async flows
- API endpoints now return counts of providers per model and filter by `x_ignored` header
- **GUI improvements**
- Updated JS labels with emoji icons, provider ignore logic, model count display
- **Model registry**
- Renamed base model `"GigaChat:latest"` ➜ `"gigachat"` in `models.py`
- **Miscellaneous**
- Added audio/video flags to GUI provider list
- Tightened error propagation in `retry_provider.raise_exceptions`
* Fix unittests
* fix: handle None conversation when accessing provider-specific data
- Modified `AnyProvider` class in `g4f/providers/any_provider.py`
- Updated logic to check if `conversation` is not None before accessing `provider.__name__` attribute
- Wrapped `getattr(conversation, provider.__name__, None)` block in an additional `if conversation is not None` condition
- Changed `setattr(conversation, provider.__name__, chunk)` to use `chunk.get_dict()` instead of the object directly
- Ensured consistent use of `JsonConversation` when modifying or assigning `conversation` data
* ```
feat: add provider string conversion & update IterListProvider call
- In g4f/client/__init__.py, within both Completions and AsyncCompletions, added a check to convert the provider from a string using convert_to_provider(provider) when applicable.
- In g4f/providers/any_provider.py, removed the second argument (False) from the IterListProvider constructor call in the async for loop.
```
---------
Co-authored-by: hlohaus <983577+hlohaus@users.noreply.github.com>
34 lines
1.4 KiB
Python
34 lines
1.4 KiB
Python
import unittest
|
|
from typing import Type
|
|
import asyncio
|
|
|
|
from g4f.models import __models__
|
|
from g4f.providers.base_provider import BaseProvider, ProviderModelMixin
|
|
from g4f.errors import MissingRequirementsError, MissingAuthError
|
|
|
|
class TestProviderHasModel(unittest.TestCase):
|
|
cache: dict = {}
|
|
|
|
def test_provider_has_model(self):
|
|
for model, providers in __models__.values():
|
|
for provider in providers:
|
|
if issubclass(provider, ProviderModelMixin):
|
|
if model.name in provider.model_aliases:
|
|
model_name = provider.model_aliases[model.name]
|
|
else:
|
|
model_name = model.name
|
|
self.provider_has_model(provider, model_name)
|
|
|
|
def provider_has_model(self, provider: Type[BaseProvider], model: str):
|
|
if provider.__name__ not in self.cache:
|
|
try:
|
|
self.cache[provider.__name__] = provider.get_models()
|
|
except (MissingRequirementsError, MissingAuthError):
|
|
return
|
|
if self.cache[provider.__name__]:
|
|
self.assertIn(model, self.cache[provider.__name__], provider.__name__)
|
|
|
|
def test_all_providers_working(self):
|
|
for model, providers in __models__.values():
|
|
for provider in providers:
|
|
self.assertTrue(provider.working, f"{provider.__name__} in {model.name}") |