diff --git a/NEXT_CHANGELOG.md b/NEXT_CHANGELOG.md index c17546bd3..15e42e4f9 100644 --- a/NEXT_CHANGELOG.md +++ b/NEXT_CHANGELOG.md @@ -12,4 +12,6 @@ ### Internal Changes +* Switch workspace addressing header on workspace-scoped API calls from `X-Databricks-Org-Id` to `X-Databricks-Workspace-Id`. The value continues to come from `Config.WorkspaceID` (`DATABRICKS_WORKSPACE_ID`), and now accepts either a classic numeric workspace ID or a CPDR connection ID (server disambiguates). Part of the DECO + Unified Workspace Addressing initiative. + ### API Changes diff --git a/config/config.go b/config/config.go index 768576123..0c148d56c 100644 --- a/config/config.go +++ b/config/config.go @@ -123,7 +123,9 @@ type Config struct { // Databricks Account ID for Accounts API. This field is used in dependencies. AccountID string `name:"account_id" env:"DATABRICKS_ACCOUNT_ID"` - // Databricks Workspace ID for Workspace clients when working with unified hosts + // Databricks Workspace ID for Workspace clients when working with unified hosts. + // Accepts either a classic numeric workspace ID or a CPDR connection ID; the + // server disambiguates. WorkspaceID string `name:"workspace_id" env:"DATABRICKS_WORKSPACE_ID"` Token string `name:"token" env:"DATABRICKS_TOKEN" auth:"pat,sensitive"` diff --git a/internal/testspecs/service/httpcallv2/impl.go b/internal/testspecs/service/httpcallv2/impl.go index 6f6f3c3a9..69cedd765 100755 --- a/internal/testspecs/service/httpcallv2/impl.go +++ b/internal/testspecs/service/httpcallv2/impl.go @@ -27,7 +27,7 @@ func (a *httpCallV2Impl) CreateResource(ctx context.Context, request CreateResou headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &resource) return &resource, err @@ -41,7 +41,7 @@ func (a *httpCallV2Impl) GetResource(ctx context.Context, request GetResourceReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &resource) return &resource, err @@ -89,7 +89,7 @@ func (a *httpCallV2Impl) UpdateResource(ctx context.Context, request UpdateResou headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Resource, &resource) return &resource, err diff --git a/internal/testspecs/service/idempotencytesting/impl.go b/internal/testspecs/service/idempotencytesting/impl.go index fe14d6e53..a7cde4d8f 100755 --- a/internal/testspecs/service/idempotencytesting/impl.go +++ b/internal/testspecs/service/idempotencytesting/impl.go @@ -32,7 +32,7 @@ func (a *idempotencyTestingImpl) CreateTestResource(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.TestResource, &testResource) return &testResource, err diff --git a/internal/testspecs/service/jsonmarshallv2/impl.go b/internal/testspecs/service/jsonmarshallv2/impl.go index d5097eb0f..1b6451482 100755 --- a/internal/testspecs/service/jsonmarshallv2/impl.go +++ b/internal/testspecs/service/jsonmarshallv2/impl.go @@ -23,7 +23,7 @@ func (a *jsonMarshallV2Impl) GetResource(ctx context.Context, request GetResourc headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &resource) return &resource, err diff --git a/internal/testspecs/service/lrotesting/impl.go b/internal/testspecs/service/lrotesting/impl.go index 92c657f35..f7f3075f9 100755 --- a/internal/testspecs/service/lrotesting/impl.go +++ b/internal/testspecs/service/lrotesting/impl.go @@ -22,7 +22,7 @@ func (a *lroTestingImpl) CancelOperation(ctx context.Context, request CancelOper headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, nil) return err @@ -37,7 +37,7 @@ func (a *lroTestingImpl) CreateTestResource(ctx context.Context, request CreateT headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Resource, &operation) return &operation, err @@ -51,7 +51,7 @@ func (a *lroTestingImpl) DeleteTestResource(ctx context.Context, request DeleteT headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -65,7 +65,7 @@ func (a *lroTestingImpl) GetOperation(ctx context.Context, request GetOperationR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &operation) return &operation, err @@ -79,7 +79,7 @@ func (a *lroTestingImpl) GetTestResource(ctx context.Context, request GetTestRes headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &testResource) return &testResource, err diff --git a/internal/unified_host_test.go b/internal/unified_host_test.go new file mode 100644 index 000000000..4eb5f5214 --- /dev/null +++ b/internal/unified_host_test.go @@ -0,0 +1,100 @@ +package internal + +import ( + "context" + "net/http" + "testing" + + "github.com/databricks/databricks-sdk-go" + "github.com/databricks/databricks-sdk-go/config" + "github.com/databricks/databricks-sdk-go/service/iam" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/require" +) + +// captureHeadersTransport snapshots the request and response headers for a +// single matching URL path and lets every other request through unchanged. +type captureHeadersTransport struct { + inner http.RoundTripper + path string + + reqHdr http.Header + respHdr http.Header +} + +func (t *captureHeadersTransport) RoundTrip(req *http.Request) (*http.Response, error) { + matched := req.URL.Path == t.path + if matched { + t.reqHdr = req.Header.Clone() + } + resp, err := t.inner.RoundTrip(req) + if matched && resp != nil { + t.respHdr = resp.Header.Clone() + } + return resp, err +} + +// TestAccUnifiedHostSendsWorkspaceIdHeader verifies, against a real unified +// (SPOG) host, that: +// 1. the SDK sends the new X-Databricks-Workspace-Id routing header on +// workspace-scoped requests, and no longer sends the legacy +// X-Databricks-Org-Id, and +// 2. the server still echoes the workspace ID on the legacy +// X-Databricks-Org-Id response header — the response-side migration has +// not yet happened, which is why CurrentWorkspaceID still reads the +// legacy name in workspace_functions.go. +// +// Uses OAuth M2M auth (ClientID/ClientSecret) and probes the SCIM Me endpoint +// as the workspace-scoped call. A transport-level header capture is hooked in +// to verify which routing header actually crosses the wire. +// +// Runs against the workspace test environment (TEST_ENVIRONMENT_TYPE in +// {WORKSPACE, UC_WORKSPACE}). Requires UNIFIED_HOST, THIS_WORKSPACE_ID, +// TEST_ACCOUNT_ID, DATABRICKS_CLIENT_ID, and DATABRICKS_CLIENT_SECRET in the +// environment. +func TestAccUnifiedHostSendsWorkspaceIdHeader(t *testing.T) { + loadDebugEnvIfRunsFromIDE(t, "workspace") + if envType := testEnvironmentType(); envType != "" && envType != "WORKSPACE" && envType != "UC_WORKSPACE" { + skipf(t)("Skipping workspace test: TEST_ENVIRONMENT_TYPE=%s", envType) + } + + const apiPath = "/api/2.0/preview/scim/v2/Me" + transport := &captureHeadersTransport{ + inner: http.DefaultTransport, + path: apiPath, + } + workspaceID := GetEnvOrSkipTest(t, "THIS_WORKSPACE_ID") + cfg := &config.Config{ + Host: GetEnvOrSkipTest(t, "UNIFIED_HOST"), + AccountID: GetEnvOrSkipTest(t, "TEST_ACCOUNT_ID"), + ClientID: GetEnvOrSkipTest(t, "DATABRICKS_CLIENT_ID"), + ClientSecret: GetEnvOrSkipTest(t, "DATABRICKS_CLIENT_SECRET"), + WorkspaceID: workspaceID, + HTTPTransport: transport, + HTTPTimeoutSeconds: 60, + } + if err := cfg.EnsureResolved(); err != nil { + skipf(t)("config resolve: %s", err) + } + t.Parallel() + + ctx := context.Background() + w := databricks.Must(databricks.NewWorkspaceClient((*databricks.Config)(cfg))) + + // (1) workspace-scoped probe against the unified host: must succeed. + me, err := w.CurrentUser.Me(ctx, iam.MeRequest{}) + require.NoError(t, err, "workspace-scoped SCIM Me should pass through the unified host gateway with the new routing header") + assert.NotEmpty(t, me.UserName, "Me should return a non-empty UserName") + + // (2) On the request side, the SDK must send the new header and not the legacy one. + require.NotNil(t, transport.reqHdr, "transport did not observe a request to %s", apiPath) + assert.Equal(t, workspaceID, transport.reqHdr.Get("X-Databricks-Workspace-Id"), + "SDK must send X-Databricks-Workspace-Id with Config.WorkspaceID") + assert.Empty(t, transport.reqHdr.Get("X-Databricks-Org-Id"), + "SDK must no longer send the legacy X-Databricks-Org-Id request header") + + // (3) On the response side, the server still echoes the legacy header name during the migration. + require.NotNil(t, transport.respHdr, "transport did not observe a response from %s", apiPath) + assert.NotEmpty(t, transport.respHdr.Get("X-Databricks-Org-Id"), + "server is expected to still echo the legacy X-Databricks-Org-Id response header during the migration") +} diff --git a/service/agentbricks/impl.go b/service/agentbricks/impl.go index 15b58cb2f..b482126d9 100755 --- a/service/agentbricks/impl.go +++ b/service/agentbricks/impl.go @@ -23,7 +23,7 @@ func (a *agentBricksImpl) CancelOptimize(ctx context.Context, request CancelCust headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -38,7 +38,7 @@ func (a *agentBricksImpl) CreateCustomLlm(ctx context.Context, request CreateCus headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &customLlm) return &customLlm, err @@ -51,7 +51,7 @@ func (a *agentBricksImpl) DeleteCustomLlm(ctx context.Context, request DeleteCus headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -65,7 +65,7 @@ func (a *agentBricksImpl) GetCustomLlm(ctx context.Context, request GetCustomLlm headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &customLlm) return &customLlm, err @@ -80,7 +80,7 @@ func (a *agentBricksImpl) StartOptimize(ctx context.Context, request StartCustom headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &customLlm) return &customLlm, err @@ -95,7 +95,7 @@ func (a *agentBricksImpl) UpdateCustomLlm(ctx context.Context, request UpdateCus headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &customLlm) return &customLlm, err diff --git a/service/apps/impl.go b/service/apps/impl.go index c91f5e53c..e81e14769 100755 --- a/service/apps/impl.go +++ b/service/apps/impl.go @@ -33,7 +33,7 @@ func (a *appsImpl) Create(ctx context.Context, request CreateAppRequest) (*App, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.App, &app) return &app, err @@ -48,7 +48,7 @@ func (a *appsImpl) CreateSpace(ctx context.Context, request CreateSpaceRequest) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Space, &operation) return &operation, err @@ -63,7 +63,7 @@ func (a *appsImpl) CreateUpdate(ctx context.Context, request AsyncUpdateAppReque headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &appUpdate) return &appUpdate, err @@ -77,7 +77,7 @@ func (a *appsImpl) Delete(ctx context.Context, request DeleteAppRequest) (*App, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &app) return &app, err @@ -90,7 +90,7 @@ func (a *appsImpl) DeleteAppThumbnail(ctx context.Context, request DeleteAppThum headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -104,7 +104,7 @@ func (a *appsImpl) DeleteSpace(ctx context.Context, request DeleteSpaceRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -119,7 +119,7 @@ func (a *appsImpl) Deploy(ctx context.Context, request CreateAppDeploymentReques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.AppDeployment, &appDeployment) return &appDeployment, err @@ -133,7 +133,7 @@ func (a *appsImpl) Get(ctx context.Context, request GetAppRequest) (*App, error) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &app) return &app, err @@ -147,7 +147,7 @@ func (a *appsImpl) GetDeployment(ctx context.Context, request GetAppDeploymentRe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &appDeployment) return &appDeployment, err @@ -161,7 +161,7 @@ func (a *appsImpl) GetPermissionLevels(ctx context.Context, request GetAppPermis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getAppPermissionLevelsResponse) return &getAppPermissionLevelsResponse, err @@ -175,7 +175,7 @@ func (a *appsImpl) GetPermissions(ctx context.Context, request GetAppPermissions headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &appPermissions) return &appPermissions, err @@ -189,7 +189,7 @@ func (a *appsImpl) GetSpace(ctx context.Context, request GetSpaceRequest) (*Spac headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &space) return &space, err @@ -203,7 +203,7 @@ func (a *appsImpl) GetSpaceOperation(ctx context.Context, request GetOperationRe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &operation) return &operation, err @@ -217,7 +217,7 @@ func (a *appsImpl) GetUpdate(ctx context.Context, request GetAppUpdateRequest) ( headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &appUpdate) return &appUpdate, err @@ -262,7 +262,7 @@ func (a *appsImpl) internalList(ctx context.Context, request ListAppsRequest) (* headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listAppsResponse) return &listAppsResponse, err @@ -307,7 +307,7 @@ func (a *appsImpl) internalListDeployments(ctx context.Context, request ListAppD headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listAppDeploymentsResponse) return &listAppDeploymentsResponse, err @@ -352,7 +352,7 @@ func (a *appsImpl) internalListSpaces(ctx context.Context, request ListSpacesReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSpacesResponse) return &listSpacesResponse, err @@ -367,7 +367,7 @@ func (a *appsImpl) SetPermissions(ctx context.Context, request AppPermissionsReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &appPermissions) return &appPermissions, err @@ -382,7 +382,7 @@ func (a *appsImpl) Start(ctx context.Context, request StartAppRequest) (*App, er headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &app) return &app, err @@ -397,7 +397,7 @@ func (a *appsImpl) Stop(ctx context.Context, request StopAppRequest) (*App, erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &app) return &app, err @@ -412,7 +412,7 @@ func (a *appsImpl) Update(ctx context.Context, request UpdateAppRequest) (*App, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.App, &app) return &app, err @@ -427,7 +427,7 @@ func (a *appsImpl) UpdateAppThumbnail(ctx context.Context, request UpdateAppThum headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &appThumbnail) return &appThumbnail, err @@ -442,7 +442,7 @@ func (a *appsImpl) UpdatePermissions(ctx context.Context, request AppPermissions headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &appPermissions) return &appPermissions, err @@ -464,7 +464,7 @@ func (a *appsImpl) UpdateSpace(ctx context.Context, request UpdateSpaceRequest) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Space, &operation) return &operation, err @@ -484,7 +484,7 @@ func (a *appsSettingsImpl) CreateCustomTemplate(ctx context.Context, request Cre headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Template, &customTemplate) return &customTemplate, err @@ -498,7 +498,7 @@ func (a *appsSettingsImpl) DeleteCustomTemplate(ctx context.Context, request Del headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &customTemplate) return &customTemplate, err @@ -512,7 +512,7 @@ func (a *appsSettingsImpl) GetCustomTemplate(ctx context.Context, request GetCus headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &customTemplate) return &customTemplate, err @@ -557,7 +557,7 @@ func (a *appsSettingsImpl) internalListCustomTemplates(ctx context.Context, requ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCustomTemplatesResponse) return &listCustomTemplatesResponse, err @@ -572,7 +572,7 @@ func (a *appsSettingsImpl) UpdateCustomTemplate(ctx context.Context, request Upd headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request.Template, &customTemplate) return &customTemplate, err diff --git a/service/bundle/impl.go b/service/bundle/impl.go index 6997117a1..45f92c0ff 100755 --- a/service/bundle/impl.go +++ b/service/bundle/impl.go @@ -26,7 +26,7 @@ func (a *bundleImpl) CompleteVersion(ctx context.Context, request CompleteVersio headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &version) return &version, err @@ -45,7 +45,7 @@ func (a *bundleImpl) CreateDeployment(ctx context.Context, request CreateDeploym headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Deployment, &deployment) return &deployment, err @@ -64,7 +64,7 @@ func (a *bundleImpl) CreateOperation(ctx context.Context, request CreateOperatio headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Operation, &operation) return &operation, err @@ -83,7 +83,7 @@ func (a *bundleImpl) CreateVersion(ctx context.Context, request CreateVersionReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Version, &version) return &version, err @@ -96,7 +96,7 @@ func (a *bundleImpl) DeleteDeployment(ctx context.Context, request DeleteDeploym headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -110,7 +110,7 @@ func (a *bundleImpl) GetDeployment(ctx context.Context, request GetDeploymentReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &deployment) return &deployment, err @@ -124,7 +124,7 @@ func (a *bundleImpl) GetOperation(ctx context.Context, request GetOperationReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &operation) return &operation, err @@ -138,7 +138,7 @@ func (a *bundleImpl) GetResource(ctx context.Context, request GetResourceRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &resource) return &resource, err @@ -152,7 +152,7 @@ func (a *bundleImpl) GetVersion(ctx context.Context, request GetVersionRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &version) return &version, err @@ -167,7 +167,7 @@ func (a *bundleImpl) Heartbeat(ctx context.Context, request HeartbeatRequest) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &heartbeatResponse) return &heartbeatResponse, err @@ -212,7 +212,7 @@ func (a *bundleImpl) internalListDeployments(ctx context.Context, request ListDe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listDeploymentsResponse) return &listDeploymentsResponse, err @@ -257,7 +257,7 @@ func (a *bundleImpl) internalListOperations(ctx context.Context, request ListOpe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listOperationsResponse) return &listOperationsResponse, err @@ -302,7 +302,7 @@ func (a *bundleImpl) internalListResources(ctx context.Context, request ListReso headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listResourcesResponse) return &listResourcesResponse, err @@ -349,7 +349,7 @@ func (a *bundleImpl) internalListVersions(ctx context.Context, request ListVersi headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listVersionsResponse) return &listVersionsResponse, err diff --git a/service/catalog/impl.go b/service/catalog/impl.go index 4bc343af4..16e5d1d89 100755 --- a/service/catalog/impl.go +++ b/service/catalog/impl.go @@ -294,7 +294,7 @@ func (a *artifactAllowlistsImpl) Get(ctx context.Context, request GetArtifactAll headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &artifactAllowlistInfo) return &artifactAllowlistInfo, err @@ -309,7 +309,7 @@ func (a *artifactAllowlistsImpl) Update(ctx context.Context, request SetArtifact headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &artifactAllowlistInfo) return &artifactAllowlistInfo, err @@ -329,7 +329,7 @@ func (a *catalogsImpl) Create(ctx context.Context, request CreateCatalog) (*Cata headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &catalogInfo) return &catalogInfo, err @@ -342,7 +342,7 @@ func (a *catalogsImpl) Delete(ctx context.Context, request DeleteCatalogRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -356,7 +356,7 @@ func (a *catalogsImpl) Get(ctx context.Context, request GetCatalogRequest) (*Cat headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &catalogInfo) return &catalogInfo, err @@ -427,7 +427,7 @@ func (a *catalogsImpl) internalList(ctx context.Context, request ListCatalogsReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCatalogsResponse) return &listCatalogsResponse, err @@ -442,7 +442,7 @@ func (a *catalogsImpl) Update(ctx context.Context, request UpdateCatalog) (*Cata headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &catalogInfo) return &catalogInfo, err @@ -462,7 +462,7 @@ func (a *connectionsImpl) Create(ctx context.Context, request CreateConnection) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &connectionInfo) return &connectionInfo, err @@ -475,7 +475,7 @@ func (a *connectionsImpl) Delete(ctx context.Context, request DeleteConnectionRe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -489,7 +489,7 @@ func (a *connectionsImpl) Get(ctx context.Context, request GetConnectionRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &connectionInfo) return &connectionInfo, err @@ -552,7 +552,7 @@ func (a *connectionsImpl) internalList(ctx context.Context, request ListConnecti headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listConnectionsResponse) return &listConnectionsResponse, err @@ -567,7 +567,7 @@ func (a *connectionsImpl) Update(ctx context.Context, request UpdateConnection) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &connectionInfo) return &connectionInfo, err @@ -587,7 +587,7 @@ func (a *credentialsImpl) CreateCredential(ctx context.Context, request CreateCr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &credentialInfo) return &credentialInfo, err @@ -600,7 +600,7 @@ func (a *credentialsImpl) DeleteCredential(ctx context.Context, request DeleteCr headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -615,7 +615,7 @@ func (a *credentialsImpl) GenerateTemporaryServiceCredential(ctx context.Context headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &temporaryCredentials) return &temporaryCredentials, err @@ -629,7 +629,7 @@ func (a *credentialsImpl) GetCredential(ctx context.Context, request GetCredenti headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &credentialInfo) return &credentialInfo, err @@ -694,7 +694,7 @@ func (a *credentialsImpl) internalListCredentials(ctx context.Context, request L headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCredentialsResponse) return &listCredentialsResponse, err @@ -709,7 +709,7 @@ func (a *credentialsImpl) UpdateCredential(ctx context.Context, request UpdateCr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &credentialInfo) return &credentialInfo, err @@ -724,7 +724,7 @@ func (a *credentialsImpl) ValidateCredential(ctx context.Context, request Valida headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &validateCredentialResponse) return &validateCredentialResponse, err @@ -744,7 +744,7 @@ func (a *entityTagAssignmentsImpl) Create(ctx context.Context, request CreateEnt headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.TagAssignment, &entityTagAssignment) return &entityTagAssignment, err @@ -757,7 +757,7 @@ func (a *entityTagAssignmentsImpl) Delete(ctx context.Context, request DeleteEnt headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -771,7 +771,7 @@ func (a *entityTagAssignmentsImpl) Get(ctx context.Context, request GetEntityTag headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &entityTagAssignment) return &entityTagAssignment, err @@ -826,7 +826,7 @@ func (a *entityTagAssignmentsImpl) internalList(ctx context.Context, request Lis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listEntityTagAssignmentsResponse) return &listEntityTagAssignmentsResponse, err @@ -845,7 +845,7 @@ func (a *entityTagAssignmentsImpl) Update(ctx context.Context, request UpdateEnt headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.TagAssignment, &entityTagAssignment) return &entityTagAssignment, err @@ -865,7 +865,7 @@ func (a *externalLineageImpl) CreateExternalLineageRelationship(ctx context.Cont headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.ExternalLineageRelationship, &externalLineageRelationship) return &externalLineageRelationship, err @@ -878,7 +878,7 @@ func (a *externalLineageImpl) DeleteExternalLineageRelationship(ctx context.Cont headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -925,7 +925,7 @@ func (a *externalLineageImpl) internalListExternalLineageRelationships(ctx conte headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExternalLineageRelationshipsResponse) return &listExternalLineageRelationshipsResponse, err @@ -944,7 +944,7 @@ func (a *externalLineageImpl) UpdateExternalLineageRelationship(ctx context.Cont headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.ExternalLineageRelationship, &externalLineageRelationship) return &externalLineageRelationship, err @@ -964,7 +964,7 @@ func (a *externalLocationsImpl) Create(ctx context.Context, request CreateExtern headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &externalLocationInfo) return &externalLocationInfo, err @@ -977,7 +977,7 @@ func (a *externalLocationsImpl) Delete(ctx context.Context, request DeleteExtern headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -991,7 +991,7 @@ func (a *externalLocationsImpl) Get(ctx context.Context, request GetExternalLoca headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &externalLocationInfo) return &externalLocationInfo, err @@ -1062,7 +1062,7 @@ func (a *externalLocationsImpl) internalList(ctx context.Context, request ListEx headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExternalLocationsResponse) return &listExternalLocationsResponse, err @@ -1077,7 +1077,7 @@ func (a *externalLocationsImpl) Update(ctx context.Context, request UpdateExtern headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &externalLocationInfo) return &externalLocationInfo, err @@ -1097,7 +1097,7 @@ func (a *externalMetadataImpl) CreateExternalMetadata(ctx context.Context, reque headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.ExternalMetadata, &externalMetadata) return &externalMetadata, err @@ -1110,7 +1110,7 @@ func (a *externalMetadataImpl) DeleteExternalMetadata(ctx context.Context, reque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1124,7 +1124,7 @@ func (a *externalMetadataImpl) GetExternalMetadata(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &externalMetadata) return &externalMetadata, err @@ -1177,7 +1177,7 @@ func (a *externalMetadataImpl) internalListExternalMetadata(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExternalMetadataResponse) return &listExternalMetadataResponse, err @@ -1196,7 +1196,7 @@ func (a *externalMetadataImpl) UpdateExternalMetadata(ctx context.Context, reque headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.ExternalMetadata, &externalMetadata) return &externalMetadata, err @@ -1216,7 +1216,7 @@ func (a *functionsImpl) Create(ctx context.Context, request CreateFunctionReques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &functionInfo) return &functionInfo, err @@ -1228,7 +1228,7 @@ func (a *functionsImpl) Delete(ctx context.Context, request DeleteFunctionReques headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1242,7 +1242,7 @@ func (a *functionsImpl) Get(ctx context.Context, request GetFunctionRequest) (*F headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &functionInfo) return &functionInfo, err @@ -1317,7 +1317,7 @@ func (a *functionsImpl) internalList(ctx context.Context, request ListFunctionsR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listFunctionsResponse) return &listFunctionsResponse, err @@ -1332,7 +1332,7 @@ func (a *functionsImpl) Update(ctx context.Context, request UpdateFunction) (*Fu headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &functionInfo) return &functionInfo, err @@ -1351,7 +1351,7 @@ func (a *grantsImpl) Get(ctx context.Context, request GetGrantRequest) (*GetPerm headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getPermissionsResponse) return &getPermissionsResponse, err @@ -1365,7 +1365,7 @@ func (a *grantsImpl) GetEffective(ctx context.Context, request GetEffectiveReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &effectivePermissionsList) return &effectivePermissionsList, err @@ -1380,7 +1380,7 @@ func (a *grantsImpl) Update(ctx context.Context, request UpdatePermissions) (*Up headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updatePermissionsResponse) return &updatePermissionsResponse, err @@ -1399,7 +1399,7 @@ func (a *metastoresImpl) Assign(ctx context.Context, request CreateMetastoreAssi headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -1414,7 +1414,7 @@ func (a *metastoresImpl) Create(ctx context.Context, request CreateMetastore) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &metastoreInfo) return &metastoreInfo, err @@ -1428,7 +1428,7 @@ func (a *metastoresImpl) Current(ctx context.Context) (*MetastoreAssignment, err headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &metastoreAssignment) return &metastoreAssignment, err @@ -1441,7 +1441,7 @@ func (a *metastoresImpl) Delete(ctx context.Context, request DeleteMetastoreRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1455,7 +1455,7 @@ func (a *metastoresImpl) Get(ctx context.Context, request GetMetastoreRequest) ( headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &metastoreInfo) return &metastoreInfo, err @@ -1522,7 +1522,7 @@ func (a *metastoresImpl) internalList(ctx context.Context, request ListMetastore headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listMetastoresResponse) return &listMetastoresResponse, err @@ -1536,7 +1536,7 @@ func (a *metastoresImpl) Summary(ctx context.Context) (*GetMetastoreSummaryRespo headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &getMetastoreSummaryResponse) return &getMetastoreSummaryResponse, err @@ -1549,7 +1549,7 @@ func (a *metastoresImpl) Unassign(ctx context.Context, request UnassignRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1564,7 +1564,7 @@ func (a *metastoresImpl) Update(ctx context.Context, request UpdateMetastore) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &metastoreInfo) return &metastoreInfo, err @@ -1578,7 +1578,7 @@ func (a *metastoresImpl) UpdateAssignment(ctx context.Context, request UpdateMet headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -1595,7 +1595,7 @@ func (a *modelVersionsImpl) Delete(ctx context.Context, request DeleteModelVersi headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1609,7 +1609,7 @@ func (a *modelVersionsImpl) Get(ctx context.Context, request GetModelVersionRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &modelVersionInfo) return &modelVersionInfo, err @@ -1623,7 +1623,7 @@ func (a *modelVersionsImpl) GetByAlias(ctx context.Context, request GetByAliasRe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &modelVersionInfo) return &modelVersionInfo, err @@ -1702,7 +1702,7 @@ func (a *modelVersionsImpl) internalList(ctx context.Context, request ListModelV headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listModelVersionsResponse) return &listModelVersionsResponse, err @@ -1717,7 +1717,7 @@ func (a *modelVersionsImpl) Update(ctx context.Context, request UpdateModelVersi headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &modelVersionInfo) return &modelVersionInfo, err @@ -1737,7 +1737,7 @@ func (a *onlineTablesImpl) Create(ctx context.Context, request CreateOnlineTable headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Table, &onlineTable) return &onlineTable, err @@ -1750,7 +1750,7 @@ func (a *onlineTablesImpl) Delete(ctx context.Context, request DeleteOnlineTable headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1764,7 +1764,7 @@ func (a *onlineTablesImpl) Get(ctx context.Context, request GetOnlineTableReques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &onlineTable) return &onlineTable, err @@ -1784,7 +1784,7 @@ func (a *policiesImpl) CreatePolicy(ctx context.Context, request CreatePolicyReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.PolicyInfo, &policyInfo) return &policyInfo, err @@ -1798,7 +1798,7 @@ func (a *policiesImpl) DeletePolicy(ctx context.Context, request DeletePolicyReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deletePolicyResponse) return &deletePolicyResponse, err @@ -1812,7 +1812,7 @@ func (a *policiesImpl) GetPolicy(ctx context.Context, request GetPolicyRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &policyInfo) return &policyInfo, err @@ -1869,7 +1869,7 @@ func (a *policiesImpl) internalListPolicies(ctx context.Context, request ListPol headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listPoliciesResponse) return &listPoliciesResponse, err @@ -1888,7 +1888,7 @@ func (a *policiesImpl) UpdatePolicy(ctx context.Context, request UpdatePolicyReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.PolicyInfo, &policyInfo) return &policyInfo, err @@ -1907,7 +1907,7 @@ func (a *qualityMonitorsImpl) CancelRefresh(ctx context.Context, request CancelR headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -1922,7 +1922,7 @@ func (a *qualityMonitorsImpl) Create(ctx context.Context, request CreateMonitor) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &monitorInfo) return &monitorInfo, err @@ -1936,7 +1936,7 @@ func (a *qualityMonitorsImpl) Delete(ctx context.Context, request DeleteQualityM headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteMonitorResponse) return &deleteMonitorResponse, err @@ -1950,7 +1950,7 @@ func (a *qualityMonitorsImpl) Get(ctx context.Context, request GetQualityMonitor headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &monitorInfo) return &monitorInfo, err @@ -1964,7 +1964,7 @@ func (a *qualityMonitorsImpl) GetRefresh(ctx context.Context, request GetRefresh headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &monitorRefreshInfo) return &monitorRefreshInfo, err @@ -1978,7 +1978,7 @@ func (a *qualityMonitorsImpl) ListRefreshes(ctx context.Context, request ListRef headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &monitorRefreshListResponse) return &monitorRefreshListResponse, err @@ -1993,7 +1993,7 @@ func (a *qualityMonitorsImpl) RegenerateDashboard(ctx context.Context, request R headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, ®enerateDashboardResponse) return ®enerateDashboardResponse, err @@ -2008,7 +2008,7 @@ func (a *qualityMonitorsImpl) RunRefresh(ctx context.Context, request RunRefresh headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &monitorRefreshInfo) return &monitorRefreshInfo, err @@ -2023,7 +2023,7 @@ func (a *qualityMonitorsImpl) Update(ctx context.Context, request UpdateMonitor) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &monitorInfo) return &monitorInfo, err @@ -2043,7 +2043,7 @@ func (a *registeredModelsImpl) Create(ctx context.Context, request CreateRegiste headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, ®isteredModelInfo) return ®isteredModelInfo, err @@ -2055,7 +2055,7 @@ func (a *registeredModelsImpl) Delete(ctx context.Context, request DeleteRegiste headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -2067,7 +2067,7 @@ func (a *registeredModelsImpl) DeleteAlias(ctx context.Context, request DeleteAl headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -2081,7 +2081,7 @@ func (a *registeredModelsImpl) Get(ctx context.Context, request GetRegisteredMod headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, ®isteredModelInfo) return ®isteredModelInfo, err @@ -2158,7 +2158,7 @@ func (a *registeredModelsImpl) internalList(ctx context.Context, request ListReg headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listRegisteredModelsResponse) return &listRegisteredModelsResponse, err @@ -2173,7 +2173,7 @@ func (a *registeredModelsImpl) SetAlias(ctx context.Context, request SetRegister headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, ®isteredModelAlias) return ®isteredModelAlias, err @@ -2188,7 +2188,7 @@ func (a *registeredModelsImpl) Update(ctx context.Context, request UpdateRegiste headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, ®isteredModelInfo) return ®isteredModelInfo, err @@ -2207,7 +2207,7 @@ func (a *resourceQuotasImpl) GetQuota(ctx context.Context, request GetQuotaReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getQuotaResponse) return &getQuotaResponse, err @@ -2266,7 +2266,7 @@ func (a *resourceQuotasImpl) internalListQuotas(ctx context.Context, request Lis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listQuotasResponse) return &listQuotasResponse, err @@ -2286,7 +2286,7 @@ func (a *rfaImpl) BatchCreateAccessRequests(ctx context.Context, request BatchCr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &batchCreateAccessRequestsResponse) return &batchCreateAccessRequestsResponse, err @@ -2300,7 +2300,7 @@ func (a *rfaImpl) GetAccessRequestDestinations(ctx context.Context, request GetA headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &accessRequestDestinations) return &accessRequestDestinations, err @@ -2319,7 +2319,7 @@ func (a *rfaImpl) UpdateAccessRequestDestinations(ctx context.Context, request U headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.AccessRequestDestinations, &accessRequestDestinations) return &accessRequestDestinations, err @@ -2339,7 +2339,7 @@ func (a *schemasImpl) Create(ctx context.Context, request CreateSchema) (*Schema headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &schemaInfo) return &schemaInfo, err @@ -2352,7 +2352,7 @@ func (a *schemasImpl) Delete(ctx context.Context, request DeleteSchemaRequest) e headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -2366,7 +2366,7 @@ func (a *schemasImpl) Get(ctx context.Context, request GetSchemaRequest) (*Schem headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &schemaInfo) return &schemaInfo, err @@ -2437,7 +2437,7 @@ func (a *schemasImpl) internalList(ctx context.Context, request ListSchemasReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSchemasResponse) return &listSchemasResponse, err @@ -2452,7 +2452,7 @@ func (a *schemasImpl) Update(ctx context.Context, request UpdateSchema) (*Schema headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &schemaInfo) return &schemaInfo, err @@ -2472,7 +2472,7 @@ func (a *secretsUcImpl) CreateSecret(ctx context.Context, request CreateSecretRe headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Secret, &secret) return &secret, err @@ -2485,7 +2485,7 @@ func (a *secretsUcImpl) DeleteSecret(ctx context.Context, request DeleteSecretRe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -2499,7 +2499,7 @@ func (a *secretsUcImpl) GetSecret(ctx context.Context, request GetSecretRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &secret) return &secret, err @@ -2558,7 +2558,7 @@ func (a *secretsUcImpl) internalListSecrets(ctx context.Context, request ListSec headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSecretsResponse) return &listSecretsResponse, err @@ -2580,7 +2580,7 @@ func (a *secretsUcImpl) UpdateSecret(ctx context.Context, request UpdateSecretRe headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Secret, &secret) return &secret, err @@ -2600,7 +2600,7 @@ func (a *storageCredentialsImpl) Create(ctx context.Context, request CreateStora headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &storageCredentialInfo) return &storageCredentialInfo, err @@ -2613,7 +2613,7 @@ func (a *storageCredentialsImpl) Delete(ctx context.Context, request DeleteStora headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -2627,7 +2627,7 @@ func (a *storageCredentialsImpl) Get(ctx context.Context, request GetStorageCred headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &storageCredentialInfo) return &storageCredentialInfo, err @@ -2698,7 +2698,7 @@ func (a *storageCredentialsImpl) internalList(ctx context.Context, request ListS headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listStorageCredentialsResponse) return &listStorageCredentialsResponse, err @@ -2713,7 +2713,7 @@ func (a *storageCredentialsImpl) Update(ctx context.Context, request UpdateStora headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &storageCredentialInfo) return &storageCredentialInfo, err @@ -2728,7 +2728,7 @@ func (a *storageCredentialsImpl) Validate(ctx context.Context, request ValidateS headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &validateStorageCredentialResponse) return &validateStorageCredentialResponse, err @@ -2746,7 +2746,7 @@ func (a *systemSchemasImpl) Disable(ctx context.Context, request DisableRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -2760,7 +2760,7 @@ func (a *systemSchemasImpl) Enable(ctx context.Context, request EnableRequest) e headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -2825,7 +2825,7 @@ func (a *systemSchemasImpl) internalList(ctx context.Context, request ListSystem headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSystemSchemasResponse) return &listSystemSchemasResponse, err @@ -2845,7 +2845,7 @@ func (a *tableConstraintsImpl) Create(ctx context.Context, request CreateTableCo headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &tableConstraint) return &tableConstraint, err @@ -2858,7 +2858,7 @@ func (a *tableConstraintsImpl) Delete(ctx context.Context, request DeleteTableCo headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -2878,7 +2878,7 @@ func (a *tablesImpl) Create(ctx context.Context, request CreateTableRequest) (*T headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &tableInfo) return &tableInfo, err @@ -2891,7 +2891,7 @@ func (a *tablesImpl) Delete(ctx context.Context, request DeleteTableRequest) err headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -2905,7 +2905,7 @@ func (a *tablesImpl) Exists(ctx context.Context, request ExistsRequest) (*TableE headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &tableExistsResponse) return &tableExistsResponse, err @@ -2919,7 +2919,7 @@ func (a *tablesImpl) Get(ctx context.Context, request GetTableRequest) (*TableIn headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &tableInfo) return &tableInfo, err @@ -2998,7 +2998,7 @@ func (a *tablesImpl) internalList(ctx context.Context, request ListTablesRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listTablesResponse) return &listTablesResponse, err @@ -3073,7 +3073,7 @@ func (a *tablesImpl) internalListSummaries(ctx context.Context, request ListSumm headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listTableSummariesResponse) return &listTableSummariesResponse, err @@ -3087,7 +3087,7 @@ func (a *tablesImpl) Update(ctx context.Context, request UpdateTableRequest) err headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -3107,7 +3107,7 @@ func (a *temporaryPathCredentialsImpl) GenerateTemporaryPathCredentials(ctx cont headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &generateTemporaryPathCredentialResponse) return &generateTemporaryPathCredentialResponse, err @@ -3127,7 +3127,7 @@ func (a *temporaryTableCredentialsImpl) GenerateTemporaryTableCredentials(ctx co headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &generateTemporaryTableCredentialResponse) return &generateTemporaryTableCredentialResponse, err @@ -3147,7 +3147,7 @@ func (a *temporaryVolumeCredentialsImpl) GenerateTemporaryVolumeCredentials(ctx headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &generateTemporaryVolumeCredentialResponse) return &generateTemporaryVolumeCredentialResponse, err @@ -3167,7 +3167,7 @@ func (a *volumesImpl) Create(ctx context.Context, request CreateVolumeRequestCon headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &volumeInfo) return &volumeInfo, err @@ -3179,7 +3179,7 @@ func (a *volumesImpl) Delete(ctx context.Context, request DeleteVolumeRequest) e headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -3254,7 +3254,7 @@ func (a *volumesImpl) internalList(ctx context.Context, request ListVolumesReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listVolumesResponseContent) return &listVolumesResponseContent, err @@ -3268,7 +3268,7 @@ func (a *volumesImpl) Read(ctx context.Context, request ReadVolumeRequest) (*Vol headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &volumeInfo) return &volumeInfo, err @@ -3283,7 +3283,7 @@ func (a *volumesImpl) Update(ctx context.Context, request UpdateVolumeRequestCon headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &volumeInfo) return &volumeInfo, err @@ -3302,7 +3302,7 @@ func (a *workspaceBindingsImpl) Get(ctx context.Context, request GetWorkspaceBin headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getCatalogWorkspaceBindingsResponse) return &getCatalogWorkspaceBindingsResponse, err @@ -3367,7 +3367,7 @@ func (a *workspaceBindingsImpl) internalGetBindings(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getWorkspaceBindingsResponse) return &getWorkspaceBindingsResponse, err @@ -3382,7 +3382,7 @@ func (a *workspaceBindingsImpl) Update(ctx context.Context, request UpdateWorksp headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateCatalogWorkspaceBindingsResponse) return &updateCatalogWorkspaceBindingsResponse, err @@ -3397,7 +3397,7 @@ func (a *workspaceBindingsImpl) UpdateBindings(ctx context.Context, request Upda headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateWorkspaceBindingsResponse) return &updateWorkspaceBindingsResponse, err diff --git a/service/cleanrooms/impl.go b/service/cleanrooms/impl.go index df70cc4ea..44557e4d7 100755 --- a/service/cleanrooms/impl.go +++ b/service/cleanrooms/impl.go @@ -25,7 +25,7 @@ func (a *cleanRoomAssetRevisionsImpl) Get(ctx context.Context, request GetCleanR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &cleanRoomAsset) return &cleanRoomAsset, err @@ -70,7 +70,7 @@ func (a *cleanRoomAssetRevisionsImpl) internalList(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCleanRoomAssetRevisionsResponse) return &listCleanRoomAssetRevisionsResponse, err @@ -90,7 +90,7 @@ func (a *cleanRoomAssetsImpl) Create(ctx context.Context, request CreateCleanRoo headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Asset, &cleanRoomAsset) return &cleanRoomAsset, err @@ -105,7 +105,7 @@ func (a *cleanRoomAssetsImpl) CreateCleanRoomAssetReview(ctx context.Context, re headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createCleanRoomAssetReviewResponse) return &createCleanRoomAssetReviewResponse, err @@ -118,7 +118,7 @@ func (a *cleanRoomAssetsImpl) Delete(ctx context.Context, request DeleteCleanRoo headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -132,7 +132,7 @@ func (a *cleanRoomAssetsImpl) Get(ctx context.Context, request GetCleanRoomAsset headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &cleanRoomAsset) return &cleanRoomAsset, err @@ -177,7 +177,7 @@ func (a *cleanRoomAssetsImpl) internalList(ctx context.Context, request ListClea headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCleanRoomAssetsResponse) return &listCleanRoomAssetsResponse, err @@ -192,7 +192,7 @@ func (a *cleanRoomAssetsImpl) Update(ctx context.Context, request UpdateCleanRoo headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Asset, &cleanRoomAsset) return &cleanRoomAsset, err @@ -212,7 +212,7 @@ func (a *cleanRoomAutoApprovalRulesImpl) Create(ctx context.Context, request Cre headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &cleanRoomAutoApprovalRule) return &cleanRoomAutoApprovalRule, err @@ -225,7 +225,7 @@ func (a *cleanRoomAutoApprovalRulesImpl) Delete(ctx context.Context, request Del headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -239,7 +239,7 @@ func (a *cleanRoomAutoApprovalRulesImpl) Get(ctx context.Context, request GetCle headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &cleanRoomAutoApprovalRule) return &cleanRoomAutoApprovalRule, err @@ -284,7 +284,7 @@ func (a *cleanRoomAutoApprovalRulesImpl) internalList(ctx context.Context, reque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCleanRoomAutoApprovalRulesResponse) return &listCleanRoomAutoApprovalRulesResponse, err @@ -299,7 +299,7 @@ func (a *cleanRoomAutoApprovalRulesImpl) Update(ctx context.Context, request Upd headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.AutoApprovalRule, &cleanRoomAutoApprovalRule) return &cleanRoomAutoApprovalRule, err @@ -349,7 +349,7 @@ func (a *cleanRoomTaskRunsImpl) internalList(ctx context.Context, request ListCl headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCleanRoomNotebookTaskRunsResponse) return &listCleanRoomNotebookTaskRunsResponse, err @@ -369,7 +369,7 @@ func (a *cleanRoomsImpl) Create(ctx context.Context, request CreateCleanRoomRequ headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.CleanRoom, &cleanRoom) return &cleanRoom, err @@ -384,7 +384,7 @@ func (a *cleanRoomsImpl) CreateOutputCatalog(ctx context.Context, request Create headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.OutputCatalog, &createCleanRoomOutputCatalogResponse) return &createCleanRoomOutputCatalogResponse, err @@ -397,7 +397,7 @@ func (a *cleanRoomsImpl) Delete(ctx context.Context, request DeleteCleanRoomRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -411,7 +411,7 @@ func (a *cleanRoomsImpl) Get(ctx context.Context, request GetCleanRoomRequest) ( headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &cleanRoom) return &cleanRoom, err @@ -458,7 +458,7 @@ func (a *cleanRoomsImpl) internalList(ctx context.Context, request ListCleanRoom headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCleanRoomsResponse) return &listCleanRoomsResponse, err @@ -473,7 +473,7 @@ func (a *cleanRoomsImpl) Update(ctx context.Context, request UpdateCleanRoomRequ headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &cleanRoom) return &cleanRoom, err diff --git a/service/compute/impl.go b/service/compute/impl.go index 579a9110c..d1f30e1b1 100755 --- a/service/compute/impl.go +++ b/service/compute/impl.go @@ -26,7 +26,7 @@ func (a *clusterPoliciesImpl) Create(ctx context.Context, request CreatePolicy) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createPolicyResponse) return &createPolicyResponse, err @@ -40,7 +40,7 @@ func (a *clusterPoliciesImpl) Delete(ctx context.Context, request DeletePolicy) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -54,7 +54,7 @@ func (a *clusterPoliciesImpl) Edit(ctx context.Context, request EditPolicy) erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -68,7 +68,7 @@ func (a *clusterPoliciesImpl) Get(ctx context.Context, request GetClusterPolicyR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &policy) return &policy, err @@ -82,7 +82,7 @@ func (a *clusterPoliciesImpl) GetPermissionLevels(ctx context.Context, request G headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getClusterPolicyPermissionLevelsResponse) return &getClusterPolicyPermissionLevelsResponse, err @@ -96,7 +96,7 @@ func (a *clusterPoliciesImpl) GetPermissions(ctx context.Context, request GetClu headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &clusterPolicyPermissions) return &clusterPolicyPermissions, err @@ -135,7 +135,7 @@ func (a *clusterPoliciesImpl) internalList(ctx context.Context, request ListClus headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listPoliciesResponse) return &listPoliciesResponse, err @@ -150,7 +150,7 @@ func (a *clusterPoliciesImpl) SetPermissions(ctx context.Context, request Cluste headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &clusterPolicyPermissions) return &clusterPolicyPermissions, err @@ -165,7 +165,7 @@ func (a *clusterPoliciesImpl) UpdatePermissions(ctx context.Context, request Clu headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &clusterPolicyPermissions) return &clusterPolicyPermissions, err @@ -184,7 +184,7 @@ func (a *clustersImpl) ChangeOwner(ctx context.Context, request ChangeClusterOwn headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -199,7 +199,7 @@ func (a *clustersImpl) Create(ctx context.Context, request CreateCluster) (*Crea headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createClusterResponse) return &createClusterResponse, err @@ -213,7 +213,7 @@ func (a *clustersImpl) Delete(ctx context.Context, request DeleteCluster) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -227,7 +227,7 @@ func (a *clustersImpl) Edit(ctx context.Context, request EditCluster) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -279,7 +279,7 @@ func (a *clustersImpl) internalEvents(ctx context.Context, request GetEvents) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &getEventsResponse) return &getEventsResponse, err @@ -293,7 +293,7 @@ func (a *clustersImpl) Get(ctx context.Context, request GetClusterRequest) (*Clu headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &clusterDetails) return &clusterDetails, err @@ -307,7 +307,7 @@ func (a *clustersImpl) GetPermissionLevels(ctx context.Context, request GetClust headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getClusterPermissionLevelsResponse) return &getClusterPermissionLevelsResponse, err @@ -321,7 +321,7 @@ func (a *clustersImpl) GetPermissions(ctx context.Context, request GetClusterPer headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &clusterPermissions) return &clusterPermissions, err @@ -370,7 +370,7 @@ func (a *clustersImpl) internalList(ctx context.Context, request ListClustersReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listClustersResponse) return &listClustersResponse, err @@ -384,7 +384,7 @@ func (a *clustersImpl) ListNodeTypes(ctx context.Context) (*ListNodeTypesRespons headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listNodeTypesResponse) return &listNodeTypesResponse, err @@ -398,7 +398,7 @@ func (a *clustersImpl) ListZones(ctx context.Context) (*ListAvailableZonesRespon headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listAvailableZonesResponse) return &listAvailableZonesResponse, err @@ -412,7 +412,7 @@ func (a *clustersImpl) PermanentDelete(ctx context.Context, request PermanentDel headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -426,7 +426,7 @@ func (a *clustersImpl) Pin(ctx context.Context, request PinCluster) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -440,7 +440,7 @@ func (a *clustersImpl) Resize(ctx context.Context, request ResizeCluster) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -454,7 +454,7 @@ func (a *clustersImpl) Restart(ctx context.Context, request RestartCluster) erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -469,7 +469,7 @@ func (a *clustersImpl) SetPermissions(ctx context.Context, request ClusterPermis headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &clusterPermissions) return &clusterPermissions, err @@ -483,7 +483,7 @@ func (a *clustersImpl) SparkVersions(ctx context.Context) (*GetSparkVersionsResp headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &getSparkVersionsResponse) return &getSparkVersionsResponse, err @@ -497,7 +497,7 @@ func (a *clustersImpl) Start(ctx context.Context, request StartCluster) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -511,7 +511,7 @@ func (a *clustersImpl) Unpin(ctx context.Context, request UnpinCluster) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -525,7 +525,7 @@ func (a *clustersImpl) Update(ctx context.Context, request UpdateCluster) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -540,7 +540,7 @@ func (a *clustersImpl) UpdatePermissions(ctx context.Context, request ClusterPer headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &clusterPermissions) return &clusterPermissions, err @@ -559,7 +559,7 @@ func (a *commandExecutionImpl) Cancel(ctx context.Context, request CancelCommand headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -573,7 +573,7 @@ func (a *commandExecutionImpl) CommandStatus(ctx context.Context, request Comman headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &commandStatusResponse) return &commandStatusResponse, err @@ -587,7 +587,7 @@ func (a *commandExecutionImpl) ContextStatus(ctx context.Context, request Contex headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &contextStatusResponse) return &contextStatusResponse, err @@ -602,7 +602,7 @@ func (a *commandExecutionImpl) Create(ctx context.Context, request CreateContext headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &created) return &created, err @@ -616,7 +616,7 @@ func (a *commandExecutionImpl) Destroy(ctx context.Context, request DestroyConte headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -631,7 +631,7 @@ func (a *commandExecutionImpl) Execute(ctx context.Context, request Command) (*C headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &created) return &created, err @@ -651,7 +651,7 @@ func (a *globalInitScriptsImpl) Create(ctx context.Context, request GlobalInitSc headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createResponse) return &createResponse, err @@ -664,7 +664,7 @@ func (a *globalInitScriptsImpl) Delete(ctx context.Context, request DeleteGlobal headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -678,7 +678,7 @@ func (a *globalInitScriptsImpl) Get(ctx context.Context, request GetGlobalInitSc headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &globalInitScriptDetailsWithContent) return &globalInitScriptDetailsWithContent, err @@ -724,7 +724,7 @@ func (a *globalInitScriptsImpl) internalList(ctx context.Context) (*ListGlobalIn headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listGlobalInitScriptsResponse) return &listGlobalInitScriptsResponse, err @@ -738,7 +738,7 @@ func (a *globalInitScriptsImpl) Update(ctx context.Context, request GlobalInitSc headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -758,7 +758,7 @@ func (a *instancePoolsImpl) Create(ctx context.Context, request CreateInstancePo headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createInstancePoolResponse) return &createInstancePoolResponse, err @@ -772,7 +772,7 @@ func (a *instancePoolsImpl) Delete(ctx context.Context, request DeleteInstancePo headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -786,7 +786,7 @@ func (a *instancePoolsImpl) Edit(ctx context.Context, request EditInstancePool) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -800,7 +800,7 @@ func (a *instancePoolsImpl) Get(ctx context.Context, request GetInstancePoolRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getInstancePool) return &getInstancePool, err @@ -814,7 +814,7 @@ func (a *instancePoolsImpl) GetPermissionLevels(ctx context.Context, request Get headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getInstancePoolPermissionLevelsResponse) return &getInstancePoolPermissionLevelsResponse, err @@ -828,7 +828,7 @@ func (a *instancePoolsImpl) GetPermissions(ctx context.Context, request GetInsta headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &instancePoolPermissions) return &instancePoolPermissions, err @@ -868,7 +868,7 @@ func (a *instancePoolsImpl) internalList(ctx context.Context) (*ListInstancePool headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listInstancePools) return &listInstancePools, err @@ -883,7 +883,7 @@ func (a *instancePoolsImpl) SetPermissions(ctx context.Context, request Instance headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &instancePoolPermissions) return &instancePoolPermissions, err @@ -898,7 +898,7 @@ func (a *instancePoolsImpl) UpdatePermissions(ctx context.Context, request Insta headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &instancePoolPermissions) return &instancePoolPermissions, err @@ -917,7 +917,7 @@ func (a *instanceProfilesImpl) Add(ctx context.Context, request AddInstanceProfi headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -931,7 +931,7 @@ func (a *instanceProfilesImpl) Edit(ctx context.Context, request InstanceProfile headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -975,7 +975,7 @@ func (a *instanceProfilesImpl) internalList(ctx context.Context) (*ListInstanceP headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listInstanceProfilesResponse) return &listInstanceProfilesResponse, err @@ -989,7 +989,7 @@ func (a *instanceProfilesImpl) Remove(ctx context.Context, request RemoveInstanc headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -1036,7 +1036,7 @@ func (a *librariesImpl) internalAllClusterStatuses(ctx context.Context) (*ListAl headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listAllClusterLibraryStatusesResponse) return &listAllClusterLibraryStatusesResponse, err @@ -1087,7 +1087,7 @@ func (a *librariesImpl) internalClusterStatus(ctx context.Context, request Clust headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &clusterLibraryStatuses) return &clusterLibraryStatuses, err @@ -1101,7 +1101,7 @@ func (a *librariesImpl) Install(ctx context.Context, request InstallLibraries) e headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -1115,7 +1115,7 @@ func (a *librariesImpl) Uninstall(ctx context.Context, request UninstallLibrarie headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -1135,7 +1135,7 @@ func (a *policyComplianceForClustersImpl) EnforceCompliance(ctx context.Context, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &enforceClusterComplianceResponse) return &enforceClusterComplianceResponse, err @@ -1149,7 +1149,7 @@ func (a *policyComplianceForClustersImpl) GetCompliance(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getClusterComplianceResponse) return &getClusterComplianceResponse, err @@ -1198,7 +1198,7 @@ func (a *policyComplianceForClustersImpl) internalListCompliance(ctx context.Con headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listClusterCompliancesResponse) return &listClusterCompliancesResponse, err @@ -1217,7 +1217,7 @@ func (a *policyFamiliesImpl) Get(ctx context.Context, request GetPolicyFamilyReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &policyFamily) return &policyFamily, err @@ -1264,7 +1264,7 @@ func (a *policyFamiliesImpl) internalList(ctx context.Context, request ListPolic headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listPolicyFamiliesResponse) return &listPolicyFamiliesResponse, err diff --git a/service/dashboards/impl.go b/service/dashboards/impl.go index 0bd43f518..de3f90ec8 100755 --- a/service/dashboards/impl.go +++ b/service/dashboards/impl.go @@ -27,7 +27,7 @@ func (a *genieImpl) CreateMessage(ctx context.Context, request GenieCreateConver headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &genieMessage) return &genieMessage, err @@ -42,7 +42,7 @@ func (a *genieImpl) CreateMessageComment(ctx context.Context, request GenieCreat headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &genieMessageComment) return &genieMessageComment, err @@ -57,7 +57,7 @@ func (a *genieImpl) CreateSpace(ctx context.Context, request GenieCreateSpaceReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &genieSpace) return &genieSpace, err @@ -70,7 +70,7 @@ func (a *genieImpl) DeleteConversation(ctx context.Context, request GenieDeleteC headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -83,7 +83,7 @@ func (a *genieImpl) DeleteConversationMessage(ctx context.Context, request Genie headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -98,7 +98,7 @@ func (a *genieImpl) ExecuteMessageAttachmentQuery(ctx context.Context, request G headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &genieGetMessageQueryResultResponse) return &genieGetMessageQueryResultResponse, err @@ -113,7 +113,7 @@ func (a *genieImpl) ExecuteMessageQuery(ctx context.Context, request GenieExecut headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &genieGetMessageQueryResultResponse) return &genieGetMessageQueryResultResponse, err @@ -128,7 +128,7 @@ func (a *genieImpl) GenerateDownloadFullQueryResult(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &genieGenerateDownloadFullQueryResultResponse) return &genieGenerateDownloadFullQueryResultResponse, err @@ -143,7 +143,7 @@ func (a *genieImpl) GenieCreateEvalRun(ctx context.Context, request GenieCreateE headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &genieEvalRunResponse) return &genieEvalRunResponse, err @@ -157,7 +157,7 @@ func (a *genieImpl) GenieGetEvalResultDetails(ctx context.Context, request Genie headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieEvalResultDetails) return &genieEvalResultDetails, err @@ -171,7 +171,7 @@ func (a *genieImpl) GenieGetEvalRun(ctx context.Context, request GenieGetEvalRun headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieEvalRunResponse) return &genieEvalRunResponse, err @@ -185,7 +185,7 @@ func (a *genieImpl) GenieListEvalResults(ctx context.Context, request GenieListE headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieListEvalResultsResponse) return &genieListEvalResultsResponse, err @@ -199,7 +199,7 @@ func (a *genieImpl) GenieListEvalRuns(ctx context.Context, request GenieListEval headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieListEvalRunsResponse) return &genieListEvalRunsResponse, err @@ -213,7 +213,7 @@ func (a *genieImpl) GetDownloadFullQueryResult(ctx context.Context, request Geni headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieGetDownloadFullQueryResultResponse) return &genieGetDownloadFullQueryResultResponse, err @@ -227,7 +227,7 @@ func (a *genieImpl) GetMessage(ctx context.Context, request GenieGetConversation headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieMessage) return &genieMessage, err @@ -241,7 +241,7 @@ func (a *genieImpl) GetMessageAttachmentQueryResult(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieGetMessageQueryResultResponse) return &genieGetMessageQueryResultResponse, err @@ -255,7 +255,7 @@ func (a *genieImpl) GetMessageQueryResult(ctx context.Context, request GenieGetM headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieGetMessageQueryResultResponse) return &genieGetMessageQueryResultResponse, err @@ -269,7 +269,7 @@ func (a *genieImpl) GetMessageQueryResultByAttachment(ctx context.Context, reque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieGetMessageQueryResultResponse) return &genieGetMessageQueryResultResponse, err @@ -283,7 +283,7 @@ func (a *genieImpl) GetSpace(ctx context.Context, request GenieGetSpaceRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieSpace) return &genieSpace, err @@ -297,7 +297,7 @@ func (a *genieImpl) ListConversationComments(ctx context.Context, request GenieL headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieListConversationCommentsResponse) return &genieListConversationCommentsResponse, err @@ -311,7 +311,7 @@ func (a *genieImpl) ListConversationMessages(ctx context.Context, request GenieL headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieListConversationMessagesResponse) return &genieListConversationMessagesResponse, err @@ -325,7 +325,7 @@ func (a *genieImpl) ListConversations(ctx context.Context, request GenieListConv headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieListConversationsResponse) return &genieListConversationsResponse, err @@ -339,7 +339,7 @@ func (a *genieImpl) ListMessageComments(ctx context.Context, request GenieListMe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieListMessageCommentsResponse) return &genieListMessageCommentsResponse, err @@ -353,7 +353,7 @@ func (a *genieImpl) ListSpaces(ctx context.Context, request GenieListSpacesReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &genieListSpacesResponse) return &genieListSpacesResponse, err @@ -367,7 +367,7 @@ func (a *genieImpl) SendMessageFeedback(ctx context.Context, request GenieSendMe headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -382,7 +382,7 @@ func (a *genieImpl) StartConversation(ctx context.Context, request GenieStartCon headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &genieStartConversationResponse) return &genieStartConversationResponse, err @@ -395,7 +395,7 @@ func (a *genieImpl) TrashSpace(ctx context.Context, request GenieTrashSpaceReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -410,7 +410,7 @@ func (a *genieImpl) UpdateSpace(ctx context.Context, request GenieUpdateSpaceReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &genieSpace) return &genieSpace, err @@ -438,7 +438,7 @@ func (a *lakeviewImpl) Create(ctx context.Context, request CreateDashboardReques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Dashboard, &dashboard) return &dashboard, err @@ -453,7 +453,7 @@ func (a *lakeviewImpl) CreateSchedule(ctx context.Context, request CreateSchedul headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Schedule, &schedule) return &schedule, err @@ -468,7 +468,7 @@ func (a *lakeviewImpl) CreateSubscription(ctx context.Context, request CreateSub headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Subscription, &subscription) return &subscription, err @@ -481,7 +481,7 @@ func (a *lakeviewImpl) DeleteSchedule(ctx context.Context, request DeleteSchedul headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -494,7 +494,7 @@ func (a *lakeviewImpl) DeleteSubscription(ctx context.Context, request DeleteSub headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -508,7 +508,7 @@ func (a *lakeviewImpl) Get(ctx context.Context, request GetDashboardRequest) (*D headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &dashboard) return &dashboard, err @@ -522,7 +522,7 @@ func (a *lakeviewImpl) GetPublished(ctx context.Context, request GetPublishedDas headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &publishedDashboard) return &publishedDashboard, err @@ -536,7 +536,7 @@ func (a *lakeviewImpl) GetSchedule(ctx context.Context, request GetScheduleReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &schedule) return &schedule, err @@ -550,7 +550,7 @@ func (a *lakeviewImpl) GetSubscription(ctx context.Context, request GetSubscript headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &subscription) return &subscription, err @@ -595,7 +595,7 @@ func (a *lakeviewImpl) internalList(ctx context.Context, request ListDashboardsR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listDashboardsResponse) return &listDashboardsResponse, err @@ -640,7 +640,7 @@ func (a *lakeviewImpl) internalListSchedules(ctx context.Context, request ListSc headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSchedulesResponse) return &listSchedulesResponse, err @@ -685,7 +685,7 @@ func (a *lakeviewImpl) internalListSubscriptions(ctx context.Context, request Li headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSubscriptionsResponse) return &listSubscriptionsResponse, err @@ -700,7 +700,7 @@ func (a *lakeviewImpl) Migrate(ctx context.Context, request MigrateDashboardRequ headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &dashboard) return &dashboard, err @@ -715,7 +715,7 @@ func (a *lakeviewImpl) Publish(ctx context.Context, request PublishRequest) (*Pu headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &publishedDashboard) return &publishedDashboard, err @@ -730,7 +730,7 @@ func (a *lakeviewImpl) Revert(ctx context.Context, request RevertDashboardReques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &revertDashboardResponse) return &revertDashboardResponse, err @@ -743,7 +743,7 @@ func (a *lakeviewImpl) Trash(ctx context.Context, request TrashDashboardRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -756,7 +756,7 @@ func (a *lakeviewImpl) Unpublish(ctx context.Context, request UnpublishDashboard headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -779,7 +779,7 @@ func (a *lakeviewImpl) Update(ctx context.Context, request UpdateDashboardReques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Dashboard, &dashboard) return &dashboard, err @@ -794,7 +794,7 @@ func (a *lakeviewImpl) UpdateSchedule(ctx context.Context, request UpdateSchedul headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request.Schedule, &schedule) return &schedule, err @@ -813,7 +813,7 @@ func (a *lakeviewEmbeddedImpl) GetPublishedDashboardTokenInfo(ctx context.Contex headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getPublishedDashboardTokenInfoResponse) return &getPublishedDashboardTokenInfoResponse, err diff --git a/service/database/impl.go b/service/database/impl.go index 0e49aac2b..3057970da 100755 --- a/service/database/impl.go +++ b/service/database/impl.go @@ -28,7 +28,7 @@ func (a *databaseImpl) CreateDatabaseCatalog(ctx context.Context, request Create headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Catalog, &databaseCatalog) return &databaseCatalog, err @@ -43,7 +43,7 @@ func (a *databaseImpl) CreateDatabaseInstance(ctx context.Context, request Creat headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.DatabaseInstance, &databaseInstance) return &databaseInstance, err @@ -62,7 +62,7 @@ func (a *databaseImpl) CreateDatabaseInstanceRole(ctx context.Context, request C headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.DatabaseInstanceRole, &databaseInstanceRole) return &databaseInstanceRole, err @@ -77,7 +77,7 @@ func (a *databaseImpl) CreateDatabaseTable(ctx context.Context, request CreateDa headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Table, &databaseTable) return &databaseTable, err @@ -92,7 +92,7 @@ func (a *databaseImpl) CreateSyncedDatabaseTable(ctx context.Context, request Cr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.SyncedTable, &syncedDatabaseTable) return &syncedDatabaseTable, err @@ -105,7 +105,7 @@ func (a *databaseImpl) DeleteDatabaseCatalog(ctx context.Context, request Delete headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -118,7 +118,7 @@ func (a *databaseImpl) DeleteDatabaseInstance(ctx context.Context, request Delet headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -131,7 +131,7 @@ func (a *databaseImpl) DeleteDatabaseInstanceRole(ctx context.Context, request D headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -144,7 +144,7 @@ func (a *databaseImpl) DeleteDatabaseTable(ctx context.Context, request DeleteDa headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -157,7 +157,7 @@ func (a *databaseImpl) DeleteSyncedDatabaseTable(ctx context.Context, request De headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -171,7 +171,7 @@ func (a *databaseImpl) FindDatabaseInstanceByUid(ctx context.Context, request Fi headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &databaseInstance) return &databaseInstance, err @@ -189,7 +189,7 @@ func (a *databaseImpl) GenerateDatabaseCredential(ctx context.Context, request G headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &databaseCredential) return &databaseCredential, err @@ -203,7 +203,7 @@ func (a *databaseImpl) GetDatabaseCatalog(ctx context.Context, request GetDataba headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &databaseCatalog) return &databaseCatalog, err @@ -217,7 +217,7 @@ func (a *databaseImpl) GetDatabaseInstance(ctx context.Context, request GetDatab headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &databaseInstance) return &databaseInstance, err @@ -231,7 +231,7 @@ func (a *databaseImpl) GetDatabaseInstanceRole(ctx context.Context, request GetD headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &databaseInstanceRole) return &databaseInstanceRole, err @@ -245,7 +245,7 @@ func (a *databaseImpl) GetDatabaseTable(ctx context.Context, request GetDatabase headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &databaseTable) return &databaseTable, err @@ -259,7 +259,7 @@ func (a *databaseImpl) GetSyncedDatabaseTable(ctx context.Context, request GetSy headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &syncedDatabaseTable) return &syncedDatabaseTable, err @@ -304,7 +304,7 @@ func (a *databaseImpl) internalListDatabaseCatalogs(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listDatabaseCatalogsResponse) return &listDatabaseCatalogsResponse, err @@ -357,7 +357,7 @@ func (a *databaseImpl) internalListDatabaseInstanceRoles(ctx context.Context, re headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listDatabaseInstanceRolesResponse) return &listDatabaseInstanceRolesResponse, err @@ -402,7 +402,7 @@ func (a *databaseImpl) internalListDatabaseInstances(ctx context.Context, reques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listDatabaseInstancesResponse) return &listDatabaseInstancesResponse, err @@ -447,7 +447,7 @@ func (a *databaseImpl) internalListSyncedDatabaseTables(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSyncedDatabaseTablesResponse) return &listSyncedDatabaseTablesResponse, err @@ -466,7 +466,7 @@ func (a *databaseImpl) UpdateDatabaseCatalog(ctx context.Context, request Update headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.DatabaseCatalog, &databaseCatalog) return &databaseCatalog, err @@ -485,7 +485,7 @@ func (a *databaseImpl) UpdateDatabaseInstance(ctx context.Context, request Updat headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.DatabaseInstance, &databaseInstance) return &databaseInstance, err @@ -504,7 +504,7 @@ func (a *databaseImpl) UpdateSyncedDatabaseTable(ctx context.Context, request Up headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.SyncedTable, &syncedDatabaseTable) return &syncedDatabaseTable, err diff --git a/service/dataclassification/impl.go b/service/dataclassification/impl.go index e3c716e07..8f1fe7f17 100755 --- a/service/dataclassification/impl.go +++ b/service/dataclassification/impl.go @@ -26,7 +26,7 @@ func (a *dataClassificationImpl) CreateCatalogConfig(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.CatalogConfig, &catalogConfig) return &catalogConfig, err @@ -39,7 +39,7 @@ func (a *dataClassificationImpl) DeleteCatalogConfig(ctx context.Context, reques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -53,7 +53,7 @@ func (a *dataClassificationImpl) GetCatalogConfig(ctx context.Context, request G headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &catalogConfig) return &catalogConfig, err @@ -75,7 +75,7 @@ func (a *dataClassificationImpl) UpdateCatalogConfig(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.CatalogConfig, &catalogConfig) return &catalogConfig, err diff --git a/service/dataquality/impl.go b/service/dataquality/impl.go index a769e0213..21c0335ac 100755 --- a/service/dataquality/impl.go +++ b/service/dataquality/impl.go @@ -26,7 +26,7 @@ func (a *dataQualityImpl) CancelRefresh(ctx context.Context, request CancelRefre headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &cancelRefreshResponse) return &cancelRefreshResponse, err @@ -41,7 +41,7 @@ func (a *dataQualityImpl) CreateMonitor(ctx context.Context, request CreateMonit headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Monitor, &monitor) return &monitor, err @@ -56,7 +56,7 @@ func (a *dataQualityImpl) CreateRefresh(ctx context.Context, request CreateRefre headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Refresh, &refresh) return &refresh, err @@ -69,7 +69,7 @@ func (a *dataQualityImpl) DeleteMonitor(ctx context.Context, request DeleteMonit headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -82,7 +82,7 @@ func (a *dataQualityImpl) DeleteRefresh(ctx context.Context, request DeleteRefre headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -96,7 +96,7 @@ func (a *dataQualityImpl) GetMonitor(ctx context.Context, request GetMonitorRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &monitor) return &monitor, err @@ -110,7 +110,7 @@ func (a *dataQualityImpl) GetRefresh(ctx context.Context, request GetRefreshRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &refresh) return &refresh, err @@ -155,7 +155,7 @@ func (a *dataQualityImpl) internalListMonitor(ctx context.Context, request ListM headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listMonitorResponse) return &listMonitorResponse, err @@ -226,7 +226,7 @@ func (a *dataQualityImpl) internalListRefresh(ctx context.Context, request ListR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listRefreshResponse) return &listRefreshResponse, err @@ -245,7 +245,7 @@ func (a *dataQualityImpl) UpdateMonitor(ctx context.Context, request UpdateMonit headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Monitor, &monitor) return &monitor, err @@ -264,7 +264,7 @@ func (a *dataQualityImpl) UpdateRefresh(ctx context.Context, request UpdateRefre headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Refresh, &refresh) return &refresh, err diff --git a/service/environments/impl.go b/service/environments/impl.go index 57e048bff..284cfcd51 100755 --- a/service/environments/impl.go +++ b/service/environments/impl.go @@ -41,7 +41,7 @@ func (a *environmentsImpl) CreateWorkspaceBaseEnvironment(ctx context.Context, r headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.WorkspaceBaseEnvironment, &operation) return &operation, err @@ -54,7 +54,7 @@ func (a *environmentsImpl) DeleteWorkspaceBaseEnvironment(ctx context.Context, r headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -68,7 +68,7 @@ func (a *environmentsImpl) GetDefaultWorkspaceBaseEnvironment(ctx context.Contex headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &defaultWorkspaceBaseEnvironment) return &defaultWorkspaceBaseEnvironment, err @@ -82,7 +82,7 @@ func (a *environmentsImpl) GetOperation(ctx context.Context, request GetOperatio headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &operation) return &operation, err @@ -96,7 +96,7 @@ func (a *environmentsImpl) GetWorkspaceBaseEnvironment(ctx context.Context, requ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &workspaceBaseEnvironment) return &workspaceBaseEnvironment, err @@ -165,7 +165,7 @@ func (a *environmentsImpl) internalListWorkspaceBaseEnvironments(ctx context.Con headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listWorkspaceBaseEnvironmentsResponse) return &listWorkspaceBaseEnvironmentsResponse, err @@ -180,7 +180,7 @@ func (a *environmentsImpl) RefreshWorkspaceBaseEnvironment(ctx context.Context, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &operation) return &operation, err @@ -202,7 +202,7 @@ func (a *environmentsImpl) UpdateDefaultWorkspaceBaseEnvironment(ctx context.Con headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.DefaultWorkspaceBaseEnvironment, &defaultWorkspaceBaseEnvironment) return &defaultWorkspaceBaseEnvironment, err @@ -217,7 +217,7 @@ func (a *environmentsImpl) UpdateWorkspaceBaseEnvironment(ctx context.Context, r headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.WorkspaceBaseEnvironment, &operation) return &operation, err diff --git a/service/files/impl.go b/service/files/impl.go index 85b82a7c3..d413a2430 100755 --- a/service/files/impl.go +++ b/service/files/impl.go @@ -27,7 +27,7 @@ func (a *dbfsImpl) AddBlock(ctx context.Context, request AddBlock) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -41,7 +41,7 @@ func (a *dbfsImpl) Close(ctx context.Context, request Close) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -56,7 +56,7 @@ func (a *dbfsImpl) Create(ctx context.Context, request Create) (*CreateResponse, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createResponse) return &createResponse, err @@ -70,7 +70,7 @@ func (a *dbfsImpl) Delete(ctx context.Context, request Delete) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -84,7 +84,7 @@ func (a *dbfsImpl) GetStatus(ctx context.Context, request GetStatusRequest) (*Fi headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &fileInfo) return &fileInfo, err @@ -143,7 +143,7 @@ func (a *dbfsImpl) internalList(ctx context.Context, request ListDbfsRequest) (* headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listStatusResponse) return &listStatusResponse, err @@ -157,7 +157,7 @@ func (a *dbfsImpl) Mkdirs(ctx context.Context, request MkDirs) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -171,7 +171,7 @@ func (a *dbfsImpl) Move(ctx context.Context, request Move) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -185,7 +185,7 @@ func (a *dbfsImpl) Put(ctx context.Context, request Put) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -199,7 +199,7 @@ func (a *dbfsImpl) Read(ctx context.Context, request ReadDbfsRequest) (*ReadResp headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &readResponse) return &readResponse, err @@ -216,7 +216,7 @@ func (a *filesImpl) CreateDirectory(ctx context.Context, request CreateDirectory headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, nil, nil) return err @@ -228,7 +228,7 @@ func (a *filesImpl) Delete(ctx context.Context, request DeleteFileRequest) error headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -240,7 +240,7 @@ func (a *filesImpl) DeleteDirectory(ctx context.Context, request DeleteDirectory headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -254,7 +254,7 @@ func (a *filesImpl) Download(ctx context.Context, request DownloadRequest) (*Dow headers["Accept"] = "application/octet-stream" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &downloadResponse) return &downloadResponse, err @@ -266,7 +266,7 @@ func (a *filesImpl) GetDirectoryMetadata(ctx context.Context, request GetDirecto headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodHead, path, headers, queryParams, request, nil) return err @@ -279,7 +279,7 @@ func (a *filesImpl) GetMetadata(ctx context.Context, request GetMetadataRequest) headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodHead, path, headers, queryParams, request, &getMetadataResponse) return &getMetadataResponse, err @@ -326,7 +326,7 @@ func (a *filesImpl) internalListDirectoryContents(ctx context.Context, request L headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listDirectoryResponse) return &listDirectoryResponse, err @@ -343,7 +343,7 @@ func (a *filesImpl) Upload(ctx context.Context, request UploadRequest) error { headers["Content-Type"] = "application/octet-stream" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request.Contents, nil) return err diff --git a/service/iam/ext_impl.go b/service/iam/ext_impl.go index 34339b971..e8b335ee5 100644 --- a/service/iam/ext_impl.go +++ b/service/iam/ext_impl.go @@ -333,7 +333,7 @@ func (a *groupsImpl) Create(ctx context.Context, request Group) (*Group, error) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &group) return &group, err @@ -345,7 +345,7 @@ func (a *groupsImpl) Delete(ctx context.Context, request DeleteGroupRequest) err headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -359,7 +359,7 @@ func (a *groupsImpl) Get(ctx context.Context, request GetGroupRequest) (*Group, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &group) return &group, err @@ -414,7 +414,7 @@ func (a *groupsImpl) internalList(ctx context.Context, request ListGroupsRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listGroupsResponse) return &listGroupsResponse, err @@ -427,7 +427,7 @@ func (a *groupsImpl) Patch(ctx context.Context, request PartialUpdate) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -440,7 +440,7 @@ func (a *groupsImpl) Update(ctx context.Context, request Group) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -460,7 +460,7 @@ func (a *servicePrincipalsImpl) Create(ctx context.Context, request ServicePrinc headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &servicePrincipal) return &servicePrincipal, err @@ -472,7 +472,7 @@ func (a *servicePrincipalsImpl) Delete(ctx context.Context, request DeleteServic headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -486,7 +486,7 @@ func (a *servicePrincipalsImpl) Get(ctx context.Context, request GetServicePrinc headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &servicePrincipal) return &servicePrincipal, err @@ -541,7 +541,7 @@ func (a *servicePrincipalsImpl) internalList(ctx context.Context, request ListSe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listServicePrincipalResponse) return &listServicePrincipalResponse, err @@ -554,7 +554,7 @@ func (a *servicePrincipalsImpl) Patch(ctx context.Context, request PartialUpdate headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -567,7 +567,7 @@ func (a *servicePrincipalsImpl) Update(ctx context.Context, request ServicePrinc headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -587,7 +587,7 @@ func (a *usersImpl) Create(ctx context.Context, request User) (*User, error) { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &user) return &user, err @@ -599,7 +599,7 @@ func (a *usersImpl) Delete(ctx context.Context, request DeleteUserRequest) error headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -613,7 +613,7 @@ func (a *usersImpl) Get(ctx context.Context, request GetUserRequest) (*User, err headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &user) return &user, err @@ -627,7 +627,7 @@ func (a *usersImpl) GetPermissionLevels(ctx context.Context) (*GetPasswordPermis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &getPasswordPermissionLevelsResponse) return &getPasswordPermissionLevelsResponse, err @@ -641,7 +641,7 @@ func (a *usersImpl) GetPermissions(ctx context.Context) (*PasswordPermissions, e headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &passwordPermissions) return &passwordPermissions, err @@ -696,7 +696,7 @@ func (a *usersImpl) internalList(ctx context.Context, request ListUsersRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listUsersResponse) return &listUsersResponse, err @@ -709,7 +709,7 @@ func (a *usersImpl) Patch(ctx context.Context, request PartialUpdate) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -724,7 +724,7 @@ func (a *usersImpl) SetPermissions(ctx context.Context, request PasswordPermissi headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &passwordPermissions) return &passwordPermissions, err @@ -737,7 +737,7 @@ func (a *usersImpl) Update(ctx context.Context, request User) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -752,7 +752,7 @@ func (a *usersImpl) UpdatePermissions(ctx context.Context, request PasswordPermi headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &passwordPermissions) return &passwordPermissions, err diff --git a/service/iam/impl.go b/service/iam/impl.go index faf5603a5..efa013542 100755 --- a/service/iam/impl.go +++ b/service/iam/impl.go @@ -25,7 +25,7 @@ func (a *accessControlImpl) CheckPolicy(ctx context.Context, request CheckPolicy headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &checkPolicyResponse) return &checkPolicyResponse, err @@ -83,7 +83,7 @@ func (a *accountAccessControlProxyImpl) GetAssignableRolesForResource(ctx contex headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getAssignableRolesForResourceResponse) return &getAssignableRolesForResourceResponse, err @@ -97,7 +97,7 @@ func (a *accountAccessControlProxyImpl) GetRuleSet(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &ruleSetResponse) return &ruleSetResponse, err @@ -112,7 +112,7 @@ func (a *accountAccessControlProxyImpl) UpdateRuleSet(ctx context.Context, reque headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &ruleSetResponse) return &ruleSetResponse, err @@ -456,7 +456,7 @@ func (a *currentUserImpl) Me(ctx context.Context, request MeRequest) (*User, err headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &user) return &user, err @@ -476,7 +476,7 @@ func (a *groupsV2Impl) Create(ctx context.Context, request CreateGroupRequest) ( headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &group) return &group, err @@ -488,7 +488,7 @@ func (a *groupsV2Impl) Delete(ctx context.Context, request DeleteGroupRequest) e headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -502,7 +502,7 @@ func (a *groupsV2Impl) Get(ctx context.Context, request GetGroupRequest) (*Group headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &group) return &group, err @@ -552,7 +552,7 @@ func (a *groupsV2Impl) internalList(ctx context.Context, request ListGroupsReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listGroupsResponse) return &listGroupsResponse, err @@ -566,7 +566,7 @@ func (a *groupsV2Impl) Patch(ctx context.Context, request PatchGroupRequest) err headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -580,7 +580,7 @@ func (a *groupsV2Impl) Update(ctx context.Context, request UpdateGroupRequest) e headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -600,7 +600,7 @@ func (a *permissionMigrationImpl) MigratePermissions(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &migratePermissionsResponse) return &migratePermissionsResponse, err @@ -619,7 +619,7 @@ func (a *permissionsImpl) Get(ctx context.Context, request GetPermissionRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &objectPermissions) return &objectPermissions, err @@ -633,7 +633,7 @@ func (a *permissionsImpl) GetPermissionLevels(ctx context.Context, request GetPe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getPermissionLevelsResponse) return &getPermissionLevelsResponse, err @@ -648,7 +648,7 @@ func (a *permissionsImpl) Set(ctx context.Context, request SetObjectPermissions) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &objectPermissions) return &objectPermissions, err @@ -663,7 +663,7 @@ func (a *permissionsImpl) Update(ctx context.Context, request UpdateObjectPermis headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &objectPermissions) return &objectPermissions, err @@ -683,7 +683,7 @@ func (a *servicePrincipalsV2Impl) Create(ctx context.Context, request CreateServ headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &servicePrincipal) return &servicePrincipal, err @@ -695,7 +695,7 @@ func (a *servicePrincipalsV2Impl) Delete(ctx context.Context, request DeleteServ headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -709,7 +709,7 @@ func (a *servicePrincipalsV2Impl) Get(ctx context.Context, request GetServicePri headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &servicePrincipal) return &servicePrincipal, err @@ -759,7 +759,7 @@ func (a *servicePrincipalsV2Impl) internalList(ctx context.Context, request List headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listServicePrincipalResponse) return &listServicePrincipalResponse, err @@ -773,7 +773,7 @@ func (a *servicePrincipalsV2Impl) Patch(ctx context.Context, request PatchServic headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -787,7 +787,7 @@ func (a *servicePrincipalsV2Impl) Update(ctx context.Context, request UpdateServ headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -807,7 +807,7 @@ func (a *usersV2Impl) Create(ctx context.Context, request CreateUserRequest) (*U headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &user) return &user, err @@ -819,7 +819,7 @@ func (a *usersV2Impl) Delete(ctx context.Context, request DeleteUserRequest) err headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -833,7 +833,7 @@ func (a *usersV2Impl) Get(ctx context.Context, request GetUserRequest) (*User, e headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &user) return &user, err @@ -847,7 +847,7 @@ func (a *usersV2Impl) GetPermissionLevels(ctx context.Context, request GetPasswo headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getPasswordPermissionLevelsResponse) return &getPasswordPermissionLevelsResponse, err @@ -861,7 +861,7 @@ func (a *usersV2Impl) GetPermissions(ctx context.Context, request GetPasswordPer headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &passwordPermissions) return &passwordPermissions, err @@ -911,7 +911,7 @@ func (a *usersV2Impl) internalList(ctx context.Context, request ListUsersRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listUsersResponse) return &listUsersResponse, err @@ -925,7 +925,7 @@ func (a *usersV2Impl) Patch(ctx context.Context, request PatchUserRequest) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -940,7 +940,7 @@ func (a *usersV2Impl) SetPermissions(ctx context.Context, request PasswordPermis headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &passwordPermissions) return &passwordPermissions, err @@ -954,7 +954,7 @@ func (a *usersV2Impl) Update(ctx context.Context, request UpdateUserRequest) err headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -969,7 +969,7 @@ func (a *usersV2Impl) UpdatePermissions(ctx context.Context, request PasswordPer headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &passwordPermissions) return &passwordPermissions, err diff --git a/service/iamv2/impl.go b/service/iamv2/impl.go index 53acb9744..4182f6c71 100755 --- a/service/iamv2/impl.go +++ b/service/iamv2/impl.go @@ -141,7 +141,7 @@ func (a *workspaceIamV2Impl) CreateWorkspaceAssignmentDetailProxy(ctx context.Co headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.WorkspaceAssignmentDetail, &workspaceAssignmentDetail) return &workspaceAssignmentDetail, err @@ -154,7 +154,7 @@ func (a *workspaceIamV2Impl) DeleteWorkspaceAssignmentDetailProxy(ctx context.Co headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -168,7 +168,7 @@ func (a *workspaceIamV2Impl) GetWorkspaceAccessDetailLocal(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &workspaceAccessDetail) return &workspaceAccessDetail, err @@ -182,7 +182,7 @@ func (a *workspaceIamV2Impl) GetWorkspaceAssignmentDetailProxy(ctx context.Conte headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &workspaceAssignmentDetail) return &workspaceAssignmentDetail, err @@ -196,7 +196,7 @@ func (a *workspaceIamV2Impl) ListWorkspaceAssignmentDetailsProxy(ctx context.Con headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listWorkspaceAssignmentDetailsResponse) return &listWorkspaceAssignmentDetailsResponse, err @@ -211,7 +211,7 @@ func (a *workspaceIamV2Impl) ResolveGroupProxy(ctx context.Context, request Reso headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &resolveGroupResponse) return &resolveGroupResponse, err @@ -226,7 +226,7 @@ func (a *workspaceIamV2Impl) ResolveServicePrincipalProxy(ctx context.Context, r headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &resolveServicePrincipalResponse) return &resolveServicePrincipalResponse, err @@ -241,7 +241,7 @@ func (a *workspaceIamV2Impl) ResolveUserProxy(ctx context.Context, request Resol headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &resolveUserResponse) return &resolveUserResponse, err @@ -263,7 +263,7 @@ func (a *workspaceIamV2Impl) UpdateWorkspaceAssignmentDetailProxy(ctx context.Co headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.WorkspaceAssignmentDetail, &workspaceAssignmentDetail) return &workspaceAssignmentDetail, err diff --git a/service/jobs/impl.go b/service/jobs/impl.go index 86eb0ede7..60c86637e 100755 --- a/service/jobs/impl.go +++ b/service/jobs/impl.go @@ -24,7 +24,7 @@ func (a *jobsImpl) CancelAllRuns(ctx context.Context, request CancelAllRuns) err headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -37,7 +37,7 @@ func (a *jobsImpl) CancelRun(ctx context.Context, request CancelRun) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -52,7 +52,7 @@ func (a *jobsImpl) Create(ctx context.Context, request CreateJob) (*CreateRespon headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createResponse) return &createResponse, err @@ -65,7 +65,7 @@ func (a *jobsImpl) Delete(ctx context.Context, request DeleteJob) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -78,7 +78,7 @@ func (a *jobsImpl) DeleteRun(ctx context.Context, request DeleteRun) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -92,7 +92,7 @@ func (a *jobsImpl) ExportRun(ctx context.Context, request ExportRunRequest) (*Ex headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &exportRunOutput) return &exportRunOutput, err @@ -106,7 +106,7 @@ func (a *jobsImpl) Get(ctx context.Context, request GetJobRequest) (*Job, error) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &job) return &job, err @@ -120,7 +120,7 @@ func (a *jobsImpl) GetPermissionLevels(ctx context.Context, request GetJobPermis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getJobPermissionLevelsResponse) return &getJobPermissionLevelsResponse, err @@ -134,7 +134,7 @@ func (a *jobsImpl) GetPermissions(ctx context.Context, request GetJobPermissions headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &jobPermissions) return &jobPermissions, err @@ -148,7 +148,7 @@ func (a *jobsImpl) GetRun(ctx context.Context, request GetRunRequest) (*Run, err headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &run) return &run, err @@ -162,7 +162,7 @@ func (a *jobsImpl) GetRunOutput(ctx context.Context, request GetRunOutputRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &runOutput) return &runOutput, err @@ -207,7 +207,7 @@ func (a *jobsImpl) internalList(ctx context.Context, request ListJobsRequest) (* headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listJobsResponse) return &listJobsResponse, err @@ -252,7 +252,7 @@ func (a *jobsImpl) internalListRuns(ctx context.Context, request ListRunsRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listRunsResponse) return &listRunsResponse, err @@ -267,7 +267,7 @@ func (a *jobsImpl) RepairRun(ctx context.Context, request RepairRun) (*RepairRun headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &repairRunResponse) return &repairRunResponse, err @@ -280,7 +280,7 @@ func (a *jobsImpl) Reset(ctx context.Context, request ResetJob) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -295,7 +295,7 @@ func (a *jobsImpl) RunNow(ctx context.Context, request RunNow) (*RunNowResponse, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &runNowResponse) return &runNowResponse, err @@ -310,7 +310,7 @@ func (a *jobsImpl) SetPermissions(ctx context.Context, request JobPermissionsReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &jobPermissions) return &jobPermissions, err @@ -325,7 +325,7 @@ func (a *jobsImpl) Submit(ctx context.Context, request SubmitRun) (*SubmitRunRes headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &submitRunResponse) return &submitRunResponse, err @@ -338,7 +338,7 @@ func (a *jobsImpl) Update(ctx context.Context, request UpdateJob) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -353,7 +353,7 @@ func (a *jobsImpl) UpdatePermissions(ctx context.Context, request JobPermissions headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &jobPermissions) return &jobPermissions, err @@ -373,7 +373,7 @@ func (a *policyComplianceForJobsImpl) EnforceCompliance(ctx context.Context, req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &enforcePolicyComplianceResponse) return &enforcePolicyComplianceResponse, err @@ -387,7 +387,7 @@ func (a *policyComplianceForJobsImpl) GetCompliance(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getPolicyComplianceResponse) return &getPolicyComplianceResponse, err @@ -438,7 +438,7 @@ func (a *policyComplianceForJobsImpl) internalListCompliance(ctx context.Context headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listJobComplianceForPolicyResponse) return &listJobComplianceForPolicyResponse, err diff --git a/service/knowledgeassistants/impl.go b/service/knowledgeassistants/impl.go index f39dd7030..feb21d77e 100755 --- a/service/knowledgeassistants/impl.go +++ b/service/knowledgeassistants/impl.go @@ -28,7 +28,7 @@ func (a *knowledgeAssistantsImpl) CreateExample(ctx context.Context, request Cre headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Example, &example) return &example, err @@ -43,7 +43,7 @@ func (a *knowledgeAssistantsImpl) CreateKnowledgeAssistant(ctx context.Context, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.KnowledgeAssistant, &knowledgeAssistant) return &knowledgeAssistant, err @@ -58,7 +58,7 @@ func (a *knowledgeAssistantsImpl) CreateKnowledgeSource(ctx context.Context, req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.KnowledgeSource, &knowledgeSource) return &knowledgeSource, err @@ -71,7 +71,7 @@ func (a *knowledgeAssistantsImpl) DeleteExample(ctx context.Context, request Del headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -84,7 +84,7 @@ func (a *knowledgeAssistantsImpl) DeleteKnowledgeAssistant(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -97,7 +97,7 @@ func (a *knowledgeAssistantsImpl) DeleteKnowledgeSource(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -111,7 +111,7 @@ func (a *knowledgeAssistantsImpl) GetExample(ctx context.Context, request GetExa headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &example) return &example, err @@ -125,7 +125,7 @@ func (a *knowledgeAssistantsImpl) GetKnowledgeAssistant(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &knowledgeAssistant) return &knowledgeAssistant, err @@ -139,7 +139,7 @@ func (a *knowledgeAssistantsImpl) GetKnowledgeSource(ctx context.Context, reques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &knowledgeSource) return &knowledgeSource, err @@ -153,7 +153,7 @@ func (a *knowledgeAssistantsImpl) GetPermissionLevels(ctx context.Context, reque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getKnowledgeAssistantPermissionLevelsResponse) return &getKnowledgeAssistantPermissionLevelsResponse, err @@ -167,7 +167,7 @@ func (a *knowledgeAssistantsImpl) GetPermissions(ctx context.Context, request Ge headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &knowledgeAssistantPermissions) return &knowledgeAssistantPermissions, err @@ -212,7 +212,7 @@ func (a *knowledgeAssistantsImpl) internalListExamples(ctx context.Context, requ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExamplesResponse) return &listExamplesResponse, err @@ -257,7 +257,7 @@ func (a *knowledgeAssistantsImpl) internalListKnowledgeAssistants(ctx context.Co headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listKnowledgeAssistantsResponse) return &listKnowledgeAssistantsResponse, err @@ -302,7 +302,7 @@ func (a *knowledgeAssistantsImpl) internalListKnowledgeSources(ctx context.Conte headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listKnowledgeSourcesResponse) return &listKnowledgeSourcesResponse, err @@ -317,7 +317,7 @@ func (a *knowledgeAssistantsImpl) SetPermissions(ctx context.Context, request Kn headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &knowledgeAssistantPermissions) return &knowledgeAssistantPermissions, err @@ -331,7 +331,7 @@ func (a *knowledgeAssistantsImpl) SyncKnowledgeSources(ctx context.Context, requ headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -353,7 +353,7 @@ func (a *knowledgeAssistantsImpl) UpdateExample(ctx context.Context, request Upd headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Example, &example) return &example, err @@ -375,7 +375,7 @@ func (a *knowledgeAssistantsImpl) UpdateKnowledgeAssistant(ctx context.Context, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.KnowledgeAssistant, &knowledgeAssistant) return &knowledgeAssistant, err @@ -397,7 +397,7 @@ func (a *knowledgeAssistantsImpl) UpdateKnowledgeSource(ctx context.Context, req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.KnowledgeSource, &knowledgeSource) return &knowledgeSource, err @@ -412,7 +412,7 @@ func (a *knowledgeAssistantsImpl) UpdatePermissions(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &knowledgeAssistantPermissions) return &knowledgeAssistantPermissions, err diff --git a/service/marketplace/impl.go b/service/marketplace/impl.go index d19721a8c..3e70f99e8 100755 --- a/service/marketplace/impl.go +++ b/service/marketplace/impl.go @@ -56,7 +56,7 @@ func (a *consumerFulfillmentsImpl) internalGet(ctx context.Context, request GetL headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getListingContentMetadataResponse) return &getListingContentMetadataResponse, err @@ -109,7 +109,7 @@ func (a *consumerFulfillmentsImpl) internalList(ctx context.Context, request Lis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listFulfillmentsResponse) return &listFulfillmentsResponse, err @@ -129,7 +129,7 @@ func (a *consumerInstallationsImpl) Create(ctx context.Context, request CreateIn headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &installation) return &installation, err @@ -142,7 +142,7 @@ func (a *consumerInstallationsImpl) Delete(ctx context.Context, request DeleteIn headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -187,7 +187,7 @@ func (a *consumerInstallationsImpl) internalList(ctx context.Context, request Li headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listAllInstallationsResponse) return &listAllInstallationsResponse, err @@ -232,7 +232,7 @@ func (a *consumerInstallationsImpl) internalListListingInstallations(ctx context headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listInstallationsResponse) return &listInstallationsResponse, err @@ -247,7 +247,7 @@ func (a *consumerInstallationsImpl) Update(ctx context.Context, request UpdateIn headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &updateInstallationResponse) return &updateInstallationResponse, err @@ -266,7 +266,7 @@ func (a *consumerListingsImpl) BatchGet(ctx context.Context, request BatchGetLis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &batchGetListingsResponse) return &batchGetListingsResponse, err @@ -280,7 +280,7 @@ func (a *consumerListingsImpl) Get(ctx context.Context, request GetListingReques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getListingResponse) return &getListingResponse, err @@ -327,7 +327,7 @@ func (a *consumerListingsImpl) internalList(ctx context.Context, request ListLis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listListingsResponse) return &listListingsResponse, err @@ -376,7 +376,7 @@ func (a *consumerListingsImpl) internalSearch(ctx context.Context, request Searc headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &searchListingsResponse) return &searchListingsResponse, err @@ -396,7 +396,7 @@ func (a *consumerPersonalizationRequestsImpl) Create(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createPersonalizationRequestResponse) return &createPersonalizationRequestResponse, err @@ -410,7 +410,7 @@ func (a *consumerPersonalizationRequestsImpl) Get(ctx context.Context, request G headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getPersonalizationRequestResponse) return &getPersonalizationRequestResponse, err @@ -455,7 +455,7 @@ func (a *consumerPersonalizationRequestsImpl) internalList(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listAllPersonalizationRequestsResponse) return &listAllPersonalizationRequestsResponse, err @@ -474,7 +474,7 @@ func (a *consumerProvidersImpl) BatchGet(ctx context.Context, request BatchGetPr headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &batchGetProvidersResponse) return &batchGetProvidersResponse, err @@ -488,7 +488,7 @@ func (a *consumerProvidersImpl) Get(ctx context.Context, request GetProviderRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getProviderResponse) return &getProviderResponse, err @@ -535,7 +535,7 @@ func (a *consumerProvidersImpl) internalList(ctx context.Context, request ListCo headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listProvidersResponse) return &listProvidersResponse, err @@ -555,7 +555,7 @@ func (a *providerExchangeFiltersImpl) Create(ctx context.Context, request Create headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createExchangeFilterResponse) return &createExchangeFilterResponse, err @@ -568,7 +568,7 @@ func (a *providerExchangeFiltersImpl) Delete(ctx context.Context, request Delete headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -613,7 +613,7 @@ func (a *providerExchangeFiltersImpl) internalList(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExchangeFiltersResponse) return &listExchangeFiltersResponse, err @@ -628,7 +628,7 @@ func (a *providerExchangeFiltersImpl) Update(ctx context.Context, request Update headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &updateExchangeFilterResponse) return &updateExchangeFilterResponse, err @@ -648,7 +648,7 @@ func (a *providerExchangesImpl) AddListingToExchange(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &addExchangeForListingResponse) return &addExchangeForListingResponse, err @@ -663,7 +663,7 @@ func (a *providerExchangesImpl) Create(ctx context.Context, request CreateExchan headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createExchangeResponse) return &createExchangeResponse, err @@ -676,7 +676,7 @@ func (a *providerExchangesImpl) Delete(ctx context.Context, request DeleteExchan headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -689,7 +689,7 @@ func (a *providerExchangesImpl) DeleteListingFromExchange(ctx context.Context, r headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -703,7 +703,7 @@ func (a *providerExchangesImpl) Get(ctx context.Context, request GetExchangeRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getExchangeResponse) return &getExchangeResponse, err @@ -748,7 +748,7 @@ func (a *providerExchangesImpl) internalList(ctx context.Context, request ListEx headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExchangesResponse) return &listExchangesResponse, err @@ -793,7 +793,7 @@ func (a *providerExchangesImpl) internalListExchangesForListing(ctx context.Cont headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExchangesForListingResponse) return &listExchangesForListingResponse, err @@ -838,7 +838,7 @@ func (a *providerExchangesImpl) internalListListingsForExchange(ctx context.Cont headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listListingsForExchangeResponse) return &listListingsForExchangeResponse, err @@ -853,7 +853,7 @@ func (a *providerExchangesImpl) Update(ctx context.Context, request UpdateExchan headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &updateExchangeResponse) return &updateExchangeResponse, err @@ -873,7 +873,7 @@ func (a *providerFilesImpl) Create(ctx context.Context, request CreateFileReques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createFileResponse) return &createFileResponse, err @@ -886,7 +886,7 @@ func (a *providerFilesImpl) Delete(ctx context.Context, request DeleteFileReques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -900,7 +900,7 @@ func (a *providerFilesImpl) Get(ctx context.Context, request GetFileRequest) (*G headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getFileResponse) return &getFileResponse, err @@ -945,7 +945,7 @@ func (a *providerFilesImpl) internalList(ctx context.Context, request ListFilesR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listFilesResponse) return &listFilesResponse, err @@ -965,7 +965,7 @@ func (a *providerListingsImpl) Create(ctx context.Context, request CreateListing headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createListingResponse) return &createListingResponse, err @@ -978,7 +978,7 @@ func (a *providerListingsImpl) Delete(ctx context.Context, request DeleteListing headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -992,7 +992,7 @@ func (a *providerListingsImpl) Get(ctx context.Context, request GetListingReques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getListingResponse) return &getListingResponse, err @@ -1037,7 +1037,7 @@ func (a *providerListingsImpl) internalList(ctx context.Context, request GetList headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getListingsResponse) return &getListingsResponse, err @@ -1052,7 +1052,7 @@ func (a *providerListingsImpl) Update(ctx context.Context, request UpdateListing headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &updateListingResponse) return &updateListingResponse, err @@ -1104,7 +1104,7 @@ func (a *providerPersonalizationRequestsImpl) internalList(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listAllPersonalizationRequestsResponse) return &listAllPersonalizationRequestsResponse, err @@ -1119,7 +1119,7 @@ func (a *providerPersonalizationRequestsImpl) Update(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &updatePersonalizationRequestResponse) return &updatePersonalizationRequestResponse, err @@ -1138,7 +1138,7 @@ func (a *providerProviderAnalyticsDashboardsImpl) Create(ctx context.Context) (* headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, nil, nil, &providerAnalyticsDashboard) return &providerAnalyticsDashboard, err @@ -1152,7 +1152,7 @@ func (a *providerProviderAnalyticsDashboardsImpl) Get(ctx context.Context) (*Lis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listProviderAnalyticsDashboardResponse) return &listProviderAnalyticsDashboardResponse, err @@ -1166,7 +1166,7 @@ func (a *providerProviderAnalyticsDashboardsImpl) GetLatestVersion(ctx context.C headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &getLatestVersionProviderAnalyticsDashboardResponse) return &getLatestVersionProviderAnalyticsDashboardResponse, err @@ -1181,7 +1181,7 @@ func (a *providerProviderAnalyticsDashboardsImpl) Update(ctx context.Context, re headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &updateProviderAnalyticsDashboardResponse) return &updateProviderAnalyticsDashboardResponse, err @@ -1201,7 +1201,7 @@ func (a *providerProvidersImpl) Create(ctx context.Context, request CreateProvid headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createProviderResponse) return &createProviderResponse, err @@ -1214,7 +1214,7 @@ func (a *providerProvidersImpl) Delete(ctx context.Context, request DeleteProvid headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1228,7 +1228,7 @@ func (a *providerProvidersImpl) Get(ctx context.Context, request GetProviderRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getProviderResponse) return &getProviderResponse, err @@ -1273,7 +1273,7 @@ func (a *providerProvidersImpl) internalList(ctx context.Context, request ListPr headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listProvidersResponse) return &listProvidersResponse, err @@ -1288,7 +1288,7 @@ func (a *providerProvidersImpl) Update(ctx context.Context, request UpdateProvid headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &updateProviderResponse) return &updateProviderResponse, err diff --git a/service/ml/impl.go b/service/ml/impl.go index 499a032aa..42d9b0cd3 100644 --- a/service/ml/impl.go +++ b/service/ml/impl.go @@ -29,7 +29,7 @@ func (a *experimentsImpl) CreateExperiment(ctx context.Context, request CreateEx headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createExperimentResponse) return &createExperimentResponse, err @@ -44,7 +44,7 @@ func (a *experimentsImpl) CreateLoggedModel(ctx context.Context, request CreateL headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createLoggedModelResponse) return &createLoggedModelResponse, err @@ -59,7 +59,7 @@ func (a *experimentsImpl) CreateRun(ctx context.Context, request CreateRun) (*Cr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createRunResponse) return &createRunResponse, err @@ -73,7 +73,7 @@ func (a *experimentsImpl) DeleteExperiment(ctx context.Context, request DeleteEx headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -86,7 +86,7 @@ func (a *experimentsImpl) DeleteLoggedModel(ctx context.Context, request DeleteL headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -99,7 +99,7 @@ func (a *experimentsImpl) DeleteLoggedModelTag(ctx context.Context, request Dele headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -113,7 +113,7 @@ func (a *experimentsImpl) DeleteRun(ctx context.Context, request DeleteRun) erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -128,7 +128,7 @@ func (a *experimentsImpl) DeleteRuns(ctx context.Context, request DeleteRuns) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &deleteRunsResponse) return &deleteRunsResponse, err @@ -142,7 +142,7 @@ func (a *experimentsImpl) DeleteTag(ctx context.Context, request DeleteTag) erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -157,7 +157,7 @@ func (a *experimentsImpl) FinalizeLoggedModel(ctx context.Context, request Final headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &finalizeLoggedModelResponse) return &finalizeLoggedModelResponse, err @@ -171,7 +171,7 @@ func (a *experimentsImpl) GetByName(ctx context.Context, request GetByNameReques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getExperimentByNameResponse) return &getExperimentByNameResponse, err @@ -185,7 +185,7 @@ func (a *experimentsImpl) GetExperiment(ctx context.Context, request GetExperime headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getExperimentResponse) return &getExperimentResponse, err @@ -230,7 +230,7 @@ func (a *experimentsImpl) internalGetHistory(ctx context.Context, request GetHis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getMetricHistoryResponse) return &getMetricHistoryResponse, err @@ -244,7 +244,7 @@ func (a *experimentsImpl) GetLoggedModel(ctx context.Context, request GetLoggedM headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getLoggedModelResponse) return &getLoggedModelResponse, err @@ -258,7 +258,7 @@ func (a *experimentsImpl) GetPermissionLevels(ctx context.Context, request GetEx headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getExperimentPermissionLevelsResponse) return &getExperimentPermissionLevelsResponse, err @@ -272,7 +272,7 @@ func (a *experimentsImpl) GetPermissions(ctx context.Context, request GetExperim headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &experimentPermissions) return &experimentPermissions, err @@ -286,7 +286,7 @@ func (a *experimentsImpl) GetRun(ctx context.Context, request GetRunRequest) (*G headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getRunResponse) return &getRunResponse, err @@ -341,7 +341,7 @@ func (a *experimentsImpl) internalListArtifacts(ctx context.Context, request Lis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listArtifactsResponse) return &listArtifactsResponse, err @@ -386,7 +386,7 @@ func (a *experimentsImpl) internalListExperiments(ctx context.Context, request L headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExperimentsResponse) return &listExperimentsResponse, err @@ -400,7 +400,7 @@ func (a *experimentsImpl) LogBatch(ctx context.Context, request LogBatch) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -414,7 +414,7 @@ func (a *experimentsImpl) LogInputs(ctx context.Context, request LogInputs) erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -428,7 +428,7 @@ func (a *experimentsImpl) LogLoggedModelParams(ctx context.Context, request LogL headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -442,7 +442,7 @@ func (a *experimentsImpl) LogMetric(ctx context.Context, request LogMetric) erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -456,7 +456,7 @@ func (a *experimentsImpl) LogModel(ctx context.Context, request LogModel) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -470,7 +470,7 @@ func (a *experimentsImpl) LogOutputs(ctx context.Context, request LogOutputsRequ headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -484,7 +484,7 @@ func (a *experimentsImpl) LogParam(ctx context.Context, request LogParam) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -498,7 +498,7 @@ func (a *experimentsImpl) RestoreExperiment(ctx context.Context, request Restore headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -512,7 +512,7 @@ func (a *experimentsImpl) RestoreRun(ctx context.Context, request RestoreRun) er headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -527,7 +527,7 @@ func (a *experimentsImpl) RestoreRuns(ctx context.Context, request RestoreRuns) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &restoreRunsResponse) return &restoreRunsResponse, err @@ -573,7 +573,7 @@ func (a *experimentsImpl) internalSearchExperiments(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &searchExperimentsResponse) return &searchExperimentsResponse, err @@ -588,7 +588,7 @@ func (a *experimentsImpl) SearchLoggedModels(ctx context.Context, request Search headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &searchLoggedModelsResponse) return &searchLoggedModelsResponse, err @@ -638,7 +638,7 @@ func (a *experimentsImpl) internalSearchRuns(ctx context.Context, request Search headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &searchRunsResponse) return &searchRunsResponse, err @@ -652,7 +652,7 @@ func (a *experimentsImpl) SetExperimentTag(ctx context.Context, request SetExper headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -666,7 +666,7 @@ func (a *experimentsImpl) SetLoggedModelTags(ctx context.Context, request SetLog headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -681,7 +681,7 @@ func (a *experimentsImpl) SetPermissions(ctx context.Context, request Experiment headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &experimentPermissions) return &experimentPermissions, err @@ -695,7 +695,7 @@ func (a *experimentsImpl) SetTag(ctx context.Context, request SetTag) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -709,7 +709,7 @@ func (a *experimentsImpl) UpdateExperiment(ctx context.Context, request UpdateEx headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -724,7 +724,7 @@ func (a *experimentsImpl) UpdatePermissions(ctx context.Context, request Experim headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &experimentPermissions) return &experimentPermissions, err @@ -739,7 +739,7 @@ func (a *experimentsImpl) UpdateRun(ctx context.Context, request UpdateRun) (*Up headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &updateRunResponse) return &updateRunResponse, err @@ -759,7 +759,7 @@ func (a *featureEngineeringImpl) BatchCreateMaterializedFeatures(ctx context.Con headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &batchCreateMaterializedFeaturesResponse) return &batchCreateMaterializedFeaturesResponse, err @@ -774,7 +774,7 @@ func (a *featureEngineeringImpl) CreateFeature(ctx context.Context, request Crea headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Feature, &feature) return &feature, err @@ -789,7 +789,7 @@ func (a *featureEngineeringImpl) CreateKafkaConfig(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.KafkaConfig, &kafkaConfig) return &kafkaConfig, err @@ -804,7 +804,7 @@ func (a *featureEngineeringImpl) CreateMaterializedFeature(ctx context.Context, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.MaterializedFeature, &materializedFeature) return &materializedFeature, err @@ -817,7 +817,7 @@ func (a *featureEngineeringImpl) DeleteFeature(ctx context.Context, request Dele headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -830,7 +830,7 @@ func (a *featureEngineeringImpl) DeleteKafkaConfig(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -843,7 +843,7 @@ func (a *featureEngineeringImpl) DeleteMaterializedFeature(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -857,7 +857,7 @@ func (a *featureEngineeringImpl) GetFeature(ctx context.Context, request GetFeat headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &feature) return &feature, err @@ -871,7 +871,7 @@ func (a *featureEngineeringImpl) GetKafkaConfig(ctx context.Context, request Get headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &kafkaConfig) return &kafkaConfig, err @@ -885,7 +885,7 @@ func (a *featureEngineeringImpl) GetMaterializedFeature(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &materializedFeature) return &materializedFeature, err @@ -930,7 +930,7 @@ func (a *featureEngineeringImpl) internalListFeatures(ctx context.Context, reque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listFeaturesResponse) return &listFeaturesResponse, err @@ -979,7 +979,7 @@ func (a *featureEngineeringImpl) internalListKafkaConfigs(ctx context.Context, r headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listKafkaConfigsResponse) return &listKafkaConfigsResponse, err @@ -1024,7 +1024,7 @@ func (a *featureEngineeringImpl) internalListMaterializedFeatures(ctx context.Co headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listMaterializedFeaturesResponse) return &listMaterializedFeaturesResponse, err @@ -1043,7 +1043,7 @@ func (a *featureEngineeringImpl) UpdateFeature(ctx context.Context, request Upda headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Feature, &feature) return &feature, err @@ -1065,7 +1065,7 @@ func (a *featureEngineeringImpl) UpdateKafkaConfig(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.KafkaConfig, &kafkaConfig) return &kafkaConfig, err @@ -1084,7 +1084,7 @@ func (a *featureEngineeringImpl) UpdateMaterializedFeature(ctx context.Context, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.MaterializedFeature, &materializedFeature) return &materializedFeature, err @@ -1104,7 +1104,7 @@ func (a *featureStoreImpl) CreateOnlineStore(ctx context.Context, request Create headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.OnlineStore, &onlineStore) return &onlineStore, err @@ -1117,7 +1117,7 @@ func (a *featureStoreImpl) DeleteOnlineStore(ctx context.Context, request Delete headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1130,7 +1130,7 @@ func (a *featureStoreImpl) DeleteOnlineTable(ctx context.Context, request Delete headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1144,7 +1144,7 @@ func (a *featureStoreImpl) GetOnlineStore(ctx context.Context, request GetOnline headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &onlineStore) return &onlineStore, err @@ -1189,7 +1189,7 @@ func (a *featureStoreImpl) internalListOnlineStores(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listOnlineStoresResponse) return &listOnlineStoresResponse, err @@ -1204,7 +1204,7 @@ func (a *featureStoreImpl) PublishTable(ctx context.Context, request PublishTabl headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &publishTableResponse) return &publishTableResponse, err @@ -1223,7 +1223,7 @@ func (a *featureStoreImpl) UpdateOnlineStore(ctx context.Context, request Update headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.OnlineStore, &onlineStore) return &onlineStore, err @@ -1243,7 +1243,7 @@ func (a *forecastingImpl) CreateExperiment(ctx context.Context, request CreateFo headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createForecastingExperimentResponse) return &createForecastingExperimentResponse, err @@ -1257,7 +1257,7 @@ func (a *forecastingImpl) GetExperiment(ctx context.Context, request GetForecast headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &forecastingExperiment) return &forecastingExperiment, err @@ -1277,7 +1277,7 @@ func (a *materializedFeaturesImpl) CreateFeatureTag(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.FeatureTag, &featureTag) return &featureTag, err @@ -1290,7 +1290,7 @@ func (a *materializedFeaturesImpl) DeleteFeatureTag(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1304,7 +1304,7 @@ func (a *materializedFeaturesImpl) GetFeatureLineage(ctx context.Context, reques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &featureLineage) return &featureLineage, err @@ -1318,7 +1318,7 @@ func (a *materializedFeaturesImpl) GetFeatureTag(ctx context.Context, request Ge headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &featureTag) return &featureTag, err @@ -1363,7 +1363,7 @@ func (a *materializedFeaturesImpl) internalListFeatureTags(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listFeatureTagsResponse) return &listFeatureTagsResponse, err @@ -1382,7 +1382,7 @@ func (a *materializedFeaturesImpl) UpdateFeatureTag(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.FeatureTag, &featureTag) return &featureTag, err @@ -1402,7 +1402,7 @@ func (a *modelRegistryImpl) ApproveTransitionRequest(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &approveTransitionRequestResponse) return &approveTransitionRequestResponse, err @@ -1417,7 +1417,7 @@ func (a *modelRegistryImpl) CreateComment(ctx context.Context, request CreateCom headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createCommentResponse) return &createCommentResponse, err @@ -1432,7 +1432,7 @@ func (a *modelRegistryImpl) CreateModel(ctx context.Context, request CreateModel headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createModelResponse) return &createModelResponse, err @@ -1447,7 +1447,7 @@ func (a *modelRegistryImpl) CreateModelVersion(ctx context.Context, request Crea headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createModelVersionResponse) return &createModelVersionResponse, err @@ -1462,7 +1462,7 @@ func (a *modelRegistryImpl) CreateTransitionRequest(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createTransitionRequestResponse) return &createTransitionRequestResponse, err @@ -1477,7 +1477,7 @@ func (a *modelRegistryImpl) CreateWebhook(ctx context.Context, request CreateReg headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createWebhookResponse) return &createWebhookResponse, err @@ -1490,7 +1490,7 @@ func (a *modelRegistryImpl) DeleteComment(ctx context.Context, request DeleteCom headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1503,7 +1503,7 @@ func (a *modelRegistryImpl) DeleteModel(ctx context.Context, request DeleteModel headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1516,7 +1516,7 @@ func (a *modelRegistryImpl) DeleteModelTag(ctx context.Context, request DeleteMo headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1529,7 +1529,7 @@ func (a *modelRegistryImpl) DeleteModelVersion(ctx context.Context, request Dele headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1542,7 +1542,7 @@ func (a *modelRegistryImpl) DeleteModelVersionTag(ctx context.Context, request D headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1556,7 +1556,7 @@ func (a *modelRegistryImpl) DeleteTransitionRequest(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteTransitionRequestResponse) return &deleteTransitionRequestResponse, err @@ -1569,7 +1569,7 @@ func (a *modelRegistryImpl) DeleteWebhook(ctx context.Context, request DeleteWeb headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1609,7 +1609,7 @@ func (a *modelRegistryImpl) internalGetLatestVersions(ctx context.Context, reque headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &getLatestVersionsResponse) return &getLatestVersionsResponse, err @@ -1623,7 +1623,7 @@ func (a *modelRegistryImpl) GetModel(ctx context.Context, request GetModelReques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getModelResponse) return &getModelResponse, err @@ -1637,7 +1637,7 @@ func (a *modelRegistryImpl) GetModelVersion(ctx context.Context, request GetMode headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getModelVersionResponse) return &getModelVersionResponse, err @@ -1651,7 +1651,7 @@ func (a *modelRegistryImpl) GetModelVersionDownloadUri(ctx context.Context, requ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getModelVersionDownloadUriResponse) return &getModelVersionDownloadUriResponse, err @@ -1665,7 +1665,7 @@ func (a *modelRegistryImpl) GetPermissionLevels(ctx context.Context, request Get headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getRegisteredModelPermissionLevelsResponse) return &getRegisteredModelPermissionLevelsResponse, err @@ -1679,7 +1679,7 @@ func (a *modelRegistryImpl) GetPermissions(ctx context.Context, request GetRegis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, ®isteredModelPermissions) return ®isteredModelPermissions, err @@ -1726,7 +1726,7 @@ func (a *modelRegistryImpl) internalListModels(ctx context.Context, request List headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listModelsResponse) return &listModelsResponse, err @@ -1765,7 +1765,7 @@ func (a *modelRegistryImpl) internalListTransitionRequests(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listTransitionRequestsResponse) return &listTransitionRequestsResponse, err @@ -1810,7 +1810,7 @@ func (a *modelRegistryImpl) internalListWebhooks(ctx context.Context, request Li headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listRegistryWebhooks) return &listRegistryWebhooks, err @@ -1825,7 +1825,7 @@ func (a *modelRegistryImpl) RejectTransitionRequest(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &rejectTransitionRequestResponse) return &rejectTransitionRequestResponse, err @@ -1840,7 +1840,7 @@ func (a *modelRegistryImpl) RenameModel(ctx context.Context, request RenameModel headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &renameModelResponse) return &renameModelResponse, err @@ -1885,7 +1885,7 @@ func (a *modelRegistryImpl) internalSearchModelVersions(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &searchModelVersionsResponse) return &searchModelVersionsResponse, err @@ -1930,7 +1930,7 @@ func (a *modelRegistryImpl) internalSearchModels(ctx context.Context, request Se headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &searchModelsResponse) return &searchModelsResponse, err @@ -1944,7 +1944,7 @@ func (a *modelRegistryImpl) SetModelTag(ctx context.Context, request SetModelTag headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -1958,7 +1958,7 @@ func (a *modelRegistryImpl) SetModelVersionTag(ctx context.Context, request SetM headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -1973,7 +1973,7 @@ func (a *modelRegistryImpl) SetPermissions(ctx context.Context, request Register headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, ®isteredModelPermissions) return ®isteredModelPermissions, err @@ -1988,7 +1988,7 @@ func (a *modelRegistryImpl) TestRegistryWebhook(ctx context.Context, request Tes headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &testRegistryWebhookResponse) return &testRegistryWebhookResponse, err @@ -2003,7 +2003,7 @@ func (a *modelRegistryImpl) TransitionStage(ctx context.Context, request Transit headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &transitionStageResponse) return &transitionStageResponse, err @@ -2018,7 +2018,7 @@ func (a *modelRegistryImpl) UpdateComment(ctx context.Context, request UpdateCom headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateCommentResponse) return &updateCommentResponse, err @@ -2033,7 +2033,7 @@ func (a *modelRegistryImpl) UpdateModel(ctx context.Context, request UpdateModel headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateModelResponse) return &updateModelResponse, err @@ -2048,7 +2048,7 @@ func (a *modelRegistryImpl) UpdateModelVersion(ctx context.Context, request Upda headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateModelVersionResponse) return &updateModelVersionResponse, err @@ -2063,7 +2063,7 @@ func (a *modelRegistryImpl) UpdatePermissions(ctx context.Context, request Regis headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, ®isteredModelPermissions) return ®isteredModelPermissions, err @@ -2078,7 +2078,7 @@ func (a *modelRegistryImpl) UpdateWebhook(ctx context.Context, request UpdateReg headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateWebhookResponse) return &updateWebhookResponse, err diff --git a/service/oauth2/impl.go b/service/oauth2/impl.go index a19e5eca0..209dcb65d 100755 --- a/service/oauth2/impl.go +++ b/service/oauth2/impl.go @@ -532,7 +532,7 @@ func (a *servicePrincipalSecretsProxyImpl) Create(ctx context.Context, request C headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createServicePrincipalSecretResponse) return &createServicePrincipalSecretResponse, err @@ -544,7 +544,7 @@ func (a *servicePrincipalSecretsProxyImpl) Delete(ctx context.Context, request D headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -593,7 +593,7 @@ func (a *servicePrincipalSecretsProxyImpl) internalList(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listServicePrincipalSecretsResponse) return &listServicePrincipalSecretsResponse, err diff --git a/service/pipelines/impl.go b/service/pipelines/impl.go index 9417a4cf6..cd859c119 100755 --- a/service/pipelines/impl.go +++ b/service/pipelines/impl.go @@ -25,7 +25,7 @@ func (a *pipelinesImpl) ApplyEnvironment(ctx context.Context, request ApplyEnvir headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, &applyEnvironmentRequestResponse) return &applyEnvironmentRequestResponse, err @@ -40,7 +40,7 @@ func (a *pipelinesImpl) Clone(ctx context.Context, request ClonePipelineRequest) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &clonePipelineResponse) return &clonePipelineResponse, err @@ -55,7 +55,7 @@ func (a *pipelinesImpl) Create(ctx context.Context, request CreatePipeline) (*Cr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createPipelineResponse) return &createPipelineResponse, err @@ -68,7 +68,7 @@ func (a *pipelinesImpl) Delete(ctx context.Context, request DeletePipelineReques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -82,7 +82,7 @@ func (a *pipelinesImpl) Get(ctx context.Context, request GetPipelineRequest) (*G headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getPipelineResponse) return &getPipelineResponse, err @@ -96,7 +96,7 @@ func (a *pipelinesImpl) GetPermissionLevels(ctx context.Context, request GetPipe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getPipelinePermissionLevelsResponse) return &getPipelinePermissionLevelsResponse, err @@ -110,7 +110,7 @@ func (a *pipelinesImpl) GetPermissions(ctx context.Context, request GetPipelineP headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &pipelinePermissions) return &pipelinePermissions, err @@ -124,7 +124,7 @@ func (a *pipelinesImpl) GetUpdate(ctx context.Context, request GetUpdateRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getUpdateResponse) return &getUpdateResponse, err @@ -169,7 +169,7 @@ func (a *pipelinesImpl) internalListPipelineEvents(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listPipelineEventsResponse) return &listPipelineEventsResponse, err @@ -214,7 +214,7 @@ func (a *pipelinesImpl) internalListPipelines(ctx context.Context, request ListP headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listPipelinesResponse) return &listPipelinesResponse, err @@ -228,7 +228,7 @@ func (a *pipelinesImpl) ListUpdates(ctx context.Context, request ListUpdatesRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listUpdatesResponse) return &listUpdatesResponse, err @@ -243,7 +243,7 @@ func (a *pipelinesImpl) SetPermissions(ctx context.Context, request PipelinePerm headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &pipelinePermissions) return &pipelinePermissions, err @@ -258,7 +258,7 @@ func (a *pipelinesImpl) StartUpdate(ctx context.Context, request StartUpdate) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &startUpdateResponse) return &startUpdateResponse, err @@ -271,7 +271,7 @@ func (a *pipelinesImpl) Stop(ctx context.Context, request StopRequest) error { headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, nil) return err @@ -285,7 +285,7 @@ func (a *pipelinesImpl) Update(ctx context.Context, request EditPipeline) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -300,7 +300,7 @@ func (a *pipelinesImpl) UpdatePermissions(ctx context.Context, request PipelineP headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &pipelinePermissions) return &pipelinePermissions, err diff --git a/service/postgres/impl.go b/service/postgres/impl.go index 71badd054..1efa6c41f 100755 --- a/service/postgres/impl.go +++ b/service/postgres/impl.go @@ -37,7 +37,7 @@ func (a *postgresImpl) CreateBranch(ctx context.Context, request CreateBranchReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Branch, &operation) return &operation, err @@ -56,7 +56,7 @@ func (a *postgresImpl) CreateCatalog(ctx context.Context, request CreateCatalogR headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Catalog, &operation) return &operation, err @@ -75,7 +75,7 @@ func (a *postgresImpl) CreateDatabase(ctx context.Context, request CreateDatabas headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Database, &operation) return &operation, err @@ -98,7 +98,7 @@ func (a *postgresImpl) CreateEndpoint(ctx context.Context, request CreateEndpoin headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Endpoint, &operation) return &operation, err @@ -117,7 +117,7 @@ func (a *postgresImpl) CreateProject(ctx context.Context, request CreateProjectR headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Project, &operation) return &operation, err @@ -136,7 +136,7 @@ func (a *postgresImpl) CreateRole(ctx context.Context, request CreateRoleRequest headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Role, &operation) return &operation, err @@ -155,7 +155,7 @@ func (a *postgresImpl) CreateSyncedTable(ctx context.Context, request CreateSync headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.SyncedTable, &operation) return &operation, err @@ -169,7 +169,7 @@ func (a *postgresImpl) DeleteBranch(ctx context.Context, request DeleteBranchReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -183,7 +183,7 @@ func (a *postgresImpl) DeleteCatalog(ctx context.Context, request DeleteCatalogR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -197,7 +197,7 @@ func (a *postgresImpl) DeleteDatabase(ctx context.Context, request DeleteDatabas headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -211,7 +211,7 @@ func (a *postgresImpl) DeleteEndpoint(ctx context.Context, request DeleteEndpoin headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -225,7 +225,7 @@ func (a *postgresImpl) DeleteProject(ctx context.Context, request DeleteProjectR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -239,7 +239,7 @@ func (a *postgresImpl) DeleteRole(ctx context.Context, request DeleteRoleRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -253,7 +253,7 @@ func (a *postgresImpl) DeleteSyncedTable(ctx context.Context, request DeleteSync headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &operation) return &operation, err @@ -268,7 +268,7 @@ func (a *postgresImpl) GenerateDatabaseCredential(ctx context.Context, request G headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &databaseCredential) return &databaseCredential, err @@ -282,7 +282,7 @@ func (a *postgresImpl) GetBranch(ctx context.Context, request GetBranchRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &branch) return &branch, err @@ -296,7 +296,7 @@ func (a *postgresImpl) GetCatalog(ctx context.Context, request GetCatalogRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &catalog) return &catalog, err @@ -310,7 +310,7 @@ func (a *postgresImpl) GetDatabase(ctx context.Context, request GetDatabaseReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &database) return &database, err @@ -324,7 +324,7 @@ func (a *postgresImpl) GetEndpoint(ctx context.Context, request GetEndpointReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &endpoint) return &endpoint, err @@ -338,7 +338,7 @@ func (a *postgresImpl) GetOperation(ctx context.Context, request GetOperationReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &operation) return &operation, err @@ -352,7 +352,7 @@ func (a *postgresImpl) GetProject(ctx context.Context, request GetProjectRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &project) return &project, err @@ -366,7 +366,7 @@ func (a *postgresImpl) GetRole(ctx context.Context, request GetRoleRequest) (*Ro headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &role) return &role, err @@ -380,7 +380,7 @@ func (a *postgresImpl) GetSyncedTable(ctx context.Context, request GetSyncedTabl headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &syncedTable) return &syncedTable, err @@ -425,7 +425,7 @@ func (a *postgresImpl) internalListBranches(ctx context.Context, request ListBra headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listBranchesResponse) return &listBranchesResponse, err @@ -470,7 +470,7 @@ func (a *postgresImpl) internalListDatabases(ctx context.Context, request ListDa headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listDatabasesResponse) return &listDatabasesResponse, err @@ -515,7 +515,7 @@ func (a *postgresImpl) internalListEndpoints(ctx context.Context, request ListEn headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listEndpointsResponse) return &listEndpointsResponse, err @@ -562,7 +562,7 @@ func (a *postgresImpl) internalListProjects(ctx context.Context, request ListPro headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listProjectsResponse) return &listProjectsResponse, err @@ -607,7 +607,7 @@ func (a *postgresImpl) internalListRoles(ctx context.Context, request ListRolesR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listRolesResponse) return &listRolesResponse, err @@ -622,7 +622,7 @@ func (a *postgresImpl) UndeleteBranch(ctx context.Context, request UndeleteBranc headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &operation) return &operation, err @@ -637,7 +637,7 @@ func (a *postgresImpl) UndeleteProject(ctx context.Context, request UndeleteProj headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &operation) return &operation, err @@ -659,7 +659,7 @@ func (a *postgresImpl) UpdateBranch(ctx context.Context, request UpdateBranchReq headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Branch, &operation) return &operation, err @@ -681,7 +681,7 @@ func (a *postgresImpl) UpdateDatabase(ctx context.Context, request UpdateDatabas headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Database, &operation) return &operation, err @@ -703,7 +703,7 @@ func (a *postgresImpl) UpdateEndpoint(ctx context.Context, request UpdateEndpoin headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Endpoint, &operation) return &operation, err @@ -725,7 +725,7 @@ func (a *postgresImpl) UpdateProject(ctx context.Context, request UpdateProjectR headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Project, &operation) return &operation, err @@ -747,7 +747,7 @@ func (a *postgresImpl) UpdateRole(ctx context.Context, request UpdateRoleRequest headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Role, &operation) return &operation, err diff --git a/service/qualitymonitorv2/impl.go b/service/qualitymonitorv2/impl.go index 5aeadac7e..636e0b202 100755 --- a/service/qualitymonitorv2/impl.go +++ b/service/qualitymonitorv2/impl.go @@ -26,7 +26,7 @@ func (a *qualityMonitorV2Impl) CreateQualityMonitor(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.QualityMonitor, &qualityMonitor) return &qualityMonitor, err @@ -39,7 +39,7 @@ func (a *qualityMonitorV2Impl) DeleteQualityMonitor(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -53,7 +53,7 @@ func (a *qualityMonitorV2Impl) GetQualityMonitor(ctx context.Context, request Ge headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &qualityMonitor) return &qualityMonitor, err @@ -100,7 +100,7 @@ func (a *qualityMonitorV2Impl) internalListQualityMonitor(ctx context.Context, r headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listQualityMonitorResponse) return &listQualityMonitorResponse, err @@ -115,7 +115,7 @@ func (a *qualityMonitorV2Impl) UpdateQualityMonitor(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request.QualityMonitor, &qualityMonitor) return &qualityMonitor, err diff --git a/service/serving/impl.go b/service/serving/impl.go index f9dde7b60..eae1e69b5 100755 --- a/service/serving/impl.go +++ b/service/serving/impl.go @@ -29,7 +29,7 @@ func (a *servingEndpointsImpl) BuildLogs(ctx context.Context, request BuildLogsR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &buildLogsResponse) return &buildLogsResponse, err @@ -44,7 +44,7 @@ func (a *servingEndpointsImpl) Create(ctx context.Context, request CreateServing headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &servingEndpointDetailed) return &servingEndpointDetailed, err @@ -59,7 +59,7 @@ func (a *servingEndpointsImpl) CreateProvisionedThroughputEndpoint(ctx context.C headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &servingEndpointDetailed) return &servingEndpointDetailed, err @@ -71,7 +71,7 @@ func (a *servingEndpointsImpl) Delete(ctx context.Context, request DeleteServing headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -85,7 +85,7 @@ func (a *servingEndpointsImpl) ExportMetrics(ctx context.Context, request Export headers["Accept"] = "text/plain" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &exportMetricsResponse) return &exportMetricsResponse, err @@ -99,7 +99,7 @@ func (a *servingEndpointsImpl) Get(ctx context.Context, request GetServingEndpoi headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &servingEndpointDetailed) return &servingEndpointDetailed, err @@ -113,7 +113,7 @@ func (a *servingEndpointsImpl) GetOpenApi(ctx context.Context, request GetOpenAp headers["Accept"] = "text/plain" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getOpenApiResponse) return &getOpenApiResponse, err @@ -127,7 +127,7 @@ func (a *servingEndpointsImpl) GetPermissionLevels(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getServingEndpointPermissionLevelsResponse) return &getServingEndpointPermissionLevelsResponse, err @@ -141,7 +141,7 @@ func (a *servingEndpointsImpl) GetPermissions(ctx context.Context, request GetSe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &servingEndpointPermissions) return &servingEndpointPermissions, err @@ -156,7 +156,7 @@ func (a *servingEndpointsImpl) HttpRequest(ctx context.Context, request External headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &httpRequestResponse) return &httpRequestResponse, err @@ -196,7 +196,7 @@ func (a *servingEndpointsImpl) internalList(ctx context.Context) (*ListEndpoints headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listEndpointsResponse) return &listEndpointsResponse, err @@ -210,7 +210,7 @@ func (a *servingEndpointsImpl) Logs(ctx context.Context, request LogsRequest) (* headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &serverLogsResponse) return &serverLogsResponse, err @@ -225,7 +225,7 @@ func (a *servingEndpointsImpl) Patch(ctx context.Context, request PatchServingEn headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &endpointTags) return &endpointTags, err @@ -240,7 +240,7 @@ func (a *servingEndpointsImpl) Put(ctx context.Context, request PutRequest) (*Pu headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &putResponse) return &putResponse, err @@ -255,7 +255,7 @@ func (a *servingEndpointsImpl) PutAiGateway(ctx context.Context, request PutAiGa headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &putAiGatewayResponse) return &putAiGatewayResponse, err @@ -270,7 +270,7 @@ func (a *servingEndpointsImpl) Query(ctx context.Context, request QueryEndpointI headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &queryEndpointResponse) return &queryEndpointResponse, err @@ -285,7 +285,7 @@ func (a *servingEndpointsImpl) SetPermissions(ctx context.Context, request Servi headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &servingEndpointPermissions) return &servingEndpointPermissions, err @@ -300,7 +300,7 @@ func (a *servingEndpointsImpl) UpdateConfig(ctx context.Context, request Endpoin headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &servingEndpointDetailed) return &servingEndpointDetailed, err @@ -315,7 +315,7 @@ func (a *servingEndpointsImpl) UpdateNotifications(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateInferenceEndpointNotificationsResponse) return &updateInferenceEndpointNotificationsResponse, err @@ -330,7 +330,7 @@ func (a *servingEndpointsImpl) UpdatePermissions(ctx context.Context, request Se headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &servingEndpointPermissions) return &servingEndpointPermissions, err @@ -345,7 +345,7 @@ func (a *servingEndpointsImpl) UpdateProvisionedThroughputEndpointConfig(ctx con headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &servingEndpointDetailed) return &servingEndpointDetailed, err @@ -392,7 +392,7 @@ func (a *servingEndpointsDataPlaneImpl) Query(ctx context.Context, request Query headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } opts := []httpclient.DoOption{} opts = append(opts, httpclient.WithRequestHeaders(headers)) diff --git a/service/settings/impl.go b/service/settings/impl.go index 3e603798a..78d87ed50 100755 --- a/service/settings/impl.go +++ b/service/settings/impl.go @@ -124,7 +124,7 @@ func (a *aibiDashboardEmbeddingAccessPolicyImpl) Delete(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteAibiDashboardEmbeddingAccessPolicySettingResponse) return &deleteAibiDashboardEmbeddingAccessPolicySettingResponse, err @@ -138,7 +138,7 @@ func (a *aibiDashboardEmbeddingAccessPolicyImpl) Get(ctx context.Context, reques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &aibiDashboardEmbeddingAccessPolicySetting) return &aibiDashboardEmbeddingAccessPolicySetting, err @@ -153,7 +153,7 @@ func (a *aibiDashboardEmbeddingAccessPolicyImpl) Update(ctx context.Context, req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &aibiDashboardEmbeddingAccessPolicySetting) return &aibiDashboardEmbeddingAccessPolicySetting, err @@ -172,7 +172,7 @@ func (a *aibiDashboardEmbeddingApprovedDomainsImpl) Delete(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteAibiDashboardEmbeddingApprovedDomainsSettingResponse) return &deleteAibiDashboardEmbeddingApprovedDomainsSettingResponse, err @@ -186,7 +186,7 @@ func (a *aibiDashboardEmbeddingApprovedDomainsImpl) Get(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &aibiDashboardEmbeddingApprovedDomainsSetting) return &aibiDashboardEmbeddingApprovedDomainsSetting, err @@ -201,7 +201,7 @@ func (a *aibiDashboardEmbeddingApprovedDomainsImpl) Update(ctx context.Context, headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &aibiDashboardEmbeddingApprovedDomainsSetting) return &aibiDashboardEmbeddingApprovedDomainsSetting, err @@ -220,7 +220,7 @@ func (a *automaticClusterUpdateImpl) Get(ctx context.Context, request GetAutomat headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &automaticClusterUpdateSetting) return &automaticClusterUpdateSetting, err @@ -235,7 +235,7 @@ func (a *automaticClusterUpdateImpl) Update(ctx context.Context, request UpdateA headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &automaticClusterUpdateSetting) return &automaticClusterUpdateSetting, err @@ -254,7 +254,7 @@ func (a *complianceSecurityProfileImpl) Get(ctx context.Context, request GetComp headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &complianceSecurityProfileSetting) return &complianceSecurityProfileSetting, err @@ -269,7 +269,7 @@ func (a *complianceSecurityProfileImpl) Update(ctx context.Context, request Upda headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &complianceSecurityProfileSetting) return &complianceSecurityProfileSetting, err @@ -289,7 +289,7 @@ func (a *credentialsManagerImpl) ExchangeToken(ctx context.Context, request Exch headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &exchangeTokenResponse) return &exchangeTokenResponse, err @@ -336,7 +336,7 @@ func (a *dashboardEmailSubscriptionsImpl) Delete(ctx context.Context, request De headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteDashboardEmailSubscriptionsResponse) return &deleteDashboardEmailSubscriptionsResponse, err @@ -350,7 +350,7 @@ func (a *dashboardEmailSubscriptionsImpl) Get(ctx context.Context, request GetDa headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &dashboardEmailSubscriptions) return &dashboardEmailSubscriptions, err @@ -365,7 +365,7 @@ func (a *dashboardEmailSubscriptionsImpl) Update(ctx context.Context, request Up headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &dashboardEmailSubscriptions) return &dashboardEmailSubscriptions, err @@ -384,7 +384,7 @@ func (a *defaultNamespaceImpl) Delete(ctx context.Context, request DeleteDefault headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteDefaultNamespaceSettingResponse) return &deleteDefaultNamespaceSettingResponse, err @@ -398,7 +398,7 @@ func (a *defaultNamespaceImpl) Get(ctx context.Context, request GetDefaultNamesp headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &defaultNamespaceSetting) return &defaultNamespaceSetting, err @@ -413,7 +413,7 @@ func (a *defaultNamespaceImpl) Update(ctx context.Context, request UpdateDefault headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &defaultNamespaceSetting) return &defaultNamespaceSetting, err @@ -432,7 +432,7 @@ func (a *defaultWarehouseIdImpl) Delete(ctx context.Context, request DeleteDefau headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteDefaultWarehouseIdResponse) return &deleteDefaultWarehouseIdResponse, err @@ -446,7 +446,7 @@ func (a *defaultWarehouseIdImpl) Get(ctx context.Context, request GetDefaultWare headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &defaultWarehouseId) return &defaultWarehouseId, err @@ -461,7 +461,7 @@ func (a *defaultWarehouseIdImpl) Update(ctx context.Context, request UpdateDefau headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &defaultWarehouseId) return &defaultWarehouseId, err @@ -480,7 +480,7 @@ func (a *disableLegacyAccessImpl) Delete(ctx context.Context, request DeleteDisa headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteDisableLegacyAccessResponse) return &deleteDisableLegacyAccessResponse, err @@ -494,7 +494,7 @@ func (a *disableLegacyAccessImpl) Get(ctx context.Context, request GetDisableLeg headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &disableLegacyAccess) return &disableLegacyAccess, err @@ -509,7 +509,7 @@ func (a *disableLegacyAccessImpl) Update(ctx context.Context, request UpdateDisa headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &disableLegacyAccess) return &disableLegacyAccess, err @@ -528,7 +528,7 @@ func (a *disableLegacyDbfsImpl) Delete(ctx context.Context, request DeleteDisabl headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteDisableLegacyDbfsResponse) return &deleteDisableLegacyDbfsResponse, err @@ -542,7 +542,7 @@ func (a *disableLegacyDbfsImpl) Get(ctx context.Context, request GetDisableLegac headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &disableLegacyDbfs) return &disableLegacyDbfs, err @@ -557,7 +557,7 @@ func (a *disableLegacyDbfsImpl) Update(ctx context.Context, request UpdateDisabl headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &disableLegacyDbfs) return &disableLegacyDbfs, err @@ -615,7 +615,7 @@ func (a *enableExportNotebookImpl) GetEnableExportNotebook(ctx context.Context) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &enableExportNotebook) return &enableExportNotebook, err @@ -630,7 +630,7 @@ func (a *enableExportNotebookImpl) PatchEnableExportNotebook(ctx context.Context headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &enableExportNotebook) return &enableExportNotebook, err @@ -688,7 +688,7 @@ func (a *enableNotebookTableClipboardImpl) GetEnableNotebookTableClipboard(ctx c headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &enableNotebookTableClipboard) return &enableNotebookTableClipboard, err @@ -703,7 +703,7 @@ func (a *enableNotebookTableClipboardImpl) PatchEnableNotebookTableClipboard(ctx headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &enableNotebookTableClipboard) return &enableNotebookTableClipboard, err @@ -722,7 +722,7 @@ func (a *enableResultsDownloadingImpl) GetEnableResultsDownloading(ctx context.C headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &enableResultsDownloading) return &enableResultsDownloading, err @@ -737,7 +737,7 @@ func (a *enableResultsDownloadingImpl) PatchEnableResultsDownloading(ctx context headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &enableResultsDownloading) return &enableResultsDownloading, err @@ -756,7 +756,7 @@ func (a *enhancedSecurityMonitoringImpl) Get(ctx context.Context, request GetEnh headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &enhancedSecurityMonitoringSetting) return &enhancedSecurityMonitoringSetting, err @@ -771,7 +771,7 @@ func (a *enhancedSecurityMonitoringImpl) Update(ctx context.Context, request Upd headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &enhancedSecurityMonitoringSetting) return &enhancedSecurityMonitoringSetting, err @@ -819,7 +819,7 @@ func (a *ipAccessListsImpl) Create(ctx context.Context, request CreateIpAccessLi headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createIpAccessListResponse) return &createIpAccessListResponse, err @@ -831,7 +831,7 @@ func (a *ipAccessListsImpl) Delete(ctx context.Context, request DeleteIpAccessLi headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -845,7 +845,7 @@ func (a *ipAccessListsImpl) Get(ctx context.Context, request GetIpAccessListRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &fetchIpAccessListResponse) return &fetchIpAccessListResponse, err @@ -885,7 +885,7 @@ func (a *ipAccessListsImpl) internalList(ctx context.Context) (*ListIpAccessList headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listIpAccessListResponse) return &listIpAccessListResponse, err @@ -898,7 +898,7 @@ func (a *ipAccessListsImpl) Replace(ctx context.Context, request ReplaceIpAccess headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -911,7 +911,7 @@ func (a *ipAccessListsImpl) Update(ctx context.Context, request UpdateIpAccessLi headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -986,7 +986,7 @@ func (a *llmProxyPartnerPoweredWorkspaceImpl) Delete(ctx context.Context, reques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteLlmProxyPartnerPoweredWorkspaceResponse) return &deleteLlmProxyPartnerPoweredWorkspaceResponse, err @@ -1000,7 +1000,7 @@ func (a *llmProxyPartnerPoweredWorkspaceImpl) Get(ctx context.Context, request G headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &llmProxyPartnerPoweredWorkspace) return &llmProxyPartnerPoweredWorkspace, err @@ -1015,7 +1015,7 @@ func (a *llmProxyPartnerPoweredWorkspaceImpl) Update(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &llmProxyPartnerPoweredWorkspace) return &llmProxyPartnerPoweredWorkspace, err @@ -1299,7 +1299,7 @@ func (a *notificationDestinationsImpl) Create(ctx context.Context, request Creat headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, ¬ificationDestination) return ¬ificationDestination, err @@ -1312,7 +1312,7 @@ func (a *notificationDestinationsImpl) Delete(ctx context.Context, request Delet headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1326,7 +1326,7 @@ func (a *notificationDestinationsImpl) Get(ctx context.Context, request GetNotif headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, ¬ificationDestination) return ¬ificationDestination, err @@ -1371,7 +1371,7 @@ func (a *notificationDestinationsImpl) internalList(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listNotificationDestinationsResponse) return &listNotificationDestinationsResponse, err @@ -1386,7 +1386,7 @@ func (a *notificationDestinationsImpl) Update(ctx context.Context, request Updat headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, ¬ificationDestination) return ¬ificationDestination, err @@ -1444,7 +1444,7 @@ func (a *restrictWorkspaceAdminsImpl) Delete(ctx context.Context, request Delete headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteRestrictWorkspaceAdminsSettingResponse) return &deleteRestrictWorkspaceAdminsSettingResponse, err @@ -1458,7 +1458,7 @@ func (a *restrictWorkspaceAdminsImpl) Get(ctx context.Context, request GetRestri headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &restrictWorkspaceAdminsSetting) return &restrictWorkspaceAdminsSetting, err @@ -1473,7 +1473,7 @@ func (a *restrictWorkspaceAdminsImpl) Update(ctx context.Context, request Update headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &restrictWorkspaceAdminsSetting) return &restrictWorkspaceAdminsSetting, err @@ -1497,7 +1497,7 @@ func (a *sqlResultsDownloadImpl) Delete(ctx context.Context, request DeleteSqlRe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteSqlResultsDownloadResponse) return &deleteSqlResultsDownloadResponse, err @@ -1511,7 +1511,7 @@ func (a *sqlResultsDownloadImpl) Get(ctx context.Context, request GetSqlResultsD headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &sqlResultsDownload) return &sqlResultsDownload, err @@ -1526,7 +1526,7 @@ func (a *sqlResultsDownloadImpl) Update(ctx context.Context, request UpdateSqlRe headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &sqlResultsDownload) return &sqlResultsDownload, err @@ -1546,7 +1546,7 @@ func (a *tokenManagementImpl) CreateOboToken(ctx context.Context, request Create headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createOboTokenResponse) return &createOboTokenResponse, err @@ -1558,7 +1558,7 @@ func (a *tokenManagementImpl) Delete(ctx context.Context, request DeleteTokenMan headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1572,7 +1572,7 @@ func (a *tokenManagementImpl) Get(ctx context.Context, request GetTokenManagemen headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getTokenResponse) return &getTokenResponse, err @@ -1586,7 +1586,7 @@ func (a *tokenManagementImpl) GetPermissionLevels(ctx context.Context) (*GetToke headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &getTokenPermissionLevelsResponse) return &getTokenPermissionLevelsResponse, err @@ -1600,7 +1600,7 @@ func (a *tokenManagementImpl) GetPermissions(ctx context.Context) (*TokenPermiss headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &tokenPermissions) return &tokenPermissions, err @@ -1639,7 +1639,7 @@ func (a *tokenManagementImpl) internalList(ctx context.Context, request ListToke headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listTokensResponse) return &listTokensResponse, err @@ -1654,7 +1654,7 @@ func (a *tokenManagementImpl) SetPermissions(ctx context.Context, request TokenP headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &tokenPermissions) return &tokenPermissions, err @@ -1669,7 +1669,7 @@ func (a *tokenManagementImpl) UpdatePermissions(ctx context.Context, request Tok headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &tokenPermissions) return &tokenPermissions, err @@ -1689,7 +1689,7 @@ func (a *tokensImpl) Create(ctx context.Context, request CreateTokenRequest) (*C headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createTokenResponse) return &createTokenResponse, err @@ -1703,7 +1703,7 @@ func (a *tokensImpl) Delete(ctx context.Context, request RevokeTokenRequest) err headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -1743,7 +1743,7 @@ func (a *tokensImpl) internalList(ctx context.Context) (*ListPublicTokensRespons headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listPublicTokensResponse) return &listPublicTokensResponse, err @@ -1758,7 +1758,7 @@ func (a *tokensImpl) Update(ctx context.Context, request UpdateTokenRequest) (*U headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateTokenResponse) return &updateTokenResponse, err @@ -1777,7 +1777,7 @@ func (a *workspaceConfImpl) GetStatus(ctx context.Context, request GetStatusRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &workspaceConf) return &workspaceConf, err @@ -1790,7 +1790,7 @@ func (a *workspaceConfImpl) SetStatus(ctx context.Context, request WorkspaceConf headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err diff --git a/service/settingsv2/impl.go b/service/settingsv2/impl.go index a2ce211fe..dce501d6e 100755 --- a/service/settingsv2/impl.go +++ b/service/settingsv2/impl.go @@ -172,7 +172,7 @@ func (a *workspaceSettingsV2Impl) GetPublicWorkspaceSetting(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &setting) return &setting, err @@ -221,7 +221,7 @@ func (a *workspaceSettingsV2Impl) internalListWorkspaceSettingsMetadata(ctx cont headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listWorkspaceSettingsMetadataResponse) return &listWorkspaceSettingsMetadataResponse, err @@ -236,7 +236,7 @@ func (a *workspaceSettingsV2Impl) PatchPublicWorkspaceSetting(ctx context.Contex headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Setting, &setting) return &setting, err diff --git a/service/sharing/ext_api.go b/service/sharing/ext_api.go index f53c78d9f..dbd2e6da9 100644 --- a/service/sharing/ext_api.go +++ b/service/sharing/ext_api.go @@ -73,7 +73,7 @@ func (a *SharesAPI) internalList(ctx context.Context, request ListSharesRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSharesResponse) return &listSharesResponse, err diff --git a/service/sharing/impl.go b/service/sharing/impl.go index 9bacb9992..4734bc2ac 100644 --- a/service/sharing/impl.go +++ b/service/sharing/impl.go @@ -26,7 +26,7 @@ func (a *providersImpl) Create(ctx context.Context, request CreateProvider) (*Pr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &providerInfo) return &providerInfo, err @@ -38,7 +38,7 @@ func (a *providersImpl) Delete(ctx context.Context, request DeleteProviderReques headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -52,7 +52,7 @@ func (a *providersImpl) Get(ctx context.Context, request GetProviderRequest) (*P headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &providerInfo) return &providerInfo, err @@ -109,7 +109,7 @@ func (a *providersImpl) internalList(ctx context.Context, request ListProvidersR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listProvidersResponse) return &listProvidersResponse, err @@ -123,7 +123,7 @@ func (a *providersImpl) ListProviderShareAssets(ctx context.Context, request Lis headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listProviderShareAssetsResponse) return &listProviderShareAssetsResponse, err @@ -174,7 +174,7 @@ func (a *providersImpl) internalListShares(ctx context.Context, request ListShar headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listProviderSharesResponse) return &listProviderSharesResponse, err @@ -189,7 +189,7 @@ func (a *providersImpl) Update(ctx context.Context, request UpdateProvider) (*Pr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &providerInfo) return &providerInfo, err @@ -207,7 +207,7 @@ func (a *recipientActivationImpl) GetActivationUrlInfo(ctx context.Context, requ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, nil) return err @@ -221,7 +221,7 @@ func (a *recipientActivationImpl) RetrieveToken(ctx context.Context, request Ret headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &retrieveTokenResponse) return &retrieveTokenResponse, err @@ -241,7 +241,7 @@ func (a *recipientFederationPoliciesImpl) Create(ctx context.Context, request Cr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Policy, &federationPolicy) return &federationPolicy, err @@ -254,7 +254,7 @@ func (a *recipientFederationPoliciesImpl) Delete(ctx context.Context, request De headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -268,7 +268,7 @@ func (a *recipientFederationPoliciesImpl) GetFederationPolicy(ctx context.Contex headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &federationPolicy) return &federationPolicy, err @@ -317,7 +317,7 @@ func (a *recipientFederationPoliciesImpl) internalList(ctx context.Context, requ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listFederationPoliciesResponse) return &listFederationPoliciesResponse, err @@ -337,7 +337,7 @@ func (a *recipientsImpl) Create(ctx context.Context, request CreateRecipient) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &recipientInfo) return &recipientInfo, err @@ -349,7 +349,7 @@ func (a *recipientsImpl) Delete(ctx context.Context, request DeleteRecipientRequ headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -363,7 +363,7 @@ func (a *recipientsImpl) Get(ctx context.Context, request GetRecipientRequest) ( headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &recipientInfo) return &recipientInfo, err @@ -416,7 +416,7 @@ func (a *recipientsImpl) internalList(ctx context.Context, request ListRecipient headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listRecipientsResponse) return &listRecipientsResponse, err @@ -431,7 +431,7 @@ func (a *recipientsImpl) RotateToken(ctx context.Context, request RotateRecipien headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &recipientInfo) return &recipientInfo, err @@ -445,7 +445,7 @@ func (a *recipientsImpl) SharePermissions(ctx context.Context, request SharePerm headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getRecipientSharePermissionsResponse) return &getRecipientSharePermissionsResponse, err @@ -460,7 +460,7 @@ func (a *recipientsImpl) Update(ctx context.Context, request UpdateRecipient) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &recipientInfo) return &recipientInfo, err @@ -480,7 +480,7 @@ func (a *sharesImpl) Create(ctx context.Context, request CreateShare) (*ShareInf headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &shareInfo) return &shareInfo, err @@ -492,7 +492,7 @@ func (a *sharesImpl) Delete(ctx context.Context, request DeleteShareRequest) err headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -506,7 +506,7 @@ func (a *sharesImpl) Get(ctx context.Context, request GetShareRequest) (*ShareIn headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &shareInfo) return &shareInfo, err @@ -559,7 +559,7 @@ func (a *sharesImpl) internalListShares(ctx context.Context, request SharesListR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSharesResponse) return &listSharesResponse, err @@ -573,7 +573,7 @@ func (a *sharesImpl) SharePermissions(ctx context.Context, request SharePermissi headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getSharePermissionsResponse) return &getSharePermissionsResponse, err @@ -588,7 +588,7 @@ func (a *sharesImpl) Update(ctx context.Context, request UpdateShare) (*ShareInf headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &shareInfo) return &shareInfo, err @@ -603,7 +603,7 @@ func (a *sharesImpl) UpdatePermissions(ctx context.Context, request UpdateShareP headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateSharePermissionsResponse) return &updateSharePermissionsResponse, err diff --git a/service/sql/impl.go b/service/sql/impl.go index 1d13caa6e..b13cb7790 100755 --- a/service/sql/impl.go +++ b/service/sql/impl.go @@ -29,7 +29,7 @@ func (a *alertsImpl) Create(ctx context.Context, request CreateAlertRequest) (*A headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &alert) return &alert, err @@ -42,7 +42,7 @@ func (a *alertsImpl) Delete(ctx context.Context, request TrashAlertRequest) erro headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -56,7 +56,7 @@ func (a *alertsImpl) Get(ctx context.Context, request GetAlertRequest) (*Alert, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &alert) return &alert, err @@ -105,7 +105,7 @@ func (a *alertsImpl) internalList(ctx context.Context, request ListAlertsRequest headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listAlertsResponse) return &listAlertsResponse, err @@ -120,7 +120,7 @@ func (a *alertsImpl) Update(ctx context.Context, request UpdateAlertRequest) (*A headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &alert) return &alert, err @@ -140,7 +140,7 @@ func (a *alertsLegacyImpl) Create(ctx context.Context, request CreateAlert) (*Le headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &legacyAlert) return &legacyAlert, err @@ -153,7 +153,7 @@ func (a *alertsLegacyImpl) Delete(ctx context.Context, request DeleteAlertsLegac headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -167,7 +167,7 @@ func (a *alertsLegacyImpl) Get(ctx context.Context, request GetAlertsLegacyReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &legacyAlert) return &legacyAlert, err @@ -181,7 +181,7 @@ func (a *alertsLegacyImpl) List(ctx context.Context) ([]LegacyAlert, error) { headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &legacyAlertList) return legacyAlertList, err @@ -195,7 +195,7 @@ func (a *alertsLegacyImpl) Update(ctx context.Context, request EditAlert) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -215,7 +215,7 @@ func (a *alertsV2Impl) CreateAlert(ctx context.Context, request CreateAlertV2Req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Alert, &alertV2) return &alertV2, err @@ -229,7 +229,7 @@ func (a *alertsV2Impl) GetAlert(ctx context.Context, request GetAlertV2Request) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &alertV2) return &alertV2, err @@ -274,7 +274,7 @@ func (a *alertsV2Impl) internalListAlerts(ctx context.Context, request ListAlert headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listAlertsV2Response) return &listAlertsV2Response, err @@ -287,7 +287,7 @@ func (a *alertsV2Impl) TrashAlert(ctx context.Context, request TrashAlertV2Reque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -306,7 +306,7 @@ func (a *alertsV2Impl) UpdateAlert(ctx context.Context, request UpdateAlertV2Req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Alert, &alertV2) return &alertV2, err @@ -326,7 +326,7 @@ func (a *dashboardWidgetsImpl) Create(ctx context.Context, request CreateWidget) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &widget) return &widget, err @@ -339,7 +339,7 @@ func (a *dashboardWidgetsImpl) Delete(ctx context.Context, request DeleteDashboa headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -354,7 +354,7 @@ func (a *dashboardWidgetsImpl) Update(ctx context.Context, request UpdateWidgetR headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &widget) return &widget, err @@ -372,7 +372,7 @@ func (a *dashboardsImpl) Delete(ctx context.Context, request DeleteDashboardRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -386,7 +386,7 @@ func (a *dashboardsImpl) Get(ctx context.Context, request GetDashboardRequest) ( headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &dashboard) return &dashboard, err @@ -450,7 +450,7 @@ func (a *dashboardsImpl) internalList(ctx context.Context, request ListDashboard headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listResponse) return &listResponse, err @@ -463,7 +463,7 @@ func (a *dashboardsImpl) Restore(ctx context.Context, request RestoreDashboardRe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, nil) return err @@ -478,7 +478,7 @@ func (a *dashboardsImpl) Update(ctx context.Context, request DashboardEditConten headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &dashboard) return &dashboard, err @@ -497,7 +497,7 @@ func (a *dataSourcesImpl) List(ctx context.Context) ([]DataSource, error) { headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &dataSourceList) return dataSourceList, err @@ -516,7 +516,7 @@ func (a *dbsqlPermissionsImpl) Get(ctx context.Context, request GetDbsqlPermissi headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getResponse) return &getResponse, err @@ -531,7 +531,7 @@ func (a *dbsqlPermissionsImpl) Set(ctx context.Context, request SetRequest) (*Se headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &setResponse) return &setResponse, err @@ -546,7 +546,7 @@ func (a *dbsqlPermissionsImpl) TransferOwnership(ctx context.Context, request Tr headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &success) return &success, err @@ -566,7 +566,7 @@ func (a *queriesImpl) Create(ctx context.Context, request CreateQueryRequest) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &query) return &query, err @@ -579,7 +579,7 @@ func (a *queriesImpl) Delete(ctx context.Context, request TrashQueryRequest) err headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -593,7 +593,7 @@ func (a *queriesImpl) Get(ctx context.Context, request GetQueryRequest) (*Query, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &query) return &query, err @@ -642,7 +642,7 @@ func (a *queriesImpl) internalList(ctx context.Context, request ListQueriesReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listQueryObjectsResponse) return &listQueryObjectsResponse, err @@ -687,7 +687,7 @@ func (a *queriesImpl) internalListVisualizations(ctx context.Context, request Li headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listVisualizationsForQueryResponse) return &listVisualizationsForQueryResponse, err @@ -702,7 +702,7 @@ func (a *queriesImpl) Update(ctx context.Context, request UpdateQueryRequest) (* headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &query) return &query, err @@ -722,7 +722,7 @@ func (a *queriesLegacyImpl) Create(ctx context.Context, request QueryPostContent headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &legacyQuery) return &legacyQuery, err @@ -735,7 +735,7 @@ func (a *queriesLegacyImpl) Delete(ctx context.Context, request DeleteQueriesLeg headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -749,7 +749,7 @@ func (a *queriesLegacyImpl) Get(ctx context.Context, request GetQueriesLegacyReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &legacyQuery) return &legacyQuery, err @@ -815,7 +815,7 @@ func (a *queriesLegacyImpl) internalList(ctx context.Context, request ListQuerie headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &queryList) return &queryList, err @@ -828,7 +828,7 @@ func (a *queriesLegacyImpl) Restore(ctx context.Context, request RestoreQueriesL headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, nil) return err @@ -843,7 +843,7 @@ func (a *queriesLegacyImpl) Update(ctx context.Context, request QueryEditContent headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &legacyQuery) return &legacyQuery, err @@ -862,7 +862,7 @@ func (a *queryHistoryImpl) List(ctx context.Context, request ListQueryHistoryReq headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listQueriesResponse) return &listQueriesResponse, err @@ -882,7 +882,7 @@ func (a *queryVisualizationsImpl) Create(ctx context.Context, request CreateVisu headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &visualization) return &visualization, err @@ -895,7 +895,7 @@ func (a *queryVisualizationsImpl) Delete(ctx context.Context, request DeleteVisu headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -910,7 +910,7 @@ func (a *queryVisualizationsImpl) Update(ctx context.Context, request UpdateVisu headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &visualization) return &visualization, err @@ -930,7 +930,7 @@ func (a *queryVisualizationsLegacyImpl) Create(ctx context.Context, request Crea headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &legacyVisualization) return &legacyVisualization, err @@ -943,7 +943,7 @@ func (a *queryVisualizationsLegacyImpl) Delete(ctx context.Context, request Dele headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -958,7 +958,7 @@ func (a *queryVisualizationsLegacyImpl) Update(ctx context.Context, request Lega headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &legacyVisualization) return &legacyVisualization, err @@ -977,7 +977,7 @@ func (a *redashConfigImpl) GetConfig(ctx context.Context) (*ClientConfig, error) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &clientConfig) return &clientConfig, err @@ -994,7 +994,7 @@ func (a *statementExecutionImpl) CancelExecution(ctx context.Context, request Ca headers := make(map[string]string) cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, nil) return err @@ -1009,7 +1009,7 @@ func (a *statementExecutionImpl) ExecuteStatement(ctx context.Context, request E headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &statementResponse) return &statementResponse, err @@ -1023,7 +1023,7 @@ func (a *statementExecutionImpl) GetStatement(ctx context.Context, request GetSt headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &statementResponse) return &statementResponse, err @@ -1037,7 +1037,7 @@ func (a *statementExecutionImpl) GetStatementResultChunkN(ctx context.Context, r headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &resultData) return &resultData, err @@ -1057,7 +1057,7 @@ func (a *warehousesImpl) Create(ctx context.Context, request CreateWarehouseRequ headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createWarehouseResponse) return &createWarehouseResponse, err @@ -1076,7 +1076,7 @@ func (a *warehousesImpl) CreateDefaultWarehouseOverride(ctx context.Context, req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.DefaultWarehouseOverride, &defaultWarehouseOverride) return &defaultWarehouseOverride, err @@ -1089,7 +1089,7 @@ func (a *warehousesImpl) Delete(ctx context.Context, request DeleteWarehouseRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1102,7 +1102,7 @@ func (a *warehousesImpl) DeleteDefaultWarehouseOverride(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -1116,7 +1116,7 @@ func (a *warehousesImpl) Edit(ctx context.Context, request EditWarehouseRequest) headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -1130,7 +1130,7 @@ func (a *warehousesImpl) Get(ctx context.Context, request GetWarehouseRequest) ( headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getWarehouseResponse) return &getWarehouseResponse, err @@ -1144,7 +1144,7 @@ func (a *warehousesImpl) GetDefaultWarehouseOverride(ctx context.Context, reques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &defaultWarehouseOverride) return &defaultWarehouseOverride, err @@ -1158,7 +1158,7 @@ func (a *warehousesImpl) GetPermissionLevels(ctx context.Context, request GetWar headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getWarehousePermissionLevelsResponse) return &getWarehousePermissionLevelsResponse, err @@ -1172,7 +1172,7 @@ func (a *warehousesImpl) GetPermissions(ctx context.Context, request GetWarehous headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &warehousePermissions) return &warehousePermissions, err @@ -1186,7 +1186,7 @@ func (a *warehousesImpl) GetWorkspaceWarehouseConfig(ctx context.Context) (*GetW headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &getWorkspaceWarehouseConfigResponse) return &getWorkspaceWarehouseConfigResponse, err @@ -1231,7 +1231,7 @@ func (a *warehousesImpl) internalList(ctx context.Context, request ListWarehouse headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listWarehousesResponse) return &listWarehousesResponse, err @@ -1278,7 +1278,7 @@ func (a *warehousesImpl) internalListDefaultWarehouseOverrides(ctx context.Conte headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listDefaultWarehouseOverridesResponse) return &listDefaultWarehouseOverridesResponse, err @@ -1293,7 +1293,7 @@ func (a *warehousesImpl) SetPermissions(ctx context.Context, request WarehousePe headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &warehousePermissions) return &warehousePermissions, err @@ -1307,7 +1307,7 @@ func (a *warehousesImpl) SetWorkspaceWarehouseConfig(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, nil) return err @@ -1320,7 +1320,7 @@ func (a *warehousesImpl) Start(ctx context.Context, request StartRequest) error headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, nil) return err @@ -1333,7 +1333,7 @@ func (a *warehousesImpl) Stop(ctx context.Context, request StopRequest) error { headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, nil) return err @@ -1359,7 +1359,7 @@ func (a *warehousesImpl) UpdateDefaultWarehouseOverride(ctx context.Context, req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.DefaultWarehouseOverride, &defaultWarehouseOverride) return &defaultWarehouseOverride, err @@ -1374,7 +1374,7 @@ func (a *warehousesImpl) UpdatePermissions(ctx context.Context, request Warehous headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &warehousePermissions) return &warehousePermissions, err diff --git a/service/supervisoragents/impl.go b/service/supervisoragents/impl.go index f28df7f24..f450f02c1 100755 --- a/service/supervisoragents/impl.go +++ b/service/supervisoragents/impl.go @@ -28,7 +28,7 @@ func (a *supervisorAgentsImpl) CreateExample(ctx context.Context, request Create headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Example, &example) return &example, err @@ -43,7 +43,7 @@ func (a *supervisorAgentsImpl) CreateSupervisorAgent(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.SupervisorAgent, &supervisorAgent) return &supervisorAgent, err @@ -62,7 +62,7 @@ func (a *supervisorAgentsImpl) CreateTool(ctx context.Context, request CreateToo headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.Tool, &tool) return &tool, err @@ -75,7 +75,7 @@ func (a *supervisorAgentsImpl) DeleteExample(ctx context.Context, request Delete headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -88,7 +88,7 @@ func (a *supervisorAgentsImpl) DeleteSupervisorAgent(ctx context.Context, reques headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -101,7 +101,7 @@ func (a *supervisorAgentsImpl) DeleteTool(ctx context.Context, request DeleteToo headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -115,7 +115,7 @@ func (a *supervisorAgentsImpl) GetExample(ctx context.Context, request GetExampl headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &example) return &example, err @@ -129,7 +129,7 @@ func (a *supervisorAgentsImpl) GetPermissionLevels(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getSupervisorAgentPermissionLevelsResponse) return &getSupervisorAgentPermissionLevelsResponse, err @@ -143,7 +143,7 @@ func (a *supervisorAgentsImpl) GetPermissions(ctx context.Context, request GetSu headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &supervisorAgentPermissions) return &supervisorAgentPermissions, err @@ -157,7 +157,7 @@ func (a *supervisorAgentsImpl) GetSupervisorAgent(ctx context.Context, request G headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &supervisorAgent) return &supervisorAgent, err @@ -171,7 +171,7 @@ func (a *supervisorAgentsImpl) GetTool(ctx context.Context, request GetToolReque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &tool) return &tool, err @@ -216,7 +216,7 @@ func (a *supervisorAgentsImpl) internalListExamples(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listExamplesResponse) return &listExamplesResponse, err @@ -261,7 +261,7 @@ func (a *supervisorAgentsImpl) internalListSupervisorAgents(ctx context.Context, headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSupervisorAgentsResponse) return &listSupervisorAgentsResponse, err @@ -306,7 +306,7 @@ func (a *supervisorAgentsImpl) internalListTools(ctx context.Context, request Li headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listToolsResponse) return &listToolsResponse, err @@ -321,7 +321,7 @@ func (a *supervisorAgentsImpl) SetPermissions(ctx context.Context, request Super headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &supervisorAgentPermissions) return &supervisorAgentPermissions, err @@ -343,7 +343,7 @@ func (a *supervisorAgentsImpl) UpdateExample(ctx context.Context, request Update headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Example, &example) return &example, err @@ -358,7 +358,7 @@ func (a *supervisorAgentsImpl) UpdatePermissions(ctx context.Context, request Su headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &supervisorAgentPermissions) return &supervisorAgentPermissions, err @@ -380,7 +380,7 @@ func (a *supervisorAgentsImpl) UpdateSupervisorAgent(ctx context.Context, reques headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.SupervisorAgent, &supervisorAgent) return &supervisorAgent, err @@ -402,7 +402,7 @@ func (a *supervisorAgentsImpl) UpdateTool(ctx context.Context, request UpdateToo headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.Tool, &tool) return &tool, err diff --git a/service/tags/impl.go b/service/tags/impl.go index 63c62fa62..d6dd96187 100755 --- a/service/tags/impl.go +++ b/service/tags/impl.go @@ -26,7 +26,7 @@ func (a *tagPoliciesImpl) CreateTagPolicy(ctx context.Context, request CreateTag headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.TagPolicy, &tagPolicy) return &tagPolicy, err @@ -39,7 +39,7 @@ func (a *tagPoliciesImpl) DeleteTagPolicy(ctx context.Context, request DeleteTag headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -53,7 +53,7 @@ func (a *tagPoliciesImpl) GetTagPolicy(ctx context.Context, request GetTagPolicy headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &tagPolicy) return &tagPolicy, err @@ -108,7 +108,7 @@ func (a *tagPoliciesImpl) internalListTagPolicies(ctx context.Context, request L headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listTagPoliciesResponse) return &listTagPoliciesResponse, err @@ -127,7 +127,7 @@ func (a *tagPoliciesImpl) UpdateTagPolicy(ctx context.Context, request UpdateTag headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.TagPolicy, &tagPolicy) return &tagPolicy, err @@ -147,7 +147,7 @@ func (a *workspaceEntityTagAssignmentsImpl) CreateTagAssignment(ctx context.Cont headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request.TagAssignment, &tagAssignment) return &tagAssignment, err @@ -160,7 +160,7 @@ func (a *workspaceEntityTagAssignmentsImpl) DeleteTagAssignment(ctx context.Cont headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -174,7 +174,7 @@ func (a *workspaceEntityTagAssignmentsImpl) GetTagAssignment(ctx context.Context headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &tagAssignment) return &tagAssignment, err @@ -219,7 +219,7 @@ func (a *workspaceEntityTagAssignmentsImpl) internalListTagAssignments(ctx conte headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listTagAssignmentsResponse) return &listTagAssignmentsResponse, err @@ -238,7 +238,7 @@ func (a *workspaceEntityTagAssignmentsImpl) UpdateTagAssignment(ctx context.Cont headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request.TagAssignment, &tagAssignment) return &tagAssignment, err diff --git a/service/vectorsearch/impl.go b/service/vectorsearch/impl.go index e03f7c619..54603d8bb 100755 --- a/service/vectorsearch/impl.go +++ b/service/vectorsearch/impl.go @@ -26,7 +26,7 @@ func (a *vectorSearchEndpointsImpl) CreateEndpoint(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &endpointInfo) return &endpointInfo, err @@ -39,7 +39,7 @@ func (a *vectorSearchEndpointsImpl) DeleteEndpoint(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -53,7 +53,7 @@ func (a *vectorSearchEndpointsImpl) GetEndpoint(ctx context.Context, request Get headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &endpointInfo) return &endpointInfo, err @@ -67,7 +67,7 @@ func (a *vectorSearchEndpointsImpl) GetPermissionLevels(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getVectorSearchEndpointPermissionLevelsResponse) return &getVectorSearchEndpointPermissionLevelsResponse, err @@ -81,7 +81,7 @@ func (a *vectorSearchEndpointsImpl) GetPermissions(ctx context.Context, request headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &vectorSearchEndpointPermissions) return &vectorSearchEndpointPermissions, err @@ -126,7 +126,7 @@ func (a *vectorSearchEndpointsImpl) internalListEndpoints(ctx context.Context, r headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listEndpointResponse) return &listEndpointResponse, err @@ -141,7 +141,7 @@ func (a *vectorSearchEndpointsImpl) PatchEndpoint(ctx context.Context, request P headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &endpointInfo) return &endpointInfo, err @@ -156,7 +156,7 @@ func (a *vectorSearchEndpointsImpl) RetrieveUserVisibleMetrics(ctx context.Conte headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &retrieveUserVisibleMetricsResponse) return &retrieveUserVisibleMetricsResponse, err @@ -171,7 +171,7 @@ func (a *vectorSearchEndpointsImpl) SetPermissions(ctx context.Context, request headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &vectorSearchEndpointPermissions) return &vectorSearchEndpointPermissions, err @@ -186,7 +186,7 @@ func (a *vectorSearchEndpointsImpl) UpdateEndpointBudgetPolicy(ctx context.Conte headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &patchEndpointBudgetPolicyResponse) return &patchEndpointBudgetPolicyResponse, err @@ -201,7 +201,7 @@ func (a *vectorSearchEndpointsImpl) UpdateEndpointCustomTags(ctx context.Context headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &updateEndpointCustomTagsResponse) return &updateEndpointCustomTagsResponse, err @@ -216,7 +216,7 @@ func (a *vectorSearchEndpointsImpl) UpdatePermissions(ctx context.Context, reque headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &vectorSearchEndpointPermissions) return &vectorSearchEndpointPermissions, err @@ -236,7 +236,7 @@ func (a *vectorSearchIndexesImpl) CreateIndex(ctx context.Context, request Creat headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &vectorIndex) return &vectorIndex, err @@ -250,7 +250,7 @@ func (a *vectorSearchIndexesImpl) DeleteDataVectorIndex(ctx context.Context, req headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, &deleteDataVectorIndexResponse) return &deleteDataVectorIndexResponse, err @@ -263,7 +263,7 @@ func (a *vectorSearchIndexesImpl) DeleteIndex(ctx context.Context, request Delet headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -277,7 +277,7 @@ func (a *vectorSearchIndexesImpl) GetIndex(ctx context.Context, request GetIndex headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &vectorIndex) return &vectorIndex, err @@ -322,7 +322,7 @@ func (a *vectorSearchIndexesImpl) internalListIndexes(ctx context.Context, reque headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listVectorIndexesResponse) return &listVectorIndexesResponse, err @@ -337,7 +337,7 @@ func (a *vectorSearchIndexesImpl) QueryIndex(ctx context.Context, request QueryV headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &queryVectorIndexResponse) return &queryVectorIndexResponse, err @@ -352,7 +352,7 @@ func (a *vectorSearchIndexesImpl) QueryNextPage(ctx context.Context, request Que headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &queryVectorIndexResponse) return &queryVectorIndexResponse, err @@ -367,7 +367,7 @@ func (a *vectorSearchIndexesImpl) ScanIndex(ctx context.Context, request ScanVec headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &scanVectorIndexResponse) return &scanVectorIndexResponse, err @@ -380,7 +380,7 @@ func (a *vectorSearchIndexesImpl) SyncIndex(ctx context.Context, request SyncInd headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, nil, nil) return err @@ -395,7 +395,7 @@ func (a *vectorSearchIndexesImpl) UpsertDataVectorIndex(ctx context.Context, req headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &upsertDataVectorIndexResponse) return &upsertDataVectorIndexResponse, err diff --git a/service/workspace/ext_utilities.go b/service/workspace/ext_utilities.go index 443f164c6..5f8ba1f55 100644 --- a/service/workspace/ext_utilities.go +++ b/service/workspace/ext_utilities.go @@ -209,7 +209,7 @@ func (a *WorkspaceAPI) Upload(ctx context.Context, path string, r io.Reader, opt } cfg := a.workspaceImpl.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } return a.workspaceImpl.client.Do(ctx, "POST", "/api/2.0/workspace/import", headers, nil, buf.Bytes(), nil) } @@ -247,7 +247,7 @@ func (a *WorkspaceAPI) Download(ctx context.Context, path string, opts ...Downlo headers := map[string]string{"Content-Type": "application/json"} cfg := a.workspaceImpl.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.workspaceImpl.client.Do(ctx, "GET", "/api/2.0/workspace/export", headers, nil, query, &buf) if err != nil { diff --git a/service/workspace/impl.go b/service/workspace/impl.go index 67e6df136..e0a584693 100755 --- a/service/workspace/impl.go +++ b/service/workspace/impl.go @@ -26,7 +26,7 @@ func (a *gitCredentialsImpl) Create(ctx context.Context, request CreateCredentia headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createCredentialsResponse) return &createCredentialsResponse, err @@ -39,7 +39,7 @@ func (a *gitCredentialsImpl) Delete(ctx context.Context, request DeleteCredentia headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -53,7 +53,7 @@ func (a *gitCredentialsImpl) Get(ctx context.Context, request GetCredentialsRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getCredentialsResponse) return &getCredentialsResponse, err @@ -92,7 +92,7 @@ func (a *gitCredentialsImpl) internalList(ctx context.Context, request ListCrede headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listCredentialsResponse) return &listCredentialsResponse, err @@ -106,7 +106,7 @@ func (a *gitCredentialsImpl) Update(ctx context.Context, request UpdateCredentia headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -126,7 +126,7 @@ func (a *reposImpl) Create(ctx context.Context, request CreateRepoRequest) (*Cre headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, &createRepoResponse) return &createRepoResponse, err @@ -139,7 +139,7 @@ func (a *reposImpl) Delete(ctx context.Context, request DeleteRepoRequest) error headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodDelete, path, headers, queryParams, request, nil) return err @@ -153,7 +153,7 @@ func (a *reposImpl) Get(ctx context.Context, request GetRepoRequest) (*GetRepoRe headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getRepoResponse) return &getRepoResponse, err @@ -167,7 +167,7 @@ func (a *reposImpl) GetPermissionLevels(ctx context.Context, request GetRepoPerm headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getRepoPermissionLevelsResponse) return &getRepoPermissionLevelsResponse, err @@ -181,7 +181,7 @@ func (a *reposImpl) GetPermissions(ctx context.Context, request GetRepoPermissio headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &repoPermissions) return &repoPermissions, err @@ -228,7 +228,7 @@ func (a *reposImpl) internalList(ctx context.Context, request ListReposRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listReposResponse) return &listReposResponse, err @@ -243,7 +243,7 @@ func (a *reposImpl) SetPermissions(ctx context.Context, request RepoPermissionsR headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &repoPermissions) return &repoPermissions, err @@ -257,7 +257,7 @@ func (a *reposImpl) Update(ctx context.Context, request UpdateRepoRequest) error headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, nil) return err @@ -272,7 +272,7 @@ func (a *reposImpl) UpdatePermissions(ctx context.Context, request RepoPermissio headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &repoPermissions) return &repoPermissions, err @@ -290,7 +290,7 @@ func (a *secretsImpl) CreateScope(ctx context.Context, request CreateScope) erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -303,7 +303,7 @@ func (a *secretsImpl) DeleteAcl(ctx context.Context, request DeleteAcl) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -316,7 +316,7 @@ func (a *secretsImpl) DeleteScope(ctx context.Context, request DeleteScope) erro headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -330,7 +330,7 @@ func (a *secretsImpl) DeleteSecret(ctx context.Context, request DeleteSecret) er headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -344,7 +344,7 @@ func (a *secretsImpl) GetAcl(ctx context.Context, request GetAclRequest) (*AclIt headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &aclItem) return &aclItem, err @@ -358,7 +358,7 @@ func (a *secretsImpl) GetSecret(ctx context.Context, request GetSecretRequest) ( headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getSecretResponse) return &getSecretResponse, err @@ -423,7 +423,7 @@ func (a *secretsImpl) internalListAcls(ctx context.Context, request ListAclsRequ headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listAclsResponse) return &listAclsResponse, err @@ -483,7 +483,7 @@ func (a *secretsImpl) internalListScopes(ctx context.Context) (*ListScopesRespon headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, nil, nil, &listScopesResponse) return &listScopesResponse, err @@ -554,7 +554,7 @@ func (a *secretsImpl) internalListSecrets(ctx context.Context, request ListSecre headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listSecretsResponse) return &listSecretsResponse, err @@ -567,7 +567,7 @@ func (a *secretsImpl) PutAcl(ctx context.Context, request PutAcl) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -580,7 +580,7 @@ func (a *secretsImpl) PutSecret(ctx context.Context, request PutSecret) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -599,7 +599,7 @@ func (a *workspaceImpl) Delete(ctx context.Context, request Delete) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -613,7 +613,7 @@ func (a *workspaceImpl) Export(ctx context.Context, request ExportRequest) (*Exp headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &exportResponse) return &exportResponse, err @@ -627,7 +627,7 @@ func (a *workspaceImpl) GetPermissionLevels(ctx context.Context, request GetWork headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &getWorkspaceObjectPermissionLevelsResponse) return &getWorkspaceObjectPermissionLevelsResponse, err @@ -641,7 +641,7 @@ func (a *workspaceImpl) GetPermissions(ctx context.Context, request GetWorkspace headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &workspaceObjectPermissions) return &workspaceObjectPermissions, err @@ -655,7 +655,7 @@ func (a *workspaceImpl) GetStatus(ctx context.Context, request GetStatusRequest) headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &objectInfo) return &objectInfo, err @@ -669,7 +669,7 @@ func (a *workspaceImpl) Import(ctx context.Context, request Import) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -712,7 +712,7 @@ func (a *workspaceImpl) internalList(ctx context.Context, request ListWorkspaceR headers["Accept"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodGet, path, headers, queryParams, request, &listResponse) return &listResponse, err @@ -726,7 +726,7 @@ func (a *workspaceImpl) Mkdirs(ctx context.Context, request Mkdirs) error { headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPost, path, headers, queryParams, request, nil) return err @@ -741,7 +741,7 @@ func (a *workspaceImpl) SetPermissions(ctx context.Context, request WorkspaceObj headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPut, path, headers, queryParams, request, &workspaceObjectPermissions) return &workspaceObjectPermissions, err @@ -756,7 +756,7 @@ func (a *workspaceImpl) UpdatePermissions(ctx context.Context, request Workspace headers["Content-Type"] = "application/json" cfg := a.client.Config if cfg.WorkspaceID != "" { - headers["X-Databricks-Org-Id"] = cfg.WorkspaceID + headers["X-Databricks-Workspace-Id"] = cfg.WorkspaceID } err := a.client.Do(ctx, http.MethodPatch, path, headers, queryParams, request, &workspaceObjectPermissions) return &workspaceObjectPermissions, err diff --git a/workspace_functions.go b/workspace_functions.go index 7e0fbb5a6..a2ba21db1 100644 --- a/workspace_functions.go +++ b/workspace_functions.go @@ -10,11 +10,12 @@ import ( // CurrentWorkspaceID returns the workspace ID of the workspace that this client is // connected to. // -// The ID is fetched from the X-Databricks-Org-Id response header on -// /api/2.0/preview/scim/v2/Me. On unified (SPOG) hosts, Config.WorkspaceID is -// sent in the X-Databricks-Org-Id request header to route the call to the -// correct workspace; on unified hosts with no WorkspaceID set, the request has -// no routing information and will fail. +// On unified (SPOG) hosts, Config.WorkspaceID is sent in the +// X-Databricks-Workspace-Id request header to route the call to the correct +// workspace; on unified hosts with no WorkspaceID set, the request has no +// routing information and will fail. The ID is read back from the +// X-Databricks-Org-Id response header on /api/2.0/preview/scim/v2/Me — the +// server still emits the legacy response header name during the migration. func (w *WorkspaceClient) CurrentWorkspaceID(ctx context.Context) (int64, error) { var workspaceIdStr string opts := []httpclient.DoOption{ @@ -22,7 +23,7 @@ func (w *WorkspaceClient) CurrentWorkspaceID(ctx context.Context) (int64, error) httpclient.WithRequestData(map[string]string{"excludedAttributes": "entitlements"}), } if w.Config != nil && w.Config.WorkspaceID != "" { - opts = append(opts, httpclient.WithRequestHeader("X-Databricks-Org-Id", w.Config.WorkspaceID)) + opts = append(opts, httpclient.WithRequestHeader("X-Databricks-Workspace-Id", w.Config.WorkspaceID)) } err := w.apiClient.Do(ctx, "GET", "/api/2.0/preview/scim/v2/Me", opts...) if err != nil { diff --git a/workspace_functions_test.go b/workspace_functions_test.go index eccf7106a..c99e139ed 100644 --- a/workspace_functions_test.go +++ b/workspace_functions_test.go @@ -9,17 +9,19 @@ import ( "github.com/stretchr/testify/require" ) -func TestCurrentWorkspaceIDSendsOrgIdHeaderWhenConfigHasWorkspaceID(t *testing.T) { +func TestCurrentWorkspaceIDSendsWorkspaceIdHeaderWhenConfigHasWorkspaceID(t *testing.T) { // On unified (SPOG) hosts, requests to /api/2.0/preview/scim/v2/Me must - // carry an X-Databricks-Org-Id header so the gateway can route them to the - // correct workspace. When Config.WorkspaceID is set we forward it on the - // request, and the server echoes it back on the response header. + // carry an X-Databricks-Workspace-Id header so the gateway can route them + // to the correct workspace. When Config.WorkspaceID is set we forward it + // on the request; the server echoes the ID back on the legacy + // X-Databricks-Org-Id response header (the response side hasn't been + // migrated yet). var meCalls int var gotOrgIdHeader string server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.URL.Path == "/api/2.0/preview/scim/v2/Me" { meCalls++ - gotOrgIdHeader = r.Header.Get("X-Databricks-Org-Id") + gotOrgIdHeader = r.Header.Get("X-Databricks-Workspace-Id") w.Header().Set("X-Databricks-Org-Id", "7474644166319138") w.Write([]byte(`{}`)) return @@ -66,17 +68,17 @@ func TestCurrentWorkspaceIDExcludesEntitlements(t *testing.T) { assert.Equal(t, "excludedAttributes=entitlements", gotRawQuery) } -func TestCurrentWorkspaceIDOmitsOrgIdHeaderWhenConfigMissingWorkspaceID(t *testing.T) { +func TestCurrentWorkspaceIDOmitsWorkspaceIdHeaderWhenConfigMissingWorkspaceID(t *testing.T) { // On legacy workspace hosts the host itself identifies the workspace, so // no routing header is needed. When Config.WorkspaceID is empty we send - // the request without X-Databricks-Org-Id and read the ID from the - // response header. + // the request without X-Databricks-Workspace-Id and read the ID from the + // X-Databricks-Org-Id response header. var meCalls int var gotOrgIdHeader string server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.URL.Path == "/api/2.0/preview/scim/v2/Me" { meCalls++ - gotOrgIdHeader = r.Header.Get("X-Databricks-Org-Id") + gotOrgIdHeader = r.Header.Get("X-Databricks-Workspace-Id") w.Header().Set("X-Databricks-Org-Id", "7474644166319138") w.Write([]byte(`{}`)) return