"""
@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
    OUTPUT_LEN_FIELD_NUMBER: builtins.int
    BAD_WORDS_LIST_FIELD_NUMBER: builtins.int
    STOP_WORDS_LIST_FIELD_NUMBER: builtins.int
    TOPK_FIELD_NUMBER: builtins.int
    SEED_FIELD_NUMBER: builtins.int
    prompt: builtins.str
    """The prompt text"""
    output_len: builtins.int
    """The maximum number of tokens for model to generate"""
    bad_words_list: builtins.str
    """The words not to avoid being generated by the model"""
    stop_words_list: builtins.str
    """The trigger words to stop generation"""
    topk: builtins.int
    """Top k for sampling"""
    seed: builtins.int
    """The seed"""
    def __init__(
        self,
        *,
        prompt: builtins.str = ...,
        output_len: builtins.int | None = ...,
        bad_words_list: builtins.str | None = ...,
        stop_words_list: builtins.str | None = ...,
        topk: builtins.int | None = ...,
        seed: builtins.int | None = ...,
    ) -> None: ...
    def HasField(self, field_name: typing_extensions.Literal["_bad_words_list", b"_bad_words_list", "_output_len", b"_output_len", "_seed", b"_seed", "_stop_words_list", b"_stop_words_list", "_topk", b"_topk", "bad_words_list", b"bad_words_list", "output_len", b"output_len", "seed", b"seed", "stop_words_list", b"stop_words_list", "topk", b"topk"]) -> builtins.bool: ...
    def ClearField(self, field_name: typing_extensions.Literal["_bad_words_list", b"_bad_words_list", "_output_len", b"_output_len", "_seed", b"_seed", "_stop_words_list", b"_stop_words_list", "_topk", b"_topk", "bad_words_list", b"bad_words_list", "output_len", b"output_len", "prompt", b"prompt", "seed", b"seed", "stop_words_list", b"stop_words_list", "topk", b"topk"]) -> None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["_bad_words_list", b"_bad_words_list"]) -> typing_extensions.Literal["bad_words_list"] | None: ...
    @typing.overload
    def WhichOneof(self, oneof_group: typing_extensions.Literal["_output_len", b"_output_len"]) -> typing_extensions.Literal["output_len"] | 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["_topk", b"_topk"]) -> typing_extensions.Literal["topk"] | 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
