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.
Files changed (30) hide show
  1. corvic/engine/_native.pyd +0 -0
  2. corvic/model/_completion_model.py +9 -0
  3. corvic/model/_proto_orm_convert.py +2 -0
  4. corvic/model/_source.py +8 -0
  5. corvic/model/_space.py +11 -4
  6. corvic/op_graph/_schema.py +2 -4
  7. corvic/op_graph/_transformations.py +3 -3
  8. corvic/op_graph/ops.py +30 -12
  9. corvic/orm/__init__.py +1 -0
  10. corvic/orm/ids.py +6 -1
  11. corvic/system/in_memory_executor.py +42 -12
  12. corvic/table/table.py +9 -3
  13. {corvic_engine-0.3.0rc56.dist-info → corvic_engine-0.3.0rc58.dist-info}/METADATA +1 -1
  14. {corvic_engine-0.3.0rc56.dist-info → corvic_engine-0.3.0rc58.dist-info}/RECORD +18 -30
  15. corvic_generated/model/v1alpha/models_pb2.py +2 -2
  16. corvic_generated/model/v1alpha/models_pb2.pyi +6 -4
  17. corvic_generated/config/__init__.py +0 -0
  18. corvic_generated/config/v1/__init__.py +0 -0
  19. corvic_generated/config/v1/settings_pb2.py +0 -128
  20. corvic_generated/config/v1/settings_pb2.pyi +0 -389
  21. corvic_generated/config/v1/settings_pb2_grpc.py +0 -4
  22. corvic_generated/config/v1/settings_pb2_grpc.pyi +0 -17
  23. corvic_generated/platform/__init__.py +0 -0
  24. corvic_generated/platform/v1/__init__.py +0 -0
  25. corvic_generated/platform/v1/platform_pb2.py +0 -101
  26. corvic_generated/platform/v1/platform_pb2.pyi +0 -147
  27. corvic_generated/platform/v1/platform_pb2_grpc.py +0 -330
  28. corvic_generated/platform/v1/platform_pb2_grpc.pyi +0 -153
  29. {corvic_engine-0.3.0rc56.dist-info → corvic_engine-0.3.0rc58.dist-info}/WHEEL +0 -0
  30. {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,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