cribl-control-plane 0.0.43__tar.gz → 0.0.44__tar.gz

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: cribl-control-plane
-Version: 0.0.43
+Version: 0.0.44
 Summary: Python Client SDK Generated by Speakeasy.
 Author: Speakeasy
 Requires-Python: >=3.9.2

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/pyproject.toml

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

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/_hooks/clientcredentials.py

@@ -188,7 +188,7 @@ class ClientCredentialsHook(SDKInitHook, BeforeRequestHook, AfterErrorHook):
 
         response_data = response.json()
 
-        if response_data.get("token_type") != "Bearer":
+        if response_data.get("token_type", "").lower() != "bearer":
             raise Exception("Unexpected token type from token endpoint")
 
         expires_at = None

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "cribl-control-plane"
-__version__: str = "0.0.43"
-__openapi_doc_version__: str = "4.14.0-alpha.1757087755104-f460003f"
-__gen_version__: str = "2.686.7"
-__user_agent__: str = "speakeasy-sdk/python 0.0.43 2.686.7 4.14.0-alpha.1757087755104-f460003f cribl-control-plane"
+__version__: str = "0.0.44"
+__openapi_doc_version__: str = "4.14.0-alpha.1757401089572-3ca67bed"
+__gen_version__: str = "2.696.0"
+__user_agent__: str = "speakeasy-sdk/python 0.0.44 2.696.0 4.14.0-alpha.1757401089572-3ca67bed cribl-control-plane"
 
 try:
     if __package__ is not None:

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/acl.py

@@ -14,13 +14,15 @@ from typing import Any, Mapping, Optional
 class ACL(BaseSDK):
     teams: Teams
 
