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.
- 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
@@ -0,0 +1,785 @@
|
|
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
|
+
import warnings
|
5
|
+
|
6
|
+
import unity_catalog_prompt_messages_pb2 as unity__catalog__prompt__messages__pb2
|
7
|
+
|
8
|
+
GRPC_GENERATED_VERSION = '1.71.0'
|
9
|
+
GRPC_VERSION = grpc.__version__
|
10
|
+
_version_not_supported = False
|
11
|
+
|
12
|
+
try:
|
13
|
+
from grpc._utilities import first_version_is_lower
|
14
|
+
_version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
|
15
|
+
except ImportError:
|
16
|
+
_version_not_supported = True
|
17
|
+
|
18
|
+
if _version_not_supported:
|
19
|
+
raise RuntimeError(
|
20
|
+
f'The grpc package installed is at version {GRPC_VERSION},'
|
21
|
+
+ f' but the generated code in unity_catalog_prompt_service_pb2_grpc.py depends on'
|
22
|
+
+ f' grpcio>={GRPC_GENERATED_VERSION}.'
|
23
|
+
+ f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
|
24
|
+
+ f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
|
25
|
+
)
|
26
|
+
|
27
|
+
|
28
|
+
class UnityCatalogPromptServiceStub(object):
|
29
|
+
"""Missing associated documentation comment in .proto file."""
|
30
|
+
|
31
|
+
def __init__(self, channel):
|
32
|
+
"""Constructor.
|
33
|
+
|
34
|
+
Args:
|
35
|
+
channel: A grpc.Channel.
|
36
|
+
"""
|
37
|
+
self.CreatePrompt = channel.unary_unary(
|
38
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/CreatePrompt',
|
39
|
+
request_serializer=unity__catalog__prompt__messages__pb2.CreatePromptRequest.SerializeToString,
|
40
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.CreatePromptResponse.FromString,
|
41
|
+
_registered_method=True)
|
42
|
+
self.UpdatePrompt = channel.unary_unary(
|
43
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/UpdatePrompt',
|
44
|
+
request_serializer=unity__catalog__prompt__messages__pb2.UpdatePromptRequest.SerializeToString,
|
45
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.UpdatePromptResponse.FromString,
|
46
|
+
_registered_method=True)
|
47
|
+
self.DeletePrompt = channel.unary_unary(
|
48
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePrompt',
|
49
|
+
request_serializer=unity__catalog__prompt__messages__pb2.DeletePromptRequest.SerializeToString,
|
50
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptResponse.FromString,
|
51
|
+
_registered_method=True)
|
52
|
+
self.GetPrompt = channel.unary_unary(
|
53
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/GetPrompt',
|
54
|
+
request_serializer=unity__catalog__prompt__messages__pb2.GetPromptRequest.SerializeToString,
|
55
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.GetPromptResponse.FromString,
|
56
|
+
_registered_method=True)
|
57
|
+
self.SearchPrompts = channel.unary_unary(
|
58
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SearchPrompts',
|
59
|
+
request_serializer=unity__catalog__prompt__messages__pb2.SearchPromptsRequest.SerializeToString,
|
60
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.SearchPromptsResponse.FromString,
|
61
|
+
_registered_method=True)
|
62
|
+
self.CreatePromptVersion = channel.unary_unary(
|
63
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/CreatePromptVersion',
|
64
|
+
request_serializer=unity__catalog__prompt__messages__pb2.CreatePromptVersionRequest.SerializeToString,
|
65
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.CreatePromptVersionResponse.FromString,
|
66
|
+
_registered_method=True)
|
67
|
+
self.UpdatePromptVersion = channel.unary_unary(
|
68
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/UpdatePromptVersion',
|
69
|
+
request_serializer=unity__catalog__prompt__messages__pb2.UpdatePromptVersionRequest.SerializeToString,
|
70
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.UpdatePromptVersionResponse.FromString,
|
71
|
+
_registered_method=True)
|
72
|
+
self.DeletePromptVersion = channel.unary_unary(
|
73
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePromptVersion',
|
74
|
+
request_serializer=unity__catalog__prompt__messages__pb2.DeletePromptVersionRequest.SerializeToString,
|
75
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptVersionResponse.FromString,
|
76
|
+
_registered_method=True)
|
77
|
+
self.GetPromptVersion = channel.unary_unary(
|
78
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/GetPromptVersion',
|
79
|
+
request_serializer=unity__catalog__prompt__messages__pb2.GetPromptVersionRequest.SerializeToString,
|
80
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.GetPromptVersionResponse.FromString,
|
81
|
+
_registered_method=True)
|
82
|
+
self.SearchPromptVersions = channel.unary_unary(
|
83
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SearchPromptVersions',
|
84
|
+
request_serializer=unity__catalog__prompt__messages__pb2.SearchPromptVersionsRequest.SerializeToString,
|
85
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.SearchPromptVersionsResponse.FromString,
|
86
|
+
_registered_method=True)
|
87
|
+
self.SetPromptAlias = channel.unary_unary(
|
88
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SetPromptAlias',
|
89
|
+
request_serializer=unity__catalog__prompt__messages__pb2.SetPromptAliasRequest.SerializeToString,
|
90
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.SetPromptAliasResponse.FromString,
|
91
|
+
_registered_method=True)
|
92
|
+
self.DeletePromptAlias = channel.unary_unary(
|
93
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePromptAlias',
|
94
|
+
request_serializer=unity__catalog__prompt__messages__pb2.DeletePromptAliasRequest.SerializeToString,
|
95
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptAliasResponse.FromString,
|
96
|
+
_registered_method=True)
|
97
|
+
self.GetPromptVersionByAlias = channel.unary_unary(
|
98
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/GetPromptVersionByAlias',
|
99
|
+
request_serializer=unity__catalog__prompt__messages__pb2.GetPromptVersionByAliasRequest.SerializeToString,
|
100
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.GetPromptVersionByAliasResponse.FromString,
|
101
|
+
_registered_method=True)
|
102
|
+
self.SetPromptTag = channel.unary_unary(
|
103
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SetPromptTag',
|
104
|
+
request_serializer=unity__catalog__prompt__messages__pb2.SetPromptTagRequest.SerializeToString,
|
105
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.SetPromptTagResponse.FromString,
|
106
|
+
_registered_method=True)
|
107
|
+
self.DeletePromptTag = channel.unary_unary(
|
108
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePromptTag',
|
109
|
+
request_serializer=unity__catalog__prompt__messages__pb2.DeletePromptTagRequest.SerializeToString,
|
110
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptTagResponse.FromString,
|
111
|
+
_registered_method=True)
|
112
|
+
self.SetPromptVersionTag = channel.unary_unary(
|
113
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SetPromptVersionTag',
|
114
|
+
request_serializer=unity__catalog__prompt__messages__pb2.SetPromptVersionTagRequest.SerializeToString,
|
115
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.SetPromptVersionTagResponse.FromString,
|
116
|
+
_registered_method=True)
|
117
|
+
self.DeletePromptVersionTag = channel.unary_unary(
|
118
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePromptVersionTag',
|
119
|
+
request_serializer=unity__catalog__prompt__messages__pb2.DeletePromptVersionTagRequest.SerializeToString,
|
120
|
+
response_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptVersionTagResponse.FromString,
|
121
|
+
_registered_method=True)
|
122
|
+
|
123
|
+
|
124
|
+
class UnityCatalogPromptServiceServicer(object):
|
125
|
+
"""Missing associated documentation comment in .proto file."""
|
126
|
+
|
127
|
+
def CreatePrompt(self, request, context):
|
128
|
+
"""Missing associated documentation comment in .proto file."""
|
129
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
130
|
+
context.set_details('Method not implemented!')
|
131
|
+
raise NotImplementedError('Method not implemented!')
|
132
|
+
|
133
|
+
def UpdatePrompt(self, request, context):
|
134
|
+
"""Missing associated documentation comment in .proto file."""
|
135
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
136
|
+
context.set_details('Method not implemented!')
|
137
|
+
raise NotImplementedError('Method not implemented!')
|
138
|
+
|
139
|
+
def DeletePrompt(self, request, context):
|
140
|
+
"""Missing associated documentation comment in .proto file."""
|
141
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
142
|
+
context.set_details('Method not implemented!')
|
143
|
+
raise NotImplementedError('Method not implemented!')
|
144
|
+
|
145
|
+
def GetPrompt(self, request, context):
|
146
|
+
"""Missing associated documentation comment in .proto file."""
|
147
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
148
|
+
context.set_details('Method not implemented!')
|
149
|
+
raise NotImplementedError('Method not implemented!')
|
150
|
+
|
151
|
+
def SearchPrompts(self, request, context):
|
152
|
+
"""Missing associated documentation comment in .proto file."""
|
153
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
154
|
+
context.set_details('Method not implemented!')
|
155
|
+
raise NotImplementedError('Method not implemented!')
|
156
|
+
|
157
|
+
def CreatePromptVersion(self, request, context):
|
158
|
+
"""Missing associated documentation comment in .proto file."""
|
159
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
160
|
+
context.set_details('Method not implemented!')
|
161
|
+
raise NotImplementedError('Method not implemented!')
|
162
|
+
|
163
|
+
def UpdatePromptVersion(self, request, context):
|
164
|
+
"""Missing associated documentation comment in .proto file."""
|
165
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
166
|
+
context.set_details('Method not implemented!')
|
167
|
+
raise NotImplementedError('Method not implemented!')
|
168
|
+
|
169
|
+
def DeletePromptVersion(self, request, context):
|
170
|
+
"""Missing associated documentation comment in .proto file."""
|
171
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
172
|
+
context.set_details('Method not implemented!')
|
173
|
+
raise NotImplementedError('Method not implemented!')
|
174
|
+
|
175
|
+
def GetPromptVersion(self, request, context):
|
176
|
+
"""Missing associated documentation comment in .proto file."""
|
177
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
178
|
+
context.set_details('Method not implemented!')
|
179
|
+
raise NotImplementedError('Method not implemented!')
|
180
|
+
|
181
|
+
def SearchPromptVersions(self, request, context):
|
182
|
+
"""Missing associated documentation comment in .proto file."""
|
183
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
184
|
+
context.set_details('Method not implemented!')
|
185
|
+
raise NotImplementedError('Method not implemented!')
|
186
|
+
|
187
|
+
def SetPromptAlias(self, request, context):
|
188
|
+
"""Missing associated documentation comment in .proto file."""
|
189
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
190
|
+
context.set_details('Method not implemented!')
|
191
|
+
raise NotImplementedError('Method not implemented!')
|
192
|
+
|
193
|
+
def DeletePromptAlias(self, request, context):
|
194
|
+
"""Missing associated documentation comment in .proto file."""
|
195
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
196
|
+
context.set_details('Method not implemented!')
|
197
|
+
raise NotImplementedError('Method not implemented!')
|
198
|
+
|
199
|
+
def GetPromptVersionByAlias(self, request, context):
|
200
|
+
"""Missing associated documentation comment in .proto file."""
|
201
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
202
|
+
context.set_details('Method not implemented!')
|
203
|
+
raise NotImplementedError('Method not implemented!')
|
204
|
+
|
205
|
+
def SetPromptTag(self, request, context):
|
206
|
+
"""Missing associated documentation comment in .proto file."""
|
207
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
208
|
+
context.set_details('Method not implemented!')
|
209
|
+
raise NotImplementedError('Method not implemented!')
|
210
|
+
|
211
|
+
def DeletePromptTag(self, request, context):
|
212
|
+
"""Missing associated documentation comment in .proto file."""
|
213
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
214
|
+
context.set_details('Method not implemented!')
|
215
|
+
raise NotImplementedError('Method not implemented!')
|
216
|
+
|
217
|
+
def SetPromptVersionTag(self, request, context):
|
218
|
+
"""Missing associated documentation comment in .proto file."""
|
219
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
220
|
+
context.set_details('Method not implemented!')
|
221
|
+
raise NotImplementedError('Method not implemented!')
|
222
|
+
|
223
|
+
def DeletePromptVersionTag(self, request, context):
|
224
|
+
"""Missing associated documentation comment in .proto file."""
|
225
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
226
|
+
context.set_details('Method not implemented!')
|
227
|
+
raise NotImplementedError('Method not implemented!')
|
228
|
+
|
229
|
+
|
230
|
+
def add_UnityCatalogPromptServiceServicer_to_server(servicer, server):
|
231
|
+
rpc_method_handlers = {
|
232
|
+
'CreatePrompt': grpc.unary_unary_rpc_method_handler(
|
233
|
+
servicer.CreatePrompt,
|
234
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.CreatePromptRequest.FromString,
|
235
|
+
response_serializer=unity__catalog__prompt__messages__pb2.CreatePromptResponse.SerializeToString,
|
236
|
+
),
|
237
|
+
'UpdatePrompt': grpc.unary_unary_rpc_method_handler(
|
238
|
+
servicer.UpdatePrompt,
|
239
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.UpdatePromptRequest.FromString,
|
240
|
+
response_serializer=unity__catalog__prompt__messages__pb2.UpdatePromptResponse.SerializeToString,
|
241
|
+
),
|
242
|
+
'DeletePrompt': grpc.unary_unary_rpc_method_handler(
|
243
|
+
servicer.DeletePrompt,
|
244
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptRequest.FromString,
|
245
|
+
response_serializer=unity__catalog__prompt__messages__pb2.DeletePromptResponse.SerializeToString,
|
246
|
+
),
|
247
|
+
'GetPrompt': grpc.unary_unary_rpc_method_handler(
|
248
|
+
servicer.GetPrompt,
|
249
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.GetPromptRequest.FromString,
|
250
|
+
response_serializer=unity__catalog__prompt__messages__pb2.GetPromptResponse.SerializeToString,
|
251
|
+
),
|
252
|
+
'SearchPrompts': grpc.unary_unary_rpc_method_handler(
|
253
|
+
servicer.SearchPrompts,
|
254
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.SearchPromptsRequest.FromString,
|
255
|
+
response_serializer=unity__catalog__prompt__messages__pb2.SearchPromptsResponse.SerializeToString,
|
256
|
+
),
|
257
|
+
'CreatePromptVersion': grpc.unary_unary_rpc_method_handler(
|
258
|
+
servicer.CreatePromptVersion,
|
259
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.CreatePromptVersionRequest.FromString,
|
260
|
+
response_serializer=unity__catalog__prompt__messages__pb2.CreatePromptVersionResponse.SerializeToString,
|
261
|
+
),
|
262
|
+
'UpdatePromptVersion': grpc.unary_unary_rpc_method_handler(
|
263
|
+
servicer.UpdatePromptVersion,
|
264
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.UpdatePromptVersionRequest.FromString,
|
265
|
+
response_serializer=unity__catalog__prompt__messages__pb2.UpdatePromptVersionResponse.SerializeToString,
|
266
|
+
),
|
267
|
+
'DeletePromptVersion': grpc.unary_unary_rpc_method_handler(
|
268
|
+
servicer.DeletePromptVersion,
|
269
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptVersionRequest.FromString,
|
270
|
+
response_serializer=unity__catalog__prompt__messages__pb2.DeletePromptVersionResponse.SerializeToString,
|
271
|
+
),
|
272
|
+
'GetPromptVersion': grpc.unary_unary_rpc_method_handler(
|
273
|
+
servicer.GetPromptVersion,
|
274
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.GetPromptVersionRequest.FromString,
|
275
|
+
response_serializer=unity__catalog__prompt__messages__pb2.GetPromptVersionResponse.SerializeToString,
|
276
|
+
),
|
277
|
+
'SearchPromptVersions': grpc.unary_unary_rpc_method_handler(
|
278
|
+
servicer.SearchPromptVersions,
|
279
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.SearchPromptVersionsRequest.FromString,
|
280
|
+
response_serializer=unity__catalog__prompt__messages__pb2.SearchPromptVersionsResponse.SerializeToString,
|
281
|
+
),
|
282
|
+
'SetPromptAlias': grpc.unary_unary_rpc_method_handler(
|
283
|
+
servicer.SetPromptAlias,
|
284
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.SetPromptAliasRequest.FromString,
|
285
|
+
response_serializer=unity__catalog__prompt__messages__pb2.SetPromptAliasResponse.SerializeToString,
|
286
|
+
),
|
287
|
+
'DeletePromptAlias': grpc.unary_unary_rpc_method_handler(
|
288
|
+
servicer.DeletePromptAlias,
|
289
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptAliasRequest.FromString,
|
290
|
+
response_serializer=unity__catalog__prompt__messages__pb2.DeletePromptAliasResponse.SerializeToString,
|
291
|
+
),
|
292
|
+
'GetPromptVersionByAlias': grpc.unary_unary_rpc_method_handler(
|
293
|
+
servicer.GetPromptVersionByAlias,
|
294
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.GetPromptVersionByAliasRequest.FromString,
|
295
|
+
response_serializer=unity__catalog__prompt__messages__pb2.GetPromptVersionByAliasResponse.SerializeToString,
|
296
|
+
),
|
297
|
+
'SetPromptTag': grpc.unary_unary_rpc_method_handler(
|
298
|
+
servicer.SetPromptTag,
|
299
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.SetPromptTagRequest.FromString,
|
300
|
+
response_serializer=unity__catalog__prompt__messages__pb2.SetPromptTagResponse.SerializeToString,
|
301
|
+
),
|
302
|
+
'DeletePromptTag': grpc.unary_unary_rpc_method_handler(
|
303
|
+
servicer.DeletePromptTag,
|
304
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptTagRequest.FromString,
|
305
|
+
response_serializer=unity__catalog__prompt__messages__pb2.DeletePromptTagResponse.SerializeToString,
|
306
|
+
),
|
307
|
+
'SetPromptVersionTag': grpc.unary_unary_rpc_method_handler(
|
308
|
+
servicer.SetPromptVersionTag,
|
309
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.SetPromptVersionTagRequest.FromString,
|
310
|
+
response_serializer=unity__catalog__prompt__messages__pb2.SetPromptVersionTagResponse.SerializeToString,
|
311
|
+
),
|
312
|
+
'DeletePromptVersionTag': grpc.unary_unary_rpc_method_handler(
|
313
|
+
servicer.DeletePromptVersionTag,
|
314
|
+
request_deserializer=unity__catalog__prompt__messages__pb2.DeletePromptVersionTagRequest.FromString,
|
315
|
+
response_serializer=unity__catalog__prompt__messages__pb2.DeletePromptVersionTagResponse.SerializeToString,
|
316
|
+
),
|
317
|
+
}
|
318
|
+
generic_handler = grpc.method_handlers_generic_handler(
|
319
|
+
'mlflow.unitycatalog.UnityCatalogPromptService', rpc_method_handlers)
|
320
|
+
server.add_generic_rpc_handlers((generic_handler,))
|
321
|
+
server.add_registered_method_handlers('mlflow.unitycatalog.UnityCatalogPromptService', rpc_method_handlers)
|
322
|
+
|
323
|
+
|
324
|
+
# This class is part of an EXPERIMENTAL API.
|
325
|
+
class UnityCatalogPromptService(object):
|
326
|
+
"""Missing associated documentation comment in .proto file."""
|
327
|
+
|
328
|
+
@staticmethod
|
329
|
+
def CreatePrompt(request,
|
330
|
+
target,
|
331
|
+
options=(),
|
332
|
+
channel_credentials=None,
|
333
|
+
call_credentials=None,
|
334
|
+
insecure=False,
|
335
|
+
compression=None,
|
336
|
+
wait_for_ready=None,
|
337
|
+
timeout=None,
|
338
|
+
metadata=None):
|
339
|
+
return grpc.experimental.unary_unary(
|
340
|
+
request,
|
341
|
+
target,
|
342
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/CreatePrompt',
|
343
|
+
unity__catalog__prompt__messages__pb2.CreatePromptRequest.SerializeToString,
|
344
|
+
unity__catalog__prompt__messages__pb2.CreatePromptResponse.FromString,
|
345
|
+
options,
|
346
|
+
channel_credentials,
|
347
|
+
insecure,
|
348
|
+
call_credentials,
|
349
|
+
compression,
|
350
|
+
wait_for_ready,
|
351
|
+
timeout,
|
352
|
+
metadata,
|
353
|
+
_registered_method=True)
|
354
|
+
|
355
|
+
@staticmethod
|
356
|
+
def UpdatePrompt(request,
|
357
|
+
target,
|
358
|
+
options=(),
|
359
|
+
channel_credentials=None,
|
360
|
+
call_credentials=None,
|
361
|
+
insecure=False,
|
362
|
+
compression=None,
|
363
|
+
wait_for_ready=None,
|
364
|
+
timeout=None,
|
365
|
+
metadata=None):
|
366
|
+
return grpc.experimental.unary_unary(
|
367
|
+
request,
|
368
|
+
target,
|
369
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/UpdatePrompt',
|
370
|
+
unity__catalog__prompt__messages__pb2.UpdatePromptRequest.SerializeToString,
|
371
|
+
unity__catalog__prompt__messages__pb2.UpdatePromptResponse.FromString,
|
372
|
+
options,
|
373
|
+
channel_credentials,
|
374
|
+
insecure,
|
375
|
+
call_credentials,
|
376
|
+
compression,
|
377
|
+
wait_for_ready,
|
378
|
+
timeout,
|
379
|
+
metadata,
|
380
|
+
_registered_method=True)
|
381
|
+
|
382
|
+
@staticmethod
|
383
|
+
def DeletePrompt(request,
|
384
|
+
target,
|
385
|
+
options=(),
|
386
|
+
channel_credentials=None,
|
387
|
+
call_credentials=None,
|
388
|
+
insecure=False,
|
389
|
+
compression=None,
|
390
|
+
wait_for_ready=None,
|
391
|
+
timeout=None,
|
392
|
+
metadata=None):
|
393
|
+
return grpc.experimental.unary_unary(
|
394
|
+
request,
|
395
|
+
target,
|
396
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePrompt',
|
397
|
+
unity__catalog__prompt__messages__pb2.DeletePromptRequest.SerializeToString,
|
398
|
+
unity__catalog__prompt__messages__pb2.DeletePromptResponse.FromString,
|
399
|
+
options,
|
400
|
+
channel_credentials,
|
401
|
+
insecure,
|
402
|
+
call_credentials,
|
403
|
+
compression,
|
404
|
+
wait_for_ready,
|
405
|
+
timeout,
|
406
|
+
metadata,
|
407
|
+
_registered_method=True)
|
408
|
+
|
409
|
+
@staticmethod
|
410
|
+
def GetPrompt(request,
|
411
|
+
target,
|
412
|
+
options=(),
|
413
|
+
channel_credentials=None,
|
414
|
+
call_credentials=None,
|
415
|
+
insecure=False,
|
416
|
+
compression=None,
|
417
|
+
wait_for_ready=None,
|
418
|
+
timeout=None,
|
419
|
+
metadata=None):
|
420
|
+
return grpc.experimental.unary_unary(
|
421
|
+
request,
|
422
|
+
target,
|
423
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/GetPrompt',
|
424
|
+
unity__catalog__prompt__messages__pb2.GetPromptRequest.SerializeToString,
|
425
|
+
unity__catalog__prompt__messages__pb2.GetPromptResponse.FromString,
|
426
|
+
options,
|
427
|
+
channel_credentials,
|
428
|
+
insecure,
|
429
|
+
call_credentials,
|
430
|
+
compression,
|
431
|
+
wait_for_ready,
|
432
|
+
timeout,
|
433
|
+
metadata,
|
434
|
+
_registered_method=True)
|
435
|
+
|
436
|
+
@staticmethod
|
437
|
+
def SearchPrompts(request,
|
438
|
+
target,
|
439
|
+
options=(),
|
440
|
+
channel_credentials=None,
|
441
|
+
call_credentials=None,
|
442
|
+
insecure=False,
|
443
|
+
compression=None,
|
444
|
+
wait_for_ready=None,
|
445
|
+
timeout=None,
|
446
|
+
metadata=None):
|
447
|
+
return grpc.experimental.unary_unary(
|
448
|
+
request,
|
449
|
+
target,
|
450
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SearchPrompts',
|
451
|
+
unity__catalog__prompt__messages__pb2.SearchPromptsRequest.SerializeToString,
|
452
|
+
unity__catalog__prompt__messages__pb2.SearchPromptsResponse.FromString,
|
453
|
+
options,
|
454
|
+
channel_credentials,
|
455
|
+
insecure,
|
456
|
+
call_credentials,
|
457
|
+
compression,
|
458
|
+
wait_for_ready,
|
459
|
+
timeout,
|
460
|
+
metadata,
|
461
|
+
_registered_method=True)
|
462
|
+
|
463
|
+
@staticmethod
|
464
|
+
def CreatePromptVersion(request,
|
465
|
+
target,
|
466
|
+
options=(),
|
467
|
+
channel_credentials=None,
|
468
|
+
call_credentials=None,
|
469
|
+
insecure=False,
|
470
|
+
compression=None,
|
471
|
+
wait_for_ready=None,
|
472
|
+
timeout=None,
|
473
|
+
metadata=None):
|
474
|
+
return grpc.experimental.unary_unary(
|
475
|
+
request,
|
476
|
+
target,
|
477
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/CreatePromptVersion',
|
478
|
+
unity__catalog__prompt__messages__pb2.CreatePromptVersionRequest.SerializeToString,
|
479
|
+
unity__catalog__prompt__messages__pb2.CreatePromptVersionResponse.FromString,
|
480
|
+
options,
|
481
|
+
channel_credentials,
|
482
|
+
insecure,
|
483
|
+
call_credentials,
|
484
|
+
compression,
|
485
|
+
wait_for_ready,
|
486
|
+
timeout,
|
487
|
+
metadata,
|
488
|
+
_registered_method=True)
|
489
|
+
|
490
|
+
@staticmethod
|
491
|
+
def UpdatePromptVersion(request,
|
492
|
+
target,
|
493
|
+
options=(),
|
494
|
+
channel_credentials=None,
|
495
|
+
call_credentials=None,
|
496
|
+
insecure=False,
|
497
|
+
compression=None,
|
498
|
+
wait_for_ready=None,
|
499
|
+
timeout=None,
|
500
|
+
metadata=None):
|
501
|
+
return grpc.experimental.unary_unary(
|
502
|
+
request,
|
503
|
+
target,
|
504
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/UpdatePromptVersion',
|
505
|
+
unity__catalog__prompt__messages__pb2.UpdatePromptVersionRequest.SerializeToString,
|
506
|
+
unity__catalog__prompt__messages__pb2.UpdatePromptVersionResponse.FromString,
|
507
|
+
options,
|
508
|
+
channel_credentials,
|
509
|
+
insecure,
|
510
|
+
call_credentials,
|
511
|
+
compression,
|
512
|
+
wait_for_ready,
|
513
|
+
timeout,
|
514
|
+
metadata,
|
515
|
+
_registered_method=True)
|
516
|
+
|
517
|
+
@staticmethod
|
518
|
+
def DeletePromptVersion(request,
|
519
|
+
target,
|
520
|
+
options=(),
|
521
|
+
channel_credentials=None,
|
522
|
+
call_credentials=None,
|
523
|
+
insecure=False,
|
524
|
+
compression=None,
|
525
|
+
wait_for_ready=None,
|
526
|
+
timeout=None,
|
527
|
+
metadata=None):
|
528
|
+
return grpc.experimental.unary_unary(
|
529
|
+
request,
|
530
|
+
target,
|
531
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePromptVersion',
|
532
|
+
unity__catalog__prompt__messages__pb2.DeletePromptVersionRequest.SerializeToString,
|
533
|
+
unity__catalog__prompt__messages__pb2.DeletePromptVersionResponse.FromString,
|
534
|
+
options,
|
535
|
+
channel_credentials,
|
536
|
+
insecure,
|
537
|
+
call_credentials,
|
538
|
+
compression,
|
539
|
+
wait_for_ready,
|
540
|
+
timeout,
|
541
|
+
metadata,
|
542
|
+
_registered_method=True)
|
543
|
+
|
544
|
+
@staticmethod
|
545
|
+
def GetPromptVersion(request,
|
546
|
+
target,
|
547
|
+
options=(),
|
548
|
+
channel_credentials=None,
|
549
|
+
call_credentials=None,
|
550
|
+
insecure=False,
|
551
|
+
compression=None,
|
552
|
+
wait_for_ready=None,
|
553
|
+
timeout=None,
|
554
|
+
metadata=None):
|
555
|
+
return grpc.experimental.unary_unary(
|
556
|
+
request,
|
557
|
+
target,
|
558
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/GetPromptVersion',
|
559
|
+
unity__catalog__prompt__messages__pb2.GetPromptVersionRequest.SerializeToString,
|
560
|
+
unity__catalog__prompt__messages__pb2.GetPromptVersionResponse.FromString,
|
561
|
+
options,
|
562
|
+
channel_credentials,
|
563
|
+
insecure,
|
564
|
+
call_credentials,
|
565
|
+
compression,
|
566
|
+
wait_for_ready,
|
567
|
+
timeout,
|
568
|
+
metadata,
|
569
|
+
_registered_method=True)
|
570
|
+
|
571
|
+
@staticmethod
|
572
|
+
def SearchPromptVersions(request,
|
573
|
+
target,
|
574
|
+
options=(),
|
575
|
+
channel_credentials=None,
|
576
|
+
call_credentials=None,
|
577
|
+
insecure=False,
|
578
|
+
compression=None,
|
579
|
+
wait_for_ready=None,
|
580
|
+
timeout=None,
|
581
|
+
metadata=None):
|
582
|
+
return grpc.experimental.unary_unary(
|
583
|
+
request,
|
584
|
+
target,
|
585
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SearchPromptVersions',
|
586
|
+
unity__catalog__prompt__messages__pb2.SearchPromptVersionsRequest.SerializeToString,
|
587
|
+
unity__catalog__prompt__messages__pb2.SearchPromptVersionsResponse.FromString,
|
588
|
+
options,
|
589
|
+
channel_credentials,
|
590
|
+
insecure,
|
591
|
+
call_credentials,
|
592
|
+
compression,
|
593
|
+
wait_for_ready,
|
594
|
+
timeout,
|
595
|
+
metadata,
|
596
|
+
_registered_method=True)
|
597
|
+
|
598
|
+
@staticmethod
|
599
|
+
def SetPromptAlias(request,
|
600
|
+
target,
|
601
|
+
options=(),
|
602
|
+
channel_credentials=None,
|
603
|
+
call_credentials=None,
|
604
|
+
insecure=False,
|
605
|
+
compression=None,
|
606
|
+
wait_for_ready=None,
|
607
|
+
timeout=None,
|
608
|
+
metadata=None):
|
609
|
+
return grpc.experimental.unary_unary(
|
610
|
+
request,
|
611
|
+
target,
|
612
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SetPromptAlias',
|
613
|
+
unity__catalog__prompt__messages__pb2.SetPromptAliasRequest.SerializeToString,
|
614
|
+
unity__catalog__prompt__messages__pb2.SetPromptAliasResponse.FromString,
|
615
|
+
options,
|
616
|
+
channel_credentials,
|
617
|
+
insecure,
|
618
|
+
call_credentials,
|
619
|
+
compression,
|
620
|
+
wait_for_ready,
|
621
|
+
timeout,
|
622
|
+
metadata,
|
623
|
+
_registered_method=True)
|
624
|
+
|
625
|
+
@staticmethod
|
626
|
+
def DeletePromptAlias(request,
|
627
|
+
target,
|
628
|
+
options=(),
|
629
|
+
channel_credentials=None,
|
630
|
+
call_credentials=None,
|
631
|
+
insecure=False,
|
632
|
+
compression=None,
|
633
|
+
wait_for_ready=None,
|
634
|
+
timeout=None,
|
635
|
+
metadata=None):
|
636
|
+
return grpc.experimental.unary_unary(
|
637
|
+
request,
|
638
|
+
target,
|
639
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePromptAlias',
|
640
|
+
unity__catalog__prompt__messages__pb2.DeletePromptAliasRequest.SerializeToString,
|
641
|
+
unity__catalog__prompt__messages__pb2.DeletePromptAliasResponse.FromString,
|
642
|
+
options,
|
643
|
+
channel_credentials,
|
644
|
+
insecure,
|
645
|
+
call_credentials,
|
646
|
+
compression,
|
647
|
+
wait_for_ready,
|
648
|
+
timeout,
|
649
|
+
metadata,
|
650
|
+
_registered_method=True)
|
651
|
+
|
652
|
+
@staticmethod
|
653
|
+
def GetPromptVersionByAlias(request,
|
654
|
+
target,
|
655
|
+
options=(),
|
656
|
+
channel_credentials=None,
|
657
|
+
call_credentials=None,
|
658
|
+
insecure=False,
|
659
|
+
compression=None,
|
660
|
+
wait_for_ready=None,
|
661
|
+
timeout=None,
|
662
|
+
metadata=None):
|
663
|
+
return grpc.experimental.unary_unary(
|
664
|
+
request,
|
665
|
+
target,
|
666
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/GetPromptVersionByAlias',
|
667
|
+
unity__catalog__prompt__messages__pb2.GetPromptVersionByAliasRequest.SerializeToString,
|
668
|
+
unity__catalog__prompt__messages__pb2.GetPromptVersionByAliasResponse.FromString,
|
669
|
+
options,
|
670
|
+
channel_credentials,
|
671
|
+
insecure,
|
672
|
+
call_credentials,
|
673
|
+
compression,
|
674
|
+
wait_for_ready,
|
675
|
+
timeout,
|
676
|
+
metadata,
|
677
|
+
_registered_method=True)
|
678
|
+
|
679
|
+
@staticmethod
|
680
|
+
def SetPromptTag(request,
|
681
|
+
target,
|
682
|
+
options=(),
|
683
|
+
channel_credentials=None,
|
684
|
+
call_credentials=None,
|
685
|
+
insecure=False,
|
686
|
+
compression=None,
|
687
|
+
wait_for_ready=None,
|
688
|
+
timeout=None,
|
689
|
+
metadata=None):
|
690
|
+
return grpc.experimental.unary_unary(
|
691
|
+
request,
|
692
|
+
target,
|
693
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SetPromptTag',
|
694
|
+
unity__catalog__prompt__messages__pb2.SetPromptTagRequest.SerializeToString,
|
695
|
+
unity__catalog__prompt__messages__pb2.SetPromptTagResponse.FromString,
|
696
|
+
options,
|
697
|
+
channel_credentials,
|
698
|
+
insecure,
|
699
|
+
call_credentials,
|
700
|
+
compression,
|
701
|
+
wait_for_ready,
|
702
|
+
timeout,
|
703
|
+
metadata,
|
704
|
+
_registered_method=True)
|
705
|
+
|
706
|
+
@staticmethod
|
707
|
+
def DeletePromptTag(request,
|
708
|
+
target,
|
709
|
+
options=(),
|
710
|
+
channel_credentials=None,
|
711
|
+
call_credentials=None,
|
712
|
+
insecure=False,
|
713
|
+
compression=None,
|
714
|
+
wait_for_ready=None,
|
715
|
+
timeout=None,
|
716
|
+
metadata=None):
|
717
|
+
return grpc.experimental.unary_unary(
|
718
|
+
request,
|
719
|
+
target,
|
720
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePromptTag',
|
721
|
+
unity__catalog__prompt__messages__pb2.DeletePromptTagRequest.SerializeToString,
|
722
|
+
unity__catalog__prompt__messages__pb2.DeletePromptTagResponse.FromString,
|
723
|
+
options,
|
724
|
+
channel_credentials,
|
725
|
+
insecure,
|
726
|
+
call_credentials,
|
727
|
+
compression,
|
728
|
+
wait_for_ready,
|
729
|
+
timeout,
|
730
|
+
metadata,
|
731
|
+
_registered_method=True)
|
732
|
+
|
733
|
+
@staticmethod
|
734
|
+
def SetPromptVersionTag(request,
|
735
|
+
target,
|
736
|
+
options=(),
|
737
|
+
channel_credentials=None,
|
738
|
+
call_credentials=None,
|
739
|
+
insecure=False,
|
740
|
+
compression=None,
|
741
|
+
wait_for_ready=None,
|
742
|
+
timeout=None,
|
743
|
+
metadata=None):
|
744
|
+
return grpc.experimental.unary_unary(
|
745
|
+
request,
|
746
|
+
target,
|
747
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/SetPromptVersionTag',
|
748
|
+
unity__catalog__prompt__messages__pb2.SetPromptVersionTagRequest.SerializeToString,
|
749
|
+
unity__catalog__prompt__messages__pb2.SetPromptVersionTagResponse.FromString,
|
750
|
+
options,
|
751
|
+
channel_credentials,
|
752
|
+
insecure,
|
753
|
+
call_credentials,
|
754
|
+
compression,
|
755
|
+
wait_for_ready,
|
756
|
+
timeout,
|
757
|
+
metadata,
|
758
|
+
_registered_method=True)
|
759
|
+
|
760
|
+
@staticmethod
|
761
|
+
def DeletePromptVersionTag(request,
|
762
|
+
target,
|
763
|
+
options=(),
|
764
|
+
channel_credentials=None,
|
765
|
+
call_credentials=None,
|
766
|
+
insecure=False,
|
767
|
+
compression=None,
|
768
|
+
wait_for_ready=None,
|
769
|
+
timeout=None,
|
770
|
+
metadata=None):
|
771
|
+
return grpc.experimental.unary_unary(
|
772
|
+
request,
|
773
|
+
target,
|
774
|
+
'/mlflow.unitycatalog.UnityCatalogPromptService/DeletePromptVersionTag',
|
775
|
+
unity__catalog__prompt__messages__pb2.DeletePromptVersionTagRequest.SerializeToString,
|
776
|
+
unity__catalog__prompt__messages__pb2.DeletePromptVersionTagResponse.FromString,
|
777
|
+
options,
|
778
|
+
channel_credentials,
|
779
|
+
insecure,
|
780
|
+
call_credentials,
|
781
|
+
compression,
|
782
|
+
wait_for_ready,
|
783
|
+
timeout,
|
784
|
+
metadata,
|
785
|
+
_registered_method=True)
|