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.
Files changed (53) hide show
  1. corvic/engine/_native.pyd +0 -0
  2. corvic/model/_completion_model.py +15 -0
  3. corvic/model/_proto_orm_convert.py +4 -0
  4. corvic/model/_source.py +8 -0
  5. corvic/op_graph/_transformations.py +3 -3
  6. corvic/orm/__init__.py +5 -0
  7. corvic/orm/ids.py +6 -1
  8. {corvic_engine-0.3.0rc57.dist-info → corvic_engine-0.3.0rc59.dist-info}/METADATA +1 -1
  9. {corvic_engine-0.3.0rc57.dist-info → corvic_engine-0.3.0rc59.dist-info}/RECORD +23 -53
  10. corvic_generated/feature/v2/feature_view_pb2.py +39 -47
  11. corvic_generated/feature/v2/feature_view_pb2.pyi +0 -14
  12. corvic_generated/feature/v2/space_pb2.py +77 -87
  13. corvic_generated/ingest/v2/pipeline_pb2.py +33 -27
  14. corvic_generated/ingest/v2/quick_mode_pb2.py +18 -13
  15. corvic_generated/ingest/v2/quick_mode_pb2.pyi +1 -0
  16. corvic_generated/ingest/v2/resource_pb2.py +64 -58
  17. corvic_generated/ingest/v2/room_pb2.py +32 -32
  18. corvic_generated/ingest/v2/source_pb2.py +43 -43
  19. corvic_generated/ingest/v2/table_pb2.py +23 -23
  20. corvic_generated/model/v1alpha/models_pb2.py +2 -2
  21. corvic_generated/model/v1alpha/models_pb2.pyi +6 -4
  22. corvic_generated/chat/__init__.py +0 -0
  23. corvic_generated/chat/v1/__init__.py +0 -0
  24. corvic_generated/chat/v1/service_pb2.py +0 -65
  25. corvic_generated/chat/v1/service_pb2.pyi +0 -132
  26. corvic_generated/chat/v1/service_pb2_grpc.py +0 -198
  27. corvic_generated/chat/v1/service_pb2_grpc.pyi +0 -97
  28. corvic_generated/completion/__init__.py +0 -0
  29. corvic_generated/completion/v1/__init__.py +0 -0
  30. corvic_generated/completion/v1/service_pb2.py +0 -69
  31. corvic_generated/completion/v1/service_pb2.pyi +0 -201
  32. corvic_generated/completion/v1/service_pb2_grpc.py +0 -198
  33. corvic_generated/completion/v1/service_pb2_grpc.pyi +0 -97
  34. corvic_generated/config/__init__.py +0 -0
  35. corvic_generated/config/v1/__init__.py +0 -0
  36. corvic_generated/config/v1/settings_pb2.py +0 -128
  37. corvic_generated/config/v1/settings_pb2.pyi +0 -389
  38. corvic_generated/config/v1/settings_pb2_grpc.py +0 -4
  39. corvic_generated/config/v1/settings_pb2_grpc.pyi +0 -17
  40. corvic_generated/platform/__init__.py +0 -0
  41. corvic_generated/platform/v1/__init__.py +0 -0
  42. corvic_generated/platform/v1/platform_pb2.py +0 -101
  43. corvic_generated/platform/v1/platform_pb2.pyi +0 -147
  44. corvic_generated/platform/v1/platform_pb2_grpc.py +0 -330
  45. corvic_generated/platform/v1/platform_pb2_grpc.pyi +0 -153
  46. corvic_generated/query/__init__.py +0 -0
  47. corvic_generated/query/v1/__init__.py +0 -0
  48. corvic_generated/query/v1/query_pb2.py +0 -36
  49. corvic_generated/query/v1/query_pb2.pyi +0 -63
  50. corvic_generated/query/v1/query_pb2_grpc.py +0 -66
  51. corvic_generated/query/v1/query_pb2_grpc.pyi +0 -41
  52. {corvic_engine-0.3.0rc57.dist-info → corvic_engine-0.3.0rc59.dist-info}/WHEEL +0 -0
  53. {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)