From 49f0e1d0e4f15c98051a6d03f0666ef1973c93c9 Mon Sep 17 00:00:00 2001 From: Tanmay Rustagi Date: Tue, 5 Aug 2025 15:21:16 +0200 Subject: [PATCH] Update SDK to latest OpenAPI spec --- .codegen/_openapi_sha | 2 +- .gitattributes | 7 +- NEXT_CHANGELOG.md | 11 ++ .../com/databricks/sdk/service/apps/App.java | 10 +- .../sdk/service/dashboards/Result.java | 19 ++- .../service/database/SyncedDatabaseTable.java | 39 +++++- .../sdk/service/jobs/TableState.java | 62 +++++++++ .../sdk/service/jobs/TableTriggerState.java | 60 ++++++++ .../sdk/service/jobs/TriggerStateProto.java | 22 ++- .../marketplace/ConsumerProvidersAPI.java | 2 +- .../marketplace/ConsumerProvidersImpl.java | 2 +- .../marketplace/ConsumerProvidersService.java | 2 +- .../ListConsumerProvidersRequest.java | 78 +++++++++++ .../marketplace/ListProvidersRequest.java | 21 +-- .../CreatePrivateAccessSettingsRequest.java | 131 ++++++++++++++++++ .../provisioning/PrivateAccessAPI.java | 4 +- .../provisioning/PrivateAccessImpl.java | 4 +- .../provisioning/PrivateAccessService.java | 4 +- ... ReplacePrivateAccessSettingsRequest.java} | 18 +-- .../serving/CreatePtEndpointRequest.java | 17 ++- .../serving/CreateServingEndpoint.java | 25 +++- .../service/serving/DataframeSplitInput.java | 6 +- .../service/serving/EmailNotifications.java | 66 +++++++++ .../EmbeddingsV1ResponseEmbeddingElement.java | 2 +- .../service/serving/QueryEndpointInput.java | 7 +- .../serving/ServedModelStateDeployment.java | 20 +-- .../serving/ServingEndpointDetailed.java | 16 +++ .../service/serving/ServingEndpointsAPI.java | 2 +- .../serving/ServingEndpointsDataPlaneAPI.java | 2 +- .../ServingEndpointsDataPlaneService.java | 2 +- .../serving/ServingEndpointsService.java | 2 +- 31 files changed, 591 insertions(+), 74 deletions(-) create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableState.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableTriggerState.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListConsumerProvidersRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreatePrivateAccessSettingsRequest.java rename databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/{UpsertPrivateAccessSettingsRequest.java => ReplacePrivateAccessSettingsRequest.java} (87%) create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmailNotifications.java diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha index 4d77e7183..a68714d08 100644 --- a/.codegen/_openapi_sha +++ b/.codegen/_openapi_sha @@ -1 +1 @@ -7de38b0552c78117c01aab884acd9b899a9f4d7f \ No newline at end of file +3a7fe4deb693ca98d89b044116aaf008efd895a5 \ No newline at end of file diff --git a/.gitattributes b/.gitattributes index 4d1e7b244..e0c0733ba 100755 --- a/.gitattributes +++ b/.gitattributes @@ -1285,6 +1285,8 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/SubmitRunRespo databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/SubmitTask.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/Subscription.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/SubscriptionSubscriber.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableState.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableTriggerState.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableUpdateTriggerConfiguration.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/Task.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TaskDependency.java linguist-generated=true @@ -1383,6 +1385,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListAll databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListAllInstallationsResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListAllPersonalizationRequestsRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListAllPersonalizationRequestsResponse.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListConsumerProvidersRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListExchangeFiltersRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListExchangeFiltersResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListExchangesForListingRequest.java linguist-generated=true @@ -1836,6 +1839,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/Create databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreateCustomerManagedKeyRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreateGcpKeyInfo.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreateNetworkRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreatePrivateAccessSettingsRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreateStorageConfigurationRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreateVpcEndpointRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreateWorkspaceRequest.java linguist-generated=true @@ -1885,6 +1889,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/Privat databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessLevel.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessService.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessSettings.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/ReplacePrivateAccessSettingsRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/RootBucketInfo.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/StorageAPI.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/StorageConfiguration.java linguist-generated=true @@ -1892,7 +1897,6 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/Storag databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/StorageService.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/StsRole.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/UpdateWorkspaceRequest.java linguist-generated=true -databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/UpsertPrivateAccessSettingsRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/VpcEndpoint.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/VpcEndpointsAPI.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/VpcEndpointsImpl.java linguist-generated=true @@ -1947,6 +1951,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/DataPlaneIn databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/DatabricksModelServingConfig.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/DataframeSplitInput.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/DeleteServingEndpointRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmailNotifications.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmbeddingsV1ResponseEmbeddingElement.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmbeddingsV1ResponseEmbeddingElementObject.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EndpointCoreConfigInput.java linguist-generated=true diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md index 277d098b5..71838392a 100644 --- a/NEXT_CHANGELOG.md +++ b/NEXT_CHANGELOG.md @@ -11,3 +11,14 @@ ### Internal Changes ### API Changes +* Added `statementIdSignature` field for `com.databricks.sdk.service.dashboards.Result`. +* Added `effectiveDatabaseInstanceName` and `effectiveLogicalDatabaseName` fields for `com.databricks.sdk.service.database.SyncedDatabaseTable`. +* Added `table` field for `com.databricks.sdk.service.jobs.TriggerStateProto`. +* Added `emailNotifications` field for `com.databricks.sdk.service.serving.CreatePtEndpointRequest`. +* Added `emailNotifications` field for `com.databricks.sdk.service.serving.CreateServingEndpoint`. +* Added `emailNotifications` field for `com.databricks.sdk.service.serving.ServingEndpointDetailed`. +* [Breaking] Changed `list()` method for `workspaceClient.consumerProviders()` service . New request type is `com.databricks.sdk.service.marketplace.ListConsumerProvidersRequest` class. +* [Breaking] Changed `create()` method for `accountClient.privateAccess()` service . New request type is `com.databricks.sdk.service.provisioning.CreatePrivateAccessSettingsRequest` class. +* [Breaking] Changed `create()` method for `accountClient.privateAccess()` service with new required argument order. +* [Breaking] Changed `replace()` method for `accountClient.privateAccess()` service . New request type is `com.databricks.sdk.service.provisioning.ReplacePrivateAccessSettingsRequest` class. +* [Breaking] Removed `isFeatured` field for `com.databricks.sdk.service.marketplace.ListProvidersRequest`. \ No newline at end of file diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/apps/App.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/apps/App.java index 97b6f3b19..1cab911c6 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/apps/App.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/apps/App.java @@ -21,7 +21,10 @@ public class App { @JsonProperty("app_status") private ApplicationStatus appStatus; - /** */ + /** + * TODO: Deprecate this field after serverless entitlements are released to all prod stages and + * the new usage_policy_id is properly populated and used. + */ @JsonProperty("budget_policy_id") private String budgetPolicyId; @@ -48,7 +51,10 @@ public class App { @JsonProperty("description") private String description; - /** */ + /** + * TODO: Deprecate this field after serverless entitlements are released to all prod stages and + * the new usage_policy_id is properly populated and used. + */ @JsonProperty("effective_budget_policy_id") private String effectiveBudgetPolicyId; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/dashboards/Result.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/dashboards/Result.java index de3b6451a..ab09e632d 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/dashboards/Result.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/dashboards/Result.java @@ -24,6 +24,10 @@ public class Result { @JsonProperty("statement_id") private String statementId; + /** JWT corresponding to the statement contained in this result */ + @JsonProperty("statement_id_signature") + private String statementIdSignature; + public Result setIsTruncated(Boolean isTruncated) { this.isTruncated = isTruncated; return this; @@ -51,6 +55,15 @@ public String getStatementId() { return statementId; } + public Result setStatementIdSignature(String statementIdSignature) { + this.statementIdSignature = statementIdSignature; + return this; + } + + public String getStatementIdSignature() { + return statementIdSignature; + } + @Override public boolean equals(Object o) { if (this == o) return true; @@ -58,12 +71,13 @@ public boolean equals(Object o) { Result that = (Result) o; return Objects.equals(isTruncated, that.isTruncated) && Objects.equals(rowCount, that.rowCount) - && Objects.equals(statementId, that.statementId); + && Objects.equals(statementId, that.statementId) + && Objects.equals(statementIdSignature, that.statementIdSignature); } @Override public int hashCode() { - return Objects.hash(isTruncated, rowCount, statementId); + return Objects.hash(isTruncated, rowCount, statementId, statementIdSignature); } @Override @@ -72,6 +86,7 @@ public String toString() { .add("isTruncated", isTruncated) .add("rowCount", rowCount) .add("statementId", statementId) + .add("statementIdSignature", statementIdSignature) .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/database/SyncedDatabaseTable.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/database/SyncedDatabaseTable.java index dd1ff0837..090724e72 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/database/SyncedDatabaseTable.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/database/SyncedDatabaseTable.java @@ -7,7 +7,7 @@ import com.fasterxml.jackson.annotation.JsonProperty; import java.util.Objects; -/** Next field marker: 12 */ +/** Next field marker: 14 */ @Generated public class SyncedDatabaseTable { /** Synced Table data synchronization status */ @@ -24,6 +24,18 @@ public class SyncedDatabaseTable { @JsonProperty("database_instance_name") private String databaseInstanceName; + /** + * The name of the database instance that this table is registered to. This field is always + * returned, and for tables inside database catalogs is inferred database instance associated with + * the catalog. + */ + @JsonProperty("effective_database_instance_name") + private String effectiveDatabaseInstanceName; + + /** The name of the logical database that this table is registered to. */ + @JsonProperty("effective_logical_database_name") + private String effectiveLogicalDatabaseName; + /** * Target Postgres database object (logical database) name for this table. * @@ -74,6 +86,25 @@ public String getDatabaseInstanceName() { return databaseInstanceName; } + public SyncedDatabaseTable setEffectiveDatabaseInstanceName( + String effectiveDatabaseInstanceName) { + this.effectiveDatabaseInstanceName = effectiveDatabaseInstanceName; + return this; + } + + public String getEffectiveDatabaseInstanceName() { + return effectiveDatabaseInstanceName; + } + + public SyncedDatabaseTable setEffectiveLogicalDatabaseName(String effectiveLogicalDatabaseName) { + this.effectiveLogicalDatabaseName = effectiveLogicalDatabaseName; + return this; + } + + public String getEffectiveLogicalDatabaseName() { + return effectiveLogicalDatabaseName; + } + public SyncedDatabaseTable setLogicalDatabaseName(String logicalDatabaseName) { this.logicalDatabaseName = logicalDatabaseName; return this; @@ -118,6 +149,8 @@ public boolean equals(Object o) { SyncedDatabaseTable that = (SyncedDatabaseTable) o; return Objects.equals(dataSynchronizationStatus, that.dataSynchronizationStatus) && Objects.equals(databaseInstanceName, that.databaseInstanceName) + && Objects.equals(effectiveDatabaseInstanceName, that.effectiveDatabaseInstanceName) + && Objects.equals(effectiveLogicalDatabaseName, that.effectiveLogicalDatabaseName) && Objects.equals(logicalDatabaseName, that.logicalDatabaseName) && Objects.equals(name, that.name) && Objects.equals(spec, that.spec) @@ -129,6 +162,8 @@ public int hashCode() { return Objects.hash( dataSynchronizationStatus, databaseInstanceName, + effectiveDatabaseInstanceName, + effectiveLogicalDatabaseName, logicalDatabaseName, name, spec, @@ -140,6 +175,8 @@ public String toString() { return new ToStringer(SyncedDatabaseTable.class) .add("dataSynchronizationStatus", dataSynchronizationStatus) .add("databaseInstanceName", databaseInstanceName) + .add("effectiveDatabaseInstanceName", effectiveDatabaseInstanceName) + .add("effectiveLogicalDatabaseName", effectiveLogicalDatabaseName) .add("logicalDatabaseName", logicalDatabaseName) .add("name", name) .add("spec", spec) diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableState.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableState.java new file mode 100755 index 000000000..bc1d76179 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableState.java @@ -0,0 +1,62 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.jobs; + +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 TableState { + /** + * Whether or not the table has seen updates since either the creation of the trigger or the last + * successful evaluation of the trigger + */ + @JsonProperty("has_seen_updates") + private Boolean hasSeenUpdates; + + /** Full table name of the table to monitor, e.g. `mycatalog.myschema.mytable` */ + @JsonProperty("table_name") + private String tableName; + + public TableState setHasSeenUpdates(Boolean hasSeenUpdates) { + this.hasSeenUpdates = hasSeenUpdates; + return this; + } + + public Boolean getHasSeenUpdates() { + return hasSeenUpdates; + } + + public TableState setTableName(String tableName) { + this.tableName = tableName; + return this; + } + + public String getTableName() { + return tableName; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + TableState that = (TableState) o; + return Objects.equals(hasSeenUpdates, that.hasSeenUpdates) + && Objects.equals(tableName, that.tableName); + } + + @Override + public int hashCode() { + return Objects.hash(hasSeenUpdates, tableName); + } + + @Override + public String toString() { + return new ToStringer(TableState.class) + .add("hasSeenUpdates", hasSeenUpdates) + .add("tableName", tableName) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableTriggerState.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableTriggerState.java new file mode 100755 index 000000000..871c96f44 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TableTriggerState.java @@ -0,0 +1,60 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.jobs; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Collection; +import java.util.Objects; + +@Generated +public class TableTriggerState { + /** */ + @JsonProperty("last_seen_table_states") + private Collection lastSeenTableStates; + + /** Indicates whether the trigger is using scalable monitoring. */ + @JsonProperty("using_scalable_monitoring") + private Boolean usingScalableMonitoring; + + public TableTriggerState setLastSeenTableStates(Collection lastSeenTableStates) { + this.lastSeenTableStates = lastSeenTableStates; + return this; + } + + public Collection getLastSeenTableStates() { + return lastSeenTableStates; + } + + public TableTriggerState setUsingScalableMonitoring(Boolean usingScalableMonitoring) { + this.usingScalableMonitoring = usingScalableMonitoring; + return this; + } + + public Boolean getUsingScalableMonitoring() { + return usingScalableMonitoring; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + TableTriggerState that = (TableTriggerState) o; + return Objects.equals(lastSeenTableStates, that.lastSeenTableStates) + && Objects.equals(usingScalableMonitoring, that.usingScalableMonitoring); + } + + @Override + public int hashCode() { + return Objects.hash(lastSeenTableStates, usingScalableMonitoring); + } + + @Override + public String toString() { + return new ToStringer(TableTriggerState.class) + .add("lastSeenTableStates", lastSeenTableStates) + .add("usingScalableMonitoring", usingScalableMonitoring) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TriggerStateProto.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TriggerStateProto.java index d95eac4f2..bd9d24f5b 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TriggerStateProto.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/TriggerStateProto.java @@ -13,6 +13,10 @@ public class TriggerStateProto { @JsonProperty("file_arrival") private FileArrivalTriggerState fileArrival; + /** */ + @JsonProperty("table") + private TableTriggerState table; + public TriggerStateProto setFileArrival(FileArrivalTriggerState fileArrival) { this.fileArrival = fileArrival; return this; @@ -22,21 +26,33 @@ public FileArrivalTriggerState getFileArrival() { return fileArrival; } + public TriggerStateProto setTable(TableTriggerState table) { + this.table = table; + return this; + } + + public TableTriggerState getTable() { + return table; + } + @Override public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; TriggerStateProto that = (TriggerStateProto) o; - return Objects.equals(fileArrival, that.fileArrival); + return Objects.equals(fileArrival, that.fileArrival) && Objects.equals(table, that.table); } @Override public int hashCode() { - return Objects.hash(fileArrival); + return Objects.hash(fileArrival, table); } @Override public String toString() { - return new ToStringer(TriggerStateProto.class).add("fileArrival", fileArrival).toString(); + return new ToStringer(TriggerStateProto.class) + .add("fileArrival", fileArrival) + .add("table", table) + .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersAPI.java index 9a6a622d5..c7eb9791a 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersAPI.java @@ -39,7 +39,7 @@ public GetProviderResponse get(GetProviderRequest request) { } /** List all providers in the Databricks Marketplace with at least one visible listing. */ - public Iterable list(ListProvidersRequest request) { + public Iterable list(ListConsumerProvidersRequest request) { return new Paginator<>( request, impl::list, diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersImpl.java index f767a09f5..8446c2275 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersImpl.java @@ -43,7 +43,7 @@ public GetProviderResponse get(GetProviderRequest request) { } @Override - public ListProvidersResponse list(ListProvidersRequest request) { + public ListProvidersResponse list(ListConsumerProvidersRequest request) { String path = "/api/2.1/marketplace-consumer/providers"; try { Request req = new Request("GET", path); diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersService.java index 9cb127aa9..9ec813960 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersService.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ConsumerProvidersService.java @@ -19,5 +19,5 @@ public interface ConsumerProvidersService { GetProviderResponse get(GetProviderRequest getProviderRequest); /** List all providers in the Databricks Marketplace with at least one visible listing. */ - ListProvidersResponse list(ListProvidersRequest listProvidersRequest); + ListProvidersResponse list(ListConsumerProvidersRequest listConsumerProvidersRequest); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListConsumerProvidersRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListConsumerProvidersRequest.java new file mode 100755 index 000000000..9206c509d --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListConsumerProvidersRequest.java @@ -0,0 +1,78 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.marketplace; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.QueryParam; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonIgnore; +import java.util.Objects; + +@Generated +public class ListConsumerProvidersRequest { + /** */ + @JsonIgnore + @QueryParam("is_featured") + private Boolean isFeatured; + + /** */ + @JsonIgnore + @QueryParam("page_size") + private Long pageSize; + + /** */ + @JsonIgnore + @QueryParam("page_token") + private String pageToken; + + public ListConsumerProvidersRequest setIsFeatured(Boolean isFeatured) { + this.isFeatured = isFeatured; + return this; + } + + public Boolean getIsFeatured() { + return isFeatured; + } + + public ListConsumerProvidersRequest setPageSize(Long pageSize) { + this.pageSize = pageSize; + return this; + } + + public Long getPageSize() { + return pageSize; + } + + public ListConsumerProvidersRequest setPageToken(String pageToken) { + this.pageToken = pageToken; + return this; + } + + public String getPageToken() { + return pageToken; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + ListConsumerProvidersRequest that = (ListConsumerProvidersRequest) o; + return Objects.equals(isFeatured, that.isFeatured) + && Objects.equals(pageSize, that.pageSize) + && Objects.equals(pageToken, that.pageToken); + } + + @Override + public int hashCode() { + return Objects.hash(isFeatured, pageSize, pageToken); + } + + @Override + public String toString() { + return new ToStringer(ListConsumerProvidersRequest.class) + .add("isFeatured", isFeatured) + .add("pageSize", pageSize) + .add("pageToken", pageToken) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListProvidersRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListProvidersRequest.java index f2a29e370..4eee78fbf 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListProvidersRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/ListProvidersRequest.java @@ -10,11 +10,6 @@ @Generated public class ListProvidersRequest { - /** */ - @JsonIgnore - @QueryParam("is_featured") - private Boolean isFeatured; - /** */ @JsonIgnore @QueryParam("page_size") @@ -25,15 +20,6 @@ public class ListProvidersRequest { @QueryParam("page_token") private String pageToken; - public ListProvidersRequest setIsFeatured(Boolean isFeatured) { - this.isFeatured = isFeatured; - return this; - } - - public Boolean getIsFeatured() { - return isFeatured; - } - public ListProvidersRequest setPageSize(Long pageSize) { this.pageSize = pageSize; return this; @@ -57,20 +43,17 @@ public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; ListProvidersRequest that = (ListProvidersRequest) o; - return Objects.equals(isFeatured, that.isFeatured) - && Objects.equals(pageSize, that.pageSize) - && Objects.equals(pageToken, that.pageToken); + return Objects.equals(pageSize, that.pageSize) && Objects.equals(pageToken, that.pageToken); } @Override public int hashCode() { - return Objects.hash(isFeatured, pageSize, pageToken); + return Objects.hash(pageSize, pageToken); } @Override public String toString() { return new ToStringer(ListProvidersRequest.class) - .add("isFeatured", isFeatured) .add("pageSize", pageSize) .add("pageToken", pageToken) .toString(); diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreatePrivateAccessSettingsRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreatePrivateAccessSettingsRequest.java new file mode 100755 index 000000000..aed606c1f --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CreatePrivateAccessSettingsRequest.java @@ -0,0 +1,131 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.provisioning; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Collection; +import java.util.Objects; + +@Generated +public class CreatePrivateAccessSettingsRequest { + /** + * An array of Databricks VPC endpoint IDs. This is the Databricks ID that is returned when + * registering the VPC endpoint configuration in your Databricks account. This is not the ID of + * the VPC endpoint in AWS. + * + *

