From 8fe9b6ff248058a1aecca8c54029fc6bcd33c03e Mon Sep 17 00:00:00 2001 From: "databricks-ci-ghec-1[bot]" <184311507+databricks-ci-ghec-1[bot]@users.noreply.github.com> Date: Wed, 4 Mar 2026 09:54:55 +0000 Subject: [PATCH] Update SDK to e884ac5d77d4458ca4d3da058c0566c1442fbada --- .codegen/_openapi_sha | 2 +- .gitattributes | 23 ++ NEXT_CHANGELOG.md | 2 + .../com/databricks/sdk/WorkspaceClient.java | 21 ++ .../sdk/service/catalog/SecurableKind.java | 2 +- .../iam/GetPermissionLevelsRequest.java | 6 +- .../sdk/service/iam/GetPermissionRequest.java | 6 +- .../sdk/service/iam/SetObjectPermissions.java | 6 +- .../service/iam/UpdateObjectPermissions.java | 6 +- .../CreateKnowledgeAssistantRequest.java | 45 ++++ .../CreateKnowledgeSourceRequest.java | 62 +++++ .../DeleteKnowledgeAssistantRequest.java | 44 ++++ .../DeleteKnowledgeSourceRequest.java | 44 ++++ .../knowledgeassistants/FileTableSpec.java | 59 +++++ .../knowledgeassistants/FilesSpec.java | 43 ++++ .../GetKnowledgeAssistantRequest.java | 44 ++++ .../GetKnowledgeSourceRequest.java | 44 ++++ .../knowledgeassistants/IndexSpec.java | 75 ++++++ .../KnowledgeAssistant.java | 223 +++++++++++++++++ .../KnowledgeAssistantState.java | 12 + .../KnowledgeAssistantsAPI.java | 129 ++++++++++ .../KnowledgeAssistantsImpl.java | 212 ++++++++++++++++ .../KnowledgeAssistantsService.java | 55 +++++ .../knowledgeassistants/KnowledgeSource.java | 227 ++++++++++++++++++ .../KnowledgeSourceState.java | 12 + .../ListKnowledgeAssistantsRequest.java | 67 ++++++ .../ListKnowledgeAssistantsResponse.java | 65 +++++ .../ListKnowledgeSourcesRequest.java | 76 ++++++ .../ListKnowledgeSourcesResponse.java | 61 +++++ .../SyncKnowledgeSourcesRequest.java | 44 ++++ .../UpdateKnowledgeAssistantRequest.java | 89 +++++++ .../UpdateKnowledgeSourceRequest.java | 87 +++++++ .../tags/DeleteTagAssignmentRequest.java | 2 +- .../service/tags/GetTagAssignmentRequest.java | 2 +- .../tags/ListTagAssignmentsRequest.java | 2 +- .../sdk/service/tags/TagAssignment.java | 2 +- .../tags/UpdateTagAssignmentRequest.java | 2 +- 37 files changed, 1884 insertions(+), 19 deletions(-) mode change 100644 => 100755 NEXT_CHANGELOG.md create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeAssistantRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeSourceRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeAssistantRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeSourceRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FileTableSpec.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FilesSpec.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeAssistantRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeSourceRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/IndexSpec.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistant.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantState.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsAPI.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsImpl.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsService.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSource.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSourceState.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsResponse.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesResponse.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/SyncKnowledgeSourcesRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeAssistantRequest.java create mode 100755 databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeSourceRequest.java diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha index 12c62ec0b..22f385417 100755 --- a/.codegen/_openapi_sha +++ b/.codegen/_openapi_sha @@ -1 +1 @@ -86481d2fa23e3fb65128ea34b045fe585f7643f1 \ No newline at end of file +e884ac5d77d4458ca4d3da058c0566c1442fbada \ No newline at end of file diff --git a/.gitattributes b/.gitattributes index 2510002e8..905072ad5 100755 --- a/.gitattributes +++ b/.gitattributes @@ -1548,6 +1548,29 @@ databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/ViewsToExport. databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/Webhook.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/WebhookNotifications.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/WidgetErrorDetail.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeAssistantRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeSourceRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeAssistantRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeSourceRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FileTableSpec.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FilesSpec.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeAssistantRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeSourceRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/IndexSpec.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistant.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantState.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsAPI.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsImpl.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsService.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSource.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSourceState.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsResponse.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesResponse.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/SyncKnowledgeSourcesRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeAssistantRequest.java linguist-generated=true +databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeSourceRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/AddExchangeForListingRequest.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/AddExchangeForListingResponse.java linguist-generated=true databricks-sdk-java/src/main/java/com/databricks/sdk/service/marketplace/AssetType.java linguist-generated=true diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md old mode 100644 new mode 100755 index f881abc99..c0b7f4311 --- a/NEXT_CHANGELOG.md +++ b/NEXT_CHANGELOG.md @@ -13,3 +13,5 @@ ### Internal Changes ### API Changes +* Add `com.databricks.sdk.service.knowledgeassistants` package. +* Add `workspaceClient.knowledgeAssistants()` service. \ No newline at end of file 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 5b71b785e..f8c1deae6 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/WorkspaceClient.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/WorkspaceClient.java @@ -136,6 +136,8 @@ import com.databricks.sdk.service.jobs.JobsService; import com.databricks.sdk.service.jobs.PolicyComplianceForJobsAPI; import com.databricks.sdk.service.jobs.PolicyComplianceForJobsService; +import com.databricks.sdk.service.knowledgeassistants.KnowledgeAssistantsAPI; +import com.databricks.sdk.service.knowledgeassistants.KnowledgeAssistantsService; import com.databricks.sdk.service.marketplace.ConsumerFulfillmentsAPI; import com.databricks.sdk.service.marketplace.ConsumerFulfillmentsService; import com.databricks.sdk.service.marketplace.ConsumerInstallationsAPI; @@ -316,6 +318,7 @@ public class WorkspaceClient { private InstanceProfilesAPI instanceProfilesAPI; private IpAccessListsAPI ipAccessListsAPI; private JobsAPI jobsAPI; + private KnowledgeAssistantsAPI knowledgeAssistantsAPI; private LakeviewAPI lakeviewAPI; private LakeviewEmbeddedAPI lakeviewEmbeddedAPI; private LibrariesAPI librariesAPI; @@ -449,6 +452,7 @@ public WorkspaceClient(DatabricksConfig config) { instanceProfilesAPI = new InstanceProfilesAPI(apiClient); ipAccessListsAPI = new IpAccessListsAPI(apiClient); jobsAPI = new JobsAPI(apiClient); + knowledgeAssistantsAPI = new KnowledgeAssistantsAPI(apiClient); lakeviewAPI = new LakeviewAPI(apiClient); lakeviewEmbeddedAPI = new LakeviewEmbeddedAPI(apiClient); librariesAPI = new LibrariesAPI(apiClient); @@ -1146,6 +1150,11 @@ public JobsAPI jobs() { return jobsAPI; } + /** Manage Knowledge Assistants and related resources. */ + public KnowledgeAssistantsAPI knowledgeAssistants() { + return knowledgeAssistantsAPI; + } + /** * These APIs provide specific management operations for Lakeview dashboards. Generic resource * management can be done with Workspace API (import, export, get-status, list, delete). @@ -2728,6 +2737,18 @@ public WorkspaceClient withJobsAPI(JobsAPI jobs) { return this; } + /** Replace the default KnowledgeAssistantsService with a custom implementation. */ + public WorkspaceClient withKnowledgeAssistantsImpl( + KnowledgeAssistantsService knowledgeAssistants) { + return this.withKnowledgeAssistantsAPI(new KnowledgeAssistantsAPI(knowledgeAssistants)); + } + + /** Replace the default KnowledgeAssistantsAPI with a custom implementation. */ + public WorkspaceClient withKnowledgeAssistantsAPI(KnowledgeAssistantsAPI knowledgeAssistants) { + this.knowledgeAssistantsAPI = knowledgeAssistants; + return this; + } + /** Replace the default LakeviewService with a custom implementation. */ public WorkspaceClient withLakeviewImpl(LakeviewService lakeview) { return this.withLakeviewAPI(new LakeviewAPI(lakeview)); diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SecurableKind.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SecurableKind.java index 9cd70ad33..f0bb25743 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SecurableKind.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SecurableKind.java @@ -4,7 +4,7 @@ import com.databricks.sdk.support.Generated; -/** Latest kind: EXTERNAL_LOCATION_ONELAKE_MANAGED = 299; Next id: 300 */ +/** Latest kind: CONNECTION_OUTLOOK_OAUTH_M2M = 300; Next id: 301 */ @Generated public enum SecurableKind { TABLE_DB_STORAGE, diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GetPermissionLevelsRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GetPermissionLevelsRequest.java index 8ae58f6a5..f7f7dd752 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GetPermissionLevelsRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GetPermissionLevelsRequest.java @@ -14,9 +14,9 @@ public class GetPermissionLevelsRequest { /** * The type of the request object. Can be one of the following: alerts, alertsv2, authorization, - * clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, - * genie, instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, - * serving-endpoints, or warehouses. + * clusters, cluster-policies, dashboards, database-projects, dbsql-dashboards, directories, + * experiments, files, genie, instance-pools, jobs, notebooks, pipelines, queries, + * registered-models, repos, serving-endpoints, or warehouses. */ @JsonIgnore private String requestObjectType; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GetPermissionRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GetPermissionRequest.java index 801a423e4..b27841373 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GetPermissionRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GetPermissionRequest.java @@ -14,9 +14,9 @@ public class GetPermissionRequest { /** * The type of the request object. Can be one of the following: alerts, alertsv2, authorization, - * clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, - * genie, instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, - * serving-endpoints, or warehouses. + * clusters, cluster-policies, dashboards, database-projects, dbsql-dashboards, directories, + * experiments, files, genie, instance-pools, jobs, notebooks, pipelines, queries, + * registered-models, repos, serving-endpoints, or warehouses. */ @JsonIgnore private String requestObjectType; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/SetObjectPermissions.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/SetObjectPermissions.java index 25ba32997..a30c83e0c 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/SetObjectPermissions.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/SetObjectPermissions.java @@ -20,9 +20,9 @@ public class SetObjectPermissions { /** * The type of the request object. Can be one of the following: alerts, alertsv2, authorization, - * clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, - * genie, instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, - * serving-endpoints, or warehouses. + * clusters, cluster-policies, dashboards, database-projects, dbsql-dashboards, directories, + * experiments, files, genie, instance-pools, jobs, notebooks, pipelines, queries, + * registered-models, repos, serving-endpoints, or warehouses. */ @JsonIgnore private String requestObjectType; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UpdateObjectPermissions.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UpdateObjectPermissions.java index b7ea0195f..2420a2d73 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UpdateObjectPermissions.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UpdateObjectPermissions.java @@ -20,9 +20,9 @@ public class UpdateObjectPermissions { /** * The type of the request object. Can be one of the following: alerts, alertsv2, authorization, - * clusters, cluster-policies, dashboards, dbsql-dashboards, directories, experiments, files, - * genie, instance-pools, jobs, notebooks, pipelines, queries, registered-models, repos, - * serving-endpoints, or warehouses. + * clusters, cluster-policies, dashboards, database-projects, dbsql-dashboards, directories, + * experiments, files, genie, instance-pools, jobs, notebooks, pipelines, queries, + * registered-models, repos, serving-endpoints, or warehouses. */ @JsonIgnore private String requestObjectType; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeAssistantRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeAssistantRequest.java new file mode 100755 index 000000000..401986af0 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeAssistantRequest.java @@ -0,0 +1,45 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 CreateKnowledgeAssistantRequest { + /** The Knowledge Assistant to create. */ + @JsonProperty("knowledge_assistant") + private KnowledgeAssistant knowledgeAssistant; + + public CreateKnowledgeAssistantRequest setKnowledgeAssistant( + KnowledgeAssistant knowledgeAssistant) { + this.knowledgeAssistant = knowledgeAssistant; + return this; + } + + public KnowledgeAssistant getKnowledgeAssistant() { + return knowledgeAssistant; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + CreateKnowledgeAssistantRequest that = (CreateKnowledgeAssistantRequest) o; + return Objects.equals(knowledgeAssistant, that.knowledgeAssistant); + } + + @Override + public int hashCode() { + return Objects.hash(knowledgeAssistant); + } + + @Override + public String toString() { + return new ToStringer(CreateKnowledgeAssistantRequest.class) + .add("knowledgeAssistant", knowledgeAssistant) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeSourceRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeSourceRequest.java new file mode 100755 index 000000000..da7e77346 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/CreateKnowledgeSourceRequest.java @@ -0,0 +1,62 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 CreateKnowledgeSourceRequest { + /** */ + @JsonProperty("knowledge_source") + private KnowledgeSource knowledgeSource; + + /** + * Parent resource where this source will be created. Format: + * knowledge-assistants/{knowledge_assistant_id} + */ + @JsonIgnore private String parent; + + public CreateKnowledgeSourceRequest setKnowledgeSource(KnowledgeSource knowledgeSource) { + this.knowledgeSource = knowledgeSource; + return this; + } + + public KnowledgeSource getKnowledgeSource() { + return knowledgeSource; + } + + public CreateKnowledgeSourceRequest setParent(String parent) { + this.parent = parent; + return this; + } + + public String getParent() { + return parent; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + CreateKnowledgeSourceRequest that = (CreateKnowledgeSourceRequest) o; + return Objects.equals(knowledgeSource, that.knowledgeSource) + && Objects.equals(parent, that.parent); + } + + @Override + public int hashCode() { + return Objects.hash(knowledgeSource, parent); + } + + @Override + public String toString() { + return new ToStringer(CreateKnowledgeSourceRequest.class) + .add("knowledgeSource", knowledgeSource) + .add("parent", parent) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeAssistantRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeAssistantRequest.java new file mode 100755 index 000000000..60d4638a0 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeAssistantRequest.java @@ -0,0 +1,44 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 DeleteKnowledgeAssistantRequest { + /** + * The resource name of the knowledge assistant to be deleted. Format: + * knowledge-assistants/{knowledge_assistant_id} + */ + @JsonIgnore private String name; + + public DeleteKnowledgeAssistantRequest 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; + DeleteKnowledgeAssistantRequest that = (DeleteKnowledgeAssistantRequest) o; + return Objects.equals(name, that.name); + } + + @Override + public int hashCode() { + return Objects.hash(name); + } + + @Override + public String toString() { + return new ToStringer(DeleteKnowledgeAssistantRequest.class).add("name", name).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeSourceRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeSourceRequest.java new file mode 100755 index 000000000..746581a50 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/DeleteKnowledgeSourceRequest.java @@ -0,0 +1,44 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 DeleteKnowledgeSourceRequest { + /** + * The resource name of the Knowledge Source to delete. Format: + * knowledge-assistants/{knowledge_assistant_id}/knowledge-sources/{knowledge_source_id} + */ + @JsonIgnore private String name; + + public DeleteKnowledgeSourceRequest 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; + DeleteKnowledgeSourceRequest that = (DeleteKnowledgeSourceRequest) o; + return Objects.equals(name, that.name); + } + + @Override + public int hashCode() { + return Objects.hash(name); + } + + @Override + public String toString() { + return new ToStringer(DeleteKnowledgeSourceRequest.class).add("name", name).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FileTableSpec.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FileTableSpec.java new file mode 100755 index 000000000..18f65a692 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FileTableSpec.java @@ -0,0 +1,59 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +/** FileTableSpec specifies a file table source configuration. */ +@Generated +public class FileTableSpec { + /** The name of the column containing BINARY file content to be indexed. */ + @JsonProperty("file_col") + private String fileCol; + + /** Full UC name of the table, in the format of {CATALOG}.{SCHEMA}.{TABLE_NAME}. */ + @JsonProperty("table_name") + private String tableName; + + public FileTableSpec setFileCol(String fileCol) { + this.fileCol = fileCol; + return this; + } + + public String getFileCol() { + return fileCol; + } + + public FileTableSpec 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; + FileTableSpec that = (FileTableSpec) o; + return Objects.equals(fileCol, that.fileCol) && Objects.equals(tableName, that.tableName); + } + + @Override + public int hashCode() { + return Objects.hash(fileCol, tableName); + } + + @Override + public String toString() { + return new ToStringer(FileTableSpec.class) + .add("fileCol", fileCol) + .add("tableName", tableName) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FilesSpec.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FilesSpec.java new file mode 100755 index 000000000..14de72a5c --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/FilesSpec.java @@ -0,0 +1,43 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +/** FilesSpec specifies a files source configuration. */ +@Generated +public class FilesSpec { + /** A UC volume path that includes a list of files. */ + @JsonProperty("path") + private String path; + + public FilesSpec setPath(String path) { + this.path = path; + return this; + } + + public String getPath() { + return path; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + FilesSpec that = (FilesSpec) o; + return Objects.equals(path, that.path); + } + + @Override + public int hashCode() { + return Objects.hash(path); + } + + @Override + public String toString() { + return new ToStringer(FilesSpec.class).add("path", path).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeAssistantRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeAssistantRequest.java new file mode 100755 index 000000000..e722347bf --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeAssistantRequest.java @@ -0,0 +1,44 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 GetKnowledgeAssistantRequest { + /** + * The resource name of the knowledge assistant. Format: + * knowledge-assistants/{knowledge_assistant_id} + */ + @JsonIgnore private String name; + + public GetKnowledgeAssistantRequest 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; + GetKnowledgeAssistantRequest that = (GetKnowledgeAssistantRequest) o; + return Objects.equals(name, that.name); + } + + @Override + public int hashCode() { + return Objects.hash(name); + } + + @Override + public String toString() { + return new ToStringer(GetKnowledgeAssistantRequest.class).add("name", name).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeSourceRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeSourceRequest.java new file mode 100755 index 000000000..268d77a0c --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/GetKnowledgeSourceRequest.java @@ -0,0 +1,44 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 GetKnowledgeSourceRequest { + /** + * The resource name of the Knowledge Source. Format: + * knowledge-assistants/{knowledge_assistant_id}/knowledge-sources/{knowledge_source_id} + */ + @JsonIgnore private String name; + + public GetKnowledgeSourceRequest 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; + GetKnowledgeSourceRequest that = (GetKnowledgeSourceRequest) o; + return Objects.equals(name, that.name); + } + + @Override + public int hashCode() { + return Objects.hash(name); + } + + @Override + public String toString() { + return new ToStringer(GetKnowledgeSourceRequest.class).add("name", name).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/IndexSpec.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/IndexSpec.java new file mode 100755 index 000000000..a30bde9c6 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/IndexSpec.java @@ -0,0 +1,75 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +/** IndexSpec specifies a vector search index source configuration. */ +@Generated +public class IndexSpec { + /** The column that specifies a link or reference to where the information came from. */ + @JsonProperty("doc_uri_col") + private String docUriCol; + + /** Full UC name of the vector search index, in the format of {CATALOG}.{SCHEMA}.{INDEX_NAME}. */ + @JsonProperty("index_name") + private String indexName; + + /** The column that includes the document text for retrieval. */ + @JsonProperty("text_col") + private String textCol; + + public IndexSpec setDocUriCol(String docUriCol) { + this.docUriCol = docUriCol; + return this; + } + + public String getDocUriCol() { + return docUriCol; + } + + public IndexSpec setIndexName(String indexName) { + this.indexName = indexName; + return this; + } + + public String getIndexName() { + return indexName; + } + + public IndexSpec setTextCol(String textCol) { + this.textCol = textCol; + return this; + } + + public String getTextCol() { + return textCol; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + IndexSpec that = (IndexSpec) o; + return Objects.equals(docUriCol, that.docUriCol) + && Objects.equals(indexName, that.indexName) + && Objects.equals(textCol, that.textCol); + } + + @Override + public int hashCode() { + return Objects.hash(docUriCol, indexName, textCol); + } + + @Override + public String toString() { + return new ToStringer(IndexSpec.class) + .add("docUriCol", docUriCol) + .add("indexName", indexName) + .add("textCol", textCol) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistant.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistant.java new file mode 100755 index 000000000..b26ecda26 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistant.java @@ -0,0 +1,223 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.google.protobuf.Timestamp; +import java.util.Objects; + +/** + * Entity message that represents a knowledge assistant. Note: REQUIRED annotations below represent + * create-time requirements. For updates, required fields are determined by the update mask. + */ +@Generated +public class KnowledgeAssistant { + /** Creation timestamp. */ + @JsonProperty("create_time") + private Timestamp createTime; + + /** The creator of the Knowledge Assistant. */ + @JsonProperty("creator") + private String creator; + + /** + * Description of what this agent can do (user-facing). Required when creating a Knowledge + * Assistant. When updating a Knowledge Assistant, optional unless included in update_mask. + */ + @JsonProperty("description") + private String description; + + /** + * The display name of the Knowledge Assistant, unique at workspace level. Required when creating + * a Knowledge Assistant. When updating a Knowledge Assistant, optional unless included in + * update_mask. + */ + @JsonProperty("display_name") + private String displayName; + + /** The name of the knowledge assistant agent endpoint. */ + @JsonProperty("endpoint_name") + private String endpointName; + + /** Error details when the Knowledge Assistant is in FAILED state. */ + @JsonProperty("error_info") + private String errorInfo; + + /** The MLflow experiment ID. */ + @JsonProperty("experiment_id") + private String experimentId; + + /** The universally unique identifier (UUID) of the Knowledge Assistant. */ + @JsonProperty("id") + private String id; + + /** + * Additional global instructions on how the agent should generate answers. Optional on create and + * update. When updating a Knowledge Assistant, include this field in update_mask to modify it. + */ + @JsonProperty("instructions") + private String instructions; + + /** + * The resource name of the Knowledge Assistant. Format: + * knowledge-assistants/{knowledge_assistant_id} + */ + @JsonProperty("name") + private String name; + + /** State of the Knowledge Assistant. Not returned in List responses. */ + @JsonProperty("state") + private KnowledgeAssistantState state; + + public KnowledgeAssistant setCreateTime(Timestamp createTime) { + this.createTime = createTime; + return this; + } + + public Timestamp getCreateTime() { + return createTime; + } + + public KnowledgeAssistant setCreator(String creator) { + this.creator = creator; + return this; + } + + public String getCreator() { + return creator; + } + + public KnowledgeAssistant setDescription(String description) { + this.description = description; + return this; + } + + public String getDescription() { + return description; + } + + public KnowledgeAssistant setDisplayName(String displayName) { + this.displayName = displayName; + return this; + } + + public String getDisplayName() { + return displayName; + } + + public KnowledgeAssistant setEndpointName(String endpointName) { + this.endpointName = endpointName; + return this; + } + + public String getEndpointName() { + return endpointName; + } + + public KnowledgeAssistant setErrorInfo(String errorInfo) { + this.errorInfo = errorInfo; + return this; + } + + public String getErrorInfo() { + return errorInfo; + } + + public KnowledgeAssistant setExperimentId(String experimentId) { + this.experimentId = experimentId; + return this; + } + + public String getExperimentId() { + return experimentId; + } + + public KnowledgeAssistant setId(String id) { + this.id = id; + return this; + } + + public String getId() { + return id; + } + + public KnowledgeAssistant setInstructions(String instructions) { + this.instructions = instructions; + return this; + } + + public String getInstructions() { + return instructions; + } + + public KnowledgeAssistant setName(String name) { + this.name = name; + return this; + } + + public String getName() { + return name; + } + + public KnowledgeAssistant setState(KnowledgeAssistantState state) { + this.state = state; + return this; + } + + public KnowledgeAssistantState getState() { + return state; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + KnowledgeAssistant that = (KnowledgeAssistant) o; + return Objects.equals(createTime, that.createTime) + && Objects.equals(creator, that.creator) + && Objects.equals(description, that.description) + && Objects.equals(displayName, that.displayName) + && Objects.equals(endpointName, that.endpointName) + && Objects.equals(errorInfo, that.errorInfo) + && Objects.equals(experimentId, that.experimentId) + && Objects.equals(id, that.id) + && Objects.equals(instructions, that.instructions) + && Objects.equals(name, that.name) + && Objects.equals(state, that.state); + } + + @Override + public int hashCode() { + return Objects.hash( + createTime, + creator, + description, + displayName, + endpointName, + errorInfo, + experimentId, + id, + instructions, + name, + state); + } + + @Override + public String toString() { + return new ToStringer(KnowledgeAssistant.class) + .add("createTime", createTime) + .add("creator", creator) + .add("description", description) + .add("displayName", displayName) + .add("endpointName", endpointName) + .add("errorInfo", errorInfo) + .add("experimentId", experimentId) + .add("id", id) + .add("instructions", instructions) + .add("name", name) + .add("state", state) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantState.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantState.java new file mode 100755 index 000000000..3a47b2cbb --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantState.java @@ -0,0 +1,12 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.support.Generated; + +@Generated +public enum KnowledgeAssistantState { + ACTIVE, + CREATING, + FAILED, +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsAPI.java new file mode 100755 index 000000000..47c062573 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsAPI.java @@ -0,0 +1,129 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.core.ApiClient; +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.Paginator; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +/** Manage Knowledge Assistants and related resources. */ +@Generated +public class KnowledgeAssistantsAPI { + private static final Logger LOG = LoggerFactory.getLogger(KnowledgeAssistantsAPI.class); + + private final KnowledgeAssistantsService impl; + + /** Regular-use constructor */ + public KnowledgeAssistantsAPI(ApiClient apiClient) { + impl = new KnowledgeAssistantsImpl(apiClient); + } + + /** Constructor for mocks */ + public KnowledgeAssistantsAPI(KnowledgeAssistantsService mock) { + impl = mock; + } + + /** Creates a Knowledge Assistant. */ + public KnowledgeAssistant createKnowledgeAssistant(CreateKnowledgeAssistantRequest request) { + return impl.createKnowledgeAssistant(request); + } + + /** Creates a Knowledge Source under a Knowledge Assistant. */ + public KnowledgeSource createKnowledgeSource(CreateKnowledgeSourceRequest request) { + return impl.createKnowledgeSource(request); + } + + public void deleteKnowledgeAssistant(String name) { + deleteKnowledgeAssistant(new DeleteKnowledgeAssistantRequest().setName(name)); + } + + /** Deletes a Knowledge Assistant. */ + public void deleteKnowledgeAssistant(DeleteKnowledgeAssistantRequest request) { + impl.deleteKnowledgeAssistant(request); + } + + public void deleteKnowledgeSource(String name) { + deleteKnowledgeSource(new DeleteKnowledgeSourceRequest().setName(name)); + } + + /** Deletes a Knowledge Source. */ + public void deleteKnowledgeSource(DeleteKnowledgeSourceRequest request) { + impl.deleteKnowledgeSource(request); + } + + public KnowledgeAssistant getKnowledgeAssistant(String name) { + return getKnowledgeAssistant(new GetKnowledgeAssistantRequest().setName(name)); + } + + /** Gets a Knowledge Assistant. */ + public KnowledgeAssistant getKnowledgeAssistant(GetKnowledgeAssistantRequest request) { + return impl.getKnowledgeAssistant(request); + } + + public KnowledgeSource getKnowledgeSource(String name) { + return getKnowledgeSource(new GetKnowledgeSourceRequest().setName(name)); + } + + /** Gets a Knowledge Source. */ + public KnowledgeSource getKnowledgeSource(GetKnowledgeSourceRequest request) { + return impl.getKnowledgeSource(request); + } + + /** List Knowledge Assistants */ + public Iterable listKnowledgeAssistants( + ListKnowledgeAssistantsRequest request) { + return new Paginator<>( + request, + impl::listKnowledgeAssistants, + ListKnowledgeAssistantsResponse::getKnowledgeAssistants, + response -> { + String token = response.getNextPageToken(); + if (token == null || token.isEmpty()) { + return null; + } + return request.setPageToken(token); + }); + } + + public Iterable listKnowledgeSources(String parent) { + return listKnowledgeSources(new ListKnowledgeSourcesRequest().setParent(parent)); + } + + /** Lists Knowledge Sources under a Knowledge Assistant. */ + public Iterable listKnowledgeSources(ListKnowledgeSourcesRequest request) { + return new Paginator<>( + request, + impl::listKnowledgeSources, + ListKnowledgeSourcesResponse::getKnowledgeSources, + response -> { + String token = response.getNextPageToken(); + if (token == null || token.isEmpty()) { + return null; + } + return request.setPageToken(token); + }); + } + + /** + * Sync all non-index Knowledge Sources for a Knowledge Assistant (index sources do not require + * sync) + */ + public void syncKnowledgeSources(SyncKnowledgeSourcesRequest request) { + impl.syncKnowledgeSources(request); + } + + /** Updates a Knowledge Assistant. */ + public KnowledgeAssistant updateKnowledgeAssistant(UpdateKnowledgeAssistantRequest request) { + return impl.updateKnowledgeAssistant(request); + } + + /** Updates a Knowledge Source. */ + public KnowledgeSource updateKnowledgeSource(UpdateKnowledgeSourceRequest request) { + return impl.updateKnowledgeSource(request); + } + + public KnowledgeAssistantsService impl() { + return impl; + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsImpl.java new file mode 100755 index 000000000..14747a16d --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsImpl.java @@ -0,0 +1,212 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.core.ApiClient; +import com.databricks.sdk.core.DatabricksException; +import com.databricks.sdk.core.http.Request; +import com.databricks.sdk.support.Generated; +import java.io.IOException; + +/** Package-local implementation of KnowledgeAssistants */ +@Generated +class KnowledgeAssistantsImpl implements KnowledgeAssistantsService { + private final ApiClient apiClient; + + public KnowledgeAssistantsImpl(ApiClient apiClient) { + this.apiClient = apiClient; + } + + @Override + public KnowledgeAssistant createKnowledgeAssistant(CreateKnowledgeAssistantRequest request) { + String path = "/api/2.1/knowledge-assistants"; + try { + Request req = new Request("POST", path, apiClient.serialize(request.getKnowledgeAssistant())); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + req.withHeader("Content-Type", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Org-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, KnowledgeAssistant.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public KnowledgeSource createKnowledgeSource(CreateKnowledgeSourceRequest request) { + String path = String.format("/api/2.1/%s/knowledge-sources", request.getParent()); + try { + Request req = new Request("POST", path, apiClient.serialize(request.getKnowledgeSource())); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + req.withHeader("Content-Type", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Org-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, KnowledgeSource.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public void deleteKnowledgeAssistant(DeleteKnowledgeAssistantRequest request) { + String path = String.format("/api/2.1/%s", 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-Org-Id", apiClient.workspaceId()); + } + apiClient.execute(req, Void.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public void deleteKnowledgeSource(DeleteKnowledgeSourceRequest request) { + String path = String.format("/api/2.1/%s", 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-Org-Id", apiClient.workspaceId()); + } + apiClient.execute(req, Void.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public KnowledgeAssistant getKnowledgeAssistant(GetKnowledgeAssistantRequest request) { + String path = String.format("/api/2.1/%s", 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-Org-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, KnowledgeAssistant.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public KnowledgeSource getKnowledgeSource(GetKnowledgeSourceRequest request) { + String path = String.format("/api/2.1/%s", 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-Org-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, KnowledgeSource.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public ListKnowledgeAssistantsResponse listKnowledgeAssistants( + ListKnowledgeAssistantsRequest request) { + String path = "/api/2.1/knowledge-assistants"; + try { + Request req = new Request("GET", path); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Org-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, ListKnowledgeAssistantsResponse.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public ListKnowledgeSourcesResponse listKnowledgeSources(ListKnowledgeSourcesRequest request) { + String path = String.format("/api/2.1/%s/knowledge-sources", request.getParent()); + try { + Request req = new Request("GET", path); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Org-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, ListKnowledgeSourcesResponse.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public void syncKnowledgeSources(SyncKnowledgeSourcesRequest request) { + String path = String.format("/api/2.1/%s/knowledge-sources:sync", request.getName()); + try { + Request req = new Request("POST", path, apiClient.serialize(request)); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + req.withHeader("Content-Type", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Org-Id", apiClient.workspaceId()); + } + apiClient.execute(req, Void.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public KnowledgeAssistant updateKnowledgeAssistant(UpdateKnowledgeAssistantRequest request) { + String path = String.format("/api/2.1/%s", request.getName()); + try { + Request req = + new Request("PATCH", path, apiClient.serialize(request.getKnowledgeAssistant())); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + req.withHeader("Content-Type", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Org-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, KnowledgeAssistant.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } + + @Override + public KnowledgeSource updateKnowledgeSource(UpdateKnowledgeSourceRequest request) { + String path = String.format("/api/2.1/%s", request.getName()); + try { + Request req = new Request("PATCH", path, apiClient.serialize(request.getKnowledgeSource())); + + ApiClient.setQuery(req, request); + req.withHeader("Accept", "application/json"); + req.withHeader("Content-Type", "application/json"); + if (apiClient.workspaceId() != null) { + req.withHeader("X-Databricks-Org-Id", apiClient.workspaceId()); + } + return apiClient.execute(req, KnowledgeSource.class); + } catch (IOException e) { + throw new DatabricksException("IO error: " + e.getMessage(), e); + } + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsService.java new file mode 100755 index 000000000..cb70f6b10 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeAssistantsService.java @@ -0,0 +1,55 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.support.Generated; + +/** + * Manage Knowledge Assistants and related resources. + * + *

