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

corvic_generated/chat/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.chat.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 ThreadServiceStub:
-    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
-    CreateThread: grpc.UnaryUnaryMultiCallable[
-        corvic_generated.chat.v1.service_pb2.CreateThreadRequest,
-        corvic_generated.chat.v1.service_pb2.CreateThreadResponse,
-    ]
-    DeleteThread: grpc.UnaryUnaryMultiCallable[
-        corvic_generated.chat.v1.service_pb2.DeleteThreadRequest,
-        corvic_generated.chat.v1.service_pb2.DeleteThreadResponse,
-    ]
-    ListThreads: grpc.UnaryStreamMultiCallable[
-        corvic_generated.chat.v1.service_pb2.ListThreadsRequest,
-        corvic_generated.chat.v1.service_pb2.ListThreadsResponse,
-    ]
-    AppendMessage: grpc.UnaryUnaryMultiCallable[
-        corvic_generated.chat.v1.service_pb2.AppendMessageRequest,
-        corvic_generated.chat.v1.service_pb2.AppendMessageResponse,
-    ]
-    GetMessages: grpc.UnaryStreamMultiCallable[
-        corvic_generated.chat.v1.service_pb2.GetMessagesRequest,
-        corvic_generated.chat.v1.service_pb2.GetMessagesResponse,
-    ]
-
-class ThreadServiceAsyncStub:
-    CreateThread: grpc.aio.UnaryUnaryMultiCallable[
-        corvic_generated.chat.v1.service_pb2.CreateThreadRequest,
-        corvic_generated.chat.v1.service_pb2.CreateThreadResponse,
-    ]
-    DeleteThread: grpc.aio.UnaryUnaryMultiCallable[
-        corvic_generated.chat.v1.service_pb2.DeleteThreadRequest,
-        corvic_generated.chat.v1.service_pb2.DeleteThreadResponse,
-    ]
-    ListThreads: grpc.aio.UnaryStreamMultiCallable[
-        corvic_generated.chat.v1.service_pb2.ListThreadsRequest,
-        corvic_generated.chat.v1.service_pb2.ListThreadsResponse,
-    ]
-    AppendMessage: grpc.aio.UnaryUnaryMultiCallable[
-        corvic_generated.chat.v1.service_pb2.AppendMessageRequest,
-        corvic_generated.chat.v1.service_pb2.AppendMessageResponse,
-    ]
-    GetMessages: grpc.aio.UnaryStreamMultiCallable[
-        corvic_generated.chat.v1.service_pb2.GetMessagesRequest,
-        corvic_generated.chat.v1.service_pb2.GetMessagesResponse,
-    ]
-
-class ThreadServiceServicer(metaclass=abc.ABCMeta):
-    @abc.abstractmethod
-    def CreateThread(
-        self,
-        request: corvic_generated.chat.v1.service_pb2.CreateThreadRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[corvic_generated.chat.v1.service_pb2.CreateThreadResponse, collections.abc.Awaitable[corvic_generated.chat.v1.service_pb2.CreateThreadResponse]]: ...
-    @abc.abstractmethod
-    def DeleteThread(
-        self,
-        request: corvic_generated.chat.v1.service_pb2.DeleteThreadRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[corvic_generated.chat.v1.service_pb2.DeleteThreadResponse, collections.abc.Awaitable[corvic_generated.chat.v1.service_pb2.DeleteThreadResponse]]: ...
-    @abc.abstractmethod
-    def ListThreads(
-        self,
-        request: corvic_generated.chat.v1.service_pb2.ListThreadsRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[collections.abc.Iterator[corvic_generated.chat.v1.service_pb2.ListThreadsResponse], collections.abc.AsyncIterator[corvic_generated.chat.v1.service_pb2.ListThreadsResponse]]: ...
-    @abc.abstractmethod
-    def AppendMessage(
-        self,
-        request: corvic_generated.chat.v1.service_pb2.AppendMessageRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[corvic_generated.chat.v1.service_pb2.AppendMessageResponse, collections.abc.Awaitable[corvic_generated.chat.v1.service_pb2.AppendMessageResponse]]: ...
-    @abc.abstractmethod
-    def GetMessages(
-        self,
-        request: corvic_generated.chat.v1.service_pb2.GetMessagesRequest,
-        context: _ServicerContext,
-    ) -> typing.Union[collections.abc.Iterator[corvic_generated.chat.v1.service_pb2.GetMessagesResponse], collections.abc.AsyncIterator[corvic_generated.chat.v1.service_pb2.GetMessagesResponse]]: ...
-
-def add_ThreadServiceServicer_to_server(servicer: ThreadServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

corvic_generated/completion/v1/service_pb2.py

@@ -1,69 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: corvic/completion/v1/service.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()
-
-
-from google.protobuf import descriptor_pb2 as google_dot_protobuf_dot_descriptor__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"corvic/completion/v1/service.proto\x12\x14\x63orvic.completion.v1\x1a google/protobuf/descriptor.proto\"\xf6\x03\n\x10\x43ompleteResponse\x12\x12\n\x04text\x18\x01 \x01(\tR\x04text\x12#\n\rfinish_reason\x18\x02 \x01(\tR\x0c\x66inishReason\x12\x18\n\x07\x63reated\x18\x03 \x01(\x03R\x07\x63reated\x12\x42\n\x05usage\x18\x04 \x01(\x0b\x32,.corvic.completion.v1.CompleteResponse.UsageR\x05usage\x12\x44\n\x06\x63hunks\x18\x05 \x03(\x0b\x32,.corvic.completion.v1.CompleteResponse.ChunkR\x06\x63hunks\x1a\xa3\x01\n\x05Usage\x12#\n\rprompt_tokens\x18\x01 \x01(\x05R\x0cpromptTokens\x12+\n\x11\x63ompletion_tokens\x18\x02 \x01(\x05R\x10\x63ompletionTokens\x12!\n\x0ctotal_tokens\x18\x03 \x01(\x05R\x0btotalTokens\x12%\n\x0e\x63ontext_tokens\x18\x04 \x01(\x05R\rcontextTokens\x1a_\n\x05\x43hunk\x12\x19\n\x08\x63hunk_id\x18\x01 \x01(\tR\x07\x63hunkId\x12\x1f\n\x0b\x64ocument_id\x18\x02 \x01(\tR\ndocumentId\x12\x1a\n\x08\x64istance\x18\x03 \x01(\x02R\x08\x64istance\"\xff\x03\n\x13\x43ompleteRagResponse\x12\x12\n\x04text\x18\x01 \x01(\tR\x04text\x12#\n\rfinish_reason\x18\x02 \x01(\tR\x0c\x66inishReason\x12\x18\n\x07\x63reated\x18\x03 \x01(\x03R\x07\x63reated\x12\x45\n\x05usage\x18\x04 \x01(\x0b\x32/.corvic.completion.v1.CompleteRagResponse.UsageR\x05usage\x12G\n\x06\x63hunks\x18\x05 \x03(\x0b\x32/.corvic.completion.v1.CompleteRagResponse.ChunkR\x06\x63hunks\x1a\xa3\x01\n\x05Usage\x12#\n\rprompt_tokens\x18\x01 \x01(\x05R\x0cpromptTokens\x12+\n\x11\x63ompletion_tokens\x18\x02 \x01(\x05R\x10\x63ompletionTokens\x12!\n\x0ctotal_tokens\x18\x03 \x01(\x05R\x0btotalTokens\x12%\n\x0e\x63ontext_tokens\x18\x04 \x01(\x05R\rcontextTokens\x1a_\n\x05\x43hunk\x12\x19\n\x08\x63hunk_id\x18\x01 \x01(\tR\x07\x63hunkId\x12\x1f\n\x0b\x64ocument_id\x18\x02 \x01(\tR\ndocumentId\x12\x1a\n\x08\x64istance\x18\x03 \x01(\x02R\x08\x64istance\"\xbb\x03\n\x12GetDefaultResponse\x12\x1d\n\nsys_prompt\x18\x01 \x01(\tR\tsysPrompt\x12\x1d\n\nmax_chunks\x18\x02 \x01(\x05R\tmaxChunks\x12>\n\nchain_type\x18\x03 \x01(\x0e\x32\x1f.corvic.completion.v1.ChainTypeR\tchainType\x12\x43\n\x0c\x65mbed_engine\x18\x04 \x01(\x0e\x32 .corvic.completion.v1.EngineTypeR\x0b\x65mbedEngine\x12(\n\x10\x65mbed_model_name\x18\x05 \x01(\tR\x0e\x65mbedModelName\x12\x1c\n\tworkspace\x18\x06 \x01(\tR\tworkspace\x12\x1e\n\ncollection\x18\x07 \x01(\tR\ncollection\x12\x10\n\x03tag\x18\x08 \x01(\tR\x03tag\x12\x1c\n\tthreshold\x18\t \x01(\x02R\tthreshold\x12J\n\x0eretriever_type\x18\n \x01(\x0e\x32#.corvic.completion.v1.RetrieverTypeR\rretrieverType\"\x82\x04\n\x12\x43ompleteRagRequest\x12\x14\n\x05model\x18\x01 \x01(\tR\x05model\x12\x16\n\x06prompt\x18\x02 \x01(\tR\x06prompt\x12\x17\n\x07room_id\x18\x03 \x01(\x03R\x06roomId\x12\x1d\n\nsys_prompt\x18\x04 \x01(\tR\tsysPrompt\x12\x1d\n\nmax_chunks\x18\x05 \x01(\x05R\tmaxChunks\x12>\n\nchain_type\x18\x06 \x01(\x0e\x32\x1f.corvic.completion.v1.ChainTypeR\tchainType\x12\x43\n\x0c\x65mbed_engine\x18\x07 \x01(\x0e\x32 .corvic.completion.v1.EngineTypeR\x0b\x65mbedEngine\x12(\n\x10\x65mbed_model_name\x18\x08 \x01(\tR\x0e\x65mbedModelName\x12\x1c\n\tworkspace\x18\t \x01(\tR\tworkspace\x12\x1e\n\ncollection\x18\n \x01(\tR\ncollection\x12\x10\n\x03tag\x18\x0b \x01(\tR\x03tag\x12\x1c\n\tthreshold\x18\x0c \x01(\x02R\tthreshold\x12J\n\x0eretriever_type\x18\r \x01(\x0e\x32#.corvic.completion.v1.RetrieverTypeR\rretrieverType\"^\n\x0f\x43ompleteRequest\x12\x14\n\x05model\x18\x01 \x01(\tR\x05model\x12\x16\n\x06prompt\x18\x02 \x01(\tR\x06prompt\x12\x1d\n\nsys_prompt\x18\x03 \x01(\tR\tsysPrompt\"\"\n\x0eGetShaResponse\x12\x10\n\x03sha\x18\x01 \x01(\tR\x03sha\"\x0f\n\rGetShaRequest\"\x12\n\x10GetModelsRequest\"\x13\n\x11GetDefaultRequest\"\x1b\n\x05Model\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\"F\n\x11GetModelsResponse\x12\x31\n\x05names\x18\x01 \x03(\x0b\x32\x1b.corvic.completion.v1.ModelR\x05names*u\n\rRetrieverType\x12\x1e\n\x1aRETRIEVER_TYPE_UNSPECIFIED\x10\x00\x12$\n RETRIEVER_TYPE_SIMILARITY_SEARCH\x10\x01\x12\x1e\n\x1aRETRIEVER_TYPE_FAKE_SEARCH\x10\x02*\xb0\x01\n\nEngineType\x12\x1b\n\x17\x45NGINE_TYPE_UNSPECIFIED\x10\x00\x12\"\n\x1e\x45NGINE_TYPE_HUGGINGFACE_REMOTE\x10\x01\x12!\n\x1d\x45NGINE_TYPE_HUGGINGFACE_LOCAL\x10\x02\x12%\n!ENGINE_TYPE_HUGGINGFACE_INFERENCE\x10\x03\x12\x17\n\x13\x45NGINE_TYPE_OPEN_AI\x10\x04*T\n\tChainType\x12\x1a\n\x16\x43HAIN_TYPE_UNSPECIFIED\x10\x00\x12\x14\n\x10\x43HAIN_TYPE_STUFF\x10\x01\x12\x15\n\x11\x43HAIN_TYPE_REFINE\x10\x02\x32\xf9\x03\n\x11\x43ompletionService\x12[\n\x08\x43omplete\x12%.corvic.completion.v1.CompleteRequest\x1a&.corvic.completion.v1.CompleteResponse\"\x00\x12\x64\n\x0b\x43ompleteRag\x12(.corvic.completion.v1.CompleteRagRequest\x1a).corvic.completion.v1.CompleteRagResponse\"\x00\x12X\n\x06GetSha\x12#.corvic.completion.v1.GetShaRequest\x1a$.corvic.completion.v1.GetShaResponse\"\x03\x90\x02\x01\x12\x61\n\tGetModels\x12&.corvic.completion.v1.GetModelsRequest\x1a\'.corvic.completion.v1.GetModelsResponse\"\x03\x90\x02\x01\x12\x64\n\nGetDefault\x12\'.corvic.completion.v1.GetDefaultRequest\x1a(.corvic.completion.v1.GetDefaultResponse\"\x03\x90\x02\x01\x62\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.completion.v1.service_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  _globals['_COMPLETIONSERVICE'].methods_by_name['GetSha']._options = None
-  _globals['_COMPLETIONSERVICE'].methods_by_name['GetSha']._serialized_options = b'\220\002\001'
-  _globals['_COMPLETIONSERVICE'].methods_by_name['GetModels']._options = None
-  _globals['_COMPLETIONSERVICE'].methods_by_name['GetModels']._serialized_options = b'\220\002\001'
-  _globals['_COMPLETIONSERVICE'].methods_by_name['GetDefault']._options = None
-  _globals['_COMPLETIONSERVICE'].methods_by_name['GetDefault']._serialized_options = b'\220\002\001'
-  _globals['_RETRIEVERTYPE']._serialized_start=2367
-  _globals['_RETRIEVERTYPE']._serialized_end=2484
-  _globals['_ENGINETYPE']._serialized_start=2487
-  _globals['_ENGINETYPE']._serialized_end=2663
-  _globals['_CHAINTYPE']._serialized_start=2665
-  _globals['_CHAINTYPE']._serialized_end=2749
-  _globals['_COMPLETERESPONSE']._serialized_start=95
-  _globals['_COMPLETERESPONSE']._serialized_end=597
-  _globals['_COMPLETERESPONSE_USAGE']._serialized_start=337
-  _globals['_COMPLETERESPONSE_USAGE']._serialized_end=500
-  _globals['_COMPLETERESPONSE_CHUNK']._serialized_start=502
-  _globals['_COMPLETERESPONSE_CHUNK']._serialized_end=597
-  _globals['_COMPLETERAGRESPONSE']._serialized_start=600
-  _globals['_COMPLETERAGRESPONSE']._serialized_end=1111
-  _globals['_COMPLETERAGRESPONSE_USAGE']._serialized_start=337
-  _globals['_COMPLETERAGRESPONSE_USAGE']._serialized_end=500
-  _globals['_COMPLETERAGRESPONSE_CHUNK']._serialized_start=502
-  _globals['_COMPLETERAGRESPONSE_CHUNK']._serialized_end=597
-  _globals['_GETDEFAULTRESPONSE']._serialized_start=1114
-  _globals['_GETDEFAULTRESPONSE']._serialized_end=1557
-  _globals['_COMPLETERAGREQUEST']._serialized_start=1560
-  _globals['_COMPLETERAGREQUEST']._serialized_end=2074
-  _globals['_COMPLETEREQUEST']._serialized_start=2076
-  _globals['_COMPLETEREQUEST']._serialized_end=2170
-  _globals['_GETSHARESPONSE']._serialized_start=2172
-  _globals['_GETSHARESPONSE']._serialized_end=2206
-  _globals['_GETSHAREQUEST']._serialized_start=2208
-  _globals['_GETSHAREQUEST']._serialized_end=2223
-  _globals['_GETMODELSREQUEST']._serialized_start=2225
-  _globals['_GETMODELSREQUEST']._serialized_end=2243
-  _globals['_GETDEFAULTREQUEST']._serialized_start=2245
-  _globals['_GETDEFAULTREQUEST']._serialized_end=2264
-  _globals['_MODEL']._serialized_start=2266
-  _globals['_MODEL']._serialized_end=2293
-  _globals['_GETMODELSRESPONSE']._serialized_start=2295
-  _globals['_GETMODELSRESPONSE']._serialized_end=2365
-  _globals['_COMPLETIONSERVICE']._serialized_start=2752
-  _globals['_COMPLETIONSERVICE']._serialized_end=3257
-# @@protoc_insertion_point(module_scope)

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)