@zhivex-ai/vertex 1.0.2 → 1.1.0

package/dist/specialized.d.ts

@@ -0,0 +1,14 @@
+import { type DocumentExtractionInput, type DocumentExtractionResult, type TextCompletionInput, type GenerateResult, type StreamEvent } from "@zhivex-ai/core/provider";
+export interface VertexOCRClient {
+    process(input: DocumentExtractionInput): Promise<DocumentExtractionResult>;
+}
+export interface VertexFIMClient {
+    generate(input: TextCompletionInput): Promise<GenerateResult>;
+    stream(input: TextCompletionInput): Promise<AsyncIterable<StreamEvent>>;
+}
+export interface VertexSpecializedClients {
+    ocr: VertexOCRClient;
+    fim: VertexFIMClient;
+}
+export declare const createVertexSpecializedClients: (baseURL: string, fetcher: typeof globalThis.fetch, assertAccess: () => void) => VertexSpecializedClients;
+//# sourceMappingURL=specialized.d.ts.map

package/dist/specialized.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"specialized.d.ts","sourceRoot":"","sources":["../src/specialized.ts"],"names":[],"mappings":"AAAA,OAAO,EAIL,KAAK,uBAAuB,EAA+B,KAAK,wBAAwB,EACxF,KAAK,mBAAmB,EAAE,KAAK,cAAc,EAAE,KAAK,WAAW,EAChE,MAAM,0BAA0B,CAAC;AAGlC,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAAC;CAC5E;AACD,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IAC9D,MAAM,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC;CACzE;AACD,MAAM,WAAW,wBAAwB;IACvC,GAAG,EAAE,eAAe,CAAC;IACrB,GAAG,EAAE,eAAe,CAAC;CACtB;AAYD,eAAO,MAAM,8BAA8B,YAAa,MAAM,WAAW,OAAO,UAAU,CAAC,KAAK,gBAAgB,MAAM,IAAI,KAAG,wBAgF5H,CAAC"}

package/dist/specialized.js

