返回提交历史
Modified
g4f/client/image_models.py
+8
-11
XFEstudio/gpt4free
refactor(image-models): replace providers with ModelUtils
24d4ea05
代码差异
1 个文件
+8
-11
@@ -2,18 +2,15 @@ from __future__ import annotations
2
2
3
3
from .types import Client, ImageProvider
4
4
5
from ..Provider.BingCreateImages import BingCreateImages
6
from ..Provider.needs_auth import Gemini, OpenaiChat
7
from ..Provider.You import You
5
from ..models import ModelUtils
8
6
9
7
class ImageModels():
10
gemini = Gemini
11
openai = OpenaiChat
12
you = You
13
14
def __init__(self, client: Client) -> None:
8
def __init__(self, client):
15
9
self.client = client
16
self.default = BingCreateImages(proxy=self.client.get_proxy())
10
self.models = ModelUtils.convert
17
11
18
def get(self, name: str, default: ImageProvider = None) -> ImageProvider:
19
return getattr(self, name) if hasattr(self, name) else default or self.default
12
def get(self, name, default=None):
13
model = self.models.get(name)
14
if model and model.best_provider:
15
return model.best_provider
16
return default