src/models/schemas/__init__.py

"""
Pydantic schemas for API requests and responses.

This package exports all schema models for easy importing throughout
the application.
"""

from .common import (
    SparseEmbedding,
    ModelInfo,
    HealthStatus,
    ErrorResponse,
    EmbeddingOptions,
)

from .requests import BaseEmbedRequest, EmbedRequest, RerankRequest

from .responses import (
    BaseEmbedResponse,
    DenseEmbedResponse,
    SparseEmbedResponse,
    RerankResponse,
    EmbeddingObject,
    TokenUsage,
    RerankResult,
    ModelsListResponse,
    RootResponse,
)

__all__ = [
    # Common
    "SparseEmbedding",
    "ModelInfo",
    "HealthStatus",
    "ErrorResponse",
    "EmbeddingOptions",
    # Requests
    "BaseEmbedRequest",
    "EmbedRequest",
    "RerankRequest",
    # Responses
    "BaseEmbedResponse",
    "DenseEmbedResponse",
    "EmbeddingObject",
    "TokenUsage",
    "SparseEmbedResponse",
    "RerankResponse",
    "RerankResult",
    "ModelsListResponse",
    "RootResponse",
]