blaxel 0.2.31__py3-none-any.whl → 0.2.31rc120__py3-none-any.whl

blaxel/__init__.py

@@ -4,9 +4,9 @@ from .core.common.autoload import autoload
 from .core.common.env import env
 from .core.common.settings import settings
 
-__version__ = "0.2.31"
-__commit__ = "43ad41354e2f27bae2471030eeaf702cf8e811ba"
+__version__ = "0.2.31.pre-120"
+__commit__ = "a23256023f1d955467e2ee297a96b3d321172acf"
 __sentry_dsn__ = "https://9711de13cd02b285ca4378c01de8dc30@o4508714045276160.ingest.us.sentry.io/4510461121462272"
 __all__ = ["autoload", "settings", "env"]
 
-autoload()
+autoload()

blaxel/core/agents/__init__.py

@@ -51,13 +51,11 @@ class BlAgent:
             body = json.dumps(body)
 
         # Merge settings headers with provided headers
-        merged_headers = {
-            **settings.headers,
-            "Content-Type": "application/json",
-            **headers,
-        }
+        merged_headers = {**settings.headers, "Content-Type": "application/json", **headers}
 
-        return client.get_httpx_client().post(url, headers=merged_headers, data=body, params=params)
+        return client.get_httpx_client().post(
+            url, headers=merged_headers, data=body, params=params
+        )
 
     async def acall(self, url, input_data, headers: dict = {}, params: dict = {}):
         logger.debug(f"Agent Calling: {self.name}")
@@ -66,11 +64,7 @@ class BlAgent:
             body = json.dumps(body)
 
         # Merge settings headers with provided headers
-        merged_headers = {
-            **settings.headers,
-            "Content-Type": "application/json",
-            **headers,
-        }
+        merged_headers = {**settings.headers, "Content-Type": "application/json", **headers}
 
         return await client.get_async_httpx_client().post(
             url, headers=merged_headers, data=body, params=params
@@ -126,5 +120,4 @@ async def get_agent_metadata(name):
     except Exception:
         return None
 
-
-__all__ = ["BlAgent", "bl_agent", "get_agent_metadata"]
+__all__ = ["BlAgent", "bl_agent", "get_agent_metadata"]

blaxel/core/authentication/__init__.py

@@ -34,8 +34,7 @@ def get_credentials() -> CredentialsType | None:
 
     if os.environ.get("BL_CLIENT_CREDENTIALS"):
         return CredentialsType(
-            client_credentials=os.environ.get("BL_CLIENT_CREDENTIALS"),
-            workspace=get_workspace(),
+            client_credentials=os.environ.get("BL_CLIENT_CREDENTIALS"), workspace=get_workspace()
         )
 
     try:

blaxel/core/authentication/devicemode.py

@@ -24,7 +24,6 @@ class DeviceLogin:
         client_id (str): The client ID for the device.
         scope (str): The scope of the authentication.
     """
-
     client_id: str
     scope: str
 
@@ -43,7 +42,6 @@ class DeviceLoginResponse:
         verification_uri (str): URI for user to verify device login.
         verification_uri_complete (str): Complete URI including the user code for verification.
     """
-
     client_id: str
     device_code: str
     user_code: str
@@ -63,7 +61,6 @@ class DeviceLoginFinalizeRequest:
         client_id (str): The client ID for finalizing the device login.
         device_code (str): The device code to finalize login.
     """
-
     grant_type: str
     client_id: str
     device_code: str
@@ -81,7 +78,6 @@ class DeviceMode(BlaxelAuth):
     """
     DeviceMode auth that authenticates requests using a device code.
     """
-
     def get_headers(self) -> Dict[str, str]:
         """
         Retrieves the authentication headers containing the Bearer token and workspace information.
@@ -165,11 +161,7 @@ class DeviceMode(BlaxelAuth):
         }
 
         try:
-            response = post(
-                url,
-                json=refresh_data,
-                headers={"Content-Type": "application/json"},
-            )
+            response = post(url, json=refresh_data, headers={"Content-Type": "application/json"})
             response.raise_for_status()
             finalize_response = DeviceLoginFinalizeResponse(**response.json())
 

blaxel/core/authentication/oauth.py

