ibee 0.1.0__py3-none-any.whl

ibee/__init__.py

@@ -0,0 +1,157 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .types import (
+        AttachNetworkRequest,
+        Bucket,
+        BucketList,
+        BucketSummary,
+        CloudVm,
+        CloudVmStatus,
+        DeleteResponse,
+        Error,
+        ErrorDetail,
+        GpuVm,
+        GpuVmStatus,
+        NetworkInterface,
+        NetworkInterfaceStatus,
+        OperationAccepted,
+        OperationStatus,
+        OperationStatusAction,
+        OperationStatusStatus,
+        PowerActionRequest,
+        Secret,
+        SecretList,
+        SecretStatus,
+        SecretStore,
+        SecretStoreList,
+        SecretStoreStatus,
+        SecretValue,
+        UpdateVmRequest,
+        VmMetrics,
+    )
+    from .errors import BadRequestError, ConflictError, ForbiddenError, NotFoundError, UnauthorizedError
+    from . import cloud_vms, gpu_vms, object_storage, secret_store
+    from ._default_clients import DefaultAioHttpClient, DefaultAsyncHttpxClient
+    from .client import AsyncIbee, Ibee
+    from .cloud_vms import CreateCloudVmRequestOsType
+    from .environment import IbeeEnvironment
+    from .gpu_vms import CreateGpuVmRequestOsType
+_dynamic_imports: typing.Dict[str, str] = {
+    "AsyncIbee": ".client",
+    "AttachNetworkRequest": ".types",
+    "BadRequestError": ".errors",
+    "Bucket": ".types",
+    "BucketList": ".types",
+    "BucketSummary": ".types",
+    "CloudVm": ".types",
+    "CloudVmStatus": ".types",
+    "ConflictError": ".errors",
+    "CreateCloudVmRequestOsType": ".cloud_vms",
+    "CreateGpuVmRequestOsType": ".gpu_vms",
+    "DefaultAioHttpClient": "._default_clients",
+    "DefaultAsyncHttpxClient": "._default_clients",
+    "DeleteResponse": ".types",
+    "Error": ".types",
+    "ErrorDetail": ".types",
+    "ForbiddenError": ".errors",
+    "GpuVm": ".types",
+    "GpuVmStatus": ".types",
+    "Ibee": ".client",
+    "IbeeEnvironment": ".environment",
+    "NetworkInterface": ".types",
+    "NetworkInterfaceStatus": ".types",
+    "NotFoundError": ".errors",
+    "OperationAccepted": ".types",
+    "OperationStatus": ".types",
+    "OperationStatusAction": ".types",
+    "OperationStatusStatus": ".types",
+    "PowerActionRequest": ".types",
+    "Secret": ".types",
+    "SecretList": ".types",
+    "SecretStatus": ".types",
+    "SecretStore": ".types",
+    "SecretStoreList": ".types",
+    "SecretStoreStatus": ".types",
+    "SecretValue": ".types",
+    "UnauthorizedError": ".errors",
+    "UpdateVmRequest": ".types",
+    "VmMetrics": ".types",
+    "cloud_vms": ".cloud_vms",
+    "gpu_vms": ".gpu_vms",
+    "object_storage": ".object_storage",
+    "secret_store": ".secret_store",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = [
+    "AsyncIbee",
+    "AttachNetworkRequest",
+    "BadRequestError",
+    "Bucket",
+    "BucketList",
+    "BucketSummary",
+    "CloudVm",
+    "CloudVmStatus",
+    "ConflictError",
+    "CreateCloudVmRequestOsType",
+    "CreateGpuVmRequestOsType",
+    "DefaultAioHttpClient",
+    "DefaultAsyncHttpxClient",
+    "DeleteResponse",
+    "Error",
+    "ErrorDetail",
+    "ForbiddenError",
+    "GpuVm",
+    "GpuVmStatus",
+    "Ibee",
+    "IbeeEnvironment",
+    "NetworkInterface",
+    "NetworkInterfaceStatus",
+    "NotFoundError",
+    "OperationAccepted",
+    "OperationStatus",
+    "OperationStatusAction",
+    "OperationStatusStatus",
+    "PowerActionRequest",
+    "Secret",
+    "SecretList",
+    "SecretStatus",
+    "SecretStore",
+    "SecretStoreList",
+    "SecretStoreStatus",
+    "SecretValue",
+    "UnauthorizedError",
+    "UpdateVmRequest",
+    "VmMetrics",
+    "cloud_vms",
+    "gpu_vms",
+    "object_storage",
+    "secret_store",
+]

ibee/_default_clients.py

@@ -0,0 +1,32 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import httpx
+
+SDK_DEFAULT_TIMEOUT = 60
+
+try:
+    import httpx_aiohttp  # type: ignore[import-not-found]
+except ImportError:
+
+    class DefaultAioHttpClient(httpx.AsyncClient):  # type: ignore
+        def __init__(self, **kwargs: typing.Any) -> None:
+            raise RuntimeError(
+                "To use the aiohttp client, install the aiohttp extra: pip install ibee[aiohttp]"
+            )
+
+else:
+
+    class DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+        def __init__(self, **kwargs: typing.Any) -> None:
+            kwargs.setdefault("timeout", SDK_DEFAULT_TIMEOUT)
+            kwargs.setdefault("follow_redirects", True)
+            super().__init__(**kwargs)
+
+
+class DefaultAsyncHttpxClient(httpx.AsyncClient):
+    def __init__(self, **kwargs: typing.Any) -> None:
+        kwargs.setdefault("timeout", SDK_DEFAULT_TIMEOUT)
+        kwargs.setdefault("follow_redirects", True)
+        super().__init__(**kwargs)

ibee/client.py

@@ -0,0 +1,290 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import httpx
+from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .core.logging import LogConfig, Logger
+from .environment import IbeeEnvironment
+
+if typing.TYPE_CHECKING:
+    from .cloud_vms.client import AsyncCloudVmsClient, CloudVmsClient
+    from .gpu_vms.client import AsyncGpuVmsClient, GpuVmsClient
+    from .object_storage.client import AsyncObjectStorageClient, ObjectStorageClient
+    from .secret_store.client import AsyncSecretStoreClient, SecretStoreClient
+
+
+class Ibee:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : typing.Optional[str]
+        The base url to use for requests from the client.
+
+    environment : IbeeEnvironment
+        The environment to use for requests from the client. from .environment import IbeeEnvironment
+
+
+
+        Defaults to IbeeEnvironment.DEFAULT
+
+
+
+    token : typing.Union[str, typing.Callable[[], str]]
+    headers : typing.Optional[typing.Dict[str, str]]
+        Additional headers to send with every request.
+
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    max_retries : typing.Optional[int]
+        The default maximum number of retries for failed requests. Defaults to 2. Per-request `max_retries` in `request_options` takes precedence over this value.
+
+    stream_reconnection_enabled : typing.Optional[bool]
+        Whether to automatically reconnect on stream disconnection for resumable streaming endpoints. Defaults to True. Per-request `stream_reconnection_enabled` in `request_options` takes precedence over this value.
+
+    max_stream_reconnection_attempts : typing.Optional[int]
+        The maximum number of reconnection attempts for resumable streaming endpoints. Defaults to no limit. Per-request `max_stream_reconnection_attempts` in `request_options` takes precedence over this value.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.Client]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    logging : typing.Optional[typing.Union[LogConfig, Logger]]
+        Configure logging for the SDK. Accepts a LogConfig dict with 'level' (debug/info/warn/error), 'logger' (custom logger implementation), and 'silent' (boolean, defaults to True) fields. You can also pass a pre-configured Logger instance.
+
+    Examples
+    --------
+    from ibee import Ibee
+
+    client = Ibee(
+        token="YOUR_TOKEN",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: IbeeEnvironment = IbeeEnvironment.DEFAULT,
+        token: typing.Union[str, typing.Callable[[], str]],
+        headers: typing.Optional[typing.Dict[str, str]] = None,
+        timeout: typing.Optional[float] = None,
+        max_retries: typing.Optional[int] = None,
+        stream_reconnection_enabled: typing.Optional[bool] = None,
+        max_stream_reconnection_attempts: typing.Optional[int] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.Client] = None,
+        logging: typing.Optional[typing.Union[LogConfig, Logger]] = None,
+    ):
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
+        _defaulted_max_retries = max_retries if max_retries is not None else 2
+        self._client_wrapper = SyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            token=token,
+            headers=headers,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
+            if follow_redirects is not None
+            else httpx.Client(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+            max_retries=_defaulted_max_retries,
+            stream_reconnection_enabled=stream_reconnection_enabled,
+            max_stream_reconnection_attempts=max_stream_reconnection_attempts,
+            logging=logging,
+        )
+        self._secret_store: typing.Optional[SecretStoreClient] = None
+        self._object_storage: typing.Optional[ObjectStorageClient] = None
+        self._cloud_vms: typing.Optional[CloudVmsClient] = None
+        self._gpu_vms: typing.Optional[GpuVmsClient] = None
+
+    @property
+    def secret_store(self):
+        if self._secret_store is None:
+            from .secret_store.client import SecretStoreClient  # noqa: E402
+
+            self._secret_store = SecretStoreClient(client_wrapper=self._client_wrapper)
+        return self._secret_store
+
+    @property
+    def object_storage(self):
+        if self._object_storage is None:
+            from .object_storage.client import ObjectStorageClient  # noqa: E402
+
+            self._object_storage = ObjectStorageClient(client_wrapper=self._client_wrapper)
+        return self._object_storage
+
+    @property
+    def cloud_vms(self):
+        if self._cloud_vms is None:
+            from .cloud_vms.client import CloudVmsClient  # noqa: E402
+
+            self._cloud_vms = CloudVmsClient(client_wrapper=self._client_wrapper)
+        return self._cloud_vms
+
+    @property
+    def gpu_vms(self):
+        if self._gpu_vms is None:
+            from .gpu_vms.client import GpuVmsClient  # noqa: E402
+
+            self._gpu_vms = GpuVmsClient(client_wrapper=self._client_wrapper)
+        return self._gpu_vms
+
+
+def _make_default_async_client(
+    timeout: typing.Optional[float],
+    follow_redirects: typing.Optional[bool],
+) -> httpx.AsyncClient:
+    try:
+        import httpx_aiohttp  # type: ignore[import-not-found]
+    except ImportError:
+        pass
+    else:
+        if follow_redirects is not None:
+            return httpx_aiohttp.HttpxAiohttpClient(timeout=timeout, follow_redirects=follow_redirects)
+        return httpx_aiohttp.HttpxAiohttpClient(timeout=timeout)
+
+    if follow_redirects is not None:
+        return httpx.AsyncClient(timeout=timeout, follow_redirects=follow_redirects)
+    return httpx.AsyncClient(timeout=timeout)
+
+
+class AsyncIbee:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : typing.Optional[str]
+        The base url to use for requests from the client.
+
+    environment : IbeeEnvironment
+        The environment to use for requests from the client. from .environment import IbeeEnvironment
+
+
+
+        Defaults to IbeeEnvironment.DEFAULT
+
+
+
+    token : typing.Union[str, typing.Callable[[], str]]
+    headers : typing.Optional[typing.Dict[str, str]]
+        Additional headers to send with every request.
+
+    async_token : typing.Optional[typing.Callable[[], typing.Awaitable[str]]]
+        An async callable that returns a bearer token. Use this when token acquisition involves async I/O (e.g., refreshing tokens via an async HTTP client). When provided, this is used instead of the synchronous token for async requests.
+
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    max_retries : typing.Optional[int]
+        The default maximum number of retries for failed requests. Defaults to 2. Per-request `max_retries` in `request_options` takes precedence over this value.
+
+    stream_reconnection_enabled : typing.Optional[bool]
+        Whether to automatically reconnect on stream disconnection for resumable streaming endpoints. Defaults to True. Per-request `stream_reconnection_enabled` in `request_options` takes precedence over this value.
+
+    max_stream_reconnection_attempts : typing.Optional[int]
+        The maximum number of reconnection attempts for resumable streaming endpoints. Defaults to no limit. Per-request `max_stream_reconnection_attempts` in `request_options` takes precedence over this value.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.AsyncClient]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    logging : typing.Optional[typing.Union[LogConfig, Logger]]
+        Configure logging for the SDK. Accepts a LogConfig dict with 'level' (debug/info/warn/error), 'logger' (custom logger implementation), and 'silent' (boolean, defaults to True) fields. You can also pass a pre-configured Logger instance.
+
+    Examples
+    --------
+    from ibee import AsyncIbee
+
+    client = AsyncIbee(
+        token="YOUR_TOKEN",
+    )
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: typing.Optional[str] = None,
+        environment: IbeeEnvironment = IbeeEnvironment.DEFAULT,
+        token: typing.Union[str, typing.Callable[[], str]],
+        headers: typing.Optional[typing.Dict[str, str]] = None,
+        async_token: typing.Optional[typing.Callable[[], typing.Awaitable[str]]] = None,
+        timeout: typing.Optional[float] = None,
+        max_retries: typing.Optional[int] = None,
+        stream_reconnection_enabled: typing.Optional[bool] = None,
+        max_stream_reconnection_attempts: typing.Optional[int] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None,
+        logging: typing.Optional[typing.Union[LogConfig, Logger]] = None,
+    ):
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
+        _defaulted_max_retries = max_retries if max_retries is not None else 2
+        self._client_wrapper = AsyncClientWrapper(
+            base_url=_get_base_url(base_url=base_url, environment=environment),
+            token=token,
+            headers=headers,
+            async_token=async_token,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else _make_default_async_client(timeout=_defaulted_timeout, follow_redirects=follow_redirects),
+            timeout=_defaulted_timeout,
+            max_retries=_defaulted_max_retries,
+            stream_reconnection_enabled=stream_reconnection_enabled,
+            max_stream_reconnection_attempts=max_stream_reconnection_attempts,
+            logging=logging,
+        )
+        self._secret_store: typing.Optional[AsyncSecretStoreClient] = None
+        self._object_storage: typing.Optional[AsyncObjectStorageClient] = None
+        self._cloud_vms: typing.Optional[AsyncCloudVmsClient] = None
+        self._gpu_vms: typing.Optional[AsyncGpuVmsClient] = None
+
+    @property
+    def secret_store(self):
+        if self._secret_store is None:
+            from .secret_store.client import AsyncSecretStoreClient  # noqa: E402
+
+            self._secret_store = AsyncSecretStoreClient(client_wrapper=self._client_wrapper)
+        return self._secret_store
+
+    @property
+    def object_storage(self):
+        if self._object_storage is None:
+            from .object_storage.client import AsyncObjectStorageClient  # noqa: E402
+
+            self._object_storage = AsyncObjectStorageClient(client_wrapper=self._client_wrapper)
+        return self._object_storage
+
+    @property
+    def cloud_vms(self):
+        if self._cloud_vms is None:
+            from .cloud_vms.client import AsyncCloudVmsClient  # noqa: E402
+
+            self._cloud_vms = AsyncCloudVmsClient(client_wrapper=self._client_wrapper)
+        return self._cloud_vms
+
+    @property
+    def gpu_vms(self):
+        if self._gpu_vms is None:
+            from .gpu_vms.client import AsyncGpuVmsClient  # noqa: E402
+
+            self._gpu_vms = AsyncGpuVmsClient(client_wrapper=self._client_wrapper)
+        return self._gpu_vms
+
+
+def _get_base_url(*, base_url: typing.Optional[str] = None, environment: IbeeEnvironment) -> str:
+    if base_url is not None:
+        return base_url
+    elif environment is not None:
+        return environment.value
+    else:
+        raise Exception("Please pass in either base_url or environment to construct the client")

ibee/cloud_vms/__init__.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .types import CreateCloudVmRequestOsType
+_dynamic_imports: typing.Dict[str, str] = {"CreateCloudVmRequestOsType": ".types"}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = ["CreateCloudVmRequestOsType"]