service_pb2.pyi
1 import assessments_pb2 as _assessments_pb2 2 import databricks_pb2 as _databricks_pb2 3 import datasets_pb2 as _datasets_pb2 4 from google.protobuf import duration_pb2 as _duration_pb2 5 from google.protobuf import field_mask_pb2 as _field_mask_pb2 6 from google.protobuf import timestamp_pb2 as _timestamp_pb2 7 import issues_pb2 as _issues_pb2 8 from opentelemetry.proto.trace.v1 import trace_pb2 as _trace_pb2 9 import prompt_optimization_pb2 as _prompt_optimization_pb2 10 from scalapb import scalapb_pb2 as _scalapb_pb2 11 from google.protobuf.internal import containers as _containers 12 from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper 13 from google.protobuf import descriptor as _descriptor 14 from google.protobuf import message as _message 15 from google.protobuf import service as _service 16 from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union 17 18 DESCRIPTOR: _descriptor.FileDescriptor 19 20 class ViewType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 21 __slots__ = () 22 ACTIVE_ONLY: _ClassVar[ViewType] 23 DELETED_ONLY: _ClassVar[ViewType] 24 ALL: _ClassVar[ViewType] 25 26 class SourceType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 27 __slots__ = () 28 NOTEBOOK: _ClassVar[SourceType] 29 JOB: _ClassVar[SourceType] 30 PROJECT: _ClassVar[SourceType] 31 LOCAL: _ClassVar[SourceType] 32 UNKNOWN: _ClassVar[SourceType] 33 34 class RunStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 35 __slots__ = () 36 RUNNING: _ClassVar[RunStatus] 37 SCHEDULED: _ClassVar[RunStatus] 38 FINISHED: _ClassVar[RunStatus] 39 FAILED: _ClassVar[RunStatus] 40 KILLED: _ClassVar[RunStatus] 41 42 class TraceStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 43 __slots__ = () 44 TRACE_STATUS_UNSPECIFIED: _ClassVar[TraceStatus] 45 OK: _ClassVar[TraceStatus] 46 ERROR: _ClassVar[TraceStatus] 47 IN_PROGRESS: _ClassVar[TraceStatus] 48 49 class MetricViewType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 50 __slots__ = () 51 TRACES: _ClassVar[MetricViewType] 52 SPANS: _ClassVar[MetricViewType] 53 ASSESSMENTS: _ClassVar[MetricViewType] 54 55 class AggregationType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 56 __slots__ = () 57 COUNT: _ClassVar[AggregationType] 58 SUM: _ClassVar[AggregationType] 59 AVG: _ClassVar[AggregationType] 60 PERCENTILE: _ClassVar[AggregationType] 61 MIN: _ClassVar[AggregationType] 62 MAX: _ClassVar[AggregationType] 63 64 class LoggedModelStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 65 __slots__ = () 66 LOGGED_MODEL_STATUS_UNSPECIFIED: _ClassVar[LoggedModelStatus] 67 LOGGED_MODEL_PENDING: _ClassVar[LoggedModelStatus] 68 LOGGED_MODEL_READY: _ClassVar[LoggedModelStatus] 69 LOGGED_MODEL_UPLOAD_FAILED: _ClassVar[LoggedModelStatus] 70 71 class RoutingStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 72 __slots__ = () 73 ROUTING_STRATEGY_UNSPECIFIED: _ClassVar[RoutingStrategy] 74 REQUEST_BASED_TRAFFIC_SPLIT: _ClassVar[RoutingStrategy] 75 76 class FallbackStrategy(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 77 __slots__ = () 78 FALLBACK_STRATEGY_UNSPECIFIED: _ClassVar[FallbackStrategy] 79 SEQUENTIAL: _ClassVar[FallbackStrategy] 80 81 class GatewayModelLinkageType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 82 __slots__ = () 83 LINKAGE_TYPE_UNSPECIFIED: _ClassVar[GatewayModelLinkageType] 84 PRIMARY: _ClassVar[GatewayModelLinkageType] 85 FALLBACK: _ClassVar[GatewayModelLinkageType] 86 87 class BudgetDurationUnit(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 88 __slots__ = () 89 DURATION_UNIT_UNSPECIFIED: _ClassVar[BudgetDurationUnit] 90 MINUTES: _ClassVar[BudgetDurationUnit] 91 HOURS: _ClassVar[BudgetDurationUnit] 92 DAYS: _ClassVar[BudgetDurationUnit] 93 WEEKS: _ClassVar[BudgetDurationUnit] 94 MONTHS: _ClassVar[BudgetDurationUnit] 95 96 class BudgetTargetScope(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 97 __slots__ = () 98 TARGET_SCOPE_UNSPECIFIED: _ClassVar[BudgetTargetScope] 99 GLOBAL: _ClassVar[BudgetTargetScope] 100 WORKSPACE: _ClassVar[BudgetTargetScope] 101 102 class BudgetAction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 103 __slots__ = () 104 BUDGET_ACTION_UNSPECIFIED: _ClassVar[BudgetAction] 105 ALERT: _ClassVar[BudgetAction] 106 REJECT: _ClassVar[BudgetAction] 107 108 class BudgetUnit(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 109 __slots__ = () 110 BUDGET_UNIT_UNSPECIFIED: _ClassVar[BudgetUnit] 111 USD: _ClassVar[BudgetUnit] 112 113 class GuardrailStage(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 114 __slots__ = () 115 GUARDRAIL_STAGE_UNSPECIFIED: _ClassVar[GuardrailStage] 116 BEFORE: _ClassVar[GuardrailStage] 117 AFTER: _ClassVar[GuardrailStage] 118 119 class GuardrailAction(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 120 __slots__ = () 121 GUARDRAIL_ACTION_UNSPECIFIED: _ClassVar[GuardrailAction] 122 VALIDATION: _ClassVar[GuardrailAction] 123 SANITIZATION: _ClassVar[GuardrailAction] 124 ACTIVE_ONLY: ViewType 125 DELETED_ONLY: ViewType 126 ALL: ViewType 127 NOTEBOOK: SourceType 128 JOB: SourceType 129 PROJECT: SourceType 130 LOCAL: SourceType 131 UNKNOWN: SourceType 132 RUNNING: RunStatus 133 SCHEDULED: RunStatus 134 FINISHED: RunStatus 135 FAILED: RunStatus 136 KILLED: RunStatus 137 TRACE_STATUS_UNSPECIFIED: TraceStatus 138 OK: TraceStatus 139 ERROR: TraceStatus 140 IN_PROGRESS: TraceStatus 141 TRACES: MetricViewType 142 SPANS: MetricViewType 143 ASSESSMENTS: MetricViewType 144 COUNT: AggregationType 145 SUM: AggregationType 146 AVG: AggregationType 147 PERCENTILE: AggregationType 148 MIN: AggregationType 149 MAX: AggregationType 150 LOGGED_MODEL_STATUS_UNSPECIFIED: LoggedModelStatus 151 LOGGED_MODEL_PENDING: LoggedModelStatus 152 LOGGED_MODEL_READY: LoggedModelStatus 153 LOGGED_MODEL_UPLOAD_FAILED: LoggedModelStatus 154 ROUTING_STRATEGY_UNSPECIFIED: RoutingStrategy 155 REQUEST_BASED_TRAFFIC_SPLIT: RoutingStrategy 156 FALLBACK_STRATEGY_UNSPECIFIED: FallbackStrategy 157 SEQUENTIAL: FallbackStrategy 158 LINKAGE_TYPE_UNSPECIFIED: GatewayModelLinkageType 159 PRIMARY: GatewayModelLinkageType 160 FALLBACK: GatewayModelLinkageType 161 DURATION_UNIT_UNSPECIFIED: BudgetDurationUnit 162 MINUTES: BudgetDurationUnit 163 HOURS: BudgetDurationUnit 164 DAYS: BudgetDurationUnit 165 WEEKS: BudgetDurationUnit 166 MONTHS: BudgetDurationUnit 167 TARGET_SCOPE_UNSPECIFIED: BudgetTargetScope 168 GLOBAL: BudgetTargetScope 169 WORKSPACE: BudgetTargetScope 170 BUDGET_ACTION_UNSPECIFIED: BudgetAction 171 ALERT: BudgetAction 172 REJECT: BudgetAction 173 BUDGET_UNIT_UNSPECIFIED: BudgetUnit 174 USD: BudgetUnit 175 GUARDRAIL_STAGE_UNSPECIFIED: GuardrailStage 176 BEFORE: GuardrailStage 177 AFTER: GuardrailStage 178 GUARDRAIL_ACTION_UNSPECIFIED: GuardrailAction 179 VALIDATION: GuardrailAction 180 SANITIZATION: GuardrailAction 181 182 class Metric(_message.Message): 183 __slots__ = ("key", "value", "timestamp", "step", "dataset_name", "dataset_digest", "model_id", "run_id") 184 KEY_FIELD_NUMBER: _ClassVar[int] 185 VALUE_FIELD_NUMBER: _ClassVar[int] 186 TIMESTAMP_FIELD_NUMBER: _ClassVar[int] 187 STEP_FIELD_NUMBER: _ClassVar[int] 188 DATASET_NAME_FIELD_NUMBER: _ClassVar[int] 189 DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] 190 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 191 RUN_ID_FIELD_NUMBER: _ClassVar[int] 192 key: str 193 value: float 194 timestamp: int 195 step: int 196 dataset_name: str 197 dataset_digest: str 198 model_id: str 199 run_id: str 200 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: ... 201 202 class Param(_message.Message): 203 __slots__ = ("key", "value") 204 KEY_FIELD_NUMBER: _ClassVar[int] 205 VALUE_FIELD_NUMBER: _ClassVar[int] 206 key: str 207 value: str 208 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 209 210 class Run(_message.Message): 211 __slots__ = ("info", "data", "inputs", "outputs") 212 INFO_FIELD_NUMBER: _ClassVar[int] 213 DATA_FIELD_NUMBER: _ClassVar[int] 214 INPUTS_FIELD_NUMBER: _ClassVar[int] 215 OUTPUTS_FIELD_NUMBER: _ClassVar[int] 216 info: RunInfo 217 data: RunData 218 inputs: RunInputs 219 outputs: RunOutputs 220 def __init__(self, info: _Optional[_Union[RunInfo, _Mapping]] = ..., data: _Optional[_Union[RunData, _Mapping]] = ..., inputs: _Optional[_Union[RunInputs, _Mapping]] = ..., outputs: _Optional[_Union[RunOutputs, _Mapping]] = ...) -> None: ... 221 222 class RunData(_message.Message): 223 __slots__ = ("metrics", "params", "tags") 224 METRICS_FIELD_NUMBER: _ClassVar[int] 225 PARAMS_FIELD_NUMBER: _ClassVar[int] 226 TAGS_FIELD_NUMBER: _ClassVar[int] 227 metrics: _containers.RepeatedCompositeFieldContainer[Metric] 228 params: _containers.RepeatedCompositeFieldContainer[Param] 229 tags: _containers.RepeatedCompositeFieldContainer[RunTag] 230 def __init__(self, metrics: _Optional[_Iterable[_Union[Metric, _Mapping]]] = ..., params: _Optional[_Iterable[_Union[Param, _Mapping]]] = ..., tags: _Optional[_Iterable[_Union[RunTag, _Mapping]]] = ...) -> None: ... 231 232 class RunInputs(_message.Message): 233 __slots__ = ("dataset_inputs", "model_inputs") 234 DATASET_INPUTS_FIELD_NUMBER: _ClassVar[int] 235 MODEL_INPUTS_FIELD_NUMBER: _ClassVar[int] 236 dataset_inputs: _containers.RepeatedCompositeFieldContainer[DatasetInput] 237 model_inputs: _containers.RepeatedCompositeFieldContainer[ModelInput] 238 def __init__(self, dataset_inputs: _Optional[_Iterable[_Union[DatasetInput, _Mapping]]] = ..., model_inputs: _Optional[_Iterable[_Union[ModelInput, _Mapping]]] = ...) -> None: ... 239 240 class RunOutputs(_message.Message): 241 __slots__ = ("model_outputs",) 242 MODEL_OUTPUTS_FIELD_NUMBER: _ClassVar[int] 243 model_outputs: _containers.RepeatedCompositeFieldContainer[ModelOutput] 244 def __init__(self, model_outputs: _Optional[_Iterable[_Union[ModelOutput, _Mapping]]] = ...) -> None: ... 245 246 class RunTag(_message.Message): 247 __slots__ = ("key", "value") 248 KEY_FIELD_NUMBER: _ClassVar[int] 249 VALUE_FIELD_NUMBER: _ClassVar[int] 250 key: str 251 value: str 252 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 253 254 class ExperimentTag(_message.Message): 255 __slots__ = ("key", "value") 256 KEY_FIELD_NUMBER: _ClassVar[int] 257 VALUE_FIELD_NUMBER: _ClassVar[int] 258 key: str 259 value: str 260 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 261 262 class RunInfo(_message.Message): 263 __slots__ = ("run_id", "run_uuid", "run_name", "experiment_id", "user_id", "status", "start_time", "end_time", "artifact_uri", "lifecycle_stage") 264 RUN_ID_FIELD_NUMBER: _ClassVar[int] 265 RUN_UUID_FIELD_NUMBER: _ClassVar[int] 266 RUN_NAME_FIELD_NUMBER: _ClassVar[int] 267 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 268 USER_ID_FIELD_NUMBER: _ClassVar[int] 269 STATUS_FIELD_NUMBER: _ClassVar[int] 270 START_TIME_FIELD_NUMBER: _ClassVar[int] 271 END_TIME_FIELD_NUMBER: _ClassVar[int] 272 ARTIFACT_URI_FIELD_NUMBER: _ClassVar[int] 273 LIFECYCLE_STAGE_FIELD_NUMBER: _ClassVar[int] 274 run_id: str 275 run_uuid: str 276 run_name: str 277 experiment_id: str 278 user_id: str 279 status: RunStatus 280 start_time: int 281 end_time: int 282 artifact_uri: str 283 lifecycle_stage: str 284 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: ... 285 286 class Experiment(_message.Message): 287 __slots__ = ("experiment_id", "name", "artifact_location", "lifecycle_stage", "last_update_time", "creation_time", "tags") 288 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 289 NAME_FIELD_NUMBER: _ClassVar[int] 290 ARTIFACT_LOCATION_FIELD_NUMBER: _ClassVar[int] 291 LIFECYCLE_STAGE_FIELD_NUMBER: _ClassVar[int] 292 LAST_UPDATE_TIME_FIELD_NUMBER: _ClassVar[int] 293 CREATION_TIME_FIELD_NUMBER: _ClassVar[int] 294 TAGS_FIELD_NUMBER: _ClassVar[int] 295 experiment_id: str 296 name: str 297 artifact_location: str 298 lifecycle_stage: str 299 last_update_time: int 300 creation_time: int 301 tags: _containers.RepeatedCompositeFieldContainer[ExperimentTag] 302 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]]] = ...) -> None: ... 303 304 class DatasetInput(_message.Message): 305 __slots__ = ("tags", "dataset") 306 TAGS_FIELD_NUMBER: _ClassVar[int] 307 DATASET_FIELD_NUMBER: _ClassVar[int] 308 tags: _containers.RepeatedCompositeFieldContainer[InputTag] 309 dataset: Dataset 310 def __init__(self, tags: _Optional[_Iterable[_Union[InputTag, _Mapping]]] = ..., dataset: _Optional[_Union[Dataset, _Mapping]] = ...) -> None: ... 311 312 class ModelInput(_message.Message): 313 __slots__ = ("model_id",) 314 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 315 model_id: str 316 def __init__(self, model_id: _Optional[str] = ...) -> None: ... 317 318 class InputTag(_message.Message): 319 __slots__ = ("key", "value") 320 KEY_FIELD_NUMBER: _ClassVar[int] 321 VALUE_FIELD_NUMBER: _ClassVar[int] 322 key: str 323 value: str 324 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 325 326 class Dataset(_message.Message): 327 __slots__ = ("name", "digest", "source_type", "source", "schema", "profile") 328 NAME_FIELD_NUMBER: _ClassVar[int] 329 DIGEST_FIELD_NUMBER: _ClassVar[int] 330 SOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] 331 SOURCE_FIELD_NUMBER: _ClassVar[int] 332 SCHEMA_FIELD_NUMBER: _ClassVar[int] 333 PROFILE_FIELD_NUMBER: _ClassVar[int] 334 name: str 335 digest: str 336 source_type: str 337 source: str 338 schema: str 339 profile: str 340 def __init__(self, name: _Optional[str] = ..., digest: _Optional[str] = ..., source_type: _Optional[str] = ..., source: _Optional[str] = ..., schema: _Optional[str] = ..., profile: _Optional[str] = ...) -> None: ... 341 342 class ModelOutput(_message.Message): 343 __slots__ = ("model_id", "step") 344 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 345 STEP_FIELD_NUMBER: _ClassVar[int] 346 model_id: str 347 step: int 348 def __init__(self, model_id: _Optional[str] = ..., step: _Optional[int] = ...) -> None: ... 349 350 class CreateExperiment(_message.Message): 351 __slots__ = ("name", "artifact_location", "tags") 352 class Response(_message.Message): 353 __slots__ = ("experiment_id",) 354 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 355 experiment_id: str 356 def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... 357 NAME_FIELD_NUMBER: _ClassVar[int] 358 ARTIFACT_LOCATION_FIELD_NUMBER: _ClassVar[int] 359 TAGS_FIELD_NUMBER: _ClassVar[int] 360 name: str 361 artifact_location: str 362 tags: _containers.RepeatedCompositeFieldContainer[ExperimentTag] 363 def __init__(self, name: _Optional[str] = ..., artifact_location: _Optional[str] = ..., tags: _Optional[_Iterable[_Union[ExperimentTag, _Mapping]]] = ...) -> None: ... 364 365 class SearchExperiments(_message.Message): 366 __slots__ = ("max_results", "page_token", "filter", "order_by", "view_type") 367 class Response(_message.Message): 368 __slots__ = ("experiments", "next_page_token") 369 EXPERIMENTS_FIELD_NUMBER: _ClassVar[int] 370 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 371 experiments: _containers.RepeatedCompositeFieldContainer[Experiment] 372 next_page_token: str 373 def __init__(self, experiments: _Optional[_Iterable[_Union[Experiment, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 374 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 375 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 376 FILTER_FIELD_NUMBER: _ClassVar[int] 377 ORDER_BY_FIELD_NUMBER: _ClassVar[int] 378 VIEW_TYPE_FIELD_NUMBER: _ClassVar[int] 379 max_results: int 380 page_token: str 381 filter: str 382 order_by: _containers.RepeatedScalarFieldContainer[str] 383 view_type: ViewType 384 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: ... 385 386 class GetExperiment(_message.Message): 387 __slots__ = ("experiment_id",) 388 class Response(_message.Message): 389 __slots__ = ("experiment",) 390 EXPERIMENT_FIELD_NUMBER: _ClassVar[int] 391 experiment: Experiment 392 def __init__(self, experiment: _Optional[_Union[Experiment, _Mapping]] = ...) -> None: ... 393 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 394 experiment_id: str 395 def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... 396 397 class DeleteExperiment(_message.Message): 398 __slots__ = ("experiment_id",) 399 class Response(_message.Message): 400 __slots__ = () 401 def __init__(self) -> None: ... 402 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 403 experiment_id: str 404 def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... 405 406 class RestoreExperiment(_message.Message): 407 __slots__ = ("experiment_id",) 408 class Response(_message.Message): 409 __slots__ = () 410 def __init__(self) -> None: ... 411 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 412 experiment_id: str 413 def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... 414 415 class UpdateExperiment(_message.Message): 416 __slots__ = ("experiment_id", "new_name") 417 class Response(_message.Message): 418 __slots__ = () 419 def __init__(self) -> None: ... 420 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 421 NEW_NAME_FIELD_NUMBER: _ClassVar[int] 422 experiment_id: str 423 new_name: str 424 def __init__(self, experiment_id: _Optional[str] = ..., new_name: _Optional[str] = ...) -> None: ... 425 426 class CreateRun(_message.Message): 427 __slots__ = ("experiment_id", "user_id", "run_name", "start_time", "tags") 428 class Response(_message.Message): 429 __slots__ = ("run",) 430 RUN_FIELD_NUMBER: _ClassVar[int] 431 run: Run 432 def __init__(self, run: _Optional[_Union[Run, _Mapping]] = ...) -> None: ... 433 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 434 USER_ID_FIELD_NUMBER: _ClassVar[int] 435 RUN_NAME_FIELD_NUMBER: _ClassVar[int] 436 START_TIME_FIELD_NUMBER: _ClassVar[int] 437 TAGS_FIELD_NUMBER: _ClassVar[int] 438 experiment_id: str 439 user_id: str 440 run_name: str 441 start_time: int 442 tags: _containers.RepeatedCompositeFieldContainer[RunTag] 443 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: ... 444 445 class UpdateRun(_message.Message): 446 __slots__ = ("run_id", "run_uuid", "status", "end_time", "run_name") 447 class Response(_message.Message): 448 __slots__ = ("run_info",) 449 RUN_INFO_FIELD_NUMBER: _ClassVar[int] 450 run_info: RunInfo 451 def __init__(self, run_info: _Optional[_Union[RunInfo, _Mapping]] = ...) -> None: ... 452 RUN_ID_FIELD_NUMBER: _ClassVar[int] 453 RUN_UUID_FIELD_NUMBER: _ClassVar[int] 454 STATUS_FIELD_NUMBER: _ClassVar[int] 455 END_TIME_FIELD_NUMBER: _ClassVar[int] 456 RUN_NAME_FIELD_NUMBER: _ClassVar[int] 457 run_id: str 458 run_uuid: str 459 status: RunStatus 460 end_time: int 461 run_name: str 462 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: ... 463 464 class DeleteRun(_message.Message): 465 __slots__ = ("run_id",) 466 class Response(_message.Message): 467 __slots__ = () 468 def __init__(self) -> None: ... 469 RUN_ID_FIELD_NUMBER: _ClassVar[int] 470 run_id: str 471 def __init__(self, run_id: _Optional[str] = ...) -> None: ... 472 473 class RestoreRun(_message.Message): 474 __slots__ = ("run_id",) 475 class Response(_message.Message): 476 __slots__ = () 477 def __init__(self) -> None: ... 478 RUN_ID_FIELD_NUMBER: _ClassVar[int] 479 run_id: str 480 def __init__(self, run_id: _Optional[str] = ...) -> None: ... 481 482 class LogMetric(_message.Message): 483 __slots__ = ("run_id", "run_uuid", "key", "value", "timestamp", "step", "model_id", "dataset_name", "dataset_digest") 484 class Response(_message.Message): 485 __slots__ = () 486 def __init__(self) -> None: ... 487 RUN_ID_FIELD_NUMBER: _ClassVar[int] 488 RUN_UUID_FIELD_NUMBER: _ClassVar[int] 489 KEY_FIELD_NUMBER: _ClassVar[int] 490 VALUE_FIELD_NUMBER: _ClassVar[int] 491 TIMESTAMP_FIELD_NUMBER: _ClassVar[int] 492 STEP_FIELD_NUMBER: _ClassVar[int] 493 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 494 DATASET_NAME_FIELD_NUMBER: _ClassVar[int] 495 DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] 496 run_id: str 497 run_uuid: str 498 key: str 499 value: float 500 timestamp: int 501 step: int 502 model_id: str 503 dataset_name: str 504 dataset_digest: str 505 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: ... 506 507 class LogParam(_message.Message): 508 __slots__ = ("run_id", "run_uuid", "key", "value") 509 class Response(_message.Message): 510 __slots__ = () 511 def __init__(self) -> None: ... 512 RUN_ID_FIELD_NUMBER: _ClassVar[int] 513 RUN_UUID_FIELD_NUMBER: _ClassVar[int] 514 KEY_FIELD_NUMBER: _ClassVar[int] 515 VALUE_FIELD_NUMBER: _ClassVar[int] 516 run_id: str 517 run_uuid: str 518 key: str 519 value: str 520 def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 521 522 class SetExperimentTag(_message.Message): 523 __slots__ = ("experiment_id", "key", "value") 524 class Response(_message.Message): 525 __slots__ = () 526 def __init__(self) -> None: ... 527 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 528 KEY_FIELD_NUMBER: _ClassVar[int] 529 VALUE_FIELD_NUMBER: _ClassVar[int] 530 experiment_id: str 531 key: str 532 value: str 533 def __init__(self, experiment_id: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 534 535 class DeleteExperimentTag(_message.Message): 536 __slots__ = ("experiment_id", "key") 537 class Response(_message.Message): 538 __slots__ = () 539 def __init__(self) -> None: ... 540 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 541 KEY_FIELD_NUMBER: _ClassVar[int] 542 experiment_id: str 543 key: str 544 def __init__(self, experiment_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... 545 546 class SetTag(_message.Message): 547 __slots__ = ("run_id", "run_uuid", "key", "value") 548 class Response(_message.Message): 549 __slots__ = () 550 def __init__(self) -> None: ... 551 RUN_ID_FIELD_NUMBER: _ClassVar[int] 552 RUN_UUID_FIELD_NUMBER: _ClassVar[int] 553 KEY_FIELD_NUMBER: _ClassVar[int] 554 VALUE_FIELD_NUMBER: _ClassVar[int] 555 run_id: str 556 run_uuid: str 557 key: str 558 value: str 559 def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 560 561 class DeleteTag(_message.Message): 562 __slots__ = ("run_id", "key") 563 class Response(_message.Message): 564 __slots__ = () 565 def __init__(self) -> None: ... 566 RUN_ID_FIELD_NUMBER: _ClassVar[int] 567 KEY_FIELD_NUMBER: _ClassVar[int] 568 run_id: str 569 key: str 570 def __init__(self, run_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... 571 572 class GetRun(_message.Message): 573 __slots__ = ("run_id", "run_uuid") 574 class Response(_message.Message): 575 __slots__ = ("run",) 576 RUN_FIELD_NUMBER: _ClassVar[int] 577 run: Run 578 def __init__(self, run: _Optional[_Union[Run, _Mapping]] = ...) -> None: ... 579 RUN_ID_FIELD_NUMBER: _ClassVar[int] 580 RUN_UUID_FIELD_NUMBER: _ClassVar[int] 581 run_id: str 582 run_uuid: str 583 def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ...) -> None: ... 584 585 class SearchRuns(_message.Message): 586 __slots__ = ("experiment_ids", "filter", "run_view_type", "max_results", "order_by", "page_token") 587 class Response(_message.Message): 588 __slots__ = ("runs", "next_page_token") 589 RUNS_FIELD_NUMBER: _ClassVar[int] 590 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 591 runs: _containers.RepeatedCompositeFieldContainer[Run] 592 next_page_token: str 593 def __init__(self, runs: _Optional[_Iterable[_Union[Run, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 594 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 595 FILTER_FIELD_NUMBER: _ClassVar[int] 596 RUN_VIEW_TYPE_FIELD_NUMBER: _ClassVar[int] 597 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 598 ORDER_BY_FIELD_NUMBER: _ClassVar[int] 599 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 600 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 601 filter: str 602 run_view_type: ViewType 603 max_results: int 604 order_by: _containers.RepeatedScalarFieldContainer[str] 605 page_token: str 606 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: ... 607 608 class ListArtifacts(_message.Message): 609 __slots__ = ("run_id", "run_uuid", "path", "page_token") 610 class Response(_message.Message): 611 __slots__ = ("root_uri", "files", "next_page_token") 612 ROOT_URI_FIELD_NUMBER: _ClassVar[int] 613 FILES_FIELD_NUMBER: _ClassVar[int] 614 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 615 root_uri: str 616 files: _containers.RepeatedCompositeFieldContainer[FileInfo] 617 next_page_token: str 618 def __init__(self, root_uri: _Optional[str] = ..., files: _Optional[_Iterable[_Union[FileInfo, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 619 RUN_ID_FIELD_NUMBER: _ClassVar[int] 620 RUN_UUID_FIELD_NUMBER: _ClassVar[int] 621 PATH_FIELD_NUMBER: _ClassVar[int] 622 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 623 run_id: str 624 run_uuid: str 625 path: str 626 page_token: str 627 def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., path: _Optional[str] = ..., page_token: _Optional[str] = ...) -> None: ... 628 629 class CreatePresignedUploadUrl(_message.Message): 630 __slots__ = ("run_id", "path", "expiration") 631 class Response(_message.Message): 632 __slots__ = ("presigned_url", "headers") 633 class HeadersEntry(_message.Message): 634 __slots__ = ("key", "value") 635 KEY_FIELD_NUMBER: _ClassVar[int] 636 VALUE_FIELD_NUMBER: _ClassVar[int] 637 key: str 638 value: str 639 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 640 PRESIGNED_URL_FIELD_NUMBER: _ClassVar[int] 641 HEADERS_FIELD_NUMBER: _ClassVar[int] 642 presigned_url: str 643 headers: _containers.ScalarMap[str, str] 644 def __init__(self, presigned_url: _Optional[str] = ..., headers: _Optional[_Mapping[str, str]] = ...) -> None: ... 645 RUN_ID_FIELD_NUMBER: _ClassVar[int] 646 PATH_FIELD_NUMBER: _ClassVar[int] 647 EXPIRATION_FIELD_NUMBER: _ClassVar[int] 648 run_id: str 649 path: str 650 expiration: int 651 def __init__(self, run_id: _Optional[str] = ..., path: _Optional[str] = ..., expiration: _Optional[int] = ...) -> None: ... 652 653 class FileInfo(_message.Message): 654 __slots__ = ("path", "is_dir", "file_size") 655 PATH_FIELD_NUMBER: _ClassVar[int] 656 IS_DIR_FIELD_NUMBER: _ClassVar[int] 657 FILE_SIZE_FIELD_NUMBER: _ClassVar[int] 658 path: str 659 is_dir: bool 660 file_size: int 661 def __init__(self, path: _Optional[str] = ..., is_dir: bool = ..., file_size: _Optional[int] = ...) -> None: ... 662 663 class GetMetricHistory(_message.Message): 664 __slots__ = ("run_id", "run_uuid", "metric_key", "page_token", "max_results") 665 class Response(_message.Message): 666 __slots__ = ("metrics", "next_page_token") 667 METRICS_FIELD_NUMBER: _ClassVar[int] 668 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 669 metrics: _containers.RepeatedCompositeFieldContainer[Metric] 670 next_page_token: str 671 def __init__(self, metrics: _Optional[_Iterable[_Union[Metric, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 672 RUN_ID_FIELD_NUMBER: _ClassVar[int] 673 RUN_UUID_FIELD_NUMBER: _ClassVar[int] 674 METRIC_KEY_FIELD_NUMBER: _ClassVar[int] 675 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 676 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 677 run_id: str 678 run_uuid: str 679 metric_key: str 680 page_token: str 681 max_results: int 682 def __init__(self, run_id: _Optional[str] = ..., run_uuid: _Optional[str] = ..., metric_key: _Optional[str] = ..., page_token: _Optional[str] = ..., max_results: _Optional[int] = ...) -> None: ... 683 684 class MetricWithRunId(_message.Message): 685 __slots__ = ("key", "value", "timestamp", "step", "run_id") 686 KEY_FIELD_NUMBER: _ClassVar[int] 687 VALUE_FIELD_NUMBER: _ClassVar[int] 688 TIMESTAMP_FIELD_NUMBER: _ClassVar[int] 689 STEP_FIELD_NUMBER: _ClassVar[int] 690 RUN_ID_FIELD_NUMBER: _ClassVar[int] 691 key: str 692 value: float 693 timestamp: int 694 step: int 695 run_id: str 696 def __init__(self, key: _Optional[str] = ..., value: _Optional[float] = ..., timestamp: _Optional[int] = ..., step: _Optional[int] = ..., run_id: _Optional[str] = ...) -> None: ... 697 698 class GetMetricHistoryBulkInterval(_message.Message): 699 __slots__ = ("run_ids", "metric_key", "start_step", "end_step", "max_results") 700 class Response(_message.Message): 701 __slots__ = ("metrics",) 702 METRICS_FIELD_NUMBER: _ClassVar[int] 703 metrics: _containers.RepeatedCompositeFieldContainer[MetricWithRunId] 704 def __init__(self, metrics: _Optional[_Iterable[_Union[MetricWithRunId, _Mapping]]] = ...) -> None: ... 705 RUN_IDS_FIELD_NUMBER: _ClassVar[int] 706 METRIC_KEY_FIELD_NUMBER: _ClassVar[int] 707 START_STEP_FIELD_NUMBER: _ClassVar[int] 708 END_STEP_FIELD_NUMBER: _ClassVar[int] 709 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 710 run_ids: _containers.RepeatedScalarFieldContainer[str] 711 metric_key: str 712 start_step: int 713 end_step: int 714 max_results: int 715 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: ... 716 717 class LogBatch(_message.Message): 718 __slots__ = ("run_id", "metrics", "params", "tags") 719 class Response(_message.Message): 720 __slots__ = () 721 def __init__(self) -> None: ... 722 RUN_ID_FIELD_NUMBER: _ClassVar[int] 723 METRICS_FIELD_NUMBER: _ClassVar[int] 724 PARAMS_FIELD_NUMBER: _ClassVar[int] 725 TAGS_FIELD_NUMBER: _ClassVar[int] 726 run_id: str 727 metrics: _containers.RepeatedCompositeFieldContainer[Metric] 728 params: _containers.RepeatedCompositeFieldContainer[Param] 729 tags: _containers.RepeatedCompositeFieldContainer[RunTag] 730 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: ... 731 732 class LogModel(_message.Message): 733 __slots__ = ("run_id", "model_json") 734 class Response(_message.Message): 735 __slots__ = () 736 def __init__(self) -> None: ... 737 RUN_ID_FIELD_NUMBER: _ClassVar[int] 738 MODEL_JSON_FIELD_NUMBER: _ClassVar[int] 739 run_id: str 740 model_json: str 741 def __init__(self, run_id: _Optional[str] = ..., model_json: _Optional[str] = ...) -> None: ... 742 743 class LogInputs(_message.Message): 744 __slots__ = ("run_id", "datasets", "models") 745 class Response(_message.Message): 746 __slots__ = () 747 def __init__(self) -> None: ... 748 RUN_ID_FIELD_NUMBER: _ClassVar[int] 749 DATASETS_FIELD_NUMBER: _ClassVar[int] 750 MODELS_FIELD_NUMBER: _ClassVar[int] 751 run_id: str 752 datasets: _containers.RepeatedCompositeFieldContainer[DatasetInput] 753 models: _containers.RepeatedCompositeFieldContainer[ModelInput] 754 def __init__(self, run_id: _Optional[str] = ..., datasets: _Optional[_Iterable[_Union[DatasetInput, _Mapping]]] = ..., models: _Optional[_Iterable[_Union[ModelInput, _Mapping]]] = ...) -> None: ... 755 756 class LogOutputs(_message.Message): 757 __slots__ = ("run_id", "models") 758 class Response(_message.Message): 759 __slots__ = () 760 def __init__(self) -> None: ... 761 RUN_ID_FIELD_NUMBER: _ClassVar[int] 762 MODELS_FIELD_NUMBER: _ClassVar[int] 763 run_id: str 764 models: _containers.RepeatedCompositeFieldContainer[ModelOutput] 765 def __init__(self, run_id: _Optional[str] = ..., models: _Optional[_Iterable[_Union[ModelOutput, _Mapping]]] = ...) -> None: ... 766 767 class GetExperimentByName(_message.Message): 768 __slots__ = ("experiment_name",) 769 class Response(_message.Message): 770 __slots__ = ("experiment",) 771 EXPERIMENT_FIELD_NUMBER: _ClassVar[int] 772 experiment: Experiment 773 def __init__(self, experiment: _Optional[_Union[Experiment, _Mapping]] = ...) -> None: ... 774 EXPERIMENT_NAME_FIELD_NUMBER: _ClassVar[int] 775 experiment_name: str 776 def __init__(self, experiment_name: _Optional[str] = ...) -> None: ... 777 778 class CreateAssessment(_message.Message): 779 __slots__ = ("assessment",) 780 class Response(_message.Message): 781 __slots__ = ("assessment",) 782 ASSESSMENT_FIELD_NUMBER: _ClassVar[int] 783 assessment: _assessments_pb2.Assessment 784 def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ...) -> None: ... 785 ASSESSMENT_FIELD_NUMBER: _ClassVar[int] 786 assessment: _assessments_pb2.Assessment 787 def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ...) -> None: ... 788 789 class UpdateAssessment(_message.Message): 790 __slots__ = ("assessment", "update_mask") 791 class Response(_message.Message): 792 __slots__ = ("assessment",) 793 ASSESSMENT_FIELD_NUMBER: _ClassVar[int] 794 assessment: _assessments_pb2.Assessment 795 def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ...) -> None: ... 796 ASSESSMENT_FIELD_NUMBER: _ClassVar[int] 797 UPDATE_MASK_FIELD_NUMBER: _ClassVar[int] 798 assessment: _assessments_pb2.Assessment 799 update_mask: _field_mask_pb2.FieldMask 800 def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ..., update_mask: _Optional[_Union[_field_mask_pb2.FieldMask, _Mapping]] = ...) -> None: ... 801 802 class DeleteAssessment(_message.Message): 803 __slots__ = ("trace_id", "assessment_id") 804 class Response(_message.Message): 805 __slots__ = () 806 def __init__(self) -> None: ... 807 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 808 ASSESSMENT_ID_FIELD_NUMBER: _ClassVar[int] 809 trace_id: str 810 assessment_id: str 811 def __init__(self, trace_id: _Optional[str] = ..., assessment_id: _Optional[str] = ...) -> None: ... 812 813 class GetAssessmentRequest(_message.Message): 814 __slots__ = ("trace_id", "assessment_id") 815 class Response(_message.Message): 816 __slots__ = ("assessment",) 817 ASSESSMENT_FIELD_NUMBER: _ClassVar[int] 818 assessment: _assessments_pb2.Assessment 819 def __init__(self, assessment: _Optional[_Union[_assessments_pb2.Assessment, _Mapping]] = ...) -> None: ... 820 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 821 ASSESSMENT_ID_FIELD_NUMBER: _ClassVar[int] 822 trace_id: str 823 assessment_id: str 824 def __init__(self, trace_id: _Optional[str] = ..., assessment_id: _Optional[str] = ...) -> None: ... 825 826 class TraceInfo(_message.Message): 827 __slots__ = ("request_id", "experiment_id", "timestamp_ms", "execution_time_ms", "status", "request_metadata", "tags") 828 REQUEST_ID_FIELD_NUMBER: _ClassVar[int] 829 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 830 TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] 831 EXECUTION_TIME_MS_FIELD_NUMBER: _ClassVar[int] 832 STATUS_FIELD_NUMBER: _ClassVar[int] 833 REQUEST_METADATA_FIELD_NUMBER: _ClassVar[int] 834 TAGS_FIELD_NUMBER: _ClassVar[int] 835 request_id: str 836 experiment_id: str 837 timestamp_ms: int 838 execution_time_ms: int 839 status: TraceStatus 840 request_metadata: _containers.RepeatedCompositeFieldContainer[TraceRequestMetadata] 841 tags: _containers.RepeatedCompositeFieldContainer[TraceTag] 842 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: ... 843 844 class TraceRequestMetadata(_message.Message): 845 __slots__ = ("key", "value") 846 KEY_FIELD_NUMBER: _ClassVar[int] 847 VALUE_FIELD_NUMBER: _ClassVar[int] 848 key: str 849 value: str 850 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 851 852 class TraceTag(_message.Message): 853 __slots__ = ("key", "value") 854 KEY_FIELD_NUMBER: _ClassVar[int] 855 VALUE_FIELD_NUMBER: _ClassVar[int] 856 key: str 857 value: str 858 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 859 860 class StartTrace(_message.Message): 861 __slots__ = ("experiment_id", "timestamp_ms", "request_metadata", "tags") 862 class Response(_message.Message): 863 __slots__ = ("trace_info",) 864 TRACE_INFO_FIELD_NUMBER: _ClassVar[int] 865 trace_info: TraceInfo 866 def __init__(self, trace_info: _Optional[_Union[TraceInfo, _Mapping]] = ...) -> None: ... 867 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 868 TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] 869 REQUEST_METADATA_FIELD_NUMBER: _ClassVar[int] 870 TAGS_FIELD_NUMBER: _ClassVar[int] 871 experiment_id: str 872 timestamp_ms: int 873 request_metadata: _containers.RepeatedCompositeFieldContainer[TraceRequestMetadata] 874 tags: _containers.RepeatedCompositeFieldContainer[TraceTag] 875 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: ... 876 877 class EndTrace(_message.Message): 878 __slots__ = ("request_id", "timestamp_ms", "status", "request_metadata", "tags") 879 class Response(_message.Message): 880 __slots__ = ("trace_info",) 881 TRACE_INFO_FIELD_NUMBER: _ClassVar[int] 882 trace_info: TraceInfo 883 def __init__(self, trace_info: _Optional[_Union[TraceInfo, _Mapping]] = ...) -> None: ... 884 REQUEST_ID_FIELD_NUMBER: _ClassVar[int] 885 TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] 886 STATUS_FIELD_NUMBER: _ClassVar[int] 887 REQUEST_METADATA_FIELD_NUMBER: _ClassVar[int] 888 TAGS_FIELD_NUMBER: _ClassVar[int] 889 request_id: str 890 timestamp_ms: int 891 status: TraceStatus 892 request_metadata: _containers.RepeatedCompositeFieldContainer[TraceRequestMetadata] 893 tags: _containers.RepeatedCompositeFieldContainer[TraceTag] 894 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: ... 895 896 class GetTraceInfo(_message.Message): 897 __slots__ = ("request_id",) 898 class Response(_message.Message): 899 __slots__ = ("trace_info",) 900 TRACE_INFO_FIELD_NUMBER: _ClassVar[int] 901 trace_info: TraceInfo 902 def __init__(self, trace_info: _Optional[_Union[TraceInfo, _Mapping]] = ...) -> None: ... 903 REQUEST_ID_FIELD_NUMBER: _ClassVar[int] 904 request_id: str 905 def __init__(self, request_id: _Optional[str] = ...) -> None: ... 906 907 class GetTraceInfoV3(_message.Message): 908 __slots__ = ("trace_id",) 909 class Response(_message.Message): 910 __slots__ = ("trace",) 911 TRACE_FIELD_NUMBER: _ClassVar[int] 912 trace: Trace 913 def __init__(self, trace: _Optional[_Union[Trace, _Mapping]] = ...) -> None: ... 914 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 915 trace_id: str 916 def __init__(self, trace_id: _Optional[str] = ...) -> None: ... 917 918 class BatchGetTraces(_message.Message): 919 __slots__ = ("trace_ids",) 920 class Response(_message.Message): 921 __slots__ = ("traces",) 922 TRACES_FIELD_NUMBER: _ClassVar[int] 923 traces: _containers.RepeatedCompositeFieldContainer[Trace] 924 def __init__(self, traces: _Optional[_Iterable[_Union[Trace, _Mapping]]] = ...) -> None: ... 925 TRACE_IDS_FIELD_NUMBER: _ClassVar[int] 926 trace_ids: _containers.RepeatedScalarFieldContainer[str] 927 def __init__(self, trace_ids: _Optional[_Iterable[str]] = ...) -> None: ... 928 929 class BatchGetTraceInfos(_message.Message): 930 __slots__ = ("trace_ids",) 931 class Response(_message.Message): 932 __slots__ = ("trace_infos",) 933 TRACE_INFOS_FIELD_NUMBER: _ClassVar[int] 934 trace_infos: _containers.RepeatedCompositeFieldContainer[TraceInfoV3] 935 def __init__(self, trace_infos: _Optional[_Iterable[_Union[TraceInfoV3, _Mapping]]] = ...) -> None: ... 936 TRACE_IDS_FIELD_NUMBER: _ClassVar[int] 937 trace_ids: _containers.RepeatedScalarFieldContainer[str] 938 def __init__(self, trace_ids: _Optional[_Iterable[str]] = ...) -> None: ... 939 940 class GetTrace(_message.Message): 941 __slots__ = ("trace_id", "allow_partial") 942 class Response(_message.Message): 943 __slots__ = ("trace",) 944 TRACE_FIELD_NUMBER: _ClassVar[int] 945 trace: Trace 946 def __init__(self, trace: _Optional[_Union[Trace, _Mapping]] = ...) -> None: ... 947 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 948 ALLOW_PARTIAL_FIELD_NUMBER: _ClassVar[int] 949 trace_id: str 950 allow_partial: bool 951 def __init__(self, trace_id: _Optional[str] = ..., allow_partial: bool = ...) -> None: ... 952 953 class SearchTraces(_message.Message): 954 __slots__ = ("experiment_ids", "filter", "max_results", "order_by", "page_token") 955 class Response(_message.Message): 956 __slots__ = ("traces", "next_page_token") 957 TRACES_FIELD_NUMBER: _ClassVar[int] 958 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 959 traces: _containers.RepeatedCompositeFieldContainer[TraceInfo] 960 next_page_token: str 961 def __init__(self, traces: _Optional[_Iterable[_Union[TraceInfo, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 962 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 963 FILTER_FIELD_NUMBER: _ClassVar[int] 964 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 965 ORDER_BY_FIELD_NUMBER: _ClassVar[int] 966 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 967 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 968 filter: str 969 max_results: int 970 order_by: _containers.RepeatedScalarFieldContainer[str] 971 page_token: str 972 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: ... 973 974 class SearchUnifiedTraces(_message.Message): 975 __slots__ = ("model_id", "sql_warehouse_id", "experiment_ids", "filter", "max_results", "order_by", "page_token") 976 class Response(_message.Message): 977 __slots__ = ("traces", "next_page_token") 978 TRACES_FIELD_NUMBER: _ClassVar[int] 979 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 980 traces: _containers.RepeatedCompositeFieldContainer[TraceInfo] 981 next_page_token: str 982 def __init__(self, traces: _Optional[_Iterable[_Union[TraceInfo, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 983 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 984 SQL_WAREHOUSE_ID_FIELD_NUMBER: _ClassVar[int] 985 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 986 FILTER_FIELD_NUMBER: _ClassVar[int] 987 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 988 ORDER_BY_FIELD_NUMBER: _ClassVar[int] 989 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 990 model_id: str 991 sql_warehouse_id: str 992 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 993 filter: str 994 max_results: int 995 order_by: _containers.RepeatedScalarFieldContainer[str] 996 page_token: str 997 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: ... 998 999 class GetOnlineTraceDetails(_message.Message): 1000 __slots__ = ("trace_id", "sql_warehouse_id", "source_inference_table", "source_databricks_request_id") 1001 class Response(_message.Message): 1002 __slots__ = ("trace_data",) 1003 TRACE_DATA_FIELD_NUMBER: _ClassVar[int] 1004 trace_data: str 1005 def __init__(self, trace_data: _Optional[str] = ...) -> None: ... 1006 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 1007 SQL_WAREHOUSE_ID_FIELD_NUMBER: _ClassVar[int] 1008 SOURCE_INFERENCE_TABLE_FIELD_NUMBER: _ClassVar[int] 1009 SOURCE_DATABRICKS_REQUEST_ID_FIELD_NUMBER: _ClassVar[int] 1010 trace_id: str 1011 sql_warehouse_id: str 1012 source_inference_table: str 1013 source_databricks_request_id: str 1014 def __init__(self, trace_id: _Optional[str] = ..., sql_warehouse_id: _Optional[str] = ..., source_inference_table: _Optional[str] = ..., source_databricks_request_id: _Optional[str] = ...) -> None: ... 1015 1016 class DeleteTraces(_message.Message): 1017 __slots__ = ("experiment_id", "max_timestamp_millis", "max_traces", "request_ids") 1018 class Response(_message.Message): 1019 __slots__ = ("traces_deleted",) 1020 TRACES_DELETED_FIELD_NUMBER: _ClassVar[int] 1021 traces_deleted: int 1022 def __init__(self, traces_deleted: _Optional[int] = ...) -> None: ... 1023 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1024 MAX_TIMESTAMP_MILLIS_FIELD_NUMBER: _ClassVar[int] 1025 MAX_TRACES_FIELD_NUMBER: _ClassVar[int] 1026 REQUEST_IDS_FIELD_NUMBER: _ClassVar[int] 1027 experiment_id: str 1028 max_timestamp_millis: int 1029 max_traces: int 1030 request_ids: _containers.RepeatedScalarFieldContainer[str] 1031 def __init__(self, experiment_id: _Optional[str] = ..., max_timestamp_millis: _Optional[int] = ..., max_traces: _Optional[int] = ..., request_ids: _Optional[_Iterable[str]] = ...) -> None: ... 1032 1033 class DeleteTracesV3(_message.Message): 1034 __slots__ = ("experiment_id", "max_timestamp_millis", "max_traces", "request_ids") 1035 class Response(_message.Message): 1036 __slots__ = ("traces_deleted",) 1037 TRACES_DELETED_FIELD_NUMBER: _ClassVar[int] 1038 traces_deleted: int 1039 def __init__(self, traces_deleted: _Optional[int] = ...) -> None: ... 1040 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1041 MAX_TIMESTAMP_MILLIS_FIELD_NUMBER: _ClassVar[int] 1042 MAX_TRACES_FIELD_NUMBER: _ClassVar[int] 1043 REQUEST_IDS_FIELD_NUMBER: _ClassVar[int] 1044 experiment_id: str 1045 max_timestamp_millis: int 1046 max_traces: int 1047 request_ids: _containers.RepeatedScalarFieldContainer[str] 1048 def __init__(self, experiment_id: _Optional[str] = ..., max_timestamp_millis: _Optional[int] = ..., max_traces: _Optional[int] = ..., request_ids: _Optional[_Iterable[str]] = ...) -> None: ... 1049 1050 class CalculateTraceFilterCorrelation(_message.Message): 1051 __slots__ = ("experiment_ids", "filter_string1", "filter_string2", "base_filter") 1052 class Response(_message.Message): 1053 __slots__ = ("npmi", "npmi_smoothed", "filter1_count", "filter2_count", "joint_count", "total_count") 1054 NPMI_FIELD_NUMBER: _ClassVar[int] 1055 NPMI_SMOOTHED_FIELD_NUMBER: _ClassVar[int] 1056 FILTER1_COUNT_FIELD_NUMBER: _ClassVar[int] 1057 FILTER2_COUNT_FIELD_NUMBER: _ClassVar[int] 1058 JOINT_COUNT_FIELD_NUMBER: _ClassVar[int] 1059 TOTAL_COUNT_FIELD_NUMBER: _ClassVar[int] 1060 npmi: float 1061 npmi_smoothed: float 1062 filter1_count: int 1063 filter2_count: int 1064 joint_count: int 1065 total_count: int 1066 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: ... 1067 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1068 FILTER_STRING1_FIELD_NUMBER: _ClassVar[int] 1069 FILTER_STRING2_FIELD_NUMBER: _ClassVar[int] 1070 BASE_FILTER_FIELD_NUMBER: _ClassVar[int] 1071 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1072 filter_string1: str 1073 filter_string2: str 1074 base_filter: str 1075 def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ..., filter_string1: _Optional[str] = ..., filter_string2: _Optional[str] = ..., base_filter: _Optional[str] = ...) -> None: ... 1076 1077 class MetricAggregation(_message.Message): 1078 __slots__ = ("aggregation_type", "percentile_value") 1079 AGGREGATION_TYPE_FIELD_NUMBER: _ClassVar[int] 1080 PERCENTILE_VALUE_FIELD_NUMBER: _ClassVar[int] 1081 aggregation_type: AggregationType 1082 percentile_value: float 1083 def __init__(self, aggregation_type: _Optional[_Union[AggregationType, str]] = ..., percentile_value: _Optional[float] = ...) -> None: ... 1084 1085 class QueryTraceMetrics(_message.Message): 1086 __slots__ = ("experiment_ids", "view_type", "metric_name", "aggregations", "dimensions", "filters", "time_interval_seconds", "start_time_ms", "end_time_ms", "max_results", "page_token") 1087 class Response(_message.Message): 1088 __slots__ = ("data_points", "next_page_token") 1089 DATA_POINTS_FIELD_NUMBER: _ClassVar[int] 1090 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1091 data_points: _containers.RepeatedCompositeFieldContainer[MetricDataPoint] 1092 next_page_token: str 1093 def __init__(self, data_points: _Optional[_Iterable[_Union[MetricDataPoint, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 1094 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1095 VIEW_TYPE_FIELD_NUMBER: _ClassVar[int] 1096 METRIC_NAME_FIELD_NUMBER: _ClassVar[int] 1097 AGGREGATIONS_FIELD_NUMBER: _ClassVar[int] 1098 DIMENSIONS_FIELD_NUMBER: _ClassVar[int] 1099 FILTERS_FIELD_NUMBER: _ClassVar[int] 1100 TIME_INTERVAL_SECONDS_FIELD_NUMBER: _ClassVar[int] 1101 START_TIME_MS_FIELD_NUMBER: _ClassVar[int] 1102 END_TIME_MS_FIELD_NUMBER: _ClassVar[int] 1103 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 1104 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1105 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1106 view_type: MetricViewType 1107 metric_name: str 1108 aggregations: _containers.RepeatedCompositeFieldContainer[MetricAggregation] 1109 dimensions: _containers.RepeatedScalarFieldContainer[str] 1110 filters: _containers.RepeatedScalarFieldContainer[str] 1111 time_interval_seconds: int 1112 start_time_ms: int 1113 end_time_ms: int 1114 max_results: int 1115 page_token: str 1116 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: ... 1117 1118 class MetricDataPoint(_message.Message): 1119 __slots__ = ("metric_name", "dimensions", "values") 1120 class DimensionsEntry(_message.Message): 1121 __slots__ = ("key", "value") 1122 KEY_FIELD_NUMBER: _ClassVar[int] 1123 VALUE_FIELD_NUMBER: _ClassVar[int] 1124 key: str 1125 value: str 1126 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1127 class ValuesEntry(_message.Message): 1128 __slots__ = ("key", "value") 1129 KEY_FIELD_NUMBER: _ClassVar[int] 1130 VALUE_FIELD_NUMBER: _ClassVar[int] 1131 key: str 1132 value: float 1133 def __init__(self, key: _Optional[str] = ..., value: _Optional[float] = ...) -> None: ... 1134 METRIC_NAME_FIELD_NUMBER: _ClassVar[int] 1135 DIMENSIONS_FIELD_NUMBER: _ClassVar[int] 1136 VALUES_FIELD_NUMBER: _ClassVar[int] 1137 metric_name: str 1138 dimensions: _containers.ScalarMap[str, str] 1139 values: _containers.ScalarMap[str, float] 1140 def __init__(self, metric_name: _Optional[str] = ..., dimensions: _Optional[_Mapping[str, str]] = ..., values: _Optional[_Mapping[str, float]] = ...) -> None: ... 1141 1142 class SetTraceTag(_message.Message): 1143 __slots__ = ("request_id", "key", "value") 1144 class Response(_message.Message): 1145 __slots__ = () 1146 def __init__(self) -> None: ... 1147 REQUEST_ID_FIELD_NUMBER: _ClassVar[int] 1148 KEY_FIELD_NUMBER: _ClassVar[int] 1149 VALUE_FIELD_NUMBER: _ClassVar[int] 1150 request_id: str 1151 key: str 1152 value: str 1153 def __init__(self, request_id: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1154 1155 class SetTraceTagV3(_message.Message): 1156 __slots__ = ("trace_id", "key", "value") 1157 class Response(_message.Message): 1158 __slots__ = () 1159 def __init__(self) -> None: ... 1160 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 1161 KEY_FIELD_NUMBER: _ClassVar[int] 1162 VALUE_FIELD_NUMBER: _ClassVar[int] 1163 trace_id: str 1164 key: str 1165 value: str 1166 def __init__(self, trace_id: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1167 1168 class DeleteTraceTag(_message.Message): 1169 __slots__ = ("request_id", "key") 1170 class Response(_message.Message): 1171 __slots__ = () 1172 def __init__(self) -> None: ... 1173 REQUEST_ID_FIELD_NUMBER: _ClassVar[int] 1174 KEY_FIELD_NUMBER: _ClassVar[int] 1175 request_id: str 1176 key: str 1177 def __init__(self, request_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... 1178 1179 class DeleteTraceTagV3(_message.Message): 1180 __slots__ = ("trace_id", "key") 1181 class Response(_message.Message): 1182 __slots__ = () 1183 def __init__(self) -> None: ... 1184 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 1185 KEY_FIELD_NUMBER: _ClassVar[int] 1186 trace_id: str 1187 key: str 1188 def __init__(self, trace_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... 1189 1190 class Trace(_message.Message): 1191 __slots__ = ("trace_info", "spans") 1192 TRACE_INFO_FIELD_NUMBER: _ClassVar[int] 1193 SPANS_FIELD_NUMBER: _ClassVar[int] 1194 trace_info: TraceInfoV3 1195 spans: _containers.RepeatedCompositeFieldContainer[_trace_pb2.Span] 1196 def __init__(self, trace_info: _Optional[_Union[TraceInfoV3, _Mapping]] = ..., spans: _Optional[_Iterable[_Union[_trace_pb2.Span, _Mapping]]] = ...) -> None: ... 1197 1198 class TraceLocation(_message.Message): 1199 __slots__ = ("type", "mlflow_experiment", "inference_table") 1200 class TraceLocationType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 1201 __slots__ = () 1202 TRACE_LOCATION_TYPE_UNSPECIFIED: _ClassVar[TraceLocation.TraceLocationType] 1203 MLFLOW_EXPERIMENT: _ClassVar[TraceLocation.TraceLocationType] 1204 INFERENCE_TABLE: _ClassVar[TraceLocation.TraceLocationType] 1205 TRACE_LOCATION_TYPE_UNSPECIFIED: TraceLocation.TraceLocationType 1206 MLFLOW_EXPERIMENT: TraceLocation.TraceLocationType 1207 INFERENCE_TABLE: TraceLocation.TraceLocationType 1208 class MlflowExperimentLocation(_message.Message): 1209 __slots__ = ("experiment_id",) 1210 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1211 experiment_id: str 1212 def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... 1213 class InferenceTableLocation(_message.Message): 1214 __slots__ = ("full_table_name",) 1215 FULL_TABLE_NAME_FIELD_NUMBER: _ClassVar[int] 1216 full_table_name: str 1217 def __init__(self, full_table_name: _Optional[str] = ...) -> None: ... 1218 TYPE_FIELD_NUMBER: _ClassVar[int] 1219 MLFLOW_EXPERIMENT_FIELD_NUMBER: _ClassVar[int] 1220 INFERENCE_TABLE_FIELD_NUMBER: _ClassVar[int] 1221 type: TraceLocation.TraceLocationType 1222 mlflow_experiment: TraceLocation.MlflowExperimentLocation 1223 inference_table: TraceLocation.InferenceTableLocation 1224 def __init__(self, type: _Optional[_Union[TraceLocation.TraceLocationType, str]] = ..., mlflow_experiment: _Optional[_Union[TraceLocation.MlflowExperimentLocation, _Mapping]] = ..., inference_table: _Optional[_Union[TraceLocation.InferenceTableLocation, _Mapping]] = ...) -> None: ... 1225 1226 class TraceInfoV3(_message.Message): 1227 __slots__ = ("trace_id", "client_request_id", "trace_location", "request", "response", "request_preview", "response_preview", "request_time", "execution_duration", "state", "trace_metadata", "assessments", "tags") 1228 class State(int, metaclass=_enum_type_wrapper.EnumTypeWrapper): 1229 __slots__ = () 1230 STATE_UNSPECIFIED: _ClassVar[TraceInfoV3.State] 1231 OK: _ClassVar[TraceInfoV3.State] 1232 ERROR: _ClassVar[TraceInfoV3.State] 1233 IN_PROGRESS: _ClassVar[TraceInfoV3.State] 1234 STATE_UNSPECIFIED: TraceInfoV3.State 1235 OK: TraceInfoV3.State 1236 ERROR: TraceInfoV3.State 1237 IN_PROGRESS: TraceInfoV3.State 1238 class TraceMetadataEntry(_message.Message): 1239 __slots__ = ("key", "value") 1240 KEY_FIELD_NUMBER: _ClassVar[int] 1241 VALUE_FIELD_NUMBER: _ClassVar[int] 1242 key: str 1243 value: str 1244 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1245 class TagsEntry(_message.Message): 1246 __slots__ = ("key", "value") 1247 KEY_FIELD_NUMBER: _ClassVar[int] 1248 VALUE_FIELD_NUMBER: _ClassVar[int] 1249 key: str 1250 value: str 1251 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1252 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 1253 CLIENT_REQUEST_ID_FIELD_NUMBER: _ClassVar[int] 1254 TRACE_LOCATION_FIELD_NUMBER: _ClassVar[int] 1255 REQUEST_FIELD_NUMBER: _ClassVar[int] 1256 RESPONSE_FIELD_NUMBER: _ClassVar[int] 1257 REQUEST_PREVIEW_FIELD_NUMBER: _ClassVar[int] 1258 RESPONSE_PREVIEW_FIELD_NUMBER: _ClassVar[int] 1259 REQUEST_TIME_FIELD_NUMBER: _ClassVar[int] 1260 EXECUTION_DURATION_FIELD_NUMBER: _ClassVar[int] 1261 STATE_FIELD_NUMBER: _ClassVar[int] 1262 TRACE_METADATA_FIELD_NUMBER: _ClassVar[int] 1263 ASSESSMENTS_FIELD_NUMBER: _ClassVar[int] 1264 TAGS_FIELD_NUMBER: _ClassVar[int] 1265 trace_id: str 1266 client_request_id: str 1267 trace_location: TraceLocation 1268 request: str 1269 response: str 1270 request_preview: str 1271 response_preview: str 1272 request_time: _timestamp_pb2.Timestamp 1273 execution_duration: _duration_pb2.Duration 1274 state: TraceInfoV3.State 1275 trace_metadata: _containers.ScalarMap[str, str] 1276 assessments: _containers.RepeatedCompositeFieldContainer[_assessments_pb2.Assessment] 1277 tags: _containers.ScalarMap[str, str] 1278 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: ... 1279 1280 class StartTraceV3(_message.Message): 1281 __slots__ = ("trace",) 1282 class Response(_message.Message): 1283 __slots__ = ("trace",) 1284 TRACE_FIELD_NUMBER: _ClassVar[int] 1285 trace: Trace 1286 def __init__(self, trace: _Optional[_Union[Trace, _Mapping]] = ...) -> None: ... 1287 TRACE_FIELD_NUMBER: _ClassVar[int] 1288 trace: Trace 1289 def __init__(self, trace: _Optional[_Union[Trace, _Mapping]] = ...) -> None: ... 1290 1291 class LinkTracesToRun(_message.Message): 1292 __slots__ = ("trace_ids", "run_id") 1293 class Response(_message.Message): 1294 __slots__ = () 1295 def __init__(self) -> None: ... 1296 TRACE_IDS_FIELD_NUMBER: _ClassVar[int] 1297 RUN_ID_FIELD_NUMBER: _ClassVar[int] 1298 trace_ids: _containers.RepeatedScalarFieldContainer[str] 1299 run_id: str 1300 def __init__(self, trace_ids: _Optional[_Iterable[str]] = ..., run_id: _Optional[str] = ...) -> None: ... 1301 1302 class LinkPromptsToTrace(_message.Message): 1303 __slots__ = ("trace_id", "prompt_versions") 1304 class PromptVersionRef(_message.Message): 1305 __slots__ = ("name", "version") 1306 NAME_FIELD_NUMBER: _ClassVar[int] 1307 VERSION_FIELD_NUMBER: _ClassVar[int] 1308 name: str 1309 version: str 1310 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... 1311 class Response(_message.Message): 1312 __slots__ = () 1313 def __init__(self) -> None: ... 1314 TRACE_ID_FIELD_NUMBER: _ClassVar[int] 1315 PROMPT_VERSIONS_FIELD_NUMBER: _ClassVar[int] 1316 trace_id: str 1317 prompt_versions: _containers.RepeatedCompositeFieldContainer[LinkPromptsToTrace.PromptVersionRef] 1318 def __init__(self, trace_id: _Optional[str] = ..., prompt_versions: _Optional[_Iterable[_Union[LinkPromptsToTrace.PromptVersionRef, _Mapping]]] = ...) -> None: ... 1319 1320 class DatasetSummary(_message.Message): 1321 __slots__ = ("experiment_id", "name", "digest", "context") 1322 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1323 NAME_FIELD_NUMBER: _ClassVar[int] 1324 DIGEST_FIELD_NUMBER: _ClassVar[int] 1325 CONTEXT_FIELD_NUMBER: _ClassVar[int] 1326 experiment_id: str 1327 name: str 1328 digest: str 1329 context: str 1330 def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., digest: _Optional[str] = ..., context: _Optional[str] = ...) -> None: ... 1331 1332 class SearchDatasets(_message.Message): 1333 __slots__ = ("experiment_ids",) 1334 class Response(_message.Message): 1335 __slots__ = ("dataset_summaries",) 1336 DATASET_SUMMARIES_FIELD_NUMBER: _ClassVar[int] 1337 dataset_summaries: _containers.RepeatedCompositeFieldContainer[DatasetSummary] 1338 def __init__(self, dataset_summaries: _Optional[_Iterable[_Union[DatasetSummary, _Mapping]]] = ...) -> None: ... 1339 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1340 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1341 def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ...) -> None: ... 1342 1343 class CreateLoggedModel(_message.Message): 1344 __slots__ = ("experiment_id", "name", "model_type", "source_run_id", "params", "tags") 1345 class Response(_message.Message): 1346 __slots__ = ("model",) 1347 MODEL_FIELD_NUMBER: _ClassVar[int] 1348 model: LoggedModel 1349 def __init__(self, model: _Optional[_Union[LoggedModel, _Mapping]] = ...) -> None: ... 1350 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1351 NAME_FIELD_NUMBER: _ClassVar[int] 1352 MODEL_TYPE_FIELD_NUMBER: _ClassVar[int] 1353 SOURCE_RUN_ID_FIELD_NUMBER: _ClassVar[int] 1354 PARAMS_FIELD_NUMBER: _ClassVar[int] 1355 TAGS_FIELD_NUMBER: _ClassVar[int] 1356 experiment_id: str 1357 name: str 1358 model_type: str 1359 source_run_id: str 1360 params: _containers.RepeatedCompositeFieldContainer[LoggedModelParameter] 1361 tags: _containers.RepeatedCompositeFieldContainer[LoggedModelTag] 1362 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: ... 1363 1364 class FinalizeLoggedModel(_message.Message): 1365 __slots__ = ("model_id", "status") 1366 class Response(_message.Message): 1367 __slots__ = ("model",) 1368 MODEL_FIELD_NUMBER: _ClassVar[int] 1369 model: LoggedModel 1370 def __init__(self, model: _Optional[_Union[LoggedModel, _Mapping]] = ...) -> None: ... 1371 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 1372 STATUS_FIELD_NUMBER: _ClassVar[int] 1373 model_id: str 1374 status: LoggedModelStatus 1375 def __init__(self, model_id: _Optional[str] = ..., status: _Optional[_Union[LoggedModelStatus, str]] = ...) -> None: ... 1376 1377 class GetLoggedModel(_message.Message): 1378 __slots__ = ("model_id",) 1379 class Response(_message.Message): 1380 __slots__ = ("model",) 1381 MODEL_FIELD_NUMBER: _ClassVar[int] 1382 model: LoggedModel 1383 def __init__(self, model: _Optional[_Union[LoggedModel, _Mapping]] = ...) -> None: ... 1384 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 1385 model_id: str 1386 def __init__(self, model_id: _Optional[str] = ...) -> None: ... 1387 1388 class DeleteLoggedModel(_message.Message): 1389 __slots__ = ("model_id",) 1390 class Response(_message.Message): 1391 __slots__ = () 1392 def __init__(self) -> None: ... 1393 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 1394 model_id: str 1395 def __init__(self, model_id: _Optional[str] = ...) -> None: ... 1396 1397 class SearchLoggedModels(_message.Message): 1398 __slots__ = ("experiment_ids", "filter", "datasets", "max_results", "order_by", "page_token") 1399 class Dataset(_message.Message): 1400 __slots__ = ("dataset_name", "dataset_digest") 1401 DATASET_NAME_FIELD_NUMBER: _ClassVar[int] 1402 DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] 1403 dataset_name: str 1404 dataset_digest: str 1405 def __init__(self, dataset_name: _Optional[str] = ..., dataset_digest: _Optional[str] = ...) -> None: ... 1406 class OrderBy(_message.Message): 1407 __slots__ = ("field_name", "ascending", "dataset_name", "dataset_digest") 1408 FIELD_NAME_FIELD_NUMBER: _ClassVar[int] 1409 ASCENDING_FIELD_NUMBER: _ClassVar[int] 1410 DATASET_NAME_FIELD_NUMBER: _ClassVar[int] 1411 DATASET_DIGEST_FIELD_NUMBER: _ClassVar[int] 1412 field_name: str 1413 ascending: bool 1414 dataset_name: str 1415 dataset_digest: str 1416 def __init__(self, field_name: _Optional[str] = ..., ascending: bool = ..., dataset_name: _Optional[str] = ..., dataset_digest: _Optional[str] = ...) -> None: ... 1417 class Response(_message.Message): 1418 __slots__ = ("models", "next_page_token") 1419 MODELS_FIELD_NUMBER: _ClassVar[int] 1420 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1421 models: _containers.RepeatedCompositeFieldContainer[LoggedModel] 1422 next_page_token: str 1423 def __init__(self, models: _Optional[_Iterable[_Union[LoggedModel, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 1424 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1425 FILTER_FIELD_NUMBER: _ClassVar[int] 1426 DATASETS_FIELD_NUMBER: _ClassVar[int] 1427 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 1428 ORDER_BY_FIELD_NUMBER: _ClassVar[int] 1429 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1430 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1431 filter: str 1432 datasets: _containers.RepeatedCompositeFieldContainer[SearchLoggedModels.Dataset] 1433 max_results: int 1434 order_by: _containers.RepeatedCompositeFieldContainer[SearchLoggedModels.OrderBy] 1435 page_token: str 1436 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: ... 1437 1438 class SetLoggedModelTags(_message.Message): 1439 __slots__ = ("model_id", "tags") 1440 class Response(_message.Message): 1441 __slots__ = ("model",) 1442 MODEL_FIELD_NUMBER: _ClassVar[int] 1443 model: LoggedModel 1444 def __init__(self, model: _Optional[_Union[LoggedModel, _Mapping]] = ...) -> None: ... 1445 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 1446 TAGS_FIELD_NUMBER: _ClassVar[int] 1447 model_id: str 1448 tags: _containers.RepeatedCompositeFieldContainer[LoggedModelTag] 1449 def __init__(self, model_id: _Optional[str] = ..., tags: _Optional[_Iterable[_Union[LoggedModelTag, _Mapping]]] = ...) -> None: ... 1450 1451 class DeleteLoggedModelTag(_message.Message): 1452 __slots__ = ("model_id", "tag_key") 1453 class Response(_message.Message): 1454 __slots__ = () 1455 def __init__(self) -> None: ... 1456 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 1457 TAG_KEY_FIELD_NUMBER: _ClassVar[int] 1458 model_id: str 1459 tag_key: str 1460 def __init__(self, model_id: _Optional[str] = ..., tag_key: _Optional[str] = ...) -> None: ... 1461 1462 class ListLoggedModelArtifacts(_message.Message): 1463 __slots__ = ("model_id", "artifact_directory_path", "page_token") 1464 class Response(_message.Message): 1465 __slots__ = ("root_uri", "files", "next_page_token") 1466 ROOT_URI_FIELD_NUMBER: _ClassVar[int] 1467 FILES_FIELD_NUMBER: _ClassVar[int] 1468 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1469 root_uri: str 1470 files: _containers.RepeatedCompositeFieldContainer[FileInfo] 1471 next_page_token: str 1472 def __init__(self, root_uri: _Optional[str] = ..., files: _Optional[_Iterable[_Union[FileInfo, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 1473 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 1474 ARTIFACT_DIRECTORY_PATH_FIELD_NUMBER: _ClassVar[int] 1475 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1476 model_id: str 1477 artifact_directory_path: str 1478 page_token: str 1479 def __init__(self, model_id: _Optional[str] = ..., artifact_directory_path: _Optional[str] = ..., page_token: _Optional[str] = ...) -> None: ... 1480 1481 class LogLoggedModelParamsRequest(_message.Message): 1482 __slots__ = ("model_id", "params") 1483 class Response(_message.Message): 1484 __slots__ = () 1485 def __init__(self) -> None: ... 1486 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 1487 PARAMS_FIELD_NUMBER: _ClassVar[int] 1488 model_id: str 1489 params: _containers.RepeatedCompositeFieldContainer[LoggedModelParameter] 1490 def __init__(self, model_id: _Optional[str] = ..., params: _Optional[_Iterable[_Union[LoggedModelParameter, _Mapping]]] = ...) -> None: ... 1491 1492 class LoggedModel(_message.Message): 1493 __slots__ = ("info", "data") 1494 INFO_FIELD_NUMBER: _ClassVar[int] 1495 DATA_FIELD_NUMBER: _ClassVar[int] 1496 info: LoggedModelInfo 1497 data: LoggedModelData 1498 def __init__(self, info: _Optional[_Union[LoggedModelInfo, _Mapping]] = ..., data: _Optional[_Union[LoggedModelData, _Mapping]] = ...) -> None: ... 1499 1500 class LoggedModelInfo(_message.Message): 1501 __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") 1502 MODEL_ID_FIELD_NUMBER: _ClassVar[int] 1503 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1504 NAME_FIELD_NUMBER: _ClassVar[int] 1505 CREATION_TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] 1506 LAST_UPDATED_TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int] 1507 ARTIFACT_URI_FIELD_NUMBER: _ClassVar[int] 1508 STATUS_FIELD_NUMBER: _ClassVar[int] 1509 CREATOR_ID_FIELD_NUMBER: _ClassVar[int] 1510 MODEL_TYPE_FIELD_NUMBER: _ClassVar[int] 1511 SOURCE_RUN_ID_FIELD_NUMBER: _ClassVar[int] 1512 STATUS_MESSAGE_FIELD_NUMBER: _ClassVar[int] 1513 TAGS_FIELD_NUMBER: _ClassVar[int] 1514 REGISTRATIONS_FIELD_NUMBER: _ClassVar[int] 1515 model_id: str 1516 experiment_id: str 1517 name: str 1518 creation_timestamp_ms: int 1519 last_updated_timestamp_ms: int 1520 artifact_uri: str 1521 status: LoggedModelStatus 1522 creator_id: int 1523 model_type: str 1524 source_run_id: str 1525 status_message: str 1526 tags: _containers.RepeatedCompositeFieldContainer[LoggedModelTag] 1527 registrations: _containers.RepeatedCompositeFieldContainer[LoggedModelRegistrationInfo] 1528 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: ... 1529 1530 class LoggedModelTag(_message.Message): 1531 __slots__ = ("key", "value") 1532 KEY_FIELD_NUMBER: _ClassVar[int] 1533 VALUE_FIELD_NUMBER: _ClassVar[int] 1534 key: str 1535 value: str 1536 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1537 1538 class LoggedModelRegistrationInfo(_message.Message): 1539 __slots__ = ("name", "version") 1540 NAME_FIELD_NUMBER: _ClassVar[int] 1541 VERSION_FIELD_NUMBER: _ClassVar[int] 1542 name: str 1543 version: str 1544 def __init__(self, name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ... 1545 1546 class LoggedModelData(_message.Message): 1547 __slots__ = ("params", "metrics") 1548 PARAMS_FIELD_NUMBER: _ClassVar[int] 1549 METRICS_FIELD_NUMBER: _ClassVar[int] 1550 params: _containers.RepeatedCompositeFieldContainer[LoggedModelParameter] 1551 metrics: _containers.RepeatedCompositeFieldContainer[Metric] 1552 def __init__(self, params: _Optional[_Iterable[_Union[LoggedModelParameter, _Mapping]]] = ..., metrics: _Optional[_Iterable[_Union[Metric, _Mapping]]] = ...) -> None: ... 1553 1554 class LoggedModelParameter(_message.Message): 1555 __slots__ = ("key", "value") 1556 KEY_FIELD_NUMBER: _ClassVar[int] 1557 VALUE_FIELD_NUMBER: _ClassVar[int] 1558 key: str 1559 value: str 1560 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1561 1562 class SearchTracesV3(_message.Message): 1563 __slots__ = ("locations", "filter", "max_results", "order_by", "page_token") 1564 class Response(_message.Message): 1565 __slots__ = ("traces", "next_page_token") 1566 TRACES_FIELD_NUMBER: _ClassVar[int] 1567 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1568 traces: _containers.RepeatedCompositeFieldContainer[TraceInfoV3] 1569 next_page_token: str 1570 def __init__(self, traces: _Optional[_Iterable[_Union[TraceInfoV3, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 1571 LOCATIONS_FIELD_NUMBER: _ClassVar[int] 1572 FILTER_FIELD_NUMBER: _ClassVar[int] 1573 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 1574 ORDER_BY_FIELD_NUMBER: _ClassVar[int] 1575 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1576 locations: _containers.RepeatedCompositeFieldContainer[TraceLocation] 1577 filter: str 1578 max_results: int 1579 order_by: _containers.RepeatedScalarFieldContainer[str] 1580 page_token: str 1581 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: ... 1582 1583 class CreateDataset(_message.Message): 1584 __slots__ = ("name", "experiment_ids", "source_type", "source", "schema", "profile", "created_by", "tags") 1585 class Response(_message.Message): 1586 __slots__ = ("dataset",) 1587 DATASET_FIELD_NUMBER: _ClassVar[int] 1588 dataset: _datasets_pb2.Dataset 1589 def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ...) -> None: ... 1590 NAME_FIELD_NUMBER: _ClassVar[int] 1591 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1592 SOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] 1593 SOURCE_FIELD_NUMBER: _ClassVar[int] 1594 SCHEMA_FIELD_NUMBER: _ClassVar[int] 1595 PROFILE_FIELD_NUMBER: _ClassVar[int] 1596 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 1597 TAGS_FIELD_NUMBER: _ClassVar[int] 1598 name: str 1599 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1600 source_type: _datasets_pb2.DatasetRecordSource.SourceType 1601 source: str 1602 schema: str 1603 profile: str 1604 created_by: str 1605 tags: str 1606 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: ... 1607 1608 class GetDataset(_message.Message): 1609 __slots__ = ("dataset_id", "page_token") 1610 class Response(_message.Message): 1611 __slots__ = ("dataset", "next_page_token") 1612 DATASET_FIELD_NUMBER: _ClassVar[int] 1613 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1614 dataset: _datasets_pb2.Dataset 1615 next_page_token: str 1616 def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 1617 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1618 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1619 dataset_id: str 1620 page_token: str 1621 def __init__(self, dataset_id: _Optional[str] = ..., page_token: _Optional[str] = ...) -> None: ... 1622 1623 class DeleteDataset(_message.Message): 1624 __slots__ = ("dataset_id",) 1625 class Response(_message.Message): 1626 __slots__ = () 1627 def __init__(self) -> None: ... 1628 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1629 dataset_id: str 1630 def __init__(self, dataset_id: _Optional[str] = ...) -> None: ... 1631 1632 class SearchEvaluationDatasets(_message.Message): 1633 __slots__ = ("experiment_ids", "filter_string", "max_results", "order_by", "page_token") 1634 class Response(_message.Message): 1635 __slots__ = ("datasets", "next_page_token") 1636 DATASETS_FIELD_NUMBER: _ClassVar[int] 1637 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1638 datasets: _containers.RepeatedCompositeFieldContainer[_datasets_pb2.Dataset] 1639 next_page_token: str 1640 def __init__(self, datasets: _Optional[_Iterable[_Union[_datasets_pb2.Dataset, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 1641 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1642 FILTER_STRING_FIELD_NUMBER: _ClassVar[int] 1643 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 1644 ORDER_BY_FIELD_NUMBER: _ClassVar[int] 1645 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1646 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1647 filter_string: str 1648 max_results: int 1649 order_by: _containers.RepeatedScalarFieldContainer[str] 1650 page_token: str 1651 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: ... 1652 1653 class SetDatasetTags(_message.Message): 1654 __slots__ = ("dataset_id", "tags") 1655 class Response(_message.Message): 1656 __slots__ = ("dataset",) 1657 DATASET_FIELD_NUMBER: _ClassVar[int] 1658 dataset: _datasets_pb2.Dataset 1659 def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ...) -> None: ... 1660 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1661 TAGS_FIELD_NUMBER: _ClassVar[int] 1662 dataset_id: str 1663 tags: str 1664 def __init__(self, dataset_id: _Optional[str] = ..., tags: _Optional[str] = ...) -> None: ... 1665 1666 class DeleteDatasetTag(_message.Message): 1667 __slots__ = ("dataset_id", "key") 1668 class Response(_message.Message): 1669 __slots__ = () 1670 def __init__(self) -> None: ... 1671 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1672 KEY_FIELD_NUMBER: _ClassVar[int] 1673 dataset_id: str 1674 key: str 1675 def __init__(self, dataset_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... 1676 1677 class UpsertDatasetRecords(_message.Message): 1678 __slots__ = ("dataset_id", "records", "updated_by") 1679 class Response(_message.Message): 1680 __slots__ = ("inserted_count", "updated_count") 1681 INSERTED_COUNT_FIELD_NUMBER: _ClassVar[int] 1682 UPDATED_COUNT_FIELD_NUMBER: _ClassVar[int] 1683 inserted_count: int 1684 updated_count: int 1685 def __init__(self, inserted_count: _Optional[int] = ..., updated_count: _Optional[int] = ...) -> None: ... 1686 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1687 RECORDS_FIELD_NUMBER: _ClassVar[int] 1688 UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 1689 dataset_id: str 1690 records: str 1691 updated_by: str 1692 def __init__(self, dataset_id: _Optional[str] = ..., records: _Optional[str] = ..., updated_by: _Optional[str] = ...) -> None: ... 1693 1694 class GetDatasetExperimentIds(_message.Message): 1695 __slots__ = ("dataset_id",) 1696 class Response(_message.Message): 1697 __slots__ = ("experiment_ids",) 1698 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1699 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1700 def __init__(self, experiment_ids: _Optional[_Iterable[str]] = ...) -> None: ... 1701 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1702 dataset_id: str 1703 def __init__(self, dataset_id: _Optional[str] = ...) -> None: ... 1704 1705 class GetDatasetRecords(_message.Message): 1706 __slots__ = ("dataset_id", "max_results", "page_token") 1707 class Response(_message.Message): 1708 __slots__ = ("records", "next_page_token") 1709 RECORDS_FIELD_NUMBER: _ClassVar[int] 1710 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1711 records: str 1712 next_page_token: str 1713 def __init__(self, records: _Optional[str] = ..., next_page_token: _Optional[str] = ...) -> None: ... 1714 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1715 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 1716 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 1717 dataset_id: str 1718 max_results: int 1719 page_token: str 1720 def __init__(self, dataset_id: _Optional[str] = ..., max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... 1721 1722 class DeleteDatasetRecords(_message.Message): 1723 __slots__ = ("dataset_id", "dataset_record_ids") 1724 class Response(_message.Message): 1725 __slots__ = ("deleted_count",) 1726 DELETED_COUNT_FIELD_NUMBER: _ClassVar[int] 1727 deleted_count: int 1728 def __init__(self, deleted_count: _Optional[int] = ...) -> None: ... 1729 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1730 DATASET_RECORD_IDS_FIELD_NUMBER: _ClassVar[int] 1731 dataset_id: str 1732 dataset_record_ids: _containers.RepeatedScalarFieldContainer[str] 1733 def __init__(self, dataset_id: _Optional[str] = ..., dataset_record_ids: _Optional[_Iterable[str]] = ...) -> None: ... 1734 1735 class AddDatasetToExperiments(_message.Message): 1736 __slots__ = ("dataset_id", "experiment_ids") 1737 class Response(_message.Message): 1738 __slots__ = ("dataset",) 1739 DATASET_FIELD_NUMBER: _ClassVar[int] 1740 dataset: _datasets_pb2.Dataset 1741 def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ...) -> None: ... 1742 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1743 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1744 dataset_id: str 1745 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1746 def __init__(self, dataset_id: _Optional[str] = ..., experiment_ids: _Optional[_Iterable[str]] = ...) -> None: ... 1747 1748 class RemoveDatasetFromExperiments(_message.Message): 1749 __slots__ = ("dataset_id", "experiment_ids") 1750 class Response(_message.Message): 1751 __slots__ = ("dataset",) 1752 DATASET_FIELD_NUMBER: _ClassVar[int] 1753 dataset: _datasets_pb2.Dataset 1754 def __init__(self, dataset: _Optional[_Union[_datasets_pb2.Dataset, _Mapping]] = ...) -> None: ... 1755 DATASET_ID_FIELD_NUMBER: _ClassVar[int] 1756 EXPERIMENT_IDS_FIELD_NUMBER: _ClassVar[int] 1757 dataset_id: str 1758 experiment_ids: _containers.RepeatedScalarFieldContainer[str] 1759 def __init__(self, dataset_id: _Optional[str] = ..., experiment_ids: _Optional[_Iterable[str]] = ...) -> None: ... 1760 1761 class RegisterScorer(_message.Message): 1762 __slots__ = ("experiment_id", "name", "serialized_scorer") 1763 class Response(_message.Message): 1764 __slots__ = ("version", "scorer_id", "experiment_id", "name", "serialized_scorer", "creation_time") 1765 VERSION_FIELD_NUMBER: _ClassVar[int] 1766 SCORER_ID_FIELD_NUMBER: _ClassVar[int] 1767 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1768 NAME_FIELD_NUMBER: _ClassVar[int] 1769 SERIALIZED_SCORER_FIELD_NUMBER: _ClassVar[int] 1770 CREATION_TIME_FIELD_NUMBER: _ClassVar[int] 1771 version: int 1772 scorer_id: str 1773 experiment_id: str 1774 name: str 1775 serialized_scorer: str 1776 creation_time: int 1777 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: ... 1778 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1779 NAME_FIELD_NUMBER: _ClassVar[int] 1780 SERIALIZED_SCORER_FIELD_NUMBER: _ClassVar[int] 1781 experiment_id: str 1782 name: str 1783 serialized_scorer: str 1784 def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., serialized_scorer: _Optional[str] = ...) -> None: ... 1785 1786 class ListScorers(_message.Message): 1787 __slots__ = ("experiment_id",) 1788 class Response(_message.Message): 1789 __slots__ = ("scorers",) 1790 SCORERS_FIELD_NUMBER: _ClassVar[int] 1791 scorers: _containers.RepeatedCompositeFieldContainer[Scorer] 1792 def __init__(self, scorers: _Optional[_Iterable[_Union[Scorer, _Mapping]]] = ...) -> None: ... 1793 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1794 experiment_id: str 1795 def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... 1796 1797 class ListScorerVersions(_message.Message): 1798 __slots__ = ("experiment_id", "name") 1799 class Response(_message.Message): 1800 __slots__ = ("scorers",) 1801 SCORERS_FIELD_NUMBER: _ClassVar[int] 1802 scorers: _containers.RepeatedCompositeFieldContainer[Scorer] 1803 def __init__(self, scorers: _Optional[_Iterable[_Union[Scorer, _Mapping]]] = ...) -> None: ... 1804 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1805 NAME_FIELD_NUMBER: _ClassVar[int] 1806 experiment_id: str 1807 name: str 1808 def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ... 1809 1810 class GetScorer(_message.Message): 1811 __slots__ = ("experiment_id", "name", "version") 1812 class Response(_message.Message): 1813 __slots__ = ("scorer",) 1814 SCORER_FIELD_NUMBER: _ClassVar[int] 1815 scorer: Scorer 1816 def __init__(self, scorer: _Optional[_Union[Scorer, _Mapping]] = ...) -> None: ... 1817 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1818 NAME_FIELD_NUMBER: _ClassVar[int] 1819 VERSION_FIELD_NUMBER: _ClassVar[int] 1820 experiment_id: str 1821 name: str 1822 version: int 1823 def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[int] = ...) -> None: ... 1824 1825 class DeleteScorer(_message.Message): 1826 __slots__ = ("experiment_id", "name", "version") 1827 class Response(_message.Message): 1828 __slots__ = () 1829 def __init__(self) -> None: ... 1830 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1831 NAME_FIELD_NUMBER: _ClassVar[int] 1832 VERSION_FIELD_NUMBER: _ClassVar[int] 1833 experiment_id: str 1834 name: str 1835 version: int 1836 def __init__(self, experiment_id: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[int] = ...) -> None: ... 1837 1838 class Scorer(_message.Message): 1839 __slots__ = ("experiment_id", "scorer_name", "scorer_version", "serialized_scorer", "creation_time", "scorer_id") 1840 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1841 SCORER_NAME_FIELD_NUMBER: _ClassVar[int] 1842 SCORER_VERSION_FIELD_NUMBER: _ClassVar[int] 1843 SERIALIZED_SCORER_FIELD_NUMBER: _ClassVar[int] 1844 CREATION_TIME_FIELD_NUMBER: _ClassVar[int] 1845 SCORER_ID_FIELD_NUMBER: _ClassVar[int] 1846 experiment_id: int 1847 scorer_name: str 1848 scorer_version: int 1849 serialized_scorer: str 1850 creation_time: int 1851 scorer_id: str 1852 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: ... 1853 1854 class GatewaySecretInfo(_message.Message): 1855 __slots__ = ("secret_id", "secret_name", "masked_values", "created_at", "last_updated_at", "provider", "created_by", "last_updated_by", "auth_config") 1856 class MaskedValuesEntry(_message.Message): 1857 __slots__ = ("key", "value") 1858 KEY_FIELD_NUMBER: _ClassVar[int] 1859 VALUE_FIELD_NUMBER: _ClassVar[int] 1860 key: str 1861 value: str 1862 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1863 class AuthConfigEntry(_message.Message): 1864 __slots__ = ("key", "value") 1865 KEY_FIELD_NUMBER: _ClassVar[int] 1866 VALUE_FIELD_NUMBER: _ClassVar[int] 1867 key: str 1868 value: str 1869 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1870 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 1871 SECRET_NAME_FIELD_NUMBER: _ClassVar[int] 1872 MASKED_VALUES_FIELD_NUMBER: _ClassVar[int] 1873 CREATED_AT_FIELD_NUMBER: _ClassVar[int] 1874 LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] 1875 PROVIDER_FIELD_NUMBER: _ClassVar[int] 1876 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 1877 LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 1878 AUTH_CONFIG_FIELD_NUMBER: _ClassVar[int] 1879 secret_id: str 1880 secret_name: str 1881 masked_values: _containers.ScalarMap[str, str] 1882 created_at: int 1883 last_updated_at: int 1884 provider: str 1885 created_by: str 1886 last_updated_by: str 1887 auth_config: _containers.ScalarMap[str, str] 1888 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: ... 1889 1890 class GatewayModelDefinition(_message.Message): 1891 __slots__ = ("model_definition_id", "name", "secret_id", "secret_name", "provider", "model_name", "created_at", "last_updated_at", "created_by", "last_updated_by") 1892 MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] 1893 NAME_FIELD_NUMBER: _ClassVar[int] 1894 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 1895 SECRET_NAME_FIELD_NUMBER: _ClassVar[int] 1896 PROVIDER_FIELD_NUMBER: _ClassVar[int] 1897 MODEL_NAME_FIELD_NUMBER: _ClassVar[int] 1898 CREATED_AT_FIELD_NUMBER: _ClassVar[int] 1899 LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] 1900 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 1901 LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 1902 model_definition_id: str 1903 name: str 1904 secret_id: str 1905 secret_name: str 1906 provider: str 1907 model_name: str 1908 created_at: int 1909 last_updated_at: int 1910 created_by: str 1911 last_updated_by: str 1912 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: ... 1913 1914 class GatewayEndpointModelMapping(_message.Message): 1915 __slots__ = ("mapping_id", "endpoint_id", "model_definition_id", "model_definition", "weight", "created_at", "created_by", "linkage_type", "fallback_order") 1916 MAPPING_ID_FIELD_NUMBER: _ClassVar[int] 1917 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 1918 MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] 1919 MODEL_DEFINITION_FIELD_NUMBER: _ClassVar[int] 1920 WEIGHT_FIELD_NUMBER: _ClassVar[int] 1921 CREATED_AT_FIELD_NUMBER: _ClassVar[int] 1922 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 1923 LINKAGE_TYPE_FIELD_NUMBER: _ClassVar[int] 1924 FALLBACK_ORDER_FIELD_NUMBER: _ClassVar[int] 1925 mapping_id: str 1926 endpoint_id: str 1927 model_definition_id: str 1928 model_definition: GatewayModelDefinition 1929 weight: float 1930 created_at: int 1931 created_by: str 1932 linkage_type: GatewayModelLinkageType 1933 fallback_order: int 1934 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: ... 1935 1936 class GatewayEndpoint(_message.Message): 1937 __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") 1938 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 1939 NAME_FIELD_NUMBER: _ClassVar[int] 1940 CREATED_AT_FIELD_NUMBER: _ClassVar[int] 1941 LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] 1942 MODEL_MAPPINGS_FIELD_NUMBER: _ClassVar[int] 1943 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 1944 LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 1945 TAGS_FIELD_NUMBER: _ClassVar[int] 1946 ROUTING_STRATEGY_FIELD_NUMBER: _ClassVar[int] 1947 FALLBACK_CONFIG_FIELD_NUMBER: _ClassVar[int] 1948 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 1949 USAGE_TRACKING_FIELD_NUMBER: _ClassVar[int] 1950 endpoint_id: str 1951 name: str 1952 created_at: int 1953 last_updated_at: int 1954 model_mappings: _containers.RepeatedCompositeFieldContainer[GatewayEndpointModelMapping] 1955 created_by: str 1956 last_updated_by: str 1957 tags: _containers.RepeatedCompositeFieldContainer[GatewayEndpointTag] 1958 routing_strategy: RoutingStrategy 1959 fallback_config: FallbackConfig 1960 experiment_id: str 1961 usage_tracking: bool 1962 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: ... 1963 1964 class GatewayEndpointTag(_message.Message): 1965 __slots__ = ("key", "value") 1966 KEY_FIELD_NUMBER: _ClassVar[int] 1967 VALUE_FIELD_NUMBER: _ClassVar[int] 1968 key: str 1969 value: str 1970 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 1971 1972 class GatewayEndpointBinding(_message.Message): 1973 __slots__ = ("endpoint_id", "resource_type", "resource_id", "created_at", "last_updated_at", "created_by", "last_updated_by", "display_name") 1974 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 1975 RESOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] 1976 RESOURCE_ID_FIELD_NUMBER: _ClassVar[int] 1977 CREATED_AT_FIELD_NUMBER: _ClassVar[int] 1978 LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] 1979 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 1980 LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 1981 DISPLAY_NAME_FIELD_NUMBER: _ClassVar[int] 1982 endpoint_id: str 1983 resource_type: str 1984 resource_id: str 1985 created_at: int 1986 last_updated_at: int 1987 created_by: str 1988 last_updated_by: str 1989 display_name: str 1990 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: ... 1991 1992 class CreateGatewaySecret(_message.Message): 1993 __slots__ = ("secret_name", "secret_value", "provider", "auth_config", "created_by") 1994 class SecretValueEntry(_message.Message): 1995 __slots__ = ("key", "value") 1996 KEY_FIELD_NUMBER: _ClassVar[int] 1997 VALUE_FIELD_NUMBER: _ClassVar[int] 1998 key: str 1999 value: str 2000 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 2001 class AuthConfigEntry(_message.Message): 2002 __slots__ = ("key", "value") 2003 KEY_FIELD_NUMBER: _ClassVar[int] 2004 VALUE_FIELD_NUMBER: _ClassVar[int] 2005 key: str 2006 value: str 2007 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 2008 class Response(_message.Message): 2009 __slots__ = ("secret",) 2010 SECRET_FIELD_NUMBER: _ClassVar[int] 2011 secret: GatewaySecretInfo 2012 def __init__(self, secret: _Optional[_Union[GatewaySecretInfo, _Mapping]] = ...) -> None: ... 2013 SECRET_NAME_FIELD_NUMBER: _ClassVar[int] 2014 SECRET_VALUE_FIELD_NUMBER: _ClassVar[int] 2015 PROVIDER_FIELD_NUMBER: _ClassVar[int] 2016 AUTH_CONFIG_FIELD_NUMBER: _ClassVar[int] 2017 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2018 secret_name: str 2019 secret_value: _containers.ScalarMap[str, str] 2020 provider: str 2021 auth_config: _containers.ScalarMap[str, str] 2022 created_by: str 2023 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: ... 2024 2025 class GetGatewaySecretInfo(_message.Message): 2026 __slots__ = ("secret_id", "secret_name") 2027 class Response(_message.Message): 2028 __slots__ = ("secret",) 2029 SECRET_FIELD_NUMBER: _ClassVar[int] 2030 secret: GatewaySecretInfo 2031 def __init__(self, secret: _Optional[_Union[GatewaySecretInfo, _Mapping]] = ...) -> None: ... 2032 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 2033 SECRET_NAME_FIELD_NUMBER: _ClassVar[int] 2034 secret_id: str 2035 secret_name: str 2036 def __init__(self, secret_id: _Optional[str] = ..., secret_name: _Optional[str] = ...) -> None: ... 2037 2038 class UpdateGatewaySecret(_message.Message): 2039 __slots__ = ("secret_id", "secret_value", "auth_config", "updated_by") 2040 class SecretValueEntry(_message.Message): 2041 __slots__ = ("key", "value") 2042 KEY_FIELD_NUMBER: _ClassVar[int] 2043 VALUE_FIELD_NUMBER: _ClassVar[int] 2044 key: str 2045 value: str 2046 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 2047 class AuthConfigEntry(_message.Message): 2048 __slots__ = ("key", "value") 2049 KEY_FIELD_NUMBER: _ClassVar[int] 2050 VALUE_FIELD_NUMBER: _ClassVar[int] 2051 key: str 2052 value: str 2053 def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 2054 class Response(_message.Message): 2055 __slots__ = ("secret",) 2056 SECRET_FIELD_NUMBER: _ClassVar[int] 2057 secret: GatewaySecretInfo 2058 def __init__(self, secret: _Optional[_Union[GatewaySecretInfo, _Mapping]] = ...) -> None: ... 2059 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 2060 SECRET_VALUE_FIELD_NUMBER: _ClassVar[int] 2061 AUTH_CONFIG_FIELD_NUMBER: _ClassVar[int] 2062 UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 2063 secret_id: str 2064 secret_value: _containers.ScalarMap[str, str] 2065 auth_config: _containers.ScalarMap[str, str] 2066 updated_by: str 2067 def __init__(self, secret_id: _Optional[str] = ..., secret_value: _Optional[_Mapping[str, str]] = ..., auth_config: _Optional[_Mapping[str, str]] = ..., updated_by: _Optional[str] = ...) -> None: ... 2068 2069 class DeleteGatewaySecret(_message.Message): 2070 __slots__ = ("secret_id",) 2071 class Response(_message.Message): 2072 __slots__ = () 2073 def __init__(self) -> None: ... 2074 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 2075 secret_id: str 2076 def __init__(self, secret_id: _Optional[str] = ...) -> None: ... 2077 2078 class ListGatewaySecretInfos(_message.Message): 2079 __slots__ = ("provider",) 2080 class Response(_message.Message): 2081 __slots__ = ("secrets",) 2082 SECRETS_FIELD_NUMBER: _ClassVar[int] 2083 secrets: _containers.RepeatedCompositeFieldContainer[GatewaySecretInfo] 2084 def __init__(self, secrets: _Optional[_Iterable[_Union[GatewaySecretInfo, _Mapping]]] = ...) -> None: ... 2085 PROVIDER_FIELD_NUMBER: _ClassVar[int] 2086 provider: str 2087 def __init__(self, provider: _Optional[str] = ...) -> None: ... 2088 2089 class CreateGatewayModelDefinition(_message.Message): 2090 __slots__ = ("name", "secret_id", "provider", "model_name", "created_by") 2091 class Response(_message.Message): 2092 __slots__ = ("model_definition",) 2093 MODEL_DEFINITION_FIELD_NUMBER: _ClassVar[int] 2094 model_definition: GatewayModelDefinition 2095 def __init__(self, model_definition: _Optional[_Union[GatewayModelDefinition, _Mapping]] = ...) -> None: ... 2096 NAME_FIELD_NUMBER: _ClassVar[int] 2097 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 2098 PROVIDER_FIELD_NUMBER: _ClassVar[int] 2099 MODEL_NAME_FIELD_NUMBER: _ClassVar[int] 2100 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2101 name: str 2102 secret_id: str 2103 provider: str 2104 model_name: str 2105 created_by: str 2106 def __init__(self, name: _Optional[str] = ..., secret_id: _Optional[str] = ..., provider: _Optional[str] = ..., model_name: _Optional[str] = ..., created_by: _Optional[str] = ...) -> None: ... 2107 2108 class GetGatewayModelDefinition(_message.Message): 2109 __slots__ = ("model_definition_id",) 2110 class Response(_message.Message): 2111 __slots__ = ("model_definition",) 2112 MODEL_DEFINITION_FIELD_NUMBER: _ClassVar[int] 2113 model_definition: GatewayModelDefinition 2114 def __init__(self, model_definition: _Optional[_Union[GatewayModelDefinition, _Mapping]] = ...) -> None: ... 2115 MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] 2116 model_definition_id: str 2117 def __init__(self, model_definition_id: _Optional[str] = ...) -> None: ... 2118 2119 class ListGatewayModelDefinitions(_message.Message): 2120 __slots__ = ("provider", "secret_id") 2121 class Response(_message.Message): 2122 __slots__ = ("model_definitions",) 2123 MODEL_DEFINITIONS_FIELD_NUMBER: _ClassVar[int] 2124 model_definitions: _containers.RepeatedCompositeFieldContainer[GatewayModelDefinition] 2125 def __init__(self, model_definitions: _Optional[_Iterable[_Union[GatewayModelDefinition, _Mapping]]] = ...) -> None: ... 2126 PROVIDER_FIELD_NUMBER: _ClassVar[int] 2127 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 2128 provider: str 2129 secret_id: str 2130 def __init__(self, provider: _Optional[str] = ..., secret_id: _Optional[str] = ...) -> None: ... 2131 2132 class UpdateGatewayModelDefinition(_message.Message): 2133 __slots__ = ("model_definition_id", "name", "secret_id", "model_name", "updated_by", "provider") 2134 class Response(_message.Message): 2135 __slots__ = ("model_definition",) 2136 MODEL_DEFINITION_FIELD_NUMBER: _ClassVar[int] 2137 model_definition: GatewayModelDefinition 2138 def __init__(self, model_definition: _Optional[_Union[GatewayModelDefinition, _Mapping]] = ...) -> None: ... 2139 MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] 2140 NAME_FIELD_NUMBER: _ClassVar[int] 2141 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 2142 MODEL_NAME_FIELD_NUMBER: _ClassVar[int] 2143 UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 2144 PROVIDER_FIELD_NUMBER: _ClassVar[int] 2145 model_definition_id: str 2146 name: str 2147 secret_id: str 2148 model_name: str 2149 updated_by: str 2150 provider: str 2151 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: ... 2152 2153 class DeleteGatewayModelDefinition(_message.Message): 2154 __slots__ = ("model_definition_id",) 2155 class Response(_message.Message): 2156 __slots__ = () 2157 def __init__(self) -> None: ... 2158 MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] 2159 model_definition_id: str 2160 def __init__(self, model_definition_id: _Optional[str] = ...) -> None: ... 2161 2162 class BudgetDuration(_message.Message): 2163 __slots__ = ("unit", "value") 2164 UNIT_FIELD_NUMBER: _ClassVar[int] 2165 VALUE_FIELD_NUMBER: _ClassVar[int] 2166 unit: BudgetDurationUnit 2167 value: int 2168 def __init__(self, unit: _Optional[_Union[BudgetDurationUnit, str]] = ..., value: _Optional[int] = ...) -> None: ... 2169 2170 class FallbackConfig(_message.Message): 2171 __slots__ = ("strategy", "max_attempts") 2172 STRATEGY_FIELD_NUMBER: _ClassVar[int] 2173 MAX_ATTEMPTS_FIELD_NUMBER: _ClassVar[int] 2174 strategy: FallbackStrategy 2175 max_attempts: int 2176 def __init__(self, strategy: _Optional[_Union[FallbackStrategy, str]] = ..., max_attempts: _Optional[int] = ...) -> None: ... 2177 2178 class GatewayEndpointModelConfig(_message.Message): 2179 __slots__ = ("model_definition_id", "linkage_type", "weight", "fallback_order") 2180 MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] 2181 LINKAGE_TYPE_FIELD_NUMBER: _ClassVar[int] 2182 WEIGHT_FIELD_NUMBER: _ClassVar[int] 2183 FALLBACK_ORDER_FIELD_NUMBER: _ClassVar[int] 2184 model_definition_id: str 2185 linkage_type: GatewayModelLinkageType 2186 weight: float 2187 fallback_order: int 2188 def __init__(self, model_definition_id: _Optional[str] = ..., linkage_type: _Optional[_Union[GatewayModelLinkageType, str]] = ..., weight: _Optional[float] = ..., fallback_order: _Optional[int] = ...) -> None: ... 2189 2190 class CreateGatewayEndpoint(_message.Message): 2191 __slots__ = ("name", "model_configs", "created_by", "routing_strategy", "fallback_config", "experiment_id", "usage_tracking") 2192 class Response(_message.Message): 2193 __slots__ = ("endpoint",) 2194 ENDPOINT_FIELD_NUMBER: _ClassVar[int] 2195 endpoint: GatewayEndpoint 2196 def __init__(self, endpoint: _Optional[_Union[GatewayEndpoint, _Mapping]] = ...) -> None: ... 2197 NAME_FIELD_NUMBER: _ClassVar[int] 2198 MODEL_CONFIGS_FIELD_NUMBER: _ClassVar[int] 2199 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2200 ROUTING_STRATEGY_FIELD_NUMBER: _ClassVar[int] 2201 FALLBACK_CONFIG_FIELD_NUMBER: _ClassVar[int] 2202 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 2203 USAGE_TRACKING_FIELD_NUMBER: _ClassVar[int] 2204 name: str 2205 model_configs: _containers.RepeatedCompositeFieldContainer[GatewayEndpointModelConfig] 2206 created_by: str 2207 routing_strategy: RoutingStrategy 2208 fallback_config: FallbackConfig 2209 experiment_id: str 2210 usage_tracking: bool 2211 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: ... 2212 2213 class GetGatewayEndpoint(_message.Message): 2214 __slots__ = ("endpoint_id", "name") 2215 class Response(_message.Message): 2216 __slots__ = ("endpoint",) 2217 ENDPOINT_FIELD_NUMBER: _ClassVar[int] 2218 endpoint: GatewayEndpoint 2219 def __init__(self, endpoint: _Optional[_Union[GatewayEndpoint, _Mapping]] = ...) -> None: ... 2220 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2221 NAME_FIELD_NUMBER: _ClassVar[int] 2222 endpoint_id: str 2223 name: str 2224 def __init__(self, endpoint_id: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ... 2225 2226 class UpdateGatewayEndpoint(_message.Message): 2227 __slots__ = ("endpoint_id", "name", "updated_by", "model_configs", "routing_strategy", "fallback_config", "experiment_id", "usage_tracking") 2228 class Response(_message.Message): 2229 __slots__ = ("endpoint",) 2230 ENDPOINT_FIELD_NUMBER: _ClassVar[int] 2231 endpoint: GatewayEndpoint 2232 def __init__(self, endpoint: _Optional[_Union[GatewayEndpoint, _Mapping]] = ...) -> None: ... 2233 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2234 NAME_FIELD_NUMBER: _ClassVar[int] 2235 UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 2236 MODEL_CONFIGS_FIELD_NUMBER: _ClassVar[int] 2237 ROUTING_STRATEGY_FIELD_NUMBER: _ClassVar[int] 2238 FALLBACK_CONFIG_FIELD_NUMBER: _ClassVar[int] 2239 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 2240 USAGE_TRACKING_FIELD_NUMBER: _ClassVar[int] 2241 endpoint_id: str 2242 name: str 2243 updated_by: str 2244 model_configs: _containers.RepeatedCompositeFieldContainer[GatewayEndpointModelConfig] 2245 routing_strategy: RoutingStrategy 2246 fallback_config: FallbackConfig 2247 experiment_id: str 2248 usage_tracking: bool 2249 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: ... 2250 2251 class DeleteGatewayEndpoint(_message.Message): 2252 __slots__ = ("endpoint_id",) 2253 class Response(_message.Message): 2254 __slots__ = () 2255 def __init__(self) -> None: ... 2256 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2257 endpoint_id: str 2258 def __init__(self, endpoint_id: _Optional[str] = ...) -> None: ... 2259 2260 class ListGatewayEndpoints(_message.Message): 2261 __slots__ = ("provider", "secret_id") 2262 class Response(_message.Message): 2263 __slots__ = ("endpoints",) 2264 ENDPOINTS_FIELD_NUMBER: _ClassVar[int] 2265 endpoints: _containers.RepeatedCompositeFieldContainer[GatewayEndpoint] 2266 def __init__(self, endpoints: _Optional[_Iterable[_Union[GatewayEndpoint, _Mapping]]] = ...) -> None: ... 2267 PROVIDER_FIELD_NUMBER: _ClassVar[int] 2268 SECRET_ID_FIELD_NUMBER: _ClassVar[int] 2269 provider: str 2270 secret_id: str 2271 def __init__(self, provider: _Optional[str] = ..., secret_id: _Optional[str] = ...) -> None: ... 2272 2273 class AttachModelToGatewayEndpoint(_message.Message): 2274 __slots__ = ("endpoint_id", "model_config", "created_by") 2275 class Response(_message.Message): 2276 __slots__ = ("mapping",) 2277 MAPPING_FIELD_NUMBER: _ClassVar[int] 2278 mapping: GatewayEndpointModelMapping 2279 def __init__(self, mapping: _Optional[_Union[GatewayEndpointModelMapping, _Mapping]] = ...) -> None: ... 2280 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2281 MODEL_CONFIG_FIELD_NUMBER: _ClassVar[int] 2282 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2283 endpoint_id: str 2284 model_config: GatewayEndpointModelConfig 2285 created_by: str 2286 def __init__(self, endpoint_id: _Optional[str] = ..., model_config: _Optional[_Union[GatewayEndpointModelConfig, _Mapping]] = ..., created_by: _Optional[str] = ...) -> None: ... 2287 2288 class DetachModelFromGatewayEndpoint(_message.Message): 2289 __slots__ = ("endpoint_id", "model_definition_id") 2290 class Response(_message.Message): 2291 __slots__ = () 2292 def __init__(self) -> None: ... 2293 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2294 MODEL_DEFINITION_ID_FIELD_NUMBER: _ClassVar[int] 2295 endpoint_id: str 2296 model_definition_id: str 2297 def __init__(self, endpoint_id: _Optional[str] = ..., model_definition_id: _Optional[str] = ...) -> None: ... 2298 2299 class CreateGatewayEndpointBinding(_message.Message): 2300 __slots__ = ("endpoint_id", "resource_type", "resource_id", "created_by") 2301 class Response(_message.Message): 2302 __slots__ = ("binding",) 2303 BINDING_FIELD_NUMBER: _ClassVar[int] 2304 binding: GatewayEndpointBinding 2305 def __init__(self, binding: _Optional[_Union[GatewayEndpointBinding, _Mapping]] = ...) -> None: ... 2306 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2307 RESOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] 2308 RESOURCE_ID_FIELD_NUMBER: _ClassVar[int] 2309 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2310 endpoint_id: str 2311 resource_type: str 2312 resource_id: str 2313 created_by: str 2314 def __init__(self, endpoint_id: _Optional[str] = ..., resource_type: _Optional[str] = ..., resource_id: _Optional[str] = ..., created_by: _Optional[str] = ...) -> None: ... 2315 2316 class DeleteGatewayEndpointBinding(_message.Message): 2317 __slots__ = ("endpoint_id", "resource_type", "resource_id") 2318 class Response(_message.Message): 2319 __slots__ = () 2320 def __init__(self) -> None: ... 2321 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2322 RESOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] 2323 RESOURCE_ID_FIELD_NUMBER: _ClassVar[int] 2324 endpoint_id: str 2325 resource_type: str 2326 resource_id: str 2327 def __init__(self, endpoint_id: _Optional[str] = ..., resource_type: _Optional[str] = ..., resource_id: _Optional[str] = ...) -> None: ... 2328 2329 class ListGatewayEndpointBindings(_message.Message): 2330 __slots__ = ("endpoint_id", "resource_type", "resource_id") 2331 class Response(_message.Message): 2332 __slots__ = ("bindings",) 2333 BINDINGS_FIELD_NUMBER: _ClassVar[int] 2334 bindings: _containers.RepeatedCompositeFieldContainer[GatewayEndpointBinding] 2335 def __init__(self, bindings: _Optional[_Iterable[_Union[GatewayEndpointBinding, _Mapping]]] = ...) -> None: ... 2336 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2337 RESOURCE_TYPE_FIELD_NUMBER: _ClassVar[int] 2338 RESOURCE_ID_FIELD_NUMBER: _ClassVar[int] 2339 endpoint_id: str 2340 resource_type: str 2341 resource_id: str 2342 def __init__(self, endpoint_id: _Optional[str] = ..., resource_type: _Optional[str] = ..., resource_id: _Optional[str] = ...) -> None: ... 2343 2344 class SetGatewayEndpointTag(_message.Message): 2345 __slots__ = ("endpoint_id", "key", "value") 2346 class Response(_message.Message): 2347 __slots__ = () 2348 def __init__(self) -> None: ... 2349 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2350 KEY_FIELD_NUMBER: _ClassVar[int] 2351 VALUE_FIELD_NUMBER: _ClassVar[int] 2352 endpoint_id: str 2353 key: str 2354 value: str 2355 def __init__(self, endpoint_id: _Optional[str] = ..., key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ... 2356 2357 class DeleteGatewayEndpointTag(_message.Message): 2358 __slots__ = ("endpoint_id", "key") 2359 class Response(_message.Message): 2360 __slots__ = () 2361 def __init__(self) -> None: ... 2362 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2363 KEY_FIELD_NUMBER: _ClassVar[int] 2364 endpoint_id: str 2365 key: str 2366 def __init__(self, endpoint_id: _Optional[str] = ..., key: _Optional[str] = ...) -> None: ... 2367 2368 class GatewayBudgetPolicy(_message.Message): 2369 __slots__ = ("budget_policy_id", "budget_unit", "budget_amount", "duration", "target_scope", "budget_action", "created_by", "created_at", "last_updated_by", "last_updated_at") 2370 BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] 2371 BUDGET_UNIT_FIELD_NUMBER: _ClassVar[int] 2372 BUDGET_AMOUNT_FIELD_NUMBER: _ClassVar[int] 2373 DURATION_FIELD_NUMBER: _ClassVar[int] 2374 TARGET_SCOPE_FIELD_NUMBER: _ClassVar[int] 2375 BUDGET_ACTION_FIELD_NUMBER: _ClassVar[int] 2376 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2377 CREATED_AT_FIELD_NUMBER: _ClassVar[int] 2378 LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 2379 LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] 2380 budget_policy_id: str 2381 budget_unit: BudgetUnit 2382 budget_amount: float 2383 duration: BudgetDuration 2384 target_scope: BudgetTargetScope 2385 budget_action: BudgetAction 2386 created_by: str 2387 created_at: int 2388 last_updated_by: str 2389 last_updated_at: int 2390 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: ... 2391 2392 class CreateGatewayBudgetPolicy(_message.Message): 2393 __slots__ = ("budget_unit", "budget_amount", "duration", "target_scope", "budget_action", "created_by") 2394 class Response(_message.Message): 2395 __slots__ = ("budget_policy",) 2396 BUDGET_POLICY_FIELD_NUMBER: _ClassVar[int] 2397 budget_policy: GatewayBudgetPolicy 2398 def __init__(self, budget_policy: _Optional[_Union[GatewayBudgetPolicy, _Mapping]] = ...) -> None: ... 2399 BUDGET_UNIT_FIELD_NUMBER: _ClassVar[int] 2400 BUDGET_AMOUNT_FIELD_NUMBER: _ClassVar[int] 2401 DURATION_FIELD_NUMBER: _ClassVar[int] 2402 TARGET_SCOPE_FIELD_NUMBER: _ClassVar[int] 2403 BUDGET_ACTION_FIELD_NUMBER: _ClassVar[int] 2404 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2405 budget_unit: BudgetUnit 2406 budget_amount: float 2407 duration: BudgetDuration 2408 target_scope: BudgetTargetScope 2409 budget_action: BudgetAction 2410 created_by: str 2411 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: ... 2412 2413 class GetGatewayBudgetPolicy(_message.Message): 2414 __slots__ = ("budget_policy_id",) 2415 class Response(_message.Message): 2416 __slots__ = ("budget_policy",) 2417 BUDGET_POLICY_FIELD_NUMBER: _ClassVar[int] 2418 budget_policy: GatewayBudgetPolicy 2419 def __init__(self, budget_policy: _Optional[_Union[GatewayBudgetPolicy, _Mapping]] = ...) -> None: ... 2420 BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] 2421 budget_policy_id: str 2422 def __init__(self, budget_policy_id: _Optional[str] = ...) -> None: ... 2423 2424 class UpdateGatewayBudgetPolicy(_message.Message): 2425 __slots__ = ("budget_policy_id", "budget_unit", "budget_amount", "duration", "target_scope", "budget_action", "updated_by") 2426 class Response(_message.Message): 2427 __slots__ = ("budget_policy",) 2428 BUDGET_POLICY_FIELD_NUMBER: _ClassVar[int] 2429 budget_policy: GatewayBudgetPolicy 2430 def __init__(self, budget_policy: _Optional[_Union[GatewayBudgetPolicy, _Mapping]] = ...) -> None: ... 2431 BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] 2432 BUDGET_UNIT_FIELD_NUMBER: _ClassVar[int] 2433 BUDGET_AMOUNT_FIELD_NUMBER: _ClassVar[int] 2434 DURATION_FIELD_NUMBER: _ClassVar[int] 2435 TARGET_SCOPE_FIELD_NUMBER: _ClassVar[int] 2436 BUDGET_ACTION_FIELD_NUMBER: _ClassVar[int] 2437 UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 2438 budget_policy_id: str 2439 budget_unit: BudgetUnit 2440 budget_amount: float 2441 duration: BudgetDuration 2442 target_scope: BudgetTargetScope 2443 budget_action: BudgetAction 2444 updated_by: str 2445 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: ... 2446 2447 class DeleteGatewayBudgetPolicy(_message.Message): 2448 __slots__ = ("budget_policy_id",) 2449 class Response(_message.Message): 2450 __slots__ = () 2451 def __init__(self) -> None: ... 2452 BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] 2453 budget_policy_id: str 2454 def __init__(self, budget_policy_id: _Optional[str] = ...) -> None: ... 2455 2456 class ListGatewayBudgetPolicies(_message.Message): 2457 __slots__ = ("max_results", "page_token") 2458 class Response(_message.Message): 2459 __slots__ = ("budget_policies", "next_page_token") 2460 BUDGET_POLICIES_FIELD_NUMBER: _ClassVar[int] 2461 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 2462 budget_policies: _containers.RepeatedCompositeFieldContainer[GatewayBudgetPolicy] 2463 next_page_token: str 2464 def __init__(self, budget_policies: _Optional[_Iterable[_Union[GatewayBudgetPolicy, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 2465 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 2466 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 2467 max_results: int 2468 page_token: str 2469 def __init__(self, max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... 2470 2471 class ListGatewayBudgetWindows(_message.Message): 2472 __slots__ = () 2473 class BudgetWindow(_message.Message): 2474 __slots__ = ("budget_policy_id", "window_start_ms", "window_end_ms", "current_spend") 2475 BUDGET_POLICY_ID_FIELD_NUMBER: _ClassVar[int] 2476 WINDOW_START_MS_FIELD_NUMBER: _ClassVar[int] 2477 WINDOW_END_MS_FIELD_NUMBER: _ClassVar[int] 2478 CURRENT_SPEND_FIELD_NUMBER: _ClassVar[int] 2479 budget_policy_id: str 2480 window_start_ms: int 2481 window_end_ms: int 2482 current_spend: float 2483 def __init__(self, budget_policy_id: _Optional[str] = ..., window_start_ms: _Optional[int] = ..., window_end_ms: _Optional[int] = ..., current_spend: _Optional[float] = ...) -> None: ... 2484 class Response(_message.Message): 2485 __slots__ = ("windows",) 2486 WINDOWS_FIELD_NUMBER: _ClassVar[int] 2487 windows: _containers.RepeatedCompositeFieldContainer[ListGatewayBudgetWindows.BudgetWindow] 2488 def __init__(self, windows: _Optional[_Iterable[_Union[ListGatewayBudgetWindows.BudgetWindow, _Mapping]]] = ...) -> None: ... 2489 def __init__(self) -> None: ... 2490 2491 class GatewayGuardrail(_message.Message): 2492 __slots__ = ("guardrail_id", "name", "scorer", "stage", "action", "action_endpoint_id", "created_by", "created_at", "last_updated_by", "last_updated_at") 2493 GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] 2494 NAME_FIELD_NUMBER: _ClassVar[int] 2495 SCORER_FIELD_NUMBER: _ClassVar[int] 2496 STAGE_FIELD_NUMBER: _ClassVar[int] 2497 ACTION_FIELD_NUMBER: _ClassVar[int] 2498 ACTION_ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2499 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2500 CREATED_AT_FIELD_NUMBER: _ClassVar[int] 2501 LAST_UPDATED_BY_FIELD_NUMBER: _ClassVar[int] 2502 LAST_UPDATED_AT_FIELD_NUMBER: _ClassVar[int] 2503 guardrail_id: str 2504 name: str 2505 scorer: Scorer 2506 stage: GuardrailStage 2507 action: GuardrailAction 2508 action_endpoint_id: str 2509 created_by: str 2510 created_at: int 2511 last_updated_by: str 2512 last_updated_at: int 2513 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: ... 2514 2515 class GatewayGuardrailConfig(_message.Message): 2516 __slots__ = ("endpoint_id", "guardrail_id", "execution_order", "created_by", "created_at", "guardrail") 2517 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2518 GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] 2519 EXECUTION_ORDER_FIELD_NUMBER: _ClassVar[int] 2520 CREATED_BY_FIELD_NUMBER: _ClassVar[int] 2521 CREATED_AT_FIELD_NUMBER: _ClassVar[int] 2522 GUARDRAIL_FIELD_NUMBER: _ClassVar[int] 2523 endpoint_id: str 2524 guardrail_id: str 2525 execution_order: int 2526 created_by: str 2527 created_at: int 2528 guardrail: GatewayGuardrail 2529 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: ... 2530 2531 class CreateGatewayGuardrail(_message.Message): 2532 __slots__ = ("name", "scorer_id", "scorer_version", "stage", "action", "action_endpoint_id") 2533 class Response(_message.Message): 2534 __slots__ = ("guardrail",) 2535 GUARDRAIL_FIELD_NUMBER: _ClassVar[int] 2536 guardrail: GatewayGuardrail 2537 def __init__(self, guardrail: _Optional[_Union[GatewayGuardrail, _Mapping]] = ...) -> None: ... 2538 NAME_FIELD_NUMBER: _ClassVar[int] 2539 SCORER_ID_FIELD_NUMBER: _ClassVar[int] 2540 SCORER_VERSION_FIELD_NUMBER: _ClassVar[int] 2541 STAGE_FIELD_NUMBER: _ClassVar[int] 2542 ACTION_FIELD_NUMBER: _ClassVar[int] 2543 ACTION_ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2544 name: str 2545 scorer_id: str 2546 scorer_version: int 2547 stage: GuardrailStage 2548 action: GuardrailAction 2549 action_endpoint_id: str 2550 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: ... 2551 2552 class GetGatewayGuardrail(_message.Message): 2553 __slots__ = ("guardrail_id",) 2554 class Response(_message.Message): 2555 __slots__ = ("guardrail",) 2556 GUARDRAIL_FIELD_NUMBER: _ClassVar[int] 2557 guardrail: GatewayGuardrail 2558 def __init__(self, guardrail: _Optional[_Union[GatewayGuardrail, _Mapping]] = ...) -> None: ... 2559 GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] 2560 guardrail_id: str 2561 def __init__(self, guardrail_id: _Optional[str] = ...) -> None: ... 2562 2563 class DeleteGatewayGuardrail(_message.Message): 2564 __slots__ = ("guardrail_id",) 2565 class Response(_message.Message): 2566 __slots__ = () 2567 def __init__(self) -> None: ... 2568 GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] 2569 guardrail_id: str 2570 def __init__(self, guardrail_id: _Optional[str] = ...) -> None: ... 2571 2572 class ListGatewayGuardrails(_message.Message): 2573 __slots__ = ("max_results", "page_token") 2574 class Response(_message.Message): 2575 __slots__ = ("guardrails", "next_page_token") 2576 GUARDRAILS_FIELD_NUMBER: _ClassVar[int] 2577 NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 2578 guardrails: _containers.RepeatedCompositeFieldContainer[GatewayGuardrail] 2579 next_page_token: str 2580 def __init__(self, guardrails: _Optional[_Iterable[_Union[GatewayGuardrail, _Mapping]]] = ..., next_page_token: _Optional[str] = ...) -> None: ... 2581 MAX_RESULTS_FIELD_NUMBER: _ClassVar[int] 2582 PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int] 2583 max_results: int 2584 page_token: str 2585 def __init__(self, max_results: _Optional[int] = ..., page_token: _Optional[str] = ...) -> None: ... 2586 2587 class AddGuardrailToEndpoint(_message.Message): 2588 __slots__ = ("endpoint_id", "guardrail_id", "execution_order") 2589 class Response(_message.Message): 2590 __slots__ = ("config",) 2591 CONFIG_FIELD_NUMBER: _ClassVar[int] 2592 config: GatewayGuardrailConfig 2593 def __init__(self, config: _Optional[_Union[GatewayGuardrailConfig, _Mapping]] = ...) -> None: ... 2594 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2595 GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] 2596 EXECUTION_ORDER_FIELD_NUMBER: _ClassVar[int] 2597 endpoint_id: str 2598 guardrail_id: str 2599 execution_order: int 2600 def __init__(self, endpoint_id: _Optional[str] = ..., guardrail_id: _Optional[str] = ..., execution_order: _Optional[int] = ...) -> None: ... 2601 2602 class RemoveGuardrailFromEndpoint(_message.Message): 2603 __slots__ = ("endpoint_id", "guardrail_id") 2604 class Response(_message.Message): 2605 __slots__ = () 2606 def __init__(self) -> None: ... 2607 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2608 GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] 2609 endpoint_id: str 2610 guardrail_id: str 2611 def __init__(self, endpoint_id: _Optional[str] = ..., guardrail_id: _Optional[str] = ...) -> None: ... 2612 2613 class ListEndpointGuardrailConfigs(_message.Message): 2614 __slots__ = ("endpoint_id",) 2615 class Response(_message.Message): 2616 __slots__ = ("configs",) 2617 CONFIGS_FIELD_NUMBER: _ClassVar[int] 2618 configs: _containers.RepeatedCompositeFieldContainer[GatewayGuardrailConfig] 2619 def __init__(self, configs: _Optional[_Iterable[_Union[GatewayGuardrailConfig, _Mapping]]] = ...) -> None: ... 2620 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2621 endpoint_id: str 2622 def __init__(self, endpoint_id: _Optional[str] = ...) -> None: ... 2623 2624 class UpdateEndpointGuardrailConfig(_message.Message): 2625 __slots__ = ("endpoint_id", "guardrail_id", "execution_order") 2626 class Response(_message.Message): 2627 __slots__ = ("config",) 2628 CONFIG_FIELD_NUMBER: _ClassVar[int] 2629 config: GatewayGuardrailConfig 2630 def __init__(self, config: _Optional[_Union[GatewayGuardrailConfig, _Mapping]] = ...) -> None: ... 2631 ENDPOINT_ID_FIELD_NUMBER: _ClassVar[int] 2632 GUARDRAIL_ID_FIELD_NUMBER: _ClassVar[int] 2633 EXECUTION_ORDER_FIELD_NUMBER: _ClassVar[int] 2634 endpoint_id: str 2635 guardrail_id: str 2636 execution_order: int 2637 def __init__(self, endpoint_id: _Optional[str] = ..., guardrail_id: _Optional[str] = ..., execution_order: _Optional[int] = ...) -> None: ... 2638 2639 class GetSecretsConfig(_message.Message): 2640 __slots__ = () 2641 class Response(_message.Message): 2642 __slots__ = ("secrets_available",) 2643 SECRETS_AVAILABLE_FIELD_NUMBER: _ClassVar[int] 2644 secrets_available: bool 2645 def __init__(self, secrets_available: bool = ...) -> None: ... 2646 def __init__(self) -> None: ... 2647 2648 class CreatePromptOptimizationJob(_message.Message): 2649 __slots__ = ("experiment_id", "source_prompt_uri", "config", "tags") 2650 class Response(_message.Message): 2651 __slots__ = ("job",) 2652 JOB_FIELD_NUMBER: _ClassVar[int] 2653 job: _prompt_optimization_pb2.PromptOptimizationJob 2654 def __init__(self, job: _Optional[_Union[_prompt_optimization_pb2.PromptOptimizationJob, _Mapping]] = ...) -> None: ... 2655 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 2656 SOURCE_PROMPT_URI_FIELD_NUMBER: _ClassVar[int] 2657 CONFIG_FIELD_NUMBER: _ClassVar[int] 2658 TAGS_FIELD_NUMBER: _ClassVar[int] 2659 experiment_id: str 2660 source_prompt_uri: str 2661 config: _prompt_optimization_pb2.PromptOptimizationJobConfig 2662 tags: _containers.RepeatedCompositeFieldContainer[_prompt_optimization_pb2.PromptOptimizationJobTag] 2663 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: ... 2664 2665 class GetPromptOptimizationJob(_message.Message): 2666 __slots__ = ("job_id",) 2667 class Response(_message.Message): 2668 __slots__ = ("job",) 2669 JOB_FIELD_NUMBER: _ClassVar[int] 2670 job: _prompt_optimization_pb2.PromptOptimizationJob 2671 def __init__(self, job: _Optional[_Union[_prompt_optimization_pb2.PromptOptimizationJob, _Mapping]] = ...) -> None: ... 2672 JOB_ID_FIELD_NUMBER: _ClassVar[int] 2673 job_id: str 2674 def __init__(self, job_id: _Optional[str] = ...) -> None: ... 2675 2676 class SearchPromptOptimizationJobs(_message.Message): 2677 __slots__ = ("experiment_id",) 2678 class Response(_message.Message): 2679 __slots__ = ("jobs",) 2680 JOBS_FIELD_NUMBER: _ClassVar[int] 2681 jobs: _containers.RepeatedCompositeFieldContainer[_prompt_optimization_pb2.PromptOptimizationJob] 2682 def __init__(self, jobs: _Optional[_Iterable[_Union[_prompt_optimization_pb2.PromptOptimizationJob, _Mapping]]] = ...) -> None: ... 2683 EXPERIMENT_ID_FIELD_NUMBER: _ClassVar[int] 2684 experiment_id: str 2685 def __init__(self, experiment_id: _Optional[str] = ...) -> None: ... 2686 2687 class CancelPromptOptimizationJob(_message.Message): 2688 __slots__ = ("job_id",) 2689 class Response(_message.Message): 2690 __slots__ = ("job",) 2691 JOB_FIELD_NUMBER: _ClassVar[int] 2692 job: _prompt_optimization_pb2.PromptOptimizationJob 2693 def __init__(self, job: _Optional[_Union[_prompt_optimization_pb2.PromptOptimizationJob, _Mapping]] = ...) -> None: ... 2694 JOB_ID_FIELD_NUMBER: _ClassVar[int] 2695 job_id: str 2696 def __init__(self, job_id: _Optional[str] = ...) -> None: ... 2697 2698 class DeletePromptOptimizationJob(_message.Message): 2699 __slots__ = ("job_id",) 2700 class Response(_message.Message): 2701 __slots__ = () 2702 def __init__(self) -> None: ... 2703 JOB_ID_FIELD_NUMBER: _ClassVar[int] 2704 job_id: str 2705 def __init__(self, job_id: _Optional[str] = ...) -> None: ... 2706 2707 class Workspace(_message.Message): 2708 __slots__ = ("name", "description", "default_artifact_root") 2709 NAME_FIELD_NUMBER: _ClassVar[int] 2710 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 2711 DEFAULT_ARTIFACT_ROOT_FIELD_NUMBER: _ClassVar[int] 2712 name: str 2713 description: str 2714 default_artifact_root: str 2715 def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., default_artifact_root: _Optional[str] = ...) -> None: ... 2716 2717 class ListWorkspaces(_message.Message): 2718 __slots__ = () 2719 class Response(_message.Message): 2720 __slots__ = ("workspaces",) 2721 WORKSPACES_FIELD_NUMBER: _ClassVar[int] 2722 workspaces: _containers.RepeatedCompositeFieldContainer[Workspace] 2723 def __init__(self, workspaces: _Optional[_Iterable[_Union[Workspace, _Mapping]]] = ...) -> None: ... 2724 def __init__(self) -> None: ... 2725 2726 class CreateWorkspace(_message.Message): 2727 __slots__ = ("name", "description", "default_artifact_root") 2728 class Response(_message.Message): 2729 __slots__ = ("workspace",) 2730 WORKSPACE_FIELD_NUMBER: _ClassVar[int] 2731 workspace: Workspace 2732 def __init__(self, workspace: _Optional[_Union[Workspace, _Mapping]] = ...) -> None: ... 2733 NAME_FIELD_NUMBER: _ClassVar[int] 2734 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 2735 DEFAULT_ARTIFACT_ROOT_FIELD_NUMBER: _ClassVar[int] 2736 name: str 2737 description: str 2738 default_artifact_root: str 2739 def __init__(self, name: _Optional[str] = ..., description: _Optional[str] = ..., default_artifact_root: _Optional[str] = ...) -> None: ... 2740 2741 class GetWorkspace(_message.Message): 2742 __slots__ = ("workspace_name",) 2743 class Response(_message.Message): 2744 __slots__ = ("workspace",) 2745 WORKSPACE_FIELD_NUMBER: _ClassVar[int] 2746 workspace: Workspace 2747 def __init__(self, workspace: _Optional[_Union[Workspace, _Mapping]] = ...) -> None: ... 2748 WORKSPACE_NAME_FIELD_NUMBER: _ClassVar[int] 2749 workspace_name: str 2750 def __init__(self, workspace_name: _Optional[str] = ...) -> None: ... 2751 2752 class UpdateWorkspace(_message.Message): 2753 __slots__ = ("workspace_name", "description", "default_artifact_root") 2754 class Response(_message.Message): 2755 __slots__ = ("workspace",) 2756 WORKSPACE_FIELD_NUMBER: _ClassVar[int] 2757 workspace: Workspace 2758 def __init__(self, workspace: _Optional[_Union[Workspace, _Mapping]] = ...) -> None: ... 2759 WORKSPACE_NAME_FIELD_NUMBER: _ClassVar[int] 2760 DESCRIPTION_FIELD_NUMBER: _ClassVar[int] 2761 DEFAULT_ARTIFACT_ROOT_FIELD_NUMBER: _ClassVar[int] 2762 workspace_name: str 2763 description: str 2764 default_artifact_root: str 2765 def __init__(self, workspace_name: _Optional[str] = ..., description: _Optional[str] = ..., default_artifact_root: _Optional[str] = ...) -> None: ... 2766 2767 class DeleteWorkspace(_message.Message): 2768 __slots__ = ("workspace_name",) 2769 class Response(_message.Message): 2770 __slots__ = () 2771 def __init__(self) -> None: ... 2772 WORKSPACE_NAME_FIELD_NUMBER: _ClassVar[int] 2773 workspace_name: str 2774 def __init__(self, workspace_name: _Optional[str] = ...) -> None: ... 2775 2776 class MlflowService(_service.service): ... 2777 2778 class MlflowService_Stub(MlflowService): ...