1 files changed, 12 insertions, 3 deletions
diff --git a/g4f/client/types.py b/g4f/client/types.py
index 1c33e6c2..b21ff03a 100644
--- a/g4f/client/types.py
+++ b/g4f/client/types.py
@@ -5,7 +5,18 @@ ImageProvider = Union[BaseProvider, object]
 Proxies = Union[dict, str]
 IterResponse = Iterator[Union[ChatCompletion, ChatCompletionChunk]]
 
-class Client():
+class ClientProxyMixin():
+    def get_proxy(self) -> Union[str, None]:
+        if isinstance(self.proxies, str):
+            return self.proxies
+        elif self.proxies is None:
+            return os.environ.get("G4F_PROXY")
+        elif "all" in self.proxies:
+            return self.proxies["all"]
+        elif "https" in self.proxies:
+            return self.proxies["https"]
+
+class Client(ClientProxyMixin):
     def __init__(
         self,
         api_key: str = None,
@@ -16,5 +27,3 @@ class Client():
     ) -> None:
         self.api_key: str = api_key
         self.proxies: Proxies = proxies
-        self.chat: Chat = Chat(self, provider)
-        self.images: Images = Images(self, image_provider)