2024-04-06 15:27:21 -04:00
|
|
|
from .Provider.BingCreateImages import BingCreateImages
|
|
|
|
from .Provider.needs_auth import Gemini, OpenaiChat
|
|
|
|
from ..Provider.You import You
|
|
|
|
|
|
|
|
from .types import Client
|
|
|
|
|
2024-04-06 15:24:48 -04:00
|
|
|
class ImageModels():
|
|
|
|
gemini = Gemini
|
|
|
|
openai = OpenaiChat
|
|
|
|
you = You
|
|
|
|
|
|
|
|
def __init__(self, client: Client) -> None:
|
|
|
|
self.client = client
|
|
|
|
self.default = BingCreateImages(proxy=self.client.get_proxy())
|
|
|
|
|
|
|
|
def get(self, name: str, default: ImageProvider = None) -> ImageProvider:
|
|
|
|
return getattr(self, name) if hasattr(self, name) else default or self.default
|