@zhivex-ai/vertex 1.0.2 → 1.1.0

package/dist/grpc.js

@@ -0,0 +1,161 @@
+import { once } from "node:events";
+import { setImmediate as yieldToIO } from "node:timers/promises";
+import { GoogleAuth, OAuth2Client } from "google-auth-library";
+import { ConfigurationError, ProviderResponseTooLargeError, withTimeoutSignal } from "@zhivex-ai/core/provider";
+import { assertVertexTensor } from "./tensors.js";
+const abortable = async (promise, signal) => {
+    signal.throwIfAborted();
+    let cancel;
+    const abort = new Promise((_, reject) => { cancel = () => reject(signal.reason); signal.addEventListener("abort", cancel, { once: true }); });
+    try {
+        return await Promise.race([promise, abort]);
+    }
+    finally {
+        signal.removeEventListener("abort", cancel);
+    }
+};
+export const createVertexGrpcClient = (baseURL, resolveEndpoint, getToken) => {
+    async function* run(input, method) {
+        const raw = method === "streamDirectRawPredict" || method === "streamingRawPredict";
+        if (input.maxRetries !== undefined && input.maxRetries !== 0)
+            throw new ConfigurationError("Vertex gRPC streams cannot be replayed; maxRetries must be 0.");
+        const endpoint = resolveEndpoint(input.endpoint, method === "serverStreamingPredict");
+        const maxBytes = input.maxResponseBytes ?? 16 * 1024 * 1024;
+        if (!Number.isSafeInteger(maxBytes) || maxBytes <= 0 || maxBytes > 2147483647)
+            throw new ConfigurationError("Vertex gRPC maxResponseBytes must be a positive int32.");
+        if (raw && !/^\/[^/\s?#]+\/[^/\s?#]+\/?$/.test(input.methodName))
+            throw new ConfigurationError("Vertex gRPC methodName must use /namespace.Service/Method.");
+        if (!input.inputs || (!(Symbol.asyncIterator in Object(input.inputs)) && !(Symbol.iterator in Object(input.inputs))))
+            throw new ConfigurationError("Vertex gRPC inputs must be iterable.");
+        if (method === "serverStreamingPredict") {
+            const request = input;
+            if (!Array.isArray(request.inputs))
+                throw new ConfigurationError("Vertex serverStreamingPredict requires a tensor array.");
+            request.inputs.forEach(value => assertVertexTensor(value));
+            if (request.parameters !== undefined)
+                assertVertexTensor(request.parameters);
+        }
+        const stop = new AbortController();
+        const timing = withTimeoutSignal(input);
+        const signal = timing.signal ? AbortSignal.any([stop.signal, timing.signal]) : stop.signal;
+        let client;
+        let stream;
+        let iterator;
+        let pump;
+        let pumpError;
+        const cancel = () => { stream?.destroy(signal.reason instanceof Error ? signal.reason : new Error("Vertex gRPC stream aborted.")); };
+        try {
+            const token = await getToken(signal);
+            const sdk = await abortable(import("@google-cloud/aiplatform"), signal);
+            const url = new URL(baseURL);
+            const api = url.pathname.startsWith("/v1beta1/") ? sdk.v1beta1 : sdk.v1;
+            const proto = url.pathname.startsWith("/v1beta1/") ? sdk.protos.google.cloud.aiplatform.v1beta1 : sdk.protos.google.cloud.aiplatform.v1;
+            const oauth = new OAuth2Client();
+            oauth.setCredentials({ access_token: token });
+            client = new api.PredictionServiceClient({ apiEndpoint: url.hostname, ...(url.port ? { port: Number(url.port) } : {}), projectId: endpoint.split("/")[1], auth: new GoogleAuth({ projectId: endpoint.split("/")[1], authClient: oauth }), "grpc.max_receive_message_length": maxBytes });
+            const callOptions = { ...(input.timeoutMs !== undefined ? { timeout: input.timeoutMs } : {}), retry: null };
+            const encodeFrame = (frame) => {
+                if (!frame || !Array.isArray(frame.inputs))
+                    throw new ConfigurationError("Vertex gRPC tensor frames require inputs.");
+                frame.inputs.forEach(value => assertVertexTensor(value));
+                if (frame.parameters !== undefined)
+                    assertVertexTensor(frame.parameters);
+                return { inputs: frame.inputs.map(value => proto.Tensor.fromObject(value)), ...(frame.parameters !== undefined ? { parameters: proto.Tensor.fromObject(frame.parameters) } : {}) };
+            };
+            stream = method === "serverStreamingPredict"
+                ? client.serverStreamingPredict({ endpoint, ...encodeFrame(input) }, callOptions)
+                : client[method](callOptions);
+            stream.on("error", () => { }); // Iterator/pump report failures; cancellation must not be unhandled.
+            signal.addEventListener("abort", cancel, { once: true });
+            signal.throwIfAborted();
+            if (method !== "serverStreamingPredict") {
+                const source = input.inputs;
+                iterator = source[Symbol.asyncIterator]?.() ?? source[Symbol.iterator]();
+                const write = async (message) => {
+                    signal.throwIfAborted();
+                    if (!stream.write(message))
+                        await once(stream, "drain", { signal });
+                };
+                pump = (async () => {
+                    await write({ endpoint, ...(raw ? { methodName: input.methodName } : {}) });
+                    let sentFrames = 0;
+                    while (true) {
+                        const next = await abortable(Promise.resolve(iterator.next()), signal);
+                        if (next.done)
+                            break;
+                        if (raw) {
+                            if (!(next.value instanceof Uint8Array))
+                                throw new ConfigurationError("Vertex gRPC raw input must contain Uint8Array frames.");
+                            await write({ input: Buffer.from(next.value) });
+                        }
+                        else {
+                            const frame = next.value;
+                            await write(encodeFrame(frame));
+                        }
+                        // Immediately resolved iterators/writes can otherwise monopolize the
+                        // microtask queue, starving gRPC reads, abort events and deadlines.
+                        if (++sentFrames % 16 === 0)
+                            await yieldToIO(undefined, { signal });
+                    }
+                    stream.end();
+                })().catch(error => { pumpError = error; stream.destroy(error instanceof Error ? error : new Error("Vertex gRPC input failed.")); });
+            }
+            let receivedBytes = 0;
+            for await (const message of stream) {
+                signal.throwIfAborted();
+                let result;
+                if (raw) {
+                    if (message.output !== undefined && !(message.output instanceof Uint8Array))
+                        throw new ConfigurationError("Invalid Vertex gRPC output bytes.");
+                    result = new Uint8Array(message.output ?? []);
+                    receivedBytes += result.byteLength;
+                }
+                else {
+                    const convert = (value) => {
+                        const tensor = proto.Tensor.toObject(proto.Tensor.fromObject(value), { longs: String, bytes: String, enums: String, json: true });
+                        assertVertexTensor(tensor);
+                        return tensor;
+                    };
+                    result = { outputs: (message.outputs ?? []).map(convert), ...(message.parameters ? { parameters: convert(message.parameters) } : {}) };
+                    receivedBytes += Buffer.byteLength(JSON.stringify(result));
+                }
+                if (receivedBytes > maxBytes)
+                    throw new ProviderResponseTooLargeError({ maxBytes, receivedBytes, provider: "vertex", endpoint: method });
+                yield result;
+            }
+            if (pumpError)
+                throw pumpError;
+        }
+        finally {
+            stop.abort(new Error("Vertex gRPC stream closed."));
+            signal.removeEventListener("abort", cancel);
+            try {
+                stream?.cancel();
+            }
+            finally {
+                // A source can throw synchronously from return(), or never resolve it.
+                // Neither may prevent releasing the transport or its deadline timer.
+                void Promise.resolve().then(() => iterator?.return?.()).catch(() => { });
+                try {
+                    await pump;
+                }
+                finally {
+                    try {
+                        await client?.close();
+                    }
+                    finally {
+                        timing.cleanup();
+                    }
+                }
+            }
+        }
+    }
+    return {
+        serverStreamingPredict: input => run(input, "serverStreamingPredict"),
+        streamDirectRawPredict: input => run(input, "streamDirectRawPredict"),
+        streamDirectPredict: input => run(input, "streamDirectPredict"),
+        streamingRawPredict: input => run(input, "streamingRawPredict"),
+        streamingPredict: input => run(input, "streamingPredict")
+    };
+};
+//# sourceMappingURL=grpc.js.map

package/dist/grpc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"grpc.js","sourceRoot":"","sources":["../src/grpc.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,aAAa,CAAC;AACnC,OAAO,EAAE,YAAY,IAAI,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,6BAA6B,EAAE,iBAAiB,EAAqB,MAAM,0BAA0B,CAAC;AACnI,OAAO,EAAE,kBAAkB,EAAqB,MAAM,cAAc,CAAC;AA+BrE,MAAM,SAAS,GAAG,KAAK,EAAK,OAAmB,EAAE,MAAmB,EAAc,EAAE;IAClF,MAAM,CAAC,cAAc,EAAE,CAAC;IACxB,IAAI,MAAmB,CAAC;IACxB,MAAM,KAAK,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,GAAG,MAAM,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrJ,IAAI,CAAC;QAAC,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;IAAC,CAAC;YAC5C,CAAC;QAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAAC,CAAC;AAC1D,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,sBAAsB,GAAG,CACpC,OAAe,EACf,eAAsE,EACtE,QAAkD,EAChC,EAAE;IACpB,KAAK,SAAS,CAAC,CAAC,GAAG,CAAC,KAAyE,EAAE,MAAuB;QACpH,MAAM,GAAG,GAAG,MAAM,KAAK,wBAAwB,IAAI,MAAM,KAAK,qBAAqB,CAAC;QACpF,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,IAAI,KAAK,CAAC,UAAU,KAAK,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,+DAA+D,CAAC,CAAC;QAC5J,MAAM,QAAQ,GAAG,eAAe,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,KAAK,wBAAwB,CAAC,CAAC;QACtF,MAAM,QAAQ,GAAG,KAAK,CAAC,gBAAgB,IAAI,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;QAC5D,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,QAAQ,IAAI,CAAC,IAAI,QAAQ,GAAG,UAAU;YAAE,MAAM,IAAI,kBAAkB,CAAC,wDAAwD,CAAC,CAAC;QACtK,IAAI,GAAG,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAE,KAA4B,CAAC,UAAU,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,4DAA4D,CAAC,CAAC;QACrL,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;YAAE,MAAM,IAAI,kBAAkB,CAAC,sCAAsC,CAAC,CAAC;QAC3L,IAAI,MAAM,KAAK,wBAAwB,EAAE,CAAC;YACxC,MAAM,OAAO,GAAG,KAA8B,CAAC;YAC/C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC;gBAAE,MAAM,IAAI,kBAAkB,CAAC,wDAAwD,CAAC,CAAC;YAC3H,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC;YAC3D,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS;gBAAE,kBAAkB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC/E,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,eAAe,EAAE,CAAC;QACnC,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;QAC3F,IAAI,MAAsJ,CAAC;QAC3J,IAAI,MAAoF,CAAC;QACzF,IAAI,QAAgE,CAAC;QACrE,IAAI,IAA+B,CAAC;QACpC,IAAI,SAAkB,CAAC;QACvB,MAAM,MAAM,GAAG,GAAG,EAAE,GAAG,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,MAAM,YAAY,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACrI,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC;YACrC,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC,0BAA0B,CAAC,EAAE,MAAM,CAAC,CAAC;YACxE,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7B,MAAM,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;YACxE,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;YACxI,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAC;YAAC,KAAK,CAAC,cAAc,CAAC,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;YAChF,MAAM,GAAG,IAAI,GAAG,CAAC,uBAAuB,CAAC,EAAE,WAAW,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,UAAU,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,EAAE,iCAAiC,EAAE,QAAQ,EAAE,CAAC,CAAC;YACzR,MAAM,WAAW,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YAC5G,MAAM,WAAW,GAAG,CAAC,KAA4B,EAAE,EAAE;gBACnD,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;oBAAE,MAAM,IAAI,kBAAkB,CAAC,2CAA2C,CAAC,CAAC;gBACtH,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC;gBACzD,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;oBAAE,kBAAkB,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACzE,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YACrL,CAAC,CAAC;YACF,MAAM,GAAG,MAAM,KAAK,wBAAwB;gBAC1C,CAAC,CAAC,MAAM,CAAC,sBAAsB,CAAC,EAAE,QAAQ,EAAE,GAAG,WAAW,CAAC,KAA8B,CAAC,EAAE,EAAE,WAAW,CAAC;gBAC1G,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,CAAC;YAChC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,qEAAqE;YACnG,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YACzD,MAAM,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,MAAM,KAAK,wBAAwB,EAAE,CAAC;gBACxC,MAAM,MAAM,GAAG,KAAK,CAAC,MAAoD,CAAC;gBAC1E,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACzE,MAAM,KAAK,GAAG,KAAK,EAAE,OAAe,EAAE,EAAE;oBACtC,MAAM,CAAC,cAAc,EAAE,CAAC;oBACxB,IAAI,CAAC,MAAO,CAAC,KAAK,CAAC,OAAO,CAAC;wBAAE,MAAM,IAAI,CAAC,MAAO,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;gBACxE,CAAC,CAAC;gBACF,IAAI,GAAG,CAAC,KAAK,IAAI,EAAE;oBACjB,MAAM,KAAK,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,UAAU,EAAG,KAA4B,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;oBACpG,IAAI,UAAU,GAAG,CAAC,CAAC;oBACnB,OAAO,IAAI,EAAE,CAAC;wBACZ,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,QAAS,CAAC,IAAI,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;wBACxE,IAAI,IAAI,CAAC,IAAI;4BAAE,MAAM;wBACrB,IAAI,GAAG,EAAE,CAAC;4BACR,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,YAAY,UAAU,CAAC;gCAAE,MAAM,IAAI,kBAAkB,CAAC,uDAAuD,CAAC,CAAC;4BAC/H,MAAM,KAAK,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;wBAClD,CAAC;6BAAM,CAAC;4BACN,MAAM,KAAK,GAAG,IAAI,CAAC,KAA8B,CAAC;4BAClD,MAAM,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;wBAClC,CAAC;wBACD,qEAAqE;wBACrE,oEAAoE;wBACpE,IAAI,EAAE,UAAU,GAAG,EAAE,KAAK,CAAC;4BAAE,MAAM,SAAS,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;oBACtE,CAAC;oBACD,MAAO,CAAC,GAAG,EAAE,CAAC;gBAChB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,GAAG,SAAS,GAAG,KAAK,CAAC,CAAC,MAAO,CAAC,OAAO,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACxI,CAAC;YACD,IAAI,aAAa,GAAG,CAAC,CAAC;YACtB,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,MAAM,EAAE,CAAC;gBACnC,MAAM,CAAC,cAAc,EAAE,CAAC;gBACxB,IAAI,MAA2C,CAAC;gBAChD,IAAI,GAAG,EAAE,CAAC;oBACR,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,YAAY,UAAU,CAAC;wBAAE,MAAM,IAAI,kBAAkB,CAAC,mCAAmC,CAAC,CAAC;oBAC/I,MAAM,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;oBAC9C,aAAa,IAAI,MAAM,CAAC,UAAU,CAAC;gBACrC,CAAC;qBAAM,CAAC;oBACN,MAAM,OAAO,GAAG,CAAC,KAAa,EAAgB,EAAE;wBAC9C,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;wBAClI,kBAAkB,CAAC,MAAM,CAAC,CAAC;wBAAC,OAAO,MAAM,CAAC;oBAC5C,CAAC,CAAC;oBACF,MAAM,GAAG,EAAE,OAAO,EAAE,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;oBACvI,aAAa,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;gBAC7D,CAAC;gBACD,IAAI,aAAa,GAAG,QAAQ;oBAAE,MAAM,IAAI,6BAA6B,CAAC,EAAE,QAAQ,EAAE,aAAa,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBACzI,MAAM,MAAM,CAAC;YACf,CAAC;YACD,IAAI,SAAS;gBAAE,MAAM,SAAS,CAAC;QACjC,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC,CAAC;YACpD,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAC5C,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,CAAC;YACnB,CAAC;oBAAS,CAAC;gBACT,uEAAuE;gBACvE,qEAAqE;gBACrE,KAAK,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBACxE,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC;gBACb,CAAC;wBAAS,CAAC;oBACT,IAAI,CAAC;wBAAC,MAAM,MAAM,EAAE,KAAK,EAAE,CAAC;oBAAC,CAAC;4BACtB,CAAC;wBAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO;QACL,sBAAsB,EAAE,KAAK,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,wBAAwB,CAA0C;QAC9G,sBAAsB,EAAE,KAAK,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,wBAAwB,CAA8B;QAClG,mBAAmB,EAAE,KAAK,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,qBAAqB,CAA0C;QACxG,mBAAmB,EAAE,KAAK,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,qBAAqB,CAA8B;QAC5F,gBAAgB,EAAE,KAAK,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,EAAE,kBAAkB,CAA0C;KACnG,CAAC;AACJ,CAAC,CAAC"}

package/dist/index.d.ts

@@ -1,4 +1,28 @@
-import { type CallableProviderAdapter, type LanguageModel, type RealtimeConnectionFactory } from "@zhivex-ai/core/provider";
+import { type VertexTranscriptionModel as VertexTranscriptionModelContract } from "./transcription.js";
+export type { VertexAudioTranscriptionConfig, VertexAudioTranscription, VertexTranscriptionOptions, VertexTranscriptionResult, VertexTranscriptionModel } from "./transcription.js";
+import { type VertexVirtualTryOnClient } from "./virtual-try-on.js";
+import { type VertexEndpointsClient } from "./endpoints.js";
+import { type VertexGrpcClient } from "./grpc.js";
+export type { VertexGrpcServerInput, VertexGrpcClient, VertexGrpcRawInput, VertexGrpcTensorInput, VertexGrpcTensorFrame, VertexGrpcTensorOutput } from "./grpc.js";
+export type { VertexTensor, VertexTensorDataType } from "./tensors.js";
+export type { VertexEndpointDirectPredictInput, VertexEndpointDirectPredictResult } from "./endpoints.js";
+export type { VertexEndpointsClient, VertexEndpointRawPredictInput, VertexEndpointRawPredictResult, VertexEndpointStreamEvent, VertexEndpointDirectRawPredictInput, VertexEndpointDirectRawPredictResult, VertexEndpointExplainInput, VertexEndpointExplainResult } from "./endpoints.js";
+export type { VertexVirtualTryOnClient, VertexVirtualTryOnInput, VertexVirtualTryOnResult } from "./virtual-try-on.js";
+import type { VertexGeminiClient } from "./token-counting.js";
+export type { VertexGeminiClient, VertexGeminiTokenCountInput, VertexGeminiTokenCountResult } from "./token-counting.js";
+import { type VertexMultimodalEmbeddingClient } from "./multimodal-embeddings.js";
+export type { VertexMultimodalEmbeddingClient, VertexMultimodalEmbeddingInput, VertexMultimodalEmbeddingResult, VertexVideoSegmentConfig } from "./multimodal-embeddings.js";
+import { type VertexInteractionsClient } from "./interactions.js";
+export type { VertexInteractionsClient, VertexInteractionResumeInput } from "./interactions.js";
+export type { VertexClaudeOptions } from "./anthropic.js";
+import { type VertexSpecializedClients } from "./specialized.js";
+export type { VertexOCRClient, VertexFIMClient, VertexSpecializedClients } from "./specialized.js";
+export type { VertexEmbeddingOptions } from "./embeddings.js";
+import { type VertexChatModelOptions } from "./chat.js";
+export type { VertexChatModelOptions } from "./chat.js";
+import { type VertexClaudeClient } from "./token-counting.js";
+export type { VertexClaudeClient, VertexClaudeTokenCountInput, VertexClaudeTokenCountResult } from "./token-counting.js";
+import { type CachedContent, type CallableProviderAdapter, type ContextCacheUpdateInput, type ContextCachesClient, type LanguageModel, type RealtimeConnectionFactory } from "@zhivex-ai/core/provider";
 export interface VertexAuthClient {
     getAccessToken: () => string | null | undefined | Promise<string | null | undefined>;
 }
@@ -26,8 +50,22 @@ export interface VertexLanguageModelOptions {
     responseMimeType?: string;
     [key: string]: unknown;
 }
-export declare const createVertex: (options?: VertexProviderOptions) => CallableProviderAdapter<LanguageModel<VertexLanguageModelOptions>> & {
+export declare const createVertex: (options?: VertexProviderOptions) => {
+    transcriptionModel: (modelId: string) => VertexTranscriptionModelContract;
+} & CallableProviderAdapter<LanguageModel<VertexLanguageModelOptions>> & VertexSpecializedClients & {
+    caches: ContextCachesClient & {
+        update(input: ContextCacheUpdateInput): Promise<CachedContent>;
+    };
+    multimodalEmbeddings: VertexMultimodalEmbeddingClient;
     rawFetch: typeof globalThis.fetch;
+    interactions: VertexInteractionsClient;
+    claude: VertexClaudeClient;
+    gemini: VertexGeminiClient;
+    embeddingModel: (modelId: string) => import("@zhivex-ai/core/provider").EmbeddingModel;
+    chatModel: (modelId: string, options?: VertexChatModelOptions) => LanguageModel;
+    responsesModel: (modelId: string) => LanguageModel;
+    virtualTryOn: VertexVirtualTryOnClient;
+    endpoints: VertexEndpointsClient & VertexGrpcClient;
 };
 export declare const vertexMcpTools: (client: import("@zhivex-ai/core/provider").McpClient, options?: import("@zhivex-ai/core/provider").McpToolSetOptions) => Promise<import("@zhivex-ai/core/provider").ToolSet>;
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAeA,OAAO,EAgCL,KAAK,uBAAuB,EAgB5B,KAAK,aAAa,EAalB,KAAK,yBAAyB,EAW/B,MAAM,0BAA0B,CAAC;AAElC,MAAM,WAAW,gBAAgB;IAC/B,cAAc,EAAE,MAAM,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC,CAAC;CACtF;AAED,MAAM,WAAW,qBAAqB;IACpC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAChD,UAAU,CAAC,EAAE,gBAAgB,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,yBAAyB,CAAC,EAAE,yBAAyB,CAAC;IACtD,iGAAiG;IACjG,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC;AAoBD,MAAM,WAAW,0BAA0B;IACzC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AA03ED,eAAO,MAAM,YAAY,aACd,qBAAqB,KAC7B,uBAAuB,CAAC,aAAa,CAAC,0BAA0B,CAAC,CAAC,GAAG;IACtE,QAAQ,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;CAkHnC,CAAC;AAEF,eAAO,MAAM,cAAc,+KAAmB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmH,KAAK,wBAAwB,IAAI,gCAAgC,EAAE,MAAM,oBAAoB,CAAC;AACxN,YAAY,EAAE,8BAA8B,EAAE,wBAAwB,EAAE,0BAA0B,EAAE,yBAAyB,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AACpL,OAAO,EAAkC,KAAK,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AACpG,OAAO,EAA+B,KAAK,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AACzF,OAAO,EAA0B,KAAK,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC1E,YAAY,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,WAAW,CAAC;AACnK,YAAY,EAAE,YAAY,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACvE,YAAY,EAAE,gCAAgC,EAAE,iCAAiC,EAAE,MAAM,gBAAgB,CAAC;AAC1G,YAAY,EAAE,qBAAqB,EAAE,6BAA6B,EAAE,8BAA8B,EAAE,yBAAyB,EAAE,mCAAmC,EAAE,oCAAoC,EAAE,0BAA0B,EAAE,2BAA2B,EAAE,MAAM,gBAAgB,CAAC;AAC1R,YAAY,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAEvH,OAAO,KAAK,EAAE,kBAAkB,EAA+B,MAAM,qBAAqB,CAAC;AAC3F,YAAY,EAAE,kBAAkB,EAAE,2BAA2B,EAAE,4BAA4B,EAAE,MAAM,qBAAqB,CAAC;AACzH,OAAO,EAA6E,KAAK,+BAA+B,EAAE,MAAM,4BAA4B,CAAC;AAC7J,YAAY,EAAE,+BAA+B,EAAE,8BAA8B,EAAE,+BAA+B,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AAC7K,OAAO,EAAkC,KAAK,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAClG,YAAY,EAAE,wBAAwB,EAAE,4BAA4B,EAAE,MAAM,mBAAmB,CAAC;AAChG,YAAY,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAAkC,KAAK,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AACjG,YAAY,EAAE,eAAe,EAAE,eAAe,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AAEnG,YAAY,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAC9D,OAAO,EAA4C,KAAK,sBAAsB,EAAE,MAAM,WAAW,CAAC;AAClG,YAAY,EAAE,sBAAsB,EAAE,MAAM,WAAW,CAAC;AAExD,OAAO,EAA4B,KAAK,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AACxF,YAAY,EAAE,kBAAkB,EAAE,2BAA2B,EAAE,4BAA4B,EAAE,MAAM,qBAAqB,CAAC;AAgBzH,OAAO,EA+BL,KAAK,aAAa,EAClB,KAAK,uBAAuB,EAE5B,KAAK,uBAAuB,EAI5B,KAAK,mBAAmB,EAQxB,KAAK,aAAa,EAalB,KAAK,yBAAyB,EAS/B,MAAM,0BAA0B,CAAC;AAElC,MAAM,WAAW,gBAAgB;IAC/B,cAAc,EAAE,MAAM,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC,CAAC;CACtF;AAED,MAAM,WAAW,qBAAqB;IACpC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAChD,UAAU,CAAC,EAAE,gBAAgB,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,yBAAyB,CAAC,EAAE,yBAAyB,CAAC;IACtD,iGAAiG;IACjG,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC;AAkCD,MAAM,WAAW,0BAA0B;IACzC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAkhFD,eAAO,MAAM,YAAY,aACd,qBAAqB,KAC7B;IAAE,kBAAkB,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,gCAAgC,CAAA;CAAE,GAAG,uBAAuB,CAAC,aAAa,CAAC,0BAA0B,CAAC,CAAC,GAAG,wBAAwB,GAAG;IACjL,MAAM,EAAE,mBAAmB,GAAG;QAAE,MAAM,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,aAAa,CAAC,CAAA;KAAE,CAAC;IACjG,oBAAoB,EAAE,+BAA+B,CAAC;IACtD,QAAQ,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;IAClC,YAAY,EAAE,wBAAwB,CAAC;IACvC,MAAM,EAAE,kBAAkB,CAAC;IAC3B,MAAM,EAAE,kBAAkB,CAAC;IAC3B,cAAc,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,OAAO,0BAA0B,EAAE,cAAc,CAAC;IACvF,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,sBAAsB,KAAK,aAAa,CAAC;IAChF,cAAc,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,aAAa,CAAC;IACnD,YAAY,EAAE,wBAAwB,CAAC;IACvC,SAAS,EAAE,qBAAqB,GAAG,gBAAgB,CAAC;CAmNrD,CAAC;AAEF,eAAO,MAAM,cAAc,+KAAmB,CAAC"}