corvic-engine 0.3.0rc58__cp38-abi3-win_amd64.whl → 0.3.0rc59__cp38-abi3-win_amd64.whl

corvic_generated/completion/v1/service_pb2.pyi

@@ -1,201 +0,0 @@
-from google.protobuf import descriptor_pb2 as _descriptor_pb2
-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 RetrieverType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    RETRIEVER_TYPE_UNSPECIFIED: _ClassVar[RetrieverType]
-    RETRIEVER_TYPE_SIMILARITY_SEARCH: _ClassVar[RetrieverType]
-    RETRIEVER_TYPE_FAKE_SEARCH: _ClassVar[RetrieverType]
-
-class EngineType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    ENGINE_TYPE_UNSPECIFIED: _ClassVar[EngineType]
-    ENGINE_TYPE_HUGGINGFACE_REMOTE: _ClassVar[EngineType]
-    ENGINE_TYPE_HUGGINGFACE_LOCAL: _ClassVar[EngineType]
-    ENGINE_TYPE_HUGGINGFACE_INFERENCE: _ClassVar[EngineType]
-    ENGINE_TYPE_OPEN_AI: _ClassVar[EngineType]
-
-class ChainType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    CHAIN_TYPE_UNSPECIFIED: _ClassVar[ChainType]
-    CHAIN_TYPE_STUFF: _ClassVar[ChainType]
-    CHAIN_TYPE_REFINE: _ClassVar[ChainType]
-RETRIEVER_TYPE_UNSPECIFIED: RetrieverType
-RETRIEVER_TYPE_SIMILARITY_SEARCH: RetrieverType
-RETRIEVER_TYPE_FAKE_SEARCH: RetrieverType
-ENGINE_TYPE_UNSPECIFIED: EngineType
-ENGINE_TYPE_HUGGINGFACE_REMOTE: EngineType
-ENGINE_TYPE_HUGGINGFACE_LOCAL: EngineType
-ENGINE_TYPE_HUGGINGFACE_INFERENCE: EngineType
-ENGINE_TYPE_OPEN_AI: EngineType
-CHAIN_TYPE_UNSPECIFIED: ChainType
-CHAIN_TYPE_STUFF: ChainType
-CHAIN_TYPE_REFINE: ChainType
-
-class CompleteResponse(_message.Message):
-    __slots__ = ("text", "finish_reason", "created", "usage", "chunks")
-    class Usage(_message.Message):
-        __slots__ = ("prompt_tokens", "completion_tokens", "total_tokens", "context_tokens")
-        PROMPT_TOKENS_FIELD_NUMBER: _ClassVar[int]
-        COMPLETION_TOKENS_FIELD_NUMBER: _ClassVar[int]
-        TOTAL_TOKENS_FIELD_NUMBER: _ClassVar[int]
-        CONTEXT_TOKENS_FIELD_NUMBER: _ClassVar[int]
-        prompt_tokens: int
-        completion_tokens: int
-        total_tokens: int
-        context_tokens: int
-        def __init__(self, prompt_tokens: _Optional[int] = ..., completion_tokens: _Optional[int] = ..., total_tokens: _Optional[int] = ..., context_tokens: _Optional[int] = ...) -> None: ...
-    class Chunk(_message.Message):
-        __slots__ = ("chunk_id", "document_id", "distance")
-        CHUNK_ID_FIELD_NUMBER: _ClassVar[int]
-        DOCUMENT_ID_FIELD_NUMBER: _ClassVar[int]
-        DISTANCE_FIELD_NUMBER: _ClassVar[int]
-        chunk_id: str
-        document_id: str
-        distance: float
-        def __init__(self, chunk_id: _Optional[str] = ..., document_id: _Optional[str] = ..., distance: _Optional[float] = ...) -> None: ...
-    TEXT_FIELD_NUMBER: _ClassVar[int]
-    FINISH_REASON_FIELD_NUMBER: _ClassVar[int]
-    CREATED_FIELD_NUMBER: _ClassVar[int]
-    USAGE_FIELD_NUMBER: _ClassVar[int]
-    CHUNKS_FIELD_NUMBER: _ClassVar[int]
-    text: str
-    finish_reason: str
-    created: int
-    usage: CompleteResponse.Usage
-    chunks: _containers.RepeatedCompositeFieldContainer[CompleteResponse.Chunk]
-    def __init__(self, text: _Optional[str] = ..., finish_reason: _Optional[str] = ..., created: _Optional[int] = ..., usage: _Optional[_Union[CompleteResponse.Usage, _Mapping]] = ..., chunks: _Optional[_Iterable[_Union[CompleteResponse.Chunk, _Mapping]]] = ...) -> None: ...
-
-class CompleteRagResponse(_message.Message):
-    __slots__ = ("text", "finish_reason", "created", "usage", "chunks")
-    class Usage(_message.Message):
-        __slots__ = ("prompt_tokens", "completion_tokens", "total_tokens", "context_tokens")
-        PROMPT_TOKENS_FIELD_NUMBER: _ClassVar[int]
-        COMPLETION_TOKENS_FIELD_NUMBER: _ClassVar[int]
-        TOTAL_TOKENS_FIELD_NUMBER: _ClassVar[int]
-        CONTEXT_TOKENS_FIELD_NUMBER: _ClassVar[int]
-        prompt_tokens: int
-        completion_tokens: int
-        total_tokens: int
-        context_tokens: int
-        def __init__(self, prompt_tokens: _Optional[int] = ..., completion_tokens: _Optional[int] = ..., total_tokens: _Optional[int] = ..., context_tokens: _Optional[int] = ...) -> None: ...
-    class Chunk(_message.Message):
-        __slots__ = ("chunk_id", "document_id", "distance")
-        CHUNK_ID_FIELD_NUMBER: _ClassVar[int]
-        DOCUMENT_ID_FIELD_NUMBER: _ClassVar[int]
-        DISTANCE_FIELD_NUMBER: _ClassVar[int]
-        chunk_id: str
-        document_id: str
-        distance: float
-        def __init__(self, chunk_id: _Optional[str] = ..., document_id: _Optional[str] = ..., distance: _Optional[float] = ...) -> None: ...
-    TEXT_FIELD_NUMBER: _ClassVar[int]
-    FINISH_REASON_FIELD_NUMBER: _ClassVar[int]
-    CREATED_FIELD_NUMBER: _ClassVar[int]
-    USAGE_FIELD_NUMBER: _ClassVar[int]
-    CHUNKS_FIELD_NUMBER: _ClassVar[int]
-    text: str
-    finish_reason: str
-    created: int
-    usage: CompleteRagResponse.Usage
-    chunks: _containers.RepeatedCompositeFieldContainer[CompleteRagResponse.Chunk]
-    def __init__(self, text: _Optional[str] = ..., finish_reason: _Optional[str] = ..., created: _Optional[int] = ..., usage: _Optional[_Union[CompleteRagResponse.Usage, _Mapping]] = ..., chunks: _Optional[_Iterable[_Union[CompleteRagResponse.Chunk, _Mapping]]] = ...) -> None: ...
-
-class GetDefaultResponse(_message.Message):
-    __slots__ = ("sys_prompt", "max_chunks", "chain_type", "embed_engine", "embed_model_name", "workspace", "collection", "tag", "threshold", "retriever_type")
-    SYS_PROMPT_FIELD_NUMBER: _ClassVar[int]
-    MAX_CHUNKS_FIELD_NUMBER: _ClassVar[int]
-    CHAIN_TYPE_FIELD_NUMBER: _ClassVar[int]
-    EMBED_ENGINE_FIELD_NUMBER: _ClassVar[int]
-    EMBED_MODEL_NAME_FIELD_NUMBER: _ClassVar[int]
-    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
-    COLLECTION_FIELD_NUMBER: _ClassVar[int]
-    TAG_FIELD_NUMBER: _ClassVar[int]
-    THRESHOLD_FIELD_NUMBER: _ClassVar[int]
-    RETRIEVER_TYPE_FIELD_NUMBER: _ClassVar[int]
-    sys_prompt: str
-    max_chunks: int
-    chain_type: ChainType
-    embed_engine: EngineType
-    embed_model_name: str
-    workspace: str
-    collection: str
-    tag: str
-    threshold: float
-    retriever_type: RetrieverType
-    def __init__(self, sys_prompt: _Optional[str] = ..., max_chunks: _Optional[int] = ..., chain_type: _Optional[_Union[ChainType, str]] = ..., embed_engine: _Optional[_Union[EngineType, str]] = ..., embed_model_name: _Optional[str] = ..., workspace: _Optional[str] = ..., collection: _Optional[str] = ..., tag: _Optional[str] = ..., threshold: _Optional[float] = ..., retriever_type: _Optional[_Union[RetrieverType, str]] = ...) -> None: ...
-
-class CompleteRagRequest(_message.Message):
-    __slots__ = ("model", "prompt", "room_id", "sys_prompt", "max_chunks", "chain_type", "embed_engine", "embed_model_name", "workspace", "collection", "tag", "threshold", "retriever_type")
-    MODEL_FIELD_NUMBER: _ClassVar[int]
-    PROMPT_FIELD_NUMBER: _ClassVar[int]
-    ROOM_ID_FIELD_NUMBER: _ClassVar[int]
-    SYS_PROMPT_FIELD_NUMBER: _ClassVar[int]
-    MAX_CHUNKS_FIELD_NUMBER: _ClassVar[int]
-    CHAIN_TYPE_FIELD_NUMBER: _ClassVar[int]
-    EMBED_ENGINE_FIELD_NUMBER: _ClassVar[int]
-    EMBED_MODEL_NAME_FIELD_NUMBER: _ClassVar[int]
-    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
-    COLLECTION_FIELD_NUMBER: _ClassVar[int]
-    TAG_FIELD_NUMBER: _ClassVar[int]
-    THRESHOLD_FIELD_NUMBER: _ClassVar[int]
-    RETRIEVER_TYPE_FIELD_NUMBER: _ClassVar[int]
-    model: str
-    prompt: str
-    room_id: int
-    sys_prompt: str
-    max_chunks: int
-    chain_type: ChainType
-    embed_engine: EngineType
-    embed_model_name: str
-    workspace: str
-    collection: str
-    tag: str
-    threshold: float
-    retriever_type: RetrieverType
-    def __init__(self, model: _Optional[str] = ..., prompt: _Optional[str] = ..., room_id: _Optional[int] = ..., sys_prompt: _Optional[str] = ..., max_chunks: _Optional[int] = ..., chain_type: _Optional[_Union[ChainType, str]] = ..., embed_engine: _Optional[_Union[EngineType, str]] = ..., embed_model_name: _Optional[str] = ..., workspace: _Optional[str] = ..., collection: _Optional[str] = ..., tag: _Optional[str] = ..., threshold: _Optional[float] = ..., retriever_type: _Optional[_Union[RetrieverType, str]] = ...) -> None: ...
-
-class CompleteRequest(_message.Message):
-    __slots__ = ("model", "prompt", "sys_prompt")
-    MODEL_FIELD_NUMBER: _ClassVar[int]
-    PROMPT_FIELD_NUMBER: _ClassVar[int]
-    SYS_PROMPT_FIELD_NUMBER: _ClassVar[int]
-    model: str
-    prompt: str
-    sys_prompt: str
-    def __init__(self, model: _Optional[str] = ..., prompt: _Optional[str] = ..., sys_prompt: _Optional[str] = ...) -> None: ...
-
-class GetShaResponse(_message.Message):
-    __slots__ = ("sha",)
-    SHA_FIELD_NUMBER: _ClassVar[int]
-    sha: str
-    def __init__(self, sha: _Optional[str] = ...) -> None: ...
-
-class GetShaRequest(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
-class GetModelsRequest(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
-class GetDefaultRequest(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
-class Model(_message.Message):
-    __slots__ = ("name",)
-    NAME_FIELD_NUMBER: _ClassVar[int]
-    name: str
-    def __init__(self, name: _Optional[str] = ...) -> None: ...
-
-class GetModelsResponse(_message.Message):
-    __slots__ = ("names",)
-    NAMES_FIELD_NUMBER: _ClassVar[int]
-    names: _containers.RepeatedCompositeFieldContainer[Model]
-    def __init__(self, names: _Optional[_Iterable[_Union[Model, _Mapping]]] = ...) -> None: ...

corvic_generated/completion/v1/service_pb2_grpc.py

@@ -1,198 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc
-
-from corvic_generated.completion.v1 import service_pb2 as corvic_dot_completion_dot_v1_dot_service__pb2
-
-
-class CompletionServiceStub(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def __init__(self, channel):
-        """Constructor.
-
-        Args:
-            channel: A grpc.Channel.
-        """
-        self.Complete = channel.unary_unary(
-                '/corvic.completion.v1.CompletionService/Complete',
-                request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRequest.SerializeToString,
-                response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteResponse.FromString,
-                )
-        self.CompleteRag = channel.unary_unary(
-                '/corvic.completion.v1.CompletionService/CompleteRag',
-                request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagRequest.SerializeToString,
-                response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagResponse.FromString,
-                )
-        self.GetSha = channel.unary_unary(
-                '/corvic.completion.v1.CompletionService/GetSha',
-                request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetShaRequest.SerializeToString,
-                response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetShaResponse.FromString,
-                )
-        self.GetModels = channel.unary_unary(
-                '/corvic.completion.v1.CompletionService/GetModels',
-                request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsRequest.SerializeToString,
-                response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsResponse.FromString,
-                )
-        self.GetDefault = channel.unary_unary(
-                '/corvic.completion.v1.CompletionService/GetDefault',
-                request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultRequest.SerializeToString,
-                response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultResponse.FromString,
-                )
-
-
-class CompletionServiceServicer(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def Complete(self, request, context):
-        """Missing associated documentation comment in .proto file."""
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def CompleteRag(self, request, context):
-        """Missing associated documentation comment in .proto file."""
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetSha(self, request, context):
-        """Missing associated documentation comment in .proto file."""
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetModels(self, request, context):
-        """Missing associated documentation comment in .proto file."""
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetDefault(self, request, context):
-        """Missing associated documentation comment in .proto file."""
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-
-def add_CompletionServiceServicer_to_server(servicer, server):
-    rpc_method_handlers = {
-            'Complete': grpc.unary_unary_rpc_method_handler(
-                    servicer.Complete,
-                    request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRequest.FromString,
-                    response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteResponse.SerializeToString,
-            ),
-            'CompleteRag': grpc.unary_unary_rpc_method_handler(
-                    servicer.CompleteRag,
-                    request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagRequest.FromString,
-                    response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagResponse.SerializeToString,
-            ),
-            'GetSha': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetSha,
-                    request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetShaRequest.FromString,
-                    response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetShaResponse.SerializeToString,
-            ),
-            'GetModels': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetModels,
-                    request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsRequest.FromString,
-                    response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsResponse.SerializeToString,
-            ),
-            'GetDefault': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetDefault,
-                    request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultRequest.FromString,
-                    response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultResponse.SerializeToString,
-            ),
-    }
-    generic_handler = grpc.method_handlers_generic_handler(
-            'corvic.completion.v1.CompletionService', rpc_method_handlers)
-    server.add_generic_rpc_handlers((generic_handler,))
-
-
- # This class is part of an EXPERIMENTAL API.
-class CompletionService(object):
-    """Missing associated documentation comment in .proto file."""
-
-    @staticmethod
-    def Complete(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/Complete',
-            corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRequest.SerializeToString,
-            corvic_dot_completion_dot_v1_dot_service__pb2.CompleteResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def CompleteRag(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/CompleteRag',
-            corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagRequest.SerializeToString,
-            corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetSha(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/GetSha',
-            corvic_dot_completion_dot_v1_dot_service__pb2.GetShaRequest.SerializeToString,
-            corvic_dot_completion_dot_v1_dot_service__pb2.GetShaResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetModels(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/GetModels',
-            corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsRequest.SerializeToString,
-            corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetDefault(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/GetDefault',
-            corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultRequest.SerializeToString,
-            corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

corvic_generated/completion/v1/service_pb2_grpc.pyi

@@ -1,97 +0,0 @@
-"""
-@generated by mypy-protobuf.  Do not edit manually!
-isort:skip_file
-"""
-import abc
-import collections.abc
-import corvic_generated.completion.v1.service_pb2
-import grpc
-import grpc.aio
-import typing
-
-_T = typing.TypeVar('_T')
-
-class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta):
-    ...
-
-class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore
-    ...
-
-class CompletionServiceStub:
-    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
-    Complete: grpc.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.CompleteRequest,
-        corvic_generated.completion.v1.service_pb2.CompleteResponse,
-    ]
-    CompleteRag: grpc.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
-        corvic_generated.completion.v1.service_pb2.CompleteRagResponse,
-    ]
-    GetSha: grpc.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.GetShaRequest,
-        corvic_generated.completion.v1.service_pb2.GetShaResponse,
-    ]
-    GetModels: grpc.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.GetModelsRequest,
-        corvic_generated.completion.v1.service_pb2.GetModelsResponse,
-    ]
-    GetDefault: grpc.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
-        corvic_generated.completion.v1.service_pb2.GetDefaultResponse,
-    ]
-
-class CompletionServiceAsyncStub:
-    Complete: grpc.aio.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.CompleteRequest,
-        corvic_generated.completion.v1.service_pb2.CompleteResponse,
-    ]
-    CompleteRag: grpc.aio.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
-        corvic_generated.completion.v1.service_pb2.CompleteRagResponse,
-    ]
-    GetSha: grpc.aio.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.GetShaRequest,
-        corvic_generated.completion.v1.service_pb2.GetShaResponse,
-    ]
-    GetModels: grpc.aio.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.GetModelsRequest,
-        corvic_generated.completion.v1.service_pb2.GetModelsResponse,
-    ]
-    GetDefault: grpc.aio.UnaryUnaryMultiCallable[
-        corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
-        corvic_generated.completion.v1.service_pb2.GetDefaultResponse,
-    ]
-
-class CompletionServiceServicer(metaclass=abc.ABCMeta):
-    @abc.abstractmethod
-    def Complete(
-        self,
-        request: corvic_generated.completion.v1.service_pb2.CompleteRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[corvic_generated.completion.v1.service_pb2.CompleteResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.CompleteResponse]]: ...
-    @abc.abstractmethod
-    def CompleteRag(
-        self,
-        request: corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[corvic_generated.completion.v1.service_pb2.CompleteRagResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.CompleteRagResponse]]: ...
-    @abc.abstractmethod
-    def GetSha(
-        self,
-        request: corvic_generated.completion.v1.service_pb2.GetShaRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetShaResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetShaResponse]]: ...
-    @abc.abstractmethod
-    def GetModels(
-        self,
-        request: corvic_generated.completion.v1.service_pb2.GetModelsRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetModelsResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetModelsResponse]]: ...
-    @abc.abstractmethod
-    def GetDefault(
-        self,
-        request: corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetDefaultResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetDefaultResponse]]: ...
-
-def add_CompletionServiceServicer_to_server(servicer: CompletionServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

corvic_generated/query/v1/query_pb2.py

@@ -1,36 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: corvic/query/v1/query.proto
-# Protobuf Python Version: 4.25.3
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x63orvic/query/v1/query.proto\x12\x0f\x63orvic.query.v1\"P\n\nTokenUsage\x12\x1f\n\x0btoken_input\x18\x01 \x01(\x01R\ntokenInput\x12!\n\x0ctoken_output\x18\x02 \x01(\x01R\x0btokenOutput\"\xd4\x01\n\x0cQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query\x12\x39\n\nrag_method\x18\x02 \x01(\x0e\x32\x1a.corvic.query.v1.RagMethodR\tragMethod\x12\x38\n\x18model_system_instruction\x18\x03 \x01(\tR\x16modelSystemInstruction\x12\x39\n\nmodel_type\x18\x04 \x01(\x0e\x32\x1a.corvic.query.v1.ModelTypeR\tmodelType\"\x83\x01\n\rQueryResponse\x12<\n\x0btoken_usage\x18\x01 \x01(\x0b\x32\x1b.corvic.query.v1.TokenUsageR\ntokenUsage\x12\x1a\n\x08response\x18\x02 \x01(\tR\x08response\x12\x18\n\x07sources\x18\x03 \x03(\tR\x07sources*T\n\tRagMethod\x12\x1a\n\x16RAG_METHOD_UNSPECIFIED\x10\x00\x12\x14\n\x10RAG_METHOD_NAIVE\x10\x01\x12\x15\n\x11RAG_METHOD_CORVIC\x10\x02*\xed\x01\n\tModelType\x12\x1a\n\x16MODEL_TYPE_UNSPECIFIED\x10\x00\x12\x1d\n\x19MODEL_TYPE_GEMINI_1_5_PRO\x10\x01\x12%\n!MODEL_TYPE_GEMINI_1_5_PRO_PREVIEW\x10\x02\x12\x1f\n\x1bMODEL_TYPE_GEMINI_1_5_FLASH\x10\x03\x12\'\n#MODEL_TYPE_GEMINI_1_5_FLASH_PREVIEW\x10\x04\x12\x16\n\x12MODEL_TYPE_GPT_4_O\x10\x05\x12\x1c\n\x18MODEL_TYPE_GPT_3_5_TURBO\x10\x06\x32Z\n\x0cQueryService\x12J\n\x05Query\x12\x1d.corvic.query.v1.QueryRequest\x1a\x1e.corvic.query.v1.QueryResponse\"\x00\x30\x01\x62\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.query.v1.query_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  _globals['_RAGMETHOD']._serialized_start=479
-  _globals['_RAGMETHOD']._serialized_end=563
-  _globals['_MODELTYPE']._serialized_start=566
-  _globals['_MODELTYPE']._serialized_end=803
-  _globals['_TOKENUSAGE']._serialized_start=48
-  _globals['_TOKENUSAGE']._serialized_end=128
-  _globals['_QUERYREQUEST']._serialized_start=131
-  _globals['_QUERYREQUEST']._serialized_end=343
-  _globals['_QUERYRESPONSE']._serialized_start=346
-  _globals['_QUERYRESPONSE']._serialized_end=477
-  _globals['_QUERYSERVICE']._serialized_start=805
-  _globals['_QUERYSERVICE']._serialized_end=895
-# @@protoc_insertion_point(module_scope)

corvic_generated/query/v1/query_pb2.pyi

@@ -1,63 +0,0 @@
-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 RagMethod(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    RAG_METHOD_UNSPECIFIED: _ClassVar[RagMethod]
-    RAG_METHOD_NAIVE: _ClassVar[RagMethod]
-    RAG_METHOD_CORVIC: _ClassVar[RagMethod]
-
-class ModelType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = ()
-    MODEL_TYPE_UNSPECIFIED: _ClassVar[ModelType]
-    MODEL_TYPE_GEMINI_1_5_PRO: _ClassVar[ModelType]
-    MODEL_TYPE_GEMINI_1_5_PRO_PREVIEW: _ClassVar[ModelType]
-    MODEL_TYPE_GEMINI_1_5_FLASH: _ClassVar[ModelType]
-    MODEL_TYPE_GEMINI_1_5_FLASH_PREVIEW: _ClassVar[ModelType]
-    MODEL_TYPE_GPT_4_O: _ClassVar[ModelType]
-    MODEL_TYPE_GPT_3_5_TURBO: _ClassVar[ModelType]
-RAG_METHOD_UNSPECIFIED: RagMethod
-RAG_METHOD_NAIVE: RagMethod
-RAG_METHOD_CORVIC: RagMethod
-MODEL_TYPE_UNSPECIFIED: ModelType
-MODEL_TYPE_GEMINI_1_5_PRO: ModelType
-MODEL_TYPE_GEMINI_1_5_PRO_PREVIEW: ModelType
-MODEL_TYPE_GEMINI_1_5_FLASH: ModelType
-MODEL_TYPE_GEMINI_1_5_FLASH_PREVIEW: ModelType
-MODEL_TYPE_GPT_4_O: ModelType
-MODEL_TYPE_GPT_3_5_TURBO: ModelType
-
-class TokenUsage(_message.Message):
-    __slots__ = ("token_input", "token_output")
-    TOKEN_INPUT_FIELD_NUMBER: _ClassVar[int]
-    TOKEN_OUTPUT_FIELD_NUMBER: _ClassVar[int]
-    token_input: float
-    token_output: float
-    def __init__(self, token_input: _Optional[float] = ..., token_output: _Optional[float] = ...) -> None: ...
-
-class QueryRequest(_message.Message):
-    __slots__ = ("query", "rag_method", "model_system_instruction", "model_type")
-    QUERY_FIELD_NUMBER: _ClassVar[int]
-    RAG_METHOD_FIELD_NUMBER: _ClassVar[int]
-    MODEL_SYSTEM_INSTRUCTION_FIELD_NUMBER: _ClassVar[int]
-    MODEL_TYPE_FIELD_NUMBER: _ClassVar[int]
-    query: str
-    rag_method: RagMethod
-    model_system_instruction: str
-    model_type: ModelType
-    def __init__(self, query: _Optional[str] = ..., rag_method: _Optional[_Union[RagMethod, str]] = ..., model_system_instruction: _Optional[str] = ..., model_type: _Optional[_Union[ModelType, str]] = ...) -> None: ...
-
-class QueryResponse(_message.Message):
-    __slots__ = ("token_usage", "response", "sources")
-    TOKEN_USAGE_FIELD_NUMBER: _ClassVar[int]
-    RESPONSE_FIELD_NUMBER: _ClassVar[int]
-    SOURCES_FIELD_NUMBER: _ClassVar[int]
-    token_usage: TokenUsage
-    response: str
-    sources: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(self, token_usage: _Optional[_Union[TokenUsage, _Mapping]] = ..., response: _Optional[str] = ..., sources: _Optional[_Iterable[str]] = ...) -> None: ...

corvic_generated/query/v1/query_pb2_grpc.py

@@ -1,66 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc
-
-from corvic_generated.query.v1 import query_pb2 as corvic_dot_query_dot_v1_dot_query__pb2
-
-
-class QueryServiceStub(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def __init__(self, channel):
-        """Constructor.
-
-        Args:
-            channel: A grpc.Channel.
-        """
-        self.Query = channel.unary_stream(
-                '/corvic.query.v1.QueryService/Query',
-                request_serializer=corvic_dot_query_dot_v1_dot_query__pb2.QueryRequest.SerializeToString,
-                response_deserializer=corvic_dot_query_dot_v1_dot_query__pb2.QueryResponse.FromString,
-                )
-
-
-class QueryServiceServicer(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def Query(self, request, context):
-        """Missing associated documentation comment in .proto file."""
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-
-def add_QueryServiceServicer_to_server(servicer, server):
-    rpc_method_handlers = {
-            'Query': grpc.unary_stream_rpc_method_handler(
-                    servicer.Query,
-                    request_deserializer=corvic_dot_query_dot_v1_dot_query__pb2.QueryRequest.FromString,
-                    response_serializer=corvic_dot_query_dot_v1_dot_query__pb2.QueryResponse.SerializeToString,
-            ),
-    }
-    generic_handler = grpc.method_handlers_generic_handler(
-            'corvic.query.v1.QueryService', rpc_method_handlers)
-    server.add_generic_rpc_handlers((generic_handler,))
-
-
- # This class is part of an EXPERIMENTAL API.
-class QueryService(object):
-    """Missing associated documentation comment in .proto file."""
-
-    @staticmethod
-    def Query(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_stream(request, target, '/corvic.query.v1.QueryService/Query',
-            corvic_dot_query_dot_v1_dot_query__pb2.QueryRequest.SerializeToString,
-            corvic_dot_query_dot_v1_dot_query__pb2.QueryResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)