.

Author: arthurbrenno

agentle/responses/pricing/default_pricing_service.py

@@ -1,7 +1,8 @@
 """Default pricing service with static pricing data."""
 
-from typing import Any
-from rsb.models.base_model import BaseModel
+from typing import Any, Literal
+
+from rsb.models import BaseModel, Field
 
 from agentle.responses.pricing.modality import Modality
 
@@ -242,24 +243,16 @@ class DefaultPricingService(BaseModel):
     Prices are approximate and based on publicly available pricing information.
     """
 
-    type: str = "default"
-
-    def __init__(
-        self,
-        custom_pricing: dict[str, dict[str, dict[str, float]]] | None = None,
-    ):
-        """
-        Initialize the default pricing service.
+    type: Literal["default"] = Field(default="default")
+    custom_pricing: dict[str, dict[str, dict[str, float]]] | None = Field(default=None)
+    pricing: dict[str, dict[str, dict[str, Any]]] = Field(default_factory=dict)
 
-        Args:
-            custom_pricing: Optional dictionary to override or extend default pricing.
-                           Format: {"model": {"modality": {"input": float, "cached_input": float, "output": float}}}
-        """
-        super().__init__()
+    def model_post_init(self, context: Any, /) -> None:
+        """Initialize pricing after model creation."""
         self.pricing = MODEL_PRICING.copy()
-        if custom_pricing:
+        if self.custom_pricing:
             # Deep merge custom pricing
-            for model, modalities in custom_pricing.items():
+            for model, modalities in self.custom_pricing.items():
                 if model in self.pricing:
                     self.pricing[model].update(modalities)
                 else:

agentle/responses/pricing/openrouter_pricing_service.py

@@ -1,10 +1,11 @@
 """OpenRouter pricing service with dynamic pricing from API."""
 
 import logging
-from typing import Any
+from typing import Any, Literal
 
 import httpx
-from rsb.models.base_model import BaseModel
+from pydantic import PrivateAttr
+from rsb.models import BaseModel, Field
 
 from agentle.responses.pricing.modality import Modality
 
@@ -28,31 +29,25 @@ class OpenRouterPricingService(BaseModel):
         _models_cache: Internal cache of model pricing data
     """
 
-    type: str = "openrouter"
+    type: Literal["openrouter"] = Field(default="openrouter")
     api_key: str | None = None
     base_url: str = "https://openrouter.ai/api/v1"
-    http_client: httpx.AsyncClient | None = None
+    _http_client: httpx.AsyncClient | None = PrivateAttr(default=None)
     _models_cache: dict[str, dict[str, Any]] | None = None
 
-    def __init__(
-        self,
-        api_key: str | None = None,
-        base_url: str = "https://openrouter.ai/api/v1",
-        http_client: httpx.AsyncClient | None = None,
-    ):
-        """
-        Initialize the OpenRouter pricing service.
+    @property
+    def http_client(self) -> httpx.AsyncClient:
+        if self._http_client is None:
+            raise ValueError("Client is None.")
 
-        Args:
-            api_key: OpenRouter API key. If not provided, reads from OPENROUTER_API_KEY env var.
-            base_url: Base URL for OpenRouter API
-            http_client: Optional custom HTTP client for requests
-        """
-        super().__init__()
-        self.api_key = api_key
-        self.base_url = base_url
-        self.http_client = http_client
-        self._models_cache = None
+        return self._http_client
+
+    def model_post_init(self, context: Any, /) -> None:
+        super().model_post_init(context)
+        self._http_client = httpx.AsyncClient()
+
+    def change_http_client(self, client: httpx.AsyncClient) -> None:
+        self._http_client = client
 
     async def _fetch_models(self) -> dict[str, dict[str, Any]]:
         """
@@ -109,8 +104,7 @@ async def _fetch_models(self) -> dict[str, dict[str, Any]]:
             self._models_cache = {}
             return self._models_cache
         finally:
-            if self.http_client is None:
-                await client.aclose()
+            await client.aclose()
 
     async def get_input_price_per_million(
         self,

examples/openrouter_responses_example.py

@@ -19,6 +19,7 @@ def add(a: int, b: int) -> int:
 
 async def main():
     """Basic text generation example."""
+
     responder = Responder.openai()
 
     print("Starting...")