corvic-engine 0.3.0rc57__cp38-abi3-win_amd64.whl → 0.3.0rc59__cp38-abi3-win_amd64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- corvic/engine/_native.pyd +0 -0
- corvic/model/_completion_model.py +15 -0
- corvic/model/_proto_orm_convert.py +4 -0
- corvic/model/_source.py +8 -0
- corvic/op_graph/_transformations.py +3 -3
- corvic/orm/__init__.py +5 -0
- corvic/orm/ids.py +6 -1
- {corvic_engine-0.3.0rc57.dist-info → corvic_engine-0.3.0rc59.dist-info}/METADATA +1 -1
- {corvic_engine-0.3.0rc57.dist-info → corvic_engine-0.3.0rc59.dist-info}/RECORD +23 -53
- corvic_generated/feature/v2/feature_view_pb2.py +39 -47
- corvic_generated/feature/v2/feature_view_pb2.pyi +0 -14
- corvic_generated/feature/v2/space_pb2.py +77 -87
- corvic_generated/ingest/v2/pipeline_pb2.py +33 -27
- corvic_generated/ingest/v2/quick_mode_pb2.py +18 -13
- corvic_generated/ingest/v2/quick_mode_pb2.pyi +1 -0
- corvic_generated/ingest/v2/resource_pb2.py +64 -58
- corvic_generated/ingest/v2/room_pb2.py +32 -32
- corvic_generated/ingest/v2/source_pb2.py +43 -43
- corvic_generated/ingest/v2/table_pb2.py +23 -23
- corvic_generated/model/v1alpha/models_pb2.py +2 -2
- corvic_generated/model/v1alpha/models_pb2.pyi +6 -4
- corvic_generated/chat/__init__.py +0 -0
- corvic_generated/chat/v1/__init__.py +0 -0
- corvic_generated/chat/v1/service_pb2.py +0 -65
- corvic_generated/chat/v1/service_pb2.pyi +0 -132
- corvic_generated/chat/v1/service_pb2_grpc.py +0 -198
- corvic_generated/chat/v1/service_pb2_grpc.pyi +0 -97
- corvic_generated/completion/__init__.py +0 -0
- corvic_generated/completion/v1/__init__.py +0 -0
- corvic_generated/completion/v1/service_pb2.py +0 -69
- corvic_generated/completion/v1/service_pb2.pyi +0 -201
- corvic_generated/completion/v1/service_pb2_grpc.py +0 -198
- corvic_generated/completion/v1/service_pb2_grpc.pyi +0 -97
- corvic_generated/config/__init__.py +0 -0
- corvic_generated/config/v1/__init__.py +0 -0
- corvic_generated/config/v1/settings_pb2.py +0 -128
- corvic_generated/config/v1/settings_pb2.pyi +0 -389
- corvic_generated/config/v1/settings_pb2_grpc.py +0 -4
- corvic_generated/config/v1/settings_pb2_grpc.pyi +0 -17
- corvic_generated/platform/__init__.py +0 -0
- corvic_generated/platform/v1/__init__.py +0 -0
- corvic_generated/platform/v1/platform_pb2.py +0 -101
- corvic_generated/platform/v1/platform_pb2.pyi +0 -147
- corvic_generated/platform/v1/platform_pb2_grpc.py +0 -330
- corvic_generated/platform/v1/platform_pb2_grpc.pyi +0 -153
- corvic_generated/query/__init__.py +0 -0
- corvic_generated/query/v1/__init__.py +0 -0
- corvic_generated/query/v1/query_pb2.py +0 -36
- corvic_generated/query/v1/query_pb2.pyi +0 -63
- corvic_generated/query/v1/query_pb2_grpc.py +0 -66
- corvic_generated/query/v1/query_pb2_grpc.pyi +0 -41
- {corvic_engine-0.3.0rc57.dist-info → corvic_engine-0.3.0rc59.dist-info}/WHEEL +0 -0
- {corvic_engine-0.3.0rc57.dist-info → corvic_engine-0.3.0rc59.dist-info}/licenses/LICENSE +0 -0
@@ -1,201 +0,0 @@
|
|
1
|
-
from google.protobuf import descriptor_pb2 as _descriptor_pb2
|
2
|
-
from google.protobuf.internal import containers as _containers
|
3
|
-
from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
|
4
|
-
from google.protobuf import descriptor as _descriptor
|
5
|
-
from google.protobuf import message as _message
|
6
|
-
from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
|
7
|
-
|
8
|
-
DESCRIPTOR: _descriptor.FileDescriptor
|
9
|
-
|
10
|
-
class RetrieverType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
|
11
|
-
__slots__ = ()
|
12
|
-
RETRIEVER_TYPE_UNSPECIFIED: _ClassVar[RetrieverType]
|
13
|
-
RETRIEVER_TYPE_SIMILARITY_SEARCH: _ClassVar[RetrieverType]
|
14
|
-
RETRIEVER_TYPE_FAKE_SEARCH: _ClassVar[RetrieverType]
|
15
|
-
|
16
|
-
class EngineType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
|
17
|
-
__slots__ = ()
|
18
|
-
ENGINE_TYPE_UNSPECIFIED: _ClassVar[EngineType]
|
19
|
-
ENGINE_TYPE_HUGGINGFACE_REMOTE: _ClassVar[EngineType]
|
20
|
-
ENGINE_TYPE_HUGGINGFACE_LOCAL: _ClassVar[EngineType]
|
21
|
-
ENGINE_TYPE_HUGGINGFACE_INFERENCE: _ClassVar[EngineType]
|
22
|
-
ENGINE_TYPE_OPEN_AI: _ClassVar[EngineType]
|
23
|
-
|
24
|
-
class ChainType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
|
25
|
-
__slots__ = ()
|
26
|
-
CHAIN_TYPE_UNSPECIFIED: _ClassVar[ChainType]
|
27
|
-
CHAIN_TYPE_STUFF: _ClassVar[ChainType]
|
28
|
-
CHAIN_TYPE_REFINE: _ClassVar[ChainType]
|
29
|
-
RETRIEVER_TYPE_UNSPECIFIED: RetrieverType
|
30
|
-
RETRIEVER_TYPE_SIMILARITY_SEARCH: RetrieverType
|
31
|
-
RETRIEVER_TYPE_FAKE_SEARCH: RetrieverType
|
32
|
-
ENGINE_TYPE_UNSPECIFIED: EngineType
|
33
|
-
ENGINE_TYPE_HUGGINGFACE_REMOTE: EngineType
|
34
|
-
ENGINE_TYPE_HUGGINGFACE_LOCAL: EngineType
|
35
|
-
ENGINE_TYPE_HUGGINGFACE_INFERENCE: EngineType
|
36
|
-
ENGINE_TYPE_OPEN_AI: EngineType
|
37
|
-
CHAIN_TYPE_UNSPECIFIED: ChainType
|
38
|
-
CHAIN_TYPE_STUFF: ChainType
|
39
|
-
CHAIN_TYPE_REFINE: ChainType
|
40
|
-
|
41
|
-
class CompleteResponse(_message.Message):
|
42
|
-
__slots__ = ("text", "finish_reason", "created", "usage", "chunks")
|
43
|
-
class Usage(_message.Message):
|
44
|
-
__slots__ = ("prompt_tokens", "completion_tokens", "total_tokens", "context_tokens")
|
45
|
-
PROMPT_TOKENS_FIELD_NUMBER: _ClassVar[int]
|
46
|
-
COMPLETION_TOKENS_FIELD_NUMBER: _ClassVar[int]
|
47
|
-
TOTAL_TOKENS_FIELD_NUMBER: _ClassVar[int]
|
48
|
-
CONTEXT_TOKENS_FIELD_NUMBER: _ClassVar[int]
|
49
|
-
prompt_tokens: int
|
50
|
-
completion_tokens: int
|
51
|
-
total_tokens: int
|
52
|
-
context_tokens: int
|
53
|
-
def __init__(self, prompt_tokens: _Optional[int] = ..., completion_tokens: _Optional[int] = ..., total_tokens: _Optional[int] = ..., context_tokens: _Optional[int] = ...) -> None: ...
|
54
|
-
class Chunk(_message.Message):
|
55
|
-
__slots__ = ("chunk_id", "document_id", "distance")
|
56
|
-
CHUNK_ID_FIELD_NUMBER: _ClassVar[int]
|
57
|
-
DOCUMENT_ID_FIELD_NUMBER: _ClassVar[int]
|
58
|
-
DISTANCE_FIELD_NUMBER: _ClassVar[int]
|
59
|
-
chunk_id: str
|
60
|
-
document_id: str
|
61
|
-
distance: float
|
62
|
-
def __init__(self, chunk_id: _Optional[str] = ..., document_id: _Optional[str] = ..., distance: _Optional[float] = ...) -> None: ...
|
63
|
-
TEXT_FIELD_NUMBER: _ClassVar[int]
|
64
|
-
FINISH_REASON_FIELD_NUMBER: _ClassVar[int]
|
65
|
-
CREATED_FIELD_NUMBER: _ClassVar[int]
|
66
|
-
USAGE_FIELD_NUMBER: _ClassVar[int]
|
67
|
-
CHUNKS_FIELD_NUMBER: _ClassVar[int]
|
68
|
-
text: str
|
69
|
-
finish_reason: str
|
70
|
-
created: int
|
71
|
-
usage: CompleteResponse.Usage
|
72
|
-
chunks: _containers.RepeatedCompositeFieldContainer[CompleteResponse.Chunk]
|
73
|
-
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: ...
|
74
|
-
|
75
|
-
class CompleteRagResponse(_message.Message):
|
76
|
-
__slots__ = ("text", "finish_reason", "created", "usage", "chunks")
|
77
|
-
class Usage(_message.Message):
|
78
|
-
__slots__ = ("prompt_tokens", "completion_tokens", "total_tokens", "context_tokens")
|
79
|
-
PROMPT_TOKENS_FIELD_NUMBER: _ClassVar[int]
|
80
|
-
COMPLETION_TOKENS_FIELD_NUMBER: _ClassVar[int]
|
81
|
-
TOTAL_TOKENS_FIELD_NUMBER: _ClassVar[int]
|
82
|
-
CONTEXT_TOKENS_FIELD_NUMBER: _ClassVar[int]
|
83
|
-
prompt_tokens: int
|
84
|
-
completion_tokens: int
|
85
|
-
total_tokens: int
|
86
|
-
context_tokens: int
|
87
|
-
def __init__(self, prompt_tokens: _Optional[int] = ..., completion_tokens: _Optional[int] = ..., total_tokens: _Optional[int] = ..., context_tokens: _Optional[int] = ...) -> None: ...
|
88
|
-
class Chunk(_message.Message):
|
89
|
-
__slots__ = ("chunk_id", "document_id", "distance")
|
90
|
-
CHUNK_ID_FIELD_NUMBER: _ClassVar[int]
|
91
|
-
DOCUMENT_ID_FIELD_NUMBER: _ClassVar[int]
|
92
|
-
DISTANCE_FIELD_NUMBER: _ClassVar[int]
|
93
|
-
chunk_id: str
|
94
|
-
document_id: str
|
95
|
-
distance: float
|
96
|
-
def __init__(self, chunk_id: _Optional[str] = ..., document_id: _Optional[str] = ..., distance: _Optional[float] = ...) -> None: ...
|
97
|
-
TEXT_FIELD_NUMBER: _ClassVar[int]
|
98
|
-
FINISH_REASON_FIELD_NUMBER: _ClassVar[int]
|
99
|
-
CREATED_FIELD_NUMBER: _ClassVar[int]
|
100
|
-
USAGE_FIELD_NUMBER: _ClassVar[int]
|
101
|
-
CHUNKS_FIELD_NUMBER: _ClassVar[int]
|
102
|
-
text: str
|
103
|
-
finish_reason: str
|
104
|
-
created: int
|
105
|
-
usage: CompleteRagResponse.Usage
|
106
|
-
chunks: _containers.RepeatedCompositeFieldContainer[CompleteRagResponse.Chunk]
|
107
|
-
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: ...
|
108
|
-
|
109
|
-
class GetDefaultResponse(_message.Message):
|
110
|
-
__slots__ = ("sys_prompt", "max_chunks", "chain_type", "embed_engine", "embed_model_name", "workspace", "collection", "tag", "threshold", "retriever_type")
|
111
|
-
SYS_PROMPT_FIELD_NUMBER: _ClassVar[int]
|
112
|
-
MAX_CHUNKS_FIELD_NUMBER: _ClassVar[int]
|
113
|
-
CHAIN_TYPE_FIELD_NUMBER: _ClassVar[int]
|
114
|
-
EMBED_ENGINE_FIELD_NUMBER: _ClassVar[int]
|
115
|
-
EMBED_MODEL_NAME_FIELD_NUMBER: _ClassVar[int]
|
116
|
-
WORKSPACE_FIELD_NUMBER: _ClassVar[int]
|
117
|
-
COLLECTION_FIELD_NUMBER: _ClassVar[int]
|
118
|
-
TAG_FIELD_NUMBER: _ClassVar[int]
|
119
|
-
THRESHOLD_FIELD_NUMBER: _ClassVar[int]
|
120
|
-
RETRIEVER_TYPE_FIELD_NUMBER: _ClassVar[int]
|
121
|
-
sys_prompt: str
|
122
|
-
max_chunks: int
|
123
|
-
chain_type: ChainType
|
124
|
-
embed_engine: EngineType
|
125
|
-
embed_model_name: str
|
126
|
-
workspace: str
|
127
|
-
collection: str
|
128
|
-
tag: str
|
129
|
-
threshold: float
|
130
|
-
retriever_type: RetrieverType
|
131
|
-
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: ...
|
132
|
-
|
133
|
-
class CompleteRagRequest(_message.Message):
|
134
|
-
__slots__ = ("model", "prompt", "room_id", "sys_prompt", "max_chunks", "chain_type", "embed_engine", "embed_model_name", "workspace", "collection", "tag", "threshold", "retriever_type")
|
135
|
-
MODEL_FIELD_NUMBER: _ClassVar[int]
|
136
|
-
PROMPT_FIELD_NUMBER: _ClassVar[int]
|
137
|
-
ROOM_ID_FIELD_NUMBER: _ClassVar[int]
|
138
|
-
SYS_PROMPT_FIELD_NUMBER: _ClassVar[int]
|
139
|
-
MAX_CHUNKS_FIELD_NUMBER: _ClassVar[int]
|
140
|
-
CHAIN_TYPE_FIELD_NUMBER: _ClassVar[int]
|
141
|
-
EMBED_ENGINE_FIELD_NUMBER: _ClassVar[int]
|
142
|
-
EMBED_MODEL_NAME_FIELD_NUMBER: _ClassVar[int]
|
143
|
-
WORKSPACE_FIELD_NUMBER: _ClassVar[int]
|
144
|
-
COLLECTION_FIELD_NUMBER: _ClassVar[int]
|
145
|
-
TAG_FIELD_NUMBER: _ClassVar[int]
|
146
|
-
THRESHOLD_FIELD_NUMBER: _ClassVar[int]
|
147
|
-
RETRIEVER_TYPE_FIELD_NUMBER: _ClassVar[int]
|
148
|
-
model: str
|
149
|
-
prompt: str
|
150
|
-
room_id: int
|
151
|
-
sys_prompt: str
|
152
|
-
max_chunks: int
|
153
|
-
chain_type: ChainType
|
154
|
-
embed_engine: EngineType
|
155
|
-
embed_model_name: str
|
156
|
-
workspace: str
|
157
|
-
collection: str
|
158
|
-
tag: str
|
159
|
-
threshold: float
|
160
|
-
retriever_type: RetrieverType
|
161
|
-
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: ...
|
162
|
-
|
163
|
-
class CompleteRequest(_message.Message):
|
164
|
-
__slots__ = ("model", "prompt", "sys_prompt")
|
165
|
-
MODEL_FIELD_NUMBER: _ClassVar[int]
|
166
|
-
PROMPT_FIELD_NUMBER: _ClassVar[int]
|
167
|
-
SYS_PROMPT_FIELD_NUMBER: _ClassVar[int]
|
168
|
-
model: str
|
169
|
-
prompt: str
|
170
|
-
sys_prompt: str
|
171
|
-
def __init__(self, model: _Optional[str] = ..., prompt: _Optional[str] = ..., sys_prompt: _Optional[str] = ...) -> None: ...
|
172
|
-
|
173
|
-
class GetShaResponse(_message.Message):
|
174
|
-
__slots__ = ("sha",)
|
175
|
-
SHA_FIELD_NUMBER: _ClassVar[int]
|
176
|
-
sha: str
|
177
|
-
def __init__(self, sha: _Optional[str] = ...) -> None: ...
|
178
|
-
|
179
|
-
class GetShaRequest(_message.Message):
|
180
|
-
__slots__ = ()
|
181
|
-
def __init__(self) -> None: ...
|
182
|
-
|
183
|
-
class GetModelsRequest(_message.Message):
|
184
|
-
__slots__ = ()
|
185
|
-
def __init__(self) -> None: ...
|
186
|
-
|
187
|
-
class GetDefaultRequest(_message.Message):
|
188
|
-
__slots__ = ()
|
189
|
-
def __init__(self) -> None: ...
|
190
|
-
|
191
|
-
class Model(_message.Message):
|
192
|
-
__slots__ = ("name",)
|
193
|
-
NAME_FIELD_NUMBER: _ClassVar[int]
|
194
|
-
name: str
|
195
|
-
def __init__(self, name: _Optional[str] = ...) -> None: ...
|
196
|
-
|
197
|
-
class GetModelsResponse(_message.Message):
|
198
|
-
__slots__ = ("names",)
|
199
|
-
NAMES_FIELD_NUMBER: _ClassVar[int]
|
200
|
-
names: _containers.RepeatedCompositeFieldContainer[Model]
|
201
|
-
def __init__(self, names: _Optional[_Iterable[_Union[Model, _Mapping]]] = ...) -> None: ...
|
@@ -1,198 +0,0 @@
|
|
1
|
-
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
|
2
|
-
"""Client and server classes corresponding to protobuf-defined services."""
|
3
|
-
import grpc
|
4
|
-
|
5
|
-
from corvic_generated.completion.v1 import service_pb2 as corvic_dot_completion_dot_v1_dot_service__pb2
|
6
|
-
|
7
|
-
|
8
|
-
class CompletionServiceStub(object):
|
9
|
-
"""Missing associated documentation comment in .proto file."""
|
10
|
-
|
11
|
-
def __init__(self, channel):
|
12
|
-
"""Constructor.
|
13
|
-
|
14
|
-
Args:
|
15
|
-
channel: A grpc.Channel.
|
16
|
-
"""
|
17
|
-
self.Complete = channel.unary_unary(
|
18
|
-
'/corvic.completion.v1.CompletionService/Complete',
|
19
|
-
request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRequest.SerializeToString,
|
20
|
-
response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteResponse.FromString,
|
21
|
-
)
|
22
|
-
self.CompleteRag = channel.unary_unary(
|
23
|
-
'/corvic.completion.v1.CompletionService/CompleteRag',
|
24
|
-
request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagRequest.SerializeToString,
|
25
|
-
response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagResponse.FromString,
|
26
|
-
)
|
27
|
-
self.GetSha = channel.unary_unary(
|
28
|
-
'/corvic.completion.v1.CompletionService/GetSha',
|
29
|
-
request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetShaRequest.SerializeToString,
|
30
|
-
response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetShaResponse.FromString,
|
31
|
-
)
|
32
|
-
self.GetModels = channel.unary_unary(
|
33
|
-
'/corvic.completion.v1.CompletionService/GetModels',
|
34
|
-
request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsRequest.SerializeToString,
|
35
|
-
response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsResponse.FromString,
|
36
|
-
)
|
37
|
-
self.GetDefault = channel.unary_unary(
|
38
|
-
'/corvic.completion.v1.CompletionService/GetDefault',
|
39
|
-
request_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultRequest.SerializeToString,
|
40
|
-
response_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultResponse.FromString,
|
41
|
-
)
|
42
|
-
|
43
|
-
|
44
|
-
class CompletionServiceServicer(object):
|
45
|
-
"""Missing associated documentation comment in .proto file."""
|
46
|
-
|
47
|
-
def Complete(self, request, context):
|
48
|
-
"""Missing associated documentation comment in .proto file."""
|
49
|
-
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
50
|
-
context.set_details('Method not implemented!')
|
51
|
-
raise NotImplementedError('Method not implemented!')
|
52
|
-
|
53
|
-
def CompleteRag(self, request, context):
|
54
|
-
"""Missing associated documentation comment in .proto file."""
|
55
|
-
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
56
|
-
context.set_details('Method not implemented!')
|
57
|
-
raise NotImplementedError('Method not implemented!')
|
58
|
-
|
59
|
-
def GetSha(self, request, context):
|
60
|
-
"""Missing associated documentation comment in .proto file."""
|
61
|
-
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
62
|
-
context.set_details('Method not implemented!')
|
63
|
-
raise NotImplementedError('Method not implemented!')
|
64
|
-
|
65
|
-
def GetModels(self, request, context):
|
66
|
-
"""Missing associated documentation comment in .proto file."""
|
67
|
-
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
68
|
-
context.set_details('Method not implemented!')
|
69
|
-
raise NotImplementedError('Method not implemented!')
|
70
|
-
|
71
|
-
def GetDefault(self, request, context):
|
72
|
-
"""Missing associated documentation comment in .proto file."""
|
73
|
-
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
74
|
-
context.set_details('Method not implemented!')
|
75
|
-
raise NotImplementedError('Method not implemented!')
|
76
|
-
|
77
|
-
|
78
|
-
def add_CompletionServiceServicer_to_server(servicer, server):
|
79
|
-
rpc_method_handlers = {
|
80
|
-
'Complete': grpc.unary_unary_rpc_method_handler(
|
81
|
-
servicer.Complete,
|
82
|
-
request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRequest.FromString,
|
83
|
-
response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteResponse.SerializeToString,
|
84
|
-
),
|
85
|
-
'CompleteRag': grpc.unary_unary_rpc_method_handler(
|
86
|
-
servicer.CompleteRag,
|
87
|
-
request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagRequest.FromString,
|
88
|
-
response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagResponse.SerializeToString,
|
89
|
-
),
|
90
|
-
'GetSha': grpc.unary_unary_rpc_method_handler(
|
91
|
-
servicer.GetSha,
|
92
|
-
request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetShaRequest.FromString,
|
93
|
-
response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetShaResponse.SerializeToString,
|
94
|
-
),
|
95
|
-
'GetModels': grpc.unary_unary_rpc_method_handler(
|
96
|
-
servicer.GetModels,
|
97
|
-
request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsRequest.FromString,
|
98
|
-
response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsResponse.SerializeToString,
|
99
|
-
),
|
100
|
-
'GetDefault': grpc.unary_unary_rpc_method_handler(
|
101
|
-
servicer.GetDefault,
|
102
|
-
request_deserializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultRequest.FromString,
|
103
|
-
response_serializer=corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultResponse.SerializeToString,
|
104
|
-
),
|
105
|
-
}
|
106
|
-
generic_handler = grpc.method_handlers_generic_handler(
|
107
|
-
'corvic.completion.v1.CompletionService', rpc_method_handlers)
|
108
|
-
server.add_generic_rpc_handlers((generic_handler,))
|
109
|
-
|
110
|
-
|
111
|
-
# This class is part of an EXPERIMENTAL API.
|
112
|
-
class CompletionService(object):
|
113
|
-
"""Missing associated documentation comment in .proto file."""
|
114
|
-
|
115
|
-
@staticmethod
|
116
|
-
def Complete(request,
|
117
|
-
target,
|
118
|
-
options=(),
|
119
|
-
channel_credentials=None,
|
120
|
-
call_credentials=None,
|
121
|
-
insecure=False,
|
122
|
-
compression=None,
|
123
|
-
wait_for_ready=None,
|
124
|
-
timeout=None,
|
125
|
-
metadata=None):
|
126
|
-
return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/Complete',
|
127
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRequest.SerializeToString,
|
128
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.CompleteResponse.FromString,
|
129
|
-
options, channel_credentials,
|
130
|
-
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
131
|
-
|
132
|
-
@staticmethod
|
133
|
-
def CompleteRag(request,
|
134
|
-
target,
|
135
|
-
options=(),
|
136
|
-
channel_credentials=None,
|
137
|
-
call_credentials=None,
|
138
|
-
insecure=False,
|
139
|
-
compression=None,
|
140
|
-
wait_for_ready=None,
|
141
|
-
timeout=None,
|
142
|
-
metadata=None):
|
143
|
-
return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/CompleteRag',
|
144
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagRequest.SerializeToString,
|
145
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.CompleteRagResponse.FromString,
|
146
|
-
options, channel_credentials,
|
147
|
-
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
148
|
-
|
149
|
-
@staticmethod
|
150
|
-
def GetSha(request,
|
151
|
-
target,
|
152
|
-
options=(),
|
153
|
-
channel_credentials=None,
|
154
|
-
call_credentials=None,
|
155
|
-
insecure=False,
|
156
|
-
compression=None,
|
157
|
-
wait_for_ready=None,
|
158
|
-
timeout=None,
|
159
|
-
metadata=None):
|
160
|
-
return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/GetSha',
|
161
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.GetShaRequest.SerializeToString,
|
162
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.GetShaResponse.FromString,
|
163
|
-
options, channel_credentials,
|
164
|
-
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
165
|
-
|
166
|
-
@staticmethod
|
167
|
-
def GetModels(request,
|
168
|
-
target,
|
169
|
-
options=(),
|
170
|
-
channel_credentials=None,
|
171
|
-
call_credentials=None,
|
172
|
-
insecure=False,
|
173
|
-
compression=None,
|
174
|
-
wait_for_ready=None,
|
175
|
-
timeout=None,
|
176
|
-
metadata=None):
|
177
|
-
return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/GetModels',
|
178
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsRequest.SerializeToString,
|
179
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.GetModelsResponse.FromString,
|
180
|
-
options, channel_credentials,
|
181
|
-
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
182
|
-
|
183
|
-
@staticmethod
|
184
|
-
def GetDefault(request,
|
185
|
-
target,
|
186
|
-
options=(),
|
187
|
-
channel_credentials=None,
|
188
|
-
call_credentials=None,
|
189
|
-
insecure=False,
|
190
|
-
compression=None,
|
191
|
-
wait_for_ready=None,
|
192
|
-
timeout=None,
|
193
|
-
metadata=None):
|
194
|
-
return grpc.experimental.unary_unary(request, target, '/corvic.completion.v1.CompletionService/GetDefault',
|
195
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultRequest.SerializeToString,
|
196
|
-
corvic_dot_completion_dot_v1_dot_service__pb2.GetDefaultResponse.FromString,
|
197
|
-
options, channel_credentials,
|
198
|
-
insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
|
@@ -1,97 +0,0 @@
|
|
1
|
-
"""
|
2
|
-
@generated by mypy-protobuf. Do not edit manually!
|
3
|
-
isort:skip_file
|
4
|
-
"""
|
5
|
-
import abc
|
6
|
-
import collections.abc
|
7
|
-
import corvic_generated.completion.v1.service_pb2
|
8
|
-
import grpc
|
9
|
-
import grpc.aio
|
10
|
-
import typing
|
11
|
-
|
12
|
-
_T = typing.TypeVar('_T')
|
13
|
-
|
14
|
-
class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta):
|
15
|
-
...
|
16
|
-
|
17
|
-
class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext): # type: ignore
|
18
|
-
...
|
19
|
-
|
20
|
-
class CompletionServiceStub:
|
21
|
-
def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
|
22
|
-
Complete: grpc.UnaryUnaryMultiCallable[
|
23
|
-
corvic_generated.completion.v1.service_pb2.CompleteRequest,
|
24
|
-
corvic_generated.completion.v1.service_pb2.CompleteResponse,
|
25
|
-
]
|
26
|
-
CompleteRag: grpc.UnaryUnaryMultiCallable[
|
27
|
-
corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
|
28
|
-
corvic_generated.completion.v1.service_pb2.CompleteRagResponse,
|
29
|
-
]
|
30
|
-
GetSha: grpc.UnaryUnaryMultiCallable[
|
31
|
-
corvic_generated.completion.v1.service_pb2.GetShaRequest,
|
32
|
-
corvic_generated.completion.v1.service_pb2.GetShaResponse,
|
33
|
-
]
|
34
|
-
GetModels: grpc.UnaryUnaryMultiCallable[
|
35
|
-
corvic_generated.completion.v1.service_pb2.GetModelsRequest,
|
36
|
-
corvic_generated.completion.v1.service_pb2.GetModelsResponse,
|
37
|
-
]
|
38
|
-
GetDefault: grpc.UnaryUnaryMultiCallable[
|
39
|
-
corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
|
40
|
-
corvic_generated.completion.v1.service_pb2.GetDefaultResponse,
|
41
|
-
]
|
42
|
-
|
43
|
-
class CompletionServiceAsyncStub:
|
44
|
-
Complete: grpc.aio.UnaryUnaryMultiCallable[
|
45
|
-
corvic_generated.completion.v1.service_pb2.CompleteRequest,
|
46
|
-
corvic_generated.completion.v1.service_pb2.CompleteResponse,
|
47
|
-
]
|
48
|
-
CompleteRag: grpc.aio.UnaryUnaryMultiCallable[
|
49
|
-
corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
|
50
|
-
corvic_generated.completion.v1.service_pb2.CompleteRagResponse,
|
51
|
-
]
|
52
|
-
GetSha: grpc.aio.UnaryUnaryMultiCallable[
|
53
|
-
corvic_generated.completion.v1.service_pb2.GetShaRequest,
|
54
|
-
corvic_generated.completion.v1.service_pb2.GetShaResponse,
|
55
|
-
]
|
56
|
-
GetModels: grpc.aio.UnaryUnaryMultiCallable[
|
57
|
-
corvic_generated.completion.v1.service_pb2.GetModelsRequest,
|
58
|
-
corvic_generated.completion.v1.service_pb2.GetModelsResponse,
|
59
|
-
]
|
60
|
-
GetDefault: grpc.aio.UnaryUnaryMultiCallable[
|
61
|
-
corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
|
62
|
-
corvic_generated.completion.v1.service_pb2.GetDefaultResponse,
|
63
|
-
]
|
64
|
-
|
65
|
-
class CompletionServiceServicer(metaclass=abc.ABCMeta):
|
66
|
-
@abc.abstractmethod
|
67
|
-
def Complete(
|
68
|
-
self,
|
69
|
-
request: corvic_generated.completion.v1.service_pb2.CompleteRequest,
|
70
|
-
context: _ServicerContext,
|
71
|
-
) -> typing.Union[corvic_generated.completion.v1.service_pb2.CompleteResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.CompleteResponse]]: ...
|
72
|
-
@abc.abstractmethod
|
73
|
-
def CompleteRag(
|
74
|
-
self,
|
75
|
-
request: corvic_generated.completion.v1.service_pb2.CompleteRagRequest,
|
76
|
-
context: _ServicerContext,
|
77
|
-
) -> typing.Union[corvic_generated.completion.v1.service_pb2.CompleteRagResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.CompleteRagResponse]]: ...
|
78
|
-
@abc.abstractmethod
|
79
|
-
def GetSha(
|
80
|
-
self,
|
81
|
-
request: corvic_generated.completion.v1.service_pb2.GetShaRequest,
|
82
|
-
context: _ServicerContext,
|
83
|
-
) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetShaResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetShaResponse]]: ...
|
84
|
-
@abc.abstractmethod
|
85
|
-
def GetModels(
|
86
|
-
self,
|
87
|
-
request: corvic_generated.completion.v1.service_pb2.GetModelsRequest,
|
88
|
-
context: _ServicerContext,
|
89
|
-
) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetModelsResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetModelsResponse]]: ...
|
90
|
-
@abc.abstractmethod
|
91
|
-
def GetDefault(
|
92
|
-
self,
|
93
|
-
request: corvic_generated.completion.v1.service_pb2.GetDefaultRequest,
|
94
|
-
context: _ServicerContext,
|
95
|
-
) -> typing.Union[corvic_generated.completion.v1.service_pb2.GetDefaultResponse, collections.abc.Awaitable[corvic_generated.completion.v1.service_pb2.GetDefaultResponse]]: ...
|
96
|
-
|
97
|
-
def add_CompletionServiceServicer_to_server(servicer: CompletionServiceServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...
|
File without changes
|
File without changes
|
@@ -1,128 +0,0 @@
|
|
1
|
-
# -*- coding: utf-8 -*-
|
2
|
-
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
3
|
-
# source: corvic/config/v1/settings.proto
|
4
|
-
# Protobuf Python Version: 4.25.3
|
5
|
-
"""Generated protocol buffer code."""
|
6
|
-
from google.protobuf import descriptor as _descriptor
|
7
|
-
from google.protobuf import descriptor_pool as _descriptor_pool
|
8
|
-
from google.protobuf import symbol_database as _symbol_database
|
9
|
-
from google.protobuf.internal import builder as _builder
|
10
|
-
# @@protoc_insertion_point(imports)
|
11
|
-
|
12
|
-
_sym_db = _symbol_database.Default()
|
13
|
-
|
14
|
-
|
15
|
-
|
16
|
-
|
17
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1f\x63orvic/config/v1/settings.proto\x12\x10\x63orvic.config.v1\"\x7f\n\nMetaConfig\x12#\n\renvvar_prefix\x18\x01 \x01(\tR\x0c\x65nvvarPrefix\x12)\n\x10nested_separator\x18\x02 \x01(\tR\x0fnestedSeparator\x12!\n\x0cvia_launcher\x18\x03 \x01(\x08R\x0bviaLauncher\"\'\n\rMetricsConfig\x12\x16\n\x06\x65xport\x18\x01 \x01(\x08R\x06\x65xport\"&\n\x10\x43hatServerConfig\x12\x12\n\x04port\x18\x01 \x01(\rR\x04port\"\x81\t\n\rRocksetConfig\x12+\n\x11\x64\x65\x66\x61ult_workspace\x18\x01 \x01(\tR\x10\x64\x65\x66\x61ultWorkspace\x12-\n\x12\x64\x65\x66\x61ult_collection\x18\x02 \x01(\tR\x11\x64\x65\x66\x61ultCollection\x12%\n\x0e\x64\x65\x66\x61ult_region\x18\x03 \x01(\tR\rdefaultRegion\x12\x1b\n\tapi_token\x18\x04 \x01(\tR\x08\x61piToken\x12U\n\x0cintegrations\x18\x05 \x03(\x0b\x32\x31.corvic.config.v1.RocksetConfig.IntegrationsEntryR\x0cintegrations\x12R\n\x0b\x63ollections\x18\x06 \x03(\x0b\x32\x30.corvic.config.v1.RocksetConfig.CollectionsEntryR\x0b\x63ollections\x12\"\n\x0corganization\x18\x07 \x01(\tR\x0corganization\x1av\n\x11IntegrationConfig\x12\x34\n\x16rockset_integration_id\x18\x01 \x01(\tR\x14rocksetIntegrationId\x12+\n\x11supported_buckets\x18\x02 \x03(\tR\x10supportedBuckets\x1ar\n\x11IntegrationsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12G\n\x05value\x18\x02 \x01(\x0b\x32\x31.corvic.config.v1.RocksetConfig.IntegrationConfigR\x05value:\x02\x38\x01\x1a\xb4\x01\n\x0e\x43ollectionArgs\x12)\n\x10integration_name\x18\x01 \x01(\tR\x0fintegrationName\x12\x1f\n\x0b\x62ucket_name\x18\x02 \x01(\tR\nbucketName\x12*\n\x11gcs_output_prefix\x18\x03 \x01(\tR\x0fgcsOutputPrefix\x12*\n\x11\x63reate_if_missing\x18\x04 \x01(\x08R\x0f\x63reateIfMissing\x1a\xeb\x01\n\x10\x43ollectionConfig\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x1c\n\tworkspace\x18\x02 \x01(\tR\tworkspace\x12\x12\n\x04host\x18\x03 \x01(\tR\x04host\x12\x16\n\x04kind\x18\x04 \x01(\tB\x02\x18\x01R\x04kind\x12 \n\x0b\x64\x65scription\x18\x05 \x01(\tR\x0b\x64\x65scription\x12W\n\x0f\x63ollection_args\x18\x06 \x01(\x0b\x32..corvic.config.v1.RocksetConfig.CollectionArgsR\x0e\x63ollectionArgs\x1ap\n\x10\x43ollectionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x46\n\x05value\x18\x02 \x01(\x0b\x32\x30.corvic.config.v1.RocksetConfig.CollectionConfigR\x05value:\x02\x38\x01\"\x89\x03\n\x0e\x45mbedgenConfig\x12\x1f\n\x0b\x64\x65\x66\x61ult_tag\x18\x01 \x01(\tR\ndefaultTag\x12\x36\n\x17rockset_collection_name\x18\x02 \x01(\tR\x15rocksetCollectionName\x12\x31\n\x15\x63hunk_text_field_name\x18\x03 \x01(\tR\x12\x63hunkTextFieldName\x12\\\n,gcs_notification_config_embed_gen_config_key\x18\x04 \x01(\tR&gcsNotificationConfigEmbedGenConfigKey\x12]\n\x12\x64ocument_processor\x18\x05 \x01(\x0b\x32..corvic.config.v1.EmbedgenConfig.DocProcConfigR\x11\x64ocumentProcessor\x1a.\n\rDocProcConfig\x12\x1d\n\nbatch_size\x18\x01 \x01(\rR\tbatchSize\"\xfe\x02\n\nChatConfig\x12\x31\n\x15max_messages_per_page\x18\x01 \x01(\rR\x12maxMessagesPerPage\x12=\n\x1bget_messages_period_seconds\x18\x02 \x01(\x01R\x18getMessagesPeriodSeconds\x12)\n\x10\x63ompletion_model\x18\x03 \x01(\tR\x0f\x63ompletionModel\x12%\n\x0eretriever_type\x18\x04 \x01(\tR\rretrieverType\x12)\n\x10obfuscation_seed\x18\x05 \x01(\tR\x0fobfuscationSeed\x12\x41\n\x06\x63lient\x18\x06 \x01(\x0b\x32).corvic.config.v1.ChatConfig.ClientConfigR\x06\x63lient\x1a>\n\x0c\x43lientConfig\x12\x1a\n\x08hostname\x18\x01 \x01(\tR\x08hostname\x12\x12\n\x04port\x18\x02 \x01(\rR\x04port\"2\n\x08\x44\x42\x43onfig\x12\x12\n\x04host\x18\x01 \x01(\tR\x04host\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\"\xfa\x02\n\x0cIngestConfig\x12\x35\n\x15store_id_metadata_key\x18\x01 \x01(\tB\x02\x18\x01R\x12storeIdMetadataKey\x12\x43\n\x06server\x18\x02 \x01(\x0b\x32+.corvic.config.v1.IngestConfig.ServerConfigR\x06server\x12\x43\n\x06\x63lient\x18\x03 \x01(\x0b\x32+.corvic.config.v1.IngestConfig.ClientConfigR\x06\x63lient\x1ai\n\x0cServerConfig\x12\x16\n\x06\x62ucket\x18\x01 \x01(\tR\x06\x62ucket\x12\x16\n\x06prefix\x18\x02 \x01(\tR\x06prefix\x12)\n\x10obfuscation_seed\x18\x03 \x01(\tR\x0fobfuscationSeed\x1a>\n\x0c\x43lientConfig\x12\x1a\n\x08hostname\x18\x01 \x01(\tR\x08hostname\x12\x12\n\x04port\x18\x02 \x01(\rR\x04port\"8\n\x10\x45xperimentConfig\x12$\n\x0egcs_run_prefix\x18\x01 \x01(\tR\x0cgcsRunPrefix\"c\n\x15\x43ompletionModelConfig\x12\x16\n\x06\x65ngine\x18\x01 \x01(\tR\x06\x65ngine\x12\x14\n\x05model\x18\x02 \x01(\tR\x05model\x12\x1c\n\ttokenizer\x18\x03 \x01(\tR\ttokenizer\"\xd7\x01\n\x14\x45mbeddingModelConfig\x12\x16\n\x06\x65ngine\x18\x01 \x01(\tR\x06\x65ngine\x12\x14\n\x05model\x18\x02 \x01(\tR\x05model\x12 \n\x0b\x63\x61rdinality\x18\x03 \x01(\x04R\x0b\x63\x61rdinality\x12H\n\x05\x65xtra\x18\x04 \x01(\x0b\x32\x32.corvic.config.v1.EmbeddingModelConfig.ExtraConfigR\x05\x65xtra\x1a%\n\x0b\x45xtraConfig\x12\x16\n\x06\x64\x65vice\x18\x01 \x01(\tR\x06\x64\x65vice\"\xd3\x08\n\x11HuggingfaceConfig\x12*\n\x0fmodel_proxy_url\x18\x01 \x01(\tB\x02\x18\x01R\rmodelProxyUrl\x12:\n\x19\x64\x65\x66\x61ult_inference_timeout\x18\x03 \x01(\rR\x17\x64\x65\x66\x61ultInferenceTimeout\x12\x33\n\x16\x64\x65\x66\x61ult_max_new_tokens\x18\x04 \x01(\x04R\x13\x64\x65\x66\x61ultMaxNewTokens\x12\x1b\n\tapi_token\x18\x05 \x01(\tR\x08\x61piToken\x12m\n\x13inference_endpoints\x18\x06 \x01(\x0b\x32<.corvic.config.v1.HuggingfaceConfig.InferenceEndpointsConfigR\x12inferenceEndpoints\x1a\x94\x06\n\x18InferenceEndpointsConfig\x12o\n\x0b\x63ompletions\x18\x01 \x03(\x0b\x32M.corvic.config.v1.HuggingfaceConfig.InferenceEndpointsConfig.CompletionsEntryR\x0b\x63ompletions\x12l\n\nembeddings\x18\x02 \x03(\x0b\x32L.corvic.config.v1.HuggingfaceConfig.InferenceEndpointsConfig.EmbeddingsEntryR\nembeddings\x1a~\n\x0f\x43ompletionModel\x12\x17\n\x07hf_host\x18\x01 \x01(\tR\x06hfHost\x12\x17\n\x07hf_name\x18\x02 \x01(\tR\x06hfName\x12!\n\x0chf_namespace\x18\x03 \x01(\tR\x0bhfNamespace\x12\x16\n\x06models\x18\x04 \x03(\tR\x06models\x1a\x8c\x01\n\x10\x43ompletionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x62\n\x05value\x18\x02 \x01(\x0b\x32L.corvic.config.v1.HuggingfaceConfig.InferenceEndpointsConfig.CompletionModelR\x05value:\x02\x38\x01\x1a}\n\x0e\x45mbeddingModel\x12\x17\n\x07hf_host\x18\x01 \x01(\tR\x06hfHost\x12\x17\n\x07hf_name\x18\x02 \x01(\tR\x06hfName\x12!\n\x0chf_namespace\x18\x03 \x01(\tR\x0bhfNamespace\x12\x16\n\x06models\x18\x04 \x03(\tR\x06models\x1a\x8a\x01\n\x0f\x45mbeddingsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x61\n\x05value\x18\x02 \x01(\x0b\x32K.corvic.config.v1.HuggingfaceConfig.InferenceEndpointsConfig.EmbeddingModelR\x05value:\x02\x38\x01\"\xa7\x03\n\x0fRetrieverConfig\x12@\n\x1c\x64\x65\x66\x61ult_similarity_threshold\x18\x01 \x01(\x01R\x1a\x64\x65\x66\x61ultSimilarityThreshold\x12,\n\x12\x64\x65\x66\x61ult_max_chunks\x18\x02 \x01(\x04R\x10\x64\x65\x66\x61ultMaxChunks\x12\x1f\n\x0b\x64\x65\x66\x61ult_tag\x18\x03 \x01(\tR\ndefaultTag\x12\x38\n\x18\x64\x65\x66\x61ult_embeddings_model\x18\x04 \x01(\tR\x16\x64\x65\x66\x61ultEmbeddingsModel\x12\x41\n\x1d\x64\x65\x66\x61ult_embeddings_model_name\x18\x05 \x01(\tR\x1a\x64\x65\x66\x61ultEmbeddingsModelName\x12:\n\x19\x64\x65\x66\x61ult_embeddings_engine\x18\x06 \x01(\tR\x17\x64\x65\x66\x61ultEmbeddingsEngine\x12\x12\n\x04type\x18\x07 \x01(\tR\x04type\x12\x36\n\x17rockset_collection_name\x18\x08 \x01(\tR\x15rocksetCollectionName\"\x9c\x01\n\x10\x43ompletionConfig\x12\x38\n\x18\x64\x65\x66\x61ult_completion_model\x18\x01 \x01(\tR\x16\x64\x65\x66\x61ultCompletionModel\x12(\n\x10no_chunk_message\x18\x02 \x01(\tR\x0enoChunkMessage\x12$\n\x0esys_prompt_rag\x18\x03 \x01(\tR\x0csysPromptRag\"7\n\x0eWorkflowConfig\x12%\n\x0estorage_prefix\x18\x01 \x01(\tR\rstoragePrefix\"\xb7\x03\n\x11\x44\x61taStagingConfig\x12,\n\x12rockset_api_server\x18\x01 \x01(\tR\x10rocksetApiServer\x12+\n\x11rockset_workspace\x18\x02 \x01(\tR\x10rocksetWorkspace\x12-\n\x12rockset_collection\x18\x03 \x01(\tR\x11rocksetCollection\x12/\n\x13rockset_integration\x18\x04 \x01(\tR\x12rocksetIntegration\x12\x1d\n\ngcs_bucket\x18\x05 \x01(\tR\tgcsBucket\x12?\n\x1cgcs_unstructured_data_prefix\x18\x06 \x01(\tR\x19gcsUnstructuredDataPrefix\x12\x31\n\x15gcs_table_data_prefix\x18\x07 \x01(\tR\x12gcsTableDataPrefix\x12(\n\x10\x62lob_name_column\x18\x08 \x01(\tR\x0e\x62lobNameColumn\x12*\n\x11row_number_column\x18\t \x01(\tR\x0frowNumberColumn\"a\n\x05\x41uth0\x12\x1b\n\tclient_id\x18\x01 \x01(\tR\x08\x63lientId\x12#\n\rclient_secret\x18\x02 \x01(\tR\x0c\x63lientSecret\x12\x16\n\x06\x64omain\x18\x03 \x01(\tR\x06\x64omain\"\xf1\x0b\n\x08Settings\x12#\n\x0bload_dotenv\x18\x01 \x01(\x08\x42\x02\x18\x01R\nloadDotenv\x12\x34\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x1c.corvic.config.v1.MetaConfigR\x06\x63onfig\x12\x19\n\x08\x64\x65v_mode\x18\x03 \x01(\tR\x07\x64\x65vMode\x12\x39\n\x07metrics\x18\x04 \x01(\x0b\x32\x1f.corvic.config.v1.MetricsConfigR\x07metrics\x12G\n\x0b\x63hat_server\x18\x05 \x01(\x0b\x32\".corvic.config.v1.ChatServerConfigB\x02\x18\x01R\nchatServer\x12@\n\x08\x65mbedgen\x18\x06 \x01(\x0b\x32 .corvic.config.v1.EmbedgenConfigB\x02\x18\x01R\x08\x65mbedgen\x12\x34\n\x04\x63hat\x18\x07 \x01(\x0b\x32\x1c.corvic.config.v1.ChatConfigB\x02\x18\x01R\x04\x63hat\x12*\n\x02\x64\x62\x18\x08 \x01(\x0b\x32\x1a.corvic.config.v1.DBConfigR\x02\x64\x62\x12\x36\n\x06ingest\x18\t \x01(\x0b\x32\x1e.corvic.config.v1.IngestConfigR\x06ingest\x12\x61\n\x11\x63ompletion_models\x18\n \x03(\x0b\x32\x30.corvic.config.v1.Settings.CompletionModelsEntryB\x02\x18\x01R\x10\x63ompletionModels\x12Z\n\x10\x65mbedding_models\x18\x0b \x03(\x0b\x32/.corvic.config.v1.Settings.EmbeddingModelsEntryR\x0f\x65mbeddingModels\x12I\n\x0bhuggingface\x18\x0c \x01(\x0b\x32#.corvic.config.v1.HuggingfaceConfigB\x02\x18\x01R\x0bhuggingface\x12\x43\n\tretriever\x18\r \x01(\x0b\x32!.corvic.config.v1.RetrieverConfigB\x02\x18\x01R\tretriever\x12\x46\n\ncompletion\x18\x0e \x01(\x0b\x32\".corvic.config.v1.CompletionConfigB\x02\x18\x01R\ncompletion\x12\x39\n\x07rockset\x18\x0f \x01(\x0b\x32\x1f.corvic.config.v1.RocksetConfigR\x07rockset\x12\x32\n\x15google_identity_token\x18\x10 \x01(\tR\x13googleIdentityToken\x12\x1d\n\ncheck_name\x18\x11 \x01(\tR\tcheckName\x12\x17\n\x07test_db\x18\x12 \x01(\tR\x06testDb\x12<\n\x08workflow\x18\x13 \x01(\x0b\x32 .corvic.config.v1.WorkflowConfigR\x08workflow\x12\x46\n\x0c\x64\x61ta_staging\x18\x14 \x01(\x0b\x32#.corvic.config.v1.DataStagingConfigR\x0b\x64\x61taStaging\x12\x42\n\nexperiment\x18\x15 \x01(\x0b\x32\".corvic.config.v1.ExperimentConfigR\nexperiment\x12-\n\x05\x61uth0\x18\x16 \x01(\x0b\x32\x17.corvic.config.v1.Auth0R\x05\x61uth0\x1al\n\x15\x43ompletionModelsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12=\n\x05value\x18\x02 \x01(\x0b\x32\'.corvic.config.v1.CompletionModelConfigR\x05value:\x02\x38\x01\x1aj\n\x14\x45mbeddingModelsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12<\n\x05value\x18\x02 \x01(\x0b\x32&.corvic.config.v1.EmbeddingModelConfigR\x05value:\x02\x38\x01\x62\x06proto3')
|
18
|
-
|
19
|
-
_globals = globals()
|
20
|
-
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
|
21
|
-
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.config.v1.settings_pb2', _globals)
|
22
|
-
if _descriptor._USE_C_DESCRIPTORS == False:
|
23
|
-
DESCRIPTOR._options = None
|
24
|
-
_globals['_ROCKSETCONFIG_INTEGRATIONSENTRY']._options = None
|
25
|
-
_globals['_ROCKSETCONFIG_INTEGRATIONSENTRY']._serialized_options = b'8\001'
|
26
|
-
_globals['_ROCKSETCONFIG_COLLECTIONCONFIG'].fields_by_name['kind']._options = None
|
27
|
-
_globals['_ROCKSETCONFIG_COLLECTIONCONFIG'].fields_by_name['kind']._serialized_options = b'\030\001'
|
28
|
-
_globals['_ROCKSETCONFIG_COLLECTIONSENTRY']._options = None
|
29
|
-
_globals['_ROCKSETCONFIG_COLLECTIONSENTRY']._serialized_options = b'8\001'
|
30
|
-
_globals['_INGESTCONFIG'].fields_by_name['store_id_metadata_key']._options = None
|
31
|
-
_globals['_INGESTCONFIG'].fields_by_name['store_id_metadata_key']._serialized_options = b'\030\001'
|
32
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_COMPLETIONSENTRY']._options = None
|
33
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_COMPLETIONSENTRY']._serialized_options = b'8\001'
|
34
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_EMBEDDINGSENTRY']._options = None
|
35
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_EMBEDDINGSENTRY']._serialized_options = b'8\001'
|
36
|
-
_globals['_HUGGINGFACECONFIG'].fields_by_name['model_proxy_url']._options = None
|
37
|
-
_globals['_HUGGINGFACECONFIG'].fields_by_name['model_proxy_url']._serialized_options = b'\030\001'
|
38
|
-
_globals['_SETTINGS_COMPLETIONMODELSENTRY']._options = None
|
39
|
-
_globals['_SETTINGS_COMPLETIONMODELSENTRY']._serialized_options = b'8\001'
|
40
|
-
_globals['_SETTINGS_EMBEDDINGMODELSENTRY']._options = None
|
41
|
-
_globals['_SETTINGS_EMBEDDINGMODELSENTRY']._serialized_options = b'8\001'
|
42
|
-
_globals['_SETTINGS'].fields_by_name['load_dotenv']._options = None
|
43
|
-
_globals['_SETTINGS'].fields_by_name['load_dotenv']._serialized_options = b'\030\001'
|
44
|
-
_globals['_SETTINGS'].fields_by_name['chat_server']._options = None
|
45
|
-
_globals['_SETTINGS'].fields_by_name['chat_server']._serialized_options = b'\030\001'
|
46
|
-
_globals['_SETTINGS'].fields_by_name['embedgen']._options = None
|
47
|
-
_globals['_SETTINGS'].fields_by_name['embedgen']._serialized_options = b'\030\001'
|
48
|
-
_globals['_SETTINGS'].fields_by_name['chat']._options = None
|
49
|
-
_globals['_SETTINGS'].fields_by_name['chat']._serialized_options = b'\030\001'
|
50
|
-
_globals['_SETTINGS'].fields_by_name['completion_models']._options = None
|
51
|
-
_globals['_SETTINGS'].fields_by_name['completion_models']._serialized_options = b'\030\001'
|
52
|
-
_globals['_SETTINGS'].fields_by_name['huggingface']._options = None
|
53
|
-
_globals['_SETTINGS'].fields_by_name['huggingface']._serialized_options = b'\030\001'
|
54
|
-
_globals['_SETTINGS'].fields_by_name['retriever']._options = None
|
55
|
-
_globals['_SETTINGS'].fields_by_name['retriever']._serialized_options = b'\030\001'
|
56
|
-
_globals['_SETTINGS'].fields_by_name['completion']._options = None
|
57
|
-
_globals['_SETTINGS'].fields_by_name['completion']._serialized_options = b'\030\001'
|
58
|
-
_globals['_METACONFIG']._serialized_start=53
|
59
|
-
_globals['_METACONFIG']._serialized_end=180
|
60
|
-
_globals['_METRICSCONFIG']._serialized_start=182
|
61
|
-
_globals['_METRICSCONFIG']._serialized_end=221
|
62
|
-
_globals['_CHATSERVERCONFIG']._serialized_start=223
|
63
|
-
_globals['_CHATSERVERCONFIG']._serialized_end=261
|
64
|
-
_globals['_ROCKSETCONFIG']._serialized_start=264
|
65
|
-
_globals['_ROCKSETCONFIG']._serialized_end=1417
|
66
|
-
_globals['_ROCKSETCONFIG_INTEGRATIONCONFIG']._serialized_start=648
|
67
|
-
_globals['_ROCKSETCONFIG_INTEGRATIONCONFIG']._serialized_end=766
|
68
|
-
_globals['_ROCKSETCONFIG_INTEGRATIONSENTRY']._serialized_start=768
|
69
|
-
_globals['_ROCKSETCONFIG_INTEGRATIONSENTRY']._serialized_end=882
|
70
|
-
_globals['_ROCKSETCONFIG_COLLECTIONARGS']._serialized_start=885
|
71
|
-
_globals['_ROCKSETCONFIG_COLLECTIONARGS']._serialized_end=1065
|
72
|
-
_globals['_ROCKSETCONFIG_COLLECTIONCONFIG']._serialized_start=1068
|
73
|
-
_globals['_ROCKSETCONFIG_COLLECTIONCONFIG']._serialized_end=1303
|
74
|
-
_globals['_ROCKSETCONFIG_COLLECTIONSENTRY']._serialized_start=1305
|
75
|
-
_globals['_ROCKSETCONFIG_COLLECTIONSENTRY']._serialized_end=1417
|
76
|
-
_globals['_EMBEDGENCONFIG']._serialized_start=1420
|
77
|
-
_globals['_EMBEDGENCONFIG']._serialized_end=1813
|
78
|
-
_globals['_EMBEDGENCONFIG_DOCPROCCONFIG']._serialized_start=1767
|
79
|
-
_globals['_EMBEDGENCONFIG_DOCPROCCONFIG']._serialized_end=1813
|
80
|
-
_globals['_CHATCONFIG']._serialized_start=1816
|
81
|
-
_globals['_CHATCONFIG']._serialized_end=2198
|
82
|
-
_globals['_CHATCONFIG_CLIENTCONFIG']._serialized_start=2136
|
83
|
-
_globals['_CHATCONFIG_CLIENTCONFIG']._serialized_end=2198
|
84
|
-
_globals['_DBCONFIG']._serialized_start=2200
|
85
|
-
_globals['_DBCONFIG']._serialized_end=2250
|
86
|
-
_globals['_INGESTCONFIG']._serialized_start=2253
|
87
|
-
_globals['_INGESTCONFIG']._serialized_end=2631
|
88
|
-
_globals['_INGESTCONFIG_SERVERCONFIG']._serialized_start=2462
|
89
|
-
_globals['_INGESTCONFIG_SERVERCONFIG']._serialized_end=2567
|
90
|
-
_globals['_INGESTCONFIG_CLIENTCONFIG']._serialized_start=2136
|
91
|
-
_globals['_INGESTCONFIG_CLIENTCONFIG']._serialized_end=2198
|
92
|
-
_globals['_EXPERIMENTCONFIG']._serialized_start=2633
|
93
|
-
_globals['_EXPERIMENTCONFIG']._serialized_end=2689
|
94
|
-
_globals['_COMPLETIONMODELCONFIG']._serialized_start=2691
|
95
|
-
_globals['_COMPLETIONMODELCONFIG']._serialized_end=2790
|
96
|
-
_globals['_EMBEDDINGMODELCONFIG']._serialized_start=2793
|
97
|
-
_globals['_EMBEDDINGMODELCONFIG']._serialized_end=3008
|
98
|
-
_globals['_EMBEDDINGMODELCONFIG_EXTRACONFIG']._serialized_start=2971
|
99
|
-
_globals['_EMBEDDINGMODELCONFIG_EXTRACONFIG']._serialized_end=3008
|
100
|
-
_globals['_HUGGINGFACECONFIG']._serialized_start=3011
|
101
|
-
_globals['_HUGGINGFACECONFIG']._serialized_end=4118
|
102
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG']._serialized_start=3330
|
103
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG']._serialized_end=4118
|
104
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_COMPLETIONMODEL']._serialized_start=3581
|
105
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_COMPLETIONMODEL']._serialized_end=3707
|
106
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_COMPLETIONSENTRY']._serialized_start=3710
|
107
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_COMPLETIONSENTRY']._serialized_end=3850
|
108
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_EMBEDDINGMODEL']._serialized_start=3852
|
109
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_EMBEDDINGMODEL']._serialized_end=3977
|
110
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_EMBEDDINGSENTRY']._serialized_start=3980
|
111
|
-
_globals['_HUGGINGFACECONFIG_INFERENCEENDPOINTSCONFIG_EMBEDDINGSENTRY']._serialized_end=4118
|
112
|
-
_globals['_RETRIEVERCONFIG']._serialized_start=4121
|
113
|
-
_globals['_RETRIEVERCONFIG']._serialized_end=4544
|
114
|
-
_globals['_COMPLETIONCONFIG']._serialized_start=4547
|
115
|
-
_globals['_COMPLETIONCONFIG']._serialized_end=4703
|
116
|
-
_globals['_WORKFLOWCONFIG']._serialized_start=4705
|
117
|
-
_globals['_WORKFLOWCONFIG']._serialized_end=4760
|
118
|
-
_globals['_DATASTAGINGCONFIG']._serialized_start=4763
|
119
|
-
_globals['_DATASTAGINGCONFIG']._serialized_end=5202
|
120
|
-
_globals['_AUTH0']._serialized_start=5204
|
121
|
-
_globals['_AUTH0']._serialized_end=5301
|
122
|
-
_globals['_SETTINGS']._serialized_start=5304
|
123
|
-
_globals['_SETTINGS']._serialized_end=6825
|
124
|
-
_globals['_SETTINGS_COMPLETIONMODELSENTRY']._serialized_start=6609
|
125
|
-
_globals['_SETTINGS_COMPLETIONMODELSENTRY']._serialized_end=6717
|
126
|
-
_globals['_SETTINGS_EMBEDDINGMODELSENTRY']._serialized_start=6719
|
127
|
-
_globals['_SETTINGS_EMBEDDINGMODELSENTRY']._serialized_end=6825
|
128
|
-
# @@protoc_insertion_point(module_scope)
|