PyPI - genesis-flow - Versions diffs - 1.0.0__py3-none-any.whl - Mend

genesis-flow 1.0.0__py3-none-any.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 (645) hide show

genesis_flow-1.0.0.dist-info/METADATA +822 -0
genesis_flow-1.0.0.dist-info/RECORD +645 -0
genesis_flow-1.0.0.dist-info/WHEEL +5 -0
genesis_flow-1.0.0.dist-info/entry_points.txt +19 -0
genesis_flow-1.0.0.dist-info/licenses/LICENSE.txt +202 -0
genesis_flow-1.0.0.dist-info/top_level.txt +1 -0
mlflow/__init__.py +367 -0
mlflow/__main__.py +3 -0
mlflow/ag2/__init__.py +56 -0
mlflow/ag2/ag2_logger.py +294 -0
mlflow/anthropic/__init__.py +40 -0
mlflow/anthropic/autolog.py +129 -0
mlflow/anthropic/chat.py +144 -0
mlflow/artifacts/__init__.py +268 -0
mlflow/autogen/__init__.py +144 -0
mlflow/autogen/chat.py +142 -0
mlflow/azure/__init__.py +26 -0
mlflow/azure/auth_handler.py +257 -0
mlflow/azure/client.py +319 -0
mlflow/azure/config.py +120 -0
mlflow/azure/connection_factory.py +340 -0
mlflow/azure/exceptions.py +27 -0
mlflow/azure/stores.py +327 -0
mlflow/azure/utils.py +183 -0
mlflow/bedrock/__init__.py +45 -0
mlflow/bedrock/_autolog.py +202 -0
mlflow/bedrock/chat.py +122 -0
mlflow/bedrock/stream.py +160 -0
mlflow/bedrock/utils.py +43 -0
mlflow/cli.py +707 -0
mlflow/client.py +12 -0
mlflow/config/__init__.py +56 -0
mlflow/crewai/__init__.py +79 -0
mlflow/crewai/autolog.py +253 -0
mlflow/crewai/chat.py +29 -0
mlflow/data/__init__.py +75 -0
mlflow/data/artifact_dataset_sources.py +170 -0
mlflow/data/code_dataset_source.py +40 -0
mlflow/data/dataset.py +123 -0
mlflow/data/dataset_registry.py +168 -0
mlflow/data/dataset_source.py +110 -0
mlflow/data/dataset_source_registry.py +219 -0
mlflow/data/delta_dataset_source.py +167 -0
mlflow/data/digest_utils.py +108 -0
mlflow/data/evaluation_dataset.py +562 -0
mlflow/data/filesystem_dataset_source.py +81 -0
mlflow/data/http_dataset_source.py +145 -0
mlflow/data/huggingface_dataset.py +258 -0
mlflow/data/huggingface_dataset_source.py +118 -0
mlflow/data/meta_dataset.py +104 -0
mlflow/data/numpy_dataset.py +223 -0
mlflow/data/pandas_dataset.py +231 -0
mlflow/data/polars_dataset.py +352 -0
mlflow/data/pyfunc_dataset_mixin.py +31 -0
mlflow/data/schema.py +76 -0
mlflow/data/sources.py +1 -0
mlflow/data/spark_dataset.py +406 -0
mlflow/data/spark_dataset_source.py +74 -0
mlflow/data/spark_delta_utils.py +118 -0
mlflow/data/tensorflow_dataset.py +350 -0
mlflow/data/uc_volume_dataset_source.py +81 -0
mlflow/db.py +27 -0
mlflow/dspy/__init__.py +17 -0
mlflow/dspy/autolog.py +197 -0
mlflow/dspy/callback.py +398 -0
mlflow/dspy/constant.py +1 -0
mlflow/dspy/load.py +93 -0
mlflow/dspy/save.py +393 -0
mlflow/dspy/util.py +109 -0
mlflow/dspy/wrapper.py +226 -0
mlflow/entities/__init__.py +104 -0
mlflow/entities/_mlflow_object.py +52 -0
mlflow/entities/assessment.py +545 -0
mlflow/entities/assessment_error.py +80 -0
mlflow/entities/assessment_source.py +141 -0
mlflow/entities/dataset.py +92 -0
mlflow/entities/dataset_input.py +51 -0
mlflow/entities/dataset_summary.py +62 -0
mlflow/entities/document.py +48 -0
mlflow/entities/experiment.py +109 -0
mlflow/entities/experiment_tag.py +35 -0
mlflow/entities/file_info.py +45 -0
mlflow/entities/input_tag.py +35 -0
mlflow/entities/lifecycle_stage.py +35 -0
mlflow/entities/logged_model.py +228 -0
mlflow/entities/logged_model_input.py +26 -0
mlflow/entities/logged_model_output.py +32 -0
mlflow/entities/logged_model_parameter.py +46 -0
mlflow/entities/logged_model_status.py +74 -0
mlflow/entities/logged_model_tag.py +33 -0
mlflow/entities/metric.py +200 -0
mlflow/entities/model_registry/__init__.py +29 -0
mlflow/entities/model_registry/_model_registry_entity.py +13 -0
mlflow/entities/model_registry/model_version.py +243 -0
mlflow/entities/model_registry/model_version_deployment_job_run_state.py +44 -0
mlflow/entities/model_registry/model_version_deployment_job_state.py +70 -0
mlflow/entities/model_registry/model_version_search.py +25 -0
mlflow/entities/model_registry/model_version_stages.py +25 -0
mlflow/entities/model_registry/model_version_status.py +35 -0
mlflow/entities/model_registry/model_version_tag.py +35 -0
mlflow/entities/model_registry/prompt.py +73 -0
mlflow/entities/model_registry/prompt_version.py +244 -0
mlflow/entities/model_registry/registered_model.py +175 -0
mlflow/entities/model_registry/registered_model_alias.py +35 -0
mlflow/entities/model_registry/registered_model_deployment_job_state.py +39 -0
mlflow/entities/model_registry/registered_model_search.py +25 -0
mlflow/entities/model_registry/registered_model_tag.py +35 -0
mlflow/entities/multipart_upload.py +74 -0
mlflow/entities/param.py +49 -0
mlflow/entities/run.py +97 -0
mlflow/entities/run_data.py +84 -0
mlflow/entities/run_info.py +188 -0
mlflow/entities/run_inputs.py +59 -0
mlflow/entities/run_outputs.py +43 -0
mlflow/entities/run_status.py +41 -0
mlflow/entities/run_tag.py +36 -0
mlflow/entities/source_type.py +31 -0
mlflow/entities/span.py +774 -0
mlflow/entities/span_event.py +96 -0
mlflow/entities/span_status.py +102 -0
mlflow/entities/trace.py +317 -0
mlflow/entities/trace_data.py +71 -0
mlflow/entities/trace_info.py +220 -0
mlflow/entities/trace_info_v2.py +162 -0
mlflow/entities/trace_location.py +173 -0
mlflow/entities/trace_state.py +39 -0
mlflow/entities/trace_status.py +68 -0
mlflow/entities/view_type.py +51 -0
mlflow/environment_variables.py +866 -0
mlflow/evaluation/__init__.py +16 -0
mlflow/evaluation/assessment.py +369 -0
mlflow/evaluation/evaluation.py +411 -0
mlflow/evaluation/evaluation_tag.py +61 -0
mlflow/evaluation/fluent.py +48 -0
mlflow/evaluation/utils.py +201 -0
mlflow/exceptions.py +213 -0
mlflow/experiments.py +140 -0
mlflow/gemini/__init__.py +81 -0
mlflow/gemini/autolog.py +186 -0
mlflow/gemini/chat.py +261 -0
mlflow/genai/__init__.py +71 -0
mlflow/genai/datasets/__init__.py +67 -0
mlflow/genai/datasets/evaluation_dataset.py +131 -0
mlflow/genai/evaluation/__init__.py +3 -0
mlflow/genai/evaluation/base.py +411 -0
mlflow/genai/evaluation/constant.py +23 -0
mlflow/genai/evaluation/utils.py +244 -0
mlflow/genai/judges/__init__.py +21 -0
mlflow/genai/judges/databricks.py +404 -0
mlflow/genai/label_schemas/__init__.py +153 -0
mlflow/genai/label_schemas/label_schemas.py +209 -0
mlflow/genai/labeling/__init__.py +159 -0
mlflow/genai/labeling/labeling.py +250 -0
mlflow/genai/optimize/__init__.py +13 -0
mlflow/genai/optimize/base.py +198 -0
mlflow/genai/optimize/optimizers/__init__.py +4 -0
mlflow/genai/optimize/optimizers/base_optimizer.py +38 -0
mlflow/genai/optimize/optimizers/dspy_mipro_optimizer.py +221 -0
mlflow/genai/optimize/optimizers/dspy_optimizer.py +91 -0
mlflow/genai/optimize/optimizers/utils/dspy_mipro_callback.py +76 -0
mlflow/genai/optimize/optimizers/utils/dspy_mipro_utils.py +18 -0
mlflow/genai/optimize/types.py +75 -0
mlflow/genai/optimize/util.py +30 -0
mlflow/genai/prompts/__init__.py +206 -0
mlflow/genai/scheduled_scorers.py +431 -0
mlflow/genai/scorers/__init__.py +26 -0
mlflow/genai/scorers/base.py +492 -0
mlflow/genai/scorers/builtin_scorers.py +765 -0
mlflow/genai/scorers/scorer_utils.py +138 -0
mlflow/genai/scorers/validation.py +165 -0
mlflow/genai/utils/data_validation.py +146 -0
mlflow/genai/utils/enum_utils.py +23 -0
mlflow/genai/utils/trace_utils.py +211 -0
mlflow/groq/__init__.py +42 -0
mlflow/groq/_groq_autolog.py +74 -0
mlflow/johnsnowlabs/__init__.py +888 -0
mlflow/langchain/__init__.py +24 -0
mlflow/langchain/api_request_parallel_processor.py +330 -0
mlflow/langchain/autolog.py +147 -0
mlflow/langchain/chat_agent_langgraph.py +340 -0
mlflow/langchain/constant.py +1 -0
mlflow/langchain/constants.py +1 -0
mlflow/langchain/databricks_dependencies.py +444 -0
mlflow/langchain/langchain_tracer.py +597 -0
mlflow/langchain/model.py +919 -0
mlflow/langchain/output_parsers.py +142 -0
mlflow/langchain/retriever_chain.py +153 -0
mlflow/langchain/runnables.py +527 -0
mlflow/langchain/utils/chat.py +402 -0
mlflow/langchain/utils/logging.py +671 -0
mlflow/langchain/utils/serialization.py +36 -0
mlflow/legacy_databricks_cli/__init__.py +0 -0
mlflow/legacy_databricks_cli/configure/__init__.py +0 -0
mlflow/legacy_databricks_cli/configure/provider.py +482 -0
mlflow/litellm/__init__.py +175 -0
mlflow/llama_index/__init__.py +22 -0
mlflow/llama_index/autolog.py +55 -0
mlflow/llama_index/chat.py +43 -0
mlflow/llama_index/constant.py +1 -0
mlflow/llama_index/model.py +577 -0
mlflow/llama_index/pyfunc_wrapper.py +332 -0
mlflow/llama_index/serialize_objects.py +188 -0
mlflow/llama_index/tracer.py +561 -0
mlflow/metrics/__init__.py +479 -0
mlflow/metrics/base.py +39 -0
mlflow/metrics/genai/__init__.py +25 -0
mlflow/metrics/genai/base.py +101 -0
mlflow/metrics/genai/genai_metric.py +771 -0
mlflow/metrics/genai/metric_definitions.py +450 -0
mlflow/metrics/genai/model_utils.py +371 -0
mlflow/metrics/genai/prompt_template.py +68 -0
mlflow/metrics/genai/prompts/__init__.py +0 -0
mlflow/metrics/genai/prompts/v1.py +422 -0
mlflow/metrics/genai/utils.py +6 -0
mlflow/metrics/metric_definitions.py +619 -0
mlflow/mismatch.py +34 -0
mlflow/mistral/__init__.py +34 -0
mlflow/mistral/autolog.py +71 -0
mlflow/mistral/chat.py +135 -0
mlflow/ml_package_versions.py +452 -0
mlflow/models/__init__.py +97 -0
mlflow/models/auth_policy.py +83 -0
mlflow/models/cli.py +354 -0
mlflow/models/container/__init__.py +294 -0
mlflow/models/container/scoring_server/__init__.py +0 -0
mlflow/models/container/scoring_server/nginx.conf +39 -0
mlflow/models/dependencies_schemas.py +287 -0
mlflow/models/display_utils.py +158 -0
mlflow/models/docker_utils.py +211 -0
mlflow/models/evaluation/__init__.py +23 -0
mlflow/models/evaluation/_shap_patch.py +64 -0
mlflow/models/evaluation/artifacts.py +194 -0
mlflow/models/evaluation/base.py +1811 -0
mlflow/models/evaluation/calibration_curve.py +109 -0
mlflow/models/evaluation/default_evaluator.py +996 -0
mlflow/models/evaluation/deprecated.py +23 -0
mlflow/models/evaluation/evaluator_registry.py +80 -0
mlflow/models/evaluation/evaluators/classifier.py +704 -0
mlflow/models/evaluation/evaluators/default.py +233 -0
mlflow/models/evaluation/evaluators/regressor.py +96 -0
mlflow/models/evaluation/evaluators/shap.py +296 -0
mlflow/models/evaluation/lift_curve.py +178 -0
mlflow/models/evaluation/utils/metric.py +123 -0
mlflow/models/evaluation/utils/trace.py +179 -0
mlflow/models/evaluation/validation.py +434 -0
mlflow/models/flavor_backend.py +93 -0
mlflow/models/flavor_backend_registry.py +53 -0
mlflow/models/model.py +1639 -0
mlflow/models/model_config.py +150 -0
mlflow/models/notebook_resources/agent_evaluation_template.html +235 -0
mlflow/models/notebook_resources/eval_with_dataset_example.py +22 -0
mlflow/models/notebook_resources/eval_with_synthetic_example.py +22 -0
mlflow/models/python_api.py +369 -0
mlflow/models/rag_signatures.py +128 -0
mlflow/models/resources.py +321 -0
mlflow/models/signature.py +662 -0
mlflow/models/utils.py +2054 -0
mlflow/models/wheeled_model.py +280 -0
mlflow/openai/__init__.py +57 -0
mlflow/openai/_agent_tracer.py +364 -0
mlflow/openai/api_request_parallel_processor.py +131 -0
mlflow/openai/autolog.py +509 -0
mlflow/openai/constant.py +1 -0
mlflow/openai/model.py +824 -0
mlflow/openai/utils/chat_schema.py +367 -0
mlflow/optuna/__init__.py +3 -0
mlflow/optuna/storage.py +646 -0
mlflow/plugins/__init__.py +72 -0
mlflow/plugins/base.py +358 -0
mlflow/plugins/builtin/__init__.py +24 -0
mlflow/plugins/builtin/pytorch_plugin.py +150 -0
mlflow/plugins/builtin/sklearn_plugin.py +158 -0
mlflow/plugins/builtin/transformers_plugin.py +187 -0
mlflow/plugins/cli.py +321 -0
mlflow/plugins/discovery.py +340 -0
mlflow/plugins/manager.py +465 -0
mlflow/plugins/registry.py +316 -0
mlflow/plugins/templates/framework_plugin_template.py +329 -0
mlflow/prompt/constants.py +20 -0
mlflow/prompt/promptlab_model.py +197 -0
mlflow/prompt/registry_utils.py +248 -0
mlflow/promptflow/__init__.py +495 -0
mlflow/protos/__init__.py +0 -0
mlflow/protos/assessments_pb2.py +174 -0
mlflow/protos/databricks_artifacts_pb2.py +489 -0
mlflow/protos/databricks_filesystem_service_pb2.py +196 -0
mlflow/protos/databricks_managed_catalog_messages_pb2.py +95 -0
mlflow/protos/databricks_managed_catalog_service_pb2.py +86 -0
mlflow/protos/databricks_pb2.py +267 -0
mlflow/protos/databricks_trace_server_pb2.py +374 -0
mlflow/protos/databricks_uc_registry_messages_pb2.py +1249 -0
mlflow/protos/databricks_uc_registry_service_pb2.py +170 -0
mlflow/protos/facet_feature_statistics_pb2.py +296 -0
mlflow/protos/internal_pb2.py +77 -0
mlflow/protos/mlflow_artifacts_pb2.py +336 -0
mlflow/protos/model_registry_pb2.py +1073 -0
mlflow/protos/scalapb/__init__.py +0 -0
mlflow/protos/scalapb/scalapb_pb2.py +104 -0
mlflow/protos/service_pb2.py +2600 -0
mlflow/protos/unity_catalog_oss_messages_pb2.py +457 -0
mlflow/protos/unity_catalog_oss_service_pb2.py +130 -0
mlflow/protos/unity_catalog_prompt_messages_pb2.py +447 -0
mlflow/protos/unity_catalog_prompt_messages_pb2_grpc.py +24 -0
mlflow/protos/unity_catalog_prompt_service_pb2.py +164 -0
mlflow/protos/unity_catalog_prompt_service_pb2_grpc.py +785 -0
mlflow/py.typed +0 -0
mlflow/pydantic_ai/__init__.py +57 -0
mlflow/pydantic_ai/autolog.py +173 -0
mlflow/pyfunc/__init__.py +3844 -0
mlflow/pyfunc/_mlflow_pyfunc_backend_predict.py +61 -0
mlflow/pyfunc/backend.py +523 -0
mlflow/pyfunc/context.py +78 -0
mlflow/pyfunc/dbconnect_artifact_cache.py +144 -0
mlflow/pyfunc/loaders/__init__.py +7 -0
mlflow/pyfunc/loaders/chat_agent.py +117 -0
mlflow/pyfunc/loaders/chat_model.py +125 -0
mlflow/pyfunc/loaders/code_model.py +31 -0
mlflow/pyfunc/loaders/responses_agent.py +112 -0
mlflow/pyfunc/mlserver.py +46 -0
mlflow/pyfunc/model.py +1473 -0
mlflow/pyfunc/scoring_server/__init__.py +604 -0
mlflow/pyfunc/scoring_server/app.py +7 -0
mlflow/pyfunc/scoring_server/client.py +146 -0
mlflow/pyfunc/spark_model_cache.py +48 -0
mlflow/pyfunc/stdin_server.py +44 -0
mlflow/pyfunc/utils/__init__.py +3 -0
mlflow/pyfunc/utils/data_validation.py +224 -0
mlflow/pyfunc/utils/environment.py +22 -0
mlflow/pyfunc/utils/input_converter.py +47 -0
mlflow/pyfunc/utils/serving_data_parser.py +11 -0
mlflow/pytorch/__init__.py +1171 -0
mlflow/pytorch/_lightning_autolog.py +580 -0
mlflow/pytorch/_pytorch_autolog.py +50 -0
mlflow/pytorch/pickle_module.py +35 -0
mlflow/rfunc/__init__.py +42 -0
mlflow/rfunc/backend.py +134 -0
mlflow/runs.py +89 -0
mlflow/server/__init__.py +302 -0
mlflow/server/auth/__init__.py +1224 -0
mlflow/server/auth/__main__.py +4 -0
mlflow/server/auth/basic_auth.ini +6 -0
mlflow/server/auth/cli.py +11 -0
mlflow/server/auth/client.py +537 -0
mlflow/server/auth/config.py +34 -0
mlflow/server/auth/db/__init__.py +0 -0
mlflow/server/auth/db/cli.py +18 -0
mlflow/server/auth/db/migrations/__init__.py +0 -0
mlflow/server/auth/db/migrations/alembic.ini +110 -0
mlflow/server/auth/db/migrations/env.py +76 -0
mlflow/server/auth/db/migrations/versions/8606fa83a998_initial_migration.py +51 -0
mlflow/server/auth/db/migrations/versions/__init__.py +0 -0
mlflow/server/auth/db/models.py +67 -0
mlflow/server/auth/db/utils.py +37 -0
mlflow/server/auth/entities.py +165 -0
mlflow/server/auth/logo.py +14 -0
mlflow/server/auth/permissions.py +65 -0
mlflow/server/auth/routes.py +18 -0
mlflow/server/auth/sqlalchemy_store.py +263 -0
mlflow/server/graphql/__init__.py +0 -0
mlflow/server/graphql/autogenerated_graphql_schema.py +353 -0
mlflow/server/graphql/graphql_custom_scalars.py +24 -0
mlflow/server/graphql/graphql_errors.py +15 -0
mlflow/server/graphql/graphql_no_batching.py +89 -0
mlflow/server/graphql/graphql_schema_extensions.py +74 -0
mlflow/server/handlers.py +3217 -0
mlflow/server/prometheus_exporter.py +17 -0
mlflow/server/validation.py +30 -0
mlflow/shap/__init__.py +691 -0
mlflow/sklearn/__init__.py +1994 -0
mlflow/sklearn/utils.py +1041 -0
mlflow/smolagents/__init__.py +66 -0
mlflow/smolagents/autolog.py +139 -0
mlflow/smolagents/chat.py +29 -0
mlflow/store/__init__.py +10 -0
mlflow/store/_unity_catalog/__init__.py +1 -0
mlflow/store/_unity_catalog/lineage/__init__.py +1 -0
mlflow/store/_unity_catalog/lineage/constants.py +2 -0
mlflow/store/_unity_catalog/registry/__init__.py +6 -0
mlflow/store/_unity_catalog/registry/prompt_info.py +75 -0
mlflow/store/_unity_catalog/registry/rest_store.py +1740 -0
mlflow/store/_unity_catalog/registry/uc_oss_rest_store.py +507 -0
mlflow/store/_unity_catalog/registry/utils.py +121 -0
mlflow/store/artifact/__init__.py +0 -0
mlflow/store/artifact/artifact_repo.py +472 -0
mlflow/store/artifact/artifact_repository_registry.py +154 -0
mlflow/store/artifact/azure_blob_artifact_repo.py +275 -0
mlflow/store/artifact/azure_data_lake_artifact_repo.py +295 -0
mlflow/store/artifact/cli.py +141 -0
mlflow/store/artifact/cloud_artifact_repo.py +332 -0
mlflow/store/artifact/databricks_artifact_repo.py +729 -0
mlflow/store/artifact/databricks_artifact_repo_resources.py +301 -0
mlflow/store/artifact/databricks_logged_model_artifact_repo.py +93 -0
mlflow/store/artifact/databricks_models_artifact_repo.py +216 -0
mlflow/store/artifact/databricks_sdk_artifact_repo.py +134 -0
mlflow/store/artifact/databricks_sdk_models_artifact_repo.py +97 -0
mlflow/store/artifact/dbfs_artifact_repo.py +240 -0
mlflow/store/artifact/ftp_artifact_repo.py +132 -0
mlflow/store/artifact/gcs_artifact_repo.py +296 -0
mlflow/store/artifact/hdfs_artifact_repo.py +209 -0
mlflow/store/artifact/http_artifact_repo.py +218 -0
mlflow/store/artifact/local_artifact_repo.py +142 -0
mlflow/store/artifact/mlflow_artifacts_repo.py +94 -0
mlflow/store/artifact/models_artifact_repo.py +259 -0
mlflow/store/artifact/optimized_s3_artifact_repo.py +356 -0
mlflow/store/artifact/presigned_url_artifact_repo.py +173 -0
mlflow/store/artifact/r2_artifact_repo.py +70 -0
mlflow/store/artifact/runs_artifact_repo.py +265 -0
mlflow/store/artifact/s3_artifact_repo.py +330 -0
mlflow/store/artifact/sftp_artifact_repo.py +141 -0
mlflow/store/artifact/uc_volume_artifact_repo.py +76 -0
mlflow/store/artifact/unity_catalog_models_artifact_repo.py +168 -0
mlflow/store/artifact/unity_catalog_oss_models_artifact_repo.py +168 -0
mlflow/store/artifact/utils/__init__.py +0 -0
mlflow/store/artifact/utils/models.py +148 -0
mlflow/store/db/__init__.py +0 -0
mlflow/store/db/base_sql_model.py +3 -0
mlflow/store/db/db_types.py +10 -0
mlflow/store/db/utils.py +314 -0
mlflow/store/db_migrations/__init__.py +0 -0
mlflow/store/db_migrations/alembic.ini +74 -0
mlflow/store/db_migrations/env.py +84 -0
mlflow/store/db_migrations/versions/0584bdc529eb_add_cascading_deletion_to_datasets_from_experiments.py +88 -0
mlflow/store/db_migrations/versions/0a8213491aaa_drop_duplicate_killed_constraint.py +49 -0
mlflow/store/db_migrations/versions/0c779009ac13_add_deleted_time_field_to_runs_table.py +24 -0
mlflow/store/db_migrations/versions/181f10493468_allow_nulls_for_metric_values.py +35 -0
mlflow/store/db_migrations/versions/27a6a02d2cf1_add_model_version_tags_table.py +38 -0
mlflow/store/db_migrations/versions/2b4d017a5e9b_add_model_registry_tables_to_db.py +77 -0
mlflow/store/db_migrations/versions/2d6e25af4d3e_increase_max_param_val_length.py +33 -0
mlflow/store/db_migrations/versions/3500859a5d39_add_model_aliases_table.py +50 -0
mlflow/store/db_migrations/versions/39d1c3be5f05_add_is_nan_constraint_for_metrics_tables_if_necessary.py +41 -0
mlflow/store/db_migrations/versions/400f98739977_add_logged_model_tables.py +123 -0
mlflow/store/db_migrations/versions/4465047574b1_increase_max_dataset_schema_size.py +38 -0
mlflow/store/db_migrations/versions/451aebb31d03_add_metric_step.py +35 -0
mlflow/store/db_migrations/versions/5b0e9adcef9c_add_cascade_deletion_to_trace_tables_fk.py +40 -0
mlflow/store/db_migrations/versions/6953534de441_add_step_to_inputs_table.py +25 -0
mlflow/store/db_migrations/versions/728d730b5ebd_add_registered_model_tags_table.py +38 -0
mlflow/store/db_migrations/versions/7ac759974ad8_update_run_tags_with_larger_limit.py +36 -0
mlflow/store/db_migrations/versions/7f2a7d5fae7d_add_datasets_inputs_input_tags_tables.py +82 -0
mlflow/store/db_migrations/versions/84291f40a231_add_run_link_to_model_version.py +26 -0
mlflow/store/db_migrations/versions/867495a8f9d4_add_trace_tables.py +90 -0
mlflow/store/db_migrations/versions/89d4b8295536_create_latest_metrics_table.py +169 -0
mlflow/store/db_migrations/versions/90e64c465722_migrate_user_column_to_tags.py +64 -0
mlflow/store/db_migrations/versions/97727af70f4d_creation_time_last_update_time_experiments.py +25 -0
mlflow/store/db_migrations/versions/__init__.py +0 -0
mlflow/store/db_migrations/versions/a8c4a736bde6_allow_nulls_for_run_id.py +27 -0
mlflow/store/db_migrations/versions/acf3f17fdcc7_add_storage_location_field_to_model_.py +29 -0
mlflow/store/db_migrations/versions/bd07f7e963c5_create_index_on_run_uuid.py +26 -0
mlflow/store/db_migrations/versions/bda7b8c39065_increase_model_version_tag_value_limit.py +38 -0
mlflow/store/db_migrations/versions/c48cb773bb87_reset_default_value_for_is_nan_in_metrics_table_for_mysql.py +41 -0
mlflow/store/db_migrations/versions/cbc13b556ace_add_v3_trace_schema_columns.py +31 -0
mlflow/store/db_migrations/versions/cc1f77228345_change_param_value_length_to_500.py +34 -0
mlflow/store/db_migrations/versions/cfd24bdc0731_update_run_status_constraint_with_killed.py +78 -0
mlflow/store/db_migrations/versions/df50e92ffc5e_add_experiment_tags_table.py +38 -0
mlflow/store/db_migrations/versions/f5a4f2784254_increase_run_tag_value_limit.py +36 -0
mlflow/store/entities/__init__.py +3 -0
mlflow/store/entities/paged_list.py +18 -0
mlflow/store/model_registry/__init__.py +10 -0
mlflow/store/model_registry/abstract_store.py +1081 -0
mlflow/store/model_registry/base_rest_store.py +44 -0
mlflow/store/model_registry/databricks_workspace_model_registry_rest_store.py +37 -0
mlflow/store/model_registry/dbmodels/__init__.py +0 -0
mlflow/store/model_registry/dbmodels/models.py +206 -0
mlflow/store/model_registry/file_store.py +1091 -0
mlflow/store/model_registry/rest_store.py +481 -0
mlflow/store/model_registry/sqlalchemy_store.py +1286 -0
mlflow/store/tracking/__init__.py +23 -0
mlflow/store/tracking/abstract_store.py +816 -0
mlflow/store/tracking/dbmodels/__init__.py +0 -0
mlflow/store/tracking/dbmodels/initial_models.py +243 -0
mlflow/store/tracking/dbmodels/models.py +1073 -0
mlflow/store/tracking/file_store.py +2438 -0
mlflow/store/tracking/postgres_managed_identity.py +146 -0
mlflow/store/tracking/rest_store.py +1131 -0
mlflow/store/tracking/sqlalchemy_store.py +2785 -0
mlflow/system_metrics/__init__.py +61 -0
mlflow/system_metrics/metrics/__init__.py +0 -0
mlflow/system_metrics/metrics/base_metrics_monitor.py +32 -0
mlflow/system_metrics/metrics/cpu_monitor.py +23 -0
mlflow/system_metrics/metrics/disk_monitor.py +21 -0
mlflow/system_metrics/metrics/gpu_monitor.py +71 -0
mlflow/system_metrics/metrics/network_monitor.py +34 -0
mlflow/system_metrics/metrics/rocm_monitor.py +123 -0
mlflow/system_metrics/system_metrics_monitor.py +198 -0
mlflow/tracing/__init__.py +16 -0
mlflow/tracing/assessment.py +356 -0
mlflow/tracing/client.py +531 -0
mlflow/tracing/config.py +125 -0
mlflow/tracing/constant.py +105 -0
mlflow/tracing/destination.py +81 -0
mlflow/tracing/display/__init__.py +40 -0
mlflow/tracing/display/display_handler.py +196 -0
mlflow/tracing/export/async_export_queue.py +186 -0
mlflow/tracing/export/inference_table.py +138 -0
mlflow/tracing/export/mlflow_v3.py +137 -0
mlflow/tracing/export/utils.py +70 -0
mlflow/tracing/fluent.py +1417 -0
mlflow/tracing/processor/base_mlflow.py +199 -0
mlflow/tracing/processor/inference_table.py +175 -0
mlflow/tracing/processor/mlflow_v3.py +47 -0
mlflow/tracing/processor/otel.py +73 -0
mlflow/tracing/provider.py +487 -0
mlflow/tracing/trace_manager.py +200 -0
mlflow/tracing/utils/__init__.py +616 -0
mlflow/tracing/utils/artifact_utils.py +28 -0
mlflow/tracing/utils/copy.py +55 -0
mlflow/tracing/utils/environment.py +55 -0
mlflow/tracing/utils/exception.py +21 -0
mlflow/tracing/utils/once.py +35 -0
mlflow/tracing/utils/otlp.py +63 -0
mlflow/tracing/utils/processor.py +54 -0
mlflow/tracing/utils/search.py +292 -0
mlflow/tracing/utils/timeout.py +250 -0
mlflow/tracing/utils/token.py +19 -0
mlflow/tracing/utils/truncation.py +124 -0
mlflow/tracing/utils/warning.py +76 -0
mlflow/tracking/__init__.py +39 -0
mlflow/tracking/_model_registry/__init__.py +1 -0
mlflow/tracking/_model_registry/client.py +764 -0
mlflow/tracking/_model_registry/fluent.py +853 -0
mlflow/tracking/_model_registry/registry.py +67 -0
mlflow/tracking/_model_registry/utils.py +251 -0
mlflow/tracking/_tracking_service/__init__.py +0 -0
mlflow/tracking/_tracking_service/client.py +883 -0
mlflow/tracking/_tracking_service/registry.py +56 -0
mlflow/tracking/_tracking_service/utils.py +275 -0
mlflow/tracking/artifact_utils.py +179 -0
mlflow/tracking/client.py +5900 -0
mlflow/tracking/context/__init__.py +0 -0
mlflow/tracking/context/abstract_context.py +35 -0
mlflow/tracking/context/databricks_cluster_context.py +15 -0
mlflow/tracking/context/databricks_command_context.py +15 -0
mlflow/tracking/context/databricks_job_context.py +49 -0
mlflow/tracking/context/databricks_notebook_context.py +41 -0
mlflow/tracking/context/databricks_repo_context.py +43 -0
mlflow/tracking/context/default_context.py +51 -0
mlflow/tracking/context/git_context.py +32 -0
mlflow/tracking/context/registry.py +98 -0
mlflow/tracking/context/system_environment_context.py +15 -0
mlflow/tracking/default_experiment/__init__.py +1 -0
mlflow/tracking/default_experiment/abstract_context.py +43 -0
mlflow/tracking/default_experiment/databricks_notebook_experiment_provider.py +44 -0
mlflow/tracking/default_experiment/registry.py +75 -0
mlflow/tracking/fluent.py +3595 -0
mlflow/tracking/metric_value_conversion_utils.py +93 -0
mlflow/tracking/multimedia.py +206 -0
mlflow/tracking/registry.py +86 -0
mlflow/tracking/request_auth/__init__.py +0 -0
mlflow/tracking/request_auth/abstract_request_auth_provider.py +34 -0
mlflow/tracking/request_auth/registry.py +60 -0
mlflow/tracking/request_header/__init__.py +0 -0
mlflow/tracking/request_header/abstract_request_header_provider.py +36 -0
mlflow/tracking/request_header/databricks_request_header_provider.py +38 -0
mlflow/tracking/request_header/default_request_header_provider.py +17 -0
mlflow/tracking/request_header/registry.py +79 -0
mlflow/transformers/__init__.py +2982 -0
mlflow/transformers/flavor_config.py +258 -0
mlflow/transformers/hub_utils.py +83 -0
mlflow/transformers/llm_inference_utils.py +468 -0
mlflow/transformers/model_io.py +301 -0
mlflow/transformers/peft.py +51 -0
mlflow/transformers/signature.py +183 -0
mlflow/transformers/torch_utils.py +55 -0
mlflow/types/__init__.py +21 -0
mlflow/types/agent.py +270 -0
mlflow/types/chat.py +240 -0
mlflow/types/llm.py +935 -0
mlflow/types/responses.py +139 -0
mlflow/types/responses_helpers.py +416 -0
mlflow/types/schema.py +1505 -0
mlflow/types/type_hints.py +647 -0
mlflow/types/utils.py +753 -0
mlflow/utils/__init__.py +283 -0
mlflow/utils/_capture_modules.py +256 -0
mlflow/utils/_capture_transformers_modules.py +75 -0
mlflow/utils/_spark_utils.py +201 -0
mlflow/utils/_unity_catalog_oss_utils.py +97 -0
mlflow/utils/_unity_catalog_utils.py +479 -0
mlflow/utils/annotations.py +218 -0
mlflow/utils/arguments_utils.py +16 -0
mlflow/utils/async_logging/__init__.py +1 -0
mlflow/utils/async_logging/async_artifacts_logging_queue.py +258 -0
mlflow/utils/async_logging/async_logging_queue.py +366 -0
mlflow/utils/async_logging/run_artifact.py +38 -0
mlflow/utils/async_logging/run_batch.py +58 -0
mlflow/utils/async_logging/run_operations.py +49 -0
mlflow/utils/autologging_utils/__init__.py +737 -0
mlflow/utils/autologging_utils/client.py +432 -0
mlflow/utils/autologging_utils/config.py +33 -0
mlflow/utils/autologging_utils/events.py +294 -0
mlflow/utils/autologging_utils/logging_and_warnings.py +328 -0
mlflow/utils/autologging_utils/metrics_queue.py +71 -0
mlflow/utils/autologging_utils/safety.py +1104 -0
mlflow/utils/autologging_utils/versioning.py +95 -0
mlflow/utils/checkpoint_utils.py +206 -0
mlflow/utils/class_utils.py +6 -0
mlflow/utils/cli_args.py +257 -0
mlflow/utils/conda.py +354 -0
mlflow/utils/credentials.py +231 -0
mlflow/utils/data_utils.py +17 -0
mlflow/utils/databricks_utils.py +1436 -0
mlflow/utils/docstring_utils.py +477 -0
mlflow/utils/doctor.py +133 -0
mlflow/utils/download_cloud_file_chunk.py +43 -0
mlflow/utils/env_manager.py +16 -0
mlflow/utils/env_pack.py +131 -0
mlflow/utils/environment.py +1009 -0
mlflow/utils/exception_utils.py +14 -0
mlflow/utils/file_utils.py +978 -0
mlflow/utils/git_utils.py +77 -0
mlflow/utils/gorilla.py +797 -0
mlflow/utils/import_hooks/__init__.py +363 -0
mlflow/utils/lazy_load.py +51 -0
mlflow/utils/logging_utils.py +168 -0
mlflow/utils/mime_type_utils.py +58 -0
mlflow/utils/mlflow_tags.py +103 -0
mlflow/utils/model_utils.py +486 -0
mlflow/utils/name_utils.py +346 -0
mlflow/utils/nfs_on_spark.py +62 -0
mlflow/utils/openai_utils.py +164 -0
mlflow/utils/os.py +12 -0
mlflow/utils/oss_registry_utils.py +29 -0
mlflow/utils/plugins.py +17 -0
mlflow/utils/process.py +182 -0
mlflow/utils/promptlab_utils.py +146 -0
mlflow/utils/proto_json_utils.py +743 -0
mlflow/utils/pydantic_utils.py +54 -0
mlflow/utils/request_utils.py +279 -0
mlflow/utils/requirements_utils.py +704 -0
mlflow/utils/rest_utils.py +673 -0
mlflow/utils/search_logged_model_utils.py +127 -0
mlflow/utils/search_utils.py +2111 -0
mlflow/utils/secure_loading.py +221 -0
mlflow/utils/security_validation.py +384 -0
mlflow/utils/server_cli_utils.py +61 -0
mlflow/utils/spark_utils.py +15 -0
mlflow/utils/string_utils.py +138 -0
mlflow/utils/thread_utils.py +63 -0
mlflow/utils/time.py +54 -0
mlflow/utils/timeout.py +42 -0
mlflow/utils/uri.py +572 -0
mlflow/utils/validation.py +662 -0
mlflow/utils/virtualenv.py +458 -0
mlflow/utils/warnings_utils.py +25 -0
mlflow/utils/yaml_utils.py +179 -0
mlflow/version.py +24 -0

