cribl-control-plane 0.0.42__py3-none-any.whl → 0.0.44__py3-none-any.whl

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/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "cribl-control-plane"
-__version__: str = "0.0.42"
-__openapi_doc_version__: str = "4.14.0-alpha.1757000684650-0bcbd7ed"
-__gen_version__: str = "2.686.7"
-__user_agent__: str = "speakeasy-sdk/python 0.0.42 2.686.7 4.14.0-alpha.1757000684650-0bcbd7ed 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/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/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/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/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/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/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/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/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,