@@ -0,0 +1,117 @@
+import { ConfigurationError, ProviderHTTPError, UnsupportedFeatureError, createChatCompletionsModel, encodeMediaFrame, readErrorBodyWithLimit, readJsonWithLimit, withRetry, withTimeoutSignal } from "@zhivex-ai/core/provider";
+import { createVertexChatModel } from "./chat.js";
+const mistralModel = (value, family) => {
+    const model = value.replace(/^mistralai\//, "");
+    if (!model.startsWith(family) || /[\\/?#\s]/.test(model))
+        throw new ConfigurationError(`Vertex requires a ${family} model ID for this API.`);
+    return model;
+};
+const assertNativeOptions = (options, reserved, surface) => {
+    for (const key of reserved)
+        if (options?.[key] !== undefined)
+            throw new ConfigurationError(`Vertex ${surface} providerOptions.${key} conflicts with its dedicated input contract.`);
+};
+export const createVertexSpecializedClients = (baseURL, fetcher, assertAccess) => {
+    const url = (model, stream = false) => `${baseURL}/publishers/mistralai/models/${encodeURIComponent(model)}:${stream ? "streamRawPredict" : "rawPredict"}`;
+    const completion = (input) => {
+        assertAccess();
+        const modelId = mistralModel(input.modelId, "codestral");
+        assertNativeOptions(input.providerOptions, ["model", "prompt", "suffix", "stream", "messages", "tools", "tool_choice", "response_format", "max_tokens", "temperature", "top_p", "stop"], "FIM");
+        if (typeof input.prompt !== "string" || (input.suffix !== undefined && typeof input.suffix !== "string"))
+            throw new ConfigurationError("Vertex FIM requires a text prompt and optional text suffix.");
+        if (input.maxTokens !== undefined && (!Number.isInteger(input.maxTokens) || input.maxTokens <= 0))
+            throw new ConfigurationError("Vertex FIM maxTokens must be a positive integer.");
+        if (input.temperature !== undefined && (!Number.isFinite(input.temperature) || input.temperature < 0))
+            throw new ConfigurationError("Vertex FIM temperature must be finite and nonnegative.");
+        if (input.topP !== undefined && (!Number.isFinite(input.topP) || input.topP <= 0 || input.topP > 1))
+            throw new ConfigurationError("Vertex FIM topP must be greater than zero and at most one.");
+        if (input.stop !== undefined && (typeof input.stop !== "string" && (!Array.isArray(input.stop) || input.stop.some((value) => typeof value !== "string"))))
+            throw new ConfigurationError("Vertex FIM stop must be text or an array of text values.");
+        return createChatCompletionsModel({ provider: "vertex", modelId,
+            capabilities: { streaming: true, tools: false, structuredOutput: false, jsonMode: false, toolChoice: false, parallelToolCalls: false, vision: false, files: false, audioInput: false, audioOutput: false, embeddings: false, reasoning: false, webSearch: false },
+            send: (body, signal) => fetcher(url(modelId, body.stream === true), {
+                method: "POST", headers: { "content-type": "application/json" }, redirect: "error", signal,
+                body: JSON.stringify({ ...input.providerOptions, model: modelId.split("@")[0], prompt: input.prompt,
+                    ...(input.suffix !== undefined ? { suffix: input.suffix } : {}),
+                    ...(input.maxTokens !== undefined ? { max_tokens: input.maxTokens } : {}),
+                    ...(input.temperature !== undefined ? { temperature: input.temperature } : {}),
+                    ...(input.topP !== undefined ? { top_p: input.topP } : {}),
+                    ...(input.stop !== undefined ? { stop: input.stop } : {}), stream: body.stream })
+            })
+        });
+    };
+    return {
+        fim: {
+            generate: (input) => completion(input).generate({ ...input, messages: [] }),
+            stream: (input) => completion(input).stream({ ...input, messages: [] })
+        },
+        ocr: {
+            async process(input) {
+                assertAccess();
+                assertNativeOptions(input.providerOptions, ["model", "document", "pages", "include_image_base64", "stream", "messages"], "OCR");
+                const media = input.document;
+                if (!media || (media.uri === undefined) === (media.data === undefined))
+                    throw new ConfigurationError("Vertex OCR requires exactly one document uri or data value.");
+                const image = /^image\/(?:png|jpeg|webp|gif)$/.test(media.mediaType);
+                if (!image && media.mediaType !== "application/pdf")
+                    throw new UnsupportedFeatureError("Vertex Mistral OCR accepts PDF or image input.");
+                if (input.pages?.some((page) => !Number.isInteger(page) || page < 0))
+                    throw new ConfigurationError("OCR pages must contain nonnegative integer indices.");
+                if (input.includeImages !== undefined && typeof input.includeImages !== "boolean")
+                    throw new ConfigurationError("OCR includeImages must be a boolean.");
+                if (media.uri !== undefined && !/^https?:\/\//.test(media.uri))
+                    throw new ConfigurationError("Vertex OCR document URLs must use HTTP(S); use inline data for local files.");
+                const source = media.uri ?? `data:${media.mediaType};base64,${encodeMediaFrame({ data: media.data, mediaType: media.mediaType })}`;
+                if (["deepseek-ai/deepseek-ocr-maas", "publishers/deepseek-ai/models/deepseek-ocr-maas"].includes(input.modelId)) {
+                    if (!image)
+                        throw new UnsupportedFeatureError("Vertex DeepSeek OCR requires one image; rasterize PDF pages before extraction.");
+                    if (input.pages !== undefined || input.includeImages !== undefined)
+                        throw new UnsupportedFeatureError("Vertex DeepSeek OCR does not expose page selection or extracted images.");
+                    if (input.prompt !== undefined && (typeof input.prompt !== "string" || !input.prompt.trim()))
+                        throw new ConfigurationError("Vertex DeepSeek OCR prompt must be nonempty text.");
+                    assertNativeOptions(input.providerOptions, ["prompt", "tools", "tool_choice", "response_format"], "DeepSeek OCR");
+                    const result = await createVertexChatModel("deepseek-ai/deepseek-ocr-maas", baseURL, fetcher).generate({
+                        ...input,
+                        messages: [{ role: "user", parts: [{ type: "text", text: input.prompt ?? "Free OCR" }, { type: "image", image: source }] }]
+                    });
+                    if (result.finishReason !== "stop")
+                        throw new ConfigurationError(`Vertex DeepSeek OCR did not complete extraction (${result.finishReason ?? "unknown"}).`);
+                    if (typeof result.text !== "string")
+                        throw new ConfigurationError("Vertex DeepSeek OCR returned no text.");
+                    // The input is one image, not a paginated document. Index zero identifies it.
+                    return { text: result.text, pages: [{ index: 0, markdown: result.text }], rawResponse: result.rawResponse };
+                }
+                const model = mistralModel(input.modelId, "mistral-ocr-");
+                if (input.prompt !== undefined)
+                    throw new UnsupportedFeatureError("Vertex Mistral OCR does not expose a text extraction prompt.");
+                const body = { ...input.providerOptions, model: model.split("@")[0],
+                    document: image ? { type: "image_url", image_url: source } : { type: "document_url", document_url: source },
+                    ...(input.pages ? { pages: input.pages } : {}), ...(input.includeImages !== undefined ? { include_image_base64: input.includeImages } : {}) };
+                const { signal, cleanup } = withTimeoutSignal(input);
+                try {
+                    const response = await withRetry(async () => {
+                        const result = await fetcher(url(model), { method: "POST", headers: { "content-type": "application/json" }, redirect: "error", signal, body: JSON.stringify(body) });
+                        if (!result.ok)
+                            throw new ProviderHTTPError(`Vertex OCR failed with status ${result.status}.`, result.status, { responseBody: await readErrorBodyWithLimit(result) });
+                        return result;
+                    }, { ...input, abortSignal: signal });
+                    const json = await readJsonWithLimit(response, { maxBytes: 128 * 1024 * 1024 });
+                    if (!Array.isArray(json.pages))
+                        throw new ConfigurationError("Vertex OCR returned no pages array.");
+                    const indices = new Set();
+                    const pages = json.pages.map((page) => {
+                        if (!page || !Number.isInteger(page.index) || page.index < 0 || typeof page.markdown !== "string" || indices.has(page.index))
+                            throw new ConfigurationError("Vertex OCR returned an invalid or duplicate page.");
+                        indices.add(page.index);
+                        return { index: page.index, markdown: page.markdown, ...(Array.isArray(page.images) ? { images: page.images } : {}), providerMetadata: page };
+                    });
+                    return { text: pages.map((page) => page.markdown).join("\n\n"), pages, rawResponse: json };
+                }
+                finally {
+                    cleanup();
+                }
+            }
+        }
+    };
+};
+//# sourceMappingURL=specialized.js.map

package/dist/specialized.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"specialized.js","sourceRoot":"","sources":["../src/specialized.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,kBAAkB,EAAE,iBAAiB,EAAE,uBAAuB,EAC9D,0BAA0B,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,iBAAiB,EACvF,SAAS,EAAE,iBAAiB,EAG7B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAclD,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,MAAoC,EAAE,EAAE;IAC3E,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;IAChD,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,MAAM,IAAI,kBAAkB,CAAC,qBAAqB,MAAM,yBAAyB,CAAC,CAAC;IAC7I,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAAC,OAA4C,EAAE,QAAkB,EAAE,OAAe,EAAE,EAAE;IAChH,KAAK,MAAM,GAAG,IAAI,QAAQ;QAAE,IAAI,OAAO,EAAE,CAAC,GAAG,CAAC,KAAK,SAAS;YAAE,MAAM,IAAI,kBAAkB,CAAC,UAAU,OAAO,oBAAoB,GAAG,+CAA+C,CAAC,CAAC;AACtL,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,OAAe,EAAE,OAAgC,EAAE,YAAwB,EAA4B,EAAE;IACtJ,MAAM,GAAG,GAAG,CAAC,KAAa,EAAE,MAAM,GAAG,KAAK,EAAE,EAAE,CAAC,GAAG,OAAO,gCAAgC,kBAAkB,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC;IACnK,MAAM,UAAU,GAAG,CAAC,KAA0B,EAAE,EAAE;QAChD,YAAY,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QACzD,mBAAmB,CAAC,KAAK,CAAC,eAAe,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,YAAY,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC;QAChM,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,6DAA6D,CAAC,CAAC;QACtM,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,kDAAkD,CAAC,CAAC;QACpL,IAAI,KAAK,CAAC,WAAW,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,WAAW,GAAG,CAAC,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,wDAAwD,CAAC,CAAC;QAC9L,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,4DAA4D,CAAC,CAAC;QAChM,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,0DAA0D,CAAC,CAAC;QACpP,OAAO,0BAA0B,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO;YAC7D,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE;YACjQ,IAAI,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,KAAK,IAAI,CAAC,EAAE;gBAClE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM;gBAC1F,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,KAAK,CAAC,eAAe,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM;oBACjG,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC/D,GAAG,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACzE,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC9E,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC1D,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;aACpF,CAAC;SACH,CAAC,CAAC;IACL,CAAC,CAAC;IACF,OAAO;QACL,GAAG,EAAE;YACH,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;YAC3E,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAO,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;SACzE;QACD,GAAG,EAAE;YACH,KAAK,CAAC,OAAO,CAAC,KAAK;gBACjB,YAAY,EAAE,CAAC;gBACf,mBAAmB,CAAC,KAAK,CAAC,eAAe,EAAE,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,sBAAsB,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;gBAChI,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC;gBAC7B,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,GAAG,KAAK,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC;oBAAE,MAAM,IAAI,kBAAkB,CAAC,6DAA6D,CAAC,CAAC;gBACpK,MAAM,KAAK,GAAG,gCAAgC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBACrE,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,SAAS,KAAK,iBAAiB;oBAAE,MAAM,IAAI,uBAAuB,CAAC,gDAAgD,CAAC,CAAC;gBACzI,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;oBAAE,MAAM,IAAI,kBAAkB,CAAC,qDAAqD,CAAC,CAAC;gBAC1J,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS,IAAI,OAAO,KAAK,CAAC,aAAa,KAAK,SAAS;oBAAE,MAAM,IAAI,kBAAkB,CAAC,sCAAsC,CAAC,CAAC;gBACxJ,IAAI,KAAK,CAAC,GAAG,KAAK,SAAS,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC;oBAAE,MAAM,IAAI,kBAAkB,CAAC,6EAA6E,CAAC,CAAC;gBAC5K,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,IAAI,QAAQ,KAAK,CAAC,SAAS,WAAW,gBAAgB,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAK,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC;gBACpI,IAAI,CAAC,+BAA+B,EAAE,iDAAiD,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;oBACjH,IAAI,CAAC,KAAK;wBAAE,MAAM,IAAI,uBAAuB,CAAC,gFAAgF,CAAC,CAAC;oBAChI,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS;wBAAE,MAAM,IAAI,uBAAuB,CAAC,yEAAyE,CAAC,CAAC;oBACjL,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;wBAAE,MAAM,IAAI,kBAAkB,CAAC,mDAAmD,CAAC,CAAC;oBAChL,mBAAmB,CAAC,KAAK,CAAC,eAAe,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,aAAa,EAAE,iBAAiB,CAAC,EAAE,cAAc,CAAC,CAAC;oBAClH,MAAM,MAAM,GAAG,MAAM,qBAAqB,CAAC,+BAA+B,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC;wBACrG,GAAG,KAAK;wBACR,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,IAAI,UAAU,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;qBAC5H,CAAC,CAAC;oBACH,IAAI,MAAM,CAAC,YAAY,KAAK,MAAM;wBAAE,MAAM,IAAI,kBAAkB,CAAC,oDAAoD,MAAM,CAAC,YAAY,IAAI,SAAS,IAAI,CAAC,CAAC;oBAC3J,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ;wBAAE,MAAM,IAAI,kBAAkB,CAAC,uCAAuC,CAAC,CAAC;oBAC3G,8EAA8E;oBAC9E,OAAO,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC;gBAC9G,CAAC;gBACD,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;gBAC1D,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS;oBAAE,MAAM,IAAI,uBAAuB,CAAC,8DAA8D,CAAC,CAAC;gBAClI,MAAM,IAAI,GAAG,EAAE,GAAG,KAAK,CAAC,eAAe,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBACjE,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE;oBAC3G,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,aAAa,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,KAAK,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;gBAChJ,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBACrD,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,KAAK,IAAI,EAAE;wBAC1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBACrK,IAAI,CAAC,MAAM,CAAC,EAAE;4BAAE,MAAM,IAAI,iBAAiB,CAAC,iCAAiC,MAAM,CAAC,MAAM,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;wBACtK,OAAO,MAAM,CAAC;oBAChB,CAAC,EAAE,EAAE,GAAG,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;oBACtC,MAAM,IAAI,GAAQ,MAAM,iBAAiB,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,GAAG,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC,CAAC;oBACrF,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;wBAAE,MAAM,IAAI,kBAAkB,CAAC,qCAAqC,CAAC,CAAC;oBACpG,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;oBAClC,MAAM,KAAK,GAA6B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;wBACnE,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;4BAAE,MAAM,IAAI,kBAAkB,CAAC,mDAAmD,CAAC,CAAC;wBAChN,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBACxB,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC;oBAChJ,CAAC,CAAC,CAAC;oBACH,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;gBAC7F,CAAC;wBAAS,CAAC;oBAAC,OAAO,EAAE,CAAC;gBAAC,CAAC;YAC1B,CAAC;SACF;KACF,CAAC;AACJ,CAAC,CAAC"}

package/dist/tensors.d.ts

@@ -0,0 +1,22 @@
+export type VertexTensorDataType = "DATA_TYPE_UNSPECIFIED" | "BOOL" | "STRING" | "FLOAT" | "DOUBLE" | "INT8" | "INT16" | "INT32" | "INT64" | "UINT8" | "UINT16" | "UINT32" | "UINT64";
+type ProtoFloat = number | "NaN" | "Infinity" | "-Infinity";
+/** Native REST Tensor. 64-bit integers stay strings to preserve precision. */
+export interface VertexTensor {
+    dtype?: VertexTensorDataType;
+    shape?: string[];
+    boolVal?: boolean[];
+    stringVal?: string[];
+    bytesVal?: string[];
+    floatVal?: ProtoFloat[];
+    doubleVal?: ProtoFloat[];
+    intVal?: number[];
+    int64Val?: string[];
+    uintVal?: number[];
+    uint64Val?: string[];
+    listVal?: VertexTensor[];
+    structVal?: Record<string, VertexTensor>;
+    tensorVal?: string;
+}
+export declare function assertVertexTensor(value: unknown, depth?: number): asserts value is VertexTensor;
+export {};
+//# sourceMappingURL=tensors.d.ts.map

package/dist/tensors.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tensors.d.ts","sourceRoot":"","sources":["../src/tensors.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,oBAAoB,GAAG,uBAAuB,GAAG,MAAM,GAAG,QAAQ,GAAG,OAAO,GAAG,QAAQ,GAAG,MAAM,GAAG,OAAO,GAAG,OAAO,GAAG,OAAO,GAAG,OAAO,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,CAAC;AACtL,KAAK,UAAU,GAAG,MAAM,GAAG,KAAK,GAAG,UAAU,GAAG,WAAW,CAAC;AAC5D,8EAA8E;AAC9E,MAAM,WAAW,YAAY;IAC3B,KAAK,CAAC,EAAE,oBAAoB,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,QAAQ,CAAC,EAAE,UAAU,EAAE,CAAC;IACxB,SAAS,CAAC,EAAE,UAAU,EAAE,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,OAAO,CAAC,EAAE,YAAY,EAAE,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IACzC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AASD,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,SAAI,GAAG,OAAO,CAAC,KAAK,IAAI,YAAY,CA0B3F"}

package/dist/tensors.js

@@ -0,0 +1,49 @@
+import { ConfigurationError } from "@zhivex-ai/core/provider";
+const types = new Set(["DATA_TYPE_UNSPECIFIED", "BOOL", "STRING", "FLOAT", "DOUBLE", "INT8", "INT16", "INT32", "INT64", "UINT8", "UINT16", "UINT32", "UINT64"]);
+const integer64 = (value, unsigned = false) => {
+    if (typeof value !== "string" || value.length > 20 || !/^-?\d+$/.test(value))
+        return false;
+    const number = BigInt(value);
+    return unsigned ? number >= 0n && number <= 18446744073709551615n : number >= -9223372036854775808n && number <= 9223372036854775807n;
+};
+const protoFloat = (v) => typeof v === "number" ? Number.isFinite(v) : ["NaN", "Infinity", "-Infinity"].includes(v);
+const base64 = (v) => typeof v === "string" && /^(?:[A-Za-z0-9+/_-]{4})*(?:[A-Za-z0-9+/_-]{2}(?:==)?|[A-Za-z0-9+/_-]{3}=?|)?$/.test(v);
+export function assertVertexTensor(value, depth = 0) {
+    const invalid = () => { throw new ConfigurationError("Invalid Vertex REST Tensor representation."); };
+    if (depth > 32 || !value || typeof value !== "object" || Array.isArray(value))
+        invalid();
+    const tensor = value;
+    if (tensor.dtype !== undefined && !types.has(tensor.dtype))
+        invalid();
+    const arrays = {
+        shape: v => integer64(v), boolVal: v => typeof v === "boolean", stringVal: v => typeof v === "string", bytesVal: base64,
+        floatVal: protoFloat, doubleVal: protoFloat,
+        intVal: v => typeof v === "number" && Number.isInteger(v) && v >= -2147483648 && v <= 2147483647,
+        uintVal: v => typeof v === "number" && Number.isInteger(v) && v >= 0 && v <= 4294967295,
+        int64Val: v => integer64(v), uint64Val: v => integer64(v, true)
+    };
+    for (const [key, valid] of Object.entries(arrays))
+        if (tensor[key] !== undefined && (!Array.isArray(tensor[key]) || !tensor[key].every(valid)))
+            invalid();
+    if (tensor.tensorVal !== undefined && !base64(tensor.tensorVal))
+        invalid();
+    if (tensor.listVal !== undefined) {
+        if (!Array.isArray(tensor.listVal))
+            invalid();
+        for (const child of tensor.listVal)
+            assertVertexTensor(child, depth + 1);
+    }
+    if (tensor.structVal !== undefined) {
+        if (!tensor.structVal || typeof tensor.structVal !== "object" || Array.isArray(tensor.structVal))
+            invalid();
+        for (const child of Object.values(tensor.structVal))
+            assertVertexTensor(child, depth + 1);
+    }
+    const scalarFields = Object.keys(arrays).filter(key => key !== "shape" && tensor[key] !== undefined);
+    if (scalarFields.length > 1)
+        invalid();
+    const allowed = { BOOL: ["boolVal"], STRING: ["stringVal", "bytesVal"], FLOAT: ["floatVal"], DOUBLE: ["doubleVal"], INT8: ["intVal"], INT16: ["intVal"], INT32: ["intVal"], INT64: ["int64Val"], UINT8: ["uintVal"], UINT16: ["uintVal"], UINT32: ["uintVal"], UINT64: ["uint64Val"] };
+    if (scalarFields.length && tensor.dtype !== undefined && !allowed[tensor.dtype]?.includes(scalarFields[0]))
+        invalid();
+}
+//# sourceMappingURL=tensors.js.map

package/dist/tensors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tensors.js","sourceRoot":"","sources":["../src/tensors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAqB9D,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,uBAAuB,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;AAChK,MAAM,SAAS,GAAG,CAAC,KAAc,EAAE,QAAQ,GAAG,KAAK,EAAE,EAAE;IACrD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC3F,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,OAAO,QAAQ,CAAC,CAAC,CAAC,MAAM,IAAI,EAAE,IAAI,MAAM,IAAI,qBAAqB,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,oBAAoB,IAAI,MAAM,IAAI,oBAAoB,CAAC;AACxI,CAAC,CAAC;AACF,MAAM,UAAU,GAAG,CAAC,CAAU,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,CAAW,CAAC,CAAC;AACvI,MAAM,MAAM,GAAG,CAAC,CAAU,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,+EAA+E,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAChJ,MAAM,UAAU,kBAAkB,CAAC,KAAc,EAAE,KAAK,GAAG,CAAC;IAC1D,MAAM,OAAO,GAAG,GAAG,EAAE,GAAG,MAAM,IAAI,kBAAkB,CAAC,4CAA4C,CAAC,CAAC,CAAC,CAAC,CAAC;IACtG,IAAI,KAAK,GAAG,EAAE,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IACzF,MAAM,MAAM,GAAG,KAAgC,CAAC;IAChD,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,KAAe,CAAC;QAAE,OAAO,EAAE,CAAC;IAChF,MAAM,MAAM,GAA4C;QACtD,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,EAAE,QAAQ,EAAE,MAAM;QACvH,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,UAAU;QAC3C,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,UAAU;QAChG,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,UAAU;QACvF,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,CAAC;KAChE,CAAC;IACF,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QAAE,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAE,MAAM,CAAC,GAAG,CAAe,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAAE,OAAO,EAAE,CAAC;IACzK,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC;QAAE,OAAO,EAAE,CAAC;IAC3E,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACjC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;YAAE,OAAO,EAAE,CAAC;QAC9C,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAoB;YAAE,kBAAkB,CAAC,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IACxF,CAAC;IACD,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QACnC,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;YAAE,OAAO,EAAE,CAAC;QAC5G,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,SAAmB,CAAC;YAAE,kBAAkB,CAAC,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IACtG,CAAC;IACD,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC,CAAC;IACrG,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IACvC,MAAM,OAAO,GAA6B,EAAE,IAAI,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,WAAW,EAAE,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC;IACjT,IAAI,YAAY,CAAC,MAAM,IAAI,MAAM,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAe,CAAC,EAAE,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAAE,OAAO,EAAE,CAAC;AAClI,CAAC"}

package/dist/token-counting.d.ts

@@ -0,0 +1,37 @@
+import { type JsonValue, type RetryOptions } from "@zhivex-ai/core/provider";
+/** Native Claude message blocks, including tools, images and documents. */
+export interface VertexClaudeTokenCountInput extends RetryOptions {
+    modelId: string;
+    messages: Array<{
+        role: "user" | "assistant";
+        content: string | Array<Record<string, JsonValue>>;
+    }>;
+    system?: string | Array<Record<string, JsonValue>>;
+    tools?: Array<Record<string, JsonValue>>;
+    thinking?: Record<string, JsonValue>;
+    toolChoice?: Record<string, JsonValue>;
+}
+export interface VertexClaudeTokenCountResult {
+    inputTokens: number;
+    rawResponse: unknown;
+}
+export interface VertexClaudeClient {
+    countTokens(input: VertexClaudeTokenCountInput): Promise<VertexClaudeTokenCountResult>;
+}
+export declare const createVertexClaudeClient: (baseURL: string, fetcher: typeof globalThis.fetch, assertAccess: () => void) => VertexClaudeClient;
+export interface VertexGeminiTokenCountInput extends RetryOptions {
+    modelId: string;
+    messages: import("@zhivex-ai/core/provider").ModelMessage[];
+    system?: string;
+    tools?: import("@zhivex-ai/core/provider").ToolCollection;
+    generationConfig?: Record<string, JsonValue>;
+}
+export interface VertexGeminiTokenCountResult {
+    inputTokens: number;
+    totalBillableCharacters?: number;
+    rawResponse: unknown;
+}
+export interface VertexGeminiClient {
+    countTokens(input: VertexGeminiTokenCountInput): Promise<VertexGeminiTokenCountResult>;
+}
+//# sourceMappingURL=token-counting.d.ts.map

package/dist/token-counting.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-counting.d.ts","sourceRoot":"","sources":["../src/token-counting.ts"],"names":[],"mappings":"AAAA,OAAO,EAEyB,KAAK,SAAS,EAAE,KAAK,YAAY,EAChE,MAAM,0BAA0B,CAAC;AAElC,2EAA2E;AAC3E,MAAM,WAAW,2BAA4B,SAAQ,YAAY;IAC/D,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;QAAC,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAA;KAAE,CAAC,CAAC;IACpG,MAAM,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC;IACnD,KAAK,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC;IACzC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACrC,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;CACxC;AACD,MAAM,WAAW,4BAA4B;IAC3C,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC;CACtB;AACD,MAAM,WAAW,kBAAkB;IACjC,WAAW,CAAC,KAAK,EAAE,2BAA2B,GAAG,OAAO,CAAC,4BAA4B,CAAC,CAAC;CACxF;AAED,eAAO,MAAM,wBAAwB,YAAa,MAAM,WAAW,OAAO,UAAU,CAAC,KAAK,gBAAgB,MAAM,IAAI,KAAG,kBAsBrH,CAAC;AAEH,MAAM,WAAW,2BAA4B,SAAQ,YAAY;IAC/D,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,OAAO,0BAA0B,EAAE,YAAY,EAAE,CAAC;IAC5D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,OAAO,0BAA0B,EAAE,cAAc,CAAC;IAC1D,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;CAC9C;AACD,MAAM,WAAW,4BAA4B;IAC3C,WAAW,EAAE,MAAM,CAAC;IACpB,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,WAAW,EAAE,OAAO,CAAC;CACtB;AACD,MAAM,WAAW,kBAAkB;IACjC,WAAW,CAAC,KAAK,EAAE,2BAA2B,GAAG,OAAO,CAAC,4BAA4B,CAAC,CAAC;CACxF"}

package/dist/token-counting.js

@@ -0,0 +1,33 @@
+import { ConfigurationError, ProviderHTTPError, readErrorBodyWithLimit, readJsonWithLimit, withRetry, withTimeoutSignal } from "@zhivex-ai/core/provider";
+export const createVertexClaudeClient = (baseURL, fetcher, assertAccess) => ({
+    async countTokens(input) {
+        assertAccess();
+        if (!/^claude-[a-z0-9@.-]+$/.test(input.modelId))
+            throw new ConfigurationError("Vertex Claude token counting requires a Claude model ID.");
+        if (!Array.isArray(input.messages) || !input.messages.length || input.messages.some((message) => !message || !["user", "assistant"].includes(message.role) || (typeof message.content !== "string" && !Array.isArray(message.content))))
+            throw new ConfigurationError("Vertex Claude token counting requires native Claude messages.");
+        const body = JSON.stringify({ model: input.modelId, messages: input.messages, system: input.system,
+            tools: input.tools, thinking: input.thinking, tool_choice: input.toolChoice });
+        if (new TextEncoder().encode(body).byteLength > 30_000_000)
+            throw new ConfigurationError("Vertex Claude token count request exceeds 30 MB.");
+        const { signal, cleanup } = withTimeoutSignal(input);
+        try {
+            const response = await withRetry(async () => {
+                const result = await fetcher(`${baseURL}/publishers/anthropic/models/count-tokens:rawPredict`, {
+                    method: "POST", headers: { "content-type": "application/json" }, body, signal, redirect: "error"
+                });
+                if (!result.ok)
+                    throw new ProviderHTTPError(`Vertex Claude token counting failed with status ${result.status}.`, result.status, { responseBody: await readErrorBodyWithLimit(result) });
+                return result;
+            }, { ...input, abortSignal: signal });
+            const result = await readJsonWithLimit(response, { maxBytes: 1024 * 1024 });
+            if (!Number.isInteger(result?.input_tokens) || result.input_tokens < 0)
+                throw new ConfigurationError("Vertex Claude returned an invalid token count.");
+            return { inputTokens: result.input_tokens, rawResponse: result };
+        }
+        finally {
+            cleanup();
+        }
+    }
+});
+//# sourceMappingURL=token-counting.js.map

package/dist/token-counting.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-counting.js","sourceRoot":"","sources":["../src/token-counting.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,kBAAkB,EAAE,iBAAiB,EAAE,sBAAsB,EAAE,iBAAiB,EAChF,SAAS,EAAE,iBAAiB,EAC7B,MAAM,0BAA0B,CAAC;AAmBlC,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,OAAe,EAAE,OAAgC,EAAE,YAAwB,EAAsB,EAAE,CAAC,CAAC;IAC5I,KAAK,CAAC,WAAW,CAAC,KAAK;QACrB,YAAY,EAAE,CAAC;QACf,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,0DAA0D,CAAC,CAAC;QAC3I,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,+DAA+D,CAAC,CAAC;QACvU,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM;YAChG,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC;QACjF,IAAI,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,UAAU,GAAG,UAAU;YAAE,MAAM,IAAI,kBAAkB,CAAC,kDAAkD,CAAC,CAAC;QAC7I,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACrD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,KAAK,IAAI,EAAE;gBAC1C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,GAAG,OAAO,sDAAsD,EAAE;oBAC7F,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO;iBACjG,CAAC,CAAC;gBACH,IAAI,CAAC,MAAM,CAAC,EAAE;oBAAE,MAAM,IAAI,iBAAiB,CAAC,mDAAmD,MAAM,CAAC,MAAM,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACxL,OAAO,MAAM,CAAC;YAChB,CAAC,EAAE,EAAE,GAAG,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;YACtC,MAAM,MAAM,GAAQ,MAAM,iBAAiB,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,IAAI,GAAG,IAAI,EAAE,CAAC,CAAC;YACjF,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC,YAAY,GAAG,CAAC;gBAAE,MAAM,IAAI,kBAAkB,CAAC,gDAAgD,CAAC,CAAC;YACvJ,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,YAAY,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC;QACnE,CAAC;gBAAS,CAAC;YAAC,OAAO,EAAE,CAAC;QAAC,CAAC;IAC1B,CAAC;CACF,CAAC,CAAC"}

package/dist/transcription.d.ts

@@ -0,0 +1,60 @@
+import { type TranscriptionModel, type TranscriptionModelInput, type TranscriptionResult } from "@zhivex-ai/core/provider";
+export interface VertexAudioTranscriptionConfig {
+    languageCodes?: string[];
+    customVocabulary?: string[];
+    wordTimestamp?: boolean;
+    diarization?: boolean;
+    mode?: "VERBATIM" | "SMART";
+}
+export interface VertexTranscriptionOptions extends Record<string, unknown> {
+    audioTranscriptionConfig?: VertexAudioTranscriptionConfig;
+}
+export interface VertexAudioTranscription {
+    text?: string;
+    finished?: boolean;
+    languageCode?: string;
+    speakerLabel?: string;
+    words?: {
+        word?: string;
+        startOffset?: string;
+        endOffset?: string;
+    }[];
+}
+export interface VertexTranscriptionResult extends TranscriptionResult {
+    transcriptions: VertexAudioTranscription[];
+}
+export interface VertexTranscriptionModel extends TranscriptionModel<VertexTranscriptionOptions> {
+    transcribe(input: TranscriptionModelInput<VertexTranscriptionOptions>): Promise<VertexTranscriptionResult>;
+}
+export declare const isVertexTranscribeModel: (model: string) => model is "gemini-3.5-transcribe-preview";
+export declare function normalizeVertexTranscriptionConfig(supplied: unknown, language?: string): VertexAudioTranscriptionConfig;
+export declare function transcriptionRequest(model: string, input: TranscriptionModelInput, data: string): {
+    contents: {
+        role: string;
+        parts: ({
+            text?: undefined;
+            inlineData: {
+                mimeType: string;
+                data: string;
+            };
+        } | {
+            inlineData?: undefined;
+            text: string;
+        })[];
+    }[];
+} | {
+    contents: {
+        role: string;
+        parts: {
+            inlineData: {
+                mimeType: string;
+                data: string;
+            };
+        }[];
+    }[];
+    generationConfig: {
+        audioTranscriptionConfig: VertexAudioTranscriptionConfig;
+    };
+};
+export declare function transcriptionResponse(json: any): VertexTranscriptionResult;
+//# sourceMappingURL=transcription.d.ts.map

package/dist/transcription.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"transcription.d.ts","sourceRoot":"","sources":["../src/transcription.ts"],"names":[],"mappings":"AAAA,OAAO,EAA+C,KAAK,kBAAkB,EAAE,KAAK,uBAAuB,EAAE,KAAK,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAExK,MAAM,WAAW,8BAA8B;IAC7C,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,IAAI,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC;CAC7B;AACD,MAAM,WAAW,0BAA2B,SAAQ,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IACzE,wBAAwB,CAAC,EAAE,8BAA8B,CAAC;CAC3D;AACD,MAAM,WAAW,wBAAwB;IACvC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;CACvE;AACD,MAAM,WAAW,yBAA0B,SAAQ,mBAAmB;IACpE,cAAc,EAAE,wBAAwB,EAAE,CAAC;CAC5C;AACD,MAAM,WAAW,wBAAyB,SAAQ,kBAAkB,CAAC,0BAA0B,CAAC;IAC9F,UAAU,CAAC,KAAK,EAAE,uBAAuB,CAAC,0BAA0B,CAAC,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAAC;CAC5G;AACD,eAAO,MAAM,uBAAuB,UAAW,MAAM,6CAA8C,CAAC;AAEpG,wBAAgB,kCAAkC,CAAC,QAAQ,EAAE,OAAO,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,8BAA8B,CAmBvH;AAED,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,uBAAuB,EAAE,IAAI,EAAE,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;EAiB/F;AAED,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,GAAG,GAAG,yBAAyB,CAqB1E"}

package/dist/transcription.js

@@ -0,0 +1,88 @@
+import { ConfigurationError, UnsupportedFeatureError } from "@zhivex-ai/core/provider";
+export const isVertexTranscribeModel = (model) => model === "gemini-3.5-transcribe-preview";
+export function normalizeVertexTranscriptionConfig(supplied, language) {
+    if (supplied !== undefined && (!supplied || typeof supplied !== "object" || Array.isArray(supplied)))
+        throw new ConfigurationError("audioTranscriptionConfig must be an object.");
+    const config = { ...supplied };
+    const allowed = ["languageCodes", "customVocabulary", "wordTimestamp", "diarization", "mode"];
+    for (const key of Object.keys(config))
+        if (!allowed.includes(key))
+            throw new ConfigurationError(`Unsupported audioTranscriptionConfig field: ${key}.`);
+    if (language !== undefined) {
+        if (typeof language !== "string" || !language.trim())
+            throw new ConfigurationError("Transcription language must be nonempty.");
+        if (config.languageCodes !== undefined && (config.languageCodes.length !== 1 || config.languageCodes[0] !== language))
+            throw new ConfigurationError("language conflicts with audioTranscriptionConfig.languageCodes.");
+        config.languageCodes = [language];
+    }
+    for (const key of ["languageCodes", "customVocabulary"]) {
+        const list = config[key];
+        if (list !== undefined && (!Array.isArray(list) || list.some(value => typeof value !== "string" || !value.trim())))
+            throw new ConfigurationError(`${key} must contain nonempty strings.`);
+    }
+    if ((config.customVocabulary?.length ?? 0) > 1000)
+        throw new ConfigurationError("customVocabulary accepts at most 1000 terms.");
+    for (const key of ["wordTimestamp", "diarization"])
+        if (config[key] !== undefined && typeof config[key] !== "boolean")
+            throw new ConfigurationError(`${key} must be a boolean.`);
+    if (config.mode !== undefined && !["SMART", "VERBATIM"].includes(config.mode))
+        throw new ConfigurationError("Transcription mode must be SMART or VERBATIM.");
+    if (config.mode === "SMART" && (config.wordTimestamp || config.diarization))
+        throw new ConfigurationError("SMART mode cannot combine with timestamps or diarization.");
+    return config;
+}
+export function transcriptionRequest(model, input, data) {
+    const options = { ...input.providerOptions };
+    const native = isVertexTranscribeModel(model);
+    if (!native) {
+        if (options.audioTranscriptionConfig !== undefined)
+            throw new UnsupportedFeatureError("audioTranscriptionConfig requires Gemini 3.5 Transcribe.");
+        return { contents: [{ role: "user", parts: [{ inlineData: { mimeType: input.audio.mediaType, data } }, { text: input.prompt ?? `Transcribe this audio${input.language ? ` in ${input.language}` : ""}. Return only the transcript.` }] }], ...options };
+    }
+    if (input.prompt !== undefined)
+        throw new UnsupportedFeatureError("Gemini 3.5 Transcribe accepts audio and transcription configuration, not a text prompt.");
+    for (const key of ["contents", "systemInstruction", "system_instruction", "tools", "toolConfig", "audio_transcription_config"]) {
+        if (options[key] !== undefined)
+            throw new ConfigurationError(`Gemini 3.5 Transcribe does not accept providerOptions.${key}.`);
+    }
+    const config = normalizeVertexTranscriptionConfig(options.audioTranscriptionConfig, input.language);
+    delete options.audioTranscriptionConfig;
+    const generation = options.generationConfig;
+    if (generation !== undefined && (!generation || typeof generation !== "object" || Array.isArray(generation)))
+        throw new ConfigurationError("generationConfig must be an object.");
+    if (generation && ("audioTranscriptionConfig" in generation || "audio_transcription_config" in generation))
+        throw new ConfigurationError("Use the dedicated audioTranscriptionConfig option, not a generationConfig override.");
+    return { ...options, contents: [{ role: "user", parts: [{ inlineData: { mimeType: input.audio.mediaType, data } }] }], generationConfig: { ...generation, audioTranscriptionConfig: config } };
+}
+export function transcriptionResponse(json) {
+    const parts = json.candidates?.[0]?.content?.parts ?? [];
+    if (!Array.isArray(parts))
+        throw new ConfigurationError("Invalid Vertex transcription parts.");
+    const transcriptions = [];
+    const texts = [];
+    for (const part of parts) {
+        if (!part || typeof part !== "object")
+            throw new ConfigurationError("Invalid Vertex transcription part.");
+        const value = part.audioTranscription;
+        if (value !== undefined) {
+            if (!value || typeof value !== "object" || Array.isArray(value))
+                throw new ConfigurationError("Invalid audioTranscription response.");
+            for (const key of ["text", "languageCode", "speakerLabel"])
+                if (value[key] !== undefined && typeof value[key] !== "string")
+                    throw new ConfigurationError(`Invalid transcription ${key}.`);
+            if (value.finished !== undefined && typeof value.finished !== "boolean")
+                throw new ConfigurationError("Invalid transcription finished flag.");
+            if (value.words !== undefined && (!Array.isArray(value.words) || value.words.some((word) => !word || typeof word !== "object" || Array.isArray(word) || ["word", "startOffset", "endOffset"].some(key => word[key] !== undefined && typeof word[key] !== "string"))))
+                throw new ConfigurationError("Invalid transcription word information.");
+            transcriptions.push(value);
+        }
+        if (part.thought !== true) {
+            if (typeof part.text === "string" && part.text)
+                texts.push(part.text);
+            else if (typeof value?.text === "string")
+                texts.push(value.text);
+        }
+    }
+    return { text: texts.join(""), transcriptions, rawResponse: json };
+}
+//# sourceMappingURL=transcription.js.map

package/dist/transcription.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transcription.js","sourceRoot":"","sources":["../src/transcription.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAmF,MAAM,0BAA0B,CAAC;AAyBxK,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,KAAK,+BAA+B,CAAC;AAEpG,MAAM,UAAU,kCAAkC,CAAC,QAAiB,EAAE,QAAiB;IACrF,IAAI,QAAQ,KAAK,SAAS,IAAI,CAAC,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAAE,MAAM,IAAI,kBAAkB,CAAC,6CAA6C,CAAC,CAAC;IAClL,MAAM,MAAM,GAAG,EAAE,GAAI,QAAuD,EAAE,CAAC;IAC/E,MAAM,OAAO,GAAG,CAAC,eAAe,EAAE,kBAAkB,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;IAC9F,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;QAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,+CAA+C,GAAG,GAAG,CAAC,CAAC;IACvJ,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE;YAAE,MAAM,IAAI,kBAAkB,CAAC,0CAA0C,CAAC,CAAC;QAC/H,IAAI,MAAM,CAAC,aAAa,KAAK,SAAS,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,KAAK,CAAC,IAAI,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,iEAAiE,CAAC,CAAC;QACvN,MAAM,CAAC,aAAa,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IACD,KAAK,MAAM,GAAG,IAAI,CAAC,eAAe,EAAE,kBAAkB,CAAU,EAAE,CAAC;QACjE,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QACzB,IAAI,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,GAAG,GAAG,iCAAiC,CAAC,CAAC;IAC5L,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,IAAI;QAAE,MAAM,IAAI,kBAAkB,CAAC,8CAA8C,CAAC,CAAC;IAChI,KAAK,MAAM,GAAG,IAAI,CAAC,eAAe,EAAE,aAAa,CAAU;QAAE,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS,IAAI,OAAO,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS;YAAE,MAAM,IAAI,kBAAkB,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;IAC1L,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC;QAAE,MAAM,IAAI,kBAAkB,CAAC,+CAA+C,CAAC,CAAC;IAC7J,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,WAAW,CAAC;QAAE,MAAM,IAAI,kBAAkB,CAAC,2DAA2D,CAAC,CAAC;IACvK,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,KAAa,EAAE,KAA8B,EAAE,IAAY;IAC9F,MAAM,OAAO,GAAG,EAAE,GAAG,KAAK,CAAC,eAAe,EAAE,CAAC;IAC7C,MAAM,MAAM,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;IAC9C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,IAAI,OAAO,CAAC,wBAAwB,KAAK,SAAS;YAAE,MAAM,IAAI,uBAAuB,CAAC,0DAA0D,CAAC,CAAC;QAClJ,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,IAAI,wBAAwB,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,+BAA+B,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,OAAO,EAAE,CAAC;IAC1P,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS;QAAE,MAAM,IAAI,uBAAuB,CAAC,yFAAyF,CAAC,CAAC;IAC7J,KAAK,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,OAAO,EAAE,YAAY,EAAE,4BAA4B,CAAC,EAAE,CAAC;QAC/H,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,SAAS;YAAE,MAAM,IAAI,kBAAkB,CAAC,yDAAyD,GAAG,GAAG,CAAC,CAAC;IAChI,CAAC;IACD,MAAM,MAAM,GAAG,kCAAkC,CAAC,OAAO,CAAC,wBAAwB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;IACpG,OAAO,OAAO,CAAC,wBAAwB,CAAC;IACxC,MAAM,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAC5C,IAAI,UAAU,KAAK,SAAS,IAAI,CAAC,CAAC,UAAU,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAAE,MAAM,IAAI,kBAAkB,CAAC,qCAAqC,CAAC,CAAC;IAClL,IAAI,UAAU,IAAI,CAAC,0BAA0B,IAAI,UAAU,IAAI,4BAA4B,IAAI,UAAU,CAAC;QAAE,MAAM,IAAI,kBAAkB,CAAC,qFAAqF,CAAC,CAAC;IAChO,OAAO,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,gBAAgB,EAAE,EAAE,GAAG,UAAoB,EAAE,wBAAwB,EAAE,MAAM,EAAE,EAAE,CAAC;AAC3M,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,IAAS;IAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;IACzD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,MAAM,IAAI,kBAAkB,CAAC,qCAAqC,CAAC,CAAC;IAC/F,MAAM,cAAc,GAA+B,EAAE,CAAC;IACtD,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,MAAM,IAAI,kBAAkB,CAAC,oCAAoC,CAAC,CAAC;QAC1G,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC;QACtC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;gBAAE,MAAM,IAAI,kBAAkB,CAAC,sCAAsC,CAAC,CAAC;YACtI,KAAK,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,cAAc,EAAE,cAAc,CAAC;gBAAE,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,SAAS,IAAI,OAAO,KAAK,CAAC,GAAG,CAAC,KAAK,QAAQ;oBAAE,MAAM,IAAI,kBAAkB,CAAC,yBAAyB,GAAG,GAAG,CAAC,CAAC;YAC1L,IAAI,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,OAAO,KAAK,CAAC,QAAQ,KAAK,SAAS;gBAAE,MAAM,IAAI,kBAAkB,CAAC,sCAAsC,CAAC,CAAC;YAC9I,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,SAAS,IAAI,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC;gBAAE,MAAM,IAAI,kBAAkB,CAAC,yCAAyC,CAAC,CAAC;YACnV,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;YAC1B,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBACjE,IAAI,OAAO,KAAK,EAAE,IAAI,KAAK,QAAQ;gBAAE,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;AACrE,CAAC"}

package/dist/virtual-try-on.d.ts

@@ -0,0 +1,23 @@
+import { type ImageGenerationResult, type MediaInput, type PredictionModel, type RetryOptions, type JsonValue } from "@zhivex-ai/core/provider";
+export interface VertexVirtualTryOnInput extends RetryOptions {
+    personImage: MediaInput;
+    productImage: MediaInput;
+    productMask?: MediaInput;
+    productImageConfig?: Record<string, JsonValue>;
+    prompt?: string;
+    count?: number;
+    outputMimeType?: "image/png" | "image/jpeg";
+    outputStorageUri?: string;
+    /** Native VirtualTryOnModelParams, e.g. baseSteps, seed, addWatermark, outputOptions. */
+    providerOptions?: Record<string, JsonValue>;
+}
+export interface VertexVirtualTryOnResult extends ImageGenerationResult {
+    filtered: Array<{
+        reason: string;
+    }>;
+}
+export interface VertexVirtualTryOnClient {
+    generate(input: VertexVirtualTryOnInput): Promise<VertexVirtualTryOnResult>;
+}
+export declare const createVertexVirtualTryOnClient: (model: PredictionModel, assertAccess: () => void, sanitize: (value: unknown) => unknown) => VertexVirtualTryOnClient;
+//# sourceMappingURL=virtual-try-on.d.ts.map

package/dist/virtual-try-on.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"virtual-try-on.d.ts","sourceRoot":"","sources":["../src/virtual-try-on.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,qBAAqB,EAAE,KAAK,UAAU,EAAE,KAAK,eAAe,EAAE,KAAK,YAAY,EAAE,KAAK,SAAS,EACrG,MAAM,0BAA0B,CAAC;AAElC,MAAM,WAAW,uBAAwB,SAAQ,YAAY;IAC3D,WAAW,EAAE,UAAU,CAAC;IACxB,YAAY,EAAE,UAAU,CAAC;IACzB,WAAW,CAAC,EAAE,UAAU,CAAC;IACzB,kBAAkB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC/C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,WAAW,GAAG,YAAY,CAAC;IAC5C,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,yFAAyF;IACzF,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;CAC7C;AACD,MAAM,WAAW,wBAAyB,SAAQ,qBAAqB;IACrE,QAAQ,EAAE,KAAK,CAAC;QAAE,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACrC;AACD,MAAM,WAAW,wBAAwB;IACvC,QAAQ,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC,CAAC;CAC7E;AAeD,eAAO,MAAM,8BAA8B,UAClC,eAAe,gBAAgB,MAAM,IAAI,YAAY,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,KACtF,wBA6CD,CAAC"}

package/dist/virtual-try-on.js

@@ -0,0 +1,80 @@
+import { ConfigurationError, decodeBase64WithLimit, encodeMediaFrame } from "@zhivex-ai/core/provider";
+const gcs = (uri) => /^gs:\/\/[^/\s?#]+\/[^\s?#]+$/.test(uri);
+const media = (image) => {
+    if (!image || !["image/png", "image/jpeg"].includes(image.mediaType) ||
+        (image.data === undefined) === (image.uri === undefined))
+        throw new ConfigurationError("Virtual Try-On requires a PNG/JPEG image with exactly one data or GCS uri.");
+    if (image.uri !== undefined) {
+        if (!gcs(image.uri))
+            throw new ConfigurationError("Virtual Try-On image URI must be a gs:// object path.");
+        return { mimeType: image.mediaType, gcsUri: image.uri };
+    }
+    const bytes = encodeMediaFrame({ data: image.data, mediaType: image.mediaType });
+    if (!bytes || !/^[A-Za-z0-9+/]+={0,2}$/.test(bytes))
+        throw new ConfigurationError("Virtual Try-On image data must be nonempty base64 or bytes.");
+    decodeBase64WithLimit(bytes, { maxBytes: 7 * 1024 * 1024, provider: "vertex", endpoint: "virtual-try-on input" });
+    return { mimeType: image.mediaType, bytesBase64Encoded: bytes };
+};
+export const createVertexVirtualTryOnClient = (model, assertAccess, sanitize) => ({
+    async generate(input) {
+        assertAccess();
+        const count = input.count ?? 1;
+        if (!Number.isInteger(count) || count < 1 || count > 4)
+            throw new ConfigurationError("Virtual Try-On count must be 1-4.");
+        if (input.prompt !== undefined && typeof input.prompt !== "string")
+            throw new ConfigurationError("Virtual Try-On prompt must be text.");
+        if (input.outputMimeType !== undefined && !["image/png", "image/jpeg"].includes(input.outputMimeType))
+            throw new ConfigurationError("Virtual Try-On output format must be PNG or JPEG.");
+        if (input.outputStorageUri !== undefined && !/^gs:\/\/[^/\s?#]+(?:\/[^\s?#]*)?$/.test(input.outputStorageUri))
+            throw new ConfigurationError("Virtual Try-On outputStorageUri must be a gs:// path.");
+        const parameters = { ...input.providerOptions };
+        for (const key of ["sampleCount", "storageUri", "instances", "parameters", "action", "model"]) {
+            if (parameters[key] !== undefined)
+                throw new ConfigurationError(`Virtual Try-On providerOptions.${key} conflicts with the dedicated contract.`);
+        }
+        const output = parameters.outputOptions;
+        if (output !== undefined && (!output || typeof output !== "object" || Array.isArray(output)))
+            throw new ConfigurationError("Virtual Try-On outputOptions must be an object.");
+        if (input.outputMimeType && output && output.mimeType !== undefined && output.mimeType !== input.outputMimeType)
+            throw new ConfigurationError("Conflicting Virtual Try-On output MIME types.");
+        if (parameters.seed !== undefined && parameters.addWatermark !== false)
+            throw new ConfigurationError("Virtual Try-On seed requires addWatermark:false.");
+        const result = await model.predictRaw({
+            abortSignal: input.abortSignal, timeoutMs: input.timeoutMs, maxRetries: input.maxRetries, retryBackoffMs: input.retryBackoffMs,
+            instances: [{ ...(input.prompt !== undefined ? { prompt: input.prompt } : {}),
+                    personImage: { image: media(input.personImage) }, productImages: [{ image: media(input.productImage),
+                            ...(input.productMask ? { maskImage: media(input.productMask) } : {}),
+                            ...(input.productImageConfig ? { productImageConfig: input.productImageConfig } : {}) }] }],
+            parameters: { ...parameters, sampleCount: count,
+                ...(input.outputMimeType ? { outputOptions: { ...output, mimeType: input.outputMimeType } } : {}),
+                ...(input.outputStorageUri ? { storageUri: input.outputStorageUri } : {}) }
+        });
+        if (!Array.isArray(result.predictions) || !result.predictions.length)
+            throw new ConfigurationError("Virtual Try-On returned no predictions.");
+        const images = [], filtered = [];
+        // Both the guide's flat predictions and the REST result schema's images wrapper.
+        const predictions = result.predictions.flatMap((p) => Array.isArray(p?.images) ? p.images : [p]);
+        if (!predictions.length || predictions.length > count)
+            throw new ConfigurationError("Virtual Try-On returned an invalid image count.");
+        for (const prediction of predictions) {
+            if (!prediction || [prediction.bytesBase64Encoded, prediction.gcsUri, prediction.raiFilteredReason].filter(v => v !== undefined).length !== 1)
+                throw new ConfigurationError("Invalid Virtual Try-On image response.");
+            if (prediction.raiFilteredReason !== undefined) {
+                if (typeof prediction.raiFilteredReason !== "string" || !prediction.raiFilteredReason)
+                    throw new ConfigurationError("Invalid Virtual Try-On filtering reason.");
+                filtered.push({ reason: prediction.raiFilteredReason });
+                continue;
+            }
+            if (!["image/png", "image/jpeg"].includes(prediction.mimeType))
+                throw new ConfigurationError("Invalid Virtual Try-On output MIME type.");
+            if (prediction.gcsUri !== undefined && (typeof prediction.gcsUri !== "string" || !gcs(prediction.gcsUri)))
+                throw new ConfigurationError("Invalid Virtual Try-On output URI.");
+            if (prediction.bytesBase64Encoded !== undefined && (typeof prediction.bytesBase64Encoded !== "string" || !/^[A-Za-z0-9+/]+={0,2}$/.test(prediction.bytesBase64Encoded)))
+                throw new ConfigurationError("Invalid Virtual Try-On output bytes.");
+            images.push({ mediaType: prediction.mimeType,
+                ...(prediction.gcsUri ? { uri: prediction.gcsUri } : { data: decodeBase64WithLimit(prediction.bytesBase64Encoded, { maxBytes: 32 * 1024 * 1024, provider: "vertex", endpoint: "virtual-try-on output" }) }) });
+        }
+        return { images, filtered, rawResponse: sanitize(result.rawResponse) };
+    }
+});
+//# sourceMappingURL=virtual-try-on.js.map