autogenerated_graphql_schema.py
1 # GENERATED FILE. PLEASE DON'T MODIFY. 2 # Run uv run ./dev/proto_to_graphql/code_generator.py to regenerate. 3 import graphene 4 import mlflow 5 from mlflow.server.graphql.graphql_custom_scalars import LongString 6 from mlflow.server.graphql.graphql_errors import ApiError 7 from mlflow.utils.proto_json_utils import parse_dict 8 9 10 class MlflowDeploymentJobConnectionState(graphene.Enum): 11 DEPLOYMENT_JOB_CONNECTION_STATE_UNSPECIFIED = 1 12 NOT_SET_UP = 2 13 CONNECTED = 3 14 NOT_FOUND = 4 15 REQUIRED_PARAMETERS_CHANGED = 5 16 17 18 class MlflowModelVersionDeploymentJobStateDeploymentJobRunState(graphene.Enum): 19 DEPLOYMENT_JOB_RUN_STATE_UNSPECIFIED = 1 20 NO_VALID_DEPLOYMENT_JOB_FOUND = 2 21 RUNNING = 3 22 SUCCEEDED = 4 23 FAILED = 5 24 PENDING = 6 25 APPROVAL = 7 26 27 28 class MlflowModelVersionStatus(graphene.Enum): 29 PENDING_REGISTRATION = 1 30 FAILED_REGISTRATION = 2 31 READY = 3 32 33 34 class MlflowRunStatus(graphene.Enum): 35 RUNNING = 1 36 SCHEDULED = 2 37 FINISHED = 3 38 FAILED = 4 39 KILLED = 5 40 41 42 class MlflowViewType(graphene.Enum): 43 ACTIVE_ONLY = 1 44 DELETED_ONLY = 2 45 ALL = 3 46 47 48 class MlflowModelVersionDeploymentJobState(graphene.ObjectType): 49 job_id = graphene.String() 50 run_id = graphene.String() 51 job_state = graphene.Field(MlflowDeploymentJobConnectionState) 52 run_state = graphene.Field(MlflowModelVersionDeploymentJobStateDeploymentJobRunState) 53 current_task_name = graphene.String() 54 55 56 class MlflowModelMetric(graphene.ObjectType): 57 key = graphene.String() 58 value = graphene.Float() 59 timestamp = LongString() 60 step = LongString() 61 dataset_name = graphene.String() 62 dataset_digest = graphene.String() 63 model_id = graphene.String() 64 run_id = graphene.String() 65 66 67 class MlflowModelParam(graphene.ObjectType): 68 name = graphene.String() 69 value = graphene.String() 70 71 72 class MlflowModelVersionTag(graphene.ObjectType): 73 key = graphene.String() 74 value = graphene.String() 75 76 77 class MlflowModelVersion(graphene.ObjectType): 78 name = graphene.String() 79 version = graphene.String() 80 creation_timestamp = LongString() 81 last_updated_timestamp = LongString() 82 user_id = graphene.String() 83 current_stage = graphene.String() 84 description = graphene.String() 85 source = graphene.String() 86 run_id = graphene.String() 87 status = graphene.Field(MlflowModelVersionStatus) 88 status_message = graphene.String() 89 tags = graphene.List(graphene.NonNull(MlflowModelVersionTag)) 90 run_link = graphene.String() 91 aliases = graphene.List(graphene.String) 92 model_id = graphene.String() 93 model_params = graphene.List(graphene.NonNull(MlflowModelParam)) 94 model_metrics = graphene.List(graphene.NonNull(MlflowModelMetric)) 95 deployment_job_state = graphene.Field(MlflowModelVersionDeploymentJobState) 96 97 98 class MlflowSearchModelVersionsResponse(graphene.ObjectType): 99 model_versions = graphene.List(graphene.NonNull(MlflowModelVersion)) 100 next_page_token = graphene.String() 101 apiError = graphene.Field(ApiError) 102 103 104 class MlflowDatasetSummary(graphene.ObjectType): 105 experiment_id = graphene.String() 106 name = graphene.String() 107 digest = graphene.String() 108 context = graphene.String() 109 110 111 class MlflowSearchDatasetsResponse(graphene.ObjectType): 112 dataset_summaries = graphene.List(graphene.NonNull(MlflowDatasetSummary)) 113 apiError = graphene.Field(ApiError) 114 115 116 class MlflowMetricWithRunId(graphene.ObjectType): 117 key = graphene.String() 118 value = graphene.Float() 119 timestamp = LongString() 120 step = LongString() 121 run_id = graphene.String() 122 123 124 class MlflowGetMetricHistoryBulkIntervalResponse(graphene.ObjectType): 125 metrics = graphene.List(graphene.NonNull(MlflowMetricWithRunId)) 126 apiError = graphene.Field(ApiError) 127 128 129 class MlflowFileInfo(graphene.ObjectType): 130 path = graphene.String() 131 is_dir = graphene.Boolean() 132 file_size = LongString() 133 134 135 class MlflowListArtifactsResponse(graphene.ObjectType): 136 root_uri = graphene.String() 137 files = graphene.List(graphene.NonNull(MlflowFileInfo)) 138 next_page_token = graphene.String() 139 apiError = graphene.Field(ApiError) 140 141 142 class MlflowModelOutput(graphene.ObjectType): 143 model_id = graphene.String() 144 step = LongString() 145 146 147 class MlflowRunOutputs(graphene.ObjectType): 148 model_outputs = graphene.List(graphene.NonNull(MlflowModelOutput)) 149 150 151 class MlflowModelInput(graphene.ObjectType): 152 model_id = graphene.String() 153 154 155 class MlflowDataset(graphene.ObjectType): 156 name = graphene.String() 157 digest = graphene.String() 158 source_type = graphene.String() 159 source = graphene.String() 160 schema = graphene.String() 161 profile = graphene.String() 162 163 164 class MlflowInputTag(graphene.ObjectType): 165 key = graphene.String() 166 value = graphene.String() 167 168 169 class MlflowDatasetInput(graphene.ObjectType): 170 tags = graphene.List(graphene.NonNull(MlflowInputTag)) 171 dataset = graphene.Field(MlflowDataset) 172 173 174 class MlflowRunInputs(graphene.ObjectType): 175 dataset_inputs = graphene.List(graphene.NonNull(MlflowDatasetInput)) 176 model_inputs = graphene.List(graphene.NonNull(MlflowModelInput)) 177 178 179 class MlflowRunTag(graphene.ObjectType): 180 key = graphene.String() 181 value = graphene.String() 182 183 184 class MlflowParam(graphene.ObjectType): 185 key = graphene.String() 186 value = graphene.String() 187 188 189 class MlflowMetric(graphene.ObjectType): 190 key = graphene.String() 191 value = graphene.Float() 192 timestamp = LongString() 193 step = LongString() 194 dataset_name = graphene.String() 195 dataset_digest = graphene.String() 196 model_id = graphene.String() 197 run_id = graphene.String() 198 199 200 class MlflowRunData(graphene.ObjectType): 201 metrics = graphene.List(graphene.NonNull('mlflow.server.graphql.graphql_schema_extensions.MlflowMetricExtension')) 202 params = graphene.List(graphene.NonNull(MlflowParam)) 203 tags = graphene.List(graphene.NonNull(MlflowRunTag)) 204 205 206 class MlflowRunInfo(graphene.ObjectType): 207 run_id = graphene.String() 208 run_uuid = graphene.String() 209 run_name = graphene.String() 210 experiment_id = graphene.String() 211 user_id = graphene.String() 212 status = graphene.Field(MlflowRunStatus) 213 start_time = LongString() 214 end_time = LongString() 215 artifact_uri = graphene.String() 216 lifecycle_stage = graphene.String() 217 218 219 class MlflowRun(graphene.ObjectType): 220 info = graphene.Field(MlflowRunInfo) 221 data = graphene.Field(MlflowRunData) 222 inputs = graphene.Field(MlflowRunInputs) 223 outputs = graphene.Field(MlflowRunOutputs) 224 225 226 class MlflowSearchRunsResponse(graphene.ObjectType): 227 runs = graphene.List(graphene.NonNull('mlflow.server.graphql.graphql_schema_extensions.MlflowRunExtension')) 228 next_page_token = graphene.String() 229 apiError = graphene.Field(ApiError) 230 231 232 class MlflowGetRunResponse(graphene.ObjectType): 233 run = graphene.Field('mlflow.server.graphql.graphql_schema_extensions.MlflowRunExtension') 234 apiError = graphene.Field(ApiError) 235 236 237 class MlflowExperimentTag(graphene.ObjectType): 238 key = graphene.String() 239 value = graphene.String() 240 241 242 class MlflowExperiment(graphene.ObjectType): 243 experiment_id = graphene.String() 244 name = graphene.String() 245 artifact_location = graphene.String() 246 lifecycle_stage = graphene.String() 247 last_update_time = LongString() 248 creation_time = LongString() 249 tags = graphene.List(graphene.NonNull(MlflowExperimentTag)) 250 251 252 class MlflowGetExperimentResponse(graphene.ObjectType): 253 experiment = graphene.Field(MlflowExperiment) 254 apiError = graphene.Field(ApiError) 255 256 257 class MlflowSearchModelVersionsInput(graphene.InputObjectType): 258 filter = graphene.String() 259 max_results = LongString() 260 order_by = graphene.List(graphene.String) 261 page_token = graphene.String() 262 263 264 class MlflowSearchDatasetsInput(graphene.InputObjectType): 265 experiment_ids = graphene.List(graphene.String) 266 267 268 class MlflowGetMetricHistoryBulkIntervalInput(graphene.InputObjectType): 269 run_ids = graphene.List(graphene.String) 270 metric_key = graphene.String() 271 start_step = graphene.Int() 272 end_step = graphene.Int() 273 max_results = graphene.Int() 274 275 276 class MlflowListArtifactsInput(graphene.InputObjectType): 277 run_id = graphene.String() 278 run_uuid = graphene.String() 279 path = graphene.String() 280 page_token = graphene.String() 281 282 283 class MlflowSearchRunsInput(graphene.InputObjectType): 284 experiment_ids = graphene.List(graphene.String) 285 filter = graphene.String() 286 run_view_type = graphene.Field(MlflowViewType) 287 max_results = graphene.Int() 288 order_by = graphene.List(graphene.String) 289 page_token = graphene.String() 290 291 292 class MlflowGetRunInput(graphene.InputObjectType): 293 run_id = graphene.String() 294 run_uuid = graphene.String() 295 296 297 class MlflowGetExperimentInput(graphene.InputObjectType): 298 experiment_id = graphene.String() 299 300 301 class QueryType(graphene.ObjectType): 302 mlflow_get_experiment = graphene.Field(MlflowGetExperimentResponse, input=MlflowGetExperimentInput()) 303 mlflow_get_metric_history_bulk_interval = graphene.Field(MlflowGetMetricHistoryBulkIntervalResponse, input=MlflowGetMetricHistoryBulkIntervalInput()) 304 mlflow_get_run = graphene.Field(MlflowGetRunResponse, input=MlflowGetRunInput()) 305 mlflow_list_artifacts = graphene.Field(MlflowListArtifactsResponse, input=MlflowListArtifactsInput()) 306 mlflow_search_model_versions = graphene.Field(MlflowSearchModelVersionsResponse, input=MlflowSearchModelVersionsInput()) 307 308 def resolve_mlflow_get_experiment(self, info, input): 309 input_dict = vars(input) 310 request_message = mlflow.protos.service_pb2.GetExperiment() 311 parse_dict(input_dict, request_message) 312 return mlflow.server.handlers.get_experiment_impl(request_message) 313 314 def resolve_mlflow_get_metric_history_bulk_interval(self, info, input): 315 input_dict = vars(input) 316 request_message = mlflow.protos.service_pb2.GetMetricHistoryBulkInterval() 317 parse_dict(input_dict, request_message) 318 return mlflow.server.handlers.get_metric_history_bulk_interval_impl(request_message) 319 320 def resolve_mlflow_get_run(self, info, input): 321 input_dict = vars(input) 322 request_message = mlflow.protos.service_pb2.GetRun() 323 parse_dict(input_dict, request_message) 324 return mlflow.server.handlers.get_run_impl(request_message) 325 326 def resolve_mlflow_list_artifacts(self, info, input): 327 input_dict = vars(input) 328 request_message = mlflow.protos.service_pb2.ListArtifacts() 329 parse_dict(input_dict, request_message) 330 return mlflow.server.handlers.list_artifacts_impl(request_message) 331 332 def resolve_mlflow_search_model_versions(self, info, input): 333 input_dict = vars(input) 334 request_message = mlflow.protos.model_registry_pb2.SearchModelVersions() 335 parse_dict(input_dict, request_message) 336 return mlflow.server.handlers.search_model_versions_impl(request_message) 337 338 339 class MutationType(graphene.ObjectType): 340 mlflow_search_datasets = graphene.Field(MlflowSearchDatasetsResponse, input=MlflowSearchDatasetsInput()) 341 mlflow_search_runs = graphene.Field(MlflowSearchRunsResponse, input=MlflowSearchRunsInput()) 342 343 def resolve_mlflow_search_datasets(self, info, input): 344 input_dict = vars(input) 345 request_message = mlflow.protos.service_pb2.SearchDatasets() 346 parse_dict(input_dict, request_message) 347 return mlflow.server.handlers.search_datasets_impl(request_message) 348 349 def resolve_mlflow_search_runs(self, info, input): 350 input_dict = vars(input) 351 request_message = mlflow.protos.service_pb2.SearchRuns() 352 parse_dict(input_dict, request_message) 353 return mlflow.server.handlers.search_runs_impl(request_message)