@databricks/sdk-uc-entitytagassignments 0.0.0-dev → 0.1.0-dev.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,80 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+ import { Temporal } from '@js-temporal/polyfill';
3
+ import { FieldMask } from '@databricks/sdk-core/wkt';
4
+ import { z } from 'zod';
5
+ /** Enum representing the source type of a tag assignment */
6
+ export var TagAssignmentSourceType;
7
+ (function (TagAssignmentSourceType) {
8
+ /** Includes (but not limited to) tags manually assigned by users */
9
+ TagAssignmentSourceType["TAG_ASSIGNMENT_SOURCE_TYPE_UNSPECIFIED"] = "TAG_ASSIGNMENT_SOURCE_TYPE_UNSPECIFIED";
10
+ /** Automatically assigned by Data Classification */
11
+ TagAssignmentSourceType["TAG_ASSIGNMENT_SOURCE_TYPE_SYSTEM_DATA_CLASSIFICATION"] = "TAG_ASSIGNMENT_SOURCE_TYPE_SYSTEM_DATA_CLASSIFICATION";
12
+ })(TagAssignmentSourceType || (TagAssignmentSourceType = {}));
13
+ export const unmarshalEntityTagAssignmentSchema = z
14
+ .object({
15
+ entity_name: z.string().optional(),
16
+ tag_key: z.string().optional(),
17
+ tag_value: z.string().optional(),
18
+ entity_type: z.string().optional(),
19
+ update_time: z
20
+ .string()
21
+ .transform(s => Temporal.Instant.from(s))
22
+ .optional(),
23
+ updated_by: z.string().optional(),
24
+ source_type: z.enum(TagAssignmentSourceType).optional(),
25
+ })
26
+ .transform(d => ({
27
+ entityName: d.entity_name,
28
+ tagKey: d.tag_key,
29
+ tagValue: d.tag_value,
30
+ entityType: d.entity_type,
31
+ updateTime: d.update_time,
32
+ updatedBy: d.updated_by,
33
+ sourceType: d.source_type,
34
+ }));
35
+ export const unmarshalListEntityTagAssignmentsResponseSchema = z
36
+ .object({
37
+ tag_assignments: z
38
+ .array(z.lazy(() => unmarshalEntityTagAssignmentSchema))
39
+ .optional(),
40
+ next_page_token: z.string().optional(),
41
+ })
42
+ .transform(d => ({
43
+ tagAssignments: d.tag_assignments,
44
+ nextPageToken: d.next_page_token,
45
+ }));
46
+ export const marshalEntityTagAssignmentSchema = z
47
+ .object({
48
+ entityName: z.string().optional(),
49
+ tagKey: z.string().optional(),
50
+ tagValue: z.string().optional(),
51
+ entityType: z.string().optional(),
52
+ updateTime: z
53
+ .any()
54
+ .transform((d) => d.toString())
55
+ .optional(),
56
+ updatedBy: z.string().optional(),
57
+ sourceType: z.enum(TagAssignmentSourceType).optional(),
58
+ })
59
+ .transform(d => ({
60
+ entity_name: d.entityName,
61
+ tag_key: d.tagKey,
62
+ tag_value: d.tagValue,
63
+ entity_type: d.entityType,
64
+ update_time: d.updateTime,
65
+ updated_by: d.updatedBy,
66
+ source_type: d.sourceType,
67
+ }));
68
+ const entityTagAssignmentFieldMaskSchema = {
69
+ entityName: { wire: 'entity_name' },
70
+ entityType: { wire: 'entity_type' },
71
+ sourceType: { wire: 'source_type' },
72
+ tagKey: { wire: 'tag_key' },
73
+ tagValue: { wire: 'tag_value' },
74
+ updateTime: { wire: 'update_time' },
75
+ updatedBy: { wire: 'updated_by' },
76
+ };
77
+ export function entityTagAssignmentFieldMask(...paths) {
78
+ return FieldMask.build(paths, entityTagAssignmentFieldMaskSchema);
79
+ }
80
+ //# sourceMappingURL=model.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/v1/model.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAE/E,OAAO,EAAC,QAAQ,EAAC,MAAM,uBAAuB,CAAC;AAC/C,OAAO,EAAC,SAAS,EAAC,MAAM,0BAA0B,CAAC;AAEnD,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,4DAA4D;AAC5D,MAAM,CAAN,IAAY,uBAKX;AALD,WAAY,uBAAuB;IACjC,oEAAoE;IACpE,4GAAiF,CAAA;IACjF,oDAAoD;IACpD,0IAA+G,CAAA;AACjH,CAAC,EALW,uBAAuB,KAAvB,uBAAuB,QAKlC;AAsED,MAAM,CAAC,MAAM,kCAAkC,GAC7C,CAAC;KACE,MAAM,CAAC;IACN,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC9B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAChC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,WAAW,EAAE,CAAC;SACX,MAAM,EAAE;SACR,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SACxC,QAAQ,EAAE;IACb,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,QAAQ,EAAE;CACxD,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,UAAU,EAAE,CAAC,CAAC,WAAW;IACzB,MAAM,EAAE,CAAC,CAAC,OAAO;IACjB,QAAQ,EAAE,CAAC,CAAC,SAAS;IACrB,UAAU,EAAE,CAAC,CAAC,WAAW;IACzB,UAAU,EAAE,CAAC,CAAC,WAAW;IACzB,SAAS,EAAE,CAAC,CAAC,UAAU;IACvB,UAAU,EAAE,CAAC,CAAC,WAAW;CAC1B,CAAC,CAAC,CAAC;AAER,MAAM,CAAC,MAAM,+CAA+C,GAC1D,CAAC;KACE,MAAM,CAAC;IACN,eAAe,EAAE,CAAC;SACf,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,kCAAkC,CAAC,CAAC;SACvD,QAAQ,EAAE;IACb,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACvC,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,cAAc,EAAE,CAAC,CAAC,eAAe;IACjC,aAAa,EAAE,CAAC,CAAC,eAAe;CACjC,CAAC,CAAC,CAAC;AAER,MAAM,CAAC,MAAM,gCAAgC,GAAc,CAAC;KACzD,MAAM,CAAC;IACN,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC7B,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,UAAU,EAAE,CAAC;SACV,GAAG,EAAE;SACL,SAAS,CAAC,CAAC,CAAmB,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;SAChD,QAAQ,EAAE;IACb,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAChC,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,QAAQ,EAAE;CACvD,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,WAAW,EAAE,CAAC,CAAC,UAAU;IACzB,OAAO,EAAE,CAAC,CAAC,MAAM;IACjB,SAAS,EAAE,CAAC,CAAC,QAAQ;IACrB,WAAW,EAAE,CAAC,CAAC,UAAU;IACzB,WAAW,EAAE,CAAC,CAAC,UAAU;IACzB,UAAU,EAAE,CAAC,CAAC,SAAS;IACvB,WAAW,EAAE,CAAC,CAAC,UAAU;CAC1B,CAAC,CAAC,CAAC;AAEN,MAAM,kCAAkC,GAAoB;IAC1D,UAAU,EAAE,EAAC,IAAI,EAAE,aAAa,EAAC;IACjC,UAAU,EAAE,EAAC,IAAI,EAAE,aAAa,EAAC;IACjC,UAAU,EAAE,EAAC,IAAI,EAAE,aAAa,EAAC;IACjC,MAAM,EAAE,EAAC,IAAI,EAAE,SAAS,EAAC;IACzB,QAAQ,EAAE,EAAC,IAAI,EAAE,WAAW,EAAC;IAC7B,UAAU,EAAE,EAAC,IAAI,EAAE,aAAa,EAAC;IACjC,SAAS,EAAE,EAAC,IAAI,EAAE,YAAY,EAAC;CAChC,CAAC;AAEF,MAAM,UAAU,4BAA4B,CAC1C,GAAG,KAAe;IAElB,OAAO,SAAS,CAAC,KAAK,CACpB,KAAK,EACL,kCAAkC,CACnC,CAAC;AACJ,CAAC"}
@@ -0,0 +1,5 @@
1
+ import type { HttpClient } from '@databricks/sdk-core/http';
2
+ import type { ClientOptions } from '@databricks/sdk-options/client';
3
+ /** Creates a new HTTP client with the given options. */
4
+ export declare function newHttpClient(options?: ClientOptions): HttpClient;
5
+ //# sourceMappingURL=transport.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"transport.d.ts","sourceRoot":"","sources":["../../src/v1/transport.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,UAAU,EAGX,MAAM,2BAA2B,CAAC;AAEnC,OAAO,KAAK,EAAC,aAAa,EAAC,MAAM,gCAAgC,CAAC;AAElE,wDAAwD;AACxD,wBAAgB,aAAa,CAAC,OAAO,CAAC,EAAE,aAAa,GAAG,UAAU,CAwBjE"}
@@ -0,0 +1,57 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+ import { defaultCredentials } from '@databricks/sdk-auth/credentials';
3
+ import { newFetchHttpClient } from '@databricks/sdk-core/http';
4
+ /** Creates a new HTTP client with the given options. */
5
+ export function newHttpClient(options) {
6
+ const opts = options ?? {};
7
+ // If an HTTP client is provided, use it as-is. Throw if other options are
8
+ // also set, since they would be silently ignored.
9
+ if (opts.httpClient !== undefined) {
10
+ if (opts.credentials !== undefined || opts.timeout !== undefined) {
11
+ throw new Error('httpClient cannot be combined with credentials or timeout');
12
+ }
13
+ return opts.httpClient;
14
+ }
15
+ const credentials = opts.credentials ?? defaultCredentials();
16
+ const base = newFetchHttpClient();
17
+ let client = new AuthHttpClient(base, credentials);
18
+ if (opts.timeout !== undefined) {
19
+ client = new TimeoutHttpClient(client, opts.timeout);
20
+ }
21
+ return client;
22
+ }
23
+ /** Wraps an HttpClient and adds authentication headers to requests. */
24
+ class AuthHttpClient {
25
+ base;
26
+ credentials;
27
+ constructor(base, credentials) {
28
+ this.base = base;
29
+ this.credentials = credentials;
30
+ }
31
+ async send(request) {
32
+ const authHeaders = await this.credentials.authHeaders();
33
+ // Do not modify the original request.
34
+ const headers = new Headers(request.headers);
35
+ for (const h of authHeaders) {
36
+ headers.set(h.key, h.value);
37
+ }
38
+ return this.base.send({ ...request, headers });
39
+ }
40
+ }
41
+ /** Wraps an HttpClient and applies a default timeout to requests. */
42
+ class TimeoutHttpClient {
43
+ base;
44
+ timeout;
45
+ constructor(base, timeout) {
46
+ this.base = base;
47
+ this.timeout = timeout;
48
+ }
49
+ async send(request) {
50
+ const timeoutSignal = AbortSignal.timeout(this.timeout);
51
+ const signal = request.signal !== undefined
52
+ ? AbortSignal.any([request.signal, timeoutSignal])
53
+ : timeoutSignal;
54
+ return this.base.send({ ...request, signal });
55
+ }
56
+ }
57
+ //# sourceMappingURL=transport.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"transport.js","sourceRoot":"","sources":["../../src/v1/transport.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,kBAAkB,EAAC,MAAM,kCAAkC,CAAC;AAMpE,OAAO,EAAC,kBAAkB,EAAC,MAAM,2BAA2B,CAAC;AAG7D,wDAAwD;AACxD,MAAM,UAAU,aAAa,CAAC,OAAuB;IACnD,MAAM,IAAI,GAAG,OAAO,IAAI,EAAE,CAAC;IAE3B,0EAA0E;IAC1E,kDAAkD;IAClD,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CACb,2DAA2D,CAC5D,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,kBAAkB,EAAE,CAAC;IAE7D,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAC;IAClC,IAAI,MAAM,GAAe,IAAI,cAAc,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAE/D,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAC/B,MAAM,GAAG,IAAI,iBAAiB,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,uEAAuE;AACvE,MAAM,cAAc;IAEC;IACA;IAFnB,YACmB,IAAgB,EAChB,WAAwB;QADxB,SAAI,GAAJ,IAAI,CAAY;QAChB,gBAAW,GAAX,WAAW,CAAa;IACxC,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;QACzD,sCAAsC;QACtC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC7C,KAAK,MAAM,CAAC,IAAI,WAAW,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAC,GAAG,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;IAC/C,CAAC;CACF;AAED,qEAAqE;AACrE,MAAM,iBAAiB;IAEF;IACA;IAFnB,YACmB,IAAgB,EAChB,OAAe;QADf,SAAI,GAAJ,IAAI,CAAY;QAChB,YAAO,GAAP,OAAO,CAAQ;IAC/B,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxD,MAAM,MAAM,GACV,OAAO,CAAC,MAAM,KAAK,SAAS;YAC1B,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAClD,CAAC,CAAC,aAAa,CAAC;QACpB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAC,GAAG,OAAO,EAAE,MAAM,EAAC,CAAC,CAAC;IAC9C,CAAC;CACF"}
@@ -0,0 +1,22 @@
1
+ import type { Call } from '@databricks/sdk-core/api';
2
+ import type { HttpClient, HttpRequest } from '@databricks/sdk-core/http';
3
+ import type { Logger } from '@databricks/sdk-core/logger';
4
+ import type { CallOptions } from '@databricks/sdk-options/call';
5
+ import type { z } from 'zod';
6
+ export interface HttpCallOptions {
7
+ readonly request: HttpRequest;
8
+ readonly httpClient: HttpClient;
9
+ readonly logger: Logger;
10
+ }
11
+ /**
12
+ * Translates public CallOptions to the internal Options shape accepted by
13
+ * execute(). Even though the shapes match today, this isolates the public
14
+ * API from the executor's internal type so they can diverge.
15
+ */
16
+ export declare function executeCall(call: Call, options?: CallOptions): Promise<void>;
17
+ export declare function executeHttpCall(opts: HttpCallOptions): Promise<Uint8Array>;
18
+ export declare function buildHttpRequest(method: string, url: string, headers: Headers, signal?: AbortSignal, body?: string | ReadableStream<Uint8Array>): HttpRequest;
19
+ export declare function parseResponse<T>(body: Uint8Array, schema: z.ZodType<T>): T;
20
+ export declare function marshalRequest(data: unknown, schema: z.ZodType): string;
21
+ export declare function flattenQueryParams(prefix: string, value: unknown, params: URLSearchParams): void;
22
+ //# sourceMappingURL=utils.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAC,IAAI,EAAU,MAAM,0BAA0B,CAAC;AAG5D,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EAEZ,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAC,MAAM,EAAC,MAAM,6BAA6B,CAAC;AACxD,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,8BAA8B,CAAC;AAE9D,OAAO,KAAK,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAO3B,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC;IAC9B,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAChC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;CACzB;AAED;;;;GAIG;AACH,wBAAsB,WAAW,CAC/B,IAAI,EAAE,IAAI,EACV,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAAC,IAAI,CAAC,CASf;AA2BD,wBAAsB,eAAe,CACnC,IAAI,EAAE,eAAe,GACpB,OAAO,CAAC,UAAU,CAAC,CA2BrB;AAED,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,MAAM,EACd,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,OAAO,EAChB,MAAM,CAAC,EAAE,WAAW,EACpB,IAAI,CAAC,EAAE,MAAM,GAAG,cAAc,CAAC,UAAU,CAAC,GACzC,WAAW,CASb;AAED,wBAAgB,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAI1E;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,GAAG,MAAM,CAEvE;AAED,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,OAAO,EACd,MAAM,EAAE,eAAe,GACtB,IAAI,CAuBN"}
@@ -0,0 +1,113 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+ import { execute } from '@databricks/sdk-core/api';
3
+ import { ApiError } from '@databricks/sdk-core/apierror';
4
+ import JSONBig from 'json-bigint';
5
+ // JSON codec that preserves int64 precision. On the way in, large integer
6
+ // literals come back as bigint instead of being rounded to JS Number. On the
7
+ // way out, bigint values are emitted as raw JSON number digits.
8
+ const jsonBigint = JSONBig({ useNativeBigInt: true });
9
+ /**
10
+ * Translates public CallOptions to the internal Options shape accepted by
11
+ * execute(). Even though the shapes match today, this isolates the public
12
+ * API from the executor's internal type so they can diverge.
13
+ */
14
+ export async function executeCall(call, options) {
15
+ const opts = {
16
+ ...(options?.retrier !== undefined && { retrier: options.retrier }),
17
+ ...(options?.rateLimiter !== undefined && {
18
+ rateLimiter: options.rateLimiter,
19
+ }),
20
+ ...(options?.timeout !== undefined && { timeout: options.timeout }),
21
+ };
22
+ return execute(options?.signal, call, opts);
23
+ }
24
+ async function readAll(body) {
25
+ if (body === null) {
26
+ return new Uint8Array(0);
27
+ }
28
+ const reader = body.getReader();
29
+ const chunks = [];
30
+ for (;;) {
31
+ const { done, value } = await reader.read();
32
+ if (done) {
33
+ break;
34
+ }
35
+ chunks.push(value);
36
+ }
37
+ const totalLength = chunks.reduce((acc, chunk) => acc + chunk.length, 0);
38
+ const result = new Uint8Array(totalLength);
39
+ let offset = 0;
40
+ for (const chunk of chunks) {
41
+ result.set(chunk, offset);
42
+ offset += chunk.length;
43
+ }
44
+ return result;
45
+ }
46
+ export async function executeHttpCall(opts) {
47
+ opts.logger.debug('HTTP request', {
48
+ method: opts.request.method,
49
+ url: opts.request.url,
50
+ });
51
+ let resp;
52
+ try {
53
+ resp = await opts.httpClient.send(opts.request);
54
+ }
55
+ catch (e) {
56
+ opts.logger.debug('HTTP request failed');
57
+ throw e;
58
+ }
59
+ const body = await readAll(resp.body);
60
+ opts.logger.debug('HTTP response', {
61
+ statusCode: resp.statusCode,
62
+ body: new TextDecoder().decode(body),
63
+ });
64
+ const apiErr = ApiError.fromHttpError(resp.statusCode, resp.headers, body);
65
+ if (apiErr !== undefined) {
66
+ throw apiErr;
67
+ }
68
+ return body;
69
+ }
70
+ export function buildHttpRequest(method, url, headers, signal, body) {
71
+ const req = { url, method, headers };
72
+ if (body !== undefined) {
73
+ req.body = body;
74
+ }
75
+ if (signal !== undefined) {
76
+ req.signal = signal;
77
+ }
78
+ return req;
79
+ }
80
+ export function parseResponse(body, schema) {
81
+ const text = new TextDecoder().decode(body);
82
+ const parsed = jsonBigint.parse(text);
83
+ return schema.parse(parsed);
84
+ }
85
+ export function marshalRequest(data, schema) {
86
+ return jsonBigint.stringify(schema.parse(data));
87
+ }
88
+ export function flattenQueryParams(prefix, value, params) {
89
+ if (value === null || value === undefined) {
90
+ return;
91
+ }
92
+ if (Array.isArray(value)) {
93
+ // arrays of objects are not yet supported
94
+ for (const item of value) {
95
+ params.append(prefix, String(item));
96
+ }
97
+ }
98
+ else if (typeof value === 'object') {
99
+ for (const [key, val] of Object.entries(value)) {
100
+ flattenQueryParams(`${prefix}.${key}`, val, params);
101
+ }
102
+ }
103
+ else if (typeof value === 'string' ||
104
+ typeof value === 'number' ||
105
+ typeof value === 'boolean' ||
106
+ typeof value === 'bigint') {
107
+ params.append(prefix, String(value));
108
+ }
109
+ else {
110
+ throw new Error(`Unsupported query parameter type: ${typeof value}`);
111
+ }
112
+ }
113
+ //# sourceMappingURL=utils.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,OAAO,EAAC,MAAM,0BAA0B,CAAC;AACjD,OAAO,EAAC,QAAQ,EAAC,MAAM,+BAA+B,CAAC;AAQvD,OAAO,OAAO,MAAM,aAAa,CAAC;AAGlC,0EAA0E;AAC1E,6EAA6E;AAC7E,gEAAgE;AAChE,MAAM,UAAU,GAAG,OAAO,CAAC,EAAC,eAAe,EAAE,IAAI,EAAC,CAAC,CAAC;AAQpD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAAU,EACV,OAAqB;IAErB,MAAM,IAAI,GAAY;QACpB,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;QACjE,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI;YACxC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC;QACF,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;KAClE,CAAC;IACF,OAAO,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC9C,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,IAAuC;IAEvC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAC3B,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,SAAS,CAAC;QACR,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,IAAI,EAAE,CAAC;YACT,MAAM;QACR,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAC3C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAqB;IAErB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;QAChC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;QAC3B,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;KACtB,CAAC,CAAC;IAEH,IAAI,IAAkB,CAAC;IACvB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC;IACV,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE;QACjC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,IAAI,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;KACrC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC3E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,MAAM,MAAM,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,GAAW,EACX,OAAgB,EAChB,MAAoB,EACpB,IAA0C;IAE1C,MAAM,GAAG,GAAgB,EAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAC,CAAC;IAChD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;IAClB,CAAC;IACD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAI,IAAgB,EAAE,MAAoB;IACrE,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAY,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAa,EAAE,MAAiB;IAC7D,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAc,EACd,KAAc,EACd,MAAuB;IAEvB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO;IACT,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,0CAA0C;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;YAC1E,kBAAkB,CAAC,GAAG,MAAM,IAAI,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;SAAM,IACL,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS;QAC1B,OAAO,KAAK,KAAK,QAAQ,EACzB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,qCAAqC,OAAO,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC;AACH,CAAC"}
package/package.json CHANGED
@@ -1,7 +1,41 @@
1
1
  {
2
2
  "name": "@databricks/sdk-uc-entitytagassignments",
3
- "version": "0.0.0-dev",
4
- "description": "Bootstrap placeholder; real contents in a later release.",
5
- "main": "index.js",
6
- "license": "Apache-2.0"
3
+ "version": "0.1.0-dev.1",
4
+ "description": "",
5
+ "type": "module",
6
+ "exports": {
7
+ "./v1": {
8
+ "types": "./dist/v1/index.d.ts",
9
+ "import": "./dist/v1/index.js"
10
+ }
11
+ },
12
+ "files": [
13
+ "dist",
14
+ "src",
15
+ "LICENSE"
16
+ ],
17
+ "scripts": {
18
+ "build": "tsc -b",
19
+ "lint": "eslint src --ext .ts",
20
+ "lint:fix": "eslint src --ext .ts --fix",
21
+ "format": "prettier --write \"src/**/*.ts\"",
22
+ "format:check": "prettier --check \"src/**/*.ts\"",
23
+ "typecheck": "tsc --noEmit",
24
+ "clean": "rm -rf dist tsconfig.tsbuildinfo",
25
+ "test": "echo 'no tests'",
26
+ "test:browser": "echo 'no tests'"
27
+ },
28
+ "author": "Databricks",
29
+ "license": "Apache-2.0",
30
+ "dependencies": {
31
+ "@databricks/sdk-auth": ">=0.1.0-dev.2 <1.0.0",
32
+ "@databricks/sdk-core": ">=0.1.0-dev.3 <1.0.0",
33
+ "@databricks/sdk-options": ">=0.1.0-dev.2 <1.0.0",
34
+ "@js-temporal/polyfill": "^0.5.0",
35
+ "json-bigint": "^1.0.0",
36
+ "zod": "^4.3.6"
37
+ },
38
+ "engines": {
39
+ "node": ">=22.0.0"
40
+ }
7
41
  }
