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,389 +0,0 @@
1
- from google.protobuf.internal import containers as _containers
2
- from google.protobuf import descriptor as _descriptor
3
- from google.protobuf import message as _message
4
- from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
5
-
6
- DESCRIPTOR: _descriptor.FileDescriptor
7
-
8
- class MetaConfig(_message.Message):
9
- __slots__ = ("envvar_prefix", "nested_separator", "via_launcher")
10
- ENVVAR_PREFIX_FIELD_NUMBER: _ClassVar[int]
11
- NESTED_SEPARATOR_FIELD_NUMBER: _ClassVar[int]
12
- VIA_LAUNCHER_FIELD_NUMBER: _ClassVar[int]
13
- envvar_prefix: str
14
- nested_separator: str
15
- via_launcher: bool
16
- def __init__(self, envvar_prefix: _Optional[str] = ..., nested_separator: _Optional[str] = ..., via_launcher: bool = ...) -> None: ...
17
-
18
- class MetricsConfig(_message.Message):
19
- __slots__ = ("export",)
20
- EXPORT_FIELD_NUMBER: _ClassVar[int]
21
- export: bool
22
- def __init__(self, export: bool = ...) -> None: ...
23
-
24
- class ChatServerConfig(_message.Message):
25
- __slots__ = ("port",)
26
- PORT_FIELD_NUMBER: _ClassVar[int]
27
- port: int
28
- def __init__(self, port: _Optional[int] = ...) -> None: ...
29
-
30
- class RocksetConfig(_message.Message):
31
- __slots__ = ("default_workspace", "default_collection", "default_region", "api_token", "integrations", "collections", "organization")
32
- class IntegrationConfig(_message.Message):
33
- __slots__ = ("rockset_integration_id", "supported_buckets")
34
- ROCKSET_INTEGRATION_ID_FIELD_NUMBER: _ClassVar[int]
35
- SUPPORTED_BUCKETS_FIELD_NUMBER: _ClassVar[int]
36
- rockset_integration_id: str
37
- supported_buckets: _containers.RepeatedScalarFieldContainer[str]
38
- def __init__(self, rockset_integration_id: _Optional[str] = ..., supported_buckets: _Optional[_Iterable[str]] = ...) -> None: ...
39
- class IntegrationsEntry(_message.Message):
40
- __slots__ = ("key", "value")
41
- KEY_FIELD_NUMBER: _ClassVar[int]
42
- VALUE_FIELD_NUMBER: _ClassVar[int]
43
- key: str
44
- value: RocksetConfig.IntegrationConfig
45
- def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[RocksetConfig.IntegrationConfig, _Mapping]] = ...) -> None: ...
46
- class CollectionArgs(_message.Message):
47
- __slots__ = ("integration_name", "bucket_name", "gcs_output_prefix", "create_if_missing")
48
- INTEGRATION_NAME_FIELD_NUMBER: _ClassVar[int]
49
- BUCKET_NAME_FIELD_NUMBER: _ClassVar[int]
50
- GCS_OUTPUT_PREFIX_FIELD_NUMBER: _ClassVar[int]
51
- CREATE_IF_MISSING_FIELD_NUMBER: _ClassVar[int]
52
- integration_name: str
53
- bucket_name: str
54
- gcs_output_prefix: str
55
- create_if_missing: bool
56
- def __init__(self, integration_name: _Optional[str] = ..., bucket_name: _Optional[str] = ..., gcs_output_prefix: _Optional[str] = ..., create_if_missing: bool = ...) -> None: ...
57
- class CollectionConfig(_message.Message):
58
- __slots__ = ("name", "workspace", "host", "kind", "description", "collection_args")
59
- NAME_FIELD_NUMBER: _ClassVar[int]
60
- WORKSPACE_FIELD_NUMBER: _ClassVar[int]
61
- HOST_FIELD_NUMBER: _ClassVar[int]
62
- KIND_FIELD_NUMBER: _ClassVar[int]
63
- DESCRIPTION_FIELD_NUMBER: _ClassVar[int]
64
- COLLECTION_ARGS_FIELD_NUMBER: _ClassVar[int]
65
- name: str
66
- workspace: str
67
- host: str
68
- kind: str
69
- description: str
70
- collection_args: RocksetConfig.CollectionArgs
71
- def __init__(self, name: _Optional[str] = ..., workspace: _Optional[str] = ..., host: _Optional[str] = ..., kind: _Optional[str] = ..., description: _Optional[str] = ..., collection_args: _Optional[_Union[RocksetConfig.CollectionArgs, _Mapping]] = ...) -> None: ...
72
- class CollectionsEntry(_message.Message):
73
- __slots__ = ("key", "value")
74
- KEY_FIELD_NUMBER: _ClassVar[int]
75
- VALUE_FIELD_NUMBER: _ClassVar[int]
76
- key: str
77
- value: RocksetConfig.CollectionConfig
78
- def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[RocksetConfig.CollectionConfig, _Mapping]] = ...) -> None: ...
79
- DEFAULT_WORKSPACE_FIELD_NUMBER: _ClassVar[int]
80
- DEFAULT_COLLECTION_FIELD_NUMBER: _ClassVar[int]
81
- DEFAULT_REGION_FIELD_NUMBER: _ClassVar[int]
82
- API_TOKEN_FIELD_NUMBER: _ClassVar[int]
83
- INTEGRATIONS_FIELD_NUMBER: _ClassVar[int]
84
- COLLECTIONS_FIELD_NUMBER: _ClassVar[int]
85
- ORGANIZATION_FIELD_NUMBER: _ClassVar[int]
86
- default_workspace: str
87
- default_collection: str
88
- default_region: str
89
- api_token: str
90
- integrations: _containers.MessageMap[str, RocksetConfig.IntegrationConfig]
91
- collections: _containers.MessageMap[str, RocksetConfig.CollectionConfig]
92
- organization: str
93
- def __init__(self, default_workspace: _Optional[str] = ..., default_collection: _Optional[str] = ..., default_region: _Optional[str] = ..., api_token: _Optional[str] = ..., integrations: _Optional[_Mapping[str, RocksetConfig.IntegrationConfig]] = ..., collections: _Optional[_Mapping[str, RocksetConfig.CollectionConfig]] = ..., organization: _Optional[str] = ...) -> None: ...
94
-
95
- class EmbedgenConfig(_message.Message):
96
- __slots__ = ("default_tag", "rockset_collection_name", "chunk_text_field_name", "gcs_notification_config_embed_gen_config_key", "document_processor")
97
- class DocProcConfig(_message.Message):
98
- __slots__ = ("batch_size",)
99
- BATCH_SIZE_FIELD_NUMBER: _ClassVar[int]
100
- batch_size: int
101
- def __init__(self, batch_size: _Optional[int] = ...) -> None: ...
102
- DEFAULT_TAG_FIELD_NUMBER: _ClassVar[int]
103
- ROCKSET_COLLECTION_NAME_FIELD_NUMBER: _ClassVar[int]
104
- CHUNK_TEXT_FIELD_NAME_FIELD_NUMBER: _ClassVar[int]
105
- GCS_NOTIFICATION_CONFIG_EMBED_GEN_CONFIG_KEY_FIELD_NUMBER: _ClassVar[int]
106
- DOCUMENT_PROCESSOR_FIELD_NUMBER: _ClassVar[int]
107
- default_tag: str
108
- rockset_collection_name: str
109
- chunk_text_field_name: str
110
- gcs_notification_config_embed_gen_config_key: str
111
- document_processor: EmbedgenConfig.DocProcConfig
112
- def __init__(self, default_tag: _Optional[str] = ..., rockset_collection_name: _Optional[str] = ..., chunk_text_field_name: _Optional[str] = ..., gcs_notification_config_embed_gen_config_key: _Optional[str] = ..., document_processor: _Optional[_Union[EmbedgenConfig.DocProcConfig, _Mapping]] = ...) -> None: ...
113
-
114
- class ChatConfig(_message.Message):
115
- __slots__ = ("max_messages_per_page", "get_messages_period_seconds", "completion_model", "retriever_type", "obfuscation_seed", "client")
116
- class ClientConfig(_message.Message):
117
- __slots__ = ("hostname", "port")
118
- HOSTNAME_FIELD_NUMBER: _ClassVar[int]
119
- PORT_FIELD_NUMBER: _ClassVar[int]
120
- hostname: str
121
- port: int
122
- def __init__(self, hostname: _Optional[str] = ..., port: _Optional[int] = ...) -> None: ...
123
- MAX_MESSAGES_PER_PAGE_FIELD_NUMBER: _ClassVar[int]
124
- GET_MESSAGES_PERIOD_SECONDS_FIELD_NUMBER: _ClassVar[int]
125
- COMPLETION_MODEL_FIELD_NUMBER: _ClassVar[int]
126
- RETRIEVER_TYPE_FIELD_NUMBER: _ClassVar[int]
127
- OBFUSCATION_SEED_FIELD_NUMBER: _ClassVar[int]
128
- CLIENT_FIELD_NUMBER: _ClassVar[int]
129
- max_messages_per_page: int
130
- get_messages_period_seconds: float
131
- completion_model: str
132
- retriever_type: str
133
- obfuscation_seed: str
134
- client: ChatConfig.ClientConfig
135
- def __init__(self, max_messages_per_page: _Optional[int] = ..., get_messages_period_seconds: _Optional[float] = ..., completion_model: _Optional[str] = ..., retriever_type: _Optional[str] = ..., obfuscation_seed: _Optional[str] = ..., client: _Optional[_Union[ChatConfig.ClientConfig, _Mapping]] = ...) -> None: ...
136
-
137
- class DBConfig(_message.Message):
138
- __slots__ = ("host", "name")
139
- HOST_FIELD_NUMBER: _ClassVar[int]
140
- NAME_FIELD_NUMBER: _ClassVar[int]
141
- host: str
142
- name: str
143
- def __init__(self, host: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
144
-
145
- class IngestConfig(_message.Message):
146
- __slots__ = ("store_id_metadata_key", "server", "client")
147
- class ServerConfig(_message.Message):
148
- __slots__ = ("bucket", "prefix", "obfuscation_seed")
149
- BUCKET_FIELD_NUMBER: _ClassVar[int]
150
- PREFIX_FIELD_NUMBER: _ClassVar[int]
151
- OBFUSCATION_SEED_FIELD_NUMBER: _ClassVar[int]
152
- bucket: str
153
- prefix: str
154
- obfuscation_seed: str
155
- def __init__(self, bucket: _Optional[str] = ..., prefix: _Optional[str] = ..., obfuscation_seed: _Optional[str] = ...) -> None: ...
156
- class ClientConfig(_message.Message):
157
- __slots__ = ("hostname", "port")
158
- HOSTNAME_FIELD_NUMBER: _ClassVar[int]
159
- PORT_FIELD_NUMBER: _ClassVar[int]
160
- hostname: str
161
- port: int
162
- def __init__(self, hostname: _Optional[str] = ..., port: _Optional[int] = ...) -> None: ...
163
- STORE_ID_METADATA_KEY_FIELD_NUMBER: _ClassVar[int]
164
- SERVER_FIELD_NUMBER: _ClassVar[int]
165
- CLIENT_FIELD_NUMBER: _ClassVar[int]
166
- store_id_metadata_key: str
167
- server: IngestConfig.ServerConfig
168
- client: IngestConfig.ClientConfig
169
- def __init__(self, store_id_metadata_key: _Optional[str] = ..., server: _Optional[_Union[IngestConfig.ServerConfig, _Mapping]] = ..., client: _Optional[_Union[IngestConfig.ClientConfig, _Mapping]] = ...) -> None: ...
170
-
171
- class ExperimentConfig(_message.Message):
172
- __slots__ = ("gcs_run_prefix",)
173
- GCS_RUN_PREFIX_FIELD_NUMBER: _ClassVar[int]
174
- gcs_run_prefix: str
175
- def __init__(self, gcs_run_prefix: _Optional[str] = ...) -> None: ...
176
-
177
- class CompletionModelConfig(_message.Message):
178
- __slots__ = ("engine", "model", "tokenizer")
179
- ENGINE_FIELD_NUMBER: _ClassVar[int]
180
- MODEL_FIELD_NUMBER: _ClassVar[int]
181
- TOKENIZER_FIELD_NUMBER: _ClassVar[int]
182
- engine: str
183
- model: str
184
- tokenizer: str
185
- def __init__(self, engine: _Optional[str] = ..., model: _Optional[str] = ..., tokenizer: _Optional[str] = ...) -> None: ...
186
-
187
- class EmbeddingModelConfig(_message.Message):
188
- __slots__ = ("engine", "model", "cardinality", "extra")
189
- class ExtraConfig(_message.Message):
190
- __slots__ = ("device",)
191
- DEVICE_FIELD_NUMBER: _ClassVar[int]
192
- device: str
193
- def __init__(self, device: _Optional[str] = ...) -> None: ...
194
- ENGINE_FIELD_NUMBER: _ClassVar[int]
195
- MODEL_FIELD_NUMBER: _ClassVar[int]
196
- CARDINALITY_FIELD_NUMBER: _ClassVar[int]
197
- EXTRA_FIELD_NUMBER: _ClassVar[int]
198
- engine: str
199
- model: str
200
- cardinality: int
201
- extra: EmbeddingModelConfig.ExtraConfig
202
- def __init__(self, engine: _Optional[str] = ..., model: _Optional[str] = ..., cardinality: _Optional[int] = ..., extra: _Optional[_Union[EmbeddingModelConfig.ExtraConfig, _Mapping]] = ...) -> None: ...
203
-
204
- class HuggingfaceConfig(_message.Message):
205
- __slots__ = ("model_proxy_url", "default_inference_timeout", "default_max_new_tokens", "api_token", "inference_endpoints")
206
- class InferenceEndpointsConfig(_message.Message):
207
- __slots__ = ("completions", "embeddings")
208
- class CompletionModel(_message.Message):
209
- __slots__ = ("hf_host", "hf_name", "hf_namespace", "models")
210
- HF_HOST_FIELD_NUMBER: _ClassVar[int]
211
- HF_NAME_FIELD_NUMBER: _ClassVar[int]
212
- HF_NAMESPACE_FIELD_NUMBER: _ClassVar[int]
213
- MODELS_FIELD_NUMBER: _ClassVar[int]
214
- hf_host: str
215
- hf_name: str
216
- hf_namespace: str
217
- models: _containers.RepeatedScalarFieldContainer[str]
218
- def __init__(self, hf_host: _Optional[str] = ..., hf_name: _Optional[str] = ..., hf_namespace: _Optional[str] = ..., models: _Optional[_Iterable[str]] = ...) -> None: ...
219
- class CompletionsEntry(_message.Message):
220
- __slots__ = ("key", "value")
221
- KEY_FIELD_NUMBER: _ClassVar[int]
222
- VALUE_FIELD_NUMBER: _ClassVar[int]
223
- key: str
224
- value: HuggingfaceConfig.InferenceEndpointsConfig.CompletionModel
225
- def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[HuggingfaceConfig.InferenceEndpointsConfig.CompletionModel, _Mapping]] = ...) -> None: ...
226
- class EmbeddingModel(_message.Message):
227
- __slots__ = ("hf_host", "hf_name", "hf_namespace", "models")
228
- HF_HOST_FIELD_NUMBER: _ClassVar[int]
229
- HF_NAME_FIELD_NUMBER: _ClassVar[int]
230
- HF_NAMESPACE_FIELD_NUMBER: _ClassVar[int]
231
- MODELS_FIELD_NUMBER: _ClassVar[int]
232
- hf_host: str
233
- hf_name: str
234
- hf_namespace: str
235
- models: _containers.RepeatedScalarFieldContainer[str]
236
- def __init__(self, hf_host: _Optional[str] = ..., hf_name: _Optional[str] = ..., hf_namespace: _Optional[str] = ..., models: _Optional[_Iterable[str]] = ...) -> None: ...
237
- class EmbeddingsEntry(_message.Message):
238
- __slots__ = ("key", "value")
239
- KEY_FIELD_NUMBER: _ClassVar[int]
240
- VALUE_FIELD_NUMBER: _ClassVar[int]
241
- key: str
242
- value: HuggingfaceConfig.InferenceEndpointsConfig.EmbeddingModel
243
- def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[HuggingfaceConfig.InferenceEndpointsConfig.EmbeddingModel, _Mapping]] = ...) -> None: ...
244
- COMPLETIONS_FIELD_NUMBER: _ClassVar[int]
245
- EMBEDDINGS_FIELD_NUMBER: _ClassVar[int]
246
- completions: _containers.MessageMap[str, HuggingfaceConfig.InferenceEndpointsConfig.CompletionModel]
247
- embeddings: _containers.MessageMap[str, HuggingfaceConfig.InferenceEndpointsConfig.EmbeddingModel]
248
- def __init__(self, completions: _Optional[_Mapping[str, HuggingfaceConfig.InferenceEndpointsConfig.CompletionModel]] = ..., embeddings: _Optional[_Mapping[str, HuggingfaceConfig.InferenceEndpointsConfig.EmbeddingModel]] = ...) -> None: ...
249
- MODEL_PROXY_URL_FIELD_NUMBER: _ClassVar[int]
250
- DEFAULT_INFERENCE_TIMEOUT_FIELD_NUMBER: _ClassVar[int]
251
- DEFAULT_MAX_NEW_TOKENS_FIELD_NUMBER: _ClassVar[int]
252
- API_TOKEN_FIELD_NUMBER: _ClassVar[int]
253
- INFERENCE_ENDPOINTS_FIELD_NUMBER: _ClassVar[int]
254
- model_proxy_url: str
255
- default_inference_timeout: int
256
- default_max_new_tokens: int
257
- api_token: str
258
- inference_endpoints: HuggingfaceConfig.InferenceEndpointsConfig
259
- def __init__(self, model_proxy_url: _Optional[str] = ..., default_inference_timeout: _Optional[int] = ..., default_max_new_tokens: _Optional[int] = ..., api_token: _Optional[str] = ..., inference_endpoints: _Optional[_Union[HuggingfaceConfig.InferenceEndpointsConfig, _Mapping]] = ...) -> None: ...
260
-
261
- class RetrieverConfig(_message.Message):
262
- __slots__ = ("default_similarity_threshold", "default_max_chunks", "default_tag", "default_embeddings_model", "default_embeddings_model_name", "default_embeddings_engine", "type", "rockset_collection_name")
263
- DEFAULT_SIMILARITY_THRESHOLD_FIELD_NUMBER: _ClassVar[int]
264
- DEFAULT_MAX_CHUNKS_FIELD_NUMBER: _ClassVar[int]
265
- DEFAULT_TAG_FIELD_NUMBER: _ClassVar[int]
266
- DEFAULT_EMBEDDINGS_MODEL_FIELD_NUMBER: _ClassVar[int]
267
- DEFAULT_EMBEDDINGS_MODEL_NAME_FIELD_NUMBER: _ClassVar[int]
268
- DEFAULT_EMBEDDINGS_ENGINE_FIELD_NUMBER: _ClassVar[int]
269
- TYPE_FIELD_NUMBER: _ClassVar[int]
270
- ROCKSET_COLLECTION_NAME_FIELD_NUMBER: _ClassVar[int]
271
- default_similarity_threshold: float
272
- default_max_chunks: int
273
- default_tag: str
274
- default_embeddings_model: str
275
- default_embeddings_model_name: str
276
- default_embeddings_engine: str
277
- type: str
278
- rockset_collection_name: str
279
- def __init__(self, default_similarity_threshold: _Optional[float] = ..., default_max_chunks: _Optional[int] = ..., default_tag: _Optional[str] = ..., default_embeddings_model: _Optional[str] = ..., default_embeddings_model_name: _Optional[str] = ..., default_embeddings_engine: _Optional[str] = ..., type: _Optional[str] = ..., rockset_collection_name: _Optional[str] = ...) -> None: ...
280
-
281
- class CompletionConfig(_message.Message):
282
- __slots__ = ("default_completion_model", "no_chunk_message", "sys_prompt_rag")
283
- DEFAULT_COMPLETION_MODEL_FIELD_NUMBER: _ClassVar[int]
284
- NO_CHUNK_MESSAGE_FIELD_NUMBER: _ClassVar[int]
285
- SYS_PROMPT_RAG_FIELD_NUMBER: _ClassVar[int]
286
- default_completion_model: str
287
- no_chunk_message: str
288
- sys_prompt_rag: str
289
- def __init__(self, default_completion_model: _Optional[str] = ..., no_chunk_message: _Optional[str] = ..., sys_prompt_rag: _Optional[str] = ...) -> None: ...
290
-
291
- class WorkflowConfig(_message.Message):
292
- __slots__ = ("storage_prefix",)
293
- STORAGE_PREFIX_FIELD_NUMBER: _ClassVar[int]
294
- storage_prefix: str
295
- def __init__(self, storage_prefix: _Optional[str] = ...) -> None: ...
296
-
297
- class DataStagingConfig(_message.Message):
298
- __slots__ = ("rockset_api_server", "rockset_workspace", "rockset_collection", "rockset_integration", "gcs_bucket", "gcs_unstructured_data_prefix", "gcs_table_data_prefix", "blob_name_column", "row_number_column")
299
- ROCKSET_API_SERVER_FIELD_NUMBER: _ClassVar[int]
300
- ROCKSET_WORKSPACE_FIELD_NUMBER: _ClassVar[int]
301
- ROCKSET_COLLECTION_FIELD_NUMBER: _ClassVar[int]
302
- ROCKSET_INTEGRATION_FIELD_NUMBER: _ClassVar[int]
303
- GCS_BUCKET_FIELD_NUMBER: _ClassVar[int]
304
- GCS_UNSTRUCTURED_DATA_PREFIX_FIELD_NUMBER: _ClassVar[int]
305
- GCS_TABLE_DATA_PREFIX_FIELD_NUMBER: _ClassVar[int]
306
- BLOB_NAME_COLUMN_FIELD_NUMBER: _ClassVar[int]
307
- ROW_NUMBER_COLUMN_FIELD_NUMBER: _ClassVar[int]
308
- rockset_api_server: str
309
- rockset_workspace: str
310
- rockset_collection: str
311
- rockset_integration: str
312
- gcs_bucket: str
313
- gcs_unstructured_data_prefix: str
314
- gcs_table_data_prefix: str
315
- blob_name_column: str
316
- row_number_column: str
317
- def __init__(self, rockset_api_server: _Optional[str] = ..., rockset_workspace: _Optional[str] = ..., rockset_collection: _Optional[str] = ..., rockset_integration: _Optional[str] = ..., gcs_bucket: _Optional[str] = ..., gcs_unstructured_data_prefix: _Optional[str] = ..., gcs_table_data_prefix: _Optional[str] = ..., blob_name_column: _Optional[str] = ..., row_number_column: _Optional[str] = ...) -> None: ...
318
-
319
- class Auth0(_message.Message):
320
- __slots__ = ("client_id", "client_secret", "domain")
321
- CLIENT_ID_FIELD_NUMBER: _ClassVar[int]
322
- CLIENT_SECRET_FIELD_NUMBER: _ClassVar[int]
323
- DOMAIN_FIELD_NUMBER: _ClassVar[int]
324
- client_id: str
325
- client_secret: str
326
- domain: str
327
- def __init__(self, client_id: _Optional[str] = ..., client_secret: _Optional[str] = ..., domain: _Optional[str] = ...) -> None: ...
328
-
329
- class Settings(_message.Message):
330
- __slots__ = ("load_dotenv", "config", "dev_mode", "metrics", "chat_server", "embedgen", "chat", "db", "ingest", "completion_models", "embedding_models", "huggingface", "retriever", "completion", "rockset", "google_identity_token", "check_name", "test_db", "workflow", "data_staging", "experiment", "auth0")
331
- class CompletionModelsEntry(_message.Message):
332
- __slots__ = ("key", "value")
333
- KEY_FIELD_NUMBER: _ClassVar[int]
334
- VALUE_FIELD_NUMBER: _ClassVar[int]
335
- key: str
336
- value: CompletionModelConfig
337
- def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[CompletionModelConfig, _Mapping]] = ...) -> None: ...
338
- class EmbeddingModelsEntry(_message.Message):
339
- __slots__ = ("key", "value")
340
- KEY_FIELD_NUMBER: _ClassVar[int]
341
- VALUE_FIELD_NUMBER: _ClassVar[int]
342
- key: str
343
- value: EmbeddingModelConfig
344
- def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[EmbeddingModelConfig, _Mapping]] = ...) -> None: ...
345
- LOAD_DOTENV_FIELD_NUMBER: _ClassVar[int]
346
- CONFIG_FIELD_NUMBER: _ClassVar[int]
347
- DEV_MODE_FIELD_NUMBER: _ClassVar[int]
348
- METRICS_FIELD_NUMBER: _ClassVar[int]
349
- CHAT_SERVER_FIELD_NUMBER: _ClassVar[int]
350
- EMBEDGEN_FIELD_NUMBER: _ClassVar[int]
351
- CHAT_FIELD_NUMBER: _ClassVar[int]
352
- DB_FIELD_NUMBER: _ClassVar[int]
353
- INGEST_FIELD_NUMBER: _ClassVar[int]
354
- COMPLETION_MODELS_FIELD_NUMBER: _ClassVar[int]
355
- EMBEDDING_MODELS_FIELD_NUMBER: _ClassVar[int]
356
- HUGGINGFACE_FIELD_NUMBER: _ClassVar[int]
357
- RETRIEVER_FIELD_NUMBER: _ClassVar[int]
358
- COMPLETION_FIELD_NUMBER: _ClassVar[int]
359
- ROCKSET_FIELD_NUMBER: _ClassVar[int]
360
- GOOGLE_IDENTITY_TOKEN_FIELD_NUMBER: _ClassVar[int]
361
- CHECK_NAME_FIELD_NUMBER: _ClassVar[int]
362
- TEST_DB_FIELD_NUMBER: _ClassVar[int]
363
- WORKFLOW_FIELD_NUMBER: _ClassVar[int]
364
- DATA_STAGING_FIELD_NUMBER: _ClassVar[int]
365
- EXPERIMENT_FIELD_NUMBER: _ClassVar[int]
366
- AUTH0_FIELD_NUMBER: _ClassVar[int]
367
- load_dotenv: bool
368
- config: MetaConfig
369
- dev_mode: str
370
- metrics: MetricsConfig
371
- chat_server: ChatServerConfig
372
- embedgen: EmbedgenConfig
373
- chat: ChatConfig
374
- db: DBConfig
375
- ingest: IngestConfig
376
- completion_models: _containers.MessageMap[str, CompletionModelConfig]
377
- embedding_models: _containers.MessageMap[str, EmbeddingModelConfig]
378
- huggingface: HuggingfaceConfig
379
- retriever: RetrieverConfig
380
- completion: CompletionConfig
381
- rockset: RocksetConfig
382
- google_identity_token: str
383
- check_name: str
384
- test_db: str
385
- workflow: WorkflowConfig
386
- data_staging: DataStagingConfig
387
- experiment: ExperimentConfig
388
- auth0: Auth0
389
- def __init__(self, load_dotenv: bool = ..., config: _Optional[_Union[MetaConfig, _Mapping]] = ..., dev_mode: _Optional[str] = ..., metrics: _Optional[_Union[MetricsConfig, _Mapping]] = ..., chat_server: _Optional[_Union[ChatServerConfig, _Mapping]] = ..., embedgen: _Optional[_Union[EmbedgenConfig, _Mapping]] = ..., chat: _Optional[_Union[ChatConfig, _Mapping]] = ..., db: _Optional[_Union[DBConfig, _Mapping]] = ..., ingest: _Optional[_Union[IngestConfig, _Mapping]] = ..., completion_models: _Optional[_Mapping[str, CompletionModelConfig]] = ..., embedding_models: _Optional[_Mapping[str, EmbeddingModelConfig]] = ..., huggingface: _Optional[_Union[HuggingfaceConfig, _Mapping]] = ..., retriever: _Optional[_Union[RetrieverConfig, _Mapping]] = ..., completion: _Optional[_Union[CompletionConfig, _Mapping]] = ..., rockset: _Optional[_Union[RocksetConfig, _Mapping]] = ..., google_identity_token: _Optional[str] = ..., check_name: _Optional[str] = ..., test_db: _Optional[str] = ..., workflow: _Optional[_Union[WorkflowConfig, _Mapping]] = ..., data_staging: _Optional[_Union[DataStagingConfig, _Mapping]] = ..., experiment: _Optional[_Union[ExperimentConfig, _Mapping]] = ..., auth0: _Optional[_Union[Auth0, _Mapping]] = ...) -> None: ...
@@ -1,4 +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
-
@@ -1,17 +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 grpc
8
- import grpc.aio
9
- import typing
10
-
11
- _T = typing.TypeVar('_T')
12
-
13
- class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta):
14
- ...
15
-
16
- class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext): # type: ignore
17
- ...
File without changes
File without changes
@@ -1,101 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: corvic/platform/v1/platform.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
- from buf.validate import validate_pb2 as buf_dot_validate_dot_validate__pb2
16
- from google.protobuf import descriptor_pb2 as google_dot_protobuf_dot_descriptor__pb2
17
- from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
18
-
19
-
20
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!corvic/platform/v1/platform.proto\x12\x12\x63orvic.platform.v1\x1a\x1b\x62uf/validate/validate.proto\x1a google/protobuf/descriptor.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\x8c\x02\n\x03Org\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x04name\x12p\n\x02id\x18\x02 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\x12\x39\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12;\n\x0bmodified_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nmodifiedAt\"\xc8\x01\n\x04User\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\x12\"\n\x08org_name\x18\x02 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\x12\x14\n\x05\x65mail\x18\x03 \x01(\tR\x05\x65mail\x12\x14\n\x05roles\x18\x04 \x03(\tR\x05roles\"/\n\x10\x43reateOrgRequest\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x04name\">\n\x11\x43reateOrgResponse\x12)\n\x03org\x18\x01 \x01(\x0b\x32\x17.corvic.platform.v1.OrgR\x03org\"#\n\rGetOrgRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\";\n\x0eGetOrgResponse\x12)\n\x03org\x18\x01 \x01(\x0b\x32\x17.corvic.platform.v1.OrgR\x03org\"\x11\n\x0fListOrgsRequest\"?\n\x10ListOrgsResponse\x12+\n\x04orgs\x18\x01 \x03(\x0b\x32\x17.corvic.platform.v1.OrgR\x04orgs\"9\n\x13ListOrgUsersRequest\x12\"\n\x08org_name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\"F\n\x14ListOrgUsersResponse\x12.\n\x05users\x18\x01 \x03(\x0b\x32\x18.corvic.platform.v1.UserR\x05users\"}\n\x11\x41\x64\x64OrgUserRequest\x12\x14\n\x05\x65mail\x18\x01 \x01(\tR\x05\x65mail\x12\"\n\x08org_name\x18\x02 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\x12.\n\x05roles\x18\x03 \x03(\x0e\x32\x18.corvic.platform.v1.RoleR\x05roles\"B\n\x12\x41\x64\x64OrgUserResponse\x12,\n\x04user\x18\x01 \x01(\x0b\x32\x18.corvic.platform.v1.UserR\x04user\"\x85\x01\n\x11GetOrgUserRequest\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\"B\n\x12GetOrgUserResponse\x12,\n\x04user\x18\x01 \x01(\x0b\x32\x18.corvic.platform.v1.UserR\x04user\"\x8c\x01\n\x18\x44\x65\x61\x63tivateOrgUserRequest\x12p\n\x02id\x18\x01 \x01(\tB`\xbaH]\xba\x01Z\n\x0estring.pattern\x12\x16value must be a number\x1a\x30this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')R\x02id\"\x1b\n\x19\x44\x65\x61\x63tivateOrgUserResponse\".\n\x0fPurgeOrgRequest\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x04name\"\x12\n\x10PurgeOrgResponse\"B\n\x1c\x44\x65\x61\x63tivateAllOrgUsersRequest\x12\"\n\x08org_name\x18\x01 \x01(\tB\x07\xbaH\x04r\x02\x10\x01R\x07orgName\"\x1f\n\x1d\x44\x65\x61\x63tivateAllOrgUsersResponse*C\n\x04Role\x12\x14\n\x10ROLE_UNSPECIFIED\x10\x00\x12\x12\n\x0eROLE_ORG_ADMIN\x10\x01\x12\x11\n\rROLE_ORG_USER\x10\x02\x32\x90\x07\n\nOrgService\x12T\n\x06GetOrg\x12!.corvic.platform.v1.GetOrgRequest\x1a\".corvic.platform.v1.GetOrgResponse\"\x03\x90\x02\x01\x12Z\n\tCreateOrg\x12$.corvic.platform.v1.CreateOrgRequest\x1a%.corvic.platform.v1.CreateOrgResponse\"\x00\x12Z\n\x08ListOrgs\x12#.corvic.platform.v1.ListOrgsRequest\x1a$.corvic.platform.v1.ListOrgsResponse\"\x03\x90\x02\x01\x12`\n\nGetOrgUser\x12%.corvic.platform.v1.GetOrgUserRequest\x1a&.corvic.platform.v1.GetOrgUserResponse\"\x03\x90\x02\x01\x12]\n\nAddOrgUser\x12%.corvic.platform.v1.AddOrgUserRequest\x1a&.corvic.platform.v1.AddOrgUserResponse\"\x00\x12\x66\n\x0cListOrgUsers\x12\'.corvic.platform.v1.ListOrgUsersRequest\x1a(.corvic.platform.v1.ListOrgUsersResponse\"\x03\x90\x02\x01\x12r\n\x11\x44\x65\x61\x63tivateOrgUser\x12,.corvic.platform.v1.DeactivateOrgUserRequest\x1a-.corvic.platform.v1.DeactivateOrgUserResponse\"\x00\x12W\n\x08PurgeOrg\x12#.corvic.platform.v1.PurgeOrgRequest\x1a$.corvic.platform.v1.PurgeOrgResponse\"\x00\x12~\n\x15\x44\x65\x61\x63tivateAllOrgUsers\x12\x30.corvic.platform.v1.DeactivateAllOrgUsersRequest\x1a\x31.corvic.platform.v1.DeactivateAllOrgUsersResponse\"\x00\x62\x06proto3')
21
-
22
- _globals = globals()
23
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
24
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'corvic.platform.v1.platform_pb2', _globals)
25
- if _descriptor._USE_C_DESCRIPTORS == False:
26
- DESCRIPTOR._options = None
27
- _globals['_ORG'].fields_by_name['name']._options = None
28
- _globals['_ORG'].fields_by_name['name']._serialized_options = b'\272H\004r\002\020\001'
29
- _globals['_ORG'].fields_by_name['id']._options = None
30
- _globals['_ORG'].fields_by_name['id']._serialized_options = b'\272H]\272\001Z\n\016string.pattern\022\026value must be a number\0320this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')'
31
- _globals['_USER'].fields_by_name['id']._options = None
32
- _globals['_USER'].fields_by_name['id']._serialized_options = b'\272H]\272\001Z\n\016string.pattern\022\026value must be a number\0320this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')'
33
- _globals['_USER'].fields_by_name['org_name']._options = None
34
- _globals['_USER'].fields_by_name['org_name']._serialized_options = b'\272H\004r\002\020\001'
35
- _globals['_CREATEORGREQUEST'].fields_by_name['name']._options = None
36
- _globals['_CREATEORGREQUEST'].fields_by_name['name']._serialized_options = b'\272H\004r\002\020\001'
37
- _globals['_LISTORGUSERSREQUEST'].fields_by_name['org_name']._options = None
38
- _globals['_LISTORGUSERSREQUEST'].fields_by_name['org_name']._serialized_options = b'\272H\004r\002\020\001'
39
- _globals['_ADDORGUSERREQUEST'].fields_by_name['org_name']._options = None
40
- _globals['_ADDORGUSERREQUEST'].fields_by_name['org_name']._serialized_options = b'\272H\004r\002\020\001'
41
- _globals['_GETORGUSERREQUEST'].fields_by_name['id']._options = None
42
- _globals['_GETORGUSERREQUEST'].fields_by_name['id']._serialized_options = b'\272H]\272\001Z\n\016string.pattern\022\026value must be a number\0320this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')'
43
- _globals['_DEACTIVATEORGUSERREQUEST'].fields_by_name['id']._options = None
44
- _globals['_DEACTIVATEORGUSERREQUEST'].fields_by_name['id']._serialized_options = b'\272H]\272\001Z\n\016string.pattern\022\026value must be a number\0320this.matches(\'^[0-9]+$\') && !this.endsWith(\'\\n\')'
45
- _globals['_PURGEORGREQUEST'].fields_by_name['name']._options = None
46
- _globals['_PURGEORGREQUEST'].fields_by_name['name']._serialized_options = b'\272H\004r\002\020\001'
47
- _globals['_DEACTIVATEALLORGUSERSREQUEST'].fields_by_name['org_name']._options = None
48
- _globals['_DEACTIVATEALLORGUSERSREQUEST'].fields_by_name['org_name']._serialized_options = b'\272H\004r\002\020\001'
49
- _globals['_ORGSERVICE'].methods_by_name['GetOrg']._options = None
50
- _globals['_ORGSERVICE'].methods_by_name['GetOrg']._serialized_options = b'\220\002\001'
51
- _globals['_ORGSERVICE'].methods_by_name['ListOrgs']._options = None
52
- _globals['_ORGSERVICE'].methods_by_name['ListOrgs']._serialized_options = b'\220\002\001'
53
- _globals['_ORGSERVICE'].methods_by_name['GetOrgUser']._options = None
54
- _globals['_ORGSERVICE'].methods_by_name['GetOrgUser']._serialized_options = b'\220\002\001'
55
- _globals['_ORGSERVICE'].methods_by_name['ListOrgUsers']._options = None
56
- _globals['_ORGSERVICE'].methods_by_name['ListOrgUsers']._serialized_options = b'\220\002\001'
57
- _globals['_ROLE']._serialized_start=1793
58
- _globals['_ROLE']._serialized_end=1860
59
- _globals['_ORG']._serialized_start=154
60
- _globals['_ORG']._serialized_end=422
61
- _globals['_USER']._serialized_start=425
62
- _globals['_USER']._serialized_end=625
63
- _globals['_CREATEORGREQUEST']._serialized_start=627
64
- _globals['_CREATEORGREQUEST']._serialized_end=674
65
- _globals['_CREATEORGRESPONSE']._serialized_start=676
66
- _globals['_CREATEORGRESPONSE']._serialized_end=738
67
- _globals['_GETORGREQUEST']._serialized_start=740
68
- _globals['_GETORGREQUEST']._serialized_end=775
69
- _globals['_GETORGRESPONSE']._serialized_start=777
70
- _globals['_GETORGRESPONSE']._serialized_end=836
71
- _globals['_LISTORGSREQUEST']._serialized_start=838
72
- _globals['_LISTORGSREQUEST']._serialized_end=855
73
- _globals['_LISTORGSRESPONSE']._serialized_start=857
74
- _globals['_LISTORGSRESPONSE']._serialized_end=920
75
- _globals['_LISTORGUSERSREQUEST']._serialized_start=922
76
- _globals['_LISTORGUSERSREQUEST']._serialized_end=979
77
- _globals['_LISTORGUSERSRESPONSE']._serialized_start=981
78
- _globals['_LISTORGUSERSRESPONSE']._serialized_end=1051
79
- _globals['_ADDORGUSERREQUEST']._serialized_start=1053
80
- _globals['_ADDORGUSERREQUEST']._serialized_end=1178
81
- _globals['_ADDORGUSERRESPONSE']._serialized_start=1180
82
- _globals['_ADDORGUSERRESPONSE']._serialized_end=1246
83
- _globals['_GETORGUSERREQUEST']._serialized_start=1249
84
- _globals['_GETORGUSERREQUEST']._serialized_end=1382
85
- _globals['_GETORGUSERRESPONSE']._serialized_start=1384
86
- _globals['_GETORGUSERRESPONSE']._serialized_end=1450
87
- _globals['_DEACTIVATEORGUSERREQUEST']._serialized_start=1453
88
- _globals['_DEACTIVATEORGUSERREQUEST']._serialized_end=1593
89
- _globals['_DEACTIVATEORGUSERRESPONSE']._serialized_start=1595
90
- _globals['_DEACTIVATEORGUSERRESPONSE']._serialized_end=1622
91
- _globals['_PURGEORGREQUEST']._serialized_start=1624
92
- _globals['_PURGEORGREQUEST']._serialized_end=1670
93
- _globals['_PURGEORGRESPONSE']._serialized_start=1672
94
- _globals['_PURGEORGRESPONSE']._serialized_end=1690
95
- _globals['_DEACTIVATEALLORGUSERSREQUEST']._serialized_start=1692
96
- _globals['_DEACTIVATEALLORGUSERSREQUEST']._serialized_end=1758
97
- _globals['_DEACTIVATEALLORGUSERSRESPONSE']._serialized_start=1760
98
- _globals['_DEACTIVATEALLORGUSERSRESPONSE']._serialized_end=1791
99
- _globals['_ORGSERVICE']._serialized_start=1863
100
- _globals['_ORGSERVICE']._serialized_end=2775
101
- # @@protoc_insertion_point(module_scope)