gr4vy 1.1.34__tar.gz → 1.2.0__tar.gz

{gr4vy-1.1.34 → gr4vy-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: gr4vy
-Version: 1.1.34
+Version: 1.2.0
 Summary: Python Client SDK Generated by Speakeasy.
 Author: Gr4vy
 Requires-Python: >=3.9.2

{gr4vy-1.1.34 → gr4vy-1.2.0}/pyproject.toml

@@ -1,7 +1,7 @@
 
 [project]
 name = "gr4vy"
-version = "1.1.34"
+version = "1.2.0"
 description = "Python Client SDK Generated by Speakeasy."
 authors = [{ name = "Gr4vy" },]
 readme = "README-PYPI.md"
@@ -44,6 +44,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.pytest.ini_options]
 asyncio_default_fixture_loop_scope = "function"
+asyncio_mode = "auto"
 pythonpath = ["src"]
 
 [tool.mypy]

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "gr4vy"
-__version__: str = "1.1.34"
+__version__: str = "1.2.0"
 __openapi_doc_version__: str = "1.0.0"
-__gen_version__: str = "2.686.7"
-__user_agent__: str = "speakeasy-sdk/python 1.1.34 2.686.7 1.0.0 gr4vy"
+__gen_version__: str = "2.687.13"
+__user_agent__: str = "speakeasy-sdk/python 1.2.0 2.687.13 1.0.0 gr4vy"
 
 try:
     if __package__ is not None:

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/account_updater.py

@@ -3,15 +3,18 @@
 from .basesdk import BaseSDK
 from .sdkconfiguration import SDKConfiguration
 from gr4vy.jobs import Jobs
+from typing import Optional
 
 
 class AccountUpdater(BaseSDK):
     jobs: Jobs
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.jobs = Jobs(self.sdk_configuration)
+        self.jobs = Jobs(self.sdk_configuration, parent_ref=self.parent_ref)

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/basesdk.py

@@ -11,9 +11,19 @@ from urllib.parse import parse_qs, urlparse
 
 class BaseSDK:
     sdk_configuration: SDKConfiguration
+    parent_ref: Optional[object] = None
+    """
+    Reference to the root SDK instance, if any. This will prevent it from
+    being garbage collected while there are active streams.
+    """
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
+    def __init__(
+        self,
+        sdk_config: SDKConfiguration,
+        parent_ref: Optional[object] = None,
+    ) -> None:
         self.sdk_configuration = sdk_config
+        self.parent_ref = parent_ref
 
     def _get_url(self, base_url, url_variables):
         sdk_url, sdk_variables = self.sdk_configuration.get_server_details()

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/buyers_sdk.py