@@ -11,7 +11,6 @@ class OauthTokenData:
     headers: dict[str, str] = field(default_factory=dict)
     authenticated: bool | None = False
 
-
 @dataclass
 class OauthTokenResponse:
     access_token: str
@@ -19,16 +18,14 @@ class OauthTokenResponse:
     expires_in: int
     token_type: str
 
-
 @dataclass
 class OauthTokenError:
     error: str
 
-
 async def oauth_token(
     options: OauthTokenData,
     client: Client | None = None,
-    throw_on_error: bool = False,
+    throw_on_error: bool = False
 ) -> Union[OauthTokenResponse, OauthTokenError]:
     """
     Get a new OAuth token.
@@ -41,17 +38,13 @@ async def oauth_token(
     Returns:
         The OAuth token response or error
     """
-    response = (
-        await (client or default_client)
-        .get_async_httpx_client()
-        .post(
-            url="/oauth/token",
-            json=options.body or {},
-            headers=options.headers or {},
-        )
+    response = await (client or default_client).get_async_httpx_client().post(
+        url="/oauth/token",
+        json=options.body or {},
+        headers=options.headers or {},
     )
     if response.status_code >= 400:
         if throw_on_error:
             raise Exception(f"Failed to get OAuth token: {response.text}")
         return OauthTokenError(error=response.text)
-    return OauthTokenResponse(**response.json())
+    return OauthTokenResponse(**response.json())

blaxel/core/authentication/types.py

@@ -17,7 +17,6 @@ class CredentialsType(BaseModel):
     expires_in: int | None = Field(default=None, description="The expiration time")
     workspace: str | None = Field(default=None, description="The workspace")
 
-
 class BlaxelAuth(Auth):
     def __init__(self, credentials: CredentialsType, workspace_name: str, base_url: str):
         """

blaxel/core/cache/cache.py

@@ -14,26 +14,19 @@ def _load_cache() -> None:
     Load cache data from .cache.yaml file if it exists.
     """
     try:
-        if os.path.exists(".cache.yaml"):
-            with open(".cache.yaml", encoding="utf-8") as file:
+        if os.path.exists('.cache.yaml'):
+            with open('.cache.yaml', encoding='utf-8') as file:
                 cache_string = file.read()
                 cache_data = yaml.safe_load_all(cache_string)
 
                 for doc in cache_data:
-                    if (
-                        doc
-                        and isinstance(doc, dict)
-                        and "kind" in doc
-                        and "metadata" in doc
-                        and "name" in doc["metadata"]
-                    ):
+                    if doc and isinstance(doc, dict) and 'kind' in doc and 'metadata' in doc and 'name' in doc['metadata']:
                         cache_key = f"{doc['kind']}/{doc['metadata']['name']}"
                         _cache[cache_key] = doc
     except Exception:
         # Silently fail if cache file doesn't exist or is invalid
         pass
 
-
 # Initialize cache on module import
 _load_cache()
 

blaxel/core/client/api/agents/list_agent_revisions.py

@@ -36,9 +36,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Revisi
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["RevisionMetadata"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["RevisionMetadata"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/compute/delete_sandbox_preview_token.py

@@ -22,9 +22,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> DeleteSandboxPreviewTokenResponse200 | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> DeleteSandboxPreviewTokenResponse200 | None:
     if response.status_code == 200:
         response_200 = DeleteSandboxPreviewTokenResponse200.from_dict(response.json())
 
@@ -35,9 +33,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[DeleteSandboxPreviewTokenResponse200]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[DeleteSandboxPreviewTokenResponse200]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/compute/start_sandbox.py

@@ -34,9 +34,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, S
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, StartSandbox]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, StartSandbox]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/compute/stop_sandbox.py

@@ -34,9 +34,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, S
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, StopSandbox]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, StopSandbox]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/default/list_sandbox_hub_definitions.py

