@databricks/sdk-modelservingquery 0.1.0-dev.1 → 0.1.0-dev.2
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/dist/v1/client.d.ts +1 -1
- package/dist/v1/client.d.ts.map +1 -1
- package/dist/v1/client.js +2 -2
- package/dist/v1/client.js.map +1 -1
- package/dist/v1/index.d.ts +1 -1
- package/dist/v1/index.js +1 -1
- package/dist/v1/model.d.ts +7 -7
- package/dist/v1/model.js +7 -7
- package/dist/v1/utils.d.ts +1 -2
- package/dist/v1/utils.d.ts.map +1 -1
- package/dist/v1/utils.js +1 -1
- package/dist/v1/utils.js.map +1 -1
- package/package.json +4 -4
- package/src/v1/client.ts +3 -4
- package/src/v1/index.ts +1 -1
- package/src/v1/model.ts +7 -7
- package/src/v1/utils.ts +3 -3
package/dist/v1/client.d.ts
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
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
|
-
export declare class
|
|
4
|
+
export declare class ModelServingQueryClient {
|
|
5
5
|
private readonly host;
|
|
6
6
|
private readonly workspaceId;
|
|
7
7
|
private readonly httpClient;
|
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":"
|
|
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,IAAI,CAAS;IAI9B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAqB;IACjD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAa;IACxC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAIhC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;gBAEvB,OAAO,EAAE,aAAa;IAelC,+BAA+B;IACzB,KAAK,CACT,GAAG,EAAE,yBAAyB,EAC9B,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAAC,qBAAqB,CAAC;CAwBlC"}
|
package/dist/v1/client.js
CHANGED
|
@@ -11,7 +11,7 @@ const PACKAGE_SEGMENT = {
|
|
|
11
11
|
key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
|
|
12
12
|
value: pkgJson.version,
|
|
13
13
|
};
|
|
14
|
-
export class
|
|
14
|
+
export class ModelServingQueryClient {
|
|
15
15
|
host;
|
|
16
16
|
// Workspace ID used to route workspace-level calls on unified hosts (SPOG).
|
|
17
17
|
// When set, workspace-level methods send X-Databricks-Org-Id on every
|
|
@@ -58,7 +58,7 @@ export class ModelservingqueryClient {
|
|
|
58
58
|
};
|
|
59
59
|
await executeCall(call, options);
|
|
60
60
|
if (resp === undefined) {
|
|
61
|
-
throw new Error('
|
|
61
|
+
throw new Error('operation completed without a result.');
|
|
62
62
|
}
|
|
63
63
|
return resp;
|
|
64
64
|
}
|
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;
|
|
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,aAAa,EAAC,MAAM,aAAa,CAAC;AAC1C,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,IAAI,CAAS;IAC9B,4EAA4E;IAC5E,sEAAsE;IACtE,WAAW;IACM,WAAW,CAAqB;IAChC,UAAU,CAAa;IACvB,MAAM,CAAS;IAChC,8DAA8D;IAC9D,qEAAqE;IACrE,qBAAqB;IACJ,SAAS,CAAS;IAEnC,YAAY,OAAsB;QAChC,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,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;QACjC,IAAI,CAAC,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,+BAA+B;IAC/B,KAAK,CAAC,KAAK,CACT,GAA8B,EAC9B,OAAqB;QAErB,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,IAAI,0BAA0B,GAAG,CAAC,IAAI,IAAI,EAAE,cAAc,CAAC;QAC/E,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,IAAI,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;YACvD,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,EAAE,IAAI,CAAC,UAAU;gBAC3B,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
|
-
export {
|
|
1
|
+
export { ModelServingQueryClient } from './client';
|
|
2
2
|
export { ChatMessageRole, EmbeddingsV1ResponseEmbeddingElementObject, QueryEndpointResponseObject, } from './model';
|
|
3
3
|
export type { ChatMessage, DataframeSplitInput, EmbeddingsV1ResponseEmbeddingElement, ExternalModelUsageElement, QueryEndpointInputRequest, QueryEndpointInputRequest_ExtraParamsEntry, QueryEndpointInputRequest_UsageContextEntry, QueryEndpointResponse, V1ResponseChoiceElement, } from './model';
|
|
4
4
|
//# sourceMappingURL=index.d.ts.map
|
package/dist/v1/index.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
1
|
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
-
export {
|
|
2
|
+
export { ModelServingQueryClient } from './client';
|
|
3
3
|
export { ChatMessageRole, EmbeddingsV1ResponseEmbeddingElementObject, QueryEndpointResponseObject, } from './model';
|
|
4
4
|
//# sourceMappingURL=index.js.map
|
package/dist/v1/model.d.ts
CHANGED
|
@@ -3,14 +3,14 @@ import { z } from 'zod';
|
|
|
3
3
|
/** The role of the message. One of [system, user, assistant]. */
|
|
4
4
|
export declare enum ChatMessageRole {
|
|
5
5
|
CHAT_MESSAGE_ROLE_UNSPECIFIED = "CHAT_MESSAGE_ROLE_UNSPECIFIED",
|
|
6
|
-
SYSTEM = "
|
|
7
|
-
USER = "
|
|
8
|
-
ASSISTANT = "
|
|
6
|
+
SYSTEM = "system",
|
|
7
|
+
USER = "user",
|
|
8
|
+
ASSISTANT = "assistant"
|
|
9
9
|
}
|
|
10
10
|
/** This will always be 'embedding'. */
|
|
11
11
|
export declare enum EmbeddingsV1ResponseEmbeddingElementObject {
|
|
12
12
|
EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED = "EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED",
|
|
13
|
-
EMBEDDING = "
|
|
13
|
+
EMBEDDING = "embedding"
|
|
14
14
|
}
|
|
15
15
|
/**
|
|
16
16
|
* The type of object returned by the __external/foundation model__ serving endpoint, one of
|
|
@@ -18,9 +18,9 @@ export declare enum EmbeddingsV1ResponseEmbeddingElementObject {
|
|
|
18
18
|
*/
|
|
19
19
|
export declare enum QueryEndpointResponseObject {
|
|
20
20
|
QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED = "QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED",
|
|
21
|
-
TEXT_COMPLETION = "
|
|
22
|
-
CHAT_COMPLETION = "
|
|
23
|
-
LIST = "
|
|
21
|
+
TEXT_COMPLETION = "text_completion",
|
|
22
|
+
CHAT_COMPLETION = "chat.completion",
|
|
23
|
+
LIST = "list"
|
|
24
24
|
}
|
|
25
25
|
export interface ChatMessage {
|
|
26
26
|
/** The role of the message. One of [system, user, assistant]. */
|
package/dist/v1/model.js
CHANGED
|
@@ -12,15 +12,15 @@ const jsonValueSchema = z.lazy(() => z.union([
|
|
|
12
12
|
export var ChatMessageRole;
|
|
13
13
|
(function (ChatMessageRole) {
|
|
14
14
|
ChatMessageRole["CHAT_MESSAGE_ROLE_UNSPECIFIED"] = "CHAT_MESSAGE_ROLE_UNSPECIFIED";
|
|
15
|
-
ChatMessageRole["SYSTEM"] = "
|
|
16
|
-
ChatMessageRole["USER"] = "
|
|
17
|
-
ChatMessageRole["ASSISTANT"] = "
|
|
15
|
+
ChatMessageRole["SYSTEM"] = "system";
|
|
16
|
+
ChatMessageRole["USER"] = "user";
|
|
17
|
+
ChatMessageRole["ASSISTANT"] = "assistant";
|
|
18
18
|
})(ChatMessageRole || (ChatMessageRole = {}));
|
|
19
19
|
/** This will always be 'embedding'. */
|
|
20
20
|
export var EmbeddingsV1ResponseEmbeddingElementObject;
|
|
21
21
|
(function (EmbeddingsV1ResponseEmbeddingElementObject) {
|
|
22
22
|
EmbeddingsV1ResponseEmbeddingElementObject["EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED"] = "EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED";
|
|
23
|
-
EmbeddingsV1ResponseEmbeddingElementObject["EMBEDDING"] = "
|
|
23
|
+
EmbeddingsV1ResponseEmbeddingElementObject["EMBEDDING"] = "embedding";
|
|
24
24
|
})(EmbeddingsV1ResponseEmbeddingElementObject || (EmbeddingsV1ResponseEmbeddingElementObject = {}));
|
|
25
25
|
/**
|
|
26
26
|
* The type of object returned by the __external/foundation model__ serving endpoint, one of
|
|
@@ -29,9 +29,9 @@ export var EmbeddingsV1ResponseEmbeddingElementObject;
|
|
|
29
29
|
export var QueryEndpointResponseObject;
|
|
30
30
|
(function (QueryEndpointResponseObject) {
|
|
31
31
|
QueryEndpointResponseObject["QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED"] = "QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED";
|
|
32
|
-
QueryEndpointResponseObject["TEXT_COMPLETION"] = "
|
|
33
|
-
QueryEndpointResponseObject["CHAT_COMPLETION"] = "
|
|
34
|
-
QueryEndpointResponseObject["LIST"] = "
|
|
32
|
+
QueryEndpointResponseObject["TEXT_COMPLETION"] = "text_completion";
|
|
33
|
+
QueryEndpointResponseObject["CHAT_COMPLETION"] = "chat.completion";
|
|
34
|
+
QueryEndpointResponseObject["LIST"] = "list";
|
|
35
35
|
})(QueryEndpointResponseObject || (QueryEndpointResponseObject = {}));
|
|
36
36
|
export const unmarshalChatMessageSchema = z
|
|
37
37
|
.object({
|
package/dist/v1/utils.d.ts
CHANGED
|
@@ -1,4 +1,3 @@
|
|
|
1
|
-
import type { Call } from '@databricks/sdk-core/api';
|
|
2
1
|
import type { HttpClient, HttpRequest } from '@databricks/sdk-core/http';
|
|
3
2
|
import type { Logger } from '@databricks/sdk-core/logger';
|
|
4
3
|
import type { CallOptions } from '@databricks/sdk-options/call';
|
|
@@ -13,7 +12,7 @@ export interface HttpCallOptions {
|
|
|
13
12
|
* execute(). Even though the shapes match today, this isolates the public
|
|
14
13
|
* API from the executor's internal type so they can diverge.
|
|
15
14
|
*/
|
|
16
|
-
export declare function executeCall(call:
|
|
15
|
+
export declare function executeCall(call: (signal?: AbortSignal) => Promise<void>, options?: CallOptions): Promise<void>;
|
|
17
16
|
export declare function executeHttpCall(opts: HttpCallOptions): Promise<Uint8Array>;
|
|
18
17
|
export declare function buildHttpRequest(method: string, url: string, headers: Headers, signal?: AbortSignal, body?: string | ReadableStream<Uint8Array>): HttpRequest;
|
|
19
18
|
export declare function parseResponse<T>(body: Uint8Array, schema: z.ZodType<T>): T;
|
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":"
|
|
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,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"}
|
package/dist/v1/utils.js
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
|
-
import { execute } from '@databricks/sdk-core/
|
|
2
|
+
import { execute } from '@databricks/sdk-core/ops';
|
|
3
3
|
import { ApiError } from '@databricks/sdk-core/apierror';
|
|
4
4
|
import JSONBig from 'json-bigint';
|
|
5
5
|
// JSON codec that preserves int64 precision. On the way in, large integer
|
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,
|
|
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;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,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@databricks/sdk-modelservingquery",
|
|
3
|
-
"version": "0.1.0-dev.
|
|
3
|
+
"version": "0.1.0-dev.2",
|
|
4
4
|
"description": "",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"exports": {
|
|
@@ -28,9 +28,9 @@
|
|
|
28
28
|
"author": "Databricks",
|
|
29
29
|
"license": "Apache-2.0",
|
|
30
30
|
"dependencies": {
|
|
31
|
-
"@databricks/sdk-auth": ">=0.1.0-dev.
|
|
32
|
-
"@databricks/sdk-core": ">=0.1.0-dev.
|
|
33
|
-
"@databricks/sdk-options": ">=0.1.0-dev.
|
|
31
|
+
"@databricks/sdk-auth": ">=0.1.0-dev.3 <1.0.0",
|
|
32
|
+
"@databricks/sdk-core": ">=0.1.0-dev.4 <1.0.0",
|
|
33
|
+
"@databricks/sdk-options": ">=0.1.0-dev.3 <1.0.0",
|
|
34
34
|
"@js-temporal/polyfill": "^0.5.0",
|
|
35
35
|
"json-bigint": "^1.0.0",
|
|
36
36
|
"zod": "^4.3.6"
|
package/src/v1/client.ts
CHANGED
|
@@ -1,7 +1,6 @@
|
|
|
1
1
|
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
2
|
|
|
3
3
|
import {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';
|
|
4
|
-
import type {Call} from '@databricks/sdk-core/api';
|
|
5
4
|
import {createDefault} from '@databricks/sdk-core/clientinfo';
|
|
6
5
|
import type {Logger} from '@databricks/sdk-core/logger';
|
|
7
6
|
import {NoOpLogger} from '@databricks/sdk-core/logger';
|
|
@@ -29,7 +28,7 @@ const PACKAGE_SEGMENT = {
|
|
|
29
28
|
value: pkgJson.version,
|
|
30
29
|
};
|
|
31
30
|
|
|
32
|
-
export class
|
|
31
|
+
export class ModelServingQueryClient {
|
|
33
32
|
private readonly host: string;
|
|
34
33
|
// Workspace ID used to route workspace-level calls on unified hosts (SPOG).
|
|
35
34
|
// When set, workspace-level methods send X-Databricks-Org-Id on every
|
|
@@ -65,7 +64,7 @@ export class ModelservingqueryClient {
|
|
|
65
64
|
const url = `${this.host}/api/serving-endpoints/${req.name ?? ''}/invocations`;
|
|
66
65
|
const body = marshalRequest(req, marshalQueryEndpointInputRequestSchema);
|
|
67
66
|
let resp: QueryEndpointResponse | undefined;
|
|
68
|
-
const call
|
|
67
|
+
const call = async (callSignal?: AbortSignal): Promise<void> => {
|
|
69
68
|
const headers = new Headers({'Content-Type': 'application/json'});
|
|
70
69
|
if (this.workspaceId !== undefined) {
|
|
71
70
|
headers.set('X-Databricks-Org-Id', this.workspaceId);
|
|
@@ -81,7 +80,7 @@ export class ModelservingqueryClient {
|
|
|
81
80
|
};
|
|
82
81
|
await executeCall(call, options);
|
|
83
82
|
if (resp === undefined) {
|
|
84
|
-
throw new Error('
|
|
83
|
+
throw new Error('operation completed without a result.');
|
|
85
84
|
}
|
|
86
85
|
return resp;
|
|
87
86
|
}
|
package/src/v1/index.ts
CHANGED
package/src/v1/model.ts
CHANGED
|
@@ -17,15 +17,15 @@ const jsonValueSchema: z.ZodType<JsonValue> = z.lazy(() =>
|
|
|
17
17
|
/** The role of the message. One of [system, user, assistant]. */
|
|
18
18
|
export enum ChatMessageRole {
|
|
19
19
|
CHAT_MESSAGE_ROLE_UNSPECIFIED = 'CHAT_MESSAGE_ROLE_UNSPECIFIED',
|
|
20
|
-
SYSTEM = '
|
|
21
|
-
USER = '
|
|
22
|
-
ASSISTANT = '
|
|
20
|
+
SYSTEM = 'system',
|
|
21
|
+
USER = 'user',
|
|
22
|
+
ASSISTANT = 'assistant',
|
|
23
23
|
}
|
|
24
24
|
|
|
25
25
|
/** This will always be 'embedding'. */
|
|
26
26
|
export enum EmbeddingsV1ResponseEmbeddingElementObject {
|
|
27
27
|
EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED = 'EMBEDDINGS_V1_RESPONSE_EMBEDDING_ELEMENT_OBJECT_UNSPECIFIED',
|
|
28
|
-
EMBEDDING = '
|
|
28
|
+
EMBEDDING = 'embedding',
|
|
29
29
|
}
|
|
30
30
|
|
|
31
31
|
/**
|
|
@@ -34,9 +34,9 @@ export enum EmbeddingsV1ResponseEmbeddingElementObject {
|
|
|
34
34
|
*/
|
|
35
35
|
export enum QueryEndpointResponseObject {
|
|
36
36
|
QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED = 'QUERY_ENDPOINT_RESPONSE_OBJECT_UNSPECIFIED',
|
|
37
|
-
TEXT_COMPLETION = '
|
|
38
|
-
CHAT_COMPLETION = '
|
|
39
|
-
LIST = '
|
|
37
|
+
TEXT_COMPLETION = 'text_completion',
|
|
38
|
+
CHAT_COMPLETION = 'chat.completion',
|
|
39
|
+
LIST = 'list',
|
|
40
40
|
}
|
|
41
41
|
|
|
42
42
|
export interface ChatMessage {
|
package/src/v1/utils.ts
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
|
|
2
2
|
|
|
3
|
-
import type {
|
|
4
|
-
import {execute} from '@databricks/sdk-core/
|
|
3
|
+
import type {Options} from '@databricks/sdk-core/ops';
|
|
4
|
+
import {execute} from '@databricks/sdk-core/ops';
|
|
5
5
|
import {ApiError} from '@databricks/sdk-core/apierror';
|
|
6
6
|
import type {
|
|
7
7
|
HttpClient,
|
|
@@ -30,7 +30,7 @@ export interface HttpCallOptions {
|
|
|
30
30
|
* API from the executor's internal type so they can diverge.
|
|
31
31
|
*/
|
|
32
32
|
export async function executeCall(
|
|
33
|
-
call:
|
|
33
|
+
call: (signal?: AbortSignal) => Promise<void>,
|
|
34
34
|
options?: CallOptions
|
|
35
35
|
): Promise<void> {
|
|
36
36
|
const opts: Options = {
|