@@ -19,15 +19,23 @@ class BuyersSDK(BaseSDK):
     gift_cards: BuyersGiftCards
     shipping_details: BuyersShippingDetails
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.payment_methods = BuyersPaymentMethods(self.sdk_configuration)
-        self.gift_cards = BuyersGiftCards(self.sdk_configuration)
-        self.shipping_details = BuyersShippingDetails(self.sdk_configuration)
+        self.payment_methods = BuyersPaymentMethods(
+            self.sdk_configuration, parent_ref=self.parent_ref
+        )
+        self.gift_cards = BuyersGiftCards(
+            self.sdk_configuration, parent_ref=self.parent_ref
+        )
+        self.shipping_details = BuyersShippingDetails(
+            self.sdk_configuration, parent_ref=self.parent_ref
+        )
 
     def list(
         self,

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/digital_wallets_sdk.py

@@ -16,14 +16,16 @@ class DigitalWalletsSDK(BaseSDK):
     sessions: Sessions
     domains: Domains
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.sessions = Sessions(self.sdk_configuration)
-        self.domains = Domains(self.sdk_configuration)
+        self.sessions = Sessions(self.sdk_configuration, parent_ref=self.parent_ref)
+        self.domains = Domains(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def create(
         self,

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/errors/__init__.py

@@ -1,8 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
+from .gr4vyerror import Gr4vyError
 from typing import TYPE_CHECKING
 from importlib import import_module
 import builtins
+import sys
 
 if TYPE_CHECKING:
     from .apierror import APIError
@@ -17,7 +19,6 @@ if TYPE_CHECKING:
     from .error500 import Error500, Error500Data
     from .error502 import Error502, Error502Data
     from .error504 import Error504, Error504Data
-    from .gr4vyerror import Gr4vyError
     from .httpvalidationerror import HTTPValidationError, HTTPValidationErrorData
     from .no_response_error import NoResponseError
     from .responsevalidationerror import ResponseValidationError
@@ -77,7 +78,6 @@ _dynamic_imports: dict[str, str] = {
     "Error502Data": ".error502",
     "Error504": ".error504",
     "Error504Data": ".error504",
-    "Gr4vyError": ".gr4vyerror",
     "HTTPValidationError": ".httpvalidationerror",
     "HTTPValidationErrorData": ".httpvalidationerror",
     "NoResponseError": ".no_response_error",
@@ -85,6 +85,18 @@ _dynamic_imports: dict[str, str] = {
 }
 
 
+def dynamic_import(modname, retries=3):
+    for attempt in range(retries):
+        try:
+            return import_module(modname, __package__)
+        except KeyError:
+            # Clear any half-initialized module and retry
+            sys.modules.pop(modname, None)
+            if attempt == retries - 1:
+                break
+    raise KeyError(f"Failed to import module '{modname}' after {retries} attempts")
+
+
 def __getattr__(attr_name: str) -> object:
     module_name = _dynamic_imports.get(attr_name)
     if module_name is None:
@@ -93,7 +105,7 @@ def __getattr__(attr_name: str) -> object:
         )
 
     try:
-        module = import_module(module_name, __package__)
+        module = dynamic_import(module_name)
         result = getattr(module, attr_name)
         return result
     except ImportError as e:

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/gift_cards_sdk.py

@@ -15,13 +15,15 @@ from typing import Any, Dict, List, Mapping, Optional, Union
 class GiftCardsSDK(BaseSDK):
     balances: Balances
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.balances = Balances(self.sdk_configuration)
+        self.balances = Balances(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def get(
         self,

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/models/__init__.py

@@ -3,6 +3,7 @@
 from typing import TYPE_CHECKING
 from importlib import import_module
 import builtins
+import sys
 
 if TYPE_CHECKING:
     from .accountsreceivablesreportspec import (
@@ -88,6 +89,7 @@ if TYPE_CHECKING:
     from .braintreedynamicdatafieldsoptions import (
         BraintreeDynamicDataFieldsOptions,
         BraintreeDynamicDataFieldsOptionsTypedDict,
+        VaultPaymentMethodCriteria,
     )
     from .braintreeoptions import BraintreeOptions, BraintreeOptionsTypedDict
     from .browserinfo import BrowserInfo, BrowserInfoTypedDict, UserDevice
@@ -1993,6 +1995,7 @@ __all__ = [
     "UserStatus",
     "ValidationError",
     "ValidationErrorTypedDict",
+    "VaultPaymentMethodCriteria",
     "VerifyCredentials",
     "VerifyCredentialsTypedDict",
     "VerifyPaymentServiceCredentialsGlobals",
@@ -2092,6 +2095,7 @@ _dynamic_imports: dict[str, str] = {
     "BillingDetailsOutputTypedDict": ".billingdetails_output",
     "BraintreeDynamicDataFieldsOptions": ".braintreedynamicdatafieldsoptions",
     "BraintreeDynamicDataFieldsOptionsTypedDict": ".braintreedynamicdatafieldsoptions",
+    "VaultPaymentMethodCriteria": ".braintreedynamicdatafieldsoptions",
     "BraintreeOptions": ".braintreeoptions",
     "BraintreeOptionsTypedDict": ".braintreeoptions",
     "BrowserInfo": ".browserinfo",
@@ -2902,6 +2906,18 @@ _dynamic_imports: dict[str, str] = {
 }
 
 
+def dynamic_import(modname, retries=3):
+    for attempt in range(retries):
+        try:
+            return import_module(modname, __package__)
+        except KeyError:
+            # Clear any half-initialized module and retry
+            sys.modules.pop(modname, None)
+            if attempt == retries - 1:
+                break
+    raise KeyError(f"Failed to import module '{modname}' after {retries} attempts")
+
+
 def __getattr__(attr_name: str) -> object:
     module_name = _dynamic_imports.get(attr_name)
     if module_name is None:
@@ -2910,7 +2926,7 @@ def __getattr__(attr_name: str) -> object:
         )
 
     try:
-        module = import_module(module_name, __package__)
+        module = dynamic_import(module_name)
         result = getattr(module, attr_name)
         return result
     except ImportError as e:

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/models/braintreedynamicdatafieldsoptions.py

@@ -1,9 +1,24 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from gr4vy.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from gr4vy.types import (
+    BaseModel,
+    Nullable,
+    OptionalNullable,
+    UNSET,
+    UNSET_SENTINEL,
+    UnrecognizedStr,
+)
+from gr4vy.utils import validate_open_enum
 from pydantic import model_serializer
-from typing_extensions import NotRequired, TypedDict
+from pydantic.functional_validators import PlainValidator
+from typing import Literal, Union
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+VaultPaymentMethodCriteria = Union[
+    Literal["ALWAYS", "ON_SUCCESSFUL_TRANSACTION"], UnrecognizedStr
+]
 
 
 class BraintreeDynamicDataFieldsOptionsTypedDict(TypedDict):
@@ -11,6 +26,8 @@ class BraintreeDynamicDataFieldsOptionsTypedDict(TypedDict):
     r"""Passes the 3DS status to the Braintree API using `customFields` with the key set to the value of `three_ds_auth_status`"""
     purchase_order_number: NotRequired[Nullable[str]]
     r"""Passes the `transaction.purchaseOrderNumber` field when creating a new transaction."""
+    vault_payment_method_criteria: NotRequired[Nullable[VaultPaymentMethodCriteria]]
+    r"""Passes the `vaultPaymentMethodCriteria` field when creating a new transaction."""
 
 
 class BraintreeDynamicDataFieldsOptions(BaseModel):
@@ -20,10 +37,24 @@ class BraintreeDynamicDataFieldsOptions(BaseModel):
     purchase_order_number: OptionalNullable[str] = UNSET
     r"""Passes the `transaction.purchaseOrderNumber` field when creating a new transaction."""
 
+    vault_payment_method_criteria: Annotated[
+        OptionalNullable[VaultPaymentMethodCriteria],
+        PlainValidator(validate_open_enum(False)),
+    ] = UNSET
+    r"""Passes the `vaultPaymentMethodCriteria` field when creating a new transaction."""
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["three_ds_auth_status", "purchase_order_number"]
-        nullable_fields = ["three_ds_auth_status", "purchase_order_number"]
+        optional_fields = [
+            "three_ds_auth_status",
+            "purchase_order_number",
+            "vault_payment_method_criteria",
+        ]
+        nullable_fields = [
+            "three_ds_auth_status",
+            "purchase_order_number",
+            "vault_payment_method_criteria",
+        ]
         null_default_fields = []
 
         serialized = handler(self)

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/models/internal/__init__.py

@@ -3,6 +3,7 @@
 from typing import TYPE_CHECKING
 from importlib import import_module
 import builtins
+import sys
 
 if TYPE_CHECKING:
     from .globals import Globals, GlobalsTypedDict
@@ -15,6 +16,18 @@ _dynamic_imports: dict[str, str] = {
 }
 
 
+def dynamic_import(modname, retries=3):
+    for attempt in range(retries):
+        try:
+            return import_module(modname, __package__)
+        except KeyError:
+            # Clear any half-initialized module and retry
+            sys.modules.pop(modname, None)
+            if attempt == retries - 1:
+                break
+    raise KeyError(f"Failed to import module '{modname}' after {retries} attempts")
+
+
 def __getattr__(attr_name: str) -> object:
     module_name = _dynamic_imports.get(attr_name)
     if module_name is None:
@@ -23,7 +36,7 @@ def __getattr__(attr_name: str) -> object:
         )
 
     try:
-        module = import_module(module_name, __package__)
+        module = dynamic_import(module_name)
         result = getattr(module, attr_name)
         return result
     except ImportError as e:

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/models/transactionevent.py

@@ -44,6 +44,7 @@ Name = Union[
         "payment-connector-response-transaction-capture-failed",
         "payment-connector-response-transaction-capture-declined",
         "payment-connector-response-transaction-cancel-succeeded",
+        "payment-connector-response-transaction-cancel-pending",
         "payment-connector-response-transaction-cancel-failed",
         "payment-connector-response-transaction-void-succeeded",
         "payment-connector-response-transaction-void-declined",

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/payment_methods_network_tokens.py

@@ -14,13 +14,17 @@ from typing import Any, Mapping, Optional
 class PaymentMethodsNetworkTokens(BaseSDK):
     cryptogram: NetworkTokensCryptogram
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.cryptogram = NetworkTokensCryptogram(self.sdk_configuration)
+        self.cryptogram = NetworkTokensCryptogram(
+            self.sdk_configuration, parent_ref=self.parent_ref
+        )
 
     def list(
         self,

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/payment_methods_sdk.py

@@ -19,16 +19,20 @@ class PaymentMethodsSDK(BaseSDK):
     payment_service_tokens: PaymentMethodsPaymentServiceTokens
     network_tokens: PaymentMethodsNetworkTokens
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
         self.payment_service_tokens = PaymentMethodsPaymentServiceTokens(
-            self.sdk_configuration
+            self.sdk_configuration, parent_ref=self.parent_ref
+        )
+        self.network_tokens = PaymentMethodsNetworkTokens(
+            self.sdk_configuration, parent_ref=self.parent_ref
         )
-        self.network_tokens = PaymentMethodsNetworkTokens(self.sdk_configuration)
 
     def list(
         self,

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/reports_sdk.py

@@ -15,13 +15,15 @@ from typing import Any, Dict, List, Mapping, Optional, Union
 class ReportsSDK(BaseSDK):
     executions: Executions
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.executions = Executions(self.sdk_configuration)
+        self.executions = Executions(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def list(
         self,

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/sdk.py

@@ -11,6 +11,7 @@ from gr4vy.models import internal
 from gr4vy.types import OptionalNullable, UNSET
 import httpx
 import importlib
+import sys
 from typing import Any, Callable, Dict, Optional, TYPE_CHECKING, Union, cast
 import weakref
 
@@ -173,6 +174,7 @@ class Gr4vy(BaseSDK):
                 timeout_ms=timeout_ms,
                 debug_logger=debug_logger,
             ),
+            parent_ref=self,
         )
 
         hooks = SDKHooks()
@@ -197,13 +199,24 @@ class Gr4vy(BaseSDK):
             self.sdk_configuration.async_client_supplied,
         )
 
+    def dynamic_import(self, modname, retries=3):
+        for attempt in range(retries):
+            try:
+                return importlib.import_module(modname)
+            except KeyError:
+                # Clear any half-initialized module and retry
+                sys.modules.pop(modname, None)
+                if attempt == retries - 1:
+                    break
+        raise KeyError(f"Failed to import module '{modname}' after {retries} attempts")
+
     def __getattr__(self, name: str):
         if name in self._sub_sdk_map:
             module_path, class_name = self._sub_sdk_map[name]
             try:
-                module = importlib.import_module(module_path)
+                module = self.dynamic_import(module_path)
                 klass = getattr(module, class_name)
-                instance = klass(self.sdk_configuration)
+                instance = klass(self.sdk_configuration, parent_ref=self)
                 setattr(self, name, instance)
                 return instance
             except ImportError as e:

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/transactions.py

@@ -20,15 +20,21 @@ class Transactions(BaseSDK):
     events: Events
     settlements: TransactionsSettlements
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.refunds = TransactionsRefunds(self.sdk_configuration)
-        self.events = Events(self.sdk_configuration)
-        self.settlements = TransactionsSettlements(self.sdk_configuration)
+        self.refunds = TransactionsRefunds(
+            self.sdk_configuration, parent_ref=self.parent_ref
+        )
+        self.events = Events(self.sdk_configuration, parent_ref=self.parent_ref)
+        self.settlements = TransactionsSettlements(
+            self.sdk_configuration, parent_ref=self.parent_ref
+        )
 
     def list(
         self,

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/transactions_refunds.py

@@ -14,13 +14,15 @@ from typing import Any, Mapping, Optional
 class TransactionsRefunds(BaseSDK):
     all: All
 
-    def __init__(self, sdk_config: SDKConfiguration) -> None:
-        BaseSDK.__init__(self, sdk_config)
+    def __init__(
+        self, sdk_config: SDKConfiguration, parent_ref: Optional[object] = None
+    ) -> None:
+        BaseSDK.__init__(self, sdk_config, parent_ref=parent_ref)
         self.sdk_configuration = sdk_config
         self._init_sdks()
 
     def _init_sdks(self):
-        self.all = All(self.sdk_configuration)
+        self.all = All(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def list(
         self,

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/utils/__init__.py

@@ -3,6 +3,7 @@
 from typing import TYPE_CHECKING
 from importlib import import_module
 import builtins
+import sys
 
 if TYPE_CHECKING:
     from .annotations import get_discriminator
@@ -162,6 +163,18 @@ _dynamic_imports: dict[str, str] = {
 }
 
 
+def dynamic_import(modname, retries=3):
+    for attempt in range(retries):
+        try:
+            return import_module(modname, __package__)
+        except KeyError:
+            # Clear any half-initialized module and retry
+            sys.modules.pop(modname, None)
+            if attempt == retries - 1:
+                break
+    raise KeyError(f"Failed to import module '{modname}' after {retries} attempts")
+
+
 def __getattr__(attr_name: str) -> object:
     module_name = _dynamic_imports.get(attr_name)
     if module_name is None:
@@ -170,9 +183,8 @@ def __getattr__(attr_name: str) -> object:
         )
 
     try:
-        module = import_module(module_name, __package__)
-        result = getattr(module, attr_name)
-        return result
+        module = dynamic_import(module_name)
+        return getattr(module, attr_name)
     except ImportError as e:
         raise ImportError(
             f"Failed to import {attr_name} from {module_name}: {e}"

{gr4vy-1.1.34 → gr4vy-1.2.0}/src/gr4vy/utils/eventstreaming.py

@@ -17,6 +17,9 @@ T = TypeVar("T")
 
 
 class EventStream(Generic[T]):
+    # Holds a reference to the SDK client to avoid it being garbage collected
+    # and cause termination of the underlying httpx client.
+    client_ref: Optional[object]
     response: httpx.Response
     generator: Generator[T, None, None]
 
@@ -25,9 +28,11 @@ class EventStream(Generic[T]):
         response: httpx.Response,
         decoder: Callable[[str], T],
         sentinel: Optional[str] = None,
+        client_ref: Optional[object] = None,
     ):
         self.response = response
         self.generator = stream_events(response, decoder, sentinel)
+        self.client_ref = client_ref
 
     def __iter__(self):
         return self
@@ -43,6 +48,9 @@ class EventStream(Generic[T]):
 
 
 class EventStreamAsync(Generic[T]):
+    # Holds a reference to the SDK client to avoid it being garbage collected
+    # and cause termination of the underlying httpx client.
+    client_ref: Optional[object]
     response: httpx.Response
     generator: AsyncGenerator[T, None]
 
@@ -51,9 +59,11 @@ class EventStreamAsync(Generic[T]):
         response: httpx.Response,
         decoder: Callable[[str], T],
         sentinel: Optional[str] = None,
+        client_ref: Optional[object] = None,
     ):
         self.response = response
         self.generator = stream_events_async(response, decoder, sentinel)
+        self.client_ref = client_ref
 
     def __aiter__(self):
         return self