ran black to fix formatting

Author: inference-stack-llc

src/electripy/ai/llm_gateway/config.py

@@ -25,8 +25,8 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:  # pragma: no cover - import-time only
-  from .domain import LlmRequest, LlmResponse
-  from .ports import PromptGuardPort, RedactorPort
+    from .domain import LlmRequest, LlmResponse
+    from .ports import PromptGuardPort, RedactorPort
 
 
 LlmCallHook = Callable[["LlmRequest", "LlmResponse", float], None]

src/electripy/ai/llm_gateway/providers.py

@@ -68,8 +68,12 @@ def _normalise_provider_name(provider: str) -> str:
 def register_llm_provider(
     name: str,
     *,
-    sync_factory: Callable[[LlmGatewaySettings, dict[str, Any]], LlmGatewaySyncClient] | None = None,
-    async_factory: Callable[[LlmGatewaySettings, dict[str, Any]], LlmGatewayAsyncClient] | None = None,
+    sync_factory: (
+        Callable[[LlmGatewaySettings, dict[str, Any]], LlmGatewaySyncClient] | None
+    ) = None,
+    async_factory: (
+        Callable[[LlmGatewaySettings, dict[str, Any]], LlmGatewayAsyncClient] | None
+    ) = None,
 ) -> None:
     """Register custom LLM provider factories.
 

tests/concurrency/test_rate_limiter_properties.py

@@ -14,7 +14,9 @@
     rate=st.floats(min_value=0.1, max_value=100.0),
     capacity_multiplier=st.floats(min_value=0.5, max_value=5.0),
 )
-async def test_available_tokens_never_exceeds_capacity(rate: float, capacity_multiplier: float) -> None:
+async def test_available_tokens_never_exceeds_capacity(
+    rate: float, capacity_multiplier: float
+) -> None:
     capacity = max(rate * capacity_multiplier, rate)
     limiter = AsyncTokenBucketRateLimiter(rate=rate, capacity=capacity)