summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--g4f/client/types.py15
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)