summaryrefslogtreecommitdiffstats
path: root/g4f
diff options
context:
space:
mode:
Diffstat (limited to 'g4f')
-rw-r--r--g4f/client/image_models.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/g4f/client/image_models.py b/g4f/client/image_models.py
new file mode 100644
index 00000000..0c8ea894
--- /dev/null
+++ b/g4f/client/image_models.py
@@ -0,0 +1,11 @@
+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