cartesia 1.0.2__tar.gz → 1.0.3__tar.gz

{cartesia-1.0.2 → cartesia-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cartesia
-Version: 1.0.2
+Version: 1.0.3
 Summary: The official Python library for the Cartesia API.
 Home-page: 
 Author: Cartesia, Inc.

cartesia-1.0.3/cartesia/utils/deprecated.py

@@ -0,0 +1,53 @@
+import os
+import warnings
+from typing import Any, Callable, TypeVar
+
+TCallable = TypeVar("TCallable", bound=Callable[..., Any])
+
+# List of statistics of deprecated functions.
+# This should only be used by the test suite to find any deprecated functions
+# that should be removed for this version.
+_TRACK_DEPRECATED_FUNCTION_STATS = os.environ.get("CARTESIA_TEST_DEPRECATED", "").lower() == "true"
+_DEPRECATED_FUNCTION_STATS = []
+
+
+def deprecated(
+    reason=None, vdeprecated=None, vremove=None, replacement=None
+) -> Callable[[TCallable], TCallable]:
+    local_vars = locals()
+
+    def fn(func: TCallable) -> TCallable:
+        msg = _get_deprecated_msg(func, reason, vdeprecated, vremove, replacement)
+        warnings.warn(msg, DeprecationWarning)
+        return func
+
+    if _TRACK_DEPRECATED_FUNCTION_STATS:  # pragma: no cover
+        _DEPRECATED_FUNCTION_STATS.append(local_vars)
+
+    return fn
+
+
+def _get_deprecated_msg(wrapped, reason, vdeprecated, vremoved, replacement=None):
+    fmt = "{name} is deprecated"
+    if vdeprecated:
+        fmt += " since v{vdeprecated}"
+    if vremoved:
+        fmt += " and will be removed in v{vremoved}"
+    fmt += "."
+
+    if reason:
+        fmt += " ({reason})"
+    if replacement:
+        fmt += " -- Use {replacement} instead."
+
+    return fmt.format(
+        name=wrapped.__name__,
+        reason=reason or "",
+        vdeprecated=vdeprecated or "",
+        vremoved=vremoved or "",
+        replacement=replacement or "",
+    )
+
+
+# This method is taken from the following source:
+# https://github.com/ad12/meddlr/blob/main/meddlr/utils/deprecated.py

cartesia-1.0.3/cartesia/utils/retry.py

@@ -0,0 +1,87 @@
+import time
+
+from aiohttp.client_exceptions import ServerDisconnectedError
+import asyncio
+from functools import wraps
+from http.client import RemoteDisconnected
+from httpx import TimeoutException
+from requests.exceptions import ConnectionError
+
+
+def retry_on_connection_error(max_retries=3, backoff_factor=1, logger=None):
+    """Retry a function if a ConnectionError, RemoteDisconnected, ServerDisconnectedError, or TimeoutException occurs.
+
+    Args:
+        max_retries (int): The maximum number of retries.
+        backoff_factor (int): The factor to increase the delay between retries.
+        logger (logging.Logger): The logger to use for logging.
+    """
+
+    def decorator(func):
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            retry_count = 0
+            while retry_count < max_retries:
+                try:
+                    return func(*args, **kwargs)
+                except (
+                    ConnectionError,
+                    RemoteDisconnected,
+                    ServerDisconnectedError,
+                    TimeoutException,
+                ) as e:
+                    logger.info(f"Retrying after exception: {e}")
+                    retry_count += 1
+                    if retry_count < max_retries:
+                        delay = backoff_factor * (2 ** (retry_count - 1))
+                        logger.warn(
+                            f"Attempt {retry_count + 1}/{max_retries} in {delay} seconds..."
+                        )
+                        time.sleep(delay)
+                    else:
+                        raise Exception(f"Exception occurred after {max_retries} tries.") from e
+
+        return wrapper
+
+    return decorator
+
+
+def retry_on_connection_error_async(max_retries=3, backoff_factor=1, logger=None):
+    """Retry an asynchronous function if a ConnectionError, RemoteDisconnected, ServerDisconnectedError, or TimeoutException occurs.
+
+    Args:
+        max_retries (int): The maximum number of retries.
+        backoff_factor (int): The factor to increase the delay between retries.
+        logger (logging.Logger): The logger to use for logging.
+    """
+
+    def decorator(func):
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            retry_count = 0
+            while retry_count < max_retries:
+                try:
+                    async for chunk in func(*args, **kwargs):
+                        yield chunk
+                    # If the function completes without raising an exception return
+                    return
+                except (
+                    ConnectionError,
+                    RemoteDisconnected,
+                    ServerDisconnectedError,
+                    TimeoutException,
+                ) as e:
+                    logger.info(f"Retrying after exception: {e}")
+                    retry_count += 1
+                    if retry_count < max_retries:
+                        delay = backoff_factor * (2 ** (retry_count - 1))
+                        logger.warn(
+                            f"Attempt {retry_count + 1}/{max_retries} in {delay} seconds..."
+                        )
+                        await asyncio.sleep(delay)
+                    else:
+                        raise Exception(f"Exception occurred after {max_retries} tries.") from e
+
+        return wrapper
+
+    return decorator

cartesia-1.0.3/cartesia/version.py

@@ -0,0 +1 @@
+__version__ = "1.0.3"

{cartesia-1.0.2 → cartesia-1.0.3}/cartesia.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cartesia
-Version: 1.0.2
+Version: 1.0.3
 Summary: The official Python library for the Cartesia API.
 Home-page: 
 Author: Cartesia, Inc.

{cartesia-1.0.2 → cartesia-1.0.3}/cartesia.egg-info/SOURCES.txt

@@ -10,5 +10,8 @@ cartesia.egg-info/SOURCES.txt
 cartesia.egg-info/dependency_links.txt
 cartesia.egg-info/requires.txt
 cartesia.egg-info/top_level.txt
+cartesia/utils/__init__.py
+cartesia/utils/deprecated.py
+cartesia/utils/retry.py
 tests/test_deprecated.py
 tests/test_tts.py

{cartesia-1.0.2 → cartesia-1.0.3}/setup.py

@@ -268,7 +268,15 @@ setup(
     author_email=EMAIL,
     python_requires=REQUIRES_PYTHON,
     url=URL,
-    packages=[PACKAGE_DIR],
+    packages=find_packages(
+        exclude=[
+            "scratch",
+            "tests",
+            "*.tests",
+            "*.tests.*",
+            "tests.*",
+        ]
+    ),
     install_requires=REQUIRED,
     extras_require=EXTRAS,
     include_package_data=True,

cartesia-1.0.2/cartesia/version.py

@@ -1 +0,0 @@
-__version__ = "1.0.2"