corvic-engine 0.3.0rc56__cp38-abi3-win_amd64.whl → 0.3.0rc58__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 +9 -0
- corvic/model/_proto_orm_convert.py +2 -0
- corvic/model/_source.py +8 -0
- corvic/model/_space.py +11 -4
- corvic/op_graph/_schema.py +2 -4
- corvic/op_graph/_transformations.py +3 -3
- corvic/op_graph/ops.py +30 -12
- corvic/orm/__init__.py +1 -0
- corvic/orm/ids.py +6 -1
- corvic/system/in_memory_executor.py +42 -12
- corvic/table/table.py +9 -3
- {corvic_engine-0.3.0rc56.dist-info → corvic_engine-0.3.0rc58.dist-info}/METADATA +1 -1
- {corvic_engine-0.3.0rc56.dist-info → corvic_engine-0.3.0rc58.dist-info}/RECORD +18 -30
- corvic_generated/model/v1alpha/models_pb2.py +2 -2
- corvic_generated/model/v1alpha/models_pb2.pyi +6 -4
- 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_engine-0.3.0rc56.dist-info → corvic_engine-0.3.0rc58.dist-info}/WHEEL +0 -0
- {corvic_engine-0.3.0rc56.dist-info → corvic_engine-0.3.0rc58.dist-info}/licenses/LICENSE +0 -0
@@ -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)
|
@@ -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,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
|