diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha index 23433148e..716a0f6fb 100644 --- a/.codegen/_openapi_sha +++ b/.codegen/_openapi_sha @@ -1 +1 @@ -596323fdb6eeb4403529e98d557a4d8100238ef3 \ No newline at end of file +0b79be80b388e6f3bac84bdc719bac68c5204f1e \ No newline at end of file diff --git a/.gitattributes b/.gitattributes index 33a5764b7..3b4407940 100644 --- a/.gitattributes +++ b/.gitattributes @@ -2076,6 +2076,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentPermis databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentPermissionsDescription.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentPermissionsRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentTag.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentTraceLocation.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsAPI.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsService.java linguist-generated=true @@ -2294,6 +2295,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/TransitionModelV databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/TransitionRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/TransitionStageResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/TumblingWindow.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/UcTraceLocation.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/UpdateComment.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/UpdateCommentResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/UpdateExperiment.java linguist-generated=true diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md index bf7e3cdaa..4d8f8dd74 100644 --- a/NEXT_CHANGELOG.md +++ b/NEXT_CHANGELOG.md @@ -15,3 +15,8 @@ ### Internal Changes ### API Changes +* Add `stableWorkspaceId` field for `com.databricks.sdk.service.disasterrecovery.StableUrl`. +* Add `deploymentId` and `versionId` fields for `com.databricks.sdk.service.jobs.BaseRun`. +* Add `deploymentId` and `versionId` fields for `com.databricks.sdk.service.jobs.Run`. +* Add `traceLocation` field for `com.databricks.sdk.service.ml.CreateExperiment`. +* Add `traceLocation` field for `com.databricks.sdk.service.ml.Experiment`. \ No newline at end of file diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/disasterrecovery/StableUrl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/disasterrecovery/StableUrl.java index c128cb2b7..0c38f8b40 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/disasterrecovery/StableUrl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/disasterrecovery/StableUrl.java @@ -32,6 +32,14 @@ public class StableUrl { @JsonProperty("name") private String name; + /** + * The stable workspace ID for this stable URL. Generated on creation and immutable thereafter; + * identifies the URL across failovers and is the same value embedded in the `url` (as the `w=` + * query parameter for SPOG URLs, or in the `conn-` hostname for Private-Link URLs). + */ + @JsonProperty("stable_workspace_id") + private String stableWorkspaceId; + /** * The stable URL endpoint. Generated on creation and immutable thereafter. For non-Private-Link * workspaces this is `https:///?w=`. For Private-Link workspaces this @@ -67,6 +75,15 @@ public String getName() { return name; } + public StableUrl setStableWorkspaceId(String stableWorkspaceId) { + this.stableWorkspaceId = stableWorkspaceId; + return this; + } + + public String getStableWorkspaceId() { + return stableWorkspaceId; + } + public StableUrl setUrl(String url) { this.url = url; return this; @@ -84,12 +101,13 @@ public boolean equals(Object o) { return Objects.equals(failoverGroupName, that.failoverGroupName) && Objects.equals(initialWorkspaceId, that.initialWorkspaceId) && Objects.equals(name, that.name) + && Objects.equals(stableWorkspaceId, that.stableWorkspaceId) && Objects.equals(url, that.url); } @Override public int hashCode() { - return Objects.hash(failoverGroupName, initialWorkspaceId, name, url); + return Objects.hash(failoverGroupName, initialWorkspaceId, name, stableWorkspaceId, url); } @Override @@ -98,6 +116,7 @@ public String toString() { .add("failoverGroupName", failoverGroupName) .add("initialWorkspaceId", initialWorkspaceId) .add("name", name) + .add("stableWorkspaceId", stableWorkspaceId) .add("url", url) .toString(); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/BaseRun.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/BaseRun.java index 4d890aad6..8824fbcfb 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/BaseRun.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/BaseRun.java @@ -47,6 +47,14 @@ public class BaseRun { @JsonProperty("creator_user_name") private String creatorUserName; + /** + * ID of the deployment that produced the job when this run was created. Used to look up + * deployment metadata from the Deployment Metadata service. Only set for job runs of jobs with a + * `BUNDLE` deployment. + */ + @JsonProperty("deployment_id") + private String deploymentId; + /** Description of the run */ @JsonProperty("description") private String description; @@ -221,6 +229,14 @@ public class BaseRun { @JsonProperty("trigger_info") private TriggerInfo triggerInfo; + /** + * ID of the deployment version that produced the job when this run was created. Identifies a + * specific snapshot of the deployment in the Deployment Metadata service. Only set for job runs + * of jobs with a `BUNDLE` deployment. + */ + @JsonProperty("version_id") + private String versionId; + public BaseRun setAttemptNumber(Long attemptNumber) { this.attemptNumber = attemptNumber; return this; @@ -266,6 +282,15 @@ public String getCreatorUserName() { return creatorUserName; } + public BaseRun setDeploymentId(String deploymentId) { + this.deploymentId = deploymentId; + return this; + } + + public String getDeploymentId() { + return deploymentId; + } + public BaseRun setDescription(String description) { this.description = description; return this; @@ -527,6 +552,15 @@ public TriggerInfo getTriggerInfo() { return triggerInfo; } + public BaseRun setVersionId(String versionId) { + this.versionId = versionId; + return this; + } + + public String getVersionId() { + return versionId; + } + @Override public boolean equals(Object o) { if (this == o) return true; @@ -537,6 +571,7 @@ public boolean equals(Object o) { && Objects.equals(clusterInstance, that.clusterInstance) && Objects.equals(clusterSpec, that.clusterSpec) && Objects.equals(creatorUserName, that.creatorUserName) + && Objects.equals(deploymentId, that.deploymentId) && Objects.equals(description, that.description) && Objects.equals(effectivePerformanceTarget, that.effectivePerformanceTarget) && Objects.equals(effectiveUsagePolicyId, that.effectiveUsagePolicyId) @@ -565,7 +600,8 @@ public boolean equals(Object o) { && Objects.equals(status, that.status) && Objects.equals(tasks, that.tasks) && Objects.equals(trigger, that.trigger) - && Objects.equals(triggerInfo, that.triggerInfo); + && Objects.equals(triggerInfo, that.triggerInfo) + && Objects.equals(versionId, that.versionId); } @Override @@ -576,6 +612,7 @@ public int hashCode() { clusterInstance, clusterSpec, creatorUserName, + deploymentId, description, effectivePerformanceTarget, effectiveUsagePolicyId, @@ -604,7 +641,8 @@ public int hashCode() { status, tasks, trigger, - triggerInfo); + triggerInfo, + versionId); } @Override @@ -615,6 +653,7 @@ public String toString() { .add("clusterInstance", clusterInstance) .add("clusterSpec", clusterSpec) .add("creatorUserName", creatorUserName) + .add("deploymentId", deploymentId) .add("description", description) .add("effectivePerformanceTarget", effectivePerformanceTarget) .add("effectiveUsagePolicyId", effectiveUsagePolicyId) @@ -644,6 +683,7 @@ public String toString() { .add("tasks", tasks) .add("trigger", trigger) .add("triggerInfo", triggerInfo) + .add("versionId", versionId) .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/Run.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/Run.java index 6258f24fa..619290fae 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/Run.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/Run.java @@ -48,6 +48,14 @@ public class Run { @JsonProperty("creator_user_name") private String creatorUserName; + /** + * ID of the deployment that produced the job when this run was created. Used to look up + * deployment metadata from the Deployment Metadata service. Only set for job runs of jobs with a + * `BUNDLE` deployment. + */ + @JsonProperty("deployment_id") + private String deploymentId; + /** Description of the run */ @JsonProperty("description") private String description; @@ -230,6 +238,14 @@ public class Run { @JsonProperty("trigger_info") private TriggerInfo triggerInfo; + /** + * ID of the deployment version that produced the job when this run was created. Identifies a + * specific snapshot of the deployment in the Deployment Metadata service. Only set for job runs + * of jobs with a `BUNDLE` deployment. + */ + @JsonProperty("version_id") + private String versionId; + public Run setAttemptNumber(Long attemptNumber) { this.attemptNumber = attemptNumber; return this; @@ -275,6 +291,15 @@ public String getCreatorUserName() { return creatorUserName; } + public Run setDeploymentId(String deploymentId) { + this.deploymentId = deploymentId; + return this; + } + + public String getDeploymentId() { + return deploymentId; + } + public Run setDescription(String description) { this.description = description; return this; @@ -554,6 +579,15 @@ public TriggerInfo getTriggerInfo() { return triggerInfo; } + public Run setVersionId(String versionId) { + this.versionId = versionId; + return this; + } + + public String getVersionId() { + return versionId; + } + @Override public boolean equals(Object o) { if (this == o) return true; @@ -564,6 +598,7 @@ public boolean equals(Object o) { && Objects.equals(clusterInstance, that.clusterInstance) && Objects.equals(clusterSpec, that.clusterSpec) && Objects.equals(creatorUserName, that.creatorUserName) + && Objects.equals(deploymentId, that.deploymentId) && Objects.equals(description, that.description) && Objects.equals(effectivePerformanceTarget, that.effectivePerformanceTarget) && Objects.equals(effectiveUsagePolicyId, that.effectiveUsagePolicyId) @@ -594,7 +629,8 @@ public boolean equals(Object o) { && Objects.equals(status, that.status) && Objects.equals(tasks, that.tasks) && Objects.equals(trigger, that.trigger) - && Objects.equals(triggerInfo, that.triggerInfo); + && Objects.equals(triggerInfo, that.triggerInfo) + && Objects.equals(versionId, that.versionId); } @Override @@ -605,6 +641,7 @@ public int hashCode() { clusterInstance, clusterSpec, creatorUserName, + deploymentId, description, effectivePerformanceTarget, effectiveUsagePolicyId, @@ -635,7 +672,8 @@ public int hashCode() { status, tasks, trigger, - triggerInfo); + triggerInfo, + versionId); } @Override @@ -646,6 +684,7 @@ public String toString() { .add("clusterInstance", clusterInstance) .add("clusterSpec", clusterSpec) .add("creatorUserName", creatorUserName) + .add("deploymentId", deploymentId) .add("description", description) .add("effectivePerformanceTarget", effectivePerformanceTarget) .add("effectiveUsagePolicyId", effectiveUsagePolicyId) @@ -677,6 +716,7 @@ public String toString() { .add("tasks", tasks) .add("trigger", trigger) .add("triggerInfo", triggerInfo) + .add("versionId", versionId) .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/CreateExperiment.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/CreateExperiment.java index dfc04e7da..9197ea2ab 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/CreateExperiment.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/CreateExperiment.java @@ -30,6 +30,14 @@ public class CreateExperiment { @JsonProperty("tags") private Collection tags; + /** + * The location where the experiment's traces are stored. When set, the underlying storage is + * provisioned and the experiment's traces are routed to it. When unset, traces are stored in the + * default MLflow backend. This field cannot be updated after the experiment is created. + */ + @JsonProperty("trace_location") + private ExperimentTraceLocation traceLocation; + public CreateExperiment setArtifactLocation(String artifactLocation) { this.artifactLocation = artifactLocation; return this; @@ -57,6 +65,15 @@ public Collection getTags() { return tags; } + public CreateExperiment setTraceLocation(ExperimentTraceLocation traceLocation) { + this.traceLocation = traceLocation; + return this; + } + + public ExperimentTraceLocation getTraceLocation() { + return traceLocation; + } + @Override public boolean equals(Object o) { if (this == o) return true; @@ -64,12 +81,13 @@ public boolean equals(Object o) { CreateExperiment that = (CreateExperiment) o; return Objects.equals(artifactLocation, that.artifactLocation) && Objects.equals(name, that.name) - && Objects.equals(tags, that.tags); + && Objects.equals(tags, that.tags) + && Objects.equals(traceLocation, that.traceLocation); } @Override public int hashCode() { - return Objects.hash(artifactLocation, name, tags); + return Objects.hash(artifactLocation, name, tags, traceLocation); } @Override @@ -78,6 +96,7 @@ public String toString() { .add("artifactLocation", artifactLocation) .add("name", name) .add("tags", tags) + .add("traceLocation", traceLocation) .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/Experiment.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/Experiment.java index b23e8994f..aba3e2783 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/Experiment.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/Experiment.java @@ -42,6 +42,13 @@ public class Experiment { @JsonProperty("tags") private Collection tags; + /** + * The location where the experiment's traces are stored. Unset when traces are stored in the + * default MLflow backend. This field cannot be updated after the experiment is created. + */ + @JsonProperty("trace_location") + private ExperimentTraceLocation traceLocation; + public Experiment setArtifactLocation(String artifactLocation) { this.artifactLocation = artifactLocation; return this; @@ -105,6 +112,15 @@ public Collection getTags() { return tags; } + public Experiment setTraceLocation(ExperimentTraceLocation traceLocation) { + this.traceLocation = traceLocation; + return this; + } + + public ExperimentTraceLocation getTraceLocation() { + return traceLocation; + } + @Override public boolean equals(Object o) { if (this == o) return true; @@ -116,13 +132,21 @@ public boolean equals(Object o) { && Objects.equals(lastUpdateTime, that.lastUpdateTime) && Objects.equals(lifecycleStage, that.lifecycleStage) && Objects.equals(name, that.name) - && Objects.equals(tags, that.tags); + && Objects.equals(tags, that.tags) + && Objects.equals(traceLocation, that.traceLocation); } @Override public int hashCode() { return Objects.hash( - artifactLocation, creationTime, experimentId, lastUpdateTime, lifecycleStage, name, tags); + artifactLocation, + creationTime, + experimentId, + lastUpdateTime, + lifecycleStage, + name, + tags, + traceLocation); } @Override @@ -135,6 +159,7 @@ public String toString() { .add("lifecycleStage", lifecycleStage) .add("name", name) .add("tags", tags) + .add("traceLocation", traceLocation) .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentTraceLocation.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentTraceLocation.java new file mode 100644 index 000000000..7e309fef3 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentTraceLocation.java @@ -0,0 +1,45 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.ml; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +/** The storage location for an experiment's traces. */ +@Generated +public class ExperimentTraceLocation { + /** A Unity Catalog schema where the experiment's traces are stored as Delta tables. */ + @JsonProperty("uc_trace_location") + private UcTraceLocation ucTraceLocation; + + public ExperimentTraceLocation setUcTraceLocation(UcTraceLocation ucTraceLocation) { + this.ucTraceLocation = ucTraceLocation; + return this; + } + + public UcTraceLocation getUcTraceLocation() { + return ucTraceLocation; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + ExperimentTraceLocation that = (ExperimentTraceLocation) o; + return Objects.equals(ucTraceLocation, that.ucTraceLocation); + } + + @Override + public int hashCode() { + return Objects.hash(ucTraceLocation); + } + + @Override + public String toString() { + return new ToStringer(ExperimentTraceLocation.class) + .add("ucTraceLocation", ucTraceLocation) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/UcTraceLocation.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/UcTraceLocation.java new file mode 100644 index 000000000..c88f33589 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/UcTraceLocation.java @@ -0,0 +1,82 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.ml; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +/** + * A Unity Catalog trace storage location. Traces are stored as Delta tables in the specified + * catalog and schema. + */ +@Generated +public class UcTraceLocation { + /** The name of the Unity Catalog catalog. */ + @JsonProperty("catalog") + private String catalog; + + /** The name of the Unity Catalog schema within `catalog`. */ + @JsonProperty("schema") + private String schema; + + /** + * The prefix for the trace tables, which are named `{catalog}.{schema}.{table_prefix}_otel_*`. + * May only contain letters, digits, and underscores, and may be at most 238 characters. When + * unset, a server-generated prefix derived from the experiment ID is used. + */ + @JsonProperty("table_prefix") + private String tablePrefix; + + public UcTraceLocation setCatalog(String catalog) { + this.catalog = catalog; + return this; + } + + public String getCatalog() { + return catalog; + } + + public UcTraceLocation setSchema(String schema) { + this.schema = schema; + return this; + } + + public String getSchema() { + return schema; + } + + public UcTraceLocation setTablePrefix(String tablePrefix) { + this.tablePrefix = tablePrefix; + return this; + } + + public String getTablePrefix() { + return tablePrefix; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + UcTraceLocation that = (UcTraceLocation) o; + return Objects.equals(catalog, that.catalog) + && Objects.equals(schema, that.schema) + && Objects.equals(tablePrefix, that.tablePrefix); + } + + @Override + public int hashCode() { + return Objects.hash(catalog, schema, tablePrefix); + } + + @Override + public String toString() { + return new ToStringer(UcTraceLocation.class) + .add("catalog", catalog) + .add("schema", schema) + .add("tablePrefix", tablePrefix) + .toString(); + } +}