"""
@generated by mypy-protobuf.  Do not edit manually!
isort:skip_file
"""
import builtins
import google.protobuf.descriptor
import google.protobuf.message
import sys
import typing

if sys.version_info >= (3, 8):
    import typing as typing_extensions
else:
    import typing_extensions

DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

@typing_extensions.final
class TextGenerationInput(google.protobuf.message.Message):
    """TextGenerationInput represents the input of text generation task"""

    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    PROMPT_FIELD_NUMBER: builtins.int
    PROMPT_IMAGE_URL_FIELD_NUMBER: builtins.int
    PROMPT_IMAGE_BASE64_FIELD_NUMBER: builtins.int
    MAX_NEW_TOKENS_FIELD_NUMBER: builtins.int
    STOP_WORDS_LIST_FIELD_NUMBER: builtins.int
    TEMPERATURE_FIELD_NUMBER: builtins.int
    TOP_K_FIELD_NUMBER: builtins.int
    SEED_FIELD_NUMBER: builtins.int
    EXTRA_PARAMS_FIELD_NUMBER: builtins.int
    prompt: builtins.str
    """The prompt text"""
    prompt_image_url: builtins.str
    """Image type URL"""
    prompt_image_base64: builtins.str
    """Image type base64"""
    max_new_tokens: builtins.int
    """The maximum number of tokens for model to generate"""
    stop_words_list: builtins.str
    """The trigger words to stop generation"""
    temperature: builtins.float
    """The temperature for sampling"""
    top_k: builtins.int
    """Top k for sampling"""
    seed: builtins.int
    """The seed"""
    extra_params: builtins.str
    """The extra parameters"""
    def __init__(
        self,
        *,
        prompt: builtins.str = ...,
        prompt_image_url: builtins.str = ...,
        prompt_image_base64: builtins.str = ...,
        max_new_tokens: builtins.int | None = ...,
        stop_words_list: builtins.str | None = ...,
        temperature: builtins.float | None = ...,
        top_k: builtins.int | None = ...,
        seed: builtins.int | None = ...,
        extra_params: builtins.str | None = ...,
    ) -> None: ...
    def HasField(self, field_name: typing_extensions.Literal["_extra_params", b"_extra_params", "_max_new_tokens", b"_max_new_tokens", "_seed", b"_seed", "_stop_words_list", b"_stop_words_list", "_temperature", b"_temperature", "_top_k", b"_top_k", "extra_params", b"extra_params", "max_new_tokens", b"max_new_tokens", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "seed", b"seed", "stop_words_list", b"stop_words_list", "temperature", b"temperature", "top_k", b"top_k", "type", b"type"]) -> builtins.bool: ...
    def ClearField(self, field_name: typing_extensions.Literal["_extra_params", b"_extra_params", "_max_new_tokens", b"_max_new_tokens", "_seed", b"_seed", "_stop_words_list", b"_stop_words_list", "_temperature", b"_temperature", "_top_k", b"_top_k", "extra_params", b"extra_params", "max_new_tokens", b"max_new_tokens", "prompt", b"prompt", "prompt_image_base64", b"prompt_image_base64", "prompt_image_url", b"prompt_image_url", "seed", b"seed", "stop_words_list", b"stop_words_list", "temperature", b"temperature", "top_k", b"top_k", "type", b"type"]) -> None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["_extra_params", b"_extra_params"]) -> typing_extensions.Literal["extra_params"] | None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["_max_new_tokens", b"_max_new_tokens"]) -> typing_extensions.Literal["max_new_tokens"] | None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["_seed", b"_seed"]) -> typing_extensions.Literal["seed"] | None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["_stop_words_list", b"_stop_words_list"]) -> typing_extensions.Literal["stop_words_list"] | None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["_temperature", b"_temperature"]) -> typing_extensions.Literal["temperature"] | None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["_top_k", b"_top_k"]) -> typing_extensions.Literal["top_k"] | None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["prompt_image_url", "prompt_image_base64"] | None: ...

global___TextGenerationInput = TextGenerationInput

@typing_extensions.final
class TextGenerationOutput(google.protobuf.message.Message):
    """TextGenerationOutput represents the output of text generation task"""

    DESCRIPTOR: google.protobuf.descriptor.Descriptor

    TEXT_FIELD_NUMBER: builtins.int
    text: builtins.str
    """The text generated by the model"""
    def __init__(
        self,
        *,
        text: builtins.str = ...,
    ) -> None: ...
    def ClearField(self, field_name: typing_extensions.Literal["text", b"text"]) -> None: ...

global___TextGenerationOutput = TextGenerationOutput