Only used when `private_access_level` is set to `ENDPOINT`. This is an allow list of VPC + * endpoints that in your account that can connect to your workspace over AWS PrivateLink. + * + *

If hybrid access to your workspace is enabled by setting `public_access_enabled` to `true`, + * this control only works for PrivateLink connections. To control how your workspace is accessed + * via public internet, see [IP access lists]. + * + *

[IP access lists]: https://docs.databricks.com/security/network/ip-access-list.html + */ + @JsonProperty("allowed_vpc_endpoint_ids") + private Collection allowedVpcEndpointIds; + + /** */ + @JsonProperty("private_access_level") + private PrivateAccessLevel privateAccessLevel; + + /** The human-readable name of the private access settings object. */ + @JsonProperty("private_access_settings_name") + private String privateAccessSettingsName; + + /** + * Determines if the workspace can be accessed over public internet. For fully private workspaces, + * you can optionally specify `false`, but only if you implement both the front-end and the + * back-end PrivateLink connections. Otherwise, specify `true`, which means that public access is + * enabled. + */ + @JsonProperty("public_access_enabled") + private Boolean publicAccessEnabled; + + /** The cloud region for workspaces associated with this private access settings object. */ + @JsonProperty("region") + private String region; + + public CreatePrivateAccessSettingsRequest setAllowedVpcEndpointIds( + Collection allowedVpcEndpointIds) { + this.allowedVpcEndpointIds = allowedVpcEndpointIds; + return this; + } + + public Collection getAllowedVpcEndpointIds() { + return allowedVpcEndpointIds; + } + + public CreatePrivateAccessSettingsRequest setPrivateAccessLevel( + PrivateAccessLevel privateAccessLevel) { + this.privateAccessLevel = privateAccessLevel; + return this; + } + + public PrivateAccessLevel getPrivateAccessLevel() { + return privateAccessLevel; + } + + public CreatePrivateAccessSettingsRequest setPrivateAccessSettingsName( + String privateAccessSettingsName) { + this.privateAccessSettingsName = privateAccessSettingsName; + return this; + } + + public String getPrivateAccessSettingsName() { + return privateAccessSettingsName; + } + + public CreatePrivateAccessSettingsRequest setPublicAccessEnabled(Boolean publicAccessEnabled) { + this.publicAccessEnabled = publicAccessEnabled; + return this; + } + + public Boolean getPublicAccessEnabled() { + return publicAccessEnabled; + } + + public CreatePrivateAccessSettingsRequest setRegion(String region) { + this.region = region; + return this; + } + + public String getRegion() { + return region; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + CreatePrivateAccessSettingsRequest that = (CreatePrivateAccessSettingsRequest) o; + return Objects.equals(allowedVpcEndpointIds, that.allowedVpcEndpointIds) + && Objects.equals(privateAccessLevel, that.privateAccessLevel) + && Objects.equals(privateAccessSettingsName, that.privateAccessSettingsName) + && Objects.equals(publicAccessEnabled, that.publicAccessEnabled) + && Objects.equals(region, that.region); + } + + @Override + public int hashCode() { + return Objects.hash( + allowedVpcEndpointIds, + privateAccessLevel, + privateAccessSettingsName, + publicAccessEnabled, + region); + } + + @Override + public String toString() { + return new ToStringer(CreatePrivateAccessSettingsRequest.class) + .add("allowedVpcEndpointIds", allowedVpcEndpointIds) + .add("privateAccessLevel", privateAccessLevel) + .add("privateAccessSettingsName", privateAccessSettingsName) + .add("publicAccessEnabled", publicAccessEnabled) + .add("region", region) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessAPI.java index 130a5809f..82b70e7a6 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessAPI.java @@ -38,7 +38,7 @@ public PrivateAccessAPI(PrivateAccessService mock) { * PrivateLink]: * https://docs.databricks.com/administration-guide/cloud-configurations/aws/privatelink.html */ - public PrivateAccessSettings create(UpsertPrivateAccessSettingsRequest request) { + public PrivateAccessSettings create(CreatePrivateAccessSettingsRequest request) { return impl.create(request); } @@ -103,7 +103,7 @@ public Iterable list() { * PrivateLink]: * https://docs.databricks.com/administration-guide/cloud-configurations/aws/privatelink.html */ - public void replace(UpsertPrivateAccessSettingsRequest request) { + public void replace(ReplacePrivateAccessSettingsRequest request) { impl.replace(request); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessImpl.java index 948896927..dcca1eeba 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessImpl.java @@ -18,7 +18,7 @@ public PrivateAccessImpl(ApiClient apiClient) { } @Override - public PrivateAccessSettings create(UpsertPrivateAccessSettingsRequest request) { + public PrivateAccessSettings create(CreatePrivateAccessSettingsRequest request) { String path = String.format( "/api/2.0/accounts/%s/private-access-settings", apiClient.configuredAccountID()); @@ -76,7 +76,7 @@ public Collection list() { } @Override - public void replace(UpsertPrivateAccessSettingsRequest request) { + public void replace(ReplacePrivateAccessSettingsRequest request) { String path = String.format( "/api/2.0/accounts/%s/private-access-settings/%s", diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessService.java index fcc1d2ec3..1b3d54252 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessService.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessService.java @@ -29,7 +29,7 @@ public interface PrivateAccessService { * https://docs.databricks.com/administration-guide/cloud-configurations/aws/privatelink.html */ PrivateAccessSettings create( - UpsertPrivateAccessSettingsRequest upsertPrivateAccessSettingsRequest); + CreatePrivateAccessSettingsRequest createPrivateAccessSettingsRequest); /** * Deletes a private access settings object, which determines how your workspace is accessed over @@ -78,5 +78,5 @@ PrivateAccessSettings create( * PrivateLink]: * https://docs.databricks.com/administration-guide/cloud-configurations/aws/privatelink.html */ - void replace(UpsertPrivateAccessSettingsRequest upsertPrivateAccessSettingsRequest); + void replace(ReplacePrivateAccessSettingsRequest replacePrivateAccessSettingsRequest); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/UpsertPrivateAccessSettingsRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/ReplacePrivateAccessSettingsRequest.java similarity index 87% rename from databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/UpsertPrivateAccessSettingsRequest.java rename to databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/ReplacePrivateAccessSettingsRequest.java index 7efe220b5..4d62abff4 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/UpsertPrivateAccessSettingsRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/ReplacePrivateAccessSettingsRequest.java @@ -10,7 +10,7 @@ import java.util.Objects; @Generated -public class UpsertPrivateAccessSettingsRequest { +public class ReplacePrivateAccessSettingsRequest { /** * An array of Databricks VPC endpoint IDs. This is the Databricks ID that is returned when * registering the VPC endpoint configuration in your Databricks account. This is not the ID of @@ -52,7 +52,7 @@ public class UpsertPrivateAccessSettingsRequest { @JsonProperty("region") private String region; - public UpsertPrivateAccessSettingsRequest setAllowedVpcEndpointIds( + public ReplacePrivateAccessSettingsRequest setAllowedVpcEndpointIds( Collection allowedVpcEndpointIds) { this.allowedVpcEndpointIds = allowedVpcEndpointIds; return this; @@ -62,7 +62,7 @@ public Collection getAllowedVpcEndpointIds() { return allowedVpcEndpointIds; } - public UpsertPrivateAccessSettingsRequest setPrivateAccessLevel( + public ReplacePrivateAccessSettingsRequest setPrivateAccessLevel( PrivateAccessLevel privateAccessLevel) { this.privateAccessLevel = privateAccessLevel; return this; @@ -72,7 +72,7 @@ public PrivateAccessLevel getPrivateAccessLevel() { return privateAccessLevel; } - public UpsertPrivateAccessSettingsRequest setPrivateAccessSettingsId( + public ReplacePrivateAccessSettingsRequest setPrivateAccessSettingsId( String privateAccessSettingsId) { this.privateAccessSettingsId = privateAccessSettingsId; return this; @@ -82,7 +82,7 @@ public String getPrivateAccessSettingsId() { return privateAccessSettingsId; } - public UpsertPrivateAccessSettingsRequest setPrivateAccessSettingsName( + public ReplacePrivateAccessSettingsRequest setPrivateAccessSettingsName( String privateAccessSettingsName) { this.privateAccessSettingsName = privateAccessSettingsName; return this; @@ -92,7 +92,7 @@ public String getPrivateAccessSettingsName() { return privateAccessSettingsName; } - public UpsertPrivateAccessSettingsRequest setPublicAccessEnabled(Boolean publicAccessEnabled) { + public ReplacePrivateAccessSettingsRequest setPublicAccessEnabled(Boolean publicAccessEnabled) { this.publicAccessEnabled = publicAccessEnabled; return this; } @@ -101,7 +101,7 @@ public Boolean getPublicAccessEnabled() { return publicAccessEnabled; } - public UpsertPrivateAccessSettingsRequest setRegion(String region) { + public ReplacePrivateAccessSettingsRequest setRegion(String region) { this.region = region; return this; } @@ -114,7 +114,7 @@ public String getRegion() { public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; - UpsertPrivateAccessSettingsRequest that = (UpsertPrivateAccessSettingsRequest) o; + ReplacePrivateAccessSettingsRequest that = (ReplacePrivateAccessSettingsRequest) o; return Objects.equals(allowedVpcEndpointIds, that.allowedVpcEndpointIds) && Objects.equals(privateAccessLevel, that.privateAccessLevel) && Objects.equals(privateAccessSettingsId, that.privateAccessSettingsId) @@ -136,7 +136,7 @@ public int hashCode() { @Override public String toString() { - return new ToStringer(UpsertPrivateAccessSettingsRequest.class) + return new ToStringer(ReplacePrivateAccessSettingsRequest.class) .add("allowedVpcEndpointIds", allowedVpcEndpointIds) .add("privateAccessLevel", privateAccessLevel) .add("privateAccessSettingsId", privateAccessSettingsId) diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/CreatePtEndpointRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/CreatePtEndpointRequest.java index e7ae83feb..f85a44169 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/CreatePtEndpointRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/CreatePtEndpointRequest.java @@ -22,6 +22,10 @@ public class CreatePtEndpointRequest { @JsonProperty("config") private PtEndpointCoreConfig config; + /** Email notification settings. */ + @JsonProperty("email_notifications") + private EmailNotifications emailNotifications; + /** * The name of the serving endpoint. This field is required and must be unique across a Databricks * workspace. An endpoint name can consist of alphanumeric characters, dashes, and underscores. @@ -60,6 +64,15 @@ public PtEndpointCoreConfig getConfig() { return config; } + public CreatePtEndpointRequest setEmailNotifications(EmailNotifications emailNotifications) { + this.emailNotifications = emailNotifications; + return this; + } + + public EmailNotifications getEmailNotifications() { + return emailNotifications; + } + public CreatePtEndpointRequest setName(String name) { this.name = name; return this; @@ -86,13 +99,14 @@ public boolean equals(Object o) { return Objects.equals(aiGateway, that.aiGateway) && Objects.equals(budgetPolicyId, that.budgetPolicyId) && Objects.equals(config, that.config) + && Objects.equals(emailNotifications, that.emailNotifications) && Objects.equals(name, that.name) && Objects.equals(tags, that.tags); } @Override public int hashCode() { - return Objects.hash(aiGateway, budgetPolicyId, config, name, tags); + return Objects.hash(aiGateway, budgetPolicyId, config, emailNotifications, name, tags); } @Override @@ -101,6 +115,7 @@ public String toString() { .add("aiGateway", aiGateway) .add("budgetPolicyId", budgetPolicyId) .add("config", config) + .add("emailNotifications", emailNotifications) .add("name", name) .add("tags", tags) .toString(); diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/CreateServingEndpoint.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/CreateServingEndpoint.java index 0c78af532..0d684fb49 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/CreateServingEndpoint.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/CreateServingEndpoint.java @@ -30,6 +30,10 @@ public class CreateServingEndpoint { @JsonProperty("description") private String description; + /** Email notification settings. */ + @JsonProperty("email_notifications") + private EmailNotifications emailNotifications; + /** * The name of the serving endpoint. This field is required and must be unique across a Databricks * workspace. An endpoint name can consist of alphanumeric characters, dashes, and underscores. @@ -88,6 +92,15 @@ public String getDescription() { return description; } + public CreateServingEndpoint setEmailNotifications(EmailNotifications emailNotifications) { + this.emailNotifications = emailNotifications; + return this; + } + + public EmailNotifications getEmailNotifications() { + return emailNotifications; + } + public CreateServingEndpoint setName(String name) { this.name = name; return this; @@ -133,6 +146,7 @@ public boolean equals(Object o) { && Objects.equals(budgetPolicyId, that.budgetPolicyId) && Objects.equals(config, that.config) && Objects.equals(description, that.description) + && Objects.equals(emailNotifications, that.emailNotifications) && Objects.equals(name, that.name) && Objects.equals(rateLimits, that.rateLimits) && Objects.equals(routeOptimized, that.routeOptimized) @@ -142,7 +156,15 @@ public boolean equals(Object o) { @Override public int hashCode() { return Objects.hash( - aiGateway, budgetPolicyId, config, description, name, rateLimits, routeOptimized, tags); + aiGateway, + budgetPolicyId, + config, + description, + emailNotifications, + name, + rateLimits, + routeOptimized, + tags); } @Override @@ -152,6 +174,7 @@ public String toString() { .add("budgetPolicyId", budgetPolicyId) .add("config", config) .add("description", description) + .add("emailNotifications", emailNotifications) .add("name", name) .add("rateLimits", rateLimits) .add("routeOptimized", routeOptimized) diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/DataframeSplitInput.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/DataframeSplitInput.java index d146ddf42..fad484657 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/DataframeSplitInput.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/DataframeSplitInput.java @@ -10,15 +10,15 @@ @Generated public class DataframeSplitInput { - /** */ + /** Columns array for the dataframe */ @JsonProperty("columns") private Collection columns; - /** */ + /** Data array for the dataframe */ @JsonProperty("data") private Collection data; - /** */ + /** Index array for the dataframe */ @JsonProperty("index") private Collection index; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmailNotifications.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmailNotifications.java new file mode 100755 index 000000000..66bfaa0f5 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmailNotifications.java @@ -0,0 +1,66 @@ +// 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.Collection; +import java.util.Objects; + +@Generated +public class EmailNotifications { + /** + * A list of email addresses to be notified when an endpoint fails to update its configuration or + * state. + */ + @JsonProperty("on_update_failure") + private Collection onUpdateFailure; + + /** + * A list of email addresses to be notified when an endpoint successfully updates its + * configuration or state. + */ + @JsonProperty("on_update_success") + private Collection onUpdateSuccess; + + public EmailNotifications setOnUpdateFailure(Collection onUpdateFailure) { + this.onUpdateFailure = onUpdateFailure; + return this; + } + + public Collection getOnUpdateFailure() { + return onUpdateFailure; + } + + public EmailNotifications setOnUpdateSuccess(Collection onUpdateSuccess) { + this.onUpdateSuccess = onUpdateSuccess; + return this; + } + + public Collection getOnUpdateSuccess() { + return onUpdateSuccess; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + EmailNotifications that = (EmailNotifications) o; + return Objects.equals(onUpdateFailure, that.onUpdateFailure) + && Objects.equals(onUpdateSuccess, that.onUpdateSuccess); + } + + @Override + public int hashCode() { + return Objects.hash(onUpdateFailure, onUpdateSuccess); + } + + @Override + public String toString() { + return new ToStringer(EmailNotifications.class) + .add("onUpdateFailure", onUpdateFailure) + .add("onUpdateSuccess", onUpdateSuccess) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmbeddingsV1ResponseEmbeddingElement.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmbeddingsV1ResponseEmbeddingElement.java index 90b426e0a..95e93d1dd 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmbeddingsV1ResponseEmbeddingElement.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/EmbeddingsV1ResponseEmbeddingElement.java @@ -10,7 +10,7 @@ @Generated public class EmbeddingsV1ResponseEmbeddingElement { - /** */ + /** The embedding vector */ @JsonProperty("embedding") private Collection embedding; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/QueryEndpointInput.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/QueryEndpointInput.java index 945a9b2b6..0541ffc62 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/QueryEndpointInput.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/QueryEndpointInput.java @@ -54,7 +54,7 @@ public class QueryEndpointInput { /** * The messages field used ONLY for __chat external & foundation model__ serving endpoints. This - * is a map of strings and should only be used with other chat query fields. + * is an array of ChatMessage objects and should only be used with other chat query fields. */ @JsonProperty("messages") private Collection messages; @@ -67,7 +67,10 @@ public class QueryEndpointInput { @JsonProperty("n") private Long n; - /** The name of the serving endpoint. This field is required. */ + /** + * The name of the serving endpoint. This field is required and is provided via the path + * parameter. + */ @JsonIgnore private String name; /** diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServedModelStateDeployment.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServedModelStateDeployment.java index e8d32a02a..9d15cdf89 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServedModelStateDeployment.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServedModelStateDeployment.java @@ -3,22 +3,12 @@ package com.databricks.sdk.service.serving; import com.databricks.sdk.support.Generated; -import com.fasterxml.jackson.annotation.JsonProperty; @Generated public enum ServedModelStateDeployment { - @JsonProperty("DEPLOYMENT_ABORTED") - ABORTED, - - @JsonProperty("DEPLOYMENT_CREATING") - CREATING, - - @JsonProperty("DEPLOYMENT_FAILED") - FAILED, - - @JsonProperty("DEPLOYMENT_READY") - READY, - - @JsonProperty("DEPLOYMENT_RECOVERING") - RECOVERING, + DEPLOYMENT_ABORTED, + DEPLOYMENT_CREATING, + DEPLOYMENT_FAILED, + DEPLOYMENT_READY, + DEPLOYMENT_RECOVERING, } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointDetailed.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointDetailed.java index 3f5297a3e..e9629fd93 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointDetailed.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointDetailed.java @@ -42,6 +42,10 @@ public class ServingEndpointDetailed { @JsonProperty("description") private String description; + /** Email notification settings. */ + @JsonProperty("email_notifications") + private EmailNotifications emailNotifications; + /** Endpoint invocation url if route optimization is enabled for endpoint */ @JsonProperty("endpoint_url") private String endpointUrl; @@ -148,6 +152,15 @@ public String getDescription() { return description; } + public ServingEndpointDetailed setEmailNotifications(EmailNotifications emailNotifications) { + this.emailNotifications = emailNotifications; + return this; + } + + public EmailNotifications getEmailNotifications() { + return emailNotifications; + } + public ServingEndpointDetailed setEndpointUrl(String endpointUrl) { this.endpointUrl = endpointUrl; return this; @@ -251,6 +264,7 @@ public boolean equals(Object o) { && Objects.equals(creator, that.creator) && Objects.equals(dataPlaneInfo, that.dataPlaneInfo) && Objects.equals(description, that.description) + && Objects.equals(emailNotifications, that.emailNotifications) && Objects.equals(endpointUrl, that.endpointUrl) && Objects.equals(id, that.id) && Objects.equals(lastUpdatedTimestamp, that.lastUpdatedTimestamp) @@ -273,6 +287,7 @@ public int hashCode() { creator, dataPlaneInfo, description, + emailNotifications, endpointUrl, id, lastUpdatedTimestamp, @@ -295,6 +310,7 @@ public String toString() { .add("creator", creator) .add("dataPlaneInfo", dataPlaneInfo) .add("description", description) + .add("emailNotifications", emailNotifications) .add("endpointUrl", endpointUrl) .add("id", id) .add("lastUpdatedTimestamp", lastUpdatedTimestamp) diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsAPI.java index b63bc4405..ce08dd45a 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsAPI.java @@ -225,7 +225,7 @@ public PutAiGatewayResponse putAiGateway(PutAiGatewayRequest request) { return impl.putAiGateway(request); } - /** Query a serving endpoint. */ + /** Query a serving endpoint */ public QueryEndpointResponse query(QueryEndpointInput request) { return impl.query(request); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsDataPlaneAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsDataPlaneAPI.java index 207d2f932..922fc5622 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsDataPlaneAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsDataPlaneAPI.java @@ -28,7 +28,7 @@ public ServingEndpointsDataPlaneAPI(ServingEndpointsDataPlaneService mock) { impl = mock; } - /** Query a serving endpoint. */ + /** Query a serving endpoint */ public QueryEndpointResponse query(QueryEndpointInput request) { return impl.query(request); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsDataPlaneService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsDataPlaneService.java index def58d4a2..ed3c7fc46 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsDataPlaneService.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsDataPlaneService.java @@ -13,6 +13,6 @@ */ @Generated public interface ServingEndpointsDataPlaneService { - /** Query a serving endpoint. */ + /** Query a serving endpoint */ QueryEndpointResponse query(QueryEndpointInput queryEndpointInput); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsService.java index 4d61dc712..b3d0a60cc 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsService.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsService.java @@ -82,7 +82,7 @@ ServingEndpointPermissions getPermissions( */ PutAiGatewayResponse putAiGateway(PutAiGatewayRequest putAiGatewayRequest); - /** Query a serving endpoint. */ + /** Query a serving endpoint */ QueryEndpointResponse query(QueryEndpointInput queryEndpointInput); /**