mlflow/protos/unity_catalog_prompt_messages_pb2.py ADDED Viewed

@@ -0,0 +1,447 @@
+import google.protobuf
+from packaging.version import Version
+if Version(google.protobuf.__version__).major >= 5:
+  # -*- coding: utf-8 -*-
+  # Generated by the protocol buffer compiler.  DO NOT EDIT!
+  # source: unity_catalog_prompt_messages.proto
+  # Protobuf Python Version: 5.26.0
+  """Generated protocol buffer code."""
+  from google.protobuf import descriptor as _descriptor
+  from google.protobuf import descriptor_pool as _descriptor_pool
+  from google.protobuf import symbol_database as _symbol_database
+  from google.protobuf.internal import builder as _builder
+  # @@protoc_insertion_point(imports)
+  _sym_db = _symbol_database.Default()
+  from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+  from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+  from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+  from . import databricks_pb2 as databricks__pb2
+  from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2
+  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#unity_catalog_prompt_messages.proto\x12\x13mlflow.unitycatalog\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x10\x64\x61tabricks.proto\x1a\x15scalapb/scalapb.proto\"\x95\x02\n\x06Prompt\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x12\x63reation_timestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12:\n\x16last_updated_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12\x31\n\x07\x61liases\x18\x06 \x03(\x0b\x32 .mlflow.unitycatalog.PromptAlias\x12,\n\x04tags\x18\x07 \x03(\x0b\x32\x1e.mlflow.unitycatalog.PromptTagJ\x04\x08\x05\x10\x06R\rexperiment_id\"\xb1\x02\n\rPromptVersion\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12\x36\n\x12\x63reation_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12:\n\x16last_updated_timestamp\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x13\n\x0b\x64\x65scription\x18\x05 \x01(\t\x12\x10\n\x08template\x18\x06 \x01(\t\x12\x31\n\x07\x61liases\x18\x07 \x03(\x0b\x32 .mlflow.unitycatalog.PromptAlias\x12\x33\n\x04tags\x18\x08 \x03(\x0b\x32%.mlflow.unitycatalog.PromptVersionTag\"\'\n\tPromptTag\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\".\n\x10PromptVersionTag\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"-\n\x0bPromptAlias\x12\r\n\x05\x61lias\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\"?\n\x12UnityCatalogSchema\x12\x14\n\x0c\x63\x61talog_name\x18\x04 \x01(\t\x12\x13\n\x0bschema_name\x18\x05 \x01(\t\"P\n\x13\x43reatePromptRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x06prompt\x18\x02 \x01(\x0b\x32\x1b.mlflow.unitycatalog.Prompt\"P\n\x13UpdatePromptRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x06prompt\x18\x02 \x01(\x0b\x32\x1b.mlflow.unitycatalog.Prompt\"#\n\x13\x44\x65letePromptRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\"\x16\n\x14\x44\x65letePromptResponse\" \n\x10GetPromptRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\"\xb3\x01\n\x14SearchPromptsRequest\x12\x0e\n\x06\x66ilter\x18\x01 \x01(\t\x12\x41\n\x0e\x63\x61talog_schema\x18\x03 \x01(\x0b\x32\'.mlflow.unitycatalog.UnityCatalogSchemaH\x00\x12\x13\n\x0bmax_results\x18\x04 \x01(\x03\x12\x12\n\npage_token\x18\x05 \x01(\tB\n\n\x08locationJ\x04\x08\x02\x10\x03R\rexperiment_id\"^\n\x15SearchPromptsResponse\x12,\n\x07prompts\x18\x01 \x03(\x0b\x32\x1b.mlflow.unitycatalog.Prompt\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"u\n\x1a\x43reatePromptVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12:\n\x0eprompt_version\x18\x03 \x01(\x0b\x32\".mlflow.unitycatalog.PromptVersionJ\x04\x08\x02\x10\x03R\x07version\"w\n\x1aUpdatePromptVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12:\n\x0eprompt_version\x18\x03 \x01(\x0b\x32\".mlflow.unitycatalog.PromptVersion\";\n\x1a\x44\x65letePromptVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\"8\n\x17GetPromptVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\"T\n\x1bSearchPromptVersionsRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0bmax_results\x18\x02 \x01(\x03\x12\x12\n\npage_token\x18\x03 \x01(\t\"t\n\x1cSearchPromptVersionsResponse\x12;\n\x0fprompt_versions\x18\x01 \x03(\x0b\x32\".mlflow.unitycatalog.PromptVersion\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"E\n\x15SetPromptAliasRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05\x61lias\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\t\"7\n\x18\x44\x65letePromptAliasRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05\x61lias\x18\x02 \x01(\t\"=\n\x1eGetPromptVersionByAliasRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05\x61lias\x18\x02 \x01(\t\"?\n\x13SetPromptTagRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0b\n\x03key\x18\x02 \x01(\t\x12\r\n\x05value\x18\x03 \x01(\t\"3\n\x16\x44\x65letePromptTagRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0b\n\x03key\x18\x02 \x01(\t\"W\n\x1aSetPromptVersionTagRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12\x0b\n\x03key\x18\x03 \x01(\t\x12\r\n\x05value\x18\x04 \x01(\t\"K\n\x1d\x44\x65letePromptVersionTagRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12\x0b\n\x03key\x18\x03 \x01(\t\"\xbe\x01\n!LinkPromptVersionsToModelsRequest\x12\x44\n\x0fprompt_versions\x18\x01 \x03(\x0b\x32+.mlflow.unitycatalog.PromptVersionLinkEntry\x12\x11\n\tmodel_ids\x18\x02 \x03(\t:@\xe2?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]\"\xb7\x01\n\x1aLinkPromptsToTracesRequest\x12\x44\n\x0fprompt_versions\x18\x01 \x03(\x0b\x32+.mlflow.unitycatalog.PromptVersionLinkEntry\x12\x11\n\ttrace_ids\x18\x02 \x03(\t:@\xe2?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]\"\xba\x01\n\x1fLinkPromptVersionsToRunsRequest\x12\x44\n\x0fprompt_versions\x18\x01 \x03(\x0b\x32+.mlflow.unitycatalog.PromptVersionLinkEntry\x12\x0f\n\x07run_ids\x18\x02 \x03(\t:@\xe2?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]\"C\n\x16PromptVersionLinkEntry\x12\x12\n\x04name\x18\x01 \x01(\tB\x04\xf8\x86\x19\x01\x12\x15\n\x07version\x18\x02 \x01(\tB\x04\xf8\x86\x19\x01\x42\x34\n\'com.databricks.api.proto.managedcatalog\x90\x01\x01\xa0\x01\x01\xe2?\x02\x10\x01')
+  _globals = globals()
+  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+  _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'unity_catalog_prompt_messages_pb2', _globals)
+  if not _descriptor._USE_C_DESCRIPTORS:
+    _globals['DESCRIPTOR']._loaded_options = None
+    _globals['DESCRIPTOR']._serialized_options = b'\n\'com.databricks.api.proto.managedcatalog\220\001\001\240\001\001\342?\002\020\001'
+    _globals['_LINKPROMPTVERSIONSTOMODELSREQUEST']._loaded_options = None
+    _globals['_LINKPROMPTVERSIONSTOMODELSREQUEST']._serialized_options = b'\342?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]'
+    _globals['_LINKPROMPTSTOTRACESREQUEST']._loaded_options = None
+    _globals['_LINKPROMPTSTOTRACESREQUEST']._serialized_options = b'\342?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]'
+    _globals['_LINKPROMPTVERSIONSTORUNSREQUEST']._loaded_options = None
+    _globals['_LINKPROMPTVERSIONSTORUNSREQUEST']._serialized_options = b'\342?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]'
+    _globals['_PROMPTVERSIONLINKENTRY'].fields_by_name['name']._loaded_options = None
+    _globals['_PROMPTVERSIONLINKENTRY'].fields_by_name['name']._serialized_options = b'\370\206\031\001'
+    _globals['_PROMPTVERSIONLINKENTRY'].fields_by_name['version']._loaded_options = None
+    _globals['_PROMPTVERSIONLINKENTRY'].fields_by_name['version']._serialized_options = b'\370\206\031\001'
+    _globals['_PROMPT']._serialized_start=194
+    _globals['_PROMPT']._serialized_end=471
+    _globals['_PROMPTVERSION']._serialized_start=474
+    _globals['_PROMPTVERSION']._serialized_end=779
+    _globals['_PROMPTTAG']._serialized_start=781
+    _globals['_PROMPTTAG']._serialized_end=820
+    _globals['_PROMPTVERSIONTAG']._serialized_start=822
+    _globals['_PROMPTVERSIONTAG']._serialized_end=868
+    _globals['_PROMPTALIAS']._serialized_start=870
+    _globals['_PROMPTALIAS']._serialized_end=915
+    _globals['_UNITYCATALOGSCHEMA']._serialized_start=917
+    _globals['_UNITYCATALOGSCHEMA']._serialized_end=980
+    _globals['_CREATEPROMPTREQUEST']._serialized_start=982
+    _globals['_CREATEPROMPTREQUEST']._serialized_end=1062
+    _globals['_UPDATEPROMPTREQUEST']._serialized_start=1064
+    _globals['_UPDATEPROMPTREQUEST']._serialized_end=1144
+    _globals['_DELETEPROMPTREQUEST']._serialized_start=1146
+    _globals['_DELETEPROMPTREQUEST']._serialized_end=1181
+    _globals['_DELETEPROMPTRESPONSE']._serialized_start=1183
+    _globals['_DELETEPROMPTRESPONSE']._serialized_end=1205
+    _globals['_GETPROMPTREQUEST']._serialized_start=1207
+    _globals['_GETPROMPTREQUEST']._serialized_end=1239
+    _globals['_SEARCHPROMPTSREQUEST']._serialized_start=1242
+    _globals['_SEARCHPROMPTSREQUEST']._serialized_end=1421
+    _globals['_SEARCHPROMPTSRESPONSE']._serialized_start=1423
+    _globals['_SEARCHPROMPTSRESPONSE']._serialized_end=1517
+    _globals['_CREATEPROMPTVERSIONREQUEST']._serialized_start=1519
+    _globals['_CREATEPROMPTVERSIONREQUEST']._serialized_end=1636
+    _globals['_UPDATEPROMPTVERSIONREQUEST']._serialized_start=1638
+    _globals['_UPDATEPROMPTVERSIONREQUEST']._serialized_end=1757
+    _globals['_DELETEPROMPTVERSIONREQUEST']._serialized_start=1759
+    _globals['_DELETEPROMPTVERSIONREQUEST']._serialized_end=1818
+    _globals['_GETPROMPTVERSIONREQUEST']._serialized_start=1820
+    _globals['_GETPROMPTVERSIONREQUEST']._serialized_end=1876
+    _globals['_SEARCHPROMPTVERSIONSREQUEST']._serialized_start=1878
+    _globals['_SEARCHPROMPTVERSIONSREQUEST']._serialized_end=1962
+    _globals['_SEARCHPROMPTVERSIONSRESPONSE']._serialized_start=1964
+    _globals['_SEARCHPROMPTVERSIONSRESPONSE']._serialized_end=2080
+    _globals['_SETPROMPTALIASREQUEST']._serialized_start=2082
+    _globals['_SETPROMPTALIASREQUEST']._serialized_end=2151
+    _globals['_DELETEPROMPTALIASREQUEST']._serialized_start=2153
+    _globals['_DELETEPROMPTALIASREQUEST']._serialized_end=2208
+    _globals['_GETPROMPTVERSIONBYALIASREQUEST']._serialized_start=2210
+    _globals['_GETPROMPTVERSIONBYALIASREQUEST']._serialized_end=2271
+    _globals['_SETPROMPTTAGREQUEST']._serialized_start=2273
+    _globals['_SETPROMPTTAGREQUEST']._serialized_end=2336
+    _globals['_DELETEPROMPTTAGREQUEST']._serialized_start=2338
+    _globals['_DELETEPROMPTTAGREQUEST']._serialized_end=2389
+    _globals['_SETPROMPTVERSIONTAGREQUEST']._serialized_start=2391
+    _globals['_SETPROMPTVERSIONTAGREQUEST']._serialized_end=2478
+    _globals['_DELETEPROMPTVERSIONTAGREQUEST']._serialized_start=2480
+    _globals['_DELETEPROMPTVERSIONTAGREQUEST']._serialized_end=2555
+    _globals['_LINKPROMPTVERSIONSTOMODELSREQUEST']._serialized_start=2558
+    _globals['_LINKPROMPTVERSIONSTOMODELSREQUEST']._serialized_end=2748
+    _globals['_LINKPROMPTSTOTRACESREQUEST']._serialized_start=2751
+    _globals['_LINKPROMPTSTOTRACESREQUEST']._serialized_end=2934
+    _globals['_LINKPROMPTVERSIONSTORUNSREQUEST']._serialized_start=2937
+    _globals['_LINKPROMPTVERSIONSTORUNSREQUEST']._serialized_end=3123
+    _globals['_PROMPTVERSIONLINKENTRY']._serialized_start=3125
+    _globals['_PROMPTVERSIONLINKENTRY']._serialized_end=3192
+  # @@protoc_insertion_point(module_scope)
+else:
+  # -*- coding: utf-8 -*-
+  # Generated by the protocol buffer compiler.  DO NOT EDIT!
+  # source: unity_catalog_prompt_messages.proto
+  """Generated protocol buffer code."""
+  from google.protobuf import descriptor as _descriptor
+  from google.protobuf import descriptor_pool as _descriptor_pool
+  from google.protobuf import message as _message
+  from google.protobuf import reflection as _reflection
+  from google.protobuf import symbol_database as _symbol_database
+  # @@protoc_insertion_point(imports)
+  _sym_db = _symbol_database.Default()
+  from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+  from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+  from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+  from . import databricks_pb2 as databricks__pb2
+  from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2
+  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#unity_catalog_prompt_messages.proto\x12\x13mlflow.unitycatalog\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x10\x64\x61tabricks.proto\x1a\x15scalapb/scalapb.proto\"\x95\x02\n\x06Prompt\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x12\x63reation_timestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12:\n\x16last_updated_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12\x31\n\x07\x61liases\x18\x06 \x03(\x0b\x32 .mlflow.unitycatalog.PromptAlias\x12,\n\x04tags\x18\x07 \x03(\x0b\x32\x1e.mlflow.unitycatalog.PromptTagJ\x04\x08\x05\x10\x06R\rexperiment_id\"\xb1\x02\n\rPromptVersion\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12\x36\n\x12\x63reation_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12:\n\x16last_updated_timestamp\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x13\n\x0b\x64\x65scription\x18\x05 \x01(\t\x12\x10\n\x08template\x18\x06 \x01(\t\x12\x31\n\x07\x61liases\x18\x07 \x03(\x0b\x32 .mlflow.unitycatalog.PromptAlias\x12\x33\n\x04tags\x18\x08 \x03(\x0b\x32%.mlflow.unitycatalog.PromptVersionTag\"\'\n\tPromptTag\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\".\n\x10PromptVersionTag\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"-\n\x0bPromptAlias\x12\r\n\x05\x61lias\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\"?\n\x12UnityCatalogSchema\x12\x14\n\x0c\x63\x61talog_name\x18\x04 \x01(\t\x12\x13\n\x0bschema_name\x18\x05 \x01(\t\"P\n\x13\x43reatePromptRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x06prompt\x18\x02 \x01(\x0b\x32\x1b.mlflow.unitycatalog.Prompt\"P\n\x13UpdatePromptRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x06prompt\x18\x02 \x01(\x0b\x32\x1b.mlflow.unitycatalog.Prompt\"#\n\x13\x44\x65letePromptRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\"\x16\n\x14\x44\x65letePromptResponse\" \n\x10GetPromptRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\"\xb3\x01\n\x14SearchPromptsRequest\x12\x0e\n\x06\x66ilter\x18\x01 \x01(\t\x12\x41\n\x0e\x63\x61talog_schema\x18\x03 \x01(\x0b\x32\'.mlflow.unitycatalog.UnityCatalogSchemaH\x00\x12\x13\n\x0bmax_results\x18\x04 \x01(\x03\x12\x12\n\npage_token\x18\x05 \x01(\tB\n\n\x08locationJ\x04\x08\x02\x10\x03R\rexperiment_id\"^\n\x15SearchPromptsResponse\x12,\n\x07prompts\x18\x01 \x03(\x0b\x32\x1b.mlflow.unitycatalog.Prompt\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"u\n\x1a\x43reatePromptVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12:\n\x0eprompt_version\x18\x03 \x01(\x0b\x32\".mlflow.unitycatalog.PromptVersionJ\x04\x08\x02\x10\x03R\x07version\"w\n\x1aUpdatePromptVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12:\n\x0eprompt_version\x18\x03 \x01(\x0b\x32\".mlflow.unitycatalog.PromptVersion\";\n\x1a\x44\x65letePromptVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\"8\n\x17GetPromptVersionRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\"T\n\x1bSearchPromptVersionsRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0bmax_results\x18\x02 \x01(\x03\x12\x12\n\npage_token\x18\x03 \x01(\t\"t\n\x1cSearchPromptVersionsResponse\x12;\n\x0fprompt_versions\x18\x01 \x03(\x0b\x32\".mlflow.unitycatalog.PromptVersion\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"E\n\x15SetPromptAliasRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05\x61lias\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\t\"7\n\x18\x44\x65letePromptAliasRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05\x61lias\x18\x02 \x01(\t\"=\n\x1eGetPromptVersionByAliasRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05\x61lias\x18\x02 \x01(\t\"?\n\x13SetPromptTagRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0b\n\x03key\x18\x02 \x01(\t\x12\r\n\x05value\x18\x03 \x01(\t\"3\n\x16\x44\x65letePromptTagRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0b\n\x03key\x18\x02 \x01(\t\"W\n\x1aSetPromptVersionTagRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12\x0b\n\x03key\x18\x03 \x01(\t\x12\r\n\x05value\x18\x04 \x01(\t\"K\n\x1d\x44\x65letePromptVersionTagRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12\x0b\n\x03key\x18\x03 \x01(\t\"\xbe\x01\n!LinkPromptVersionsToModelsRequest\x12\x44\n\x0fprompt_versions\x18\x01 \x03(\x0b\x32+.mlflow.unitycatalog.PromptVersionLinkEntry\x12\x11\n\tmodel_ids\x18\x02 \x03(\t:@\xe2?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]\"\xb7\x01\n\x1aLinkPromptsToTracesRequest\x12\x44\n\x0fprompt_versions\x18\x01 \x03(\x0b\x32+.mlflow.unitycatalog.PromptVersionLinkEntry\x12\x11\n\ttrace_ids\x18\x02 \x03(\t:@\xe2?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]\"\xba\x01\n\x1fLinkPromptVersionsToRunsRequest\x12\x44\n\x0fprompt_versions\x18\x01 \x03(\x0b\x32+.mlflow.unitycatalog.PromptVersionLinkEntry\x12\x0f\n\x07run_ids\x18\x02 \x03(\t:@\xe2?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]\"C\n\x16PromptVersionLinkEntry\x12\x12\n\x04name\x18\x01 \x01(\tB\x04\xf8\x86\x19\x01\x12\x15\n\x07version\x18\x02 \x01(\tB\x04\xf8\x86\x19\x01\x42\x34\n\'com.databricks.api.proto.managedcatalog\x90\x01\x01\xa0\x01\x01\xe2?\x02\x10\x01')
+  _PROMPT = DESCRIPTOR.message_types_by_name['Prompt']
+  _PROMPTVERSION = DESCRIPTOR.message_types_by_name['PromptVersion']
+  _PROMPTTAG = DESCRIPTOR.message_types_by_name['PromptTag']
+  _PROMPTVERSIONTAG = DESCRIPTOR.message_types_by_name['PromptVersionTag']
+  _PROMPTALIAS = DESCRIPTOR.message_types_by_name['PromptAlias']
+  _UNITYCATALOGSCHEMA = DESCRIPTOR.message_types_by_name['UnityCatalogSchema']
+  _CREATEPROMPTREQUEST = DESCRIPTOR.message_types_by_name['CreatePromptRequest']
+  _UPDATEPROMPTREQUEST = DESCRIPTOR.message_types_by_name['UpdatePromptRequest']
+  _DELETEPROMPTREQUEST = DESCRIPTOR.message_types_by_name['DeletePromptRequest']
+  _DELETEPROMPTRESPONSE = DESCRIPTOR.message_types_by_name['DeletePromptResponse']
+  _GETPROMPTREQUEST = DESCRIPTOR.message_types_by_name['GetPromptRequest']
+  _SEARCHPROMPTSREQUEST = DESCRIPTOR.message_types_by_name['SearchPromptsRequest']
+  _SEARCHPROMPTSRESPONSE = DESCRIPTOR.message_types_by_name['SearchPromptsResponse']
+  _CREATEPROMPTVERSIONREQUEST = DESCRIPTOR.message_types_by_name['CreatePromptVersionRequest']
+  _UPDATEPROMPTVERSIONREQUEST = DESCRIPTOR.message_types_by_name['UpdatePromptVersionRequest']
+  _DELETEPROMPTVERSIONREQUEST = DESCRIPTOR.message_types_by_name['DeletePromptVersionRequest']
+  _GETPROMPTVERSIONREQUEST = DESCRIPTOR.message_types_by_name['GetPromptVersionRequest']
+  _SEARCHPROMPTVERSIONSREQUEST = DESCRIPTOR.message_types_by_name['SearchPromptVersionsRequest']
+  _SEARCHPROMPTVERSIONSRESPONSE = DESCRIPTOR.message_types_by_name['SearchPromptVersionsResponse']
+  _SETPROMPTALIASREQUEST = DESCRIPTOR.message_types_by_name['SetPromptAliasRequest']
+  _DELETEPROMPTALIASREQUEST = DESCRIPTOR.message_types_by_name['DeletePromptAliasRequest']
+  _GETPROMPTVERSIONBYALIASREQUEST = DESCRIPTOR.message_types_by_name['GetPromptVersionByAliasRequest']
+  _SETPROMPTTAGREQUEST = DESCRIPTOR.message_types_by_name['SetPromptTagRequest']
+  _DELETEPROMPTTAGREQUEST = DESCRIPTOR.message_types_by_name['DeletePromptTagRequest']
+  _SETPROMPTVERSIONTAGREQUEST = DESCRIPTOR.message_types_by_name['SetPromptVersionTagRequest']
+  _DELETEPROMPTVERSIONTAGREQUEST = DESCRIPTOR.message_types_by_name['DeletePromptVersionTagRequest']
+  _LINKPROMPTVERSIONSTOMODELSREQUEST = DESCRIPTOR.message_types_by_name['LinkPromptVersionsToModelsRequest']
+  _LINKPROMPTSTOTRACESREQUEST = DESCRIPTOR.message_types_by_name['LinkPromptsToTracesRequest']
+  _LINKPROMPTVERSIONSTORUNSREQUEST = DESCRIPTOR.message_types_by_name['LinkPromptVersionsToRunsRequest']
+  _PROMPTVERSIONLINKENTRY = DESCRIPTOR.message_types_by_name['PromptVersionLinkEntry']
+  Prompt = _reflection.GeneratedProtocolMessageType('Prompt', (_message.Message,), {
+    'DESCRIPTOR' : _PROMPT,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.Prompt)
+    })
+  _sym_db.RegisterMessage(Prompt)
+  PromptVersion = _reflection.GeneratedProtocolMessageType('PromptVersion', (_message.Message,), {
+    'DESCRIPTOR' : _PROMPTVERSION,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.PromptVersion)
+    })
+  _sym_db.RegisterMessage(PromptVersion)
+  PromptTag = _reflection.GeneratedProtocolMessageType('PromptTag', (_message.Message,), {
+    'DESCRIPTOR' : _PROMPTTAG,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.PromptTag)
+    })
+  _sym_db.RegisterMessage(PromptTag)
+  PromptVersionTag = _reflection.GeneratedProtocolMessageType('PromptVersionTag', (_message.Message,), {
+    'DESCRIPTOR' : _PROMPTVERSIONTAG,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.PromptVersionTag)
+    })
+  _sym_db.RegisterMessage(PromptVersionTag)
+  PromptAlias = _reflection.GeneratedProtocolMessageType('PromptAlias', (_message.Message,), {
+    'DESCRIPTOR' : _PROMPTALIAS,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.PromptAlias)
+    })
+  _sym_db.RegisterMessage(PromptAlias)
+  UnityCatalogSchema = _reflection.GeneratedProtocolMessageType('UnityCatalogSchema', (_message.Message,), {
+    'DESCRIPTOR' : _UNITYCATALOGSCHEMA,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.UnityCatalogSchema)
+    })
+  _sym_db.RegisterMessage(UnityCatalogSchema)
+  CreatePromptRequest = _reflection.GeneratedProtocolMessageType('CreatePromptRequest', (_message.Message,), {
+    'DESCRIPTOR' : _CREATEPROMPTREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.CreatePromptRequest)
+    })
+  _sym_db.RegisterMessage(CreatePromptRequest)
+  UpdatePromptRequest = _reflection.GeneratedProtocolMessageType('UpdatePromptRequest', (_message.Message,), {
+    'DESCRIPTOR' : _UPDATEPROMPTREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.UpdatePromptRequest)
+    })
+  _sym_db.RegisterMessage(UpdatePromptRequest)
+  DeletePromptRequest = _reflection.GeneratedProtocolMessageType('DeletePromptRequest', (_message.Message,), {
+    'DESCRIPTOR' : _DELETEPROMPTREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.DeletePromptRequest)
+    })
+  _sym_db.RegisterMessage(DeletePromptRequest)
+  DeletePromptResponse = _reflection.GeneratedProtocolMessageType('DeletePromptResponse', (_message.Message,), {
+    'DESCRIPTOR' : _DELETEPROMPTRESPONSE,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.DeletePromptResponse)
+    })
+  _sym_db.RegisterMessage(DeletePromptResponse)
+  GetPromptRequest = _reflection.GeneratedProtocolMessageType('GetPromptRequest', (_message.Message,), {
+    'DESCRIPTOR' : _GETPROMPTREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.GetPromptRequest)
+    })
+  _sym_db.RegisterMessage(GetPromptRequest)
+  SearchPromptsRequest = _reflection.GeneratedProtocolMessageType('SearchPromptsRequest', (_message.Message,), {
+    'DESCRIPTOR' : _SEARCHPROMPTSREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.SearchPromptsRequest)
+    })
+  _sym_db.RegisterMessage(SearchPromptsRequest)
+  SearchPromptsResponse = _reflection.GeneratedProtocolMessageType('SearchPromptsResponse', (_message.Message,), {
+    'DESCRIPTOR' : _SEARCHPROMPTSRESPONSE,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.SearchPromptsResponse)
+    })
+  _sym_db.RegisterMessage(SearchPromptsResponse)
+  CreatePromptVersionRequest = _reflection.GeneratedProtocolMessageType('CreatePromptVersionRequest', (_message.Message,), {
+    'DESCRIPTOR' : _CREATEPROMPTVERSIONREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.CreatePromptVersionRequest)
+    })
+  _sym_db.RegisterMessage(CreatePromptVersionRequest)
+  UpdatePromptVersionRequest = _reflection.GeneratedProtocolMessageType('UpdatePromptVersionRequest', (_message.Message,), {
+    'DESCRIPTOR' : _UPDATEPROMPTVERSIONREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.UpdatePromptVersionRequest)
+    })
+  _sym_db.RegisterMessage(UpdatePromptVersionRequest)
+  DeletePromptVersionRequest = _reflection.GeneratedProtocolMessageType('DeletePromptVersionRequest', (_message.Message,), {
+    'DESCRIPTOR' : _DELETEPROMPTVERSIONREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.DeletePromptVersionRequest)
+    })
+  _sym_db.RegisterMessage(DeletePromptVersionRequest)
+  GetPromptVersionRequest = _reflection.GeneratedProtocolMessageType('GetPromptVersionRequest', (_message.Message,), {
+    'DESCRIPTOR' : _GETPROMPTVERSIONREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.GetPromptVersionRequest)
+    })
+  _sym_db.RegisterMessage(GetPromptVersionRequest)
+  SearchPromptVersionsRequest = _reflection.GeneratedProtocolMessageType('SearchPromptVersionsRequest', (_message.Message,), {
+    'DESCRIPTOR' : _SEARCHPROMPTVERSIONSREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.SearchPromptVersionsRequest)
+    })
+  _sym_db.RegisterMessage(SearchPromptVersionsRequest)
+  SearchPromptVersionsResponse = _reflection.GeneratedProtocolMessageType('SearchPromptVersionsResponse', (_message.Message,), {
+    'DESCRIPTOR' : _SEARCHPROMPTVERSIONSRESPONSE,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.SearchPromptVersionsResponse)
+    })
+  _sym_db.RegisterMessage(SearchPromptVersionsResponse)
+  SetPromptAliasRequest = _reflection.GeneratedProtocolMessageType('SetPromptAliasRequest', (_message.Message,), {
+    'DESCRIPTOR' : _SETPROMPTALIASREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.SetPromptAliasRequest)
+    })
+  _sym_db.RegisterMessage(SetPromptAliasRequest)
+  DeletePromptAliasRequest = _reflection.GeneratedProtocolMessageType('DeletePromptAliasRequest', (_message.Message,), {
+    'DESCRIPTOR' : _DELETEPROMPTALIASREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.DeletePromptAliasRequest)
+    })
+  _sym_db.RegisterMessage(DeletePromptAliasRequest)
+  GetPromptVersionByAliasRequest = _reflection.GeneratedProtocolMessageType('GetPromptVersionByAliasRequest', (_message.Message,), {
+    'DESCRIPTOR' : _GETPROMPTVERSIONBYALIASREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.GetPromptVersionByAliasRequest)
+    })
+  _sym_db.RegisterMessage(GetPromptVersionByAliasRequest)
+  SetPromptTagRequest = _reflection.GeneratedProtocolMessageType('SetPromptTagRequest', (_message.Message,), {
+    'DESCRIPTOR' : _SETPROMPTTAGREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.SetPromptTagRequest)
+    })
+  _sym_db.RegisterMessage(SetPromptTagRequest)
+  DeletePromptTagRequest = _reflection.GeneratedProtocolMessageType('DeletePromptTagRequest', (_message.Message,), {
+    'DESCRIPTOR' : _DELETEPROMPTTAGREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.DeletePromptTagRequest)
+    })
+  _sym_db.RegisterMessage(DeletePromptTagRequest)
+  SetPromptVersionTagRequest = _reflection.GeneratedProtocolMessageType('SetPromptVersionTagRequest', (_message.Message,), {
+    'DESCRIPTOR' : _SETPROMPTVERSIONTAGREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.SetPromptVersionTagRequest)
+    })
+  _sym_db.RegisterMessage(SetPromptVersionTagRequest)
+  DeletePromptVersionTagRequest = _reflection.GeneratedProtocolMessageType('DeletePromptVersionTagRequest', (_message.Message,), {
+    'DESCRIPTOR' : _DELETEPROMPTVERSIONTAGREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.DeletePromptVersionTagRequest)
+    })
+  _sym_db.RegisterMessage(DeletePromptVersionTagRequest)
+  LinkPromptVersionsToModelsRequest = _reflection.GeneratedProtocolMessageType('LinkPromptVersionsToModelsRequest', (_message.Message,), {
+    'DESCRIPTOR' : _LINKPROMPTVERSIONSTOMODELSREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.LinkPromptVersionsToModelsRequest)
+    })
+  _sym_db.RegisterMessage(LinkPromptVersionsToModelsRequest)
+  LinkPromptsToTracesRequest = _reflection.GeneratedProtocolMessageType('LinkPromptsToTracesRequest', (_message.Message,), {
+    'DESCRIPTOR' : _LINKPROMPTSTOTRACESREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.LinkPromptsToTracesRequest)
+    })
+  _sym_db.RegisterMessage(LinkPromptsToTracesRequest)
+  LinkPromptVersionsToRunsRequest = _reflection.GeneratedProtocolMessageType('LinkPromptVersionsToRunsRequest', (_message.Message,), {
+    'DESCRIPTOR' : _LINKPROMPTVERSIONSTORUNSREQUEST,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.LinkPromptVersionsToRunsRequest)
+    })
+  _sym_db.RegisterMessage(LinkPromptVersionsToRunsRequest)
+  PromptVersionLinkEntry = _reflection.GeneratedProtocolMessageType('PromptVersionLinkEntry', (_message.Message,), {
+    'DESCRIPTOR' : _PROMPTVERSIONLINKENTRY,
+    '__module__' : 'unity_catalog_prompt_messages_pb2'
+    # @@protoc_insertion_point(class_scope:mlflow.unitycatalog.PromptVersionLinkEntry)
+    })
+  _sym_db.RegisterMessage(PromptVersionLinkEntry)
+  if _descriptor._USE_C_DESCRIPTORS == False:
+    DESCRIPTOR._options = None
+    DESCRIPTOR._serialized_options = b'\n\'com.databricks.api.proto.managedcatalog\220\001\001\240\001\001\342?\002\020\001'
+    _LINKPROMPTVERSIONSTOMODELSREQUEST._options = None
+    _LINKPROMPTVERSIONSTOMODELSREQUEST._serialized_options = b'\342?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]'
+    _LINKPROMPTSTOTRACESREQUEST._options = None
+    _LINKPROMPTSTOTRACESREQUEST._serialized_options = b'\342?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]'
+    _LINKPROMPTVERSIONSTORUNSREQUEST._options = None
+    _LINKPROMPTVERSIONSTORUNSREQUEST._serialized_options = b'\342?=\n;UcModelRegistryRoutableRPC[com.google.protobuf.empty.Empty]'
+    _PROMPTVERSIONLINKENTRY.fields_by_name['name']._options = None
+    _PROMPTVERSIONLINKENTRY.fields_by_name['name']._serialized_options = b'\370\206\031\001'
+    _PROMPTVERSIONLINKENTRY.fields_by_name['version']._options = None
+    _PROMPTVERSIONLINKENTRY.fields_by_name['version']._serialized_options = b'\370\206\031\001'
+    _PROMPT._serialized_start=194
+    _PROMPT._serialized_end=471
+    _PROMPTVERSION._serialized_start=474
+    _PROMPTVERSION._serialized_end=779
+    _PROMPTTAG._serialized_start=781
+    _PROMPTTAG._serialized_end=820
+    _PROMPTVERSIONTAG._serialized_start=822
+    _PROMPTVERSIONTAG._serialized_end=868
+    _PROMPTALIAS._serialized_start=870
+    _PROMPTALIAS._serialized_end=915
+    _UNITYCATALOGSCHEMA._serialized_start=917
+    _UNITYCATALOGSCHEMA._serialized_end=980
+    _CREATEPROMPTREQUEST._serialized_start=982
+    _CREATEPROMPTREQUEST._serialized_end=1062
+    _UPDATEPROMPTREQUEST._serialized_start=1064
+    _UPDATEPROMPTREQUEST._serialized_end=1144
+    _DELETEPROMPTREQUEST._serialized_start=1146
+    _DELETEPROMPTREQUEST._serialized_end=1181
+    _DELETEPROMPTRESPONSE._serialized_start=1183
+    _DELETEPROMPTRESPONSE._serialized_end=1205
+    _GETPROMPTREQUEST._serialized_start=1207
+    _GETPROMPTREQUEST._serialized_end=1239
+    _SEARCHPROMPTSREQUEST._serialized_start=1242
+    _SEARCHPROMPTSREQUEST._serialized_end=1421
+    _SEARCHPROMPTSRESPONSE._serialized_start=1423
+    _SEARCHPROMPTSRESPONSE._serialized_end=1517
+    _CREATEPROMPTVERSIONREQUEST._serialized_start=1519
+    _CREATEPROMPTVERSIONREQUEST._serialized_end=1636
+    _UPDATEPROMPTVERSIONREQUEST._serialized_start=1638
+    _UPDATEPROMPTVERSIONREQUEST._serialized_end=1757
+    _DELETEPROMPTVERSIONREQUEST._serialized_start=1759
+    _DELETEPROMPTVERSIONREQUEST._serialized_end=1818
+    _GETPROMPTVERSIONREQUEST._serialized_start=1820
+    _GETPROMPTVERSIONREQUEST._serialized_end=1876
+    _SEARCHPROMPTVERSIONSREQUEST._serialized_start=1878
+    _SEARCHPROMPTVERSIONSREQUEST._serialized_end=1962
+    _SEARCHPROMPTVERSIONSRESPONSE._serialized_start=1964
+    _SEARCHPROMPTVERSIONSRESPONSE._serialized_end=2080
+    _SETPROMPTALIASREQUEST._serialized_start=2082
+    _SETPROMPTALIASREQUEST._serialized_end=2151
+    _DELETEPROMPTALIASREQUEST._serialized_start=2153
+    _DELETEPROMPTALIASREQUEST._serialized_end=2208
+    _GETPROMPTVERSIONBYALIASREQUEST._serialized_start=2210
+    _GETPROMPTVERSIONBYALIASREQUEST._serialized_end=2271
+    _SETPROMPTTAGREQUEST._serialized_start=2273
+    _SETPROMPTTAGREQUEST._serialized_end=2336
+    _DELETEPROMPTTAGREQUEST._serialized_start=2338
+    _DELETEPROMPTTAGREQUEST._serialized_end=2389
+    _SETPROMPTVERSIONTAGREQUEST._serialized_start=2391
+    _SETPROMPTVERSIONTAGREQUEST._serialized_end=2478
+    _DELETEPROMPTVERSIONTAGREQUEST._serialized_start=2480
+    _DELETEPROMPTVERSIONTAGREQUEST._serialized_end=2555
+    _LINKPROMPTVERSIONSTOMODELSREQUEST._serialized_start=2558
+    _LINKPROMPTVERSIONSTOMODELSREQUEST._serialized_end=2748
+    _LINKPROMPTSTOTRACESREQUEST._serialized_start=2751
+    _LINKPROMPTSTOTRACESREQUEST._serialized_end=2934
+    _LINKPROMPTVERSIONSTORUNSREQUEST._serialized_start=2937
+    _LINKPROMPTVERSIONSTORUNSREQUEST._serialized_end=3123
+    _PROMPTVERSIONLINKENTRY._serialized_start=3125
+    _PROMPTVERSIONLINKENTRY._serialized_end=3192
+  # @@protoc_insertion_point(module_scope)

mlflow/protos/unity_catalog_prompt_messages_pb2_grpc.py ADDED Viewed

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+GRPC_GENERATED_VERSION = '1.71.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in unity_catalog_prompt_messages_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )