keycycle 0.2.0__tar.gz → 0.2.2__tar.gz

{keycycle-0.2.0 → keycycle-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keycycle
-Version: 0.2.0
+Version: 0.2.2
 Summary: A thread-safe key rotation and rate limiting manager for API keys.
 Author-email: William Jin <jinwilliam.jin@gmail.com>
 Classifier: Programming Language :: Python :: 3

{keycycle-0.2.0 → keycycle-0.2.2}/keycycle/__init__.py

@@ -19,7 +19,6 @@ from .adapters.generic_adapter import (
     SyncGenericRotatingClient,
     AsyncGenericRotatingClient,
 )
-from .adapters.openai_adapter import RotatingAsyncOpenAIClient, RotatingOpenAIClient
 
 __all__ = [
     # New primary wrapper (multi-provider support)
@@ -32,8 +31,6 @@ __all__ = [
     "KeyLimitOverride",
     "KeyEntry",
     "RotatingKeyManager",
-    "RotatingAsyncOpenAIClient",
-    "RotatingOpenAIClient",
     # Generic rotating client
     "create_rotating_client",
     "detect_async_client",

{keycycle-0.2.0 → keycycle-0.2.2}/keycycle/adapters/__init__.py

@@ -1,4 +1,3 @@
-from .openai_adapter import RotatingOpenAIClient, RotatingAsyncOpenAIClient
 from .generic_adapter import (
     create_rotating_client,
     detect_async_client,
@@ -11,9 +10,6 @@ from .generic_adapter import (
 )
 
 __all__ = [
-    # OpenAI-specific
-    "RotatingOpenAIClient",
-    "RotatingAsyncOpenAIClient",
     # Generic adapter
     "create_rotating_client",
     "detect_async_client",

{keycycle-0.2.0 → keycycle-0.2.2}/keycycle/key_rotation/rotating_mixin.py

@@ -109,8 +109,8 @@ class RotatingCredentialsMixin:
             return
 
         actual_tokens = 0
-        
-        if response:
+
+        if response and response.response_usage:
             actual_tokens = response.response_usage.total_tokens
 
         # Fallback: If no usage found (or streaming), use the estimate

{keycycle-0.2.0 → keycycle-0.2.2}/keycycle/legacy_multi_provider_wrapper.py

@@ -31,7 +31,7 @@ from .core.exceptions import NoAvailableKeyError, KeyNotFoundError
 from .core.backoff import ExponentialBackoff, BackoffConfig
 from .usage.db_logic import UsageDatabase
 from .config.log_config import default_logger
-from .adapters import RotatingOpenAIClient, RotatingAsyncOpenAIClient
+from .adapters.openai_adapter import RotatingOpenAIClient, RotatingAsyncOpenAIClient
 from .adapters.generic_adapter import (
     create_rotating_client,
     SyncGenericRotatingClient,

{keycycle-0.2.0 → keycycle-0.2.2}/keycycle/multi_client_wrapper.py

@@ -39,6 +39,7 @@ class ProviderConfig:
     """Internal config for a registered provider."""
     default_model: Optional[str] = None
     limits: Optional[Dict[str, RateLimits]] = None
+    excluded_kwargs: Optional[List[str]] = None
 
 
 @dataclass
@@ -252,6 +253,10 @@ class MultiClientWrapper:
         if api_key_param is None:
             api_key_param = manager.api_key_param
 
+        # Use config's excluded_kwargs as default if not specified
+        if excluded_kwargs is None:
+            excluded_kwargs = config.excluded_kwargs
+
         return create_rotating_client(
             client_class=client_class,
             manager=manager,
@@ -330,6 +335,9 @@ class MultiClientWrapper:
                 limits=config.limits,
                 api_key_param=config.api_key_param,
             )
+            # Store excluded_kwargs from env config
+            if config.excluded_kwargs:
+                instance._configs[provider.lower()].excluded_kwargs = config.excluded_kwargs
         return instance
 
     @staticmethod

{keycycle-0.2.0 → keycycle-0.2.2}/keycycle.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keycycle
-Version: 0.2.0
+Version: 0.2.2
 Summary: A thread-safe key rotation and rate limiting manager for API keys.
 Author-email: William Jin <jinwilliam.jin@gmail.com>
 Classifier: Programming Language :: Python :: 3

{keycycle-0.2.0 → keycycle-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "keycycle"
-version = "0.2.0"
+version = "0.2.2"
 authors = [
   { name="William Jin", email="jinwilliam.jin@gmail.com" },
 ]