diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha index 118140862..f26953677 100644 --- a/.codegen/_openapi_sha +++ b/.codegen/_openapi_sha @@ -1 +1 @@ -64504edbdf47f4e1fd6866d208093a0e5370436b \ No newline at end of file +463ed2cce38f6306936a13c3a51822d7999553c3 \ No newline at end of file diff --git a/.gitattributes b/.gitattributes index c2748531a..dba46a6bf 100644 --- a/.gitattributes +++ b/.gitattributes @@ -428,6 +428,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateFunct databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateFunctionSecurityType.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateFunctionSqlDataAccess.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMcpServiceRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMcpServiceUserMappedCredentialRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMetastore.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMetastoreAssignment.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateModelProviderServiceRequest.java linguist-generated=true @@ -468,6 +469,8 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteExter databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteExternalMetadataRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteFunctionRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceUserMappedCredentialRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceUserMappedCredentialResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMetastoreRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteModelProviderServiceRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteModelServiceRequest.java linguist-generated=true @@ -577,6 +580,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetExternal databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetFunctionRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetGrantRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetMcpServiceRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetMcpServiceUserMappedCredentialRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetMetastoreRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetMetastoreSummaryResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetModelProviderServiceRequest.java linguist-generated=true @@ -666,6 +670,8 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MatchType.j databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpService.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceConfig.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceConfigSourceConnection.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceUserMappedCredential.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceUserMappedCredentialLogin.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MetastoreAssignment.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MetastoreInfo.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MetastoresAPI.java linguist-generated=true @@ -2635,10 +2641,12 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ApiSource databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ApplyEnvironmentRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ApplyEnvironmentRequestResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/AutoFullRefreshPolicy.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/AvroTransformerOptions.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/CloneMode.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ClonePipelineRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ClonePipelineResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ConfluenceConnectorOptions.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ConfluentSchemaRegistryOptions.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ConnectionParameters.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ConnectorOptions.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ConnectorType.java linguist-generated=true @@ -2707,6 +2715,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/Origin.ja databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/OutlookAttachmentMode.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/OutlookBodyFormat.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/OutlookOptions.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ParseMode.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PathPattern.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PipelineAccessControlRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PipelineAccessControlResponse.java linguist-generated=true @@ -2732,6 +2741,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/Pipelines databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PipelinesService.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PostgresCatalogConfig.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PostgresSlotConfig.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ProtobufTransformerOptions.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PublishingMode.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RabbitmqOptions.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RedditAdsOptions.java linguist-generated=true @@ -2742,6 +2752,7 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RestartWi databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RewindDatasetSpec.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RewindSpec.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RunAs.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/SchemaRegistryConfig.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/SchemaSpec.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/Sequencing.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/SerializedException.java linguist-generated=true diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md index e38f1e61c..aa04a2ce9 100644 --- a/NEXT_CHANGELOG.md +++ b/NEXT_CHANGELOG.md @@ -15,3 +15,7 @@ ### Internal Changes ### API Changes +* Add `createMcpServiceUserMappedCredential()`, `deleteMcpServiceUserMappedCredential()` and `getMcpServiceUserMappedCredential()` methods for `workspaceClient.aiGateway()` service. +* Add `groupName` field for `com.databricks.sdk.service.pipelines.RunAs`. +* Add `avroOptions` and `protobufOptions` fields for `com.databricks.sdk.service.pipelines.Transformer`. +* Add `AVRO` and `PROTOBUF` enum values for `com.databricks.sdk.service.pipelines.TransformerFormat`. \ No newline at end of file diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/AccountClient.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/AccountClient.java index 648348bd6..da266f117 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/AccountClient.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/AccountClient.java @@ -328,7 +328,7 @@ public AccountGroupsV2API groupsV2() { /** * These APIs are used to manage identities and the workspace access of these identities in - * . + * Databricks. */ public AccountIamV2API iamV2() { return iamV2API; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/WorkspaceClient.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/WorkspaceClient.java index ec2f6a317..9134c86fe 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/WorkspaceClient.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/WorkspaceClient.java @@ -2228,7 +2228,7 @@ public WorkspaceEntityTagAssignmentsAPI workspaceEntityTagAssignments() { /** * These APIs are used to manage identities and the workspace access of these identities in - * . + * Databricks. */ public WorkspaceIamV2API workspaceIamV2() { return workspaceIamV2API; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/aifunctions/AiFunctionsAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/aifunctions/AiFunctionsAPI.java index c5d2a719a..84ae787e5 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/aifunctions/AiFunctionsAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/aifunctions/AiFunctionsAPI.java @@ -41,11 +41,7 @@ public AiExtractResponse aiExtract(AiExtractRequest request) { return impl.aiExtract(request); } - /** - * Parse structured content from unstructured documents. For REST API requests, the default rate - * limit is 120 pages per minute per workspace. Contact your Databricks account team to request a - * higher limit. - */ + /** Parse structured content from unstructured documents. */ public AiParseDocumentResponse aiParseDocument(AiParseDocumentRequest request) { return impl.aiParseDocument(request); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/aifunctions/AiFunctionsService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/aifunctions/AiFunctionsService.java index f971cf0db..53be8cc4d 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/aifunctions/AiFunctionsService.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/aifunctions/AiFunctionsService.java @@ -26,10 +26,6 @@ public interface AiFunctionsService { */ AiExtractResponse aiExtract(AiExtractRequest aiExtractRequest); - /** - * Parse structured content from unstructured documents. For REST API requests, the default rate - * limit is 120 pages per minute per workspace. Contact your Databricks account team to request a - * higher limit. - */ + /** Parse structured content from unstructured documents. */ AiParseDocumentResponse aiParseDocument(AiParseDocumentRequest aiParseDocumentRequest); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/apps/ComputeStatus.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/apps/ComputeStatus.java index ed741af12..19bc1f492 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/apps/ComputeStatus.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/apps/ComputeStatus.java @@ -9,10 +9,7 @@ @Generated public class ComputeStatus { - /** - * The number of compute instances currently serving requests for this application. An instance is - * considered active if it is reachable and ready to handle requests. - */ + /** The number of compute instances used and billed for this application. */ @JsonProperty("active_instances") private Long activeInstances; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayAPI.java index 04f24a5b6..81e8288e0 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayAPI.java @@ -43,6 +43,18 @@ public McpService createMcpService(CreateMcpServiceRequest request) { return impl.createMcpService(request); } + /** + * Logs the caller in to an MCP service: creates their per-user OAuth credential, or + * re-authenticates it if one already exists. The request body carries the OAuth exchange fields. + * + *

