douyin/grpc_m/vector/vector_pb2.pyi

from google.protobuf.internal import containers as _containers
from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union

DESCRIPTOR: _descriptor.FileDescriptor

class ErrorCode(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
    __slots__ = ()
    SUCCESS: _ClassVar[ErrorCode]
    DOC_CONVERT_ERROR: _ClassVar[ErrorCode]
    VECTOR_SERVER_ERROR: _ClassVar[ErrorCode]
SUCCESS: ErrorCode
DOC_CONVERT_ERROR: ErrorCode
VECTOR_SERVER_ERROR: ErrorCode

class SaveDocToVectorRequest(_message.Message):
    __slots__ = ("collection_name", "user_doc_relative_path")
    COLLECTION_NAME_FIELD_NUMBER: _ClassVar[int]
    USER_DOC_RELATIVE_PATH_FIELD_NUMBER: _ClassVar[int]
    collection_name: str
    user_doc_relative_path: str
    def __init__(self, collection_name: _Optional[str] = ..., user_doc_relative_path: _Optional[str] = ...) -> None: ...

class SaveDocToVectorResponse(_message.Message):
    __slots__ = ("status",)
    STATUS_FIELD_NUMBER: _ClassVar[int]
    status: ErrorCode
    def __init__(self, status: _Optional[_Union[ErrorCode, str]] = ...) -> None: ...

class DocChatRequest(_message.Message):
    __slots__ = ("collection_name", "prompt")
    COLLECTION_NAME_FIELD_NUMBER: _ClassVar[int]
    PROMPT_FIELD_NUMBER: _ClassVar[int]
    collection_name: str
    prompt: str
    def __init__(self, collection_name: _Optional[str] = ..., prompt: _Optional[str] = ...) -> None: ...

class DocChatResponse(_message.Message):
    __slots__ = ("reply",)
    REPLY_FIELD_NUMBER: _ClassVar[int]
    reply: str
    def __init__(self, reply: _Optional[str] = ...) -> None: ...

class SearchRequest(_message.Message):
    __slots__ = ("collection_name", "query")
    COLLECTION_NAME_FIELD_NUMBER: _ClassVar[int]
    QUERY_FIELD_NUMBER: _ClassVar[int]
    collection_name: str
    query: str
    def __init__(self, collection_name: _Optional[str] = ..., query: _Optional[str] = ...) -> None: ...

class SearchWithIdsRequest(_message.Message):
    __slots__ = ("uuid",)
    UUID_FIELD_NUMBER: _ClassVar[int]
    uuid: str
    def __init__(self, uuid: _Optional[str] = ...) -> None: ...

class SearchWithIdsRequests(_message.Message):
    __slots__ = ("request",)
    REQUEST_FIELD_NUMBER: _ClassVar[int]
    request: _containers.RepeatedCompositeFieldContainer[SearchWithIdsRequest]
    def __init__(self, request: _Optional[_Iterable[_Union[SearchWithIdsRequest, _Mapping]]] = ...) -> None: ...

class SearchResponse(_message.Message):
    __slots__ = ("chunk", "metadata", "score", "uuid")
    class MetadataEntry(_message.Message):
        __slots__ = ("key", "value")
        KEY_FIELD_NUMBER: _ClassVar[int]
        VALUE_FIELD_NUMBER: _ClassVar[int]
        key: str
        value: str
        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
    CHUNK_FIELD_NUMBER: _ClassVar[int]
    METADATA_FIELD_NUMBER: _ClassVar[int]
    SCORE_FIELD_NUMBER: _ClassVar[int]
    UUID_FIELD_NUMBER: _ClassVar[int]
    chunk: str
    metadata: _containers.ScalarMap[str, str]
    score: float
    uuid: str
    def __init__(self, chunk: _Optional[str] = ..., metadata: _Optional[_Mapping[str, str]] = ..., score: _Optional[float] = ..., uuid: _Optional[str] = ...) -> None: ...

class SearchResponses(_message.Message):
    __slots__ = ("data",)
    DATA_FIELD_NUMBER: _ClassVar[int]
    data: _containers.RepeatedCompositeFieldContainer[SearchResponse]
    def __init__(self, data: _Optional[_Iterable[_Union[SearchResponse, _Mapping]]] = ...) -> None: ...