From a7f8e43ae6b9456d65658f54808abb7cc76ac1df Mon Sep 17 00:00:00 2001 From: "databricks-ci-ghec-1[bot]" <184311507+databricks-ci-ghec-1[bot]@users.noreply.github.com> Date: Wed, 22 Jul 2026 04:18:38 +0000 Subject: [PATCH] Update SDK to 3d409cc7565ce0f35a2e8c0ff1e629f34e51106f --- .codegen/_openapi_sha | 2 +- .gitattributes | 1 + NEXT_CHANGELOG.md | 3 + .../iamv2/WorkspaceAssignmentDetail.java | 22 +++- .../sdk/service/jobs/JobCluster.java | 23 +++- .../sdk/service/serving/TelemetryConfig.java | 40 ++++++- .../serving/UnityCatalogTableNames.java | 101 ++++++++++++++++++ tagging.py | 13 ++- 8 files changed, 198 insertions(+), 7 deletions(-) create mode 100644 databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/UnityCatalogTableNames.java diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha index 8d5536c65..d5e2a8aac 100644 --- a/.codegen/_openapi_sha +++ b/.codegen/_openapi_sha @@ -1 +1 @@ -68404f483a3150291f360721997bbc7be7cd2d84 \ No newline at end of file +3d409cc7565ce0f35a2e8c0ff1e629f34e51106f \ No newline at end of file diff --git a/.gitattributes b/.gitattributes index 78feb4920..00d403dda 100644 --- a/.gitattributes +++ b/.gitattributes @@ -2898,6 +2898,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingMode databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/TelemetryConfig.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/TelemetryInferenceTableConfig.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/TrafficConfig.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/UnityCatalogTableNames.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/UpdateInferenceEndpointNotifications.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/UpdateInferenceEndpointNotificationsResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/UpdateProvisionedThroughputEndpointConfigRequest.java linguist-generated=true diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md index 1e76f7cc7..c25ec7126 100644 --- a/NEXT_CHANGELOG.md +++ b/NEXT_CHANGELOG.md @@ -15,3 +15,6 @@ ### Internal Changes ### API Changes +* Add `effectiveEntitlements` field for `com.databricks.sdk.service.iamv2.WorkspaceAssignmentDetail`. +* Add `serverlessComputeId` field for `com.databricks.sdk.service.jobs.JobCluster`. +* Add `tableNames` and `telemetryProfileId` fields for `com.databricks.sdk.service.serving.TelemetryConfig`. \ No newline at end of file diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceAssignmentDetail.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceAssignmentDetail.java index 91084a6ed..a50c29e48 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceAssignmentDetail.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceAssignmentDetail.java @@ -15,6 +15,13 @@ public class WorkspaceAssignmentDetail { @JsonProperty("account_id") private String accountId; + /** + * The principal's full effective entitlements granted in this workspace: every entitlement it + * holds whether granted directly or via group membership. Populated on Get; empty on List. + */ + @JsonProperty("effective_entitlements") + private Collection effectiveEntitlements; + /** * Entitlements granted directly to the principal on this workspace. The only client-settable * field: create and update manage exactly this set (including entitlements the principal also @@ -45,6 +52,16 @@ public String getAccountId() { return accountId; } + public WorkspaceAssignmentDetail setEffectiveEntitlements( + Collection effectiveEntitlements) { + this.effectiveEntitlements = effectiveEntitlements; + return this; + } + + public Collection getEffectiveEntitlements() { + return effectiveEntitlements; + } + public WorkspaceAssignmentDetail setEntitlements(Collection entitlements) { this.entitlements = entitlements; return this; @@ -87,6 +104,7 @@ public boolean equals(Object o) { if (o == null || getClass() != o.getClass()) return false; WorkspaceAssignmentDetail that = (WorkspaceAssignmentDetail) o; return Objects.equals(accountId, that.accountId) + && Objects.equals(effectiveEntitlements, that.effectiveEntitlements) && Objects.equals(entitlements, that.entitlements) && Objects.equals(principalId, that.principalId) && Objects.equals(principalType, that.principalType) @@ -95,13 +113,15 @@ public boolean equals(Object o) { @Override public int hashCode() { - return Objects.hash(accountId, entitlements, principalId, principalType, workspaceId); + return Objects.hash( + accountId, effectiveEntitlements, entitlements, principalId, principalType, workspaceId); } @Override public String toString() { return new ToStringer(WorkspaceAssignmentDetail.class) .add("accountId", accountId) + .add("effectiveEntitlements", effectiveEntitlements) .add("entitlements", entitlements) .add("principalId", principalId) .add("principalType", principalType) diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobCluster.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobCluster.java index 25d519e34..d8087d986 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobCluster.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobCluster.java @@ -21,6 +21,14 @@ public class JobCluster { @JsonProperty("new_cluster") private com.databricks.sdk.service.compute.ClusterSpec newCluster; + /** + * The ID of the serverless compute object to bind this cluster to. At most one JobCluster per job + * may set this field; the rate limit defined on the referenced serverless compute applies across + * all tasks bound to this cluster. + */ + @JsonProperty("serverless_compute_id") + private String serverlessComputeId; + public JobCluster setJobClusterKey(String jobClusterKey) { this.jobClusterKey = jobClusterKey; return this; @@ -39,18 +47,28 @@ public com.databricks.sdk.service.compute.ClusterSpec getNewCluster() { return newCluster; } + public JobCluster setServerlessComputeId(String serverlessComputeId) { + this.serverlessComputeId = serverlessComputeId; + return this; + } + + public String getServerlessComputeId() { + return serverlessComputeId; + } + @Override public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; JobCluster that = (JobCluster) o; return Objects.equals(jobClusterKey, that.jobClusterKey) - && Objects.equals(newCluster, that.newCluster); + && Objects.equals(newCluster, that.newCluster) + && Objects.equals(serverlessComputeId, that.serverlessComputeId); } @Override public int hashCode() { - return Objects.hash(jobClusterKey, newCluster); + return Objects.hash(jobClusterKey, newCluster, serverlessComputeId); } @Override @@ -58,6 +76,7 @@ public String toString() { return new ToStringer(JobCluster.class) .add("jobClusterKey", jobClusterKey) .add("newCluster", newCluster) + .add("serverlessComputeId", serverlessComputeId) .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/TelemetryConfig.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/TelemetryConfig.java index bc3288bbd..4743a1446 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/TelemetryConfig.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/TelemetryConfig.java @@ -13,6 +13,20 @@ public class TelemetryConfig { @JsonProperty("inference_table_config") private TelemetryInferenceTableConfig inferenceTableConfig; + /** + * The Unity Catalog tables to which endpoint telemetry (logs, traces, and metrics) is exported. + * Provide this to create a new telemetry profile for the endpoint from the given tables. + */ + @JsonProperty("table_names") + private UnityCatalogTableNames tableNames; + + /** + * The ID of an existing telemetry profile to apply to this endpoint. Provide this to reuse a + * telemetry profile that has already been created, instead of specifying table_names. + */ + @JsonProperty("telemetry_profile_id") + private String telemetryProfileId; + public TelemetryConfig setInferenceTableConfig( TelemetryInferenceTableConfig inferenceTableConfig) { this.inferenceTableConfig = inferenceTableConfig; @@ -23,23 +37,45 @@ public TelemetryInferenceTableConfig getInferenceTableConfig() { return inferenceTableConfig; } + public TelemetryConfig setTableNames(UnityCatalogTableNames tableNames) { + this.tableNames = tableNames; + return this; + } + + public UnityCatalogTableNames getTableNames() { + return tableNames; + } + + public TelemetryConfig setTelemetryProfileId(String telemetryProfileId) { + this.telemetryProfileId = telemetryProfileId; + return this; + } + + public String getTelemetryProfileId() { + return telemetryProfileId; + } + @Override public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; TelemetryConfig that = (TelemetryConfig) o; - return Objects.equals(inferenceTableConfig, that.inferenceTableConfig); + return Objects.equals(inferenceTableConfig, that.inferenceTableConfig) + && Objects.equals(tableNames, that.tableNames) + && Objects.equals(telemetryProfileId, that.telemetryProfileId); } @Override public int hashCode() { - return Objects.hash(inferenceTableConfig); + return Objects.hash(inferenceTableConfig, tableNames, telemetryProfileId); } @Override public String toString() { return new ToStringer(TelemetryConfig.class) .add("inferenceTableConfig", inferenceTableConfig) + .add("tableNames", tableNames) + .add("telemetryProfileId", telemetryProfileId) .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/UnityCatalogTableNames.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/UnityCatalogTableNames.java new file mode 100644 index 000000000..5a74ce089 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/UnityCatalogTableNames.java @@ -0,0 +1,101 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.serving; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class UnityCatalogTableNames { + /** + * The full three-level Unity Catalog name (catalog.schema.table) of the table that receives + * exported annotations. + */ + @JsonProperty("annotations_table") + private String annotationsTable; + + /** + * The full three-level Unity Catalog name (catalog.schema.table) of the table that receives + * exported logs. + */ + @JsonProperty("logs_table") + private String logsTable; + + /** + * The full three-level Unity Catalog name (catalog.schema.table) of the table that receives + * exported metrics. + */ + @JsonProperty("metrics_table") + private String metricsTable; + + /** + * The full three-level Unity Catalog name (catalog.schema.table) of the table that receives + * exported traces (spans). + */ + @JsonProperty("traces_table") + private String tracesTable; + + public UnityCatalogTableNames setAnnotationsTable(String annotationsTable) { + this.annotationsTable = annotationsTable; + return this; + } + + public String getAnnotationsTable() { + return annotationsTable; + } + + public UnityCatalogTableNames setLogsTable(String logsTable) { + this.logsTable = logsTable; + return this; + } + + public String getLogsTable() { + return logsTable; + } + + public UnityCatalogTableNames setMetricsTable(String metricsTable) { + this.metricsTable = metricsTable; + return this; + } + + public String getMetricsTable() { + return metricsTable; + } + + public UnityCatalogTableNames setTracesTable(String tracesTable) { + this.tracesTable = tracesTable; + return this; + } + + public String getTracesTable() { + return tracesTable; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + UnityCatalogTableNames that = (UnityCatalogTableNames) o; + return Objects.equals(annotationsTable, that.annotationsTable) + && Objects.equals(logsTable, that.logsTable) + && Objects.equals(metricsTable, that.metricsTable) + && Objects.equals(tracesTable, that.tracesTable); + } + + @Override + public int hashCode() { + return Objects.hash(annotationsTable, logsTable, metricsTable, tracesTable); + } + + @Override + public String toString() { + return new ToStringer(UnityCatalogTableNames.class) + .add("annotationsTable", annotationsTable) + .add("logsTable", logsTable) + .add("metricsTable", metricsTable) + .add("tracesTable", tracesTable) + .toString(); + } +} diff --git a/tagging.py b/tagging.py index 7c1ad5f45..3a75f3728 100644 --- a/tagging.py +++ b/tagging.py @@ -184,6 +184,13 @@ def add_file(self, loc: str, content: str): element = InputGitTreeElement(path=local_path, mode="100644", type="blob", sha=blob.sha) self.changed_files.append(element) + # Replaces "git rm file" + def delete_file(self, loc: str): + """``git rm`` equivalent for GitHubRepo: stage a tree deletion (sha=None).""" + local_path = os.path.relpath(loc, os.getcwd()) + print(f"Deleting file {local_path}") + self.changed_files.append(InputGitTreeElement(path=local_path, mode="100644", type="blob", sha=None)) + # Replaces "git commit && git push" def commit_and_push(self, message: str): head_ref = self.repo.get_git_ref(self.ref) @@ -1069,7 +1076,11 @@ def validate_git_root(): raise Exception("Please run this script from the root of the repository.") -if __name__ == "__main__": +def main(): validate_git_root() init_github() process() + + +if __name__ == "__main__": + main()