-    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.teams = Teams(self.sdk_configuration)
+        self.teams = Teams(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def get(
         self,

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/auth_sdk.py

@@ -3,15 +3,18 @@
 from .basesdk import BaseSDK
 from .sdkconfiguration import SDKConfiguration
 from cribl_control_plane.tokens import Tokens
+from typing import Optional
 
 
 class AuthSDK(BaseSDK):
     tokens: Tokens
 
-    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.tokens = Tokens(self.sdk_configuration)
+        self.tokens = Tokens(self.sdk_configuration, parent_ref=self.parent_ref)

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/basesdk.py

@@ -19,9 +19,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()

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/commits.py

@@ -14,13 +14,15 @@ from typing import Any, List, Mapping, Optional
 class Commits(BaseSDK):
     files: CommitsFiles
 
-    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.files = CommitsFiles(self.sdk_configuration)
+        self.files = CommitsFiles(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def create(
         self,

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/destinations.py

@@ -18,14 +18,16 @@ class Destinations(BaseSDK):
     pq: DestinationsPq
     samples: Samples
 
-    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.pq = DestinationsPq(self.sdk_configuration)
-        self.samples = Samples(self.sdk_configuration)
+        self.pq = DestinationsPq(self.sdk_configuration, parent_ref=self.parent_ref)
+        self.samples = Samples(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def list(
         self,

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/errors/__init__.py

@@ -1,12 +1,13 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
+from .criblcontrolplaneerror import CriblControlPlaneError
 from typing import TYPE_CHECKING
 from importlib import import_module
 import builtins
+import sys
 
 if TYPE_CHECKING:
     from .apierror import APIError
-    from .criblcontrolplaneerror import CriblControlPlaneError
     from .error import Error, ErrorData
     from .healthstatus_error import HealthStatusError, HealthStatusErrorData
     from .no_response_error import NoResponseError
@@ -25,7 +26,6 @@ __all__ = [
 
 _dynamic_imports: dict[str, str] = {
     "APIError": ".apierror",
-    "CriblControlPlaneError": ".criblcontrolplaneerror",
     "Error": ".error",
     "ErrorData": ".error",
     "HealthStatusError": ".healthstatus_error",
@@ -35,6 +35,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:
@@ -43,7 +55,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:

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/groups_configs.py

@@ -3,15 +3,20 @@
 from .basesdk import BaseSDK
 from .sdkconfiguration import SDKConfiguration
 from cribl_control_plane.configs_versions import ConfigsVersions
+from typing import Optional
 
 
 class GroupsConfigs(BaseSDK):
     versions: ConfigsVersions
 
-    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.versions = ConfigsVersions(self.sdk_configuration)
+        self.versions = ConfigsVersions(
+            self.sdk_configuration, parent_ref=self.parent_ref
+        )

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/groups_sdk.py

@@ -18,14 +18,16 @@ class GroupsSDK(BaseSDK):
     configs: GroupsConfigs
     acl: ACL
 
-    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.configs = GroupsConfigs(self.sdk_configuration)
-        self.acl = ACL(self.sdk_configuration)
+        self.configs = GroupsConfigs(self.sdk_configuration, parent_ref=self.parent_ref)
+        self.acl = ACL(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def list(
         self,

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/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 .addhectokenrequest import (
@@ -2971,6 +2972,8 @@ if TYPE_CHECKING:
         OutputSyslog,
         OutputSyslogBackpressureBehavior,
         OutputSyslogCompression,
+        OutputSyslogHost,
+        OutputSyslogHostTypedDict,
         OutputSyslogMaximumTLSVersion,
         OutputSyslogMessageFormat,
         OutputSyslogMinimumTLSVersion,
@@ -2980,6 +2983,7 @@ if TYPE_CHECKING:
         OutputSyslogProtocol,
         OutputSyslogQueueFullBehavior,
         OutputSyslogSeverity,
+        OutputSyslogTLS,
         OutputSyslogTLSSettingsClientSide,
         OutputSyslogTLSSettingsClientSideTypedDict,
         OutputSyslogType,
@@ -5892,6 +5896,8 @@ __all__ = [
     "OutputSyslog",
     "OutputSyslogBackpressureBehavior",
     "OutputSyslogCompression",
+    "OutputSyslogHost",
+    "OutputSyslogHostTypedDict",
     "OutputSyslogMaximumTLSVersion",
     "OutputSyslogMessageFormat",
     "OutputSyslogMinimumTLSVersion",
@@ -5901,6 +5907,7 @@ __all__ = [
     "OutputSyslogProtocol",
     "OutputSyslogQueueFullBehavior",
     "OutputSyslogSeverity",
+    "OutputSyslogTLS",
     "OutputSyslogTLSSettingsClientSide",
     "OutputSyslogTLSSettingsClientSideTypedDict",
     "OutputSyslogType",
@@ -8855,6 +8862,8 @@ _dynamic_imports: dict[str, str] = {
     "OutputSyslog": ".outputsyslog",
     "OutputSyslogBackpressureBehavior": ".outputsyslog",
     "OutputSyslogCompression": ".outputsyslog",
+    "OutputSyslogHost": ".outputsyslog",
+    "OutputSyslogHostTypedDict": ".outputsyslog",
     "OutputSyslogMaximumTLSVersion": ".outputsyslog",
     "OutputSyslogMessageFormat": ".outputsyslog",
     "OutputSyslogMinimumTLSVersion": ".outputsyslog",
@@ -8864,6 +8873,7 @@ _dynamic_imports: dict[str, str] = {
     "OutputSyslogProtocol": ".outputsyslog",
     "OutputSyslogQueueFullBehavior": ".outputsyslog",
     "OutputSyslogSeverity": ".outputsyslog",
+    "OutputSyslogTLS": ".outputsyslog",
     "OutputSyslogTLSSettingsClientSide": ".outputsyslog",
     "OutputSyslogTLSSettingsClientSideTypedDict": ".outputsyslog",
     "OutputSyslogType": ".outputsyslog",
@@ -9117,6 +9127,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:
@@ -9125,7 +9147,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:

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/models/inputconfluentcloud.py

@@ -341,6 +341,8 @@ class InputConfluentCloudAuthenticationTypedDict(TypedDict):
 
     disabled: NotRequired[bool]
     mechanism: NotRequired[InputConfluentCloudSASLMechanism]
+    oauth_enabled: NotRequired[bool]
+    r"""Enable OAuth authentication"""
 
 
 class InputConfluentCloudAuthentication(BaseModel):
@@ -352,6 +354,11 @@ class InputConfluentCloudAuthentication(BaseModel):
         InputConfluentCloudSASLMechanism.PLAIN
     )
 
+    oauth_enabled: Annotated[Optional[bool], pydantic.Field(alias="oauthEnabled")] = (
+        False
+    )
+    r"""Enable OAuth authentication"""
+
 
 class InputConfluentCloudMetadatumTypedDict(TypedDict):
     name: str

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/models/inputkafka.py

@@ -260,6 +260,8 @@ class InputKafkaAuthenticationTypedDict(TypedDict):
 
     disabled: NotRequired[bool]
     mechanism: NotRequired[InputKafkaSASLMechanism]
+    oauth_enabled: NotRequired[bool]
+    r"""Enable OAuth authentication"""
 
 
 class InputKafkaAuthentication(BaseModel):
@@ -269,6 +271,11 @@ class InputKafkaAuthentication(BaseModel):
 
     mechanism: Optional[InputKafkaSASLMechanism] = InputKafkaSASLMechanism.PLAIN
 
+    oauth_enabled: Annotated[Optional[bool], pydantic.Field(alias="oauthEnabled")] = (
+        False
+    )
+    r"""Enable OAuth authentication"""
+
 
 class InputKafkaMinimumTLSVersion(str, Enum):
     TL_SV1 = "TLSv1"

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/models/outputazuredataexplorer.py

@@ -336,6 +336,8 @@ class OutputAzureDataExplorerTypedDict(TypedDict):
     pq_mode: NotRequired[OutputAzureDataExplorerMode]
     r"""In Error mode, PQ writes events to the filesystem if the Destination is unavailable. In Backpressure mode, PQ writes events to the filesystem when it detects backpressure from the Destination. In Always On mode, PQ always writes events to the filesystem."""
     pq_controls: NotRequired[OutputAzureDataExplorerPqControlsTypedDict]
+    empty_dir_cleanup_sec: NotRequired[float]
+    r"""How frequently, in seconds, to clean up empty directories"""
 
 
 class OutputAzureDataExplorer(BaseModel):
@@ -611,3 +613,8 @@ class OutputAzureDataExplorer(BaseModel):
     pq_controls: Annotated[
         Optional[OutputAzureDataExplorerPqControls], pydantic.Field(alias="pqControls")
     ] = None
+
+    empty_dir_cleanup_sec: Annotated[
+        Optional[float], pydantic.Field(alias="emptyDirCleanupSec")
+    ] = 300
+    r"""How frequently, in seconds, to clean up empty directories"""

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/models/outputconfluentcloud.py

@@ -294,6 +294,8 @@ class OutputConfluentCloudAuthenticationTypedDict(TypedDict):
 
     disabled: NotRequired[bool]
     mechanism: NotRequired[OutputConfluentCloudSASLMechanism]
+    oauth_enabled: NotRequired[bool]
+    r"""Enable OAuth authentication"""
 
 
 class OutputConfluentCloudAuthentication(BaseModel):
@@ -305,6 +307,11 @@ class OutputConfluentCloudAuthentication(BaseModel):
         OutputConfluentCloudSASLMechanism.PLAIN
     )
 
+    oauth_enabled: Annotated[Optional[bool], pydantic.Field(alias="oauthEnabled")] = (
+        False
+    )
+    r"""Enable OAuth authentication"""
+
 
 class OutputConfluentCloudBackpressureBehavior(str, Enum):
     r"""How to handle events when all receivers are exerting backpressure"""

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/models/outputkafka.py

@@ -215,6 +215,8 @@ class OutputKafkaAuthenticationTypedDict(TypedDict):
 
     disabled: NotRequired[bool]
     mechanism: NotRequired[OutputKafkaSASLMechanism]
+    oauth_enabled: NotRequired[bool]
+    r"""Enable OAuth authentication"""
 
 
 class OutputKafkaAuthentication(BaseModel):
@@ -224,6 +226,11 @@ class OutputKafkaAuthentication(BaseModel):
 
     mechanism: Optional[OutputKafkaSASLMechanism] = OutputKafkaSASLMechanism.PLAIN
 
+    oauth_enabled: Annotated[Optional[bool], pydantic.Field(alias="oauthEnabled")] = (
+        False
+    )
+    r"""Enable OAuth authentication"""
+
 
 class OutputKafkaMinimumTLSVersion(str, Enum):
     TL_SV1 = "TLSv1"

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/models/outputsyslog.py

@@ -186,6 +186,43 @@ class OutputSyslogPqControls(BaseModel):
     pass
 
 
+class OutputSyslogTLS(str, Enum):
+    r"""Whether to inherit TLS configs from group setting or disable TLS"""
+
+    INHERIT = "inherit"
+    OFF = "off"
+
+
+class OutputSyslogHostTypedDict(TypedDict):
+    host: str
+    r"""The hostname of the receiver"""
+    port: NotRequired[float]
+    r"""The port to connect to on the provided host"""
+    tls: NotRequired[OutputSyslogTLS]
+    r"""Whether to inherit TLS configs from group setting or disable TLS"""
+    servername: NotRequired[str]
+    r"""Servername to use if establishing a TLS connection. If not specified, defaults to connection host (if not an IP); otherwise, uses the global TLS settings."""
+    weight: NotRequired[float]
+    r"""Assign a weight (>0) to each endpoint to indicate its traffic-handling capability"""
+
+
+class OutputSyslogHost(BaseModel):
+    host: str
+    r"""The hostname of the receiver"""
+
+    port: Optional[float] = 9997
+    r"""The port to connect to on the provided host"""
+
+    tls: Optional[OutputSyslogTLS] = OutputSyslogTLS.INHERIT
+    r"""Whether to inherit TLS configs from group setting or disable TLS"""
+
+    servername: Optional[str] = None
+    r"""Servername to use if establishing a TLS connection. If not specified, defaults to connection host (if not an IP); otherwise, uses the global TLS settings."""
+
+    weight: Optional[float] = 1
+    r"""Assign a weight (>0) to each endpoint to indicate its traffic-handling capability"""
+
+
 class OutputSyslogTypedDict(TypedDict):
     type: OutputSyslogType
     id: NotRequired[str]
@@ -247,6 +284,16 @@ class OutputSyslogTypedDict(TypedDict):
     pq_mode: NotRequired[OutputSyslogMode]
     r"""In Error mode, PQ writes events to the filesystem if the Destination is unavailable. In Backpressure mode, PQ writes events to the filesystem when it detects backpressure from the Destination. In Always On mode, PQ always writes events to the filesystem."""
     pq_controls: NotRequired[OutputSyslogPqControlsTypedDict]
+    dns_resolve_period_sec: NotRequired[float]
+    r"""The interval in which to re-resolve any hostnames and pick up destinations from A records"""
+    load_balance_stats_period_sec: NotRequired[float]
+    r"""How far back in time to keep traffic stats for load balancing purposes"""
+    max_concurrent_senders: NotRequired[float]
+    r"""Maximum number of concurrent connections (per Worker Process). A random set of IPs will be picked on every DNS resolution period. Use 0 for unlimited."""
+    exclude_self: NotRequired[bool]
+    r"""Exclude all IPs of the current host from the list of any resolved hostnames"""
+    hosts: NotRequired[List[OutputSyslogHostTypedDict]]
+    r"""Set of hosts to load-balance data to."""
 
 
 class OutputSyslog(BaseModel):
@@ -379,3 +426,24 @@ class OutputSyslog(BaseModel):
     pq_controls: Annotated[
         Optional[OutputSyslogPqControls], pydantic.Field(alias="pqControls")
     ] = None
+
+    dns_resolve_period_sec: Annotated[
+        Optional[float], pydantic.Field(alias="dnsResolvePeriodSec")
+    ] = 600
+    r"""The interval in which to re-resolve any hostnames and pick up destinations from A records"""
+
+    load_balance_stats_period_sec: Annotated[
+        Optional[float], pydantic.Field(alias="loadBalanceStatsPeriodSec")
+    ] = 300
+    r"""How far back in time to keep traffic stats for load balancing purposes"""
+
+    max_concurrent_senders: Annotated[
+        Optional[float], pydantic.Field(alias="maxConcurrentSenders")
+    ] = 0
+    r"""Maximum number of concurrent connections (per Worker Process). A random set of IPs will be picked on every DNS resolution period. Use 0 for unlimited."""
+
+    exclude_self: Annotated[Optional[bool], pydantic.Field(alias="excludeSelf")] = False
+    r"""Exclude all IPs of the current host from the list of any resolved hostnames"""
+
+    hosts: Optional[List[OutputSyslogHost]] = None
+    r"""Set of hosts to load-balance data to."""

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/nodes.py

@@ -14,13 +14,15 @@ from typing import Any, Mapping, Optional
 class Nodes(BaseSDK):
     summaries: Summaries
 
-    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.summaries = Summaries(self.sdk_configuration)
+        self.summaries = Summaries(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def list(
         self,

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/sdk.py

@@ -10,6 +10,7 @@ from cribl_control_plane._hooks import SDKHooks
 from cribl_control_plane.types import OptionalNullable, UNSET
 import httpx
 import importlib
+import sys
 from typing import Callable, Optional, TYPE_CHECKING, Union, cast
 import weakref
 
@@ -119,6 +120,7 @@ class CriblControlPlane(BaseSDK):
                 timeout_ms=timeout_ms,
                 debug_logger=debug_logger,
             ),
+            parent_ref=self,
         )
 
         hooks = SDKHooks()
@@ -138,13 +140,24 @@ class CriblControlPlane(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:

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/sources.py

@@ -16,13 +16,15 @@ class Sources(BaseSDK):
 
     hec_tokens: HecTokens
 
-    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.hec_tokens = HecTokens(self.sdk_configuration)
+        self.hec_tokens = HecTokens(self.sdk_configuration, parent_ref=self.parent_ref)
 
     def list(
         self,

{cribl_control_plane-0.0.43 → cribl_control_plane-0.0.44}/src/cribl_control_plane/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}"