sdkrouter 0.1.1__py3-none-any.whl

sdkrouter/_api/generated/models/logger.py

@@ -0,0 +1,256 @@
+# Auto-generated by DjangoCFG - see CLAUDE.md
+"""
+API Logger with Rich
+Beautiful console logging for API requests and responses
+
+Installation:
+    pip install rich
+"""
+
+from __future__ import annotations
+
+import time
+from dataclasses import dataclass, field
+from typing import Any, Dict, Optional
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from rich.text import Text
+
+
+@dataclass
+class RequestLog:
+    """Request log data."""
+
+    method: str
+    url: str
+    headers: Optional[Dict[str, str]] = None
+    body: Optional[Any] = None
+    timestamp: float = field(default_factory=time.time)
+
+
+@dataclass
+class ResponseLog:
+    """Response log data."""
+
+    status: int
+    status_text: str
+    data: Optional[Any] = None
+    duration: float = 0.0
+    timestamp: float = field(default_factory=time.time)
+
+
+@dataclass
+class ErrorLog:
+    """Error log data."""
+
+    message: str
+    status_code: Optional[int] = None
+    field_errors: Optional[Dict[str, list[str]]] = None
+    duration: float = 0.0
+    timestamp: float = field(default_factory=time.time)
+
+
+@dataclass
+class LoggerConfig:
+    """Logger configuration."""
+
+    enabled: bool = True
+    log_requests: bool = True
+    log_responses: bool = True
+    log_errors: bool = True
+    log_bodies: bool = True
+    log_headers: bool = False
+    console: Optional[Console] = None
+
+
+# Sensitive header names to filter out
+SENSITIVE_HEADERS = [
+    "authorization",
+    "cookie",
+    "set-cookie",
+    "x-api-key",
+    "x-csrf-token",
+]
+
+
+class APILogger:
+    """API Logger class."""
+
+    def __init__(self, config: Optional[LoggerConfig] = None):
+        """Initialize logger."""
+        self.config = config or LoggerConfig()
+        self.console = self.config.console or Console()
+
+    def enable(self) -> None:
+        """Enable logging."""
+        self.config.enabled = True
+
+    def disable(self) -> None:
+        """Disable logging."""
+        self.config.enabled = False
+
+    def set_config(self, **kwargs: Any) -> None:
+        """Update configuration."""
+        for key, value in kwargs.items():
+            if hasattr(self.config, key):
+                setattr(self.config, key, value)
+
+    def _filter_headers(self, headers: Optional[Dict[str, str]]) -> Dict[str, str]:
+        """Filter sensitive headers."""
+        if not headers:
+            return {}
+
+        filtered = {}
+        for key, value in headers.items():
+            if key.lower() in SENSITIVE_HEADERS:
+                filtered[key] = "***"
+            else:
+                filtered[key] = value
+
+        return filtered
+
+    def log_request(self, request: RequestLog) -> None:
+        """Log request."""
+        if not self.config.enabled or not self.config.log_requests:
+            return
+
+        # Create request info
+        text = Text()
+        text.append("→ ", style="bold blue")
+        text.append(request.method, style="bold yellow")
+        text.append(" ", style="")
+        text.append(request.url, style="cyan")
+
+        self.console.print(text)
+
+        if self.config.log_headers and request.headers:
+            headers = self._filter_headers(request.headers)
+            self.console.print("  Headers:", style="dim")
+            for key, value in headers.items():
+                self.console.print(f"    {key}: {value}", style="dim")
+
+        if self.config.log_bodies and request.body:
+            self.console.print("  Body:", style="dim")
+            self.console.print(request.body, style="dim")
+
+    def log_response(self, request: RequestLog, response: ResponseLog) -> None:
+        """Log response."""
+        if not self.config.enabled or not self.config.log_responses:
+            return
+
+        # Determine color based on status
+        if response.status >= 500:
+            status_style = "bold red"
+        elif response.status >= 400:
+            status_style = "bold yellow"
+        elif response.status >= 300:
+            status_style = "bold cyan"
+        else:
+            status_style = "bold green"
+
+        # Create response info
+        text = Text()
+        text.append("← ", style="bold green")
+        text.append(request.method, style="bold yellow")
+        text.append(" ", style="")
+        text.append(request.url, style="cyan")
+        text.append(" ", style="")
+        text.append(str(response.status), style=status_style)
+        text.append(" ", style="")
+        text.append(response.status_text, style=status_style)
+        text.append(f" ({response.duration:.0f}ms)", style="dim")
+
+        self.console.print(text)
+
+        if self.config.log_bodies and response.data:
+            self.console.print("  Response:", style="dim")
+            self.console.print(response.data, style="dim")
+
+    def log_error(self, request: RequestLog, error: ErrorLog) -> None:
+        """Log error."""
+        if not self.config.enabled or not self.config.log_errors:
+            return
+
+        # Create error header
+        text = Text()
+        text.append("✗ ", style="bold red")
+        text.append(request.method, style="bold yellow")
+        text.append(" ", style="")
+        text.append(request.url, style="cyan")
+        text.append(" ", style="")
+        text.append(
+            str(error.status_code) if error.status_code else "Network",
+            style="bold red",
+        )
+        text.append(" Error", style="bold red")
+        text.append(f" ({error.duration:.0f}ms)", style="dim")
+
+        self.console.print(text)
+        self.console.print(f"  Message: {error.message}", style="red")
+
+        if error.field_errors:
+            self.console.print("  Field Errors:", style="red")
+            for field, errors in error.field_errors.items():
+                for err in errors:
+                    self.console.print(f"    • {field}: {err}", style="red dim")
+
+    def info(self, message: str, **kwargs: Any) -> None:
+        """Log info message."""
+        if not self.config.enabled:
+            return
+        self.console.print(f"ℹ {message}", style="blue", **kwargs)
+
+    def warn(self, message: str, **kwargs: Any) -> None:
+        """Log warning message."""
+        if not self.config.enabled:
+            return
+        self.console.print(f"⚠ {message}", style="yellow", **kwargs)
+
+    def error(self, message: str, **kwargs: Any) -> None:
+        """Log error message."""
+        if not self.config.enabled:
+            return
+        self.console.print(f"✗ {message}", style="red", **kwargs)
+
+    def success(self, message: str, **kwargs: Any) -> None:
+        """Log success message."""
+        if not self.config.enabled:
+            return
+        self.console.print(f"✓ {message}", style="green", **kwargs)
+
+    def debug(self, message: str, **kwargs: Any) -> None:
+        """Log debug message."""
+        if not self.config.enabled:
+            return
+        self.console.print(f"🔍 {message}", style="dim", **kwargs)
+
+    def panel(self, content: Any, title: str, style: str = "blue") -> None:
+        """Log content in a panel."""
+        if not self.config.enabled:
+            return
+        self.console.print(Panel(content, title=title, border_style=style))
+
+    def table(
+        self,
+        headers: list[str],
+        rows: list[list[Any]],
+        title: Optional[str] = None,
+    ) -> None:
+        """Log data in a table."""
+        if not self.config.enabled:
+            return
+
+        table = Table(title=title)
+        for header in headers:
+            table.add_column(header, style="cyan")
+
+        for row in rows:
+            table.add_row(*[str(cell) for cell in row])
+
+        self.console.print(table)
+
+
+# Default logger instance
+default_logger = APILogger()

