@databricks/sdk-modelservingquery 0.1.0-dev.3 → 0.1.0-dev.5
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.
- package/README.md +28 -0
- package/dist/v1/client.d.ts +3 -3
- package/dist/v1/client.d.ts.map +1 -1
- package/dist/v1/client.js +16 -17
- package/dist/v1/client.js.map +1 -1
- package/dist/v1/index.d.ts +1 -1
- package/dist/v1/index.d.ts.map +1 -1
- package/dist/v1/model.d.ts +19 -24
- package/dist/v1/model.d.ts.map +1 -1
- package/dist/v1/model.js +23 -23
- package/dist/v1/model.js.map +1 -1
- package/dist/v1/transport.d.ts +30 -2
- package/dist/v1/transport.d.ts.map +1 -1
- package/dist/v1/transport.js +33 -16
- package/dist/v1/transport.js.map +1 -1
- package/dist/v1/utils.d.ts.map +1 -1
- package/dist/v1/utils.js +2 -1
- package/dist/v1/utils.js.map +1 -1
- package/package.json +9 -5
- package/src/v1/client.ts +0 -87
- package/src/v1/index.ts +0 -21
- package/src/v1/model.ts +0 -345
- package/src/v1/transport.ts +0 -73
- package/src/v1/utils.ts +0 -156
package/README.md
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
# @databricks/sdk-modelservingquery
|
|
2
|
+
|
|
3
|
+
> [!WARNING]
|
|
4
|
+
>
|
|
5
|
+
> ## PREVIEW - NOT FOR PRODUCTION USE
|
|
6
|
+
>
|
|
7
|
+
> **This SDK is in preview and is subject to change without notice.**
|
|
8
|
+
>
|
|
9
|
+
> - **Do NOT use in production environments**
|
|
10
|
+
> - **Breaking changes may occur at any time**
|
|
11
|
+
> - **APIs are experimental and unstable**
|
|
12
|
+
> - **Use for development and testing only**
|
|
13
|
+
|
|
14
|
+
## Installation
|
|
15
|
+
|
|
16
|
+
```bash
|
|
17
|
+
npm install @databricks/sdk-modelservingquery
|
|
18
|
+
```
|
|
19
|
+
|
|
20
|
+
## Usage
|
|
21
|
+
|
|
22
|
+
```typescript
|
|
23
|
+
import {ModelServingQueryClient} from '@databricks/sdk-modelservingquery/v1';
|
|
24
|
+
|
|
25
|
+
const client = new ModelServingQueryClient({});
|
|
26
|
+
```
|
|
27
|
+
|
|
28
|
+
For a full getting-started guide, see the [root README](../../README.md).
|
package/dist/v1/client.d.ts
CHANGED
|
@@ -2,12 +2,12 @@ import type { CallOptions } from '@databricks/sdk-options/call';
|
|
|
2
2
|
import type { ClientOptions } from '@databricks/sdk-options/client';
|
|
3
3
|
import type { QueryEndpointInputRequest, QueryEndpointResponse } from './model';
|
|
4
4
|
export declare class ModelServingQueryClient {
|
|
5
|
-
private readonly
|
|
6
|
-
private readonly workspaceId;
|
|
7
|
-
private readonly httpClient;
|
|
5
|
+
private readonly options;
|
|
8
6
|
private readonly logger;
|
|
9
7
|
private readonly userAgent;
|
|
8
|
+
private config;
|
|
10
9
|
constructor(options: ClientOptions);
|
|
10
|
+
private resolveConfig;
|
|
11
11
|
/** Query a serving endpoint */
|
|
12
12
|
query(req: QueryEndpointInputRequest, options?: CallOptions): Promise<QueryEndpointResponse>;
|
|
13
13
|
}
|
package/dist/v1/client.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/v1/client.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,8BAA8B,CAAC;AAC9D,OAAO,KAAK,EAAC,aAAa,EAAC,MAAM,gCAAgC,CAAC;AAWlE,OAAO,KAAK,EAAC,yBAAyB,EAAE,qBAAqB,EAAC,MAAM,SAAS,CAAC;AAY9E,qBAAa,uBAAuB;IAClC,OAAO,CAAC,QAAQ,CAAC,
|
|
1
|
+
{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/v1/client.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,8BAA8B,CAAC;AAC9D,OAAO,KAAK,EAAC,aAAa,EAAC,MAAM,gCAAgC,CAAC;AAWlE,OAAO,KAAK,EAAC,yBAAyB,EAAE,qBAAqB,EAAC,MAAM,SAAS,CAAC;AAY9E,qBAAa,uBAAuB;IAClC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAgB;IACxC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAIhC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IAInC,OAAO,CAAC,MAAM,CAA4C;gBAE9C,OAAO,EAAE,aAAa;IAUlC,OAAO,CAAC,aAAa;IAKrB,+BAA+B;IACzB,KAAK,CACT,GAAG,EAAE,yBAAyB,EAC9B,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAAC,qBAAqB,CAAC;CAyBlC"}
|
package/dist/v1/client.js
CHANGED
|
@@ -2,7 +2,7 @@
|
|
|
2
2
|
import { VERSION as AUTH_VERSION } from '@databricks/sdk-auth';
|
|
3
3
|
import { createDefault } from '@databricks/sdk-core/clientinfo';
|
|
4
4
|
import { NoOpLogger } from '@databricks/sdk-core/logger';
|
|
5
|
-
import {
|
|
5
|
+
import { resolveClientConfig } from './transport';
|
|
6
6
|
import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse, } from './utils';
|
|
7
7
|
import pkgJson from '../../package.json' with { type: 'json' };
|
|
8
8
|
import { marshalQueryEndpointInputRequestSchema, unmarshalQueryEndpointResponseSchema, } from './model';
|
|
@@ -12,46 +12,45 @@ const PACKAGE_SEGMENT = {
|
|
|
12
12
|
value: pkgJson.version,
|
|
13
13
|
};
|
|
14
14
|
export class ModelServingQueryClient {
|
|
15
|
-
|
|
16
|
-
// Workspace ID used to route workspace-level calls on unified hosts (SPOG).
|
|
17
|
-
// When set, workspace-level methods send X-Databricks-Org-Id on every
|
|
18
|
-
// request.
|
|
19
|
-
workspaceId;
|
|
20
|
-
httpClient;
|
|
15
|
+
options;
|
|
21
16
|
logger;
|
|
22
17
|
// User-Agent header value. Composed once at construction from
|
|
23
18
|
// createDefault() merged with this package's identity and the active
|
|
24
19
|
// credential's name.
|
|
25
20
|
userAgent;
|
|
21
|
+
// Memoized configuration. The profile is resolved once, lazily, on the first
|
|
22
|
+
// request, then reused; host, workspaceId/accountId, and credentials are
|
|
23
|
+
// filled from it when not set explicitly on the options.
|
|
24
|
+
config;
|
|
26
25
|
constructor(options) {
|
|
27
|
-
|
|
28
|
-
throw new Error('Host is required.');
|
|
29
|
-
}
|
|
30
|
-
this.host = options.host.replace(/\/$/, '');
|
|
31
|
-
this.workspaceId = options.workspaceId;
|
|
26
|
+
this.options = options;
|
|
32
27
|
this.logger = options.logger ?? new NoOpLogger();
|
|
33
28
|
const info = createDefault()
|
|
34
29
|
.with(PACKAGE_SEGMENT)
|
|
35
30
|
.with({ key: 'sdk-js-auth', value: AUTH_VERSION })
|
|
36
31
|
.with({ key: 'auth', value: options.credentials?.name() ?? 'default' });
|
|
37
32
|
this.userAgent = info.toString();
|
|
38
|
-
|
|
33
|
+
}
|
|
34
|
+
resolveConfig() {
|
|
35
|
+
this.config ??= resolveClientConfig(this.options);
|
|
36
|
+
return this.config;
|
|
39
37
|
}
|
|
40
38
|
/** Query a serving endpoint */
|
|
41
39
|
async query(req, options) {
|
|
42
|
-
const
|
|
40
|
+
const { host, workspaceId, httpClient } = await this.resolveConfig();
|
|
41
|
+
const url = `${host}/api/serving-endpoints/${req.name ?? ''}/invocations`;
|
|
43
42
|
const body = marshalRequest(req, marshalQueryEndpointInputRequestSchema);
|
|
44
43
|
let resp;
|
|
45
44
|
const call = async (callSignal) => {
|
|
46
45
|
const headers = new Headers({ 'Content-Type': 'application/json' });
|
|
47
|
-
if (
|
|
48
|
-
headers.set('X-Databricks-Org-Id',
|
|
46
|
+
if (workspaceId !== undefined) {
|
|
47
|
+
headers.set('X-Databricks-Org-Id', workspaceId);
|
|
49
48
|
}
|
|
50
49
|
headers.set('User-Agent', this.userAgent);
|
|
51
50
|
const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
|
|
52
51
|
const respBody = await executeHttpCall({
|
|
53
52
|
request: httpReq,
|
|
54
|
-
httpClient
|
|
53
|
+
httpClient,
|
|
55
54
|
logger: this.logger,
|
|
56
55
|
});
|
|
57
56
|
resp = parseResponse(respBody, unmarshalQueryEndpointResponseSchema);
|
package/dist/v1/client.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/v1/client.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAE/E,OAAO,EAAC,OAAO,IAAI,YAAY,EAAC,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EAAC,aAAa,EAAC,MAAM,iCAAiC,CAAC;AAE9D,OAAO,EAAC,UAAU,EAAC,MAAM,6BAA6B,CAAC;AAIvD,OAAO,EAAC,
|
|
1
|
+
{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/v1/client.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAE/E,OAAO,EAAC,OAAO,IAAI,YAAY,EAAC,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EAAC,aAAa,EAAC,MAAM,iCAAiC,CAAC;AAE9D,OAAO,EAAC,UAAU,EAAC,MAAM,6BAA6B,CAAC;AAIvD,OAAO,EAAC,mBAAmB,EAAC,MAAM,aAAa,CAAC;AAChD,OAAO,EACL,gBAAgB,EAChB,WAAW,EACX,eAAe,EACf,cAAc,EACd,aAAa,GACd,MAAM,SAAS,CAAC;AACjB,OAAO,OAAO,MAAM,oBAAoB,CAAC,OAAM,IAAI,EAAE,MAAM,EAAC,CAAC;AAE7D,OAAO,EACL,sCAAsC,EACtC,oCAAoC,GACrC,MAAM,SAAS,CAAC;AAEjB,gFAAgF;AAChF,MAAM,eAAe,GAAG;IACtB,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC;IAC1D,KAAK,EAAE,OAAO,CAAC,OAAO;CACvB,CAAC;AAEF,MAAM,OAAO,uBAAuB;IACjB,OAAO,CAAgB;IACvB,MAAM,CAAS;IAChC,8DAA8D;IAC9D,qEAAqE;IACrE,qBAAqB;IACJ,SAAS,CAAS;IACnC,6EAA6E;IAC7E,yEAAyE;IACzE,yDAAyD;IACjD,MAAM,CAA4C;IAE1D,YAAY,OAAsB;QAChC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QACjD,MAAM,IAAI,GAAG,aAAa,EAAE;aACzB,IAAI,CAAC,eAAe,CAAC;aACrB,IAAI,CAAC,EAAC,GAAG,EAAE,aAAa,EAAE,KAAK,EAAE,YAAY,EAAC,CAAC;aAC/C,IAAI,CAAC,EAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,SAAS,EAAC,CAAC,CAAC;QACxE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;IACnC,CAAC;IAEO,aAAa;QACnB,IAAI,CAAC,MAAM,KAAK,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClD,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,+BAA+B;IAC/B,KAAK,CAAC,KAAK,CACT,GAA8B,EAC9B,OAAqB;QAErB,MAAM,EAAC,IAAI,EAAE,WAAW,EAAE,UAAU,EAAC,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QACnE,MAAM,GAAG,GAAG,GAAG,IAAI,0BAA0B,GAAG,CAAC,IAAI,IAAI,EAAE,cAAc,CAAC;QAC1E,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,sCAAsC,CAAC,CAAC;QACzE,IAAI,IAAuC,CAAC;QAC5C,MAAM,IAAI,GAAG,KAAK,EAAE,UAAwB,EAAiB,EAAE;YAC7D,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,EAAC,cAAc,EAAE,kBAAkB,EAAC,CAAC,CAAC;YAClE,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBAC9B,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,WAAW,CAAC,CAAC;YAClD,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC;YACzE,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC;gBACrC,OAAO,EAAE,OAAO;gBAChB,UAAU;gBACV,MAAM,EAAE,IAAI,CAAC,MAAM;aACpB,CAAC,CAAC;YACH,IAAI,GAAG,aAAa,CAAC,QAAQ,EAAE,oCAAoC,CAAC,CAAC;QACvE,CAAC,CAAC;QACF,MAAM,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACjC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
|
package/dist/v1/index.d.ts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
1
|
export { ModelServingQueryClient } from './client';
|
|
2
2
|
export { ChatMessageRole, EmbeddingsV1ResponseEmbeddingElementObject, QueryEndpointResponseObject, } from './model';
|
|
3
|
-
export type { ChatMessage, DataframeSplitInput, EmbeddingsV1ResponseEmbeddingElement, ExternalModelUsageElement, QueryEndpointInputRequest,
|
|
3
|
+
export type { ChatMessage, DataframeSplitInput, EmbeddingsV1ResponseEmbeddingElement, ExternalModelUsageElement, QueryEndpointInputRequest, QueryEndpointResponse, V1ResponseChoiceElement, } from './model';
|
|
4
4
|
//# sourceMappingURL=index.d.ts.map
|
package/dist/v1/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/v1/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAC,uBAAuB,EAAC,MAAM,UAAU,CAAC;AAEjD,OAAO,EACL,eAAe,EACf,0CAA0C,EAC1C,2BAA2B,GAC5B,MAAM,SAAS,CAAC;AAEjB,YAAY,EACV,WAAW,EACX,mBAAmB,EACnB,oCAAoC,EACpC,yBAAyB,EACzB,yBAAyB,EACzB,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/v1/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAC,uBAAuB,EAAC,MAAM,UAAU,CAAC;AAEjD,OAAO,EACL,eAAe,EACf,0CAA0C,EAC1C,2BAA2B,GAC5B,MAAM,SAAS,CAAC;AAEjB,YAAY,EACV,WAAW,EACX,mBAAmB,EACnB,oCAAoC,EACpC,yBAAyB,EACzB,yBAAyB,EACzB,qBAAqB,EACrB,uBAAuB,GACxB,MAAM,SAAS,CAAC"}
|
package/dist/v1/model.d.ts
CHANGED
|
@@ -1,27 +1,30 @@
|
|
|
1
1
|
import type { JsonValue } from '@databricks/sdk-core/wkt';
|
|
2
2
|
import { z } from 'zod';
|
|
3
3
|
/** The role of the message. One of [system, user, assistant]. */
|
|
4
|
-
export declare
|
|
5
|
-
CHAT_MESSAGE_ROLE_UNSPECIFIED
|
|
6
|
-
SYSTEM
|
|
7
|
-
USER
|
|
8
|
-
ASSISTANT
|
|
9
|
-
}
|
|
4
|
+
export declare const ChatMessageRole: {
|
|
5
|
+
readonly CHAT_MESSAGE_ROLE_UNSPECIFIED: "CHAT_MESSAGE_ROLE_UNSPECIFIED";
|
|
6
|
+
readonly SYSTEM: "system";
|
|
7
|
+
readonly USER: "user";
|
|
8
|
+
readonly ASSISTANT: "assistant";
|
|
9
|
+
};
|
|
10
|
+
export type ChatMessageRole = (typeof ChatMessageRole)[keyof typeof ChatMessageRole] | (string & {});
|
|
10
11
|
/** This will always be 'embedding'. */
|
|
11
|
-
export declare
|
|
12
|
-
EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED
|
|
13
|
-
EMBEDDING
|
|
14
|
-
}
|
|
12
|
+
export declare const EmbeddingsV1ResponseEmbeddingElementObject: {
|
|
13
|
+
readonly EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED: "EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED";
|
|
14
|
+
readonly EMBEDDING: "embedding";
|
|
15
|
+
};
|
|
16
|
+
export type EmbeddingsV1ResponseEmbeddingElementObject = (typeof EmbeddingsV1ResponseEmbeddingElementObject)[keyof typeof EmbeddingsV1ResponseEmbeddingElementObject] | (string & {});
|
|
15
17
|
/**
|
|
16
18
|
* The type of object returned by the __external/foundation model__ serving endpoint, one of
|
|
17
19
|
* [text_completion, chat.completion, list (of embeddings)].
|
|
18
20
|
*/
|
|
19
|
-
export declare
|
|
20
|
-
QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED
|
|
21
|
-
TEXT_COMPLETION
|
|
22
|
-
CHAT_COMPLETION
|
|
23
|
-
LIST
|
|
24
|
-
}
|
|
21
|
+
export declare const QueryEndpointResponseObject: {
|
|
22
|
+
readonly QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED: "QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED";
|
|
23
|
+
readonly TEXT_COMPLETION: "text_completion";
|
|
24
|
+
readonly CHAT_COMPLETION: "chat.completion";
|
|
25
|
+
readonly LIST: "list";
|
|
26
|
+
};
|
|
27
|
+
export type QueryEndpointResponseObject = (typeof QueryEndpointResponseObject)[keyof typeof QueryEndpointResponseObject] | (string & {});
|
|
25
28
|
export interface ChatMessage {
|
|
26
29
|
/** The role of the message. One of [system, user, assistant]. */
|
|
27
30
|
role?: ChatMessageRole | undefined;
|
|
@@ -116,14 +119,6 @@ export interface QueryEndpointInputRequest {
|
|
|
116
119
|
/** Optional user-provided context that will be recorded in the usage tracking table. */
|
|
117
120
|
usageContext?: Record<string, string> | undefined;
|
|
118
121
|
}
|
|
119
|
-
export interface QueryEndpointInputRequest_ExtraParamsEntry {
|
|
120
|
-
key?: string | undefined;
|
|
121
|
-
value?: string | undefined;
|
|
122
|
-
}
|
|
123
|
-
export interface QueryEndpointInputRequest_UsageContextEntry {
|
|
124
|
-
key?: string | undefined;
|
|
125
|
-
value?: string | undefined;
|
|
126
|
-
}
|
|
127
122
|
export interface QueryEndpointResponse {
|
|
128
123
|
/** The list of choices returned by the __chat or completions external/foundation model__ serving endpoint. */
|
|
129
124
|
choices?: V1ResponseChoiceElement[] | undefined;
|
package/dist/v1/model.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/v1/model.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAatB,iEAAiE;
|
|
1
|
+
{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/v1/model.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAC,SAAS,EAAC,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAatB,iEAAiE;AAEjE,eAAO,MAAM,eAAe;;;;;CAKlB,CAAC;AACX,MAAM,MAAM,eAAe,GACvB,CAAC,OAAO,eAAe,CAAC,CAAC,MAAM,OAAO,eAAe,CAAC,GACtD,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAElB,uCAAuC;AAEvC,eAAO,MAAM,0CAA0C;;;CAI7C,CAAC;AACX,MAAM,MAAM,0CAA0C,GAClD,CAAC,OAAO,0CAA0C,CAAC,CAAC,MAAM,OAAO,0CAA0C,CAAC,GAC5G,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAElB;;;GAGG;AAEH,eAAO,MAAM,2BAA2B;;;;;CAM9B,CAAC;AACX,MAAM,MAAM,2BAA2B,GACnC,CAAC,OAAO,2BAA2B,CAAC,CAAC,MAAM,OAAO,2BAA2B,CAAC,GAC9E,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAElB,MAAM,WAAW,WAAW;IAC1B,iEAAiE;IACjE,IAAI,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC;IACnC,kCAAkC;IAClC,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B;AAED,MAAM,WAAW,mBAAmB;IAClC,oCAAoC;IACpC,KAAK,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IAC7B,sCAAsC;IACtC,OAAO,CAAC,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;IAClC,mCAAmC;IACnC,IAAI,CAAC,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;CAChC;AAED,MAAM,WAAW,oCAAoC;IACnD,2BAA2B;IAC3B,SAAS,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IACjC,kDAAkD;IAClD,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,uCAAuC;IACvC,MAAM,CAAC,EAAE,0CAA0C,GAAG,SAAS,CAAC;CACjE;AAED,MAAM,WAAW,yBAAyB;IACxC,0CAA0C;IAC1C,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,6DAA6D;IAC7D,gBAAgB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACtC,6DAA6D;IAC7D,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAClC;AAED,MAAM,WAAW,yBAAyB;IACxC,uGAAuG;IACvG,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B;;;OAGG;IACH,MAAM,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;IAC/B;;;OAGG;IACH,KAAK,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;IAC9B;;;OAGG;IACH,QAAQ,CAAC,EAAE,WAAW,EAAE,GAAG,SAAS,CAAC;IACrC;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACjC;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IAC5B;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC/B;;;;OAIG;IACH,CAAC,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACvB;;;OAGG;IACH,MAAM,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC7B;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,SAAS,CAAC;IACjD,yDAAyD;IACzD,gBAAgB,CAAC,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;IAC3C,uDAAuD;IACvD,cAAc,CAAC,EAAE,mBAAmB,GAAG,SAAS,CAAC;IACjD,wCAAwC;IACxC,SAAS,CAAC,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;IACpC,6CAA6C;IAC7C,MAAM,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;IAC/B,2HAA2H;IAC3H,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACrC,wFAAwF;IACxF,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,SAAS,CAAC;CACnD;AAED,MAAM,WAAW,qBAAqB;IACpC,8GAA8G;IAC9G,OAAO,CAAC,EAAE,uBAAuB,EAAE,GAAG,SAAS,CAAC;IAChD,4GAA4G;IAC5G,IAAI,CAAC,EAAE,oCAAoC,EAAE,GAAG,SAAS,CAAC;IAC1D;;;OAGG;IACH,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B;;;OAGG;IACH,KAAK,CAAC,EAAE,yBAAyB,GAAG,SAAS,CAAC;IAC9C,wHAAwH;IACxH,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,yJAAyJ;IACzJ,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B;;;OAGG;IACH,MAAM,CAAC,EAAE,2BAA2B,GAAG,SAAS,CAAC;IACjD,wDAAwD;IACxD,WAAW,CAAC,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;IACtC,mDAAmD;IACnD,OAAO,CAAC,EAAE,SAAS,EAAE,GAAG,SAAS,CAAC;IAClC,uJAAuJ;IACvJ,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACtC;AAED,MAAM,WAAW,uBAAuB;IACtC,2DAA2D;IAC3D,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,uDAAuD;IACvD,OAAO,CAAC,EAAE,WAAW,GAAG,SAAS,CAAC;IAClC,uEAAuE;IACvE,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC3B,kDAAkD;IAClD,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAClC,kEAAkE;IAClE,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CAC/B;AAED,eAAO,MAAM,0BAA0B,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAQzD,CAAC;AAEN,eAAO,MAAM,mDAAmD,EAAE,CAAC,CAAC,OAAO,CAAC,oCAAoC,CAWzG,CAAC;AAER,eAAO,MAAM,wCAAwC,EAAE,CAAC,CAAC,OAAO,CAAC,yBAAyB,CAWnF,CAAC;AAER,eAAO,MAAM,oCAAoC,EAAE,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAkC3E,CAAC;AAER,eAAO,MAAM,sCAAsC,EAAE,CAAC,CAAC,OAAO,CAAC,uBAAuB,CAe/E,CAAC;AAER,eAAO,MAAM,wBAAwB,EAAE,CAAC,CAAC,OAQpC,CAAC;AAEN,eAAO,MAAM,gCAAgC,EAAE,CAAC,CAAC,OAU5C,CAAC;AAEN,eAAO,MAAM,sCAAsC,EAAE,CAAC,CAAC,OAoClD,CAAC"}
|
package/dist/v1/model.js
CHANGED
|
@@ -9,33 +9,33 @@ const jsonValueSchema = z.lazy(() => z.union([
|
|
|
9
9
|
z.array(jsonValueSchema),
|
|
10
10
|
]));
|
|
11
11
|
/** The role of the message. One of [system, user, assistant]. */
|
|
12
|
-
|
|
13
|
-
|
|
14
|
-
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
}
|
|
12
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
|
|
13
|
+
export const ChatMessageRole = {
|
|
14
|
+
CHAT_MESSAGE_ROLE_UNSPECIFIED: 'CHAT_MESSAGE_ROLE_UNSPECIFIED',
|
|
15
|
+
SYSTEM: 'system',
|
|
16
|
+
USER: 'user',
|
|
17
|
+
ASSISTANT: 'assistant',
|
|
18
|
+
};
|
|
19
19
|
/** This will always be 'embedding'. */
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
}
|
|
20
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
|
|
21
|
+
export const EmbeddingsV1ResponseEmbeddingElementObject = {
|
|
22
|
+
EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED: 'EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED',
|
|
23
|
+
EMBEDDING: 'embedding',
|
|
24
|
+
};
|
|
25
25
|
/**
|
|
26
26
|
* The type of object returned by the __external/foundation model__ serving endpoint, one of
|
|
27
27
|
* [text_completion, chat.completion, list (of embeddings)].
|
|
28
28
|
*/
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
}
|
|
29
|
+
// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
|
|
30
|
+
export const QueryEndpointResponseObject = {
|
|
31
|
+
QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED: 'QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED',
|
|
32
|
+
TEXT_COMPLETION: 'text_completion',
|
|
33
|
+
CHAT_COMPLETION: 'chat.completion',
|
|
34
|
+
LIST: 'list',
|
|
35
|
+
};
|
|
36
36
|
export const unmarshalChatMessageSchema = z
|
|
37
37
|
.object({
|
|
38
|
-
role: z.
|
|
38
|
+
role: z.string().optional(),
|
|
39
39
|
content: z.string().optional(),
|
|
40
40
|
})
|
|
41
41
|
.transform(d => ({
|
|
@@ -46,7 +46,7 @@ export const unmarshalEmbeddingsV1ResponseEmbeddingElementSchema = z
|
|
|
46
46
|
.object({
|
|
47
47
|
embedding: z.array(z.number()).optional(),
|
|
48
48
|
index: z.number().optional(),
|
|
49
|
-
object: z.
|
|
49
|
+
object: z.string().optional(),
|
|
50
50
|
})
|
|
51
51
|
.transform(d => ({
|
|
52
52
|
embedding: d.embedding,
|
|
@@ -79,7 +79,7 @@ export const unmarshalQueryEndpointResponseSchema = z
|
|
|
79
79
|
.union([z.number(), z.bigint()])
|
|
80
80
|
.transform(v => BigInt(v))
|
|
81
81
|
.optional(),
|
|
82
|
-
object: z.
|
|
82
|
+
object: z.string().optional(),
|
|
83
83
|
predictions: z.array(jsonValueSchema).optional(),
|
|
84
84
|
outputs: z.array(jsonValueSchema).optional(),
|
|
85
85
|
'served-model-name': z.string().optional(),
|
|
@@ -113,7 +113,7 @@ export const unmarshalV1ResponseChoiceElementSchema = z
|
|
|
113
113
|
}));
|
|
114
114
|
export const marshalChatMessageSchema = z
|
|
115
115
|
.object({
|
|
116
|
-
role: z.
|
|
116
|
+
role: z.string().optional(),
|
|
117
117
|
content: z.string().optional(),
|
|
118
118
|
})
|
|
119
119
|
.transform(d => ({
|
package/dist/v1/model.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/v1/model.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,MAAM,eAAe,GAAyB,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CACxD,CAAC,CAAC,KAAK,CAAC;IACN,CAAC,CAAC,IAAI,EAAE;IACR,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,OAAO,EAAE;IACX,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,eAAe,CAAC;IACrC,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC;CACzB,CAAC,CACH,CAAC;AAEF,iEAAiE;AACjE,MAAM,
|
|
1
|
+
{"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/v1/model.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAEtB,MAAM,eAAe,GAAyB,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CACxD,CAAC,CAAC,KAAK,CAAC;IACN,CAAC,CAAC,IAAI,EAAE;IACR,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,OAAO,EAAE;IACX,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,eAAe,CAAC;IACrC,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC;CACzB,CAAC,CACH,CAAC;AAEF,iEAAiE;AACjE,4FAA4F;AAC5F,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,6BAA6B,EAAE,+BAA+B;IAC9D,MAAM,EAAE,QAAQ;IAChB,IAAI,EAAE,MAAM;IACZ,SAAS,EAAE,WAAW;CACd,CAAC;AAKX,uCAAuC;AACvC,4FAA4F;AAC5F,MAAM,CAAC,MAAM,0CAA0C,GAAG;IACxD,2DAA2D,EACzD,6DAA6D;IAC/D,SAAS,EAAE,WAAW;CACd,CAAC;AAKX;;;GAGG;AACH,4FAA4F;AAC5F,MAAM,CAAC,MAAM,2BAA2B,GAAG;IACzC,0CAA0C,EACxC,4CAA4C;IAC9C,eAAe,EAAE,iBAAiB;IAClC,eAAe,EAAE,iBAAiB;IAClC,IAAI,EAAE,MAAM;CACJ,CAAC;AAqJX,MAAM,CAAC,MAAM,0BAA0B,GAA2B,CAAC;KAChE,MAAM,CAAC;IACN,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC/B,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,IAAI,EAAE,CAAC,CAAC,IAAI;IACZ,OAAO,EAAE,CAAC,CAAC,OAAO;CACnB,CAAC,CAAC,CAAC;AAEN,MAAM,CAAC,MAAM,mDAAmD,GAC9D,CAAC;KACE,MAAM,CAAC;IACN,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACzC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC5B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC9B,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,SAAS,EAAE,CAAC,CAAC,SAAS;IACtB,KAAK,EAAE,CAAC,CAAC,KAAK;IACd,MAAM,EAAE,CAAC,CAAC,MAAM;CACjB,CAAC,CAAC,CAAC;AAER,MAAM,CAAC,MAAM,wCAAwC,GACnD,CAAC;KACE,MAAM,CAAC;IACN,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACpC,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACxC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACpC,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,YAAY,EAAE,CAAC,CAAC,aAAa;IAC7B,gBAAgB,EAAE,CAAC,CAAC,iBAAiB;IACrC,WAAW,EAAE,CAAC,CAAC,YAAY;CAC5B,CAAC,CAAC,CAAC;AAER,MAAM,CAAC,MAAM,oCAAoC,GAC/C,CAAC;KACE,MAAM,CAAC;IACN,OAAO,EAAE,CAAC;SACP,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,sCAAsC,CAAC,CAAC;SAC3D,QAAQ,EAAE;IACb,IAAI,EAAE,CAAC;SACJ,KAAK,CACJ,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,mDAAmD,CAAC,CAClE;SACA,QAAQ,EAAE;IACb,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC5B,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,wCAAwC,CAAC,CAAC,QAAQ,EAAE;IACxE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACzB,OAAO,EAAE,CAAC;SACP,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;SAC/B,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;SACzB,QAAQ,EAAE;IACb,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC7B,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE;IAChD,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE;IAC5C,mBAAmB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC3C,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,OAAO,EAAE,CAAC,CAAC,OAAO;IAClB,IAAI,EAAE,CAAC,CAAC,IAAI;IACZ,KAAK,EAAE,CAAC,CAAC,KAAK;IACd,KAAK,EAAE,CAAC,CAAC,KAAK;IACd,EAAE,EAAE,CAAC,CAAC,EAAE;IACR,OAAO,EAAE,CAAC,CAAC,OAAO;IAClB,MAAM,EAAE,CAAC,CAAC,MAAM;IAChB,WAAW,EAAE,CAAC,CAAC,WAAW;IAC1B,OAAO,EAAE,CAAC,CAAC,OAAO;IAClB,eAAe,EAAE,CAAC,CAAC,mBAAmB,CAAC;CACxC,CAAC,CAAC,CAAC;AAER,MAAM,CAAC,MAAM,sCAAsC,GACjD,CAAC;KACE,MAAM,CAAC;IACN,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,0BAA0B,CAAC,CAAC,QAAQ,EAAE;IAC5D,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC5B,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACnC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAChC,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,IAAI,EAAE,CAAC,CAAC,IAAI;IACZ,OAAO,EAAE,CAAC,CAAC,OAAO;IAClB,KAAK,EAAE,CAAC,CAAC,KAAK;IACd,YAAY,EAAE,CAAC,CAAC,YAAY;IAC5B,QAAQ,EAAE,CAAC,CAAC,QAAQ;CACrB,CAAC,CAAC,CAAC;AAER,MAAM,CAAC,MAAM,wBAAwB,GAAc,CAAC;KACjD,MAAM,CAAC;IACN,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC/B,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,IAAI,EAAE,CAAC,CAAC,IAAI;IACZ,OAAO,EAAE,CAAC,CAAC,OAAO;CACnB,CAAC,CAAC,CAAC;AAEN,MAAM,CAAC,MAAM,gCAAgC,GAAc,CAAC;KACzD,MAAM,CAAC;IACN,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACrC,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE;IAC5C,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE;CAC1C,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,KAAK,EAAE,CAAC,CAAC,KAAK;IACd,OAAO,EAAE,CAAC,CAAC,OAAO;IAClB,IAAI,EAAE,CAAC,CAAC,IAAI;CACb,CAAC,CAAC,CAAC;AAEN,MAAM,CAAC,MAAM,sCAAsC,GAAc,CAAC;KAC/D,MAAM,CAAC;IACN,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC3B,MAAM,EAAE,eAAe,CAAC,QAAQ,EAAE;IAClC,KAAK,EAAE,eAAe,CAAC,QAAQ,EAAE;IACjC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC,QAAQ,EAAE;IACpE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACpC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAChC,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACxB,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IAC9B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACxD,gBAAgB,EAAE,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE;IACrD,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,gCAAgC,CAAC,CAAC,QAAQ,EAAE;IACzE,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,QAAQ,EAAE;IAC9C,MAAM,EAAE,eAAe,CAAC,QAAQ,EAAE;IAClC,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACtC,YAAY,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CAC1D,CAAC;KACD,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACf,IAAI,EAAE,CAAC,CAAC,IAAI;IACZ,MAAM,EAAE,CAAC,CAAC,MAAM;IAChB,KAAK,EAAE,CAAC,CAAC,KAAK;IACd,QAAQ,EAAE,CAAC,CAAC,QAAQ;IACpB,WAAW,EAAE,CAAC,CAAC,WAAW;IAC1B,IAAI,EAAE,CAAC,CAAC,IAAI;IACZ,UAAU,EAAE,CAAC,CAAC,SAAS;IACvB,CAAC,EAAE,CAAC,CAAC,CAAC;IACN,MAAM,EAAE,CAAC,CAAC,MAAM;IAChB,YAAY,EAAE,CAAC,CAAC,WAAW;IAC3B,iBAAiB,EAAE,CAAC,CAAC,gBAAgB;IACrC,eAAe,EAAE,CAAC,CAAC,cAAc;IACjC,SAAS,EAAE,CAAC,CAAC,SAAS;IACtB,MAAM,EAAE,CAAC,CAAC,MAAM;IAChB,iBAAiB,EAAE,CAAC,CAAC,eAAe;IACpC,aAAa,EAAE,CAAC,CAAC,YAAY;CAC9B,CAAC,CAAC,CAAC"}
|
package/dist/v1/transport.d.ts
CHANGED
|
@@ -1,5 +1,33 @@
|
|
|
1
1
|
import type { HttpClient } from '@databricks/sdk-core/http';
|
|
2
2
|
import type { ClientOptions } from '@databricks/sdk-options/client';
|
|
3
|
-
/**
|
|
4
|
-
|
|
3
|
+
/**
|
|
4
|
+
* The configuration a client needs to issue requests, resolved from
|
|
5
|
+
* {@link ClientOptions} and a configuration profile.
|
|
6
|
+
*/
|
|
7
|
+
export interface ResolvedClientConfig {
|
|
8
|
+
/** Host with any trailing slash removed. */
|
|
9
|
+
host: string;
|
|
10
|
+
/**
|
|
11
|
+
* Default account ID for account-level paths that contain an account_id
|
|
12
|
+
* segment. A request's own accountId still wins.
|
|
13
|
+
*/
|
|
14
|
+
accountId?: string;
|
|
15
|
+
/**
|
|
16
|
+
* Workspace ID used to route workspace-level calls on unified hosts (SPOG).
|
|
17
|
+
*/
|
|
18
|
+
workspaceId?: string;
|
|
19
|
+
/** HTTP client with authentication, and any configured timeout, applied. */
|
|
20
|
+
httpClient: HttpClient;
|
|
21
|
+
}
|
|
22
|
+
/**
|
|
23
|
+
* Resolves {@link ClientOptions} into a {@link ResolvedClientConfig}.
|
|
24
|
+
*
|
|
25
|
+
* A configuration profile is always resolved from the config file and
|
|
26
|
+
* environment variables (per options.profileOptions); it supplies host,
|
|
27
|
+
* accountId, workspaceId, and credentials wherever the caller did not set them
|
|
28
|
+
* explicitly. Explicit options always take precedence.
|
|
29
|
+
*
|
|
30
|
+
* @throws if host is neither provided nor present in the resolved profile.
|
|
31
|
+
*/
|
|
32
|
+
export declare function resolveClientConfig(options: ClientOptions): Promise<ResolvedClientConfig>;
|
|
5
33
|
//# sourceMappingURL=transport.d.ts.map
|
|
@@ -1 +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;
|
|
1
|
+
{"version":3,"file":"transport.d.ts","sourceRoot":"","sources":["../../src/v1/transport.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,UAAU,EAGX,MAAM,2BAA2B,CAAC;AAInC,OAAO,KAAK,EAAC,aAAa,EAAC,MAAM,gCAAgC,CAAC;AAElE;;;GAGG;AACH,MAAM,WAAW,oBAAoB;IACnC,4CAA4C;IAC5C,IAAI,EAAE,MAAM,CAAC;IAEb;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IAEnB;;OAEG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,4EAA4E;IAC5E,UAAU,EAAE,UAAU,CAAC;CACxB;AAED;;;;;;;;;GASG;AACH,wBAAsB,mBAAmB,CACvC,OAAO,EAAE,aAAa,GACrB,OAAO,CAAC,oBAAoB,CAAC,CA4B/B"}
|
package/dist/v1/transport.js
CHANGED
|
@@ -1,24 +1,41 @@
|
|
|
1
1
|
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
2
|
import { defaultCredentials } from '@databricks/sdk-auth/credentials';
|
|
3
3
|
import { newFetchHttpClient } from '@databricks/sdk-core/http';
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
|
|
12
|
-
|
|
13
|
-
|
|
4
|
+
import { resolve } from '@databricks/sdk-core/profiles';
|
|
5
|
+
/**
|
|
6
|
+
* Resolves {@link ClientOptions} into a {@link ResolvedClientConfig}.
|
|
7
|
+
*
|
|
8
|
+
* A configuration profile is always resolved from the config file and
|
|
9
|
+
* environment variables (per options.profileOptions); it supplies host,
|
|
10
|
+
* accountId, workspaceId, and credentials wherever the caller did not set them
|
|
11
|
+
* explicitly. Explicit options always take precedence.
|
|
12
|
+
*
|
|
13
|
+
* @throws if host is neither provided nor present in the resolved profile.
|
|
14
|
+
*/
|
|
15
|
+
export async function resolveClientConfig(options) {
|
|
16
|
+
const profile = await resolve(options.profileOptions);
|
|
17
|
+
const host = options.host ?? profile.host;
|
|
18
|
+
if (host === undefined) {
|
|
19
|
+
throw new Error('Host is required.');
|
|
14
20
|
}
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
|
|
19
|
-
|
|
21
|
+
// The provided httpClient (or the default fetch client) is the base wire;
|
|
22
|
+
// authentication and the optional timeout are layered on top of it. The
|
|
23
|
+
// default credential chain reuses the profile resolved above so that the
|
|
24
|
+
// same profileOptions govern host and authentication alike.
|
|
25
|
+
const base = options.httpClient ?? newFetchHttpClient();
|
|
26
|
+
const credentials = options.credentials ?? defaultCredentials({ profile });
|
|
27
|
+
let httpClient = new AuthHttpClient(base, credentials);
|
|
28
|
+
if (options.timeout !== undefined) {
|
|
29
|
+
httpClient = new TimeoutHttpClient(httpClient, options.timeout);
|
|
20
30
|
}
|
|
21
|
-
|
|
31
|
+
const accountId = options.accountId ?? profile.accountId;
|
|
32
|
+
const workspaceId = options.workspaceId ?? profile.workspaceId;
|
|
33
|
+
return {
|
|
34
|
+
host: host.replace(/\/$/, ''),
|
|
35
|
+
httpClient,
|
|
36
|
+
...(accountId !== undefined && { accountId }),
|
|
37
|
+
...(workspaceId !== undefined && { workspaceId }),
|
|
38
|
+
};
|
|
22
39
|
}
|
|
23
40
|
/** Wraps an HttpClient and adds authentication headers to requests. */
|
|
24
41
|
class AuthHttpClient {
|
package/dist/v1/transport.js.map
CHANGED
|
@@ -1 +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;
|
|
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;AAE7D,OAAO,EAAC,OAAO,EAAC,MAAM,+BAA+B,CAAC;AA0BtD;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,OAAsB;IAEtB,MAAM,OAAO,GAAY,MAAM,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IAE/D,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAC1C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;IACvC,CAAC;IAED,0EAA0E;IAC1E,wEAAwE;IACxE,yEAAyE;IACzE,4DAA4D;IAC5D,MAAM,IAAI,GAAG,OAAO,CAAC,UAAU,IAAI,kBAAkB,EAAE,CAAC;IACxD,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,kBAAkB,CAAC,EAAC,OAAO,EAAC,CAAC,CAAC;IACzE,IAAI,UAAU,GAAe,IAAI,cAAc,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IACnE,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAClC,UAAU,GAAG,IAAI,iBAAiB,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC;IACzD,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,CAAC;IAE/D,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;QAC7B,UAAU;QACV,GAAG,CAAC,SAAS,KAAK,SAAS,IAAI,EAAC,SAAS,EAAC,CAAC;QAC3C,GAAG,CAAC,WAAW,KAAK,SAAS,IAAI,EAAC,WAAW,EAAC,CAAC;KAChD,CAAC;AACJ,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"}
|
package/dist/v1/utils.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAKA,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,CAAC,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,EAC7C,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,
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAKA,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,CAAC,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,EAC7C,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,CAK1E;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"}
|
package/dist/v1/utils.js
CHANGED
|
@@ -79,7 +79,8 @@ export function buildHttpRequest(method, url, headers, signal, body) {
|
|
|
79
79
|
}
|
|
80
80
|
export function parseResponse(body, schema) {
|
|
81
81
|
const text = new TextDecoder().decode(body);
|
|
82
|
-
|
|
82
|
+
// 204 responses return an empty body; treat as `{}`.
|
|
83
|
+
const parsed = text === '' ? {} : jsonBigint.parse(text);
|
|
83
84
|
return schema.parse(parsed);
|
|
84
85
|
}
|
|
85
86
|
export function marshalRequest(data, schema) {
|
package/dist/v1/utils.js.map
CHANGED
|
@@ -1 +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,IAA6C,EAC7C,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;
|
|
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,IAA6C,EAC7C,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,qDAAqD;IACrD,MAAM,MAAM,GAAY,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClE,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,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@databricks/sdk-modelservingquery",
|
|
3
|
-
"version": "0.1.0-dev.
|
|
3
|
+
"version": "0.1.0-dev.5",
|
|
4
4
|
"description": "",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"exports": {
|
|
@@ -12,7 +12,6 @@
|
|
|
12
12
|
},
|
|
13
13
|
"files": [
|
|
14
14
|
"dist",
|
|
15
|
-
"src",
|
|
16
15
|
"LICENSE"
|
|
17
16
|
],
|
|
18
17
|
"scripts": {
|
|
@@ -28,10 +27,15 @@
|
|
|
28
27
|
},
|
|
29
28
|
"author": "Databricks",
|
|
30
29
|
"license": "Apache-2.0",
|
|
30
|
+
"repository": {
|
|
31
|
+
"type": "git",
|
|
32
|
+
"url": "https://github.com/databricks/sdk-js.git",
|
|
33
|
+
"directory": "packages/modelservingquery"
|
|
34
|
+
},
|
|
31
35
|
"dependencies": {
|
|
32
|
-
"@databricks/sdk-auth": ">=0.1.0-dev.
|
|
33
|
-
"@databricks/sdk-core": ">=0.1.0-dev.
|
|
34
|
-
"@databricks/sdk-options": ">=0.1.0-dev.
|
|
36
|
+
"@databricks/sdk-auth": ">=0.1.0-dev.6 <1.0.0",
|
|
37
|
+
"@databricks/sdk-core": ">=0.1.0-dev.7 <1.0.0",
|
|
38
|
+
"@databricks/sdk-options": ">=0.1.0-dev.6 <1.0.0",
|
|
35
39
|
"@js-temporal/polyfill": "^0.5.0",
|
|
36
40
|
"json-bigint": "^1.0.0",
|
|
37
41
|
"zod": "^4.3.6"
|
package/src/v1/client.ts
DELETED
|
@@ -1,87 +0,0 @@
|
|
|
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 {createDefault} from '@databricks/sdk-core/clientinfo';
|
|
5
|
-
import type {Logger} from '@databricks/sdk-core/logger';
|
|
6
|
-
import {NoOpLogger} from '@databricks/sdk-core/logger';
|
|
7
|
-
import type {CallOptions} from '@databricks/sdk-options/call';
|
|
8
|
-
import type {ClientOptions} from '@databricks/sdk-options/client';
|
|
9
|
-
import type {HttpClient} from '@databricks/sdk-core/http';
|
|
10
|
-
import {newHttpClient} from './transport';
|
|
11
|
-
import {
|
|
12
|
-
buildHttpRequest,
|
|
13
|
-
executeCall,
|
|
14
|
-
executeHttpCall,
|
|
15
|
-
marshalRequest,
|
|
16
|
-
parseResponse,
|
|
17
|
-
} from './utils';
|
|
18
|
-
import pkgJson from '../../package.json' with {type: 'json'};
|
|
19
|
-
import type {QueryEndpointInputRequest, QueryEndpointResponse} from './model';
|
|
20
|
-
import {
|
|
21
|
-
marshalQueryEndpointInputRequestSchema,
|
|
22
|
-
unmarshalQueryEndpointResponseSchema,
|
|
23
|
-
} from './model';
|
|
24
|
-
|
|
25
|
-
// Package identity segment for this client to be used in the User-Agent header.
|
|
26
|
-
const PACKAGE_SEGMENT = {
|
|
27
|
-
key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
|
|
28
|
-
value: pkgJson.version,
|
|
29
|
-
};
|
|
30
|
-
|
|
31
|
-
export class ModelServingQueryClient {
|
|
32
|
-
private readonly host: string;
|
|
33
|
-
// Workspace ID used to route workspace-level calls on unified hosts (SPOG).
|
|
34
|
-
// When set, workspace-level methods send X-Databricks-Org-Id on every
|
|
35
|
-
// request.
|
|
36
|
-
private readonly workspaceId: string | undefined;
|
|
37
|
-
private readonly httpClient: HttpClient;
|
|
38
|
-
private readonly logger: Logger;
|
|
39
|
-
// User-Agent header value. Composed once at construction from
|
|
40
|
-
// createDefault() merged with this package's identity and the active
|
|
41
|
-
// credential's name.
|
|
42
|
-
private readonly userAgent: string;
|
|
43
|
-
|
|
44
|
-
constructor(options: ClientOptions) {
|
|
45
|
-
if (options.host === undefined) {
|
|
46
|
-
throw new Error('Host is required.');
|
|
47
|
-
}
|
|
48
|
-
this.host = options.host.replace(/\/$/, '');
|
|
49
|
-
this.workspaceId = options.workspaceId;
|
|
50
|
-
this.logger = options.logger ?? new NoOpLogger();
|
|
51
|
-
const info = createDefault()
|
|
52
|
-
.with(PACKAGE_SEGMENT)
|
|
53
|
-
.with({key: 'sdk-js-auth', value: AUTH_VERSION})
|
|
54
|
-
.with({key: 'auth', value: options.credentials?.name() ?? 'default'});
|
|
55
|
-
this.userAgent = info.toString();
|
|
56
|
-
this.httpClient = newHttpClient(options);
|
|
57
|
-
}
|
|
58
|
-
|
|
59
|
-
/** Query a serving endpoint */
|
|
60
|
-
async query(
|
|
61
|
-
req: QueryEndpointInputRequest,
|
|
62
|
-
options?: CallOptions
|
|
63
|
-
): Promise<QueryEndpointResponse> {
|
|
64
|
-
const url = `${this.host}/api/serving-endpoints/${req.name ?? ''}/invocations`;
|
|
65
|
-
const body = marshalRequest(req, marshalQueryEndpointInputRequestSchema);
|
|
66
|
-
let resp: QueryEndpointResponse | undefined;
|
|
67
|
-
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
68
|
-
const headers = new Headers({'Content-Type': 'application/json'});
|
|
69
|
-
if (this.workspaceId !== undefined) {
|
|
70
|
-
headers.set('X-Databricks-Org-Id', this.workspaceId);
|
|
71
|
-
}
|
|
72
|
-
headers.set('User-Agent', this.userAgent);
|
|
73
|
-
const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
|
|
74
|
-
const respBody = await executeHttpCall({
|
|
75
|
-
request: httpReq,
|
|
76
|
-
httpClient: this.httpClient,
|
|
77
|
-
logger: this.logger,
|
|
78
|
-
});
|
|
79
|
-
resp = parseResponse(respBody, unmarshalQueryEndpointResponseSchema);
|
|
80
|
-
};
|
|
81
|
-
await executeCall(call, options);
|
|
82
|
-
if (resp === undefined) {
|
|
83
|
-
throw new Error('operation completed without a result.');
|
|
84
|
-
}
|
|
85
|
-
return resp;
|
|
86
|
-
}
|
|
87
|
-
}
|
package/src/v1/index.ts
DELETED
|
@@ -1,21 +0,0 @@
|
|
|
1
|
-
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
-
|
|
3
|
-
export {ModelServingQueryClient} from './client';
|
|
4
|
-
|
|
5
|
-
export {
|
|
6
|
-
ChatMessageRole,
|
|
7
|
-
EmbeddingsV1ResponseEmbeddingElementObject,
|
|
8
|
-
QueryEndpointResponseObject,
|
|
9
|
-
} from './model';
|
|
10
|
-
|
|
11
|
-
export type {
|
|
12
|
-
ChatMessage,
|
|
13
|
-
DataframeSplitInput,
|
|
14
|
-
EmbeddingsV1ResponseEmbeddingElement,
|
|
15
|
-
ExternalModelUsageElement,
|
|
16
|
-
QueryEndpointInputRequest,
|
|
17
|
-
QueryEndpointInputRequest_ExtraParamsEntry,
|
|
18
|
-
QueryEndpointInputRequest_UsageContextEntry,
|
|
19
|
-
QueryEndpointResponse,
|
|
20
|
-
V1ResponseChoiceElement,
|
|
21
|
-
} from './model';
|
package/src/v1/model.ts
DELETED
|
@@ -1,345 +0,0 @@
|
|
|
1
|
-
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
-
|
|
3
|
-
import type {JsonValue} from '@databricks/sdk-core/wkt';
|
|
4
|
-
import {z} from 'zod';
|
|
5
|
-
|
|
6
|
-
const jsonValueSchema: z.ZodType<JsonValue> = z.lazy(() =>
|
|
7
|
-
z.union([
|
|
8
|
-
z.null(),
|
|
9
|
-
z.number(),
|
|
10
|
-
z.string(),
|
|
11
|
-
z.boolean(),
|
|
12
|
-
z.record(z.string(), jsonValueSchema),
|
|
13
|
-
z.array(jsonValueSchema),
|
|
14
|
-
])
|
|
15
|
-
);
|
|
16
|
-
|
|
17
|
-
/** The role of the message. One of [system, user, assistant]. */
|
|
18
|
-
export enum ChatMessageRole {
|
|
19
|
-
CHAT_MESSAGE_ROLE_UNSPECIFIED = 'CHAT_MESSAGE_ROLE_UNSPECIFIED',
|
|
20
|
-
SYSTEM = 'system',
|
|
21
|
-
USER = 'user',
|
|
22
|
-
ASSISTANT = 'assistant',
|
|
23
|
-
}
|
|
24
|
-
|
|
25
|
-
/** This will always be 'embedding'. */
|
|
26
|
-
export enum EmbeddingsV1ResponseEmbeddingElementObject {
|
|
27
|
-
EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED = 'EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED',
|
|
28
|
-
EMBEDDING = 'embedding',
|
|
29
|
-
}
|
|
30
|
-
|
|
31
|
-
/**
|
|
32
|
-
* The type of object returned by the __external/foundation model__ serving endpoint, one of
|
|
33
|
-
* [text_completion, chat.completion, list (of embeddings)].
|
|
34
|
-
*/
|
|
35
|
-
export enum QueryEndpointResponseObject {
|
|
36
|
-
QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED = 'QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED',
|
|
37
|
-
TEXT_COMPLETION = 'text_completion',
|
|
38
|
-
CHAT_COMPLETION = 'chat.completion',
|
|
39
|
-
LIST = 'list',
|
|
40
|
-
}
|
|
41
|
-
|
|
42
|
-
export interface ChatMessage {
|
|
43
|
-
/** The role of the message. One of [system, user, assistant]. */
|
|
44
|
-
role?: ChatMessageRole | undefined;
|
|
45
|
-
/** The content of the message. */
|
|
46
|
-
content?: string | undefined;
|
|
47
|
-
}
|
|
48
|
-
|
|
49
|
-
export interface DataframeSplitInput {
|
|
50
|
-
/** Index array for the dataframe */
|
|
51
|
-
index?: number[] | undefined;
|
|
52
|
-
/** Columns array for the dataframe */
|
|
53
|
-
columns?: JsonValue[] | undefined;
|
|
54
|
-
/** Data array for the dataframe */
|
|
55
|
-
data?: JsonValue[] | undefined;
|
|
56
|
-
}
|
|
57
|
-
|
|
58
|
-
export interface EmbeddingsV1ResponseEmbeddingElement {
|
|
59
|
-
/** The embedding vector */
|
|
60
|
-
embedding?: number[] | undefined;
|
|
61
|
-
/** The index of the embedding in the response. */
|
|
62
|
-
index?: number | undefined;
|
|
63
|
-
/** This will always be 'embedding'. */
|
|
64
|
-
object?: EmbeddingsV1ResponseEmbeddingElementObject | undefined;
|
|
65
|
-
}
|
|
66
|
-
|
|
67
|
-
export interface ExternalModelUsageElement {
|
|
68
|
-
/** The number of tokens in the prompt. */
|
|
69
|
-
promptTokens?: number | undefined;
|
|
70
|
-
/** The number of tokens in the chat/completions response. */
|
|
71
|
-
completionTokens?: number | undefined;
|
|
72
|
-
/** The total number of tokens in the prompt and response. */
|
|
73
|
-
totalTokens?: number | undefined;
|
|
74
|
-
}
|
|
75
|
-
|
|
76
|
-
export interface QueryEndpointInputRequest {
|
|
77
|
-
/** The name of the serving endpoint. This field is required and is provided via the path parameter. */
|
|
78
|
-
name?: string | undefined;
|
|
79
|
-
/**
|
|
80
|
-
* The prompt string (or array of strings) field used ONLY for __completions external & foundation model__
|
|
81
|
-
* serving endpoints and should only be used with other completions query fields.
|
|
82
|
-
*/
|
|
83
|
-
prompt?: JsonValue | undefined;
|
|
84
|
-
/**
|
|
85
|
-
* The input string (or array of strings) field used ONLY for __embeddings external & foundation model__
|
|
86
|
-
* serving endpoints and is the only field (along with extra_params if needed) used by embeddings queries.
|
|
87
|
-
*/
|
|
88
|
-
input?: JsonValue | undefined;
|
|
89
|
-
/**
|
|
90
|
-
* The messages field used ONLY for __chat external & foundation model__ serving endpoints.
|
|
91
|
-
* This is an array of ChatMessage objects and should only be used with other chat query fields.
|
|
92
|
-
*/
|
|
93
|
-
messages?: ChatMessage[] | undefined;
|
|
94
|
-
/**
|
|
95
|
-
* The temperature field used ONLY for __completions__ and __chat external & foundation model__ serving
|
|
96
|
-
* endpoints. This is a float between 0.0 and 2.0 with a default of 1.0 and should only be used with other
|
|
97
|
-
* chat/completions query fields.
|
|
98
|
-
*/
|
|
99
|
-
temperature?: number | undefined;
|
|
100
|
-
/**
|
|
101
|
-
* The stop sequences field used ONLY for __completions__ and __chat external & foundation model__ serving
|
|
102
|
-
* endpoints. This is a list of strings and should only be used with other chat/completions query fields.
|
|
103
|
-
*/
|
|
104
|
-
stop?: string[] | undefined;
|
|
105
|
-
/**
|
|
106
|
-
* The max tokens field used ONLY for __completions__ and __chat external & foundation model__ serving
|
|
107
|
-
* endpoints. This is an integer and should only be used with other chat/completions query fields.
|
|
108
|
-
*/
|
|
109
|
-
maxTokens?: number | undefined;
|
|
110
|
-
/**
|
|
111
|
-
* The n (number of candidates) field used ONLY for __completions__ and __chat external & foundation model__
|
|
112
|
-
* serving endpoints. This is an integer between 1 and 5 with a default of 1 and should only be used with
|
|
113
|
-
* other chat/completions query fields.
|
|
114
|
-
*/
|
|
115
|
-
n?: number | undefined;
|
|
116
|
-
/**
|
|
117
|
-
* The stream field used ONLY for __completions__ and __chat external & foundation model__ serving endpoints.
|
|
118
|
-
* This is a boolean defaulting to false and should only be used with other chat/completions query fields.
|
|
119
|
-
*/
|
|
120
|
-
stream?: boolean | undefined;
|
|
121
|
-
/**
|
|
122
|
-
* The extra parameters field used ONLY for __completions, chat,__ and __embeddings external & foundation
|
|
123
|
-
* model__ serving endpoints. This is a map of strings and should only be used with other external/foundation
|
|
124
|
-
* model query fields.
|
|
125
|
-
*/
|
|
126
|
-
extraParams?: Record<string, string> | undefined;
|
|
127
|
-
/** Pandas Dataframe input in the records orientation. */
|
|
128
|
-
dataframeRecords?: JsonValue[] | undefined;
|
|
129
|
-
/** Pandas Dataframe input in the split orientation. */
|
|
130
|
-
dataframeSplit?: DataframeSplitInput | undefined;
|
|
131
|
-
/** Tensor-based input in row format. */
|
|
132
|
-
instances?: JsonValue[] | undefined;
|
|
133
|
-
/** Tensor-based input in columnar format. */
|
|
134
|
-
inputs?: JsonValue | undefined;
|
|
135
|
-
/** Optional user-provided request identifier that will be recorded in the inference table and the usage tracking table. */
|
|
136
|
-
clientRequestId?: string | undefined;
|
|
137
|
-
/** Optional user-provided context that will be recorded in the usage tracking table. */
|
|
138
|
-
usageContext?: Record<string, string> | undefined;
|
|
139
|
-
}
|
|
140
|
-
|
|
141
|
-
// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
|
|
142
|
-
export interface QueryEndpointInputRequest_ExtraParamsEntry {
|
|
143
|
-
key?: string | undefined;
|
|
144
|
-
value?: string | undefined;
|
|
145
|
-
}
|
|
146
|
-
|
|
147
|
-
// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
|
|
148
|
-
export interface QueryEndpointInputRequest_UsageContextEntry {
|
|
149
|
-
key?: string | undefined;
|
|
150
|
-
value?: string | undefined;
|
|
151
|
-
}
|
|
152
|
-
|
|
153
|
-
export interface QueryEndpointResponse {
|
|
154
|
-
/** The list of choices returned by the __chat or completions external/foundation model__ serving endpoint. */
|
|
155
|
-
choices?: V1ResponseChoiceElement[] | undefined;
|
|
156
|
-
/** The list of the embeddings returned by the __embeddings external/foundation model__ serving endpoint. */
|
|
157
|
-
data?: EmbeddingsV1ResponseEmbeddingElement[] | undefined;
|
|
158
|
-
/**
|
|
159
|
-
* The name of the __external/foundation model__ used for querying. This is the name of the model that was
|
|
160
|
-
* specified in the endpoint config.
|
|
161
|
-
*/
|
|
162
|
-
model?: string | undefined;
|
|
163
|
-
/**
|
|
164
|
-
* The usage object that may be returned by the __external/foundation model__ serving endpoint. This
|
|
165
|
-
* contains information about the number of tokens used in the prompt and response.
|
|
166
|
-
*/
|
|
167
|
-
usage?: ExternalModelUsageElement | undefined;
|
|
168
|
-
/** The ID of the query that may be returned by a __completions or chat external/foundation model__ serving endpoint. */
|
|
169
|
-
id?: string | undefined;
|
|
170
|
-
/** The timestamp in seconds when the query was created in Unix time returned by a __completions or chat external/foundation model__ serving endpoint. */
|
|
171
|
-
created?: bigint | undefined;
|
|
172
|
-
/**
|
|
173
|
-
* The type of object returned by the __external/foundation model__ serving endpoint, one of
|
|
174
|
-
* [text_completion, chat.completion, list (of embeddings)].
|
|
175
|
-
*/
|
|
176
|
-
object?: QueryEndpointResponseObject | undefined;
|
|
177
|
-
/** The predictions returned by the serving endpoint. */
|
|
178
|
-
predictions?: JsonValue[] | undefined;
|
|
179
|
-
/** The outputs of the feature serving endpoint. */
|
|
180
|
-
outputs?: JsonValue[] | undefined;
|
|
181
|
-
/** The name of the served model that served the request. This is useful when there are multiple models behind the same endpoint with traffic split. */
|
|
182
|
-
servedModelName?: string | undefined;
|
|
183
|
-
}
|
|
184
|
-
|
|
185
|
-
export interface V1ResponseChoiceElement {
|
|
186
|
-
/** The text response from the __completions__ endpoint. */
|
|
187
|
-
text?: string | undefined;
|
|
188
|
-
/** The message response from the __chat__ endpoint. */
|
|
189
|
-
message?: ChatMessage | undefined;
|
|
190
|
-
/** The index of the choice in the __chat or completions__ response. */
|
|
191
|
-
index?: number | undefined;
|
|
192
|
-
/** The finish reason returned by the endpoint. */
|
|
193
|
-
finishReason?: string | undefined;
|
|
194
|
-
/** The logprobs returned only by the __completions__ endpoint. */
|
|
195
|
-
logprobs?: number | undefined;
|
|
196
|
-
}
|
|
197
|
-
|
|
198
|
-
export const unmarshalChatMessageSchema: z.ZodType<ChatMessage> = z
|
|
199
|
-
.object({
|
|
200
|
-
role: z.enum(ChatMessageRole).optional(),
|
|
201
|
-
content: z.string().optional(),
|
|
202
|
-
})
|
|
203
|
-
.transform(d => ({
|
|
204
|
-
role: d.role,
|
|
205
|
-
content: d.content,
|
|
206
|
-
}));
|
|
207
|
-
|
|
208
|
-
export const unmarshalEmbeddingsV1ResponseEmbeddingElementSchema: z.ZodType<EmbeddingsV1ResponseEmbeddingElement> =
|
|
209
|
-
z
|
|
210
|
-
.object({
|
|
211
|
-
embedding: z.array(z.number()).optional(),
|
|
212
|
-
index: z.number().optional(),
|
|
213
|
-
object: z.enum(EmbeddingsV1ResponseEmbeddingElementObject).optional(),
|
|
214
|
-
})
|
|
215
|
-
.transform(d => ({
|
|
216
|
-
embedding: d.embedding,
|
|
217
|
-
index: d.index,
|
|
218
|
-
object: d.object,
|
|
219
|
-
}));
|
|
220
|
-
|
|
221
|
-
export const unmarshalExternalModelUsageElementSchema: z.ZodType<ExternalModelUsageElement> =
|
|
222
|
-
z
|
|
223
|
-
.object({
|
|
224
|
-
prompt_tokens: z.number().optional(),
|
|
225
|
-
completion_tokens: z.number().optional(),
|
|
226
|
-
total_tokens: z.number().optional(),
|
|
227
|
-
})
|
|
228
|
-
.transform(d => ({
|
|
229
|
-
promptTokens: d.prompt_tokens,
|
|
230
|
-
completionTokens: d.completion_tokens,
|
|
231
|
-
totalTokens: d.total_tokens,
|
|
232
|
-
}));
|
|
233
|
-
|
|
234
|
-
export const unmarshalQueryEndpointResponseSchema: z.ZodType<QueryEndpointResponse> =
|
|
235
|
-
z
|
|
236
|
-
.object({
|
|
237
|
-
choices: z
|
|
238
|
-
.array(z.lazy(() => unmarshalV1ResponseChoiceElementSchema))
|
|
239
|
-
.optional(),
|
|
240
|
-
data: z
|
|
241
|
-
.array(
|
|
242
|
-
z.lazy(() => unmarshalEmbeddingsV1ResponseEmbeddingElementSchema)
|
|
243
|
-
)
|
|
244
|
-
.optional(),
|
|
245
|
-
model: z.string().optional(),
|
|
246
|
-
usage: z.lazy(() => unmarshalExternalModelUsageElementSchema).optional(),
|
|
247
|
-
id: z.string().optional(),
|
|
248
|
-
created: z
|
|
249
|
-
.union([z.number(), z.bigint()])
|
|
250
|
-
.transform(v => BigInt(v))
|
|
251
|
-
.optional(),
|
|
252
|
-
object: z.enum(QueryEndpointResponseObject).optional(),
|
|
253
|
-
predictions: z.array(jsonValueSchema).optional(),
|
|
254
|
-
outputs: z.array(jsonValueSchema).optional(),
|
|
255
|
-
'served-model-name': z.string().optional(),
|
|
256
|
-
})
|
|
257
|
-
.transform(d => ({
|
|
258
|
-
choices: d.choices,
|
|
259
|
-
data: d.data,
|
|
260
|
-
model: d.model,
|
|
261
|
-
usage: d.usage,
|
|
262
|
-
id: d.id,
|
|
263
|
-
created: d.created,
|
|
264
|
-
object: d.object,
|
|
265
|
-
predictions: d.predictions,
|
|
266
|
-
outputs: d.outputs,
|
|
267
|
-
servedModelName: d['served-model-name'],
|
|
268
|
-
}));
|
|
269
|
-
|
|
270
|
-
export const unmarshalV1ResponseChoiceElementSchema: z.ZodType<V1ResponseChoiceElement> =
|
|
271
|
-
z
|
|
272
|
-
.object({
|
|
273
|
-
text: z.string().optional(),
|
|
274
|
-
message: z.lazy(() => unmarshalChatMessageSchema).optional(),
|
|
275
|
-
index: z.number().optional(),
|
|
276
|
-
finishReason: z.string().optional(),
|
|
277
|
-
logprobs: z.number().optional(),
|
|
278
|
-
})
|
|
279
|
-
.transform(d => ({
|
|
280
|
-
text: d.text,
|
|
281
|
-
message: d.message,
|
|
282
|
-
index: d.index,
|
|
283
|
-
finishReason: d.finishReason,
|
|
284
|
-
logprobs: d.logprobs,
|
|
285
|
-
}));
|
|
286
|
-
|
|
287
|
-
export const marshalChatMessageSchema: z.ZodType = z
|
|
288
|
-
.object({
|
|
289
|
-
role: z.enum(ChatMessageRole).optional(),
|
|
290
|
-
content: z.string().optional(),
|
|
291
|
-
})
|
|
292
|
-
.transform(d => ({
|
|
293
|
-
role: d.role,
|
|
294
|
-
content: d.content,
|
|
295
|
-
}));
|
|
296
|
-
|
|
297
|
-
export const marshalDataframeSplitInputSchema: z.ZodType = z
|
|
298
|
-
.object({
|
|
299
|
-
index: z.array(z.number()).optional(),
|
|
300
|
-
columns: z.array(jsonValueSchema).optional(),
|
|
301
|
-
data: z.array(jsonValueSchema).optional(),
|
|
302
|
-
})
|
|
303
|
-
.transform(d => ({
|
|
304
|
-
index: d.index,
|
|
305
|
-
columns: d.columns,
|
|
306
|
-
data: d.data,
|
|
307
|
-
}));
|
|
308
|
-
|
|
309
|
-
export const marshalQueryEndpointInputRequestSchema: z.ZodType = z
|
|
310
|
-
.object({
|
|
311
|
-
name: z.string().optional(),
|
|
312
|
-
prompt: jsonValueSchema.optional(),
|
|
313
|
-
input: jsonValueSchema.optional(),
|
|
314
|
-
messages: z.array(z.lazy(() => marshalChatMessageSchema)).optional(),
|
|
315
|
-
temperature: z.number().optional(),
|
|
316
|
-
stop: z.array(z.string()).optional(),
|
|
317
|
-
maxTokens: z.number().optional(),
|
|
318
|
-
n: z.number().optional(),
|
|
319
|
-
stream: z.boolean().optional(),
|
|
320
|
-
extraParams: z.record(z.string(), z.string()).optional(),
|
|
321
|
-
dataframeRecords: z.array(jsonValueSchema).optional(),
|
|
322
|
-
dataframeSplit: z.lazy(() => marshalDataframeSplitInputSchema).optional(),
|
|
323
|
-
instances: z.array(jsonValueSchema).optional(),
|
|
324
|
-
inputs: jsonValueSchema.optional(),
|
|
325
|
-
clientRequestId: z.string().optional(),
|
|
326
|
-
usageContext: z.record(z.string(), z.string()).optional(),
|
|
327
|
-
})
|
|
328
|
-
.transform(d => ({
|
|
329
|
-
name: d.name,
|
|
330
|
-
prompt: d.prompt,
|
|
331
|
-
input: d.input,
|
|
332
|
-
messages: d.messages,
|
|
333
|
-
temperature: d.temperature,
|
|
334
|
-
stop: d.stop,
|
|
335
|
-
max_tokens: d.maxTokens,
|
|
336
|
-
n: d.n,
|
|
337
|
-
stream: d.stream,
|
|
338
|
-
extra_params: d.extraParams,
|
|
339
|
-
dataframe_records: d.dataframeRecords,
|
|
340
|
-
dataframe_split: d.dataframeSplit,
|
|
341
|
-
instances: d.instances,
|
|
342
|
-
inputs: d.inputs,
|
|
343
|
-
client_request_id: d.clientRequestId,
|
|
344
|
-
usage_context: d.usageContext,
|
|
345
|
-
}));
|
package/src/v1/transport.ts
DELETED
|
@@ -1,73 +0,0 @@
|
|
|
1
|
-
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
-
|
|
3
|
-
import type {Credentials} from '@databricks/sdk-auth';
|
|
4
|
-
import {defaultCredentials} from '@databricks/sdk-auth/credentials';
|
|
5
|
-
import type {
|
|
6
|
-
HttpClient,
|
|
7
|
-
HttpRequest,
|
|
8
|
-
HttpResponse,
|
|
9
|
-
} from '@databricks/sdk-core/http';
|
|
10
|
-
import {newFetchHttpClient} from '@databricks/sdk-core/http';
|
|
11
|
-
import type {ClientOptions} from '@databricks/sdk-options/client';
|
|
12
|
-
|
|
13
|
-
/** Creates a new HTTP client with the given options. */
|
|
14
|
-
export function newHttpClient(options?: ClientOptions): HttpClient {
|
|
15
|
-
const opts = options ?? {};
|
|
16
|
-
|
|
17
|
-
// If an HTTP client is provided, use it as-is. Throw if other options are
|
|
18
|
-
// also set, since they would be silently ignored.
|
|
19
|
-
if (opts.httpClient !== undefined) {
|
|
20
|
-
if (opts.credentials !== undefined || opts.timeout !== undefined) {
|
|
21
|
-
throw new Error(
|
|
22
|
-
'httpClient cannot be combined with credentials or timeout'
|
|
23
|
-
);
|
|
24
|
-
}
|
|
25
|
-
return opts.httpClient;
|
|
26
|
-
}
|
|
27
|
-
|
|
28
|
-
const credentials = opts.credentials ?? defaultCredentials();
|
|
29
|
-
|
|
30
|
-
const base = newFetchHttpClient();
|
|
31
|
-
let client: HttpClient = new AuthHttpClient(base, credentials);
|
|
32
|
-
|
|
33
|
-
if (opts.timeout !== undefined) {
|
|
34
|
-
client = new TimeoutHttpClient(client, opts.timeout);
|
|
35
|
-
}
|
|
36
|
-
|
|
37
|
-
return client;
|
|
38
|
-
}
|
|
39
|
-
|
|
40
|
-
/** Wraps an HttpClient and adds authentication headers to requests. */
|
|
41
|
-
class AuthHttpClient implements HttpClient {
|
|
42
|
-
constructor(
|
|
43
|
-
private readonly base: HttpClient,
|
|
44
|
-
private readonly credentials: Credentials
|
|
45
|
-
) {}
|
|
46
|
-
|
|
47
|
-
async send(request: HttpRequest): Promise<HttpResponse> {
|
|
48
|
-
const authHeaders = await this.credentials.authHeaders();
|
|
49
|
-
// Do not modify the original request.
|
|
50
|
-
const headers = new Headers(request.headers);
|
|
51
|
-
for (const h of authHeaders) {
|
|
52
|
-
headers.set(h.key, h.value);
|
|
53
|
-
}
|
|
54
|
-
return this.base.send({...request, headers});
|
|
55
|
-
}
|
|
56
|
-
}
|
|
57
|
-
|
|
58
|
-
/** Wraps an HttpClient and applies a default timeout to requests. */
|
|
59
|
-
class TimeoutHttpClient implements HttpClient {
|
|
60
|
-
constructor(
|
|
61
|
-
private readonly base: HttpClient,
|
|
62
|
-
private readonly timeout: number
|
|
63
|
-
) {}
|
|
64
|
-
|
|
65
|
-
async send(request: HttpRequest): Promise<HttpResponse> {
|
|
66
|
-
const timeoutSignal = AbortSignal.timeout(this.timeout);
|
|
67
|
-
const signal =
|
|
68
|
-
request.signal !== undefined
|
|
69
|
-
? AbortSignal.any([request.signal, timeoutSignal])
|
|
70
|
-
: timeoutSignal;
|
|
71
|
-
return this.base.send({...request, signal});
|
|
72
|
-
}
|
|
73
|
-
}
|
package/src/v1/utils.ts
DELETED
|
@@ -1,156 +0,0 @@
|
|
|
1
|
-
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
-
|
|
3
|
-
import type {Options} from '@databricks/sdk-core/ops';
|
|
4
|
-
import {execute} from '@databricks/sdk-core/ops';
|
|
5
|
-
import {ApiError} from '@databricks/sdk-core/apierror';
|
|
6
|
-
import type {
|
|
7
|
-
HttpClient,
|
|
8
|
-
HttpRequest,
|
|
9
|
-
HttpResponse,
|
|
10
|
-
} from '@databricks/sdk-core/http';
|
|
11
|
-
import type {Logger} from '@databricks/sdk-core/logger';
|
|
12
|
-
import type {CallOptions} from '@databricks/sdk-options/call';
|
|
13
|
-
import JSONBig from 'json-bigint';
|
|
14
|
-
import type {z} from 'zod';
|
|
15
|
-
|
|
16
|
-
// JSON codec that preserves int64 precision. On the way in, large integer
|
|
17
|
-
// literals come back as bigint instead of being rounded to JS Number. On the
|
|
18
|
-
// way out, bigint values are emitted as raw JSON number digits.
|
|
19
|
-
const jsonBigint = JSONBig({useNativeBigInt: true});
|
|
20
|
-
|
|
21
|
-
export interface HttpCallOptions {
|
|
22
|
-
readonly request: HttpRequest;
|
|
23
|
-
readonly httpClient: HttpClient;
|
|
24
|
-
readonly logger: Logger;
|
|
25
|
-
}
|
|
26
|
-
|
|
27
|
-
/**
|
|
28
|
-
* Translates public CallOptions to the internal Options shape accepted by
|
|
29
|
-
* execute(). Even though the shapes match today, this isolates the public
|
|
30
|
-
* API from the executor's internal type so they can diverge.
|
|
31
|
-
*/
|
|
32
|
-
export async function executeCall(
|
|
33
|
-
call: (signal?: AbortSignal) => Promise<void>,
|
|
34
|
-
options?: CallOptions
|
|
35
|
-
): Promise<void> {
|
|
36
|
-
const opts: Options = {
|
|
37
|
-
...(options?.retrier !== undefined && {retrier: options.retrier}),
|
|
38
|
-
...(options?.rateLimiter !== undefined && {
|
|
39
|
-
rateLimiter: options.rateLimiter,
|
|
40
|
-
}),
|
|
41
|
-
...(options?.timeout !== undefined && {timeout: options.timeout}),
|
|
42
|
-
};
|
|
43
|
-
return execute(options?.signal, call, opts);
|
|
44
|
-
}
|
|
45
|
-
|
|
46
|
-
async function readAll(
|
|
47
|
-
body: ReadableStream<Uint8Array> | null
|
|
48
|
-
): Promise<Uint8Array> {
|
|
49
|
-
if (body === null) {
|
|
50
|
-
return new Uint8Array(0);
|
|
51
|
-
}
|
|
52
|
-
const reader = body.getReader();
|
|
53
|
-
const chunks: Uint8Array[] = [];
|
|
54
|
-
for (;;) {
|
|
55
|
-
const {done, value} = await reader.read();
|
|
56
|
-
if (done) {
|
|
57
|
-
break;
|
|
58
|
-
}
|
|
59
|
-
chunks.push(value);
|
|
60
|
-
}
|
|
61
|
-
const totalLength = chunks.reduce((acc, chunk) => acc + chunk.length, 0);
|
|
62
|
-
const result = new Uint8Array(totalLength);
|
|
63
|
-
let offset = 0;
|
|
64
|
-
for (const chunk of chunks) {
|
|
65
|
-
result.set(chunk, offset);
|
|
66
|
-
offset += chunk.length;
|
|
67
|
-
}
|
|
68
|
-
return result;
|
|
69
|
-
}
|
|
70
|
-
|
|
71
|
-
export async function executeHttpCall(
|
|
72
|
-
opts: HttpCallOptions
|
|
73
|
-
): Promise<Uint8Array> {
|
|
74
|
-
opts.logger.debug('HTTP request', {
|
|
75
|
-
method: opts.request.method,
|
|
76
|
-
url: opts.request.url,
|
|
77
|
-
});
|
|
78
|
-
|
|
79
|
-
let resp: HttpResponse;
|
|
80
|
-
try {
|
|
81
|
-
resp = await opts.httpClient.send(opts.request);
|
|
82
|
-
} catch (e: unknown) {
|
|
83
|
-
opts.logger.debug('HTTP request failed');
|
|
84
|
-
throw e;
|
|
85
|
-
}
|
|
86
|
-
|
|
87
|
-
const body = await readAll(resp.body);
|
|
88
|
-
|
|
89
|
-
opts.logger.debug('HTTP response', {
|
|
90
|
-
statusCode: resp.statusCode,
|
|
91
|
-
body: new TextDecoder().decode(body),
|
|
92
|
-
});
|
|
93
|
-
|
|
94
|
-
const apiErr = ApiError.fromHttpError(resp.statusCode, resp.headers, body);
|
|
95
|
-
if (apiErr !== undefined) {
|
|
96
|
-
throw apiErr;
|
|
97
|
-
}
|
|
98
|
-
|
|
99
|
-
return body;
|
|
100
|
-
}
|
|
101
|
-
|
|
102
|
-
export function buildHttpRequest(
|
|
103
|
-
method: string,
|
|
104
|
-
url: string,
|
|
105
|
-
headers: Headers,
|
|
106
|
-
signal?: AbortSignal,
|
|
107
|
-
body?: string | ReadableStream<Uint8Array>
|
|
108
|
-
): HttpRequest {
|
|
109
|
-
const req: HttpRequest = {url, method, headers};
|
|
110
|
-
if (body !== undefined) {
|
|
111
|
-
req.body = body;
|
|
112
|
-
}
|
|
113
|
-
if (signal !== undefined) {
|
|
114
|
-
req.signal = signal;
|
|
115
|
-
}
|
|
116
|
-
return req;
|
|
117
|
-
}
|
|
118
|
-
|
|
119
|
-
export function parseResponse<T>(body: Uint8Array, schema: z.ZodType<T>): T {
|
|
120
|
-
const text = new TextDecoder().decode(body);
|
|
121
|
-
const parsed: unknown = jsonBigint.parse(text);
|
|
122
|
-
return schema.parse(parsed);
|
|
123
|
-
}
|
|
124
|
-
|
|
125
|
-
export function marshalRequest(data: unknown, schema: z.ZodType): string {
|
|
126
|
-
return jsonBigint.stringify(schema.parse(data));
|
|
127
|
-
}
|
|
128
|
-
|
|
129
|
-
export function flattenQueryParams(
|
|
130
|
-
prefix: string,
|
|
131
|
-
value: unknown,
|
|
132
|
-
params: URLSearchParams
|
|
133
|
-
): void {
|
|
134
|
-
if (value === null || value === undefined) {
|
|
135
|
-
return;
|
|
136
|
-
}
|
|
137
|
-
if (Array.isArray(value)) {
|
|
138
|
-
// arrays of objects are not yet supported
|
|
139
|
-
for (const item of value) {
|
|
140
|
-
params.append(prefix, String(item));
|
|
141
|
-
}
|
|
142
|
-
} else if (typeof value === 'object') {
|
|
143
|
-
for (const [key, val] of Object.entries(value as Record<string, unknown>)) {
|
|
144
|
-
flattenQueryParams(`${prefix}.${key}`, val, params);
|
|
145
|
-
}
|
|
146
|
-
} else if (
|
|
147
|
-
typeof value === 'string' ||
|
|
148
|
-
typeof value === 'number' ||
|
|
149
|
-
typeof value === 'boolean' ||
|
|
150
|
-
typeof value === 'bigint'
|
|
151
|
-
) {
|
|
152
|
-
params.append(prefix, String(value));
|
|
153
|
-
} else {
|
|
154
|
-
throw new Error(`Unsupported query parameter type: ${typeof value}`);
|
|
155
|
-
}
|
|
156
|
-
}
|