sdkrouter 0.1.1__py3-none-any.whl

sdkrouter/_api/generated/vision/sync_client.py

@@ -0,0 +1,58 @@
+from __future__ import annotations
+
+from typing import Any, Optional
+
+import httpx
+
+from .vision__api__vision.sync_client import SyncVisionVisionAPI
+from .logger import APILogger, LoggerConfig
+from .retry import RetryConfig, RetryAsyncClient
+
+
+class SyncAPIClient:
+    """
+    Synchronous API client for SDKRouter API.
+
+    Usage:
+        >>> with SyncAPIClient(base_url='https://api.example.com') as client:
+        ...     users = client.users.list()
+        ...     post = client.posts.create(data=new_post)
+    """
+
+    def __init__(
+        self,
+        base_url: str,
+        logger_config: Optional[LoggerConfig] = None,
+        **kwargs: Any,
+    ):
+        """
+        Initialize sync API client.
+
+        Args:
+            base_url: Base API URL (e.g., 'https://api.example.com')
+            logger_config: Logger configuration (None to disable logging)
+            **kwargs: Additional httpx.Client kwargs
+        """
+        self.base_url = base_url.rstrip('/')
+        self._client = httpx.Client(
+            base_url=self.base_url,
+            **kwargs,
+        )
+
+        # Initialize logger
+        self.logger: Optional[APILogger] = None
+        if logger_config is not None:
+            self.logger = APILogger(logger_config)
+
+        # Initialize sub-clients
+        self.vision_vision = SyncVisionVisionAPI(self._client)
+
+    def __enter__(self) -> 'SyncAPIClient':
+        return self
+
+    def __exit__(self, *args: Any) -> None:
+        self._client.close()
+
+    def close(self) -> None:
+        """Close HTTP client."""
+        self._client.close()

sdkrouter/_api/generated/vision/vision__api__vision/__init__.py

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

sdkrouter/_api/generated/vision/vision__api__vision/client.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+import httpx
+
+from .models import *
+
+
+class VisionVisionAPI:
+    """API endpoints for Vision."""
+
+    def __init__(self, client: httpx.AsyncClient):
+        """Initialize sub-client with shared httpx client."""
+        self._client = client
+
+    async def analyze_create(self, data: VisionAnalyzeRequestRequest) -> VisionAnalyzeResponse:
+        """
+        Analyze image
+
+        Analyze an image with vision model.
+        """
+        url = "/api/vision/analyze/"
+        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 VisionAnalyzeResponse.model_validate(response.json())
+
+
+    async def models_retrieve(self) -> VisionModelsResponse:
+        """
+        Get vision models
+
+        Get supported vision models.
+        """
+        url = "/api/vision/models/"
+        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 VisionModelsResponse.model_validate(response.json())
+
+
+    async def ocr_create(self, data: OCRRequestRequest) -> OCRResponse:
+        """
+        OCR extraction
+
+        Extract text from image using OCR.
+        """
+        url = "/api/vision/ocr/"
+        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 OCRResponse.model_validate(response.json())
+
+

sdkrouter/_api/generated/vision/vision__api__vision/models.py

