aphrodite-engine/aphrodite/inputs/__init__.py

from .data import (ExplicitEncoderDecoderPrompt, LLMInputs, ParsedText,
                   ParsedTokens, PromptInputs, SingletonPromptInputs,
                   TextPrompt, TokensPrompt, get_prompt_type,
                   is_valid_encoder_decoder_llm_inputs, parse_and_batch_prompt)
from .registry import InputContext, InputRegistry

INPUT_REGISTRY = InputRegistry()
"""
The global :class:`~InputRegistry` which is used by
:class:`~aphrodite.AphroditeEngine`
to dispatch data processing according to the target model.
See also:
    :ref:`input_processing_pipeline`
"""

__all__ = [
    "ParsedText",
    "ParsedTokens",
    "parse_and_batch_prompt",
    "TextPrompt",
    "TokensPrompt",
    "PromptInputs",
    "LLMInputs",
    "INPUT_REGISTRY",
    "InputContext",
    "InputRegistry",
    "get_prompt_type",
    "is_valid_encoder_decoder_llm_inputs",
    "ExplicitEncoderDecoderPrompt",
    "SingletonPromptInputs",
]