sdkrouter/_api/generated/models/models__api__llm_models/__init__.py

@@ -0,0 +1,7 @@
+# Auto-generated by DjangoCFG - see CLAUDE.md
+from .client import ModelsLlmModelsAPI
+from .models import *
+
+__all__ = [
+    "ModelsLlmModelsAPI",
+]

sdkrouter/_api/generated/models/models__api__llm_models/client.py

@@ -0,0 +1,99 @@
+from __future__ import annotations
+
+import httpx
+
+from .models import *
+
+
+class ModelsLlmModelsAPI:
+    """API endpoints for Llm Models."""
+
+    def __init__(self, client: httpx.AsyncClient):
+        """Initialize sub-client with shared httpx client."""
+        self._client = client
+
+    async def list(self, page: int | None = None, page_size: int | None = None) -> list[PaginatedLLMModelListList]:
+        """
+        List models
+
+        Get list of available LLM models with filtering.
+        """
+        url = "/api/llm_models/"
+        response = await self._client.get(url, params={"page": page if page is not None else None, "page_size": page_size if page_size is not None else None})
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return PaginatedLLMModelListList.model_validate(response.json())
+
+
+    async def retrieve(self, model_id: str) -> LLMModelDetail:
+        """
+        Get model details
+
+        Get detailed information about a specific model.
+        """
+        url = f"/api/llm_models/{model_id}/"
+        response = await self._client.get(url)
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return LLMModelDetail.model_validate(response.json())
+
+
+    async def calculate_cost_create(self, model_id: str, data: CostCalculationRequestRequest) -> CostCalculationResponse:
+        """
+        Calculate cost
+
+        Calculate cost for a request.
+        """
+        url = f"/api/llm_models/{model_id}/calculate-cost/"
+        response = await self._client.post(url, json=data.model_dump(exclude_unset=True))
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return CostCalculationResponse.model_validate(response.json())
+
+
+    async def providers_retrieve(self) -> ProvidersResponse:
+        """
+        List providers
+
+        Get list of available providers.
+        """
+        url = "/api/llm_models/providers/"
+        response = await self._client.get(url)
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return ProvidersResponse.model_validate(response.json())
+
+
+    async def stats_retrieve(self) -> StatsResponse:
+        """
+        Get statistics
+
+        Get model statistics.
+        """
+        url = "/api/llm_models/stats/"
+        response = await self._client.get(url)
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return StatsResponse.model_validate(response.json())
+
+

