diff --git a/src/python/library/tritonclient/http/aio/__init__.py b/src/python/library/tritonclient/http/aio/__init__.py index 405474975..78badb1af 100755 --- a/src/python/library/tritonclient/http/aio/__init__.py +++ b/src/python/library/tritonclient/http/aio/__init__.py @@ -107,6 +107,8 @@ def __init__( conn_timeout=60.0, ssl=False, ssl_context=None, + proxy: aiohttp.typedefs.StrOrURL | None = None, + proxy_auth: aiohttp.helpers.BasicAuth | None = None ): super().__init__() if url.startswith("http://") or url.startswith("https://"): @@ -118,6 +120,8 @@ def __init__( connector=self._conn, timeout=aiohttp.ClientTimeout(total=conn_timeout), auto_decompress=False, + proxy=proxy, + proxy_auth=proxy_auth, ) self._verbose = verbose