This is the high-level interface, that contains generated methods. + * + *

Evolving: this interface is under development. Method signatures may change. + */ +@Generated +public interface KnowledgeAssistantsService { + /** Creates a Knowledge Assistant. */ + KnowledgeAssistant createKnowledgeAssistant( + CreateKnowledgeAssistantRequest createKnowledgeAssistantRequest); + + /** Creates a Knowledge Source under a Knowledge Assistant. */ + KnowledgeSource createKnowledgeSource(CreateKnowledgeSourceRequest createKnowledgeSourceRequest); + + /** Deletes a Knowledge Assistant. */ + void deleteKnowledgeAssistant(DeleteKnowledgeAssistantRequest deleteKnowledgeAssistantRequest); + + /** Deletes a Knowledge Source. */ + void deleteKnowledgeSource(DeleteKnowledgeSourceRequest deleteKnowledgeSourceRequest); + + /** Gets a Knowledge Assistant. */ + KnowledgeAssistant getKnowledgeAssistant( + GetKnowledgeAssistantRequest getKnowledgeAssistantRequest); + + /** Gets a Knowledge Source. */ + KnowledgeSource getKnowledgeSource(GetKnowledgeSourceRequest getKnowledgeSourceRequest); + + /** List Knowledge Assistants */ + ListKnowledgeAssistantsResponse listKnowledgeAssistants( + ListKnowledgeAssistantsRequest listKnowledgeAssistantsRequest); + + /** Lists Knowledge Sources under a Knowledge Assistant. */ + ListKnowledgeSourcesResponse listKnowledgeSources( + ListKnowledgeSourcesRequest listKnowledgeSourcesRequest); + + /** + * Sync all non-index Knowledge Sources for a Knowledge Assistant (index sources do not require + * sync) + */ + void syncKnowledgeSources(SyncKnowledgeSourcesRequest syncKnowledgeSourcesRequest); + + /** Updates a Knowledge Assistant. */ + KnowledgeAssistant updateKnowledgeAssistant( + UpdateKnowledgeAssistantRequest updateKnowledgeAssistantRequest); + + /** Updates a Knowledge Source. */ + KnowledgeSource updateKnowledgeSource(UpdateKnowledgeSourceRequest updateKnowledgeSourceRequest); +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSource.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSource.java new file mode 100755 index 000000000..705511930 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSource.java @@ -0,0 +1,227 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import com.google.protobuf.Timestamp; +import java.util.Objects; + +/** + * KnowledgeSource represents a source of knowledge for the KnowledgeAssistant. Used in + * create/update requests and returned in Get/List responses. Note: REQUIRED annotations below + * represent create-time requirements. For updates, required fields are determined by the update + * mask. + */ +@Generated +public class KnowledgeSource { + /** Timestamp when this knowledge source was created. */ + @JsonProperty("create_time") + private Timestamp createTime; + + /** + * Description of the knowledge source. Required when creating a Knowledge Source. When updating a + * Knowledge Source, optional unless included in update_mask. + */ + @JsonProperty("description") + private String description; + + /** + * Human-readable display name of the knowledge source. Required when creating a Knowledge Source. + * When updating a Knowledge Source, optional unless included in update_mask. + */ + @JsonProperty("display_name") + private String displayName; + + /** */ + @JsonProperty("file_table") + private FileTableSpec fileTable; + + /** */ + @JsonProperty("files") + private FilesSpec files; + + /** */ + @JsonProperty("id") + private String id; + + /** */ + @JsonProperty("index") + private IndexSpec index; + + /** + * Timestamp representing the cutoff before which content in this knowledge source is being + * ingested. + */ + @JsonProperty("knowledge_cutoff_time") + private Timestamp knowledgeCutoffTime; + + /** + * Full resource name: + * knowledge-assistants/{knowledge_assistant_id}/knowledge-sources/{knowledge_source_id} + */ + @JsonProperty("name") + private String name; + + /** + * The type of the source: "index", "files", or "file_table". Required when creating a Knowledge + * Source. When updating a Knowledge Source, this field is ignored. + */ + @JsonProperty("source_type") + private String sourceType; + + /** */ + @JsonProperty("state") + private KnowledgeSourceState state; + + public KnowledgeSource setCreateTime(Timestamp createTime) { + this.createTime = createTime; + return this; + } + + public Timestamp getCreateTime() { + return createTime; + } + + public KnowledgeSource setDescription(String description) { + this.description = description; + return this; + } + + public String getDescription() { + return description; + } + + public KnowledgeSource setDisplayName(String displayName) { + this.displayName = displayName; + return this; + } + + public String getDisplayName() { + return displayName; + } + + public KnowledgeSource setFileTable(FileTableSpec fileTable) { + this.fileTable = fileTable; + return this; + } + + public FileTableSpec getFileTable() { + return fileTable; + } + + public KnowledgeSource setFiles(FilesSpec files) { + this.files = files; + return this; + } + + public FilesSpec getFiles() { + return files; + } + + public KnowledgeSource setId(String id) { + this.id = id; + return this; + } + + public String getId() { + return id; + } + + public KnowledgeSource setIndex(IndexSpec index) { + this.index = index; + return this; + } + + public IndexSpec getIndex() { + return index; + } + + public KnowledgeSource setKnowledgeCutoffTime(Timestamp knowledgeCutoffTime) { + this.knowledgeCutoffTime = knowledgeCutoffTime; + return this; + } + + public Timestamp getKnowledgeCutoffTime() { + return knowledgeCutoffTime; + } + + public KnowledgeSource setName(String name) { + this.name = name; + return this; + } + + public String getName() { + return name; + } + + public KnowledgeSource setSourceType(String sourceType) { + this.sourceType = sourceType; + return this; + } + + public String getSourceType() { + return sourceType; + } + + public KnowledgeSource setState(KnowledgeSourceState state) { + this.state = state; + return this; + } + + public KnowledgeSourceState getState() { + return state; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + KnowledgeSource that = (KnowledgeSource) o; + return Objects.equals(createTime, that.createTime) + && Objects.equals(description, that.description) + && Objects.equals(displayName, that.displayName) + && Objects.equals(fileTable, that.fileTable) + && Objects.equals(files, that.files) + && Objects.equals(id, that.id) + && Objects.equals(index, that.index) + && Objects.equals(knowledgeCutoffTime, that.knowledgeCutoffTime) + && Objects.equals(name, that.name) + && Objects.equals(sourceType, that.sourceType) + && Objects.equals(state, that.state); + } + + @Override + public int hashCode() { + return Objects.hash( + createTime, + description, + displayName, + fileTable, + files, + id, + index, + knowledgeCutoffTime, + name, + sourceType, + state); + } + + @Override + public String toString() { + return new ToStringer(KnowledgeSource.class) + .add("createTime", createTime) + .add("description", description) + .add("displayName", displayName) + .add("fileTable", fileTable) + .add("files", files) + .add("id", id) + .add("index", index) + .add("knowledgeCutoffTime", knowledgeCutoffTime) + .add("name", name) + .add("sourceType", sourceType) + .add("state", state) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSourceState.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSourceState.java new file mode 100755 index 000000000..f831c5d19 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/KnowledgeSourceState.java @@ -0,0 +1,12 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +import com.databricks.sdk.support.Generated; + +@Generated +public enum KnowledgeSourceState { + FAILED_UPDATE, + UPDATED, + UPDATING, +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsRequest.java new file mode 100755 index 000000000..228407b6f --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsRequest.java @@ -0,0 +1,67 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 ListKnowledgeAssistantsRequest { + /** + * The maximum number of knowledge assistants to return. If unspecified, at most 100 knowledge + * assistants will be returned. The maximum value is 100; values above 100 will be coerced to 100. + */ + @JsonIgnore + @QueryParam("page_size") + private Long pageSize; + + /** + * A page token, received from a previous `ListKnowledgeAssistants` call. Provide this to retrieve + * the subsequent page. If unspecified, the first page will be returned. + */ + @JsonIgnore + @QueryParam("page_token") + private String pageToken; + + public ListKnowledgeAssistantsRequest setPageSize(Long pageSize) { + this.pageSize = pageSize; + return this; + } + + public Long getPageSize() { + return pageSize; + } + + public ListKnowledgeAssistantsRequest 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; + ListKnowledgeAssistantsRequest that = (ListKnowledgeAssistantsRequest) o; + return Objects.equals(pageSize, that.pageSize) && Objects.equals(pageToken, that.pageToken); + } + + @Override + public int hashCode() { + return Objects.hash(pageSize, pageToken); + } + + @Override + public String toString() { + return new ToStringer(ListKnowledgeAssistantsRequest.class) + .add("pageSize", pageSize) + .add("pageToken", pageToken) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsResponse.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsResponse.java new file mode 100755 index 000000000..ec55ca4b4 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeAssistantsResponse.java @@ -0,0 +1,65 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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; + +/** A list of Knowledge Assistants. */ +@Generated +public class ListKnowledgeAssistantsResponse { + /** */ + @JsonProperty("knowledge_assistants") + private Collection knowledgeAssistants; + + /** + * A token that can be sent as `page_token` to retrieve the next page. If this field is omitted, + * there are no subsequent pages. + */ + @JsonProperty("next_page_token") + private String nextPageToken; + + public ListKnowledgeAssistantsResponse setKnowledgeAssistants( + Collection knowledgeAssistants) { + this.knowledgeAssistants = knowledgeAssistants; + return this; + } + + public Collection getKnowledgeAssistants() { + return knowledgeAssistants; + } + + public ListKnowledgeAssistantsResponse setNextPageToken(String nextPageToken) { + this.nextPageToken = nextPageToken; + return this; + } + + public String getNextPageToken() { + return nextPageToken; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + ListKnowledgeAssistantsResponse that = (ListKnowledgeAssistantsResponse) o; + return Objects.equals(knowledgeAssistants, that.knowledgeAssistants) + && Objects.equals(nextPageToken, that.nextPageToken); + } + + @Override + public int hashCode() { + return Objects.hash(knowledgeAssistants, nextPageToken); + } + + @Override + public String toString() { + return new ToStringer(ListKnowledgeAssistantsResponse.class) + .add("knowledgeAssistants", knowledgeAssistants) + .add("nextPageToken", nextPageToken) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesRequest.java new file mode 100755 index 000000000..9488a1268 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesRequest.java @@ -0,0 +1,76 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 ListKnowledgeSourcesRequest { + /** */ + @JsonIgnore + @QueryParam("page_size") + private Long pageSize; + + /** */ + @JsonIgnore + @QueryParam("page_token") + private String pageToken; + + /** Parent resource to list from. Format: knowledge-assistants/{knowledge_assistant_id} */ + @JsonIgnore private String parent; + + public ListKnowledgeSourcesRequest setPageSize(Long pageSize) { + this.pageSize = pageSize; + return this; + } + + public Long getPageSize() { + return pageSize; + } + + public ListKnowledgeSourcesRequest setPageToken(String pageToken) { + this.pageToken = pageToken; + return this; + } + + public String getPageToken() { + return pageToken; + } + + public ListKnowledgeSourcesRequest setParent(String parent) { + this.parent = parent; + return this; + } + + public String getParent() { + return parent; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + ListKnowledgeSourcesRequest that = (ListKnowledgeSourcesRequest) o; + return Objects.equals(pageSize, that.pageSize) + && Objects.equals(pageToken, that.pageToken) + && Objects.equals(parent, that.parent); + } + + @Override + public int hashCode() { + return Objects.hash(pageSize, pageToken, parent); + } + + @Override + public String toString() { + return new ToStringer(ListKnowledgeSourcesRequest.class) + .add("pageSize", pageSize) + .add("pageToken", pageToken) + .add("parent", parent) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesResponse.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesResponse.java new file mode 100755 index 000000000..e4cb8b24d --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/ListKnowledgeSourcesResponse.java @@ -0,0 +1,61 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 ListKnowledgeSourcesResponse { + /** */ + @JsonProperty("knowledge_sources") + private Collection knowledgeSources; + + /** */ + @JsonProperty("next_page_token") + private String nextPageToken; + + public ListKnowledgeSourcesResponse setKnowledgeSources( + Collection knowledgeSources) { + this.knowledgeSources = knowledgeSources; + return this; + } + + public Collection getKnowledgeSources() { + return knowledgeSources; + } + + public ListKnowledgeSourcesResponse setNextPageToken(String nextPageToken) { + this.nextPageToken = nextPageToken; + return this; + } + + public String getNextPageToken() { + return nextPageToken; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + ListKnowledgeSourcesResponse that = (ListKnowledgeSourcesResponse) o; + return Objects.equals(knowledgeSources, that.knowledgeSources) + && Objects.equals(nextPageToken, that.nextPageToken); + } + + @Override + public int hashCode() { + return Objects.hash(knowledgeSources, nextPageToken); + } + + @Override + public String toString() { + return new ToStringer(ListKnowledgeSourcesResponse.class) + .add("knowledgeSources", knowledgeSources) + .add("nextPageToken", nextPageToken) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/SyncKnowledgeSourcesRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/SyncKnowledgeSourcesRequest.java new file mode 100755 index 000000000..4cd8ab302 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/SyncKnowledgeSourcesRequest.java @@ -0,0 +1,44 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 SyncKnowledgeSourcesRequest { + /** + * The resource name of the Knowledge Assistant. Format: + * knowledge-assistants/{knowledge_assistant_id} + */ + @JsonIgnore private String name; + + public SyncKnowledgeSourcesRequest 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; + SyncKnowledgeSourcesRequest that = (SyncKnowledgeSourcesRequest) o; + return Objects.equals(name, that.name); + } + + @Override + public int hashCode() { + return Objects.hash(name); + } + + @Override + public String toString() { + return new ToStringer(SyncKnowledgeSourcesRequest.class).add("name", name).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeAssistantRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeAssistantRequest.java new file mode 100755 index 000000000..f2ef10436 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeAssistantRequest.java @@ -0,0 +1,89 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 com.fasterxml.jackson.annotation.JsonProperty; +import com.google.protobuf.FieldMask; +import java.util.Objects; + +@Generated +public class UpdateKnowledgeAssistantRequest { + /** + * The Knowledge Assistant update payload. Only fields listed in update_mask are updated. REQUIRED + * annotations on Knowledge Assistant fields describe create-time requirements and do not mean all + * those fields are required for update. + */ + @JsonProperty("knowledge_assistant") + private KnowledgeAssistant knowledgeAssistant; + + /** + * The resource name of the Knowledge Assistant. Format: + * knowledge-assistants/{knowledge_assistant_id} + */ + @JsonIgnore private String name; + + /** + * Comma-delimited list of fields to update on the Knowledge Assistant. Allowed values: + * `display_name`, `description`, `instructions`. Examples: - `display_name` - + * `description,instructions` + */ + @JsonIgnore + @QueryParam("update_mask") + private FieldMask updateMask; + + public UpdateKnowledgeAssistantRequest setKnowledgeAssistant( + KnowledgeAssistant knowledgeAssistant) { + this.knowledgeAssistant = knowledgeAssistant; + return this; + } + + public KnowledgeAssistant getKnowledgeAssistant() { + return knowledgeAssistant; + } + + public UpdateKnowledgeAssistantRequest setName(String name) { + this.name = name; + return this; + } + + public String getName() { + return name; + } + + public UpdateKnowledgeAssistantRequest setUpdateMask(FieldMask updateMask) { + this.updateMask = updateMask; + return this; + } + + public FieldMask getUpdateMask() { + return updateMask; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + UpdateKnowledgeAssistantRequest that = (UpdateKnowledgeAssistantRequest) o; + return Objects.equals(knowledgeAssistant, that.knowledgeAssistant) + && Objects.equals(name, that.name) + && Objects.equals(updateMask, that.updateMask); + } + + @Override + public int hashCode() { + return Objects.hash(knowledgeAssistant, name, updateMask); + } + + @Override + public String toString() { + return new ToStringer(UpdateKnowledgeAssistantRequest.class) + .add("knowledgeAssistant", knowledgeAssistant) + .add("name", name) + .add("updateMask", updateMask) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeSourceRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeSourceRequest.java new file mode 100755 index 000000000..1750025e8 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/knowledgeassistants/UpdateKnowledgeSourceRequest.java @@ -0,0 +1,87 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.knowledgeassistants; + +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 com.fasterxml.jackson.annotation.JsonProperty; +import com.google.protobuf.FieldMask; +import java.util.Objects; + +@Generated +public class UpdateKnowledgeSourceRequest { + /** + * The Knowledge Source update payload. Only fields listed in update_mask are updated. REQUIRED + * annotations on Knowledge Source fields describe create-time requirements and do not mean all + * those fields are required for update. + */ + @JsonProperty("knowledge_source") + private KnowledgeSource knowledgeSource; + + /** + * The resource name of the Knowledge Source to update. Format: + * knowledge-assistants/{knowledge_assistant_id}/knowledge-sources/{knowledge_source_id} + */ + @JsonIgnore private String name; + + /** + * Comma-delimited list of fields to update on the Knowledge Source. Allowed values: + * `display_name`, `description`. Examples: - `display_name` - `display_name,description` + */ + @JsonIgnore + @QueryParam("update_mask") + private FieldMask updateMask; + + public UpdateKnowledgeSourceRequest setKnowledgeSource(KnowledgeSource knowledgeSource) { + this.knowledgeSource = knowledgeSource; + return this; + } + + public KnowledgeSource getKnowledgeSource() { + return knowledgeSource; + } + + public UpdateKnowledgeSourceRequest setName(String name) { + this.name = name; + return this; + } + + public String getName() { + return name; + } + + public UpdateKnowledgeSourceRequest setUpdateMask(FieldMask updateMask) { + this.updateMask = updateMask; + return this; + } + + public FieldMask getUpdateMask() { + return updateMask; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + UpdateKnowledgeSourceRequest that = (UpdateKnowledgeSourceRequest) o; + return Objects.equals(knowledgeSource, that.knowledgeSource) + && Objects.equals(name, that.name) + && Objects.equals(updateMask, that.updateMask); + } + + @Override + public int hashCode() { + return Objects.hash(knowledgeSource, name, updateMask); + } + + @Override + public String toString() { + return new ToStringer(UpdateKnowledgeSourceRequest.class) + .add("knowledgeSource", knowledgeSource) + .add("name", name) + .add("updateMask", updateMask) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/DeleteTagAssignmentRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/DeleteTagAssignmentRequest.java index c6d63c761..a4dd8def2 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/DeleteTagAssignmentRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/DeleteTagAssignmentRequest.java @@ -17,7 +17,7 @@ public class DeleteTagAssignmentRequest { /** * The type of entity to which the tag is assigned. Allowed values are apps, dashboards, - * geniespaces + * geniespaces, notebooks */ @JsonIgnore private String entityType; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/GetTagAssignmentRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/GetTagAssignmentRequest.java index dc1707406..05227b7e1 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/GetTagAssignmentRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/GetTagAssignmentRequest.java @@ -17,7 +17,7 @@ public class GetTagAssignmentRequest { /** * The type of entity to which the tag is assigned. Allowed values are apps, dashboards, - * geniespaces + * geniespaces, notebooks */ @JsonIgnore private String entityType; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/ListTagAssignmentsRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/ListTagAssignmentsRequest.java index 103a7b10c..926202986 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/ListTagAssignmentsRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/ListTagAssignmentsRequest.java @@ -18,7 +18,7 @@ public class ListTagAssignmentsRequest { /** * The type of entity to which the tag is assigned. Allowed values are apps, dashboards, - * geniespaces + * geniespaces, notebooks */ @JsonIgnore private String entityType; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/TagAssignment.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/TagAssignment.java index 10b567652..2cddc0d88 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/TagAssignment.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/TagAssignment.java @@ -18,7 +18,7 @@ public class TagAssignment { /** * The type of entity to which the tag is assigned. Allowed values are apps, dashboards, - * geniespaces + * geniespaces, notebooks */ @JsonProperty("entity_type") private String entityType; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/UpdateTagAssignmentRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/UpdateTagAssignmentRequest.java index f25f26f9b..21f6e0c33 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/UpdateTagAssignmentRequest.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/tags/UpdateTagAssignmentRequest.java @@ -19,7 +19,7 @@ public class UpdateTagAssignmentRequest { /** * The type of entity to which the tag is assigned. Allowed values are apps, dashboards, - * geniespaces + * geniespaces, notebooks */ @JsonIgnore private String entityType;