docstrange/services/__init__.py

"""Services for authentication, API key pooling, and local LLM processing."""

from .ollama_service import OllamaFieldExtractor
from .api_key_pool import (
    ApiKeyPool,
    get_pool,
    add_api_key,
    remove_api_key,
    list_api_keys,
    get_available_key,
)

__all__ = [
    "OllamaFieldExtractor",
    "ApiKeyPool",
    "get_pool",
    "add_api_key",
    "remove_api_key",
    "list_api_keys",
    "get_available_key",
]