import assessments_pb2 as _assessments_pb2 import databricks_pb2 as _databricks_pb2 import datasets_pb2 as _datasets_pb2 from google.protobuf import duration_pb2 as _duration_pb2 from google.protobuf import field_mask_pb2 as _field_mask_pb2 from google.protobuf import timestamp_pb2 as _timestamp_pb2 import issues_pb2 as _issues_pb2 import label_schemas_pb2 as _label_schemas_pb2 from opentelemetry.proto.trace.v1 import trace_pb2 as _trace_pb2 import prompt_optimization_pb2 as _prompt_optimization_pb2 import review_queues_pb2 as _review_queues_pb2 from scalapb import scalapb_pb2 as _scalapb_pb2 from google.protobuf.internal import containers as _containers from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper from google.protobuf import descriptor as _descriptor from google.protobuf import message as _message from google.protobuf import service as _service from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union DESCRIPTOR: _descriptor.FileDescriptor class ViewType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () ACTIVE_ONLY: _ClassVar[ViewType] DELETED_ONLY: _ClassVar[ViewType] ALL: _ClassVar[ViewType] class SourceType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () NOTEBOOK: _ClassVar[SourceType] JOB: _ClassVar[SourceType] PROJECT: _ClassVar[SourceType] LOCAL: _ClassVar[SourceType] UNKNOWN: _ClassVar[SourceType] class RunStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () RUNNING: _ClassVar[RunStatus] SCHEDULED: _ClassVar[RunStatus] FINISHED: _ClassVar[RunStatus] FAILED: _ClassVar[RunStatus] KILLED: _ClassVar[RunStatus] class TraceStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () TRACE_STATUS_UNSPECIFIED: _ClassVar[TraceStatus] OK: _ClassVar[TraceStatus] ERROR: _ClassVar[TraceStatus] IN_PROGRESS: _ClassVar[TraceStatus] class MetricViewType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () TRACES: _ClassVar[MetricViewType] SPANS: _ClassVar[MetricViewType] ASSESSMENTS: _ClassVar[MetricViewType] class AggregationType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () COUNT: _ClassVar[AggregationType] SUM: _ClassVar[AggregationType] AVG: _ClassVar[AggregationType] PERCENTILE: _ClassVar[AggregationType] MIN: _ClassVar[AggregationType] MAX: _ClassVar[AggregationType] class LoggedModelStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () LOGGED_MODEL_STATUS_UNSPECIFIED: _ClassVar[LoggedModelStatus] LOGGED_MODEL_PENDING: _ClassVar[LoggedModelStatus] LOGGED_MODEL_READY: _ClassVar[LoggedModelStatus] LOGGED_MODEL_UPLOAD_FAILED: _ClassVar[LoggedModelStatus] class RoutingStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () ROUTING_STRATEGY_UNSPECIFIED: _ClassVar[RoutingStrategy] REQUEST_BASED_TRAFFIC_SPLIT: _ClassVar[RoutingStrategy] class FallbackStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () FALLBACK_STRATEGY_UNSPECIFIED: _ClassVar[FallbackStrategy] SEQUENTIAL: _ClassVar[FallbackStrategy] class GatewayModelLinkageType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () LINKAGE_TYPE_UNSPECIFIED: _ClassVar[GatewayModelLinkageType] PRIMARY: _ClassVar[GatewayModelLinkageType] FALLBACK: _ClassVar[GatewayModelLinkageType] class BudgetDurationUnit(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () DURATION_UNIT_UNSPECIFIED: _ClassVar[BudgetDurationUnit] MINUTES: _ClassVar[BudgetDurationUnit] HOURS: _ClassVar[BudgetDurationUnit] DAYS: _ClassVar[BudgetDurationUnit] WEEKS: _ClassVar[BudgetDurationUnit] MONTHS: _ClassVar[BudgetDurationUnit] class BudgetTargetScope(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () TARGET_SCOPE_UNSPECIFIED: _ClassVar[BudgetTargetScope] GLOBAL: _ClassVar[BudgetTargetScope] WORKSPACE: _ClassVar[BudgetTargetScope] class BudgetAction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () BUDGET_ACTION_UNSPECIFIED: _ClassVar[BudgetAction] ALERT: _ClassVar[BudgetAction] REJECT: _ClassVar[BudgetAction] class BudgetUnit(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () BUDGET_UNIT_UNSPECIFIED: _ClassVar[BudgetUnit] USD: _ClassVar[BudgetUnit] class GuardrailStage(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () GUARDRAIL_STAGE_UNSPECIFIED: _ClassVar[GuardrailStage] BEFORE: _ClassVar[GuardrailStage] AFTER: _ClassVar[GuardrailStage] class GuardrailAction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () GUARDRAIL_ACTION_UNSPECIFIED: _ClassVar[GuardrailAction] VALIDATION: _ClassVar[GuardrailAction] SANITIZATION: _ClassVar[GuardrailAction] ACTIVE_ONLY: ViewType DELETED_ONLY: ViewType ALL: ViewType NOTEBOOK: SourceType JOB: SourceType PROJECT: SourceType LOCAL: SourceType UNKNOWN: SourceType RUNNING: RunStatus SCHEDULED: RunStatus FINISHED: RunStatus FAILED: RunStatus KILLED: RunStatus TRACE_STATUS_UNSPECIFIED: TraceStatus OK: TraceStatus ERROR: TraceStatus IN_PROGRESS: TraceStatus TRACES: MetricViewType SPANS: MetricViewType ASSESSMENTS: MetricViewType COUNT: AggregationType SUM: AggregationType AVG: AggregationType PERCENTILE: AggregationType MIN: AggregationType MAX: AggregationType LOGGED_MODEL_STATUS_UNSPECIFIED: LoggedModelStatus LOGGED_MODEL_PENDING: LoggedModelStatus LOGGED_MODEL_READY: LoggedModelStatus LOGGED_MODEL_UPLOAD_FAILED: LoggedModelStatus ROUTING_STRATEGY_UNSPECIFIED: RoutingStrategy REQUEST_BASED_TRAFFIC_SPLIT: RoutingStrategy FALLBACK_STRATEGY_UNSPECIFIED: FallbackStrategy SEQUENTIAL: FallbackStrategy LINKAGE_TYPE_UNSPECIFIED: GatewayModelLinkageType PRIMARY: GatewayModelLinkageType FALLBACK: GatewayModelLinkageType DURATION_UNIT_UNSPECIFIED: BudgetDurationUnit MINUTES: BudgetDurationUnit HOURS: BudgetDurationUnit DAYS: BudgetDurationUnit WEEKS: BudgetDurationUnit MONTHS: BudgetDurationUnit TARGET_SCOPE_UNSPECIFIED: BudgetTargetScope GLOBAL: BudgetTargetScope WORKSPACE: BudgetTargetScope BUDGET_ACTION_UNSPECIFIED: BudgetAction ALERT: BudgetAction REJECT: BudgetAction BUDGET_UNIT_UNSPECIFIED: BudgetUnit USD: BudgetUnit GUARDRAIL_STAGE_UNSPECIFIED: GuardrailStage BEFORE: GuardrailStage AFTER: GuardrailStage GUARDRAIL_ACTION_UNSPECIFIED: GuardrailAction VALIDATION: GuardrailAction SANITIZATION: GuardrailAction class Metric(_message.Message): __slots__ = ("key", "value", "timestamp", "step", "dataset_name", "dataset_digest", "model_id", "run_id") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] TIMESTAMP_FIELD_NUMBER: _ClassVar[int] STEP_FIELD_NUMBER: _ClassVar[int] DATASET_NAME_FIELD_NUMBER: _ClassVar[int] DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] MODEL_ID_FIELD_NUMBER: _ClassVar[int] RUN_ID_FIELD_NUMBER: _ClassVar[int] key: str value: float timestamp: int step: int dataset_name: str dataset_digest: str model_id: str run_id: str def __init__(self, key: _Optional[str] = ..., value: _Optional[float] = ..., timestamp: _Optional[int] = ..., step: _Optional[int] = ..., dataset_name: _Optional[str] = ..., dataset_digest: _Optional[str] = ..., model_id: _Optional[str] = ..., run_id: _Optional[str] = ...) -> None: ... class Param(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class Run(_message.Message): __slots__ = ("info", "data", "inputs", "outputs") INFO_FIELD_NUMBER: _ClassVar[int] DATA_FIELD_NUMBER: _ClassVar[int] INPUTS_FIELD_NUMBER: _ClassVar[int] OUTPUTS_FIELD_NUMBER: _ClassVar[int] info: RunInfo data: RunData inputs: RunInputs outputs: RunOutputs def __init__(self, info: _Optional[_Union[RunInfo, _Mapping]] = ..., data: _Optional[_Union[RunData, _Mapping]] = ..., inputs: _Optional[_Union[RunInputs, _Mapping]] = ..., outputs: _Optional[_Union[RunOutputs, _Mapping]] = ...) -> None: ... class RunData(_message.Message): __slots__ = ("metrics", "params", "tags") METRICS_FIELD_NUMBER: _ClassVar[int] PARAMS_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] metrics: _containers.RepeatedCompositeFieldContainer[Metric] params: _containers.RepeatedCompositeFieldContainer[Param] tags: _containers.RepeatedCompositeFieldContainer[RunTag] def __init__(self, metrics: _Optional[_Iterable[_Union[Metric, _Mapping]]] = ..., params: _Optional[_Iterable[_Union[Param, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[RunTag, _Mapping]]] = ...) -> None: ... class RunInputs(_message.Message): __slots__ = ("dataset_inputs", "model_inputs") DATASET_INPUTS_FIELD_NUMBER: _ClassVar[int] MODEL_INPUTS_FIELD_NUMBER: _ClassVar[int] dataset_inputs: _containers.RepeatedCompositeFieldContainer[DatasetInput] model_inputs: _containers.RepeatedCompositeFieldContainer[ModelInput] def __init__(self, dataset_inputs: _Optional[_Iterable[_Union[DatasetInput, _Mapping]]] = ..., model_inputs: _Optional[_Iterable[_Union[ModelInput, _Mapping]]] = ...) -> None: ... class RunOutputs(_message.Message): __slots__ = ("model_outputs",) MODEL_OUTPUTS_FIELD_NUMBER: _ClassVar[int] model_outputs: _containers.RepeatedCompositeFieldContainer[ModelOutput] def __init__(self, model_outputs: _Optional[_Iterable[_Union[ModelOutput, _Mapping]]] = ...) -> None: ... class RunTag(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class ExperimentTag(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class RunInfo(_message.Message): __slots__ = ("run_id", "run_uuid", "run_name", "experiment_id", "user_id", "status", "start_time", "end_time", "artifact_uri", "lifecycle_stage") RUN_ID_FIELD_NUMBER: _ClassVar[int] RUN_UUID_FIELD_NUMBER: _ClassVar[int] RUN_NAME_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] USER_ID_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] START_TIME_FIELD_NUMBER: _ClassVar[int] END_TIME_FIELD_NUMBER: _ClassVar[int] ARTIFACT_URI_FIELD_NUMBER: _ClassVar[int] LIFECYCLE_STAGE_FIELD_NUMBER: _ClassVar[int] run_id: str run_uuid: str run_name: str experiment_id: str user_id: str status: RunStatus start_time: int end_time: int artifact_uri: str lifecycle_stage: str def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., run_name: _Optional[str] = ..., experiment_id: _Optional[str] = ..., user_id: _Optional[str] = ..., status: _Optional[_Union[RunStatus, str]] = ..., start_time: _Optional[int] = ..., end_time: _Optional[int] = ..., artifact_uri: _Optional[str] = ..., lifecycle_stage: _Optional[str] = ...) -> None: ... class Experiment(_message.Message): __slots__ = ("experiment_id", "name", "artifact_location", "lifecycle_stage", "last_update_time", "creation_time", "tags", "effective_trace_archival_retention", "workspace") EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] ARTIFACT_LOCATION_FIELD_NUMBER: _ClassVar[int] LIFECYCLE_STAGE_FIELD_NUMBER: _ClassVar[int] LAST_UPDATE_TIME_FIELD_NUMBER: _ClassVar[int] CREATION_TIME_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] EFFECTIVE_TRACE_ARCHIVAL_RETENTION_FIELD_NUMBER: _ClassVar[int] WORKSPACE_FIELD_NUMBER: _ClassVar[int] experiment_id: str name: str artifact_location: str lifecycle_stage: str last_update_time: int creation_time: int tags: _containers.RepeatedCompositeFieldContainer[ExperimentTag] effective_trace_archival_retention: str workspace: str def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., artifact_location: _Optional[str] = ..., lifecycle_stage: _Optional[str] = ..., last_update_time: _Optional[int] = ..., creation_time: _Optional[int] = ..., tags: _Optional[_Iterable[_Union[ExperimentTag, _Mapping]]] = ..., effective_trace_archival_retention: _Optional[str] = ..., workspace: _Optional[str] = ...) -> None: ... class DatasetInput(_message.Message): __slots__ = ("tags", "dataset") TAGS_FIELD_NUMBER: _ClassVar[int] DATASET_FIELD_NUMBER: _ClassVar[int] tags: _containers.RepeatedCompositeFieldContainer[InputTag] dataset: Dataset def __init__(self, tags: _Optional[_Iterable[_Union[InputTag, _Mapping]]] = ..., dataset: _Optional[_Union[Dataset, _Mapping]] = ...) -> None: ... class ModelInput(_message.Message): __slots__ = ("model_id",) MODEL_ID_FIELD_NUMBER: _ClassVar[int] model_id: str def __init__(self, model_id: _Optional[str] = ...) -> None: ... class InputTag(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class Dataset(_message.Message): __slots__ = ("name", "digest", "source_type", "source", "schema", "profile") NAME_FIELD_NUMBER: _ClassVar[int] DIGEST_FIELD_NUMBER: _ClassVar[int] SOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] SOURCE_FIELD_NUMBER: _ClassVar[int] SCHEMA_FIELD_NUMBER: _ClassVar[int] PROFILE_FIELD_NUMBER: _ClassVar[int] name: str digest: str source_type: str source: str schema: str profile: str def __init__(self, name: _Optional[str] = ..., digest: _Optional[str] = ..., source_type: _Optional[str] = ..., source: _Optional[str] = ..., schema: _Optional[str] = ..., profile: _Optional[str] = ...) -> None: ... class ModelOutput(_message.Message): __slots__ = ("model_id", "step") MODEL_ID_FIELD_NUMBER: _ClassVar[int] STEP_FIELD_NUMBER: _ClassVar[int] model_id: str step: int def __init__(self, model_id: _Optional[str] = ..., step: _Optional[int] = ...) -> None: ... class CreateExperiment(_message.Message): __slots__ = ("name", "artifact_location", "tags") class Response(_message.Message): __slots__ = ("experiment_id",) EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] experiment_id: str def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... NAME_FIELD_NUMBER: _ClassVar[int] ARTIFACT_LOCATION_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] name: str artifact_location: str tags: _containers.RepeatedCompositeFieldContainer[ExperimentTag] def __init__(self, name: _Optional[str] = ..., artifact_location: _Optional[str] = ..., tags: _Optional[_Iterable[_Union[ExperimentTag, _Mapping]]] = ...) -> None: ... class SearchExperiments(_message.Message): __slots__ = ("max_results", "page_token", "filter", "order_by", "view_type") class Response(_message.Message): __slots__ = ("experiments", "next_page_token") EXPERIMENTS_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] experiments: _containers.RepeatedCompositeFieldContainer[Experiment] next_page_token: str def __init__(self, experiments: _Optional[_Iterable[_Union[Experiment, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] FILTER_FIELD_NUMBER: _ClassVar[int] ORDER_BY_FIELD_NUMBER: _ClassVar[int] VIEW_TYPE_FIELD_NUMBER: _ClassVar[int] max_results: int page_token: str filter: str order_by: _containers.RepeatedScalarFieldContainer[str] view_type: ViewType def __init__(self, max_results: _Optional[int] = ..., page_token: _Optional[str] = ..., filter: _Optional[str] = ..., order_by: _Optional[_Iterable[str]] = ..., view_type: _Optional[_Union[ViewType, str]] = ...) -> None: ... class GetExperiment(_message.Message): __slots__ = ("experiment_id",) class Response(_message.Message): __slots__ = ("experiment",) EXPERIMENT_FIELD_NUMBER: _ClassVar[int] experiment: Experiment def __init__(self, experiment: _Optional[_Union[Experiment, _Mapping]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] experiment_id: str def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... class DeleteExperiment(_message.Message): __slots__ = ("experiment_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] experiment_id: str def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... class RestoreExperiment(_message.Message): __slots__ = ("experiment_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] experiment_id: str def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... class UpdateExperiment(_message.Message): __slots__ = ("experiment_id", "new_name") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NEW_NAME_FIELD_NUMBER: _ClassVar[int] experiment_id: str new_name: str def __init__(self, experiment_id: _Optional[str] = ..., new_name: _Optional[str] = ...) -> None: ... class CreateRun(_message.Message): __slots__ = ("experiment_id", "user_id", "run_name", "start_time", "tags") class Response(_message.Message): __slots__ = ("run",) RUN_FIELD_NUMBER: _ClassVar[int] run: Run def __init__(self, run: _Optional[_Union[Run, _Mapping]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] USER_ID_FIELD_NUMBER: _ClassVar[int] RUN_NAME_FIELD_NUMBER: _ClassVar[int] START_TIME_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] experiment_id: str user_id: str run_name: str start_time: int tags: _containers.RepeatedCompositeFieldContainer[RunTag] def __init__(self, experiment_id: _Optional[str] = ..., user_id: _Optional[str] = ..., run_name: _Optional[str] = ..., start_time: _Optional[int] = ..., tags: _Optional[_Iterable[_Union[RunTag, _Mapping]]] = ...) -> None: ... class UpdateRun(_message.Message): __slots__ = ("run_id", "run_uuid", "status", "end_time", "run_name") class Response(_message.Message): __slots__ = ("run_info",) RUN_INFO_FIELD_NUMBER: _ClassVar[int] run_info: RunInfo def __init__(self, run_info: _Optional[_Union[RunInfo, _Mapping]] = ...) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] RUN_UUID_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] END_TIME_FIELD_NUMBER: _ClassVar[int] RUN_NAME_FIELD_NUMBER: _ClassVar[int] run_id: str run_uuid: str status: RunStatus end_time: int run_name: str def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., status: _Optional[_Union[RunStatus, str]] = ..., end_time: _Optional[int] = ..., run_name: _Optional[str] = ...) -> None: ... class DeleteRun(_message.Message): __slots__ = ("run_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] run_id: str def __init__(self, run_id: _Optional[str] = ...) -> None: ... class RestoreRun(_message.Message): __slots__ = ("run_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] run_id: str def __init__(self, run_id: _Optional[str] = ...) -> None: ... class LogMetric(_message.Message): __slots__ = ("run_id", "run_uuid", "key", "value", "timestamp", "step", "model_id", "dataset_name", "dataset_digest") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] RUN_UUID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] TIMESTAMP_FIELD_NUMBER: _ClassVar[int] STEP_FIELD_NUMBER: _ClassVar[int] MODEL_ID_FIELD_NUMBER: _ClassVar[int] DATASET_NAME_FIELD_NUMBER: _ClassVar[int] DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] run_id: str run_uuid: str key: str value: float timestamp: int step: int model_id: str dataset_name: str dataset_digest: str def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[float] = ..., timestamp: _Optional[int] = ..., step: _Optional[int] = ..., model_id: _Optional[str] = ..., dataset_name: _Optional[str] = ..., dataset_digest: _Optional[str] = ...) -> None: ... class LogParam(_message.Message): __slots__ = ("run_id", "run_uuid", "key", "value") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] RUN_UUID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] run_id: str run_uuid: str key: str value: str def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class SetExperimentTag(_message.Message): __slots__ = ("experiment_id", "key", "value") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] experiment_id: str key: str value: str def __init__(self, experiment_id: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class DeleteExperimentTag(_message.Message): __slots__ = ("experiment_id", "key") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] experiment_id: str key: str def __init__(self, experiment_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... class SetTag(_message.Message): __slots__ = ("run_id", "run_uuid", "key", "value") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] RUN_UUID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] run_id: str run_uuid: str key: str value: str def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class DeleteTag(_message.Message): __slots__ = ("run_id", "key") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] run_id: str key: str def __init__(self, run_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... class GetRun(_message.Message): __slots__ = ("run_id", "run_uuid") class Response(_message.Message): __slots__ = ("run",) RUN_FIELD_NUMBER: _ClassVar[int] run: Run def __init__(self, run: _Optional[_Union[Run, _Mapping]] = ...) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] RUN_UUID_FIELD_NUMBER: _ClassVar[int] run_id: str run_uuid: str def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ...) -> None: ... class SearchRuns(_message.Message): __slots__ = ("experiment_ids", "filter", "run_view_type", "max_results", "order_by", "page_token") class Response(_message.Message): __slots__ = ("runs", "next_page_token") RUNS_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] runs: _containers.RepeatedCompositeFieldContainer[Run] next_page_token: str def __init__(self, runs: _Optional[_Iterable[_Union[Run, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] FILTER_FIELD_NUMBER: _ClassVar[int] RUN_VIEW_TYPE_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] ORDER_BY_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] experiment_ids: _containers.RepeatedScalarFieldContainer[str] filter: str run_view_type: ViewType max_results: int order_by: _containers.RepeatedScalarFieldContainer[str] page_token: str def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ..., filter: _Optional[str] = ..., run_view_type: _Optional[_Union[ViewType, str]] = ..., max_results: _Optional[int] = ..., order_by: _Optional[_Iterable[str]] = ..., page_token: _Optional[str] = ...) -> None: ... class ListArtifacts(_message.Message): __slots__ = ("run_id", "run_uuid", "path", "page_token") class Response(_message.Message): __slots__ = ("root_uri", "files", "next_page_token") ROOT_URI_FIELD_NUMBER: _ClassVar[int] FILES_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] root_uri: str files: _containers.RepeatedCompositeFieldContainer[FileInfo] next_page_token: str def __init__(self, root_uri: _Optional[str] = ..., files: _Optional[_Iterable[_Union[FileInfo, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] RUN_UUID_FIELD_NUMBER: _ClassVar[int] PATH_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] run_id: str run_uuid: str path: str page_token: str def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., path: _Optional[str] = ..., page_token: _Optional[str] = ...) -> None: ... class CreatePresignedUploadUrl(_message.Message): __slots__ = ("run_id", "path", "expiration") class Response(_message.Message): __slots__ = ("presigned_url", "headers") class HeadersEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... PRESIGNED_URL_FIELD_NUMBER: _ClassVar[int] HEADERS_FIELD_NUMBER: _ClassVar[int] presigned_url: str headers: _containers.ScalarMap[str, str] def __init__(self, presigned_url: _Optional[str] = ..., headers: _Optional[_Mapping[str, str]] = ...) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] PATH_FIELD_NUMBER: _ClassVar[int] EXPIRATION_FIELD_NUMBER: _ClassVar[int] run_id: str path: str expiration: int def __init__(self, run_id: _Optional[str] = ..., path: _Optional[str] = ..., expiration: _Optional[int] = ...) -> None: ... class FileInfo(_message.Message): __slots__ = ("path", "is_dir", "file_size") PATH_FIELD_NUMBER: _ClassVar[int] IS_DIR_FIELD_NUMBER: _ClassVar[int] FILE_SIZE_FIELD_NUMBER: _ClassVar[int] path: str is_dir: bool file_size: int def __init__(self, path: _Optional[str] = ..., is_dir: bool = ..., file_size: _Optional[int] = ...) -> None: ... class GetMetricHistory(_message.Message): __slots__ = ("run_id", "run_uuid", "metric_key", "page_token", "max_results") class Response(_message.Message): __slots__ = ("metrics", "next_page_token") METRICS_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] metrics: _containers.RepeatedCompositeFieldContainer[Metric] next_page_token: str def __init__(self, metrics: _Optional[_Iterable[_Union[Metric, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] RUN_UUID_FIELD_NUMBER: _ClassVar[int] METRIC_KEY_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] run_id: str run_uuid: str metric_key: str page_token: str max_results: int def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., metric_key: _Optional[str] = ..., page_token: _Optional[str] = ..., max_results: _Optional[int] = ...) -> None: ... class MetricWithRunId(_message.Message): __slots__ = ("key", "value", "timestamp", "step", "run_id") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] TIMESTAMP_FIELD_NUMBER: _ClassVar[int] STEP_FIELD_NUMBER: _ClassVar[int] RUN_ID_FIELD_NUMBER: _ClassVar[int] key: str value: float timestamp: int step: int run_id: str def __init__(self, key: _Optional[str] = ..., value: _Optional[float] = ..., timestamp: _Optional[int] = ..., step: _Optional[int] = ..., run_id: _Optional[str] = ...) -> None: ... class GetMetricHistoryBulkInterval(_message.Message): __slots__ = ("run_ids", "metric_key", "start_step", "end_step", "max_results") class Response(_message.Message): __slots__ = ("metrics",) METRICS_FIELD_NUMBER: _ClassVar[int] metrics: _containers.RepeatedCompositeFieldContainer[MetricWithRunId] def __init__(self, metrics: _Optional[_Iterable[_Union[MetricWithRunId, _Mapping]]] = ...) -> None: ... RUN_IDS_FIELD_NUMBER: _ClassVar[int] METRIC_KEY_FIELD_NUMBER: _ClassVar[int] START_STEP_FIELD_NUMBER: _ClassVar[int] END_STEP_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] run_ids: _containers.RepeatedScalarFieldContainer[str] metric_key: str start_step: int end_step: int max_results: int def __init__(self, run_ids: _Optional[_Iterable[str]] = ..., metric_key: _Optional[str] = ..., start_step: _Optional[int] = ..., end_step: _Optional[int] = ..., max_results: _Optional[int] = ...) -> None: ... class LogBatch(_message.Message): __slots__ = ("run_id", "metrics", "params", "tags") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] METRICS_FIELD_NUMBER: _ClassVar[int] PARAMS_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] run_id: str metrics: _containers.RepeatedCompositeFieldContainer[Metric] params: _containers.RepeatedCompositeFieldContainer[Param] tags: _containers.RepeatedCompositeFieldContainer[RunTag] def __init__(self, run_id: _Optional[str] = ..., metrics: _Optional[_Iterable[_Union[Metric, _Mapping]]] = ..., params: _Optional[_Iterable[_Union[Param, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[RunTag, _Mapping]]] = ...) -> None: ... class LogModel(_message.Message): __slots__ = ("run_id", "model_json") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] MODEL_JSON_FIELD_NUMBER: _ClassVar[int] run_id: str model_json: str def __init__(self, run_id: _Optional[str] = ..., model_json: _Optional[str] = ...) -> None: ... class LogInputs(_message.Message): __slots__ = ("run_id", "datasets", "models") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] DATASETS_FIELD_NUMBER: _ClassVar[int] MODELS_FIELD_NUMBER: _ClassVar[int] run_id: str datasets: _containers.RepeatedCompositeFieldContainer[DatasetInput] models: _containers.RepeatedCompositeFieldContainer[ModelInput] def __init__(self, run_id: _Optional[str] = ..., datasets: _Optional[_Iterable[_Union[DatasetInput, _Mapping]]] = ..., models: _Optional[_Iterable[_Union[ModelInput, _Mapping]]] = ...) -> None: ... class LogOutputs(_message.Message): __slots__ = ("run_id", "models") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... RUN_ID_FIELD_NUMBER: _ClassVar[int] MODELS_FIELD_NUMBER: _ClassVar[int] run_id: str models: _containers.RepeatedCompositeFieldContainer[ModelOutput] def __init__(self, run_id: _Optional[str] = ..., models: _Optional[_Iterable[_Union[ModelOutput, _Mapping]]] = ...) -> None: ... class GetExperimentByName(_message.Message): __slots__ = ("experiment_name",) class Response(_message.Message): __slots__ = ("experiment",) EXPERIMENT_FIELD_NUMBER: _ClassVar[int] experiment: Experiment def __init__(self, experiment: _Optional[_Union[Experiment, _Mapping]] = ...) -> None: ... EXPERIMENT_NAME_FIELD_NUMBER: _ClassVar[int] experiment_name: str def __init__(self, experiment_name: _Optional[str] = ...) -> None: ... class CreateAssessment(_message.Message): __slots__ = ("assessment",) class Response(_message.Message): __slots__ = ("assessment",) ASSESSMENT_FIELD_NUMBER: _ClassVar[int] assessment: _assessments_pb2.Assessment def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ...) -> None: ... ASSESSMENT_FIELD_NUMBER: _ClassVar[int] assessment: _assessments_pb2.Assessment def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ...) -> None: ... class UpdateAssessment(_message.Message): __slots__ = ("assessment", "update_mask") class Response(_message.Message): __slots__ = ("assessment",) ASSESSMENT_FIELD_NUMBER: _ClassVar[int] assessment: _assessments_pb2.Assessment def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ...) -> None: ... ASSESSMENT_FIELD_NUMBER: _ClassVar[int] UPDATE_MASK_FIELD_NUMBER: _ClassVar[int] assessment: _assessments_pb2.Assessment update_mask: _field_mask_pb2.FieldMask def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ..., update_mask: _Optional[_Union[_field_mask_pb2.FieldMask, _Mapping]] = ...) -> None: ... class DeleteAssessment(_message.Message): __slots__ = ("trace_id", "assessment_id") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] ASSESSMENT_ID_FIELD_NUMBER: _ClassVar[int] trace_id: str assessment_id: str def __init__(self, trace_id: _Optional[str] = ..., assessment_id: _Optional[str] = ...) -> None: ... class GetAssessmentRequest(_message.Message): __slots__ = ("trace_id", "assessment_id") class Response(_message.Message): __slots__ = ("assessment",) ASSESSMENT_FIELD_NUMBER: _ClassVar[int] assessment: _assessments_pb2.Assessment def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ...) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] ASSESSMENT_ID_FIELD_NUMBER: _ClassVar[int] trace_id: str assessment_id: str def __init__(self, trace_id: _Optional[str] = ..., assessment_id: _Optional[str] = ...) -> None: ... class TraceInfo(_message.Message): __slots__ = ("request_id", "experiment_id", "timestamp_ms", "execution_time_ms", "status", "request_metadata", "tags") REQUEST_ID_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] EXECUTION_TIME_MS_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] REQUEST_METADATA_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] request_id: str experiment_id: str timestamp_ms: int execution_time_ms: int status: TraceStatus request_metadata: _containers.RepeatedCompositeFieldContainer[TraceRequestMetadata] tags: _containers.RepeatedCompositeFieldContainer[TraceTag] def __init__(self, request_id: _Optional[str] = ..., experiment_id: _Optional[str] = ..., timestamp_ms: _Optional[int] = ..., execution_time_ms: _Optional[int] = ..., status: _Optional[_Union[TraceStatus, str]] = ..., request_metadata: _Optional[_Iterable[_Union[TraceRequestMetadata, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[TraceTag, _Mapping]]] = ...) -> None: ... class TraceRequestMetadata(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class TraceTag(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class StartTrace(_message.Message): __slots__ = ("experiment_id", "timestamp_ms", "request_metadata", "tags") class Response(_message.Message): __slots__ = ("trace_info",) TRACE_INFO_FIELD_NUMBER: _ClassVar[int] trace_info: TraceInfo def __init__(self, trace_info: _Optional[_Union[TraceInfo, _Mapping]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] REQUEST_METADATA_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] experiment_id: str timestamp_ms: int request_metadata: _containers.RepeatedCompositeFieldContainer[TraceRequestMetadata] tags: _containers.RepeatedCompositeFieldContainer[TraceTag] def __init__(self, experiment_id: _Optional[str] = ..., timestamp_ms: _Optional[int] = ..., request_metadata: _Optional[_Iterable[_Union[TraceRequestMetadata, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[TraceTag, _Mapping]]] = ...) -> None: ... class EndTrace(_message.Message): __slots__ = ("request_id", "timestamp_ms", "status", "request_metadata", "tags") class Response(_message.Message): __slots__ = ("trace_info",) TRACE_INFO_FIELD_NUMBER: _ClassVar[int] trace_info: TraceInfo def __init__(self, trace_info: _Optional[_Union[TraceInfo, _Mapping]] = ...) -> None: ... REQUEST_ID_FIELD_NUMBER: _ClassVar[int] TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] REQUEST_METADATA_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] request_id: str timestamp_ms: int status: TraceStatus request_metadata: _containers.RepeatedCompositeFieldContainer[TraceRequestMetadata] tags: _containers.RepeatedCompositeFieldContainer[TraceTag] def __init__(self, request_id: _Optional[str] = ..., timestamp_ms: _Optional[int] = ..., status: _Optional[_Union[TraceStatus, str]] = ..., request_metadata: _Optional[_Iterable[_Union[TraceRequestMetadata, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[TraceTag, _Mapping]]] = ...) -> None: ... class GetTraceInfo(_message.Message): __slots__ = ("request_id",) class Response(_message.Message): __slots__ = ("trace_info",) TRACE_INFO_FIELD_NUMBER: _ClassVar[int] trace_info: TraceInfo def __init__(self, trace_info: _Optional[_Union[TraceInfo, _Mapping]] = ...) -> None: ... REQUEST_ID_FIELD_NUMBER: _ClassVar[int] request_id: str def __init__(self, request_id: _Optional[str] = ...) -> None: ... class GetTraceInfoV3(_message.Message): __slots__ = ("trace_id",) class Response(_message.Message): __slots__ = ("trace",) TRACE_FIELD_NUMBER: _ClassVar[int] trace: Trace def __init__(self, trace: _Optional[_Union[Trace, _Mapping]] = ...) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] trace_id: str def __init__(self, trace_id: _Optional[str] = ...) -> None: ... class BatchGetTraces(_message.Message): __slots__ = ("trace_ids",) class Response(_message.Message): __slots__ = ("traces",) TRACES_FIELD_NUMBER: _ClassVar[int] traces: _containers.RepeatedCompositeFieldContainer[Trace] def __init__(self, traces: _Optional[_Iterable[_Union[Trace, _Mapping]]] = ...) -> None: ... TRACE_IDS_FIELD_NUMBER: _ClassVar[int] trace_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, trace_ids: _Optional[_Iterable[str]] = ...) -> None: ... class BatchGetTraceInfos(_message.Message): __slots__ = ("trace_ids",) class Response(_message.Message): __slots__ = ("trace_infos",) TRACE_INFOS_FIELD_NUMBER: _ClassVar[int] trace_infos: _containers.RepeatedCompositeFieldContainer[TraceInfoV3] def __init__(self, trace_infos: _Optional[_Iterable[_Union[TraceInfoV3, _Mapping]]] = ...) -> None: ... TRACE_IDS_FIELD_NUMBER: _ClassVar[int] trace_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, trace_ids: _Optional[_Iterable[str]] = ...) -> None: ... class GetTrace(_message.Message): __slots__ = ("trace_id", "allow_partial") class Response(_message.Message): __slots__ = ("trace",) TRACE_FIELD_NUMBER: _ClassVar[int] trace: Trace def __init__(self, trace: _Optional[_Union[Trace, _Mapping]] = ...) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] ALLOW_PARTIAL_FIELD_NUMBER: _ClassVar[int] trace_id: str allow_partial: bool def __init__(self, trace_id: _Optional[str] = ..., allow_partial: bool = ...) -> None: ... class SearchTraces(_message.Message): __slots__ = ("experiment_ids", "filter", "max_results", "order_by", "page_token") class Response(_message.Message): __slots__ = ("traces", "next_page_token") TRACES_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] traces: _containers.RepeatedCompositeFieldContainer[TraceInfo] next_page_token: str def __init__(self, traces: _Optional[_Iterable[_Union[TraceInfo, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] FILTER_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] ORDER_BY_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] experiment_ids: _containers.RepeatedScalarFieldContainer[str] filter: str max_results: int order_by: _containers.RepeatedScalarFieldContainer[str] page_token: str def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ..., filter: _Optional[str] = ..., max_results: _Optional[int] = ..., order_by: _Optional[_Iterable[str]] = ..., page_token: _Optional[str] = ...) -> None: ... class SearchUnifiedTraces(_message.Message): __slots__ = ("model_id", "sql_warehouse_id", "experiment_ids", "filter", "max_results", "order_by", "page_token") class Response(_message.Message): __slots__ = ("traces", "next_page_token") TRACES_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] traces: _containers.RepeatedCompositeFieldContainer[TraceInfo] next_page_token: str def __init__(self, traces: _Optional[_Iterable[_Union[TraceInfo, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... MODEL_ID_FIELD_NUMBER: _ClassVar[int] SQL_WAREHOUSE_ID_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] FILTER_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] ORDER_BY_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] model_id: str sql_warehouse_id: str experiment_ids: _containers.RepeatedScalarFieldContainer[str] filter: str max_results: int order_by: _containers.RepeatedScalarFieldContainer[str] page_token: str def __init__(self, model_id: _Optional[str] = ..., sql_warehouse_id: _Optional[str] = ..., experiment_ids: _Optional[_Iterable[str]] = ..., filter: _Optional[str] = ..., max_results: _Optional[int] = ..., order_by: _Optional[_Iterable[str]] = ..., page_token: _Optional[str] = ...) -> None: ... class GetOnlineTraceDetails(_message.Message): __slots__ = ("trace_id", "sql_warehouse_id", "source_inference_table", "source_databricks_request_id") class Response(_message.Message): __slots__ = ("trace_data",) TRACE_DATA_FIELD_NUMBER: _ClassVar[int] trace_data: str def __init__(self, trace_data: _Optional[str] = ...) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] SQL_WAREHOUSE_ID_FIELD_NUMBER: _ClassVar[int] SOURCE_INFERENCE_TABLE_FIELD_NUMBER: _ClassVar[int] SOURCE_DATABRICKS_REQUEST_ID_FIELD_NUMBER: _ClassVar[int] trace_id: str sql_warehouse_id: str source_inference_table: str source_databricks_request_id: str def __init__(self, trace_id: _Optional[str] = ..., sql_warehouse_id: _Optional[str] = ..., source_inference_table: _Optional[str] = ..., source_databricks_request_id: _Optional[str] = ...) -> None: ... class DeleteTraces(_message.Message): __slots__ = ("experiment_id", "max_timestamp_millis", "max_traces", "request_ids") class Response(_message.Message): __slots__ = ("traces_deleted",) TRACES_DELETED_FIELD_NUMBER: _ClassVar[int] traces_deleted: int def __init__(self, traces_deleted: _Optional[int] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] MAX_TIMESTAMP_MILLIS_FIELD_NUMBER: _ClassVar[int] MAX_TRACES_FIELD_NUMBER: _ClassVar[int] REQUEST_IDS_FIELD_NUMBER: _ClassVar[int] experiment_id: str max_timestamp_millis: int max_traces: int request_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, experiment_id: _Optional[str] = ..., max_timestamp_millis: _Optional[int] = ..., max_traces: _Optional[int] = ..., request_ids: _Optional[_Iterable[str]] = ...) -> None: ... class DeleteTracesV3(_message.Message): __slots__ = ("experiment_id", "max_timestamp_millis", "max_traces", "request_ids") class Response(_message.Message): __slots__ = ("traces_deleted",) TRACES_DELETED_FIELD_NUMBER: _ClassVar[int] traces_deleted: int def __init__(self, traces_deleted: _Optional[int] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] MAX_TIMESTAMP_MILLIS_FIELD_NUMBER: _ClassVar[int] MAX_TRACES_FIELD_NUMBER: _ClassVar[int] REQUEST_IDS_FIELD_NUMBER: _ClassVar[int] experiment_id: str max_timestamp_millis: int max_traces: int request_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, experiment_id: _Optional[str] = ..., max_timestamp_millis: _Optional[int] = ..., max_traces: _Optional[int] = ..., request_ids: _Optional[_Iterable[str]] = ...) -> None: ... class CalculateTraceFilterCorrelation(_message.Message): __slots__ = ("experiment_ids", "filter_string1", "filter_string2", "base_filter") class Response(_message.Message): __slots__ = ("npmi", "npmi_smoothed", "filter1_count", "filter2_count", "joint_count", "total_count") NPMI_FIELD_NUMBER: _ClassVar[int] NPMI_SMOOTHED_FIELD_NUMBER: _ClassVar[int] FILTER1_COUNT_FIELD_NUMBER: _ClassVar[int] FILTER2_COUNT_FIELD_NUMBER: _ClassVar[int] JOINT_COUNT_FIELD_NUMBER: _ClassVar[int] TOTAL_COUNT_FIELD_NUMBER: _ClassVar[int] npmi: float npmi_smoothed: float filter1_count: int filter2_count: int joint_count: int total_count: int def __init__(self, npmi: _Optional[float] = ..., npmi_smoothed: _Optional[float] = ..., filter1_count: _Optional[int] = ..., filter2_count: _Optional[int] = ..., joint_count: _Optional[int] = ..., total_count: _Optional[int] = ...) -> None: ... EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] FILTER_STRING1_FIELD_NUMBER: _ClassVar[int] FILTER_STRING2_FIELD_NUMBER: _ClassVar[int] BASE_FILTER_FIELD_NUMBER: _ClassVar[int] experiment_ids: _containers.RepeatedScalarFieldContainer[str] filter_string1: str filter_string2: str base_filter: str def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ..., filter_string1: _Optional[str] = ..., filter_string2: _Optional[str] = ..., base_filter: _Optional[str] = ...) -> None: ... class MetricAggregation(_message.Message): __slots__ = ("aggregation_type", "percentile_value") AGGREGATION_TYPE_FIELD_NUMBER: _ClassVar[int] PERCENTILE_VALUE_FIELD_NUMBER: _ClassVar[int] aggregation_type: AggregationType percentile_value: float def __init__(self, aggregation_type: _Optional[_Union[AggregationType, str]] = ..., percentile_value: _Optional[float] = ...) -> None: ... class QueryTraceMetrics(_message.Message): __slots__ = ("experiment_ids", "view_type", "metric_name", "aggregations", "dimensions", "filters", "time_interval_seconds", "start_time_ms", "end_time_ms", "max_results", "page_token") class Response(_message.Message): __slots__ = ("data_points", "next_page_token") DATA_POINTS_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] data_points: _containers.RepeatedCompositeFieldContainer[MetricDataPoint] next_page_token: str def __init__(self, data_points: _Optional[_Iterable[_Union[MetricDataPoint, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] VIEW_TYPE_FIELD_NUMBER: _ClassVar[int] METRIC_NAME_FIELD_NUMBER: _ClassVar[int] AGGREGATIONS_FIELD_NUMBER: _ClassVar[int] DIMENSIONS_FIELD_NUMBER: _ClassVar[int] FILTERS_FIELD_NUMBER: _ClassVar[int] TIME_INTERVAL_SECONDS_FIELD_NUMBER: _ClassVar[int] START_TIME_MS_FIELD_NUMBER: _ClassVar[int] END_TIME_MS_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] experiment_ids: _containers.RepeatedScalarFieldContainer[str] view_type: MetricViewType metric_name: str aggregations: _containers.RepeatedCompositeFieldContainer[MetricAggregation] dimensions: _containers.RepeatedScalarFieldContainer[str] filters: _containers.RepeatedScalarFieldContainer[str] time_interval_seconds: int start_time_ms: int end_time_ms: int max_results: int page_token: str def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ..., view_type: _Optional[_Union[MetricViewType, str]] = ..., metric_name: _Optional[str] = ..., aggregations: _Optional[_Iterable[_Union[MetricAggregation, _Mapping]]] = ..., dimensions: _Optional[_Iterable[str]] = ..., filters: _Optional[_Iterable[str]] = ..., time_interval_seconds: _Optional[int] = ..., start_time_ms: _Optional[int] = ..., end_time_ms: _Optional[int] = ..., max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... class MetricDataPoint(_message.Message): __slots__ = ("metric_name", "dimensions", "values") class DimensionsEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class ValuesEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: float def __init__(self, key: _Optional[str] = ..., value: _Optional[float] = ...) -> None: ... METRIC_NAME_FIELD_NUMBER: _ClassVar[int] DIMENSIONS_FIELD_NUMBER: _ClassVar[int] VALUES_FIELD_NUMBER: _ClassVar[int] metric_name: str dimensions: _containers.ScalarMap[str, str] values: _containers.ScalarMap[str, float] def __init__(self, metric_name: _Optional[str] = ..., dimensions: _Optional[_Mapping[str, str]] = ..., values: _Optional[_Mapping[str, float]] = ...) -> None: ... class SetTraceTag(_message.Message): __slots__ = ("request_id", "key", "value") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... REQUEST_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] request_id: str key: str value: str def __init__(self, request_id: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class SetTraceTagV3(_message.Message): __slots__ = ("trace_id", "key", "value") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] trace_id: str key: str value: str def __init__(self, trace_id: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class DeleteTraceTag(_message.Message): __slots__ = ("request_id", "key") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... REQUEST_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] request_id: str key: str def __init__(self, request_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... class DeleteTraceTagV3(_message.Message): __slots__ = ("trace_id", "key") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] trace_id: str key: str def __init__(self, trace_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... class Trace(_message.Message): __slots__ = ("trace_info", "spans") TRACE_INFO_FIELD_NUMBER: _ClassVar[int] SPANS_FIELD_NUMBER: _ClassVar[int] trace_info: TraceInfoV3 spans: _containers.RepeatedCompositeFieldContainer[_trace_pb2.Span] def __init__(self, trace_info: _Optional[_Union[TraceInfoV3, _Mapping]] = ..., spans: _Optional[_Iterable[_Union[_trace_pb2.Span, _Mapping]]] = ...) -> None: ... class TraceLocation(_message.Message): __slots__ = ("type", "mlflow_experiment", "inference_table") class TraceLocationType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () TRACE_LOCATION_TYPE_UNSPECIFIED: _ClassVar[TraceLocation.TraceLocationType] MLFLOW_EXPERIMENT: _ClassVar[TraceLocation.TraceLocationType] INFERENCE_TABLE: _ClassVar[TraceLocation.TraceLocationType] TRACE_LOCATION_TYPE_UNSPECIFIED: TraceLocation.TraceLocationType MLFLOW_EXPERIMENT: TraceLocation.TraceLocationType INFERENCE_TABLE: TraceLocation.TraceLocationType class MlflowExperimentLocation(_message.Message): __slots__ = ("experiment_id",) EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] experiment_id: str def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... class InferenceTableLocation(_message.Message): __slots__ = ("full_table_name",) FULL_TABLE_NAME_FIELD_NUMBER: _ClassVar[int] full_table_name: str def __init__(self, full_table_name: _Optional[str] = ...) -> None: ... TYPE_FIELD_NUMBER: _ClassVar[int] MLFLOW_EXPERIMENT_FIELD_NUMBER: _ClassVar[int] INFERENCE_TABLE_FIELD_NUMBER: _ClassVar[int] type: TraceLocation.TraceLocationType mlflow_experiment: TraceLocation.MlflowExperimentLocation inference_table: TraceLocation.InferenceTableLocation def __init__(self, type: _Optional[_Union[TraceLocation.TraceLocationType, str]] = ..., mlflow_experiment: _Optional[_Union[TraceLocation.MlflowExperimentLocation, _Mapping]] = ..., inference_table: _Optional[_Union[TraceLocation.InferenceTableLocation, _Mapping]] = ...) -> None: ... class TraceInfoV3(_message.Message): __slots__ = ("trace_id", "client_request_id", "trace_location", "request", "response", "request_preview", "response_preview", "request_time", "execution_duration", "state", "trace_metadata", "assessments", "tags") class State(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): __slots__ = () STATE_UNSPECIFIED: _ClassVar[TraceInfoV3.State] OK: _ClassVar[TraceInfoV3.State] ERROR: _ClassVar[TraceInfoV3.State] IN_PROGRESS: _ClassVar[TraceInfoV3.State] STATE_UNSPECIFIED: TraceInfoV3.State OK: TraceInfoV3.State ERROR: TraceInfoV3.State IN_PROGRESS: TraceInfoV3.State class TraceMetadataEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class TagsEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] CLIENT_REQUEST_ID_FIELD_NUMBER: _ClassVar[int] TRACE_LOCATION_FIELD_NUMBER: _ClassVar[int] REQUEST_FIELD_NUMBER: _ClassVar[int] RESPONSE_FIELD_NUMBER: _ClassVar[int] REQUEST_PREVIEW_FIELD_NUMBER: _ClassVar[int] RESPONSE_PREVIEW_FIELD_NUMBER: _ClassVar[int] REQUEST_TIME_FIELD_NUMBER: _ClassVar[int] EXECUTION_DURATION_FIELD_NUMBER: _ClassVar[int] STATE_FIELD_NUMBER: _ClassVar[int] TRACE_METADATA_FIELD_NUMBER: _ClassVar[int] ASSESSMENTS_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] trace_id: str client_request_id: str trace_location: TraceLocation request: str response: str request_preview: str response_preview: str request_time: _timestamp_pb2.Timestamp execution_duration: _duration_pb2.Duration state: TraceInfoV3.State trace_metadata: _containers.ScalarMap[str, str] assessments: _containers.RepeatedCompositeFieldContainer[_assessments_pb2.Assessment] tags: _containers.ScalarMap[str, str] def __init__(self, trace_id: _Optional[str] = ..., client_request_id: _Optional[str] = ..., trace_location: _Optional[_Union[TraceLocation, _Mapping]] = ..., request: _Optional[str] = ..., response: _Optional[str] = ..., request_preview: _Optional[str] = ..., response_preview: _Optional[str] = ..., request_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., execution_duration: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ..., state: _Optional[_Union[TraceInfoV3.State, str]] = ..., trace_metadata: _Optional[_Mapping[str, str]] = ..., assessments: _Optional[_Iterable[_Union[_assessments_pb2.Assessment, _Mapping]]] = ..., tags: _Optional[_Mapping[str, str]] = ...) -> None: ... class StartTraceV3(_message.Message): __slots__ = ("trace",) class Response(_message.Message): __slots__ = ("trace",) TRACE_FIELD_NUMBER: _ClassVar[int] trace: Trace def __init__(self, trace: _Optional[_Union[Trace, _Mapping]] = ...) -> None: ... TRACE_FIELD_NUMBER: _ClassVar[int] trace: Trace def __init__(self, trace: _Optional[_Union[Trace, _Mapping]] = ...) -> None: ... class LinkTracesToRun(_message.Message): __slots__ = ("trace_ids", "run_id") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... TRACE_IDS_FIELD_NUMBER: _ClassVar[int] RUN_ID_FIELD_NUMBER: _ClassVar[int] trace_ids: _containers.RepeatedScalarFieldContainer[str] run_id: str def __init__(self, trace_ids: _Optional[_Iterable[str]] = ..., run_id: _Optional[str] = ...) -> None: ... class LinkPromptsToTrace(_message.Message): __slots__ = ("trace_id", "prompt_versions") class PromptVersionRef(_message.Message): __slots__ = ("name", "version") NAME_FIELD_NUMBER: _ClassVar[int] VERSION_FIELD_NUMBER: _ClassVar[int] name: str version: str def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... TRACE_ID_FIELD_NUMBER: _ClassVar[int] PROMPT_VERSIONS_FIELD_NUMBER: _ClassVar[int] trace_id: str prompt_versions: _containers.RepeatedCompositeFieldContainer[LinkPromptsToTrace.PromptVersionRef] def __init__(self, trace_id: _Optional[str] = ..., prompt_versions: _Optional[_Iterable[_Union[LinkPromptsToTrace.PromptVersionRef, _Mapping]]] = ...) -> None: ... class DatasetSummary(_message.Message): __slots__ = ("experiment_id", "name", "digest", "context") EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] DIGEST_FIELD_NUMBER: _ClassVar[int] CONTEXT_FIELD_NUMBER: _ClassVar[int] experiment_id: str name: str digest: str context: str def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., digest: _Optional[str] = ..., context: _Optional[str] = ...) -> None: ... class SearchDatasets(_message.Message): __slots__ = ("experiment_ids",) class Response(_message.Message): __slots__ = ("dataset_summaries",) DATASET_SUMMARIES_FIELD_NUMBER: _ClassVar[int] dataset_summaries: _containers.RepeatedCompositeFieldContainer[DatasetSummary] def __init__(self, dataset_summaries: _Optional[_Iterable[_Union[DatasetSummary, _Mapping]]] = ...) -> None: ... EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] experiment_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ...) -> None: ... class CreateLoggedModel(_message.Message): __slots__ = ("experiment_id", "name", "model_type", "source_run_id", "params", "tags") class Response(_message.Message): __slots__ = ("model",) MODEL_FIELD_NUMBER: _ClassVar[int] model: LoggedModel def __init__(self, model: _Optional[_Union[LoggedModel, _Mapping]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] MODEL_TYPE_FIELD_NUMBER: _ClassVar[int] SOURCE_RUN_ID_FIELD_NUMBER: _ClassVar[int] PARAMS_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] experiment_id: str name: str model_type: str source_run_id: str params: _containers.RepeatedCompositeFieldContainer[LoggedModelParameter] tags: _containers.RepeatedCompositeFieldContainer[LoggedModelTag] def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., model_type: _Optional[str] = ..., source_run_id: _Optional[str] = ..., params: _Optional[_Iterable[_Union[LoggedModelParameter, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[LoggedModelTag, _Mapping]]] = ...) -> None: ... class FinalizeLoggedModel(_message.Message): __slots__ = ("model_id", "status") class Response(_message.Message): __slots__ = ("model",) MODEL_FIELD_NUMBER: _ClassVar[int] model: LoggedModel def __init__(self, model: _Optional[_Union[LoggedModel, _Mapping]] = ...) -> None: ... MODEL_ID_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] model_id: str status: LoggedModelStatus def __init__(self, model_id: _Optional[str] = ..., status: _Optional[_Union[LoggedModelStatus, str]] = ...) -> None: ... class GetLoggedModel(_message.Message): __slots__ = ("model_id",) class Response(_message.Message): __slots__ = ("model",) MODEL_FIELD_NUMBER: _ClassVar[int] model: LoggedModel def __init__(self, model: _Optional[_Union[LoggedModel, _Mapping]] = ...) -> None: ... MODEL_ID_FIELD_NUMBER: _ClassVar[int] model_id: str def __init__(self, model_id: _Optional[str] = ...) -> None: ... class DeleteLoggedModel(_message.Message): __slots__ = ("model_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... MODEL_ID_FIELD_NUMBER: _ClassVar[int] model_id: str def __init__(self, model_id: _Optional[str] = ...) -> None: ... class SearchLoggedModels(_message.Message): __slots__ = ("experiment_ids", "filter", "datasets", "max_results", "order_by", "page_token") class Dataset(_message.Message): __slots__ = ("dataset_name", "dataset_digest") DATASET_NAME_FIELD_NUMBER: _ClassVar[int] DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] dataset_name: str dataset_digest: str def __init__(self, dataset_name: _Optional[str] = ..., dataset_digest: _Optional[str] = ...) -> None: ... class OrderBy(_message.Message): __slots__ = ("field_name", "ascending", "dataset_name", "dataset_digest") FIELD_NAME_FIELD_NUMBER: _ClassVar[int] ASCENDING_FIELD_NUMBER: _ClassVar[int] DATASET_NAME_FIELD_NUMBER: _ClassVar[int] DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] field_name: str ascending: bool dataset_name: str dataset_digest: str def __init__(self, field_name: _Optional[str] = ..., ascending: bool = ..., dataset_name: _Optional[str] = ..., dataset_digest: _Optional[str] = ...) -> None: ... class Response(_message.Message): __slots__ = ("models", "next_page_token") MODELS_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] models: _containers.RepeatedCompositeFieldContainer[LoggedModel] next_page_token: str def __init__(self, models: _Optional[_Iterable[_Union[LoggedModel, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] FILTER_FIELD_NUMBER: _ClassVar[int] DATASETS_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] ORDER_BY_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] experiment_ids: _containers.RepeatedScalarFieldContainer[str] filter: str datasets: _containers.RepeatedCompositeFieldContainer[SearchLoggedModels.Dataset] max_results: int order_by: _containers.RepeatedCompositeFieldContainer[SearchLoggedModels.OrderBy] page_token: str def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ..., filter: _Optional[str] = ..., datasets: _Optional[_Iterable[_Union[SearchLoggedModels.Dataset, _Mapping]]] = ..., max_results: _Optional[int] = ..., order_by: _Optional[_Iterable[_Union[SearchLoggedModels.OrderBy, _Mapping]]] = ..., page_token: _Optional[str] = ...) -> None: ... class SetLoggedModelTags(_message.Message): __slots__ = ("model_id", "tags") class Response(_message.Message): __slots__ = ("model",) MODEL_FIELD_NUMBER: _ClassVar[int] model: LoggedModel def __init__(self, model: _Optional[_Union[LoggedModel, _Mapping]] = ...) -> None: ... MODEL_ID_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] model_id: str tags: _containers.RepeatedCompositeFieldContainer[LoggedModelTag] def __init__(self, model_id: _Optional[str] = ..., tags: _Optional[_Iterable[_Union[LoggedModelTag, _Mapping]]] = ...) -> None: ... class DeleteLoggedModelTag(_message.Message): __slots__ = ("model_id", "tag_key") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... MODEL_ID_FIELD_NUMBER: _ClassVar[int] TAG_KEY_FIELD_NUMBER: _ClassVar[int] model_id: str tag_key: str def __init__(self, model_id: _Optional[str] = ..., tag_key: _Optional[str] = ...) -> None: ... class ListLoggedModelArtifacts(_message.Message): __slots__ = ("model_id", "artifact_directory_path", "page_token") class Response(_message.Message): __slots__ = ("root_uri", "files", "next_page_token") ROOT_URI_FIELD_NUMBER: _ClassVar[int] FILES_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] root_uri: str files: _containers.RepeatedCompositeFieldContainer[FileInfo] next_page_token: str def __init__(self, root_uri: _Optional[str] = ..., files: _Optional[_Iterable[_Union[FileInfo, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... MODEL_ID_FIELD_NUMBER: _ClassVar[int] ARTIFACT_DIRECTORY_PATH_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] model_id: str artifact_directory_path: str page_token: str def __init__(self, model_id: _Optional[str] = ..., artifact_directory_path: _Optional[str] = ..., page_token: _Optional[str] = ...) -> None: ... class LogLoggedModelParamsRequest(_message.Message): __slots__ = ("model_id", "params") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... MODEL_ID_FIELD_NUMBER: _ClassVar[int] PARAMS_FIELD_NUMBER: _ClassVar[int] model_id: str params: _containers.RepeatedCompositeFieldContainer[LoggedModelParameter] def __init__(self, model_id: _Optional[str] = ..., params: _Optional[_Iterable[_Union[LoggedModelParameter, _Mapping]]] = ...) -> None: ... class LoggedModel(_message.Message): __slots__ = ("info", "data") INFO_FIELD_NUMBER: _ClassVar[int] DATA_FIELD_NUMBER: _ClassVar[int] info: LoggedModelInfo data: LoggedModelData def __init__(self, info: _Optional[_Union[LoggedModelInfo, _Mapping]] = ..., data: _Optional[_Union[LoggedModelData, _Mapping]] = ...) -> None: ... class LoggedModelInfo(_message.Message): __slots__ = ("model_id", "experiment_id", "name", "creation_timestamp_ms", "last_updated_timestamp_ms", "artifact_uri", "status", "creator_id", "model_type", "source_run_id", "status_message", "tags", "registrations") MODEL_ID_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] CREATION_TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] ARTIFACT_URI_FIELD_NUMBER: _ClassVar[int] STATUS_FIELD_NUMBER: _ClassVar[int] CREATOR_ID_FIELD_NUMBER: _ClassVar[int] MODEL_TYPE_FIELD_NUMBER: _ClassVar[int] SOURCE_RUN_ID_FIELD_NUMBER: _ClassVar[int] STATUS_MESSAGE_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] REGISTRATIONS_FIELD_NUMBER: _ClassVar[int] model_id: str experiment_id: str name: str creation_timestamp_ms: int last_updated_timestamp_ms: int artifact_uri: str status: LoggedModelStatus creator_id: int model_type: str source_run_id: str status_message: str tags: _containers.RepeatedCompositeFieldContainer[LoggedModelTag] registrations: _containers.RepeatedCompositeFieldContainer[LoggedModelRegistrationInfo] def __init__(self, model_id: _Optional[str] = ..., experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., creation_timestamp_ms: _Optional[int] = ..., last_updated_timestamp_ms: _Optional[int] = ..., artifact_uri: _Optional[str] = ..., status: _Optional[_Union[LoggedModelStatus, str]] = ..., creator_id: _Optional[int] = ..., model_type: _Optional[str] = ..., source_run_id: _Optional[str] = ..., status_message: _Optional[str] = ..., tags: _Optional[_Iterable[_Union[LoggedModelTag, _Mapping]]] = ..., registrations: _Optional[_Iterable[_Union[LoggedModelRegistrationInfo, _Mapping]]] = ...) -> None: ... class LoggedModelTag(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class LoggedModelRegistrationInfo(_message.Message): __slots__ = ("name", "version") NAME_FIELD_NUMBER: _ClassVar[int] VERSION_FIELD_NUMBER: _ClassVar[int] name: str version: str def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... class LoggedModelData(_message.Message): __slots__ = ("params", "metrics") PARAMS_FIELD_NUMBER: _ClassVar[int] METRICS_FIELD_NUMBER: _ClassVar[int] params: _containers.RepeatedCompositeFieldContainer[LoggedModelParameter] metrics: _containers.RepeatedCompositeFieldContainer[Metric] def __init__(self, params: _Optional[_Iterable[_Union[LoggedModelParameter, _Mapping]]] = ..., metrics: _Optional[_Iterable[_Union[Metric, _Mapping]]] = ...) -> None: ... class LoggedModelParameter(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class SearchTracesV3(_message.Message): __slots__ = ("locations", "filter", "max_results", "order_by", "page_token") class Response(_message.Message): __slots__ = ("traces", "next_page_token") TRACES_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] traces: _containers.RepeatedCompositeFieldContainer[TraceInfoV3] next_page_token: str def __init__(self, traces: _Optional[_Iterable[_Union[TraceInfoV3, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... LOCATIONS_FIELD_NUMBER: _ClassVar[int] FILTER_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] ORDER_BY_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] locations: _containers.RepeatedCompositeFieldContainer[TraceLocation] filter: str max_results: int order_by: _containers.RepeatedScalarFieldContainer[str] page_token: str def __init__(self, locations: _Optional[_Iterable[_Union[TraceLocation, _Mapping]]] = ..., filter: _Optional[str] = ..., max_results: _Optional[int] = ..., order_by: _Optional[_Iterable[str]] = ..., page_token: _Optional[str] = ...) -> None: ... class CreateDataset(_message.Message): __slots__ = ("name", "experiment_ids", "source_type", "source", "schema", "profile", "created_by", "tags") class Response(_message.Message): __slots__ = ("dataset",) DATASET_FIELD_NUMBER: _ClassVar[int] dataset: _datasets_pb2.Dataset def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ...) -> None: ... NAME_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] SOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] SOURCE_FIELD_NUMBER: _ClassVar[int] SCHEMA_FIELD_NUMBER: _ClassVar[int] PROFILE_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] name: str experiment_ids: _containers.RepeatedScalarFieldContainer[str] source_type: _datasets_pb2.DatasetRecordSource.SourceType source: str schema: str profile: str created_by: str tags: str def __init__(self, name: _Optional[str] = ..., experiment_ids: _Optional[_Iterable[str]] = ..., source_type: _Optional[_Union[_datasets_pb2.DatasetRecordSource.SourceType, str]] = ..., source: _Optional[str] = ..., schema: _Optional[str] = ..., profile: _Optional[str] = ..., created_by: _Optional[str] = ..., tags: _Optional[str] = ...) -> None: ... class GetDataset(_message.Message): __slots__ = ("dataset_id", "page_token") class Response(_message.Message): __slots__ = ("dataset", "next_page_token") DATASET_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] dataset: _datasets_pb2.Dataset next_page_token: str def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ..., next_page_token: _Optional[str] = ...) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] dataset_id: str page_token: str def __init__(self, dataset_id: _Optional[str] = ..., page_token: _Optional[str] = ...) -> None: ... class DeleteDataset(_message.Message): __slots__ = ("dataset_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] dataset_id: str def __init__(self, dataset_id: _Optional[str] = ...) -> None: ... class SearchEvaluationDatasets(_message.Message): __slots__ = ("experiment_ids", "filter_string", "max_results", "order_by", "page_token") class Response(_message.Message): __slots__ = ("datasets", "next_page_token") DATASETS_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] datasets: _containers.RepeatedCompositeFieldContainer[_datasets_pb2.Dataset] next_page_token: str def __init__(self, datasets: _Optional[_Iterable[_Union[_datasets_pb2.Dataset, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] FILTER_STRING_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] ORDER_BY_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] experiment_ids: _containers.RepeatedScalarFieldContainer[str] filter_string: str max_results: int order_by: _containers.RepeatedScalarFieldContainer[str] page_token: str def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ..., filter_string: _Optional[str] = ..., max_results: _Optional[int] = ..., order_by: _Optional[_Iterable[str]] = ..., page_token: _Optional[str] = ...) -> None: ... class SetDatasetTags(_message.Message): __slots__ = ("dataset_id", "tags") class Response(_message.Message): __slots__ = ("dataset",) DATASET_FIELD_NUMBER: _ClassVar[int] dataset: _datasets_pb2.Dataset def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ...) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] dataset_id: str tags: str def __init__(self, dataset_id: _Optional[str] = ..., tags: _Optional[str] = ...) -> None: ... class DeleteDatasetTag(_message.Message): __slots__ = ("dataset_id", "key") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] dataset_id: str key: str def __init__(self, dataset_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... class UpsertDatasetRecords(_message.Message): __slots__ = ("dataset_id", "records", "updated_by") class Response(_message.Message): __slots__ = ("inserted_count", "updated_count") INSERTED_COUNT_FIELD_NUMBER: _ClassVar[int] UPDATED_COUNT_FIELD_NUMBER: _ClassVar[int] inserted_count: int updated_count: int def __init__(self, inserted_count: _Optional[int] = ..., updated_count: _Optional[int] = ...) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] RECORDS_FIELD_NUMBER: _ClassVar[int] UPDATED_BY_FIELD_NUMBER: _ClassVar[int] dataset_id: str records: str updated_by: str def __init__(self, dataset_id: _Optional[str] = ..., records: _Optional[str] = ..., updated_by: _Optional[str] = ...) -> None: ... class GetDatasetExperimentIds(_message.Message): __slots__ = ("dataset_id",) class Response(_message.Message): __slots__ = ("experiment_ids",) EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] experiment_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ...) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] dataset_id: str def __init__(self, dataset_id: _Optional[str] = ...) -> None: ... class GetDatasetRecords(_message.Message): __slots__ = ("dataset_id", "max_results", "page_token") class Response(_message.Message): __slots__ = ("records", "next_page_token") RECORDS_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] records: str next_page_token: str def __init__(self, records: _Optional[str] = ..., next_page_token: _Optional[str] = ...) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] dataset_id: str max_results: int page_token: str def __init__(self, dataset_id: _Optional[str] = ..., max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... class DeleteDatasetRecords(_message.Message): __slots__ = ("dataset_id", "dataset_record_ids") class Response(_message.Message): __slots__ = ("deleted_count",) DELETED_COUNT_FIELD_NUMBER: _ClassVar[int] deleted_count: int def __init__(self, deleted_count: _Optional[int] = ...) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] DATASET_RECORD_IDS_FIELD_NUMBER: _ClassVar[int] dataset_id: str dataset_record_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, dataset_id: _Optional[str] = ..., dataset_record_ids: _Optional[_Iterable[str]] = ...) -> None: ... class AddDatasetToExperiments(_message.Message): __slots__ = ("dataset_id", "experiment_ids") class Response(_message.Message): __slots__ = ("dataset",) DATASET_FIELD_NUMBER: _ClassVar[int] dataset: _datasets_pb2.Dataset def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ...) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] dataset_id: str experiment_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, dataset_id: _Optional[str] = ..., experiment_ids: _Optional[_Iterable[str]] = ...) -> None: ... class RemoveDatasetFromExperiments(_message.Message): __slots__ = ("dataset_id", "experiment_ids") class Response(_message.Message): __slots__ = ("dataset",) DATASET_FIELD_NUMBER: _ClassVar[int] dataset: _datasets_pb2.Dataset def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ...) -> None: ... DATASET_ID_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] dataset_id: str experiment_ids: _containers.RepeatedScalarFieldContainer[str] def __init__(self, dataset_id: _Optional[str] = ..., experiment_ids: _Optional[_Iterable[str]] = ...) -> None: ... class RegisterScorer(_message.Message): __slots__ = ("experiment_id", "name", "serialized_scorer") class Response(_message.Message): __slots__ = ("version", "scorer_id", "experiment_id", "name", "serialized_scorer", "creation_time") VERSION_FIELD_NUMBER: _ClassVar[int] SCORER_ID_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] SERIALIZED_SCORER_FIELD_NUMBER: _ClassVar[int] CREATION_TIME_FIELD_NUMBER: _ClassVar[int] version: int scorer_id: str experiment_id: str name: str serialized_scorer: str creation_time: int def __init__(self, version: _Optional[int] = ..., scorer_id: _Optional[str] = ..., experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., serialized_scorer: _Optional[str] = ..., creation_time: _Optional[int] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] SERIALIZED_SCORER_FIELD_NUMBER: _ClassVar[int] experiment_id: str name: str serialized_scorer: str def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., serialized_scorer: _Optional[str] = ...) -> None: ... class ListScorers(_message.Message): __slots__ = ("experiment_id",) class Response(_message.Message): __slots__ = ("scorers",) SCORERS_FIELD_NUMBER: _ClassVar[int] scorers: _containers.RepeatedCompositeFieldContainer[Scorer] def __init__(self, scorers: _Optional[_Iterable[_Union[Scorer, _Mapping]]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] experiment_id: str def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... class ListScorerVersions(_message.Message): __slots__ = ("experiment_id", "name") class Response(_message.Message): __slots__ = ("scorers",) SCORERS_FIELD_NUMBER: _ClassVar[int] scorers: _containers.RepeatedCompositeFieldContainer[Scorer] def __init__(self, scorers: _Optional[_Iterable[_Union[Scorer, _Mapping]]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] experiment_id: str name: str def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ... class GetScorer(_message.Message): __slots__ = ("experiment_id", "name", "version") class Response(_message.Message): __slots__ = ("scorer",) SCORER_FIELD_NUMBER: _ClassVar[int] scorer: Scorer def __init__(self, scorer: _Optional[_Union[Scorer, _Mapping]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] VERSION_FIELD_NUMBER: _ClassVar[int] experiment_id: str name: str version: int def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[int] = ...) -> None: ... class DeleteScorer(_message.Message): __slots__ = ("experiment_id", "name", "version") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] VERSION_FIELD_NUMBER: _ClassVar[int] experiment_id: str name: str version: int def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[int] = ...) -> None: ... class Scorer(_message.Message): __slots__ = ("experiment_id", "scorer_name", "scorer_version", "serialized_scorer", "creation_time", "scorer_id") EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] SCORER_NAME_FIELD_NUMBER: _ClassVar[int] SCORER_VERSION_FIELD_NUMBER: _ClassVar[int] SERIALIZED_SCORER_FIELD_NUMBER: _ClassVar[int] CREATION_TIME_FIELD_NUMBER: _ClassVar[int] SCORER_ID_FIELD_NUMBER: _ClassVar[int] experiment_id: int scorer_name: str scorer_version: int serialized_scorer: str creation_time: int scorer_id: str def __init__(self, experiment_id: _Optional[int] = ..., scorer_name: _Optional[str] = ..., scorer_version: _Optional[int] = ..., serialized_scorer: _Optional[str] = ..., creation_time: _Optional[int] = ..., scorer_id: _Optional[str] = ...) -> None: ... class GatewaySecretInfo(_message.Message): __slots__ = ("secret_id", "secret_name", "masked_values", "created_at", "last_updated_at", "provider", "created_by", "last_updated_by", "auth_config") class MaskedValuesEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class AuthConfigEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... SECRET_ID_FIELD_NUMBER: _ClassVar[int] SECRET_NAME_FIELD_NUMBER: _ClassVar[int] MASKED_VALUES_FIELD_NUMBER: _ClassVar[int] CREATED_AT_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] PROVIDER_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] AUTH_CONFIG_FIELD_NUMBER: _ClassVar[int] secret_id: str secret_name: str masked_values: _containers.ScalarMap[str, str] created_at: int last_updated_at: int provider: str created_by: str last_updated_by: str auth_config: _containers.ScalarMap[str, str] def __init__(self, secret_id: _Optional[str] = ..., secret_name: _Optional[str] = ..., masked_values: _Optional[_Mapping[str, str]] = ..., created_at: _Optional[int] = ..., last_updated_at: _Optional[int] = ..., provider: _Optional[str] = ..., created_by: _Optional[str] = ..., last_updated_by: _Optional[str] = ..., auth_config: _Optional[_Mapping[str, str]] = ...) -> None: ... class GatewayModelDefinition(_message.Message): __slots__ = ("model_definition_id", "name", "secret_id", "secret_name", "provider", "model_name", "created_at", "last_updated_at", "created_by", "last_updated_by") MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] SECRET_ID_FIELD_NUMBER: _ClassVar[int] SECRET_NAME_FIELD_NUMBER: _ClassVar[int] PROVIDER_FIELD_NUMBER: _ClassVar[int] MODEL_NAME_FIELD_NUMBER: _ClassVar[int] CREATED_AT_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] model_definition_id: str name: str secret_id: str secret_name: str provider: str model_name: str created_at: int last_updated_at: int created_by: str last_updated_by: str def __init__(self, model_definition_id: _Optional[str] = ..., name: _Optional[str] = ..., secret_id: _Optional[str] = ..., secret_name: _Optional[str] = ..., provider: _Optional[str] = ..., model_name: _Optional[str] = ..., created_at: _Optional[int] = ..., last_updated_at: _Optional[int] = ..., created_by: _Optional[str] = ..., last_updated_by: _Optional[str] = ...) -> None: ... class GatewayEndpointModelMapping(_message.Message): __slots__ = ("mapping_id", "endpoint_id", "model_definition_id", "model_definition", "weight", "created_at", "created_by", "linkage_type", "fallback_order") MAPPING_ID_FIELD_NUMBER: _ClassVar[int] ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] MODEL_DEFINITION_FIELD_NUMBER: _ClassVar[int] WEIGHT_FIELD_NUMBER: _ClassVar[int] CREATED_AT_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] LINKAGE_TYPE_FIELD_NUMBER: _ClassVar[int] FALLBACK_ORDER_FIELD_NUMBER: _ClassVar[int] mapping_id: str endpoint_id: str model_definition_id: str model_definition: GatewayModelDefinition weight: float created_at: int created_by: str linkage_type: GatewayModelLinkageType fallback_order: int def __init__(self, mapping_id: _Optional[str] = ..., endpoint_id: _Optional[str] = ..., model_definition_id: _Optional[str] = ..., model_definition: _Optional[_Union[GatewayModelDefinition, _Mapping]] = ..., weight: _Optional[float] = ..., created_at: _Optional[int] = ..., created_by: _Optional[str] = ..., linkage_type: _Optional[_Union[GatewayModelLinkageType, str]] = ..., fallback_order: _Optional[int] = ...) -> None: ... class GatewayEndpoint(_message.Message): __slots__ = ("endpoint_id", "name", "created_at", "last_updated_at", "model_mappings", "created_by", "last_updated_by", "tags", "routing_strategy", "fallback_config", "experiment_id", "usage_tracking") ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] CREATED_AT_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] MODEL_MAPPINGS_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] ROUTING_STRATEGY_FIELD_NUMBER: _ClassVar[int] FALLBACK_CONFIG_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] USAGE_TRACKING_FIELD_NUMBER: _ClassVar[int] endpoint_id: str name: str created_at: int last_updated_at: int model_mappings: _containers.RepeatedCompositeFieldContainer[GatewayEndpointModelMapping] created_by: str last_updated_by: str tags: _containers.RepeatedCompositeFieldContainer[GatewayEndpointTag] routing_strategy: RoutingStrategy fallback_config: FallbackConfig experiment_id: str usage_tracking: bool def __init__(self, endpoint_id: _Optional[str] = ..., name: _Optional[str] = ..., created_at: _Optional[int] = ..., last_updated_at: _Optional[int] = ..., model_mappings: _Optional[_Iterable[_Union[GatewayEndpointModelMapping, _Mapping]]] = ..., created_by: _Optional[str] = ..., last_updated_by: _Optional[str] = ..., tags: _Optional[_Iterable[_Union[GatewayEndpointTag, _Mapping]]] = ..., routing_strategy: _Optional[_Union[RoutingStrategy, str]] = ..., fallback_config: _Optional[_Union[FallbackConfig, _Mapping]] = ..., experiment_id: _Optional[str] = ..., usage_tracking: bool = ...) -> None: ... class GatewayEndpointTag(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class GatewayEndpointBinding(_message.Message): __slots__ = ("endpoint_id", "resource_type", "resource_id", "created_at", "last_updated_at", "created_by", "last_updated_by", "display_name") ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] RESOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] RESOURCE_ID_FIELD_NUMBER: _ClassVar[int] CREATED_AT_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] DISPLAY_NAME_FIELD_NUMBER: _ClassVar[int] endpoint_id: str resource_type: str resource_id: str created_at: int last_updated_at: int created_by: str last_updated_by: str display_name: str def __init__(self, endpoint_id: _Optional[str] = ..., resource_type: _Optional[str] = ..., resource_id: _Optional[str] = ..., created_at: _Optional[int] = ..., last_updated_at: _Optional[int] = ..., created_by: _Optional[str] = ..., last_updated_by: _Optional[str] = ..., display_name: _Optional[str] = ...) -> None: ... class CreateGatewaySecret(_message.Message): __slots__ = ("secret_name", "secret_value", "provider", "auth_config", "created_by") class SecretValueEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class AuthConfigEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class Response(_message.Message): __slots__ = ("secret",) SECRET_FIELD_NUMBER: _ClassVar[int] secret: GatewaySecretInfo def __init__(self, secret: _Optional[_Union[GatewaySecretInfo, _Mapping]] = ...) -> None: ... SECRET_NAME_FIELD_NUMBER: _ClassVar[int] SECRET_VALUE_FIELD_NUMBER: _ClassVar[int] PROVIDER_FIELD_NUMBER: _ClassVar[int] AUTH_CONFIG_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] secret_name: str secret_value: _containers.ScalarMap[str, str] provider: str auth_config: _containers.ScalarMap[str, str] created_by: str def __init__(self, secret_name: _Optional[str] = ..., secret_value: _Optional[_Mapping[str, str]] = ..., provider: _Optional[str] = ..., auth_config: _Optional[_Mapping[str, str]] = ..., created_by: _Optional[str] = ...) -> None: ... class GetGatewaySecretInfo(_message.Message): __slots__ = ("secret_id", "secret_name") class Response(_message.Message): __slots__ = ("secret",) SECRET_FIELD_NUMBER: _ClassVar[int] secret: GatewaySecretInfo def __init__(self, secret: _Optional[_Union[GatewaySecretInfo, _Mapping]] = ...) -> None: ... SECRET_ID_FIELD_NUMBER: _ClassVar[int] SECRET_NAME_FIELD_NUMBER: _ClassVar[int] secret_id: str secret_name: str def __init__(self, secret_id: _Optional[str] = ..., secret_name: _Optional[str] = ...) -> None: ... class UpdateGatewaySecret(_message.Message): __slots__ = ("secret_id", "secret_value", "auth_config", "updated_by") class SecretValueEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class AuthConfigEntry(_message.Message): __slots__ = ("key", "value") KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] key: str value: str def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class Response(_message.Message): __slots__ = ("secret",) SECRET_FIELD_NUMBER: _ClassVar[int] secret: GatewaySecretInfo def __init__(self, secret: _Optional[_Union[GatewaySecretInfo, _Mapping]] = ...) -> None: ... SECRET_ID_FIELD_NUMBER: _ClassVar[int] SECRET_VALUE_FIELD_NUMBER: _ClassVar[int] AUTH_CONFIG_FIELD_NUMBER: _ClassVar[int] UPDATED_BY_FIELD_NUMBER: _ClassVar[int] secret_id: str secret_value: _containers.ScalarMap[str, str] auth_config: _containers.ScalarMap[str, str] updated_by: str def __init__(self, secret_id: _Optional[str] = ..., secret_value: _Optional[_Mapping[str, str]] = ..., auth_config: _Optional[_Mapping[str, str]] = ..., updated_by: _Optional[str] = ...) -> None: ... class DeleteGatewaySecret(_message.Message): __slots__ = ("secret_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... SECRET_ID_FIELD_NUMBER: _ClassVar[int] secret_id: str def __init__(self, secret_id: _Optional[str] = ...) -> None: ... class ListGatewaySecretInfos(_message.Message): __slots__ = ("provider",) class Response(_message.Message): __slots__ = ("secrets",) SECRETS_FIELD_NUMBER: _ClassVar[int] secrets: _containers.RepeatedCompositeFieldContainer[GatewaySecretInfo] def __init__(self, secrets: _Optional[_Iterable[_Union[GatewaySecretInfo, _Mapping]]] = ...) -> None: ... PROVIDER_FIELD_NUMBER: _ClassVar[int] provider: str def __init__(self, provider: _Optional[str] = ...) -> None: ... class CreateGatewayModelDefinition(_message.Message): __slots__ = ("name", "secret_id", "provider", "model_name", "created_by") class Response(_message.Message): __slots__ = ("model_definition",) MODEL_DEFINITION_FIELD_NUMBER: _ClassVar[int] model_definition: GatewayModelDefinition def __init__(self, model_definition: _Optional[_Union[GatewayModelDefinition, _Mapping]] = ...) -> None: ... NAME_FIELD_NUMBER: _ClassVar[int] SECRET_ID_FIELD_NUMBER: _ClassVar[int] PROVIDER_FIELD_NUMBER: _ClassVar[int] MODEL_NAME_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] name: str secret_id: str provider: str model_name: str created_by: str def __init__(self, name: _Optional[str] = ..., secret_id: _Optional[str] = ..., provider: _Optional[str] = ..., model_name: _Optional[str] = ..., created_by: _Optional[str] = ...) -> None: ... class GetGatewayModelDefinition(_message.Message): __slots__ = ("model_definition_id",) class Response(_message.Message): __slots__ = ("model_definition",) MODEL_DEFINITION_FIELD_NUMBER: _ClassVar[int] model_definition: GatewayModelDefinition def __init__(self, model_definition: _Optional[_Union[GatewayModelDefinition, _Mapping]] = ...) -> None: ... MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] model_definition_id: str def __init__(self, model_definition_id: _Optional[str] = ...) -> None: ... class ListGatewayModelDefinitions(_message.Message): __slots__ = ("provider", "secret_id") class Response(_message.Message): __slots__ = ("model_definitions",) MODEL_DEFINITIONS_FIELD_NUMBER: _ClassVar[int] model_definitions: _containers.RepeatedCompositeFieldContainer[GatewayModelDefinition] def __init__(self, model_definitions: _Optional[_Iterable[_Union[GatewayModelDefinition, _Mapping]]] = ...) -> None: ... PROVIDER_FIELD_NUMBER: _ClassVar[int] SECRET_ID_FIELD_NUMBER: _ClassVar[int] provider: str secret_id: str def __init__(self, provider: _Optional[str] = ..., secret_id: _Optional[str] = ...) -> None: ... class UpdateGatewayModelDefinition(_message.Message): __slots__ = ("model_definition_id", "name", "secret_id", "model_name", "updated_by", "provider") class Response(_message.Message): __slots__ = ("model_definition",) MODEL_DEFINITION_FIELD_NUMBER: _ClassVar[int] model_definition: GatewayModelDefinition def __init__(self, model_definition: _Optional[_Union[GatewayModelDefinition, _Mapping]] = ...) -> None: ... MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] SECRET_ID_FIELD_NUMBER: _ClassVar[int] MODEL_NAME_FIELD_NUMBER: _ClassVar[int] UPDATED_BY_FIELD_NUMBER: _ClassVar[int] PROVIDER_FIELD_NUMBER: _ClassVar[int] model_definition_id: str name: str secret_id: str model_name: str updated_by: str provider: str def __init__(self, model_definition_id: _Optional[str] = ..., name: _Optional[str] = ..., secret_id: _Optional[str] = ..., model_name: _Optional[str] = ..., updated_by: _Optional[str] = ..., provider: _Optional[str] = ...) -> None: ... class DeleteGatewayModelDefinition(_message.Message): __slots__ = ("model_definition_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] model_definition_id: str def __init__(self, model_definition_id: _Optional[str] = ...) -> None: ... class BudgetDuration(_message.Message): __slots__ = ("unit", "value") UNIT_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] unit: BudgetDurationUnit value: int def __init__(self, unit: _Optional[_Union[BudgetDurationUnit, str]] = ..., value: _Optional[int] = ...) -> None: ... class FallbackConfig(_message.Message): __slots__ = ("strategy", "max_attempts") STRATEGY_FIELD_NUMBER: _ClassVar[int] MAX_ATTEMPTS_FIELD_NUMBER: _ClassVar[int] strategy: FallbackStrategy max_attempts: int def __init__(self, strategy: _Optional[_Union[FallbackStrategy, str]] = ..., max_attempts: _Optional[int] = ...) -> None: ... class GatewayEndpointModelConfig(_message.Message): __slots__ = ("model_definition_id", "linkage_type", "weight", "fallback_order") MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] LINKAGE_TYPE_FIELD_NUMBER: _ClassVar[int] WEIGHT_FIELD_NUMBER: _ClassVar[int] FALLBACK_ORDER_FIELD_NUMBER: _ClassVar[int] model_definition_id: str linkage_type: GatewayModelLinkageType weight: float fallback_order: int def __init__(self, model_definition_id: _Optional[str] = ..., linkage_type: _Optional[_Union[GatewayModelLinkageType, str]] = ..., weight: _Optional[float] = ..., fallback_order: _Optional[int] = ...) -> None: ... class CreateGatewayEndpoint(_message.Message): __slots__ = ("name", "model_configs", "created_by", "routing_strategy", "fallback_config", "experiment_id", "usage_tracking") class Response(_message.Message): __slots__ = ("endpoint",) ENDPOINT_FIELD_NUMBER: _ClassVar[int] endpoint: GatewayEndpoint def __init__(self, endpoint: _Optional[_Union[GatewayEndpoint, _Mapping]] = ...) -> None: ... NAME_FIELD_NUMBER: _ClassVar[int] MODEL_CONFIGS_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] ROUTING_STRATEGY_FIELD_NUMBER: _ClassVar[int] FALLBACK_CONFIG_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] USAGE_TRACKING_FIELD_NUMBER: _ClassVar[int] name: str model_configs: _containers.RepeatedCompositeFieldContainer[GatewayEndpointModelConfig] created_by: str routing_strategy: RoutingStrategy fallback_config: FallbackConfig experiment_id: str usage_tracking: bool def __init__(self, name: _Optional[str] = ..., model_configs: _Optional[_Iterable[_Union[GatewayEndpointModelConfig, _Mapping]]] = ..., created_by: _Optional[str] = ..., routing_strategy: _Optional[_Union[RoutingStrategy, str]] = ..., fallback_config: _Optional[_Union[FallbackConfig, _Mapping]] = ..., experiment_id: _Optional[str] = ..., usage_tracking: bool = ...) -> None: ... class GetGatewayEndpoint(_message.Message): __slots__ = ("endpoint_id", "name") class Response(_message.Message): __slots__ = ("endpoint",) ENDPOINT_FIELD_NUMBER: _ClassVar[int] endpoint: GatewayEndpoint def __init__(self, endpoint: _Optional[_Union[GatewayEndpoint, _Mapping]] = ...) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] endpoint_id: str name: str def __init__(self, endpoint_id: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ... class UpdateGatewayEndpoint(_message.Message): __slots__ = ("endpoint_id", "name", "updated_by", "model_configs", "routing_strategy", "fallback_config", "experiment_id", "usage_tracking") class Response(_message.Message): __slots__ = ("endpoint",) ENDPOINT_FIELD_NUMBER: _ClassVar[int] endpoint: GatewayEndpoint def __init__(self, endpoint: _Optional[_Union[GatewayEndpoint, _Mapping]] = ...) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] UPDATED_BY_FIELD_NUMBER: _ClassVar[int] MODEL_CONFIGS_FIELD_NUMBER: _ClassVar[int] ROUTING_STRATEGY_FIELD_NUMBER: _ClassVar[int] FALLBACK_CONFIG_FIELD_NUMBER: _ClassVar[int] EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] USAGE_TRACKING_FIELD_NUMBER: _ClassVar[int] endpoint_id: str name: str updated_by: str model_configs: _containers.RepeatedCompositeFieldContainer[GatewayEndpointModelConfig] routing_strategy: RoutingStrategy fallback_config: FallbackConfig experiment_id: str usage_tracking: bool def __init__(self, endpoint_id: _Optional[str] = ..., name: _Optional[str] = ..., updated_by: _Optional[str] = ..., model_configs: _Optional[_Iterable[_Union[GatewayEndpointModelConfig, _Mapping]]] = ..., routing_strategy: _Optional[_Union[RoutingStrategy, str]] = ..., fallback_config: _Optional[_Union[FallbackConfig, _Mapping]] = ..., experiment_id: _Optional[str] = ..., usage_tracking: bool = ...) -> None: ... class DeleteGatewayEndpoint(_message.Message): __slots__ = ("endpoint_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] endpoint_id: str def __init__(self, endpoint_id: _Optional[str] = ...) -> None: ... class ListGatewayEndpoints(_message.Message): __slots__ = ("provider", "secret_id") class Response(_message.Message): __slots__ = ("endpoints",) ENDPOINTS_FIELD_NUMBER: _ClassVar[int] endpoints: _containers.RepeatedCompositeFieldContainer[GatewayEndpoint] def __init__(self, endpoints: _Optional[_Iterable[_Union[GatewayEndpoint, _Mapping]]] = ...) -> None: ... PROVIDER_FIELD_NUMBER: _ClassVar[int] SECRET_ID_FIELD_NUMBER: _ClassVar[int] provider: str secret_id: str def __init__(self, provider: _Optional[str] = ..., secret_id: _Optional[str] = ...) -> None: ... class AttachModelToGatewayEndpoint(_message.Message): __slots__ = ("endpoint_id", "model_config", "created_by") class Response(_message.Message): __slots__ = ("mapping",) MAPPING_FIELD_NUMBER: _ClassVar[int] mapping: GatewayEndpointModelMapping def __init__(self, mapping: _Optional[_Union[GatewayEndpointModelMapping, _Mapping]] = ...) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] MODEL_CONFIG_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] endpoint_id: str model_config: GatewayEndpointModelConfig created_by: str def __init__(self, endpoint_id: _Optional[str] = ..., model_config: _Optional[_Union[GatewayEndpointModelConfig, _Mapping]] = ..., created_by: _Optional[str] = ...) -> None: ... class DetachModelFromGatewayEndpoint(_message.Message): __slots__ = ("endpoint_id", "model_definition_id") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] endpoint_id: str model_definition_id: str def __init__(self, endpoint_id: _Optional[str] = ..., model_definition_id: _Optional[str] = ...) -> None: ... class CreateGatewayEndpointBinding(_message.Message): __slots__ = ("endpoint_id", "resource_type", "resource_id", "created_by") class Response(_message.Message): __slots__ = ("binding",) BINDING_FIELD_NUMBER: _ClassVar[int] binding: GatewayEndpointBinding def __init__(self, binding: _Optional[_Union[GatewayEndpointBinding, _Mapping]] = ...) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] RESOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] RESOURCE_ID_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] endpoint_id: str resource_type: str resource_id: str created_by: str def __init__(self, endpoint_id: _Optional[str] = ..., resource_type: _Optional[str] = ..., resource_id: _Optional[str] = ..., created_by: _Optional[str] = ...) -> None: ... class DeleteGatewayEndpointBinding(_message.Message): __slots__ = ("endpoint_id", "resource_type", "resource_id") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] RESOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] RESOURCE_ID_FIELD_NUMBER: _ClassVar[int] endpoint_id: str resource_type: str resource_id: str def __init__(self, endpoint_id: _Optional[str] = ..., resource_type: _Optional[str] = ..., resource_id: _Optional[str] = ...) -> None: ... class ListGatewayEndpointBindings(_message.Message): __slots__ = ("endpoint_id", "resource_type", "resource_id") class Response(_message.Message): __slots__ = ("bindings",) BINDINGS_FIELD_NUMBER: _ClassVar[int] bindings: _containers.RepeatedCompositeFieldContainer[GatewayEndpointBinding] def __init__(self, bindings: _Optional[_Iterable[_Union[GatewayEndpointBinding, _Mapping]]] = ...) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] RESOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] RESOURCE_ID_FIELD_NUMBER: _ClassVar[int] endpoint_id: str resource_type: str resource_id: str def __init__(self, endpoint_id: _Optional[str] = ..., resource_type: _Optional[str] = ..., resource_id: _Optional[str] = ...) -> None: ... class SetGatewayEndpointTag(_message.Message): __slots__ = ("endpoint_id", "key", "value") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] VALUE_FIELD_NUMBER: _ClassVar[int] endpoint_id: str key: str value: str def __init__(self, endpoint_id: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... class DeleteGatewayEndpointTag(_message.Message): __slots__ = ("endpoint_id", "key") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] KEY_FIELD_NUMBER: _ClassVar[int] endpoint_id: str key: str def __init__(self, endpoint_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... class GatewayBudgetPolicy(_message.Message): __slots__ = ("budget_policy_id", "budget_unit", "budget_amount", "duration", "target_scope", "budget_action", "created_by", "created_at", "last_updated_by", "last_updated_at") BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] BUDGET_UNIT_FIELD_NUMBER: _ClassVar[int] BUDGET_AMOUNT_FIELD_NUMBER: _ClassVar[int] DURATION_FIELD_NUMBER: _ClassVar[int] TARGET_SCOPE_FIELD_NUMBER: _ClassVar[int] BUDGET_ACTION_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] CREATED_AT_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] budget_policy_id: str budget_unit: BudgetUnit budget_amount: float duration: BudgetDuration target_scope: BudgetTargetScope budget_action: BudgetAction created_by: str created_at: int last_updated_by: str last_updated_at: int def __init__(self, budget_policy_id: _Optional[str] = ..., budget_unit: _Optional[_Union[BudgetUnit, str]] = ..., budget_amount: _Optional[float] = ..., duration: _Optional[_Union[BudgetDuration, _Mapping]] = ..., target_scope: _Optional[_Union[BudgetTargetScope, str]] = ..., budget_action: _Optional[_Union[BudgetAction, str]] = ..., created_by: _Optional[str] = ..., created_at: _Optional[int] = ..., last_updated_by: _Optional[str] = ..., last_updated_at: _Optional[int] = ...) -> None: ... class CreateGatewayBudgetPolicy(_message.Message): __slots__ = ("budget_unit", "budget_amount", "duration", "target_scope", "budget_action", "created_by") class Response(_message.Message): __slots__ = ("budget_policy",) BUDGET_POLICY_FIELD_NUMBER: _ClassVar[int] budget_policy: GatewayBudgetPolicy def __init__(self, budget_policy: _Optional[_Union[GatewayBudgetPolicy, _Mapping]] = ...) -> None: ... BUDGET_UNIT_FIELD_NUMBER: _ClassVar[int] BUDGET_AMOUNT_FIELD_NUMBER: _ClassVar[int] DURATION_FIELD_NUMBER: _ClassVar[int] TARGET_SCOPE_FIELD_NUMBER: _ClassVar[int] BUDGET_ACTION_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] budget_unit: BudgetUnit budget_amount: float duration: BudgetDuration target_scope: BudgetTargetScope budget_action: BudgetAction created_by: str def __init__(self, budget_unit: _Optional[_Union[BudgetUnit, str]] = ..., budget_amount: _Optional[float] = ..., duration: _Optional[_Union[BudgetDuration, _Mapping]] = ..., target_scope: _Optional[_Union[BudgetTargetScope, str]] = ..., budget_action: _Optional[_Union[BudgetAction, str]] = ..., created_by: _Optional[str] = ...) -> None: ... class GetGatewayBudgetPolicy(_message.Message): __slots__ = ("budget_policy_id",) class Response(_message.Message): __slots__ = ("budget_policy",) BUDGET_POLICY_FIELD_NUMBER: _ClassVar[int] budget_policy: GatewayBudgetPolicy def __init__(self, budget_policy: _Optional[_Union[GatewayBudgetPolicy, _Mapping]] = ...) -> None: ... BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] budget_policy_id: str def __init__(self, budget_policy_id: _Optional[str] = ...) -> None: ... class UpdateGatewayBudgetPolicy(_message.Message): __slots__ = ("budget_policy_id", "budget_unit", "budget_amount", "duration", "target_scope", "budget_action", "updated_by") class Response(_message.Message): __slots__ = ("budget_policy",) BUDGET_POLICY_FIELD_NUMBER: _ClassVar[int] budget_policy: GatewayBudgetPolicy def __init__(self, budget_policy: _Optional[_Union[GatewayBudgetPolicy, _Mapping]] = ...) -> None: ... BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] BUDGET_UNIT_FIELD_NUMBER: _ClassVar[int] BUDGET_AMOUNT_FIELD_NUMBER: _ClassVar[int] DURATION_FIELD_NUMBER: _ClassVar[int] TARGET_SCOPE_FIELD_NUMBER: _ClassVar[int] BUDGET_ACTION_FIELD_NUMBER: _ClassVar[int] UPDATED_BY_FIELD_NUMBER: _ClassVar[int] budget_policy_id: str budget_unit: BudgetUnit budget_amount: float duration: BudgetDuration target_scope: BudgetTargetScope budget_action: BudgetAction updated_by: str def __init__(self, budget_policy_id: _Optional[str] = ..., budget_unit: _Optional[_Union[BudgetUnit, str]] = ..., budget_amount: _Optional[float] = ..., duration: _Optional[_Union[BudgetDuration, _Mapping]] = ..., target_scope: _Optional[_Union[BudgetTargetScope, str]] = ..., budget_action: _Optional[_Union[BudgetAction, str]] = ..., updated_by: _Optional[str] = ...) -> None: ... class DeleteGatewayBudgetPolicy(_message.Message): __slots__ = ("budget_policy_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] budget_policy_id: str def __init__(self, budget_policy_id: _Optional[str] = ...) -> None: ... class ListGatewayBudgetPolicies(_message.Message): __slots__ = ("max_results", "page_token") class Response(_message.Message): __slots__ = ("budget_policies", "next_page_token") BUDGET_POLICIES_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] budget_policies: _containers.RepeatedCompositeFieldContainer[GatewayBudgetPolicy] next_page_token: str def __init__(self, budget_policies: _Optional[_Iterable[_Union[GatewayBudgetPolicy, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] max_results: int page_token: str def __init__(self, max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... class ListGatewayBudgetWindows(_message.Message): __slots__ = () class BudgetWindow(_message.Message): __slots__ = ("budget_policy_id", "window_start_ms", "window_end_ms", "current_spend") BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] WINDOW_START_MS_FIELD_NUMBER: _ClassVar[int] WINDOW_END_MS_FIELD_NUMBER: _ClassVar[int] CURRENT_SPEND_FIELD_NUMBER: _ClassVar[int] budget_policy_id: str window_start_ms: int window_end_ms: int current_spend: float def __init__(self, budget_policy_id: _Optional[str] = ..., window_start_ms: _Optional[int] = ..., window_end_ms: _Optional[int] = ..., current_spend: _Optional[float] = ...) -> None: ... class Response(_message.Message): __slots__ = ("windows",) WINDOWS_FIELD_NUMBER: _ClassVar[int] windows: _containers.RepeatedCompositeFieldContainer[ListGatewayBudgetWindows.BudgetWindow] def __init__(self, windows: _Optional[_Iterable[_Union[ListGatewayBudgetWindows.BudgetWindow, _Mapping]]] = ...) -> None: ... def __init__(self) -> None: ... class GatewayGuardrail(_message.Message): __slots__ = ("guardrail_id", "name", "scorer", "stage", "action", "action_endpoint_id", "created_by", "created_at", "last_updated_by", "last_updated_at") GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] NAME_FIELD_NUMBER: _ClassVar[int] SCORER_FIELD_NUMBER: _ClassVar[int] STAGE_FIELD_NUMBER: _ClassVar[int] ACTION_FIELD_NUMBER: _ClassVar[int] ACTION_ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] CREATED_AT_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] guardrail_id: str name: str scorer: Scorer stage: GuardrailStage action: GuardrailAction action_endpoint_id: str created_by: str created_at: int last_updated_by: str last_updated_at: int def __init__(self, guardrail_id: _Optional[str] = ..., name: _Optional[str] = ..., scorer: _Optional[_Union[Scorer, _Mapping]] = ..., stage: _Optional[_Union[GuardrailStage, str]] = ..., action: _Optional[_Union[GuardrailAction, str]] = ..., action_endpoint_id: _Optional[str] = ..., created_by: _Optional[str] = ..., created_at: _Optional[int] = ..., last_updated_by: _Optional[str] = ..., last_updated_at: _Optional[int] = ...) -> None: ... class GatewayGuardrailConfig(_message.Message): __slots__ = ("endpoint_id", "guardrail_id", "execution_order", "created_by", "created_at", "guardrail") ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] EXECUTION_ORDER_FIELD_NUMBER: _ClassVar[int] CREATED_BY_FIELD_NUMBER: _ClassVar[int] CREATED_AT_FIELD_NUMBER: _ClassVar[int] GUARDRAIL_FIELD_NUMBER: _ClassVar[int] endpoint_id: str guardrail_id: str execution_order: int created_by: str created_at: int guardrail: GatewayGuardrail def __init__(self, endpoint_id: _Optional[str] = ..., guardrail_id: _Optional[str] = ..., execution_order: _Optional[int] = ..., created_by: _Optional[str] = ..., created_at: _Optional[int] = ..., guardrail: _Optional[_Union[GatewayGuardrail, _Mapping]] = ...) -> None: ... class CreateGatewayGuardrail(_message.Message): __slots__ = ("name", "scorer_id", "scorer_version", "stage", "action", "action_endpoint_id") class Response(_message.Message): __slots__ = ("guardrail",) GUARDRAIL_FIELD_NUMBER: _ClassVar[int] guardrail: GatewayGuardrail def __init__(self, guardrail: _Optional[_Union[GatewayGuardrail, _Mapping]] = ...) -> None: ... NAME_FIELD_NUMBER: _ClassVar[int] SCORER_ID_FIELD_NUMBER: _ClassVar[int] SCORER_VERSION_FIELD_NUMBER: _ClassVar[int] STAGE_FIELD_NUMBER: _ClassVar[int] ACTION_FIELD_NUMBER: _ClassVar[int] ACTION_ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] name: str scorer_id: str scorer_version: int stage: GuardrailStage action: GuardrailAction action_endpoint_id: str def __init__(self, name: _Optional[str] = ..., scorer_id: _Optional[str] = ..., scorer_version: _Optional[int] = ..., stage: _Optional[_Union[GuardrailStage, str]] = ..., action: _Optional[_Union[GuardrailAction, str]] = ..., action_endpoint_id: _Optional[str] = ...) -> None: ... class GetGatewayGuardrail(_message.Message): __slots__ = ("guardrail_id",) class Response(_message.Message): __slots__ = ("guardrail",) GUARDRAIL_FIELD_NUMBER: _ClassVar[int] guardrail: GatewayGuardrail def __init__(self, guardrail: _Optional[_Union[GatewayGuardrail, _Mapping]] = ...) -> None: ... GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] guardrail_id: str def __init__(self, guardrail_id: _Optional[str] = ...) -> None: ... class DeleteGatewayGuardrail(_message.Message): __slots__ = ("guardrail_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] guardrail_id: str def __init__(self, guardrail_id: _Optional[str] = ...) -> None: ... class ListGatewayGuardrails(_message.Message): __slots__ = ("max_results", "page_token") class Response(_message.Message): __slots__ = ("guardrails", "next_page_token") GUARDRAILS_FIELD_NUMBER: _ClassVar[int] NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] guardrails: _containers.RepeatedCompositeFieldContainer[GatewayGuardrail] next_page_token: str def __init__(self, guardrails: _Optional[_Iterable[_Union[GatewayGuardrail, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] max_results: int page_token: str def __init__(self, max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... class AddGuardrailToEndpoint(_message.Message): __slots__ = ("endpoint_id", "guardrail_id", "execution_order") class Response(_message.Message): __slots__ = ("config",) CONFIG_FIELD_NUMBER: _ClassVar[int] config: GatewayGuardrailConfig def __init__(self, config: _Optional[_Union[GatewayGuardrailConfig, _Mapping]] = ...) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] EXECUTION_ORDER_FIELD_NUMBER: _ClassVar[int] endpoint_id: str guardrail_id: str execution_order: int def __init__(self, endpoint_id: _Optional[str] = ..., guardrail_id: _Optional[str] = ..., execution_order: _Optional[int] = ...) -> None: ... class RemoveGuardrailFromEndpoint(_message.Message): __slots__ = ("endpoint_id", "guardrail_id") class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] endpoint_id: str guardrail_id: str def __init__(self, endpoint_id: _Optional[str] = ..., guardrail_id: _Optional[str] = ...) -> None: ... class ListEndpointGuardrailConfigs(_message.Message): __slots__ = ("endpoint_id",) class Response(_message.Message): __slots__ = ("configs",) CONFIGS_FIELD_NUMBER: _ClassVar[int] configs: _containers.RepeatedCompositeFieldContainer[GatewayGuardrailConfig] def __init__(self, configs: _Optional[_Iterable[_Union[GatewayGuardrailConfig, _Mapping]]] = ...) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] endpoint_id: str def __init__(self, endpoint_id: _Optional[str] = ...) -> None: ... class UpdateEndpointGuardrailConfig(_message.Message): __slots__ = ("endpoint_id", "guardrail_id", "execution_order") class Response(_message.Message): __slots__ = ("config",) CONFIG_FIELD_NUMBER: _ClassVar[int] config: GatewayGuardrailConfig def __init__(self, config: _Optional[_Union[GatewayGuardrailConfig, _Mapping]] = ...) -> None: ... ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] EXECUTION_ORDER_FIELD_NUMBER: _ClassVar[int] endpoint_id: str guardrail_id: str execution_order: int def __init__(self, endpoint_id: _Optional[str] = ..., guardrail_id: _Optional[str] = ..., execution_order: _Optional[int] = ...) -> None: ... class GetSecretsConfig(_message.Message): __slots__ = () class Response(_message.Message): __slots__ = ("secrets_available",) SECRETS_AVAILABLE_FIELD_NUMBER: _ClassVar[int] secrets_available: bool def __init__(self, secrets_available: bool = ...) -> None: ... def __init__(self) -> None: ... class CreatePromptOptimizationJob(_message.Message): __slots__ = ("experiment_id", "source_prompt_uri", "config", "tags") class Response(_message.Message): __slots__ = ("job",) JOB_FIELD_NUMBER: _ClassVar[int] job: _prompt_optimization_pb2.PromptOptimizationJob def __init__(self, job: _Optional[_Union[_prompt_optimization_pb2.PromptOptimizationJob, _Mapping]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] SOURCE_PROMPT_URI_FIELD_NUMBER: _ClassVar[int] CONFIG_FIELD_NUMBER: _ClassVar[int] TAGS_FIELD_NUMBER: _ClassVar[int] experiment_id: str source_prompt_uri: str config: _prompt_optimization_pb2.PromptOptimizationJobConfig tags: _containers.RepeatedCompositeFieldContainer[_prompt_optimization_pb2.PromptOptimizationJobTag] def __init__(self, experiment_id: _Optional[str] = ..., source_prompt_uri: _Optional[str] = ..., config: _Optional[_Union[_prompt_optimization_pb2.PromptOptimizationJobConfig, _Mapping]] = ..., tags: _Optional[_Iterable[_Union[_prompt_optimization_pb2.PromptOptimizationJobTag, _Mapping]]] = ...) -> None: ... class GetPromptOptimizationJob(_message.Message): __slots__ = ("job_id",) class Response(_message.Message): __slots__ = ("job",) JOB_FIELD_NUMBER: _ClassVar[int] job: _prompt_optimization_pb2.PromptOptimizationJob def __init__(self, job: _Optional[_Union[_prompt_optimization_pb2.PromptOptimizationJob, _Mapping]] = ...) -> None: ... JOB_ID_FIELD_NUMBER: _ClassVar[int] job_id: str def __init__(self, job_id: _Optional[str] = ...) -> None: ... class SearchPromptOptimizationJobs(_message.Message): __slots__ = ("experiment_id",) class Response(_message.Message): __slots__ = ("jobs",) JOBS_FIELD_NUMBER: _ClassVar[int] jobs: _containers.RepeatedCompositeFieldContainer[_prompt_optimization_pb2.PromptOptimizationJob] def __init__(self, jobs: _Optional[_Iterable[_Union[_prompt_optimization_pb2.PromptOptimizationJob, _Mapping]]] = ...) -> None: ... EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] experiment_id: str def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... class CancelPromptOptimizationJob(_message.Message): __slots__ = ("job_id",) class Response(_message.Message): __slots__ = ("job",) JOB_FIELD_NUMBER: _ClassVar[int] job: _prompt_optimization_pb2.PromptOptimizationJob def __init__(self, job: _Optional[_Union[_prompt_optimization_pb2.PromptOptimizationJob, _Mapping]] = ...) -> None: ... JOB_ID_FIELD_NUMBER: _ClassVar[int] job_id: str def __init__(self, job_id: _Optional[str] = ...) -> None: ... class DeletePromptOptimizationJob(_message.Message): __slots__ = ("job_id",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... JOB_ID_FIELD_NUMBER: _ClassVar[int] job_id: str def __init__(self, job_id: _Optional[str] = ...) -> None: ... class TraceArchivalConfig(_message.Message): __slots__ = ("location", "retention") LOCATION_FIELD_NUMBER: _ClassVar[int] RETENTION_FIELD_NUMBER: _ClassVar[int] location: str retention: str def __init__(self, location: _Optional[str] = ..., retention: _Optional[str] = ...) -> None: ... class Workspace(_message.Message): __slots__ = ("name", "description", "default_artifact_root", "trace_archival_config") NAME_FIELD_NUMBER: _ClassVar[int] DESCRIPTION_FIELD_NUMBER: _ClassVar[int] DEFAULT_ARTIFACT_ROOT_FIELD_NUMBER: _ClassVar[int] TRACE_ARCHIVAL_CONFIG_FIELD_NUMBER: _ClassVar[int] name: str description: str default_artifact_root: str trace_archival_config: TraceArchivalConfig def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., default_artifact_root: _Optional[str] = ..., trace_archival_config: _Optional[_Union[TraceArchivalConfig, _Mapping]] = ...) -> None: ... class ListWorkspaces(_message.Message): __slots__ = () class Response(_message.Message): __slots__ = ("workspaces",) WORKSPACES_FIELD_NUMBER: _ClassVar[int] workspaces: _containers.RepeatedCompositeFieldContainer[Workspace] def __init__(self, workspaces: _Optional[_Iterable[_Union[Workspace, _Mapping]]] = ...) -> None: ... def __init__(self) -> None: ... class CreateWorkspace(_message.Message): __slots__ = ("name", "description", "default_artifact_root", "trace_archival_config") class Response(_message.Message): __slots__ = ("workspace",) WORKSPACE_FIELD_NUMBER: _ClassVar[int] workspace: Workspace def __init__(self, workspace: _Optional[_Union[Workspace, _Mapping]] = ...) -> None: ... NAME_FIELD_NUMBER: _ClassVar[int] DESCRIPTION_FIELD_NUMBER: _ClassVar[int] DEFAULT_ARTIFACT_ROOT_FIELD_NUMBER: _ClassVar[int] TRACE_ARCHIVAL_CONFIG_FIELD_NUMBER: _ClassVar[int] name: str description: str default_artifact_root: str trace_archival_config: TraceArchivalConfig def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., default_artifact_root: _Optional[str] = ..., trace_archival_config: _Optional[_Union[TraceArchivalConfig, _Mapping]] = ...) -> None: ... class GetWorkspace(_message.Message): __slots__ = ("workspace_name",) class Response(_message.Message): __slots__ = ("workspace",) WORKSPACE_FIELD_NUMBER: _ClassVar[int] workspace: Workspace def __init__(self, workspace: _Optional[_Union[Workspace, _Mapping]] = ...) -> None: ... WORKSPACE_NAME_FIELD_NUMBER: _ClassVar[int] workspace_name: str def __init__(self, workspace_name: _Optional[str] = ...) -> None: ... class UpdateWorkspace(_message.Message): __slots__ = ("workspace_name", "description", "default_artifact_root", "trace_archival_config") class Response(_message.Message): __slots__ = ("workspace",) WORKSPACE_FIELD_NUMBER: _ClassVar[int] workspace: Workspace def __init__(self, workspace: _Optional[_Union[Workspace, _Mapping]] = ...) -> None: ... WORKSPACE_NAME_FIELD_NUMBER: _ClassVar[int] DESCRIPTION_FIELD_NUMBER: _ClassVar[int] DEFAULT_ARTIFACT_ROOT_FIELD_NUMBER: _ClassVar[int] TRACE_ARCHIVAL_CONFIG_FIELD_NUMBER: _ClassVar[int] workspace_name: str description: str default_artifact_root: str trace_archival_config: TraceArchivalConfig def __init__(self, workspace_name: _Optional[str] = ..., description: _Optional[str] = ..., default_artifact_root: _Optional[str] = ..., trace_archival_config: _Optional[_Union[TraceArchivalConfig, _Mapping]] = ...) -> None: ... class DeleteWorkspace(_message.Message): __slots__ = ("workspace_name",) class Response(_message.Message): __slots__ = () def __init__(self) -> None: ... WORKSPACE_NAME_FIELD_NUMBER: _ClassVar[int] workspace_name: str def __init__(self, workspace_name: _Optional[str] = ...) -> None: ... class MlflowService(_service.service): ... class MlflowService_Stub(MlflowService): ...