@@ -18,9 +18,7 @@ def _get_kwargs() -> dict[str, Any]:
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> list["SandboxDefinition"] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> list["SandboxDefinition"] | None:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -36,9 +34,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["SandboxDefinition"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["SandboxDefinition"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/functions/list_function_revisions.py

@@ -36,9 +36,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Revisi
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["RevisionMetadata"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["RevisionMetadata"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/images/cleanup_images.py

@@ -29,9 +29,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> CleanupImage
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[CleanupImagesResponse200]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[CleanupImagesResponse200]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/integrations/list_integration_connections.py

@@ -18,9 +18,7 @@ def _get_kwargs() -> dict[str, Any]:
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> list["IntegrationConnection"] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> list["IntegrationConnection"] | None:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -36,9 +34,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["IntegrationConnection"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["IntegrationConnection"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/invitations/list_all_pending_invitations.py

@@ -18,9 +18,7 @@ def _get_kwargs() -> dict[str, Any]:
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[Any, list["PendingInvitationRender"]] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, list["PendingInvitationRender"]] | None:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()

blaxel/core/client/api/jobs/create_job_execution.py

@@ -51,9 +51,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, J
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, JobExecution]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, JobExecution]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/jobs/delete_job_execution.py

@@ -41,9 +41,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, J
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, JobExecution]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, JobExecution]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/jobs/get_job_execution.py

@@ -41,9 +41,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, J
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, JobExecution]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, JobExecution]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/jobs/list_job_executions.py

@@ -32,9 +32,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[Any, list["JobExecution"]] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, list["JobExecution"]] | None:
     if response.status_code == 200:
         response_200 = []
         _response_200 = response.json()
@@ -56,9 +54,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, list["JobExecution"]]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, list["JobExecution"]]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/jobs/list_job_revisions.py