You must be the owner of the MCP service or have `EXECUTE` on it, plus `USE_CATALOG` on the + * parent catalog and `USE_SCHEMA` on the parent schema. + */ + public McpServiceUserMappedCredential createMcpServiceUserMappedCredential( + CreateMcpServiceUserMappedCredentialRequest request) { + return impl.createMcpServiceUserMappedCredential(request); + } + /** * Creates a model provider service in a Unity Catalog schema. A model provider service stores * authentication and request configuration for an external model provider, such as OpenAI, Azure @@ -90,6 +102,23 @@ public void deleteMcpService(DeleteMcpServiceRequest request) { impl.deleteMcpService(request); } + public DeleteMcpServiceUserMappedCredentialResponse deleteMcpServiceUserMappedCredential( + String name) { + return deleteMcpServiceUserMappedCredential( + new DeleteMcpServiceUserMappedCredentialRequest().setName(name)); + } + + /** + * Revokes (deletes) the caller's per-user OAuth credential for an MCP service (logout). + * + *

You must be the owner of the MCP service or have `EXECUTE` on it, plus `USE_CATALOG` on the + * parent catalog and `USE_SCHEMA` on the parent schema. + */ + public DeleteMcpServiceUserMappedCredentialResponse deleteMcpServiceUserMappedCredential( + DeleteMcpServiceUserMappedCredentialRequest request) { + return impl.deleteMcpServiceUserMappedCredential(request); + } + public void deleteModelProviderService(String name) { deleteModelProviderService(new DeleteModelProviderServiceRequest().setName(name)); } @@ -135,6 +164,26 @@ public McpService getMcpService(GetMcpServiceRequest request) { return impl.getMcpService(request); } + public McpServiceUserMappedCredential getMcpServiceUserMappedCredential(String name) { + return getMcpServiceUserMappedCredential( + new GetMcpServiceUserMappedCredentialRequest().setName(name)); + } + + /** + * Returns the caller's per-user OAuth login state for an MCP service. Read + * `provisioning_info.state`: `ACTIVE` means the caller is logged in and the credential is usable; + * any other state (for example a failed or still-provisioning login) means the login has not + * completed and the caller should log in again. If the caller has no credential yet, the RPC + * returns `NOT_FOUND`. + * + *

You must be the owner of the MCP service or have `EXECUTE` on it, plus `USE_CATALOG` on the + * parent catalog and `USE_SCHEMA` on the parent schema. + */ + public McpServiceUserMappedCredential getMcpServiceUserMappedCredential( + GetMcpServiceUserMappedCredentialRequest request) { + return impl.getMcpServiceUserMappedCredential(request); + } + public ModelProviderService getModelProviderService(String name) { return getModelProviderService(new GetModelProviderServiceRequest().setName(name)); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayImpl.java index cd966387d..e0497f189 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayImpl.java @@ -34,6 +34,25 @@ public McpService createMcpService(CreateMcpServiceRequest request) { } } + @Override + public McpServiceUserMappedCredential createMcpServiceUserMappedCredential( + CreateMcpServiceUserMappedCredentialRequest request) { + String path = String.format("/api/2.1/unity-catalog/%s/user-credentials", request.getName()); + try { + Request req = new Request("POST", path, apiClient.serialize(request.getLogin())); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + req.withHeader("Content-Type", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Workspace-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, McpServiceUserMappedCredential.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + @Override public ModelProviderService createModelProviderService( CreateModelProviderServiceRequest request) { @@ -89,6 +108,24 @@ public void deleteMcpService(DeleteMcpServiceRequest request) { } } + @Override + public DeleteMcpServiceUserMappedCredentialResponse deleteMcpServiceUserMappedCredential( + DeleteMcpServiceUserMappedCredentialRequest request) { + String path = String.format("/api/2.1/unity-catalog/%s/user-credentials", request.getName()); + try { + Request req = new Request("DELETE", path); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Workspace-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, DeleteMcpServiceUserMappedCredentialResponse.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + @Override public void deleteModelProviderService(DeleteModelProviderServiceRequest request) { String path = String.format("/api/2.1/unity-catalog/%s", request.getName()); @@ -140,6 +177,24 @@ public McpService getMcpService(GetMcpServiceRequest request) { } } + @Override + public McpServiceUserMappedCredential getMcpServiceUserMappedCredential( + GetMcpServiceUserMappedCredentialRequest request) { + String path = String.format("/api/2.1/unity-catalog/%s/user-credentials", request.getName()); + try { + Request req = new Request("GET", path); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Workspace-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, McpServiceUserMappedCredential.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + @Override public ModelProviderService getModelProviderService(GetModelProviderServiceRequest request) { String path = String.format("/api/2.1/unity-catalog/%s", request.getName()); diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayService.java index 5c62fe964..01d02623b 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayService.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AiGatewayService.java @@ -27,6 +27,16 @@ public interface AiGatewayService { */ McpService createMcpService(CreateMcpServiceRequest createMcpServiceRequest); + /** + * Logs the caller in to an MCP service: creates their per-user OAuth credential, or + * re-authenticates it if one already exists. The request body carries the OAuth exchange fields. + * + *

You must be the owner of the MCP service or have `EXECUTE` on it, plus `USE_CATALOG` on the + * parent catalog and `USE_SCHEMA` on the parent schema. + */ + McpServiceUserMappedCredential createMcpServiceUserMappedCredential( + CreateMcpServiceUserMappedCredentialRequest createMcpServiceUserMappedCredentialRequest); + /** * Creates a model provider service in a Unity Catalog schema. A model provider service stores * authentication and request configuration for an external model provider, such as OpenAI, Azure @@ -64,6 +74,15 @@ ModelProviderService createModelProviderService( */ void deleteMcpService(DeleteMcpServiceRequest deleteMcpServiceRequest); + /** + * Revokes (deletes) the caller's per-user OAuth credential for an MCP service (logout). + * + *

You must be the owner of the MCP service or have `EXECUTE` on it, plus `USE_CATALOG` on the + * parent catalog and `USE_SCHEMA` on the parent schema. + */ + DeleteMcpServiceUserMappedCredentialResponse deleteMcpServiceUserMappedCredential( + DeleteMcpServiceUserMappedCredentialRequest deleteMcpServiceUserMappedCredentialRequest); + /** * Deletes the model provider service identified by its resource name. Optionally supply an `etag` * to make the delete conditional on the model provider service not having changed since it was @@ -92,6 +111,19 @@ void deleteModelProviderService( */ McpService getMcpService(GetMcpServiceRequest getMcpServiceRequest); + /** + * Returns the caller's per-user OAuth login state for an MCP service. Read + * `provisioning_info.state`: `ACTIVE` means the caller is logged in and the credential is usable; + * any other state (for example a failed or still-provisioning login) means the login has not + * completed and the caller should log in again. If the caller has no credential yet, the RPC + * returns `NOT_FOUND`. + * + *

You must be the owner of the MCP service or have `EXECUTE` on it, plus `USE_CATALOG` on the + * parent catalog and `USE_SCHEMA` on the parent schema. + */ + McpServiceUserMappedCredential getMcpServiceUserMappedCredential( + GetMcpServiceUserMappedCredentialRequest getMcpServiceUserMappedCredentialRequest); + /** * Returns the model provider service identified by its resource name. * diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMcpServiceUserMappedCredentialRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMcpServiceUserMappedCredentialRequest.java new file mode 100644 index 000000000..b3edf4568 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMcpServiceUserMappedCredentialRequest.java @@ -0,0 +1,60 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonIgnore; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class CreateMcpServiceUserMappedCredentialRequest { + /** */ + @JsonProperty("login") + private McpServiceUserMappedCredentialLogin login; + + /** Resource name of the MCP service. Format: `mcp-services/{catalog}.{schema}.{mcp_service}`. */ + @JsonIgnore private String name; + + public CreateMcpServiceUserMappedCredentialRequest setLogin( + McpServiceUserMappedCredentialLogin login) { + this.login = login; + return this; + } + + public McpServiceUserMappedCredentialLogin getLogin() { + return login; + } + + public CreateMcpServiceUserMappedCredentialRequest setName(String name) { + this.name = name; + return this; + } + + public String getName() { + return name; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + CreateMcpServiceUserMappedCredentialRequest that = + (CreateMcpServiceUserMappedCredentialRequest) o; + return Objects.equals(login, that.login) && Objects.equals(name, that.name); + } + + @Override + public int hashCode() { + return Objects.hash(login, name); + } + + @Override + public String toString() { + return new ToStringer(CreateMcpServiceUserMappedCredentialRequest.class) + .add("login", login) + .add("name", name) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceUserMappedCredentialRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceUserMappedCredentialRequest.java new file mode 100644 index 000000000..7f346469d --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceUserMappedCredentialRequest.java @@ -0,0 +1,44 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonIgnore; +import java.util.Objects; + +@Generated +public class DeleteMcpServiceUserMappedCredentialRequest { + /** Resource name of the MCP service. Format: `mcp-services/{catalog}.{schema}.{mcp_service}`. */ + @JsonIgnore private String name; + + public DeleteMcpServiceUserMappedCredentialRequest setName(String name) { + this.name = name; + return this; + } + + public String getName() { + return name; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + DeleteMcpServiceUserMappedCredentialRequest that = + (DeleteMcpServiceUserMappedCredentialRequest) o; + return Objects.equals(name, that.name); + } + + @Override + public int hashCode() { + return Objects.hash(name); + } + + @Override + public String toString() { + return new ToStringer(DeleteMcpServiceUserMappedCredentialRequest.class) + .add("name", name) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceUserMappedCredentialResponse.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceUserMappedCredentialResponse.java new file mode 100644 index 000000000..1baee599a --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/DeleteMcpServiceUserMappedCredentialResponse.java @@ -0,0 +1,32 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import java.util.Objects; + +/** + * Delete returns no resource; a dedicated (empty) response keeps the revoke RPC's shape owned here + * rather than google.protobuf.Empty. + */ +@Generated +public class DeleteMcpServiceUserMappedCredentialResponse { + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + return true; + } + + @Override + public int hashCode() { + return Objects.hash(); + } + + @Override + public String toString() { + return new ToStringer(DeleteMcpServiceUserMappedCredentialResponse.class).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetMcpServiceUserMappedCredentialRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetMcpServiceUserMappedCredentialRequest.java new file mode 100644 index 000000000..aba3b5990 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetMcpServiceUserMappedCredentialRequest.java @@ -0,0 +1,43 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonIgnore; +import java.util.Objects; + +@Generated +public class GetMcpServiceUserMappedCredentialRequest { + /** Resource name of the MCP service. Format: `mcp-services/{catalog}.{schema}.{mcp_service}`. */ + @JsonIgnore private String name; + + public GetMcpServiceUserMappedCredentialRequest setName(String name) { + this.name = name; + return this; + } + + public String getName() { + return name; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + GetMcpServiceUserMappedCredentialRequest that = (GetMcpServiceUserMappedCredentialRequest) o; + return Objects.equals(name, that.name); + } + + @Override + public int hashCode() { + return Objects.hash(name); + } + + @Override + public String toString() { + return new ToStringer(GetMcpServiceUserMappedCredentialRequest.class) + .add("name", name) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceUserMappedCredential.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceUserMappedCredential.java new file mode 100644 index 000000000..99bf726fe --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceUserMappedCredential.java @@ -0,0 +1,68 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Map; +import java.util.Objects; + +/** A caller's per-user OAuth credential for an MCP service. */ +@Generated +public class McpServiceUserMappedCredential { + /** + * Token-expiry info for the credential, returned as a flat map: `access_token_expiration` (always + * set) and `refresh_token_expiration` (set when the credential has a refresh token). Both values + * are timestamps. + */ + @JsonProperty("options") + private Map options; + + /** + * Provisioning state of the credential. `ACTIVE` means the caller is logged in and the credential + * is usable; any other state means the login has not completed. + */ + @JsonProperty("provisioning_info") + private ProvisioningInfo provisioningInfo; + + public McpServiceUserMappedCredential setOptions(Map options) { + this.options = options; + return this; + } + + public Map getOptions() { + return options; + } + + public McpServiceUserMappedCredential setProvisioningInfo(ProvisioningInfo provisioningInfo) { + this.provisioningInfo = provisioningInfo; + return this; + } + + public ProvisioningInfo getProvisioningInfo() { + return provisioningInfo; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + McpServiceUserMappedCredential that = (McpServiceUserMappedCredential) o; + return Objects.equals(options, that.options) + && Objects.equals(provisioningInfo, that.provisioningInfo); + } + + @Override + public int hashCode() { + return Objects.hash(options, provisioningInfo); + } + + @Override + public String toString() { + return new ToStringer(McpServiceUserMappedCredential.class) + .add("options", options) + .add("provisioningInfo", provisioningInfo) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceUserMappedCredentialLogin.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceUserMappedCredentialLogin.java new file mode 100644 index 000000000..26d563cab --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/McpServiceUserMappedCredentialLogin.java @@ -0,0 +1,48 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Map; +import java.util.Objects; + +/** + * Login input for an MCP service user credential. Carries the OAuth exchange fields as a flat map. + */ +@Generated +public class McpServiceUserMappedCredentialLogin { + /** OAuth exchange fields: `pkce_verifier`, `authorization_code`, and `oauth_redirect_uri`. */ + @JsonProperty("options") + private Map options; + + public McpServiceUserMappedCredentialLogin setOptions(Map options) { + this.options = options; + return this; + } + + public Map getOptions() { + return options; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + McpServiceUserMappedCredentialLogin that = (McpServiceUserMappedCredentialLogin) o; + return Objects.equals(options, that.options); + } + + @Override + public int hashCode() { + return Objects.hash(options); + } + + @Override + public String toString() { + return new ToStringer(McpServiceUserMappedCredentialLogin.class) + .add("options", options) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/AccountIamV2API.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/AccountIamV2API.java index fb761b9c4..39d8db2af 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/AccountIamV2API.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/AccountIamV2API.java @@ -9,7 +9,7 @@ /** * These APIs are used to manage identities and the workspace access of these identities in - * . + * Databricks. */ @Generated public class AccountIamV2API { diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/AccountIamV2Service.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/AccountIamV2Service.java index 270009962..d7f4307b5 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/AccountIamV2Service.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/AccountIamV2Service.java @@ -5,7 +5,7 @@ /** * These APIs are used to manage identities and the workspace access of these identities in - * . + * Databricks. * *

This is the high-level interface, that contains generated methods. * diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateGroupProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateGroupProxyRequest.java index 3b3b114de..968828918 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateGroupProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateGroupProxyRequest.java @@ -9,7 +9,7 @@ @Generated public class CreateGroupProxyRequest { - /** Required. Group to be created in */ + /** Required. Group to be created in Databricks */ @JsonProperty("group") private Group group; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateGroupRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateGroupRequest.java index 62c97720d..ee163beb3 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateGroupRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateGroupRequest.java @@ -9,7 +9,7 @@ @Generated public class CreateGroupRequest { - /** Required. Group to be created in */ + /** Required. Group to be created in Databricks */ @JsonProperty("group") private Group group; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateServicePrincipalProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateServicePrincipalProxyRequest.java index 43953b199..26d5752ca 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateServicePrincipalProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateServicePrincipalProxyRequest.java @@ -9,7 +9,7 @@ @Generated public class CreateServicePrincipalProxyRequest { - /** Required. Service principal to be created in */ + /** Required. Service principal to be created in Databricks */ @JsonProperty("service_principal") private ServicePrincipal servicePrincipal; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateServicePrincipalRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateServicePrincipalRequest.java index fadf9a546..9010d9d50 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateServicePrincipalRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateServicePrincipalRequest.java @@ -9,7 +9,7 @@ @Generated public class CreateServicePrincipalRequest { - /** Required. Service principal to be created in */ + /** Required. Service principal to be created in Databricks */ @JsonProperty("service_principal") private ServicePrincipal servicePrincipal; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateUserProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateUserProxyRequest.java index b35fec791..638007ef8 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateUserProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateUserProxyRequest.java @@ -9,7 +9,7 @@ @Generated public class CreateUserProxyRequest { - /** Required. User to be created in */ + /** Required. User to be created in Databricks */ @JsonProperty("user") private User user; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateUserRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateUserRequest.java index 0e85e104b..dbf3ec858 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateUserRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateUserRequest.java @@ -9,7 +9,7 @@ @Generated public class CreateUserRequest { - /** Required. User to be created in */ + /** Required. User to be created in Databricks */ @JsonProperty("user") private User user; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentDetailProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentDetailProxyRequest.java index ea0827e77..395945fed 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentDetailProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentDetailProxyRequest.java @@ -9,7 +9,7 @@ @Generated public class CreateWorkspaceAssignmentDetailProxyRequest { - /** Required. Workspace assignment detail to be created in . */ + /** Required. Workspace assignment detail to be created in Databricks. */ @JsonProperty("workspace_assignment_detail") private WorkspaceAssignmentDetail workspaceAssignmentDetail; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentDetailRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentDetailRequest.java index e832956f1..c3009b32f 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentDetailRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentDetailRequest.java @@ -10,7 +10,7 @@ @Generated public class CreateWorkspaceAssignmentDetailRequest { - /** Required. Workspace assignment detail to be created in . */ + /** Required. Workspace assignment detail to be created in Databricks. */ @JsonProperty("workspace_assignment_detail") private WorkspaceAssignmentDetail workspaceAssignmentDetail; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentProxyRequest.java index 58e3b2875..4e1c219ed 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentProxyRequest.java @@ -9,7 +9,7 @@ @Generated public class CreateWorkspaceAssignmentProxyRequest { - /** Required. Workspace assignment to be created in . */ + /** Required. Workspace assignment to be created in Databricks. */ @JsonProperty("workspace_assignment") private WorkspaceAssignment workspaceAssignment; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentRequest.java index 3a036f9b3..028998469 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/CreateWorkspaceAssignmentRequest.java @@ -10,7 +10,7 @@ @Generated public class CreateWorkspaceAssignmentRequest { - /** Required. Workspace assignment to be created in . */ + /** Required. Workspace assignment to be created in Databricks. */ @JsonProperty("workspace_assignment") private WorkspaceAssignment workspaceAssignment; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateGroupProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateGroupProxyRequest.java index c5a9e6bc5..92a78a90c 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateGroupProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateGroupProxyRequest.java @@ -11,7 +11,7 @@ @Generated public class UpdateGroupProxyRequest { - /** Required. Group to be updated in */ + /** Required. Group to be updated in Databricks */ @JsonProperty("group") private Group group; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateGroupRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateGroupRequest.java index de1a4952f..3a4147719 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateGroupRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateGroupRequest.java @@ -11,7 +11,7 @@ @Generated public class UpdateGroupRequest { - /** Required. Group to be updated in */ + /** Required. Group to be updated in Databricks */ @JsonProperty("group") private Group group; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateServicePrincipalProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateServicePrincipalProxyRequest.java index 091de5312..334d7f1bb 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateServicePrincipalProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateServicePrincipalProxyRequest.java @@ -11,7 +11,7 @@ @Generated public class UpdateServicePrincipalProxyRequest { - /** Required. Service principal to be updated in */ + /** Required. Service principal to be updated in Databricks */ @JsonProperty("service_principal") private ServicePrincipal servicePrincipal; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateServicePrincipalRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateServicePrincipalRequest.java index 1c5f67d44..a7b5de289 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateServicePrincipalRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateServicePrincipalRequest.java @@ -11,7 +11,7 @@ @Generated public class UpdateServicePrincipalRequest { - /** Required. Service Principal to be updated in */ + /** Required. Service Principal to be updated in Databricks */ @JsonProperty("service_principal") private ServicePrincipal servicePrincipal; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateUserProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateUserProxyRequest.java index b755e5214..5eae94056 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateUserProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateUserProxyRequest.java @@ -16,7 +16,7 @@ public class UpdateUserProxyRequest { @QueryParam("update_mask") private String updateMask; - /** Required. User to be updated in */ + /** Required. User to be updated in Databricks */ @JsonProperty("user") private User user; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateUserRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateUserRequest.java index 209ba0890..52d0f32c2 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateUserRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateUserRequest.java @@ -16,7 +16,7 @@ public class UpdateUserRequest { @QueryParam("update_mask") private String updateMask; - /** Required. User to be updated in */ + /** Required. User to be updated in Databricks */ @JsonProperty("user") private User user; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentDetailProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentDetailProxyRequest.java index 202897f76..ec482237c 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentDetailProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentDetailProxyRequest.java @@ -20,7 +20,7 @@ public class UpdateWorkspaceAssignmentDetailProxyRequest { @QueryParam("update_mask") private FieldMask updateMask; - /** Required. Workspace assignment detail to be updated in . */ + /** Required. Workspace assignment detail to be updated in Databricks. */ @JsonProperty("workspace_assignment_detail") private WorkspaceAssignmentDetail workspaceAssignmentDetail; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentDetailRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentDetailRequest.java index e9d0b4e7f..1d9d27198 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentDetailRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentDetailRequest.java @@ -20,7 +20,7 @@ public class UpdateWorkspaceAssignmentDetailRequest { @QueryParam("update_mask") private FieldMask updateMask; - /** Required. Workspace assignment detail to be updated in . */ + /** Required. Workspace assignment detail to be updated in Databricks. */ @JsonProperty("workspace_assignment_detail") private WorkspaceAssignmentDetail workspaceAssignmentDetail; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentProxyRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentProxyRequest.java index 98ca1233f..0a567e274 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentProxyRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentProxyRequest.java @@ -20,7 +20,7 @@ public class UpdateWorkspaceAssignmentProxyRequest { @QueryParam("update_mask") private FieldMask updateMask; - /** Required. Workspace assignment to be updated in . */ + /** Required. Workspace assignment to be updated in Databricks. */ @JsonProperty("workspace_assignment") private WorkspaceAssignment workspaceAssignment; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentRequest.java index 042954c1f..058be4896 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceAssignmentRequest.java @@ -20,7 +20,7 @@ public class UpdateWorkspaceAssignmentRequest { @QueryParam("update_mask") private FieldMask updateMask; - /** Required. Workspace assignment to be updated in . */ + /** Required. Workspace assignment to be updated in Databricks. */ @JsonProperty("workspace_assignment") private WorkspaceAssignment workspaceAssignment; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceIdentityDetailRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceIdentityDetailRequest.java index 392b6bc6c..2a27582ba 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceIdentityDetailRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/UpdateWorkspaceIdentityDetailRequest.java @@ -20,7 +20,7 @@ public class UpdateWorkspaceIdentityDetailRequest { @QueryParam("update_mask") private FieldMask updateMask; - /** Required. Workspace identity detail to be updated in . */ + /** Required. Workspace identity detail to be updated in Databricks. */ @JsonProperty("workspace_identity_detail") private WorkspaceIdentityDetail workspaceIdentityDetail; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceIamV2API.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceIamV2API.java index aff51c3d1..c42acb7d3 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceIamV2API.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceIamV2API.java @@ -9,7 +9,7 @@ /** * These APIs are used to manage identities and the workspace access of these identities in - * . + * Databricks. */ @Generated public class WorkspaceIamV2API { diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceIamV2Service.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceIamV2Service.java index d61b80c7d..217ccf0f1 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceIamV2Service.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iamv2/WorkspaceIamV2Service.java @@ -5,7 +5,7 @@ /** * These APIs are used to manage identities and the workspace access of these identities in - * . + * Databricks. * *

This is the high-level interface, that contains generated methods. * diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobRunAs.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobRunAs.java index dcbf39048..6dc0a7333 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobRunAs.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobRunAs.java @@ -8,17 +8,18 @@ import java.util.Objects; /** - * Write-only setting. Specifies the user or service principal that the job runs as. If not + * Write-only setting. Specifies the user, service principal, or group that the job runs as. If not * specified, the job runs as the user who created the job. * - *

Either `user_name` or `service_principal_name` should be specified. If not, an error is - * thrown. + *

One of `user_name`, `service_principal_name`, or `group_name` should be specified. If not, an + * error is thrown. */ @Generated public class JobRunAs { /** - * Group name of an account group assigned to the workspace. Setting this field requires being a - * member of the group. + * Group name of an account group assigned to the workspace. When set, all tasks run as the group + * and the group's permissions are used for data access. Setting this field requires being a + * member of the group, or having the `Assume` permission on the group. */ @JsonProperty("group_name") private String groupName; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/FeatureStoreAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/FeatureStoreAPI.java index 06d49bfc8..ff122bf67 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/FeatureStoreAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/FeatureStoreAPI.java @@ -83,7 +83,13 @@ public PublishTableResponse publishTable(PublishTableRequest request) { return impl.publishTable(request); } - /** Update an Online Feature Store. */ + /** + * Update an Online Feature Store. + * + *

This update is not guaranteed to be atomic: when a request changes multiple fields, some may + * be applied while others fail. On a failed response, treat the update as partially applied and + * retry until it succeeds. + */ public OnlineStore updateOnlineStore(UpdateOnlineStoreRequest request) { return impl.updateOnlineStore(request); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/FeatureStoreService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/FeatureStoreService.java index c9de2f866..ac91114d9 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/FeatureStoreService.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/FeatureStoreService.java @@ -35,6 +35,12 @@ public interface FeatureStoreService { /** Publish features. */ PublishTableResponse publishTable(PublishTableRequest publishTableRequest); - /** Update an Online Feature Store. */ + /** + * Update an Online Feature Store. + * + *

This update is not guaranteed to be atomic: when a request changes multiple fields, some may + * be applied while others fail. On a failed response, treat the update as partially applied and + * retry until it succeeds. + */ OnlineStore updateOnlineStore(UpdateOnlineStoreRequest updateOnlineStoreRequest); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/AvroTransformerOptions.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/AvroTransformerOptions.java new file mode 100644 index 000000000..e867d6693 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/AvroTransformerOptions.java @@ -0,0 +1,94 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.pipelines; + +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 AvroTransformerOptions { + /** + * (Optional) Parse mode for Avro data. Valid values: FAILFAST, PERMISSIVE. Defaults to FAILFAST. + */ + @JsonProperty("parse_mode") + private ParseMode parseMode; + + /** Inline Avro JSON schema string. */ + @JsonProperty("schema") + private String schema; + + /** Path to a schema file (.avsc). */ + @JsonProperty("schema_file_path") + private String schemaFilePath; + + /** + * (Optional) Schema registry to resolve the Avro schema at runtime instead of providing it inline + * or via a file path. + */ + @JsonProperty("schema_registry") + private SchemaRegistryConfig schemaRegistry; + + public AvroTransformerOptions setParseMode(ParseMode parseMode) { + this.parseMode = parseMode; + return this; + } + + public ParseMode getParseMode() { + return parseMode; + } + + public AvroTransformerOptions setSchema(String schema) { + this.schema = schema; + return this; + } + + public String getSchema() { + return schema; + } + + public AvroTransformerOptions setSchemaFilePath(String schemaFilePath) { + this.schemaFilePath = schemaFilePath; + return this; + } + + public String getSchemaFilePath() { + return schemaFilePath; + } + + public AvroTransformerOptions setSchemaRegistry(SchemaRegistryConfig schemaRegistry) { + this.schemaRegistry = schemaRegistry; + return this; + } + + public SchemaRegistryConfig getSchemaRegistry() { + return schemaRegistry; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + AvroTransformerOptions that = (AvroTransformerOptions) o; + return Objects.equals(parseMode, that.parseMode) + && Objects.equals(schema, that.schema) + && Objects.equals(schemaFilePath, that.schemaFilePath) + && Objects.equals(schemaRegistry, that.schemaRegistry); + } + + @Override + public int hashCode() { + return Objects.hash(parseMode, schema, schemaFilePath, schemaRegistry); + } + + @Override + public String toString() { + return new ToStringer(AvroTransformerOptions.class) + .add("parseMode", parseMode) + .add("schema", schema) + .add("schemaFilePath", schemaFilePath) + .add("schemaRegistry", schemaRegistry) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ConfluentSchemaRegistryOptions.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ConfluentSchemaRegistryOptions.java new file mode 100644 index 000000000..f2e196a90 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ConfluentSchemaRegistryOptions.java @@ -0,0 +1,42 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.pipelines; + +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 ConfluentSchemaRegistryOptions { + /** Required: subject name to resolve in the registry. */ + @JsonProperty("subject") + private String subject; + + public ConfluentSchemaRegistryOptions setSubject(String subject) { + this.subject = subject; + return this; + } + + public String getSubject() { + return subject; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + ConfluentSchemaRegistryOptions that = (ConfluentSchemaRegistryOptions) o; + return Objects.equals(subject, that.subject); + } + + @Override + public int hashCode() { + return Objects.hash(subject); + } + + @Override + public String toString() { + return new ToStringer(ConfluentSchemaRegistryOptions.class).add("subject", subject).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ParseMode.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ParseMode.java new file mode 100644 index 000000000..b82030ec3 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ParseMode.java @@ -0,0 +1,12 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.pipelines; + +import com.databricks.sdk.support.Generated; + +/** Determines how errors encountered while deserializing records are handled. */ +@Generated +public enum ParseMode { + FAILFAST, + PERMISSIVE, +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ProtobufTransformerOptions.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ProtobufTransformerOptions.java new file mode 100644 index 000000000..3431a361a --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/ProtobufTransformerOptions.java @@ -0,0 +1,115 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.pipelines; + +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 ProtobufTransformerOptions { + /** Required: path to the .desc file (dbfs:/... or /Volumes/...). */ + @JsonProperty("desc_file_path") + private String descFilePath; + + /** Required: fully-qualified message type name. */ + @JsonProperty("message_name") + private String messageName; + + /** + * (Optional) Parse mode for Protobuf data. Valid values: FAILFAST, PERMISSIVE. Defaults to + * FAILFAST. + */ + @JsonProperty("parse_mode") + private ParseMode parseMode; + + /** + * (Optional) Maximum expansion depth for recursive protobuf fields. Spark SQL does not natively + * support recursive types, so recursive fields are expanded up to this depth and truncated beyond + * it. Valid values: -1 (disallow recursive fields), 0 (drop), 1-10. + */ + @JsonProperty("recursive_fields_max_depth") + private Long recursiveFieldsMaxDepth; + + /** + * (Optional) Schema registry to resolve the Protobuf schema at runtime instead of providing it + * via desc_file_path. + */ + @JsonProperty("schema_registry") + private SchemaRegistryConfig schemaRegistry; + + public ProtobufTransformerOptions setDescFilePath(String descFilePath) { + this.descFilePath = descFilePath; + return this; + } + + public String getDescFilePath() { + return descFilePath; + } + + public ProtobufTransformerOptions setMessageName(String messageName) { + this.messageName = messageName; + return this; + } + + public String getMessageName() { + return messageName; + } + + public ProtobufTransformerOptions setParseMode(ParseMode parseMode) { + this.parseMode = parseMode; + return this; + } + + public ParseMode getParseMode() { + return parseMode; + } + + public ProtobufTransformerOptions setRecursiveFieldsMaxDepth(Long recursiveFieldsMaxDepth) { + this.recursiveFieldsMaxDepth = recursiveFieldsMaxDepth; + return this; + } + + public Long getRecursiveFieldsMaxDepth() { + return recursiveFieldsMaxDepth; + } + + public ProtobufTransformerOptions setSchemaRegistry(SchemaRegistryConfig schemaRegistry) { + this.schemaRegistry = schemaRegistry; + return this; + } + + public SchemaRegistryConfig getSchemaRegistry() { + return schemaRegistry; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + ProtobufTransformerOptions that = (ProtobufTransformerOptions) o; + return Objects.equals(descFilePath, that.descFilePath) + && Objects.equals(messageName, that.messageName) + && Objects.equals(parseMode, that.parseMode) + && Objects.equals(recursiveFieldsMaxDepth, that.recursiveFieldsMaxDepth) + && Objects.equals(schemaRegistry, that.schemaRegistry); + } + + @Override + public int hashCode() { + return Objects.hash( + descFilePath, messageName, parseMode, recursiveFieldsMaxDepth, schemaRegistry); + } + + @Override + public String toString() { + return new ToStringer(ProtobufTransformerOptions.class) + .add("descFilePath", descFilePath) + .add("messageName", messageName) + .add("parseMode", parseMode) + .add("recursiveFieldsMaxDepth", recursiveFieldsMaxDepth) + .add("schemaRegistry", schemaRegistry) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RunAs.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RunAs.java index 0258b123e..53ea5207f 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RunAs.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/RunAs.java @@ -17,6 +17,14 @@ */ @Generated public class RunAs { + /** + * Group name of an account group assigned to the workspace. When set, the pipeline runs as the + * group and the group's permissions are used for data access. Setting this field requires being a + * member of the group, or having the `Assume` permission on the group. + */ + @JsonProperty("group_name") + private String groupName; + /** * Application ID of an active service principal. Setting this field requires the * `servicePrincipal/user` role. @@ -28,6 +36,15 @@ public class RunAs { @JsonProperty("user_name") private String userName; + public RunAs setGroupName(String groupName) { + this.groupName = groupName; + return this; + } + + public String getGroupName() { + return groupName; + } + public RunAs setServicePrincipalName(String servicePrincipalName) { this.servicePrincipalName = servicePrincipalName; return this; @@ -51,18 +68,20 @@ public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; RunAs that = (RunAs) o; - return Objects.equals(servicePrincipalName, that.servicePrincipalName) + return Objects.equals(groupName, that.groupName) + && Objects.equals(servicePrincipalName, that.servicePrincipalName) && Objects.equals(userName, that.userName); } @Override public int hashCode() { - return Objects.hash(servicePrincipalName, userName); + return Objects.hash(groupName, servicePrincipalName, userName); } @Override public String toString() { return new ToStringer(RunAs.class) + .add("groupName", groupName) .add("servicePrincipalName", servicePrincipalName) .add("userName", userName) .toString(); diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/SchemaRegistryConfig.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/SchemaRegistryConfig.java new file mode 100644 index 000000000..e298a90b3 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/SchemaRegistryConfig.java @@ -0,0 +1,81 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.pipelines; + +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 SchemaRegistryConfig { + /** Required: Confluent-compatible schema registry options. */ + @JsonProperty("confluent_options") + private ConfluentSchemaRegistryOptions confluentOptions; + + /** + * (Optional) UC connection for registry authentication. Specify if different from the top-level + * source connection. + */ + @JsonProperty("connection_name") + private String connectionName; + + /** + * (Optional, Protobuf only) Selects a specific message from a schema that defines multiple + * Protobuf messages. Simple ("Location") or fully-qualified ("com.example.protos.Location"). + * Defaults to the first message. + */ + @JsonProperty("protobuf_message_name") + private String protobufMessageName; + + public SchemaRegistryConfig setConfluentOptions(ConfluentSchemaRegistryOptions confluentOptions) { + this.confluentOptions = confluentOptions; + return this; + } + + public ConfluentSchemaRegistryOptions getConfluentOptions() { + return confluentOptions; + } + + public SchemaRegistryConfig setConnectionName(String connectionName) { + this.connectionName = connectionName; + return this; + } + + public String getConnectionName() { + return connectionName; + } + + public SchemaRegistryConfig setProtobufMessageName(String protobufMessageName) { + this.protobufMessageName = protobufMessageName; + return this; + } + + public String getProtobufMessageName() { + return protobufMessageName; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + SchemaRegistryConfig that = (SchemaRegistryConfig) o; + return Objects.equals(confluentOptions, that.confluentOptions) + && Objects.equals(connectionName, that.connectionName) + && Objects.equals(protobufMessageName, that.protobufMessageName); + } + + @Override + public int hashCode() { + return Objects.hash(confluentOptions, connectionName, protobufMessageName); + } + + @Override + public String toString() { + return new ToStringer(SchemaRegistryConfig.class) + .add("confluentOptions", confluentOptions) + .add("connectionName", connectionName) + .add("protobufMessageName", protobufMessageName) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/Transformer.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/Transformer.java index c237b131a..de9ceb7f3 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/Transformer.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/Transformer.java @@ -10,6 +10,10 @@ /** Specifies how to transform binary data into structured data. */ @Generated public class Transformer { + /** */ + @JsonProperty("avro_options") + private AvroTransformerOptions avroOptions; + /** Required: the wire format of the data. */ @JsonProperty("format") private TransformerFormat format; @@ -32,6 +36,19 @@ public class Transformer { @JsonProperty("output_column") private String outputColumn; + /** */ + @JsonProperty("protobuf_options") + private ProtobufTransformerOptions protobufOptions; + + public Transformer setAvroOptions(AvroTransformerOptions avroOptions) { + this.avroOptions = avroOptions; + return this; + } + + public AvroTransformerOptions getAvroOptions() { + return avroOptions; + } + public Transformer setFormat(TransformerFormat format) { this.format = format; return this; @@ -68,29 +85,43 @@ public String getOutputColumn() { return outputColumn; } + public Transformer setProtobufOptions(ProtobufTransformerOptions protobufOptions) { + this.protobufOptions = protobufOptions; + return this; + } + + public ProtobufTransformerOptions getProtobufOptions() { + return protobufOptions; + } + @Override public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; Transformer that = (Transformer) o; - return Objects.equals(format, that.format) + return Objects.equals(avroOptions, that.avroOptions) + && Objects.equals(format, that.format) && Objects.equals(inputColumn, that.inputColumn) && Objects.equals(jsonOptions, that.jsonOptions) - && Objects.equals(outputColumn, that.outputColumn); + && Objects.equals(outputColumn, that.outputColumn) + && Objects.equals(protobufOptions, that.protobufOptions); } @Override public int hashCode() { - return Objects.hash(format, inputColumn, jsonOptions, outputColumn); + return Objects.hash( + avroOptions, format, inputColumn, jsonOptions, outputColumn, protobufOptions); } @Override public String toString() { return new ToStringer(Transformer.class) + .add("avroOptions", avroOptions) .add("format", format) .add("inputColumn", inputColumn) .add("jsonOptions", jsonOptions) .add("outputColumn", outputColumn) + .add("protobufOptions", protobufOptions) .toString(); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/TransformerFormat.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/TransformerFormat.java index 06de84ce9..ce8912439 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/TransformerFormat.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/TransformerFormat.java @@ -6,6 +6,8 @@ @Generated public enum TransformerFormat { + AVRO, JSON, + PROTOBUF, STRING, }