@@ -0,0 +1,138 @@
+# 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
+
+from ..enums import OCRRequestRequestMode, VisionAnalyzeRequestRequestModelQuality
+
+
+class VisionAnalyzeRequestRequest(BaseModel):
+    """
+    Request serializer for vision analysis.
+
+    Request model (no read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    image: str | None = Field(None, description='Base64 encoded image data')
+    image_url: str | None = Field(None, description='URL of the image to analyze')
+    prompt: str | None = Field(None, description='Custom prompt for analysis (optional)', max_length=2000)
+    model: str | None = Field(None, description="Specific model to use (e.g., 'openai/gpt-4o')", max_length=100)
+    model_quality: VisionAnalyzeRequestRequestModelQuality | None = Field(None, description='Model quality tier (fast, balanced, best)\n\n* `fast` - Fast - lower quality, cheaper\n* `balanced` - Balanced - good quality, reasonable cost\n* `best` - Best - highest quality, most expensive')
+    fetch_image: bool | None = Field(None, description='Whether to fetch image from URL (default: True)')
+    max_tokens: int | None = Field(None, description='Maximum tokens in response', ge=1, le=4096)
+
+
+
+class Usage(BaseModel):
+    """
+    Token usage statistics.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    prompt_tokens: int = Field(description='Tokens in prompt')
+    completion_tokens: int = Field(description='Tokens in completion')
+    total_tokens: int = Field(description='Total tokens used')
+
+
+
+class VisionAnalyzeResponse(BaseModel):
+    """
+    Response serializer for vision analysis.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    extracted_text: str = Field(description='Text extracted from the image')
+    description: str = Field(description='Description/analysis of the image')
+    language: str | None = Field(None, description='Detected language of text in image')
+    model: str = Field(description='Model used for analysis')
+    usage: Usage | None = None
+    cost_usd: float = Field(description='Cost of this request in USD')
+    cached: bool | None = Field(None, description='Whether result was from cache')
+    request_uuid: str | None = Field(None, description='UUID of the tracked request')
+
+
+
+class VisionModelsResponse(BaseModel):
+    """
+    Response serializer for supported vision models.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    vision_models: dict[str, Any] = Field(description='Vision models by quality tier. Options: fast, balanced, best')
+    ocr_models: dict[str, Any] = Field(description='OCR models by mode. Options: tiny, small, base, maximum')
+
+
+
+class OCRRequestRequest(BaseModel):
+    """
+    Request serializer for OCR extraction.
+
+    Request model (no read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    image: str | None = Field(None, description='Base64 encoded image data')
+    image_url: str | None = Field(None, description='URL of the image for OCR')
+    mode: OCRRequestRequestMode | None = Field(None, description='OCR mode affecting quality and speed\n\n* `tiny` - Tiny - fastest, basic text extraction\n* `small` - Small - fast, good for simple text\n* `base` - Base - balanced speed and accuracy\n* `maximum` - Maximum - best accuracy, slower')
+    language_hint: str | None = Field(None, description="Hint for expected language (e.g., 'ko', 'en', 'zh')", max_length=10)
+
+
+
+class OCRResponse(BaseModel):
+    """
+    Response serializer for OCR extraction.
+
+    Response model (includes read-only fields).
+    """
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        extra="allow",
+        frozen=False,
+    )
+
+    text: str = Field(description='Extracted text from the image')
+    confidence: float | None = Field(None, description='Confidence score of OCR extraction', ge=0.0, le=1.0)
+    language: str | None = Field(None, description='Detected language of extracted text')
+    model: str = Field(description='OCR model used')
+    cost_usd: float = Field(description='Cost of this request in USD')
+    cached: bool | None = Field(None, description='Whether result was from cache')
+    request_uuid: str | None = Field(None, description='UUID of the tracked request')
+
+
+

sdkrouter/_api/generated/vision/vision__api__vision/sync_client.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+import httpx
+
+from .models import *
+
+
+class SyncVisionVisionAPI:
+    """Synchronous API endpoints for Vision."""
+
+    def __init__(self, client: httpx.Client):
+        """Initialize sync sub-client with shared httpx client."""
+        self._client = client
+
+    def analyze_create(self, data: VisionAnalyzeRequestRequest) -> VisionAnalyzeResponse:
+        """
+        Analyze image
+
+        Analyze an image with vision model.
+        """
+        url = "/api/vision/analyze/"
+        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 VisionAnalyzeResponse.model_validate(response.json())
+
+
+    def models_retrieve(self) -> VisionModelsResponse:
+        """
+        Get vision models
+
+        Get supported vision models.
+        """
+        url = "/api/vision/models/"
+        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 VisionModelsResponse.model_validate(response.json())
+
+
+    def ocr_create(self, data: OCRRequestRequest) -> OCRResponse:
+        """
+        OCR extraction
+
+        Extract text from image using OCR.
+        """
+        url = "/api/vision/ocr/"
+        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 OCRResponse.model_validate(response.json())
+
+