@@ -36,9 +36,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Revisi
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["RevisionMetadata"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["RevisionMetadata"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/locations/list_locations.py

@@ -34,9 +34,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Locati
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["LocationResponse"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["LocationResponse"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/models/list_model_revisions.py

@@ -36,9 +36,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Revisi
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["RevisionMetadata"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["RevisionMetadata"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/service_accounts/create_workspace_service_account.py

@@ -35,9 +35,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> CreateWorkspaceServiceAccountResponse200 | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> CreateWorkspaceServiceAccountResponse200 | None:
     if response.status_code == 200:
         response_200 = CreateWorkspaceServiceAccountResponse200.from_dict(response.json())
 
@@ -48,9 +46,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[CreateWorkspaceServiceAccountResponse200]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[CreateWorkspaceServiceAccountResponse200]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/service_accounts/delete_workspace_service_account.py

@@ -22,9 +22,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> DeleteWorkspaceServiceAccountResponse200 | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> DeleteWorkspaceServiceAccountResponse200 | None:
     if response.status_code == 200:
         response_200 = DeleteWorkspaceServiceAccountResponse200.from_dict(response.json())
 
@@ -35,9 +33,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[DeleteWorkspaceServiceAccountResponse200]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[DeleteWorkspaceServiceAccountResponse200]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/service_accounts/get_workspace_service_accounts.py

@@ -27,9 +27,7 @@ def _parse_response(
         response_200 = []
         _response_200 = response.json()
         for response_200_item_data in _response_200:
-            response_200_item = GetWorkspaceServiceAccountsResponse200Item.from_dict(
-                response_200_item_data
-            )
+            response_200_item = GetWorkspaceServiceAccountsResponse200Item.from_dict(response_200_item_data)
 
             response_200.append(response_200_item)
 

blaxel/core/client/api/service_accounts/update_workspace_service_account.py

@@ -36,9 +36,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> UpdateWorkspaceServiceAccountResponse200 | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> UpdateWorkspaceServiceAccountResponse200 | None:
     if response.status_code == 200:
         response_200 = UpdateWorkspaceServiceAccountResponse200.from_dict(response.json())
 
@@ -49,9 +47,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[UpdateWorkspaceServiceAccountResponse200]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[UpdateWorkspaceServiceAccountResponse200]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/volume_templates/list_volume_templates.py

@@ -34,9 +34,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> list["Volume
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[list["VolumeTemplate"]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[list["VolumeTemplate"]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/workspaces/accept_workspace_invitation.py

@@ -20,9 +20,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[Any, PendingInvitationAccept] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, PendingInvitationAccept] | None:
     if response.status_code == 200:
         response_200 = PendingInvitationAccept.from_dict(response.json())
 
@@ -36,9 +34,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, PendingInvitationAccept]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, PendingInvitationAccept]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/workspaces/invite_workspace_user.py

@@ -33,9 +33,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[Any, PendingInvitation] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, PendingInvitation] | None:
     if response.status_code == 200:
         response_200 = PendingInvitation.from_dict(response.json())
 
@@ -52,9 +50,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, PendingInvitation]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, PendingInvitation]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/api/workspaces/update_workspace_user_role.py

@@ -34,9 +34,7 @@ def _get_kwargs(
     return _kwargs
 
 
-def _parse_response(
-    *, client: Client, response: httpx.Response
-) -> Union[Any, WorkspaceUser] | None:
+def _parse_response(*, client: Client, response: httpx.Response) -> Union[Any, WorkspaceUser] | None:
     if response.status_code == 200:
         response_200 = WorkspaceUser.from_dict(response.json())
 
@@ -53,9 +51,7 @@ def _parse_response(
         return None
 
 
-def _build_response(
-    *, client: Client, response: httpx.Response
-) -> Response[Union[Any, WorkspaceUser]]:
+def _build_response(*, client: Client, response: httpx.Response) -> Response[Union[Any, WorkspaceUser]]:
     return Response(
         status_code=HTTPStatus(response.status_code),
         content=response.content,

blaxel/core/client/client.py

@@ -43,9 +43,7 @@ class Client:
     _headers: dict[str, str] = field(factory=dict, kw_only=True, alias="headers")
     _auth: httpx.Auth = field(default=None, alias="auth")
     _timeout: httpx.Timeout | None = field(default=None, kw_only=True, alias="timeout")
-    _verify_ssl: Union[str, bool, ssl.SSLContext] = field(
-        default=True, kw_only=True, alias="verify_ssl"
-    )
+    _verify_ssl: Union[str, bool, ssl.SSLContext] = field(default=True, kw_only=True, alias="verify_ssl")
     _follow_redirects: bool = field(default=False, kw_only=True, alias="follow_redirects")
     _httpx_args: dict[str, Any] = field(factory=dict, kw_only=True, alias="httpx_args")
     _client: httpx.Client | None = field(default=None, init=False)
@@ -107,6 +105,12 @@ class Client:
     def get_httpx_client(self) -> httpx.Client:
         """Get the underlying httpx.Client, constructing a new one if not previously set"""
         if self._client is None:
+            # Set default limits if not provided in httpx_args to prevent TLS handshake issues
+            limits = self._httpx_args.get("limits")
+            if limits is None:
+                limits = httpx.Limits(max_connections=100, max_keepalive_connections=20)
+            # Disable HTTP/2 to prevent TLS handshake issues in certain environments
+            http2 = self._httpx_args.get("http2", False)
             self._client = httpx.Client(
                 base_url=self._base_url,
                 cookies=self._cookies,
@@ -115,7 +119,9 @@ class Client:
                 verify=self._verify_ssl,
                 follow_redirects=self._follow_redirects,
                 auth=self._auth,
-                **self._httpx_args,
+                limits=limits,
+                http2=http2,
+                **{k: v for k, v in self._httpx_args.items() if k not in ("limits", "http2")},
             )
         return self._client
 
@@ -139,6 +145,12 @@ class Client:
     def get_async_httpx_client(self) -> httpx.AsyncClient:
         """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
         if self._async_client is None:
+            # Set default limits if not provided in httpx_args to prevent TLS handshake issues
+            limits = self._httpx_args.get("limits")
+            if limits is None:
+                limits = httpx.Limits(max_connections=100, max_keepalive_connections=20)
+            # Disable HTTP/2 to prevent TLS handshake issues in certain environments
+            http2 = self._httpx_args.get("http2", False)
             self._async_client = httpx.AsyncClient(
                 base_url=self._base_url,
                 cookies=self._cookies,
@@ -147,7 +159,9 @@ class Client:
                 verify=self._verify_ssl,
                 follow_redirects=self._follow_redirects,
                 auth=self._auth,
-                **self._httpx_args,
+                limits=limits,
+                http2=http2,
+                **{k: v for k, v in self._httpx_args.items() if k not in ("limits", "http2")},
             )
         return self._async_client