sdkrouter/_api/generated/models/models__api__llm_models/models.py

@@ -0,0 +1,206 @@
+# Auto-generated by DjangoCFG - see CLAUDE.md
+from __future__ import annotations
+
+from datetime import datetime
+from typing import Any
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class LLMModelList(BaseModel):
+    """
+    Lightweight serializer for model listing.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    model_id: str = Field(description="Model identifier (e.g., 'openai/gpt-4o')", max_length=100)
+    name: str = Field(description='Human-readable model name', max_length=200)
+    owned_by: str = Field(description="Provider/owner name (e.g., 'OpenAI')", max_length=100)
+    context_length: int = Field(description='Maximum context length in tokens', ge=-2147483648, le=2147483647)
+    supports_vision: bool | None = Field(None, description='Whether model supports image inputs')
+    supports_tools: bool | None = Field(None, description='Whether model supports tool/function calling')
+    pricing: dict[str, Any] = Field(description='Get simplified pricing info.')
+
+
+
+class PaginatedLLMModelListList(BaseModel):
+    """
+    
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    count: int = Field(description='Total number of items across all pages')
+    page: int = Field(description='Current page number (1-based)')
+    pages: int = Field(description='Total number of pages')
+    page_size: int = Field(description='Number of items per page')
+    has_next: bool = Field(description='Whether there is a next page')
+    has_previous: bool = Field(description='Whether there is a previous page')
+    next_page: int | None = Field(None, description='Next page number (null if no next page)')
+    previous_page: int | None = Field(None, description='Previous page number (null if no previous page)')
+    results: list[LLMModelList] = Field(description='Array of items for current page')
+
+
+
+class LLMModelPricing(BaseModel):
+    """
+    Pricing information serializer.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    prompt: str = Field(description='Price per million prompt tokens (USD)', pattern='^-?\\d{0,4}(?:\\.\\d{0,6})?$')
+    completion: str = Field(description='Price per million completion tokens (USD)', pattern='^-?\\d{0,4}(?:\\.\\d{0,6})?$')
+    image: str | None = Field(None, description='Price per image (USD)', pattern='^-?\\d{0,4}(?:\\.\\d{0,6})?$')
+
+
+
+class LLMModelDetail(BaseModel):
+    """
+    Detailed serializer for single model.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    id: Any = ...
+    model_id: str = Field(description="Model identifier (e.g., 'openai/gpt-4o')", max_length=100)
+    name: str = Field(description='Human-readable model name', max_length=200)
+    owned_by: str = Field(description="Provider/owner name (e.g., 'OpenAI')", max_length=100)
+    provider: Any = ...
+    description: str | None = Field(None, description='Model description')
+    context_length: int = Field(description='Maximum context length in tokens', ge=-2147483648, le=2147483647)
+    max_output_tokens: int | None = Field(None, description='Maximum output tokens (if limited)', ge=-2147483648, le=2147483647)
+    supports_vision: bool | None = Field(None, description='Whether model supports image inputs')
+    supports_tools: bool | None = Field(None, description='Whether model supports tool/function calling')
+    supports_streaming: bool | None = Field(None, description='Whether model supports streaming responses')
+    supports_json_mode: bool | None = Field(None, description='Whether model supports JSON output mode')
+    pricing: LLMModelPricing = ...
+    architecture: dict[str, Any] | None = Field(None, description='Architecture details (modality, tokenizer, etc.)')
+    top_provider: dict[str, Any] | None = Field(None, description='Top provider information from OpenRouter')
+    is_active: bool | None = Field(None, description='Whether the model is available for use')
+    last_synced_at: Any = Field(description='Last time this model was synced from OpenRouter')
+    created_at: Any = ...
+    updated_at: Any = ...
+
+
+
+class CostCalculationRequestRequest(BaseModel):
+    """
+    Request serializer for cost calculation.
+
+    Request model (no read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    input_tokens: int = Field(description='Number of input tokens', ge=0)
+    output_tokens: int = Field(description='Number of output tokens', ge=0)
+
+
+
+class CostCalculationResponse(BaseModel):
+    """
+    Response serializer for cost calculation.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    model_id: str = Field(description='Model ID')
+    input_cost_usd: float = Field(description='Cost of input tokens in USD')
+    output_cost_usd: float = Field(description='Cost of output tokens in USD')
+    total_cost_usd: float = Field(description='Total cost in USD')
+    input_tokens: int = Field(description='Number of input tokens')
+    output_tokens: int = Field(description='Number of output tokens')
+
+
+
+class ProviderInfo(BaseModel):
+    """
+    Provider info serializer.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    name: str = Field(description='Provider name')
+    model_count: int = Field(description='Number of models')
+
+
+
+class ProvidersResponse(BaseModel):
+    """
+    Response serializer for providers list.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    providers: list[ProviderInfo] = Field(description='List of providers')
+
+
+
+class StatsResponse(BaseModel):
+    """
+    Response serializer for model statistics.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    total_models: int = Field(description='Total number of models')
+    active_models: int = Field(description='Number of active models')
+    vision_models: int = Field(description='Number of vision-capable models')
+    tool_models: int = Field(description='Number of tool-capable models')
+    provider_count: int = Field(description='Number of providers')
+
+
+

sdkrouter/_api/generated/models/models__api__llm_models/sync_client.py

@@ -0,0 +1,99 @@
+from __future__ import annotations
+
+import httpx
+
+from .models import *
+
+
+class SyncModelsLlmModelsAPI:
+    """Synchronous API endpoints for Llm Models."""
+
+    def __init__(self, client: httpx.Client):
+        """Initialize sync sub-client with shared httpx client."""
+        self._client = client
+
+    def list(self, page: int | None = None, page_size: int | None = None) -> list[PaginatedLLMModelListList]:
+        """
+        List models
+
+        Get list of available LLM models with filtering.
+        """
+        url = "/api/llm_models/"
+        response = self._client.get(url, params={"page": page if page is not None else None, "page_size": page_size if page_size is not None else None})
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return PaginatedLLMModelListList.model_validate(response.json())
+
+
+    def retrieve(self, model_id: str) -> LLMModelDetail:
+        """
+        Get model details
+
+        Get detailed information about a specific model.
+        """
+        url = f"/api/llm_models/{model_id}/"
+        response = self._client.get(url)
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return LLMModelDetail.model_validate(response.json())
+
+
+    def calculate_cost_create(self, model_id: str, data: CostCalculationRequestRequest) -> CostCalculationResponse:
+        """
+        Calculate cost
+
+        Calculate cost for a request.
+        """
+        url = f"/api/llm_models/{model_id}/calculate-cost/"
+        response = self._client.post(url, json=data.model_dump(exclude_unset=True))
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return CostCalculationResponse.model_validate(response.json())
+
+
+    def providers_retrieve(self) -> ProvidersResponse:
+        """
+        List providers
+
+        Get list of available providers.
+        """
+        url = "/api/llm_models/providers/"
+        response = self._client.get(url)
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return ProvidersResponse.model_validate(response.json())
+
+
+    def stats_retrieve(self) -> StatsResponse:
+        """
+        Get statistics
+
+        Get model statistics.
+        """
+        url = "/api/llm_models/stats/"
+        response = self._client.get(url)
+        if not response.is_success:
+            try:
+                error_body = response.json()
+            except Exception:
+                error_body = response.text
+            raise httpx.HTTPStatusError(f"{response.status_code}: {error_body}", request=response.request, response=response)
+        return StatsResponse.model_validate(response.json())
+
+