@@ -0,0 +1,283 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+
3
+ import {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';
4
+ import type {Call} from '@databricks/sdk-core/api';
5
+ import {createDefault} from '@databricks/sdk-core/clientinfo';
6
+ import type {Logger} from '@databricks/sdk-core/logger';
7
+ import {NoOpLogger} from '@databricks/sdk-core/logger';
8
+ import type {CallOptions} from '@databricks/sdk-options/call';
9
+ import type {ClientOptions} from '@databricks/sdk-options/client';
10
+ import type {HttpClient} from '@databricks/sdk-core/http';
11
+ import {newHttpClient} from './transport';
12
+ import {
13
+ buildHttpRequest,
14
+ executeCall,
15
+ executeHttpCall,
16
+ marshalRequest,
17
+ parseResponse,
18
+ } from './utils';
19
+ import pkgJson from '../../package.json' with {type: 'json'};
20
+ import type {
21
+ CreateEntityTagAssignmentRequest,
22
+ DeleteEntityTagAssignmentRequest,
23
+ EntityTagAssignment,
24
+ GetEntityTagAssignmentRequest,
25
+ ListEntityTagAssignmentsRequest,
26
+ ListEntityTagAssignmentsResponse,
27
+ UpdateEntityTagAssignmentRequest,
28
+ } from './model';
29
+ import {
30
+ marshalEntityTagAssignmentSchema,
31
+ unmarshalEntityTagAssignmentSchema,
32
+ unmarshalListEntityTagAssignmentsResponseSchema,
33
+ } from './model';
34
+
35
+ // Package identity segment for this client to be used in the User-Agent header.
36
+ const PACKAGE_SEGMENT = {
37
+ key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
38
+ value: pkgJson.version,
39
+ };
40
+
41
+ export class EntitytagassignmentsClient {
42
+ private readonly host: string;
43
+ // Workspace ID used to route workspace-level calls on unified hosts (SPOG).
44
+ // When set, workspace-level methods send X-Databricks-Org-Id on every
45
+ // request.
46
+ private readonly workspaceId: string | undefined;
47
+ private readonly httpClient: HttpClient;
48
+ private readonly logger: Logger;
49
+ // User-Agent header value. Composed once at construction from
50
+ // createDefault() merged with this package's identity and the active
51
+ // credential's name.
52
+ private readonly userAgent: string;
53
+
54
+ constructor(options: ClientOptions) {
55
+ if (options.host === undefined) {
56
+ throw new Error('Host is required.');
57
+ }
58
+ this.host = options.host.replace(/\/$/, '');
59
+ this.workspaceId = options.workspaceId;
60
+ this.logger = options.logger ?? new NoOpLogger();
61
+ const info = createDefault()
62
+ .with(PACKAGE_SEGMENT)
63
+ .with({key: 'sdk-js-auth', value: AUTH_VERSION})
64
+ .with({key: 'auth', value: options.credentials?.name() ?? 'default'});
65
+ this.userAgent = info.toString();
66
+ this.httpClient = newHttpClient(options);
67
+ }
68
+
69
+ /**
70
+ * Creates a tag assignment for an Unity Catalog entity.
71
+ *
72
+ * To add tags to Unity Catalog entities, you must own the entity or have the following privileges:
73
+ * - **APPLY TAG** on the entity
74
+ * - **USE SCHEMA** on the entity's parent schema
75
+ * - **USE CATALOG** on the entity's parent catalog
76
+ *
77
+ * To add a governed tag to Unity Catalog entities, you must also have the **ASSIGN** or **MANAGE** permission on the tag policy. See [Manage tag policy permissions](https://docs.databricks.com/aws/en/admin/tag-policies/manage-permissions).
78
+ */
79
+ async createEntityTagAssignment(
80
+ req: CreateEntityTagAssignmentRequest,
81
+ options?: CallOptions
82
+ ): Promise<EntityTagAssignment> {
83
+ const url = `${this.host}/api/2.1/unity-catalog/entity-tag-assignments`;
84
+ const body = marshalRequest(
85
+ req.tagAssignment,
86
+ marshalEntityTagAssignmentSchema
87
+ );
88
+ let resp: EntityTagAssignment | undefined;
89
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
90
+ const headers = new Headers({'Content-Type': 'application/json'});
91
+ if (this.workspaceId !== undefined) {
92
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
93
+ }
94
+ headers.set('User-Agent', this.userAgent);
95
+ const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
96
+ const respBody = await executeHttpCall({
97
+ request: httpReq,
98
+ httpClient: this.httpClient,
99
+ logger: this.logger,
100
+ });
101
+ resp = parseResponse(respBody, unmarshalEntityTagAssignmentSchema);
102
+ };
103
+ await executeCall(call, options);
104
+ if (resp === undefined) {
105
+ throw new Error('API call completed without a result.');
106
+ }
107
+ return resp;
108
+ }
109
+
110
+ /**
111
+ * Deletes a tag assignment for an Unity Catalog entity by its key.
112
+ *
113
+ * To delete tags from Unity Catalog entities, you must own the entity or have the following privileges:
114
+ * - **APPLY TAG** on the entity
115
+ * - **USE_SCHEMA** on the entity's parent schema
116
+ * - **USE_CATALOG** on the entity's parent catalog
117
+ *
118
+ * To delete a governed tag from Unity Catalog entities, you must also have the **ASSIGN** or **MANAGE** permission on the tag policy. See [Manage tag policy permissions](https://docs.databricks.com/aws/en/admin/tag-policies/manage-permissions).
119
+ */
120
+ async deleteEntityTagAssignment(
121
+ req: DeleteEntityTagAssignmentRequest,
122
+ options?: CallOptions
123
+ ): Promise<void> {
124
+ const url = `${this.host}/api/2.1/unity-catalog/entity-tag-assignments/${req.entityType ?? ''}/${req.entityName ?? ''}/tags/${req.tagKey ?? ''}`;
125
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
126
+ const headers = new Headers();
127
+ if (this.workspaceId !== undefined) {
128
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
129
+ }
130
+ headers.set('User-Agent', this.userAgent);
131
+ const httpReq = buildHttpRequest('DELETE', url, headers, callSignal);
132
+ await executeHttpCall({
133
+ request: httpReq,
134
+ httpClient: this.httpClient,
135
+ logger: this.logger,
136
+ });
137
+ };
138
+ await executeCall(call, options);
139
+ }
140
+
141
+ /** Gets a tag assignment for an Unity Catalog entity by tag key. */
142
+ async getEntityTagAssignment(
143
+ req: GetEntityTagAssignmentRequest,
144
+ options?: CallOptions
145
+ ): Promise<EntityTagAssignment> {
146
+ const url = `${this.host}/api/2.1/unity-catalog/entity-tag-assignments/${req.entityType ?? ''}/${req.entityName ?? ''}/tags/${req.tagKey ?? ''}`;
147
+ let resp: EntityTagAssignment | undefined;
148
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
149
+ const headers = new Headers();
150
+ if (this.workspaceId !== undefined) {
151
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
152
+ }
153
+ headers.set('User-Agent', this.userAgent);
154
+ const httpReq = buildHttpRequest('GET', url, headers, callSignal);
155
+ const respBody = await executeHttpCall({
156
+ request: httpReq,
157
+ httpClient: this.httpClient,
158
+ logger: this.logger,
159
+ });
160
+ resp = parseResponse(respBody, unmarshalEntityTagAssignmentSchema);
161
+ };
162
+ await executeCall(call, options);
163
+ if (resp === undefined) {
164
+ throw new Error('API call completed without a result.');
165
+ }
166
+ return resp;
167
+ }
168
+
169
+ /**
170
+ * List tag assignments for an Unity Catalog entity
171
+ *
172
+ * PAGINATION BEHAVIOR: The API is by default paginated, a page may contain zero results while still providing a next_page_token.
173
+ * Clients must continue reading pages until next_page_token is absent, which is the only indication that the end of results has been reached.
174
+ */
175
+ async listEntityTagAssignments(
176
+ req: ListEntityTagAssignmentsRequest,
177
+ options?: CallOptions
178
+ ): Promise<ListEntityTagAssignmentsResponse> {
179
+ const url = `${this.host}/api/2.1/unity-catalog/entity-tag-assignments/${req.entityType ?? ''}/${req.entityName ?? ''}/tags`;
180
+ const params = new URLSearchParams();
181
+ if (req.maxResults !== undefined) {
182
+ params.append('max_results', String(req.maxResults));
183
+ }
184
+ if (req.pageToken !== undefined) {
185
+ params.append('page_token', req.pageToken);
186
+ }
187
+ const query = params.toString();
188
+ const fullUrl = query !== '' ? `${url}?${query}` : url;
189
+ let resp: ListEntityTagAssignmentsResponse | undefined;
190
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
191
+ const headers = new Headers();
192
+ if (this.workspaceId !== undefined) {
193
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
194
+ }
195
+ headers.set('User-Agent', this.userAgent);
196
+ const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);
197
+ const respBody = await executeHttpCall({
198
+ request: httpReq,
199
+ httpClient: this.httpClient,
200
+ logger: this.logger,
201
+ });
202
+ resp = parseResponse(
203
+ respBody,
204
+ unmarshalListEntityTagAssignmentsResponseSchema
205
+ );
206
+ };
207
+ await executeCall(call, options);
208
+ if (resp === undefined) {
209
+ throw new Error('API call completed without a result.');
210
+ }
211
+ return resp;
212
+ }
213
+
214
+ async *listEntityTagAssignmentsIter(
215
+ req: ListEntityTagAssignmentsRequest,
216
+ options?: CallOptions
217
+ ): AsyncGenerator<EntityTagAssignment> {
218
+ const pageReq: ListEntityTagAssignmentsRequest = {...req};
219
+ for (;;) {
220
+ const resp = await this.listEntityTagAssignments(pageReq, options);
221
+ for (const item of resp.tagAssignments ?? []) {
222
+ yield item;
223
+ }
224
+ if (resp.nextPageToken === undefined || resp.nextPageToken === '') {
225
+ return;
226
+ }
227
+ pageReq.pageToken = resp.nextPageToken;
228
+ }
229
+ }
230
+
231
+ /**
232
+ * Updates an existing tag assignment for an Unity Catalog entity.
233
+ *
234
+ * To update tags to Unity Catalog entities, you must own the entity or have the following privileges:
235
+ * - **APPLY TAG** on the entity
236
+ * - **USE SCHEMA** on the entity's parent schema
237
+ * - **USE CATALOG** on the entity's parent catalog
238
+ *
239
+ * To update a governed tag to Unity Catalog entities, you must also have the **ASSIGN** or **MANAGE** permission on the tag policy. See [Manage tag policy permissions](https://docs.databricks.com/aws/en/admin/tag-policies/manage-permissions).
240
+ */
241
+ async updateEntityTagAssignment(
242
+ req: UpdateEntityTagAssignmentRequest,
243
+ options?: CallOptions
244
+ ): Promise<EntityTagAssignment> {
245
+ const url = `${this.host}/api/2.1/unity-catalog/entity-tag-assignments/${req.tagAssignment?.entityType ?? ''}/${req.tagAssignment?.entityName ?? ''}/tags/${req.tagAssignment?.tagKey ?? ''}`;
246
+ const params = new URLSearchParams();
247
+ if (req.updateMask !== undefined) {
248
+ params.append('update_mask', req.updateMask.toString());
249
+ }
250
+ const query = params.toString();
251
+ const fullUrl = query !== '' ? `${url}?${query}` : url;
252
+ const body = marshalRequest(
253
+ req.tagAssignment,
254
+ marshalEntityTagAssignmentSchema
255
+ );
256
+ let resp: EntityTagAssignment | undefined;
257
+ const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
258
+ const headers = new Headers({'Content-Type': 'application/json'});
259
+ if (this.workspaceId !== undefined) {
260
+ headers.set('X-Databricks-Org-Id', this.workspaceId);
261
+ }
262
+ headers.set('User-Agent', this.userAgent);
263
+ const httpReq = buildHttpRequest(
264
+ 'PATCH',
265
+ fullUrl,
266
+ headers,
267
+ callSignal,
268
+ body
269
+ );
270
+ const respBody = await executeHttpCall({
271
+ request: httpReq,
272
+ httpClient: this.httpClient,
273
+ logger: this.logger,
274
+ });
275
+ resp = parseResponse(respBody, unmarshalEntityTagAssignmentSchema);
276
+ };
277
+ await executeCall(call, options);
278
+ if (resp === undefined) {
279
+ throw new Error('API call completed without a result.');
280
+ }
281
+ return resp;
282
+ }
283
+ }
@@ -0,0 +1,17 @@
1
+ // Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
2
+
3
+ export {EntitytagassignmentsClient} from './client';
4
+
5
+ export {TagAssignmentSourceType} from './model';
6
+
7
+ export type {
8
+ CreateEntityTagAssignmentRequest,
9
+ DeleteEntityTagAssignmentRequest,
10
+ EntityTagAssignment,
11
+ GetEntityTagAssignmentRequest,
12
+ ListEntityTagAssignmentsRequest,
13
+ ListEntityTagAssignmentsResponse,
14
+ UpdateEntityTagAssignmentRequest,
15
+ } from './model';
16
+
17
+ export {entityTagAssignmentFieldMask} from './model';