@truefoundry/ai-gateway-sdk 0.0.0-stage → 0.0.2

package/dist/providers/google-vertex-ai/createSpeech.js

@@ -0,0 +1,185 @@
+/**
+ * ================================
+ * TFY Managed File
+ * Vertex AI Text-to-Speech implementation
+ * Single-speaker: Google Cloud Text-to-Speech API (text:synthesize)
+ * Reference: https://cloud.google.com/text-to-speech/docs/reference/rest/v1/text/synthesize
+ * ================================
+ */
+import { CONTENT_TYPES, fileExtensionMimeTypeMap, GOOGLE_VERTEX_AI } from '../../globals.js';
+import { safeFetch } from '../../http/index.js';
+import { providerApiArgsFromHandler } from '../requestHandlerApiArgs.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+import VertexApiConfig from './api.js';
+import { GoogleErrorResponseTransform, mapVoiceToVertexFormat } from './utils.js';
+// Default values
+const DEFAULT_VOICE_NAME = 'en-US-Studio-O';
+/** Voice name mappings from OpenAI voices to Google Cloud / Vertex AI voices */
+const VERTEX_VOICE_MAPPINGS = {
+    alloy: 'en-US-Studio-O',
+    echo: 'en-US-Studio-M',
+    fable: 'en-GB-Studio-B',
+    onyx: 'en-US-Wavenet-D',
+    nova: 'en-US-Studio-O',
+    shimmer: 'en-US-Wavenet-F',
+    ash: 'en-US-Studio-O',
+    ballad: 'en-US-Studio-O',
+    coral: 'en-US-Studio-O',
+    sage: 'en-US-Studio-O',
+    verse: 'en-US-Studio-O',
+};
+// Google Cloud audio encoding format constants
+const AUDIO_ENCODING_FORMATS = {
+    MP3: 'MP3',
+    OGG_OPUS: 'OGG_OPUS',
+    FLAC: 'FLAC',
+    LINEAR16: 'LINEAR16',
+};
+const DEFAULT_AUDIO_ENCODING = AUDIO_ENCODING_FORMATS.LINEAR16;
+// Google Cloud TTS API endpoint
+const TTS_API_URL = 'https://texttospeech.googleapis.com/v1/text:synthesize';
+// Response format mappings from OpenAI to Google Cloud audio encoding
+const FORMAT_MAPPINGS = {
+    mp3: AUDIO_ENCODING_FORMATS.MP3,
+    opus: AUDIO_ENCODING_FORMATS.OGG_OPUS,
+    aac: AUDIO_ENCODING_FORMATS.MP3, // Google doesn't have AAC, use MP3
+    flac: AUDIO_ENCODING_FORMATS.FLAC,
+    wav: AUDIO_ENCODING_FORMATS.LINEAR16,
+    pcm: AUDIO_ENCODING_FORMATS.LINEAR16,
+};
+export const VertexAICreateSpeechConfig = {
+    model: {
+        param: 'model',
+        required: true,
+        default: 'gemini-2.5-flash-preview-tts',
+    },
+    input: {
+        param: 'input',
+        required: true,
+    },
+    voice: {
+        param: 'voice',
+        required: false,
+        default: DEFAULT_VOICE_NAME,
+    },
+    response_format: {
+        param: 'response_format',
+        required: false,
+        default: 'mp3',
+    },
+    speed: {
+        param: 'speed',
+        required: false,
+        default: 1,
+    },
+    instructions: {
+        param: 'instructions',
+        removeFromParams: true, // Vertex AI doesn't support instructions parameter
+    },
+    stream_format: {
+        param: 'stream_format',
+        removeFromParams: true, // Vertex AI doesn't support streaming format parameter
+    },
+};
+/**
+ * Custom request handler for Vertex AI TTS
+ * Single-speaker: Google Cloud text:synthesize API
+ */
+export const VertexAICreateSpeechRequestHandler = async (args) => {
+    const requestBody = args.request.body;
+    // TTS only accepts string input, not arrays or EmbedInput objects
+    if (typeof requestBody.input !== 'string' || !requestBody.input) {
+        return new Response(JSON.stringify({
+            error: {
+                message: 'Input must be a non-empty string for text-to-speech.',
+                type: 'invalid_request_error',
+            },
+        }), { status: 400, headers: { 'Content-Type': 'application/json' } });
+    }
+    const input = requestBody.input;
+    const voice = requestBody.voice;
+    const responseFormatRaw = requestBody.response_format;
+    const responseFormat = typeof responseFormatRaw === 'string' && responseFormatRaw ? responseFormatRaw : 'mp3';
+    const speed = requestBody.speed ?? 1.0;
+    // --- Single-speaker: classic Google Cloud text:synthesize ---
+    const vertexVoice = mapVoiceToVertexFormat(voice, VERTEX_VOICE_MAPPINGS, 'en-US');
+    if (!vertexVoice) {
+        return new Response(JSON.stringify({
+            error: {
+                message: 'Voice is required for single-speaker text-to-speech.',
+                type: 'invalid_request_error',
+            },
+        }), { status: 400, headers: { 'Content-Type': 'application/json' } });
+    }
+    const audioEncoding = FORMAT_MAPPINGS[responseFormat.toLowerCase()] || DEFAULT_AUDIO_ENCODING;
+    const vertexRequestBody = {
+        input: { text: input },
+        voice: vertexVoice,
+        audioConfig: { audioEncoding, speakingRate: String(speed) },
+    };
+    const headers = (await VertexApiConfig.headers({
+        ...providerApiArgsFromHandler(args, { endpoint: 'createSpeech', body: requestBody }),
+        transformedRequestBody: vertexRequestBody,
+        transformedRequestUrl: TTS_API_URL,
+    }));
+    let response;
+    try {
+        response = await safeFetch(TTS_API_URL, {
+            ssrf: 'guard',
+            method: 'POST',
+            signal: args.signal,
+            headers,
+            body: JSON.stringify(vertexRequestBody),
+        });
+    }
+    catch (error) {
+        const errorMessage = error.message || 'Text-to-speech processing failed';
+        return new Response(JSON.stringify({
+            error: { message: errorMessage, type: 'invalid_request_error' },
+        }), { status: 400, headers: { 'Content-Type': 'application/json' } });
+    }
+    if (!response.ok) {
+        return response;
+    }
+    const jsonResponse = await response.json();
+    const audioContent = jsonResponse.audioContent;
+    if (!audioContent) {
+        throw new Error('No audioContent in Vertex AI TTS response');
+    }
+    const binaryData = Buffer.from(audioContent, 'base64');
+    const contentTypeMap = {
+        MP3: CONTENT_TYPES.AUDIO_MPEG,
+        OGG_OPUS: 'audio/ogg',
+        FLAC: 'audio/flac',
+        LINEAR16: fileExtensionMimeTypeMap.wav,
+    };
+    const contentType = contentTypeMap[audioEncoding] || CONTENT_TYPES.AUDIO_MPEG;
+    return new Response(binaryData, {
+        status: 200,
+        headers: { 'Content-Type': contentType },
+    });
+};
+/**
+ * Response transformer for Vertex AI TTS
+ * Since we handle response transformation in the request handler,
+ * this transformer just passes through the response for success cases
+ * and handles error transformation for error cases.
+ */
+export const VertexAICreateSpeechResponseTransform = (response, responseStatus) => {
+    // Check for error responses and transform them
+    if (responseStatus !== 200 && 'error' in response) {
+        const errorResponse = GoogleErrorResponseTransform(response);
+        if (errorResponse) {
+            return errorResponse;
+        }
+        return generateInvalidProviderResponseError({
+            error: {
+                message: 'Failed to process Vertex AI TTS error response',
+                type: 'invalid_request_error',
+            },
+        }, GOOGLE_VERTEX_AI);
+    }
+    // For success cases, response transformation is already handled in the request handler
+    return response;
+};
+//# sourceMappingURL=createSpeech.js.map

package/dist/providers/google-vertex-ai/createSpeech.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createSpeech.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/createSpeech.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,EAAE,aAAa,EAAE,wBAAwB,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAC1F,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAEtE,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAChE,OAAO,eAAe,MAAM,OAAO,CAAC;AAEpC,OAAO,EAAE,4BAA4B,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAE/E,iBAAiB;AACjB,MAAM,kBAAkB,GAAG,gBAAgB,CAAC;AAE5C,gFAAgF;AAChF,MAAM,qBAAqB,GAA2B;IACpD,KAAK,EAAE,gBAAgB;IACvB,IAAI,EAAE,gBAAgB;IACtB,KAAK,EAAE,gBAAgB;IACvB,IAAI,EAAE,iBAAiB;IACvB,IAAI,EAAE,gBAAgB;IACtB,OAAO,EAAE,iBAAiB;IAC1B,GAAG,EAAE,gBAAgB;IACrB,MAAM,EAAE,gBAAgB;IACxB,KAAK,EAAE,gBAAgB;IACvB,IAAI,EAAE,gBAAgB;IACtB,KAAK,EAAE,gBAAgB;CACxB,CAAC;AAOF,+CAA+C;AAC/C,MAAM,sBAAsB,GAAG;IAC7B,GAAG,EAAE,KAAK;IACV,QAAQ,EAAE,UAAU;IACpB,IAAI,EAAE,MAAM;IACZ,QAAQ,EAAE,UAAU;CACZ,CAAC;AAEX,MAAM,sBAAsB,GAAG,sBAAsB,CAAC,QAAQ,CAAC;AAE/D,gCAAgC;AAChC,MAAM,WAAW,GAAG,wDAAwD,CAAC;AAE7E,sEAAsE;AACtE,MAAM,eAAe,GAA2B;IAC9C,GAAG,EAAE,sBAAsB,CAAC,GAAG;IAC/B,IAAI,EAAE,sBAAsB,CAAC,QAAQ;IACrC,GAAG,EAAE,sBAAsB,CAAC,GAAG,EAAE,mCAAmC;IACpE,IAAI,EAAE,sBAAsB,CAAC,IAAI;IACjC,GAAG,EAAE,sBAAsB,CAAC,QAAQ;IACpC,GAAG,EAAE,sBAAsB,CAAC,QAAQ;CACrC,CAAC;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAAmB;IACxD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,8BAA8B;KACxC;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,IAAI;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;QACf,OAAO,EAAE,kBAAkB;KAC5B;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;QACf,OAAO,EAAE,KAAK;KACf;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,KAAK;QACf,OAAO,EAAE,CAAC;KACX;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,cAAc;QACrB,gBAAgB,EAAE,IAAI,EAAE,mDAAmD;KAC5E;IACD,aAAa,EAAE;QACb,KAAK,EAAE,eAAe;QACtB,gBAAgB,EAAE,IAAI,EAAE,uDAAuD;KAChF;CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,kCAAkC,GAA2B,KAAK,EAAC,IAAI,EAAC,EAAE;IACrF,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IACtC,kEAAkE;IAClE,IAAI,OAAO,WAAW,CAAC,KAAK,KAAK,QAAQ,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QAChE,OAAO,IAAI,QAAQ,CACjB,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE;gBACL,OAAO,EAAE,sDAAsD;gBAC/D,IAAI,EAAE,uBAAuB;aAC9B;SACF,CAAC,EACF,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CAAC;IACJ,CAAC;IACD,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC;IAChC,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC;IAChC,MAAM,iBAAiB,GAAG,WAAW,CAAC,eAAe,CAAC;IACtD,MAAM,cAAc,GAAG,OAAO,iBAAiB,KAAK,QAAQ,IAAI,iBAAiB,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,KAAK,CAAC;IAC9G,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,IAAI,GAAG,CAAC;IAEvC,+DAA+D;IAC/D,MAAM,WAAW,GAAG,sBAAsB,CAAC,KAAK,EAAE,qBAAqB,EAAE,OAAO,CAAC,CAAC;IAClF,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,IAAI,QAAQ,CACjB,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE;gBACL,OAAO,EAAE,sDAAsD;gBAC/D,IAAI,EAAE,uBAAuB;aAC9B;SACF,CAAC,EACF,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CAAC;IACJ,CAAC;IAED,MAAM,aAAa,GAAG,eAAe,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,IAAI,sBAAsB,CAAC;IAC9F,MAAM,iBAAiB,GAInB;QACF,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE;QACtB,KAAK,EAAE,WAAW;QAClB,WAAW,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;KAC5D,CAAC;IAEF,MAAM,OAAO,GAAG,CAAC,MAAM,eAAe,CAAC,OAAO,CAAC;QAC7C,GAAG,0BAA0B,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,cAAc,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;QACpF,sBAAsB,EAAE,iBAAiB;QACzC,qBAAqB,EAAE,WAAW;KACnC,CAAC,CAA2B,CAAC;IAE9B,IAAI,QAAkB,CAAC;IACvB,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,SAAS,CAAC,WAAW,EAAE;YACtC,IAAI,EAAE,OAAO;YACb,MAAM,EAAE,MAAM;YACd,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;SACxC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,YAAY,GAAI,KAAe,CAAC,OAAO,IAAI,kCAAkC,CAAC;QACpF,OAAO,IAAI,QAAQ,CACjB,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,uBAAuB,EAAE;SAChE,CAAC,EACF,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CACjE,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,YAAY,GAAwB,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAChE,MAAM,YAAY,GAAG,YAAY,CAAC,YAAY,CAAC;IAC/C,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACvD,MAAM,cAAc,GAA2B;QAC7C,GAAG,EAAE,aAAa,CAAC,UAAU;QAC7B,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,YAAY;QAClB,QAAQ,EAAE,wBAAwB,CAAC,GAAG;KACvC,CAAC;IACF,MAAM,WAAW,GAAG,cAAc,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,UAAU,CAAC;IAE9E,OAAO,IAAI,QAAQ,CAAC,UAAU,EAAE;QAC9B,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,EAAE,cAAc,EAAE,WAAW,EAAE;KACzC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF;;;;;GAKG;AACH,MAAM,CAAC,MAAM,qCAAqC,GAGlB,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IAC3D,+CAA+C;IAC/C,IAAI,cAAc,KAAK,GAAG,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;QAClD,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAA+B,CAAC,CAAC;QACpF,IAAI,aAAa,EAAE,CAAC;YAClB,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,OAAO,oCAAoC,CACzC;YACE,KAAK,EAAE;gBACL,OAAO,EAAE,gDAAgD;gBACzD,IAAI,EAAE,uBAAuB;aAC9B;SACF,EACD,gBAAgB,CACjB,CAAC;IACJ,CAAC;IAED,uFAAuF;IACvF,OAAO,QAAoB,CAAC;AAC9B,CAAC,CAAC"}

package/dist/providers/google-vertex-ai/embed.d.ts

@@ -0,0 +1,32 @@
+import { EmbedParams, EmbedResponse } from '../../types/embedRequestBody.js';
+import { Params } from '../../types/requestBody.js';
+import { ErrorResponse, ProviderConfig } from '../types.js';
+import { GoogleEmbedContentResponse, GoogleEmbedResponse, GoogleErrorResponse } from './types.js';
+declare enum TASK_TYPE {
+    RETRIEVAL_QUERY = "RETRIEVAL_QUERY",
+    RETRIEVAL_DOCUMENT = "RETRIEVAL_DOCUMENT",
+    SEMANTIC_SIMILARITY = "SEMANTIC_SIMILARITY",
+    CLASSIFICATION = "CLASSIFICATION",
+    CLUSTERING = "CLUSTERING",
+    QUESTION_ANSWERING = "QUESTION_ANSWERING",
+    FACT_VERIFICATION = "FACT_VERIFICATION",
+    CODE_RETRIEVAL_QUERY = "CODE_RETRIEVAL_QUERY"
+}
+export interface GoogleEmbedParams extends EmbedParams {
+    task_type: TASK_TYPE | string;
+    parameters?: {
+        outputDimensionality: number;
+    };
+}
+export declare const GoogleEmbedConfig: ProviderConfig;
+export declare const GoogleEmbedResponseTransform: (response: GoogleEmbedResponse | GoogleErrorResponse, responseStatus: number, responseHeaders: Headers, strictOpenAiCompliance: boolean, gatewayRequestUrl: string, gatewayRequest: Params) => EmbedResponse | ErrorResponse;
+export declare const VertexBatchEmbedConfig: ProviderConfig;
+/** Vertex `gemini-embedding-2*` models routed to `:embedContent` (vs `:predict`). */
+export declare const isEmbedContentModel: (model: string) => boolean;
+/** Vertex multimodal embedding models — `:predict` instances use `{text}` (not `{content}`). */
+export declare const MULTIMODAL_EMBED_MODELS: ReadonlySet<string>;
+export declare const isMultimodalEmbedModel: (model: string) => boolean;
+/** @see https://docs.cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1beta1/projects.locations.publishers.models/embedContent#request-body */
+export declare const GoogleEmbedContentConfig: ProviderConfig;
+export declare const GoogleEmbedContentResponseTransform: (response: GoogleEmbedContentResponse | GoogleErrorResponse, responseStatus: number, responseHeaders: Headers, strictOpenAiCompliance: boolean, gatewayRequestUrl: string, gatewayRequest: Params) => EmbedResponse | ErrorResponse;
+export {};

package/dist/providers/google-vertex-ai/embed.js

@@ -0,0 +1,275 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { GOOGLE_VERTEX_AI } from '../../globals.js';
+import { generateInvalidProviderResponseError, postProcessEmbeddingResponse, validateEncodingFormatAndRemove, } from '../utils.js';
+import { transformEmbeddingInputs, transformEmbeddingsParameters } from './transformGenerationConfig.js';
+import { getMimeType, GoogleErrorResponseTransform } from './utils.js';
+var TASK_TYPE;
+(function (TASK_TYPE) {
+    TASK_TYPE["RETRIEVAL_QUERY"] = "RETRIEVAL_QUERY";
+    TASK_TYPE["RETRIEVAL_DOCUMENT"] = "RETRIEVAL_DOCUMENT";
+    TASK_TYPE["SEMANTIC_SIMILARITY"] = "SEMANTIC_SIMILARITY";
+    TASK_TYPE["CLASSIFICATION"] = "CLASSIFICATION";
+    TASK_TYPE["CLUSTERING"] = "CLUSTERING";
+    TASK_TYPE["QUESTION_ANSWERING"] = "QUESTION_ANSWERING";
+    TASK_TYPE["FACT_VERIFICATION"] = "FACT_VERIFICATION";
+    TASK_TYPE["CODE_RETRIEVAL_QUERY"] = "CODE_RETRIEVAL_QUERY";
+})(TASK_TYPE || (TASK_TYPE = {}));
+export const GoogleEmbedConfig = {
+    input: {
+        param: 'instances',
+        required: true,
+        transform: (params) => transformEmbeddingInputs(params),
+    },
+    parameters: {
+        param: 'parameters',
+        required: false,
+    },
+    dimensions: {
+        param: 'parameters',
+        required: false,
+        transform: (params) => transformEmbeddingsParameters(params),
+    },
+    encoding_format: {
+        param: 'embedding_types',
+        required: false,
+        removeFromParams: validateEncodingFormatAndRemove,
+    },
+    task_type: {
+        param: 'task_type',
+        removeFromParams: true,
+    },
+    // Vertex has no end-user identifier field and rejects unknown fields.
+    user: {
+        param: 'user',
+        removeFromParams: true,
+    },
+};
+export const GoogleEmbedResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200) {
+        const errorResponse = GoogleErrorResponseTransform(response);
+        if (errorResponse)
+            return errorResponse;
+    }
+    const model = gatewayRequest.model || '';
+    if ('predictions' in response) {
+        const data = [];
+        let tokenCount = 0;
+        response.predictions.forEach((prediction, index) => {
+            const item = {
+                object: 'embedding',
+                index: index,
+                ...(prediction.imageEmbedding && {
+                    image_embedding: prediction.imageEmbedding,
+                }),
+                ...(prediction.videoEmbeddings && {
+                    video_embeddings: prediction.videoEmbeddings.map((videoEmbedding, idx) => ({
+                        object: 'embedding',
+                        embedding: videoEmbedding.embedding,
+                        index: idx,
+                        start_offset: videoEmbedding.startOffsetSec,
+                        end_offset: videoEmbedding.endOffsetSec,
+                    })),
+                }),
+                ...(prediction.textEmbedding && {
+                    embedding: prediction.textEmbedding,
+                }),
+                ...(prediction.embeddings && {
+                    embedding: prediction.embeddings.values,
+                }),
+            };
+            tokenCount += prediction?.embeddings?.statistics?.token_count || 0;
+            data.push(item);
+        });
+        data.forEach((item, index) => {
+            item.index = index;
+        });
+        const openAIResponse = {
+            object: 'list',
+            data: data,
+            provider: GOOGLE_VERTEX_AI,
+            model,
+            usage: {
+                prompt_tokens: tokenCount,
+                total_tokens: tokenCount,
+            },
+        };
+        // Post-process to convert to base64 if requested
+        return postProcessEmbeddingResponse(openAIResponse, gatewayRequest.encoding_format);
+    }
+    return generateInvalidProviderResponseError(response, GOOGLE_VERTEX_AI);
+};
+export const VertexBatchEmbedConfig = {
+    input: {
+        param: 'content',
+        required: true,
+        transform: (value) => {
+            if (typeof value.input === 'string') {
+                return value.input;
+            }
+            return value.input.map(item => item).join('\n');
+        },
+    },
+};
+/** Vertex `gemini-embedding-2*` models routed to `:embedContent` (vs `:predict`). */
+export const isEmbedContentModel = (model) => {
+    if (!model)
+        return false;
+    return model.includes('gemini-embedding-2');
+};
+/** Vertex multimodal embedding models — `:predict` instances use `{text}` (not `{content}`). */
+export const MULTIMODAL_EMBED_MODELS = new Set(['multimodalembedding']);
+export const isMultimodalEmbedModel = (model) => {
+    if (!model)
+        return false;
+    for (const needle of MULTIMODAL_EMBED_MODELS) {
+        if (model.includes(needle))
+            return true;
+    }
+    return false;
+};
+const DEFAULT_INLINE_IMAGE_MIME_TYPE = 'image/jpeg';
+const DEFAULT_INLINE_VIDEO_MIME_TYPE = 'video/mp4';
+const buildInlinePart = (base64, fallbackMime) => {
+    // Accept `data:<mime>;base64,<data>` or raw base64.
+    if (base64.startsWith('data:') && base64.includes(';base64,')) {
+        const [meta, data] = base64.split(',', 2);
+        const mimeType = meta.slice(5).split(';')[0] || fallbackMime;
+        return { inlineData: { mimeType, data } };
+    }
+    return { inlineData: { mimeType: fallbackMime, data: base64 } };
+};
+const buildFilePart = (uri, fallbackMime) => {
+    const mimeType = getMimeType(uri) ?? fallbackMime;
+    return { fileData: { mimeType, fileUri: uri } };
+};
+/**
+ * Seconds -> Vertex `Duration` ("<n>s"). `interval` is dropped (no direct analog).
+ * @see https://docs.cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1beta1/Content#VideoMetadata
+ */
+const buildVideoMetadata = (v) => {
+    const meta = {};
+    if (typeof v.start_offset === 'number')
+        meta.startOffset = `${v.start_offset}s`;
+    if (typeof v.end_offset === 'number')
+        meta.endOffset = `${v.end_offset}s`;
+    return Object.keys(meta).length > 0 ? meta : undefined;
+};
+/**
+ * Build `content.parts[]` from OpenAI-style `input`. Multi-element inputs are
+ * collapsed into a single `content` (Vertex returns one embedding per call).
+ *
+ * Dropped (no `:embedContent` analog): `image.text`, `video.text` (captions),
+ * `video.interval` (only `fps` exists, inverse/lossy).
+ */
+const transformEmbedContentInput = (params) => {
+    const parts = [];
+    const pushFromString = (s) => {
+        parts.push({ text: s });
+    };
+    const pushFromEmbedInput = (item) => {
+        if (item.text) {
+            parts.push({ text: item.text });
+        }
+        if (item.image?.url) {
+            parts.push(buildFilePart(item.image.url, DEFAULT_INLINE_IMAGE_MIME_TYPE));
+        }
+        if (item.image?.base64) {
+            parts.push(buildInlinePart(item.image.base64, DEFAULT_INLINE_IMAGE_MIME_TYPE));
+        }
+        if (item.video) {
+            const metadata = buildVideoMetadata(item.video);
+            if (item.video.url) {
+                const part = buildFilePart(item.video.url, DEFAULT_INLINE_VIDEO_MIME_TYPE);
+                if (metadata)
+                    part.videoMetadata = metadata;
+                parts.push(part);
+            }
+            if (item.video.base64) {
+                const part = buildInlinePart(item.video.base64, DEFAULT_INLINE_VIDEO_MIME_TYPE);
+                if (metadata)
+                    part.videoMetadata = metadata;
+                parts.push(part);
+            }
+        }
+    };
+    if (typeof params.input === 'string') {
+        pushFromString(params.input);
+    }
+    else if (Array.isArray(params.input)) {
+        params.input.forEach(item => {
+            if (typeof item === 'string') {
+                pushFromString(item);
+            }
+            else if (item && typeof item === 'object') {
+                pushFromEmbedInput(item);
+            }
+        });
+    }
+    return { parts };
+};
+/** @see https://docs.cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1beta1/projects.locations.publishers.models/embedContent#request-body */
+export const GoogleEmbedContentConfig = {
+    input: {
+        param: 'content',
+        required: true,
+        transform: (params) => transformEmbedContentInput(params),
+    },
+    dimensions: {
+        param: 'outputDimensionality',
+        required: false,
+    },
+    task_type: {
+        param: 'taskType',
+        required: false,
+    },
+    parameters: {
+        param: 'parameters',
+        removeFromParams: true,
+    },
+    encoding_format: {
+        param: 'encoding_format',
+        required: false,
+        removeFromParams: validateEncodingFormatAndRemove,
+    },
+    // Vertex has no end-user identifier field and rejects unknown fields.
+    user: {
+        param: 'user',
+        removeFromParams: true,
+    },
+};
+export const GoogleEmbedContentResponseTransform = (response, responseStatus, _responseHeaders, _strictOpenAiCompliance, _gatewayRequestUrl, gatewayRequest) => {
+    if (responseStatus !== 200) {
+        const errorResponse = GoogleErrorResponseTransform(response);
+        if (errorResponse)
+            return errorResponse;
+    }
+    const model = gatewayRequest.model || '';
+    if ('embedding' in response && response.embedding && Array.isArray(response.embedding.values)) {
+        const promptTokens = response.usageMetadata?.promptTokenCount ?? 0;
+        const totalTokens = response.usageMetadata?.totalTokenCount ?? promptTokens;
+        const data = [
+            {
+                object: 'embedding',
+                embedding: response.embedding.values,
+                index: 0,
+            },
+        ];
+        const openAIResponse = {
+            object: 'list',
+            data,
+            provider: GOOGLE_VERTEX_AI,
+            model,
+            usage: {
+                prompt_tokens: promptTokens,
+                total_tokens: totalTokens,
+            },
+        };
+        return postProcessEmbeddingResponse(openAIResponse, gatewayRequest.encoding_format);
+    }
+    return generateInvalidProviderResponseError(response, GOOGLE_VERTEX_AI);
+};
+//# sourceMappingURL=embed.js.map

package/dist/providers/google-vertex-ai/embed.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embed.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/embed.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAIjD,OAAO,EACL,oCAAoC,EACpC,4BAA4B,EAC5B,+BAA+B,GAChC,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,wBAAwB,EAAE,6BAA6B,EAAE,MAAM,6BAA6B,CAAC;AAEtG,OAAO,EAAE,WAAW,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAEpE,IAAK,SASJ;AATD,WAAK,SAAS;IACZ,gDAAmC,CAAA;IACnC,sDAAyC,CAAA;IACzC,wDAA2C,CAAA;IAC3C,8CAAiC,CAAA;IACjC,sCAAyB,CAAA;IACzB,sDAAyC,CAAA;IACzC,oDAAuC,CAAA;IACvC,0DAA6C,CAAA;AAC/C,CAAC,EATI,SAAS,KAAT,SAAS,QASb;AASD,MAAM,CAAC,MAAM,iBAAiB,GAAmB;IAC/C,KAAK,EAAE;QACL,KAAK,EAAE,WAAW;QAClB,QAAQ,EAAE,IAAI;QACd,SAAS,EAAE,CAAC,MAAyB,EAAE,EAAE,CAAC,wBAAwB,CAAC,MAAM,CAAC;KAC3E;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,QAAQ,EAAE,KAAK;KAChB;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,QAAQ,EAAE,KAAK;QACf,SAAS,EAAE,CAAC,MAAyB,EAAE,EAAE,CAAC,6BAA6B,CAAC,MAAM,CAAC;KAChF;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,+BAA+B;KAClD;IACD,SAAS,EAAE;QACT,KAAK,EAAE,WAAW;QAClB,gBAAgB,EAAE,IAAI;KACvB;IACD,sEAAsE;IACtE,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;QACb,gBAAgB,EAAE,IAAI;KACvB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,4BAA4B,GAOJ,CACnC,QAAQ,EACR,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAA+B,CAAC,CAAC;QACpF,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,MAAM,KAAK,GAAI,cAAc,CAAC,KAAgB,IAAI,EAAE,CAAC;IAErD,IAAI,aAAa,IAAI,QAAQ,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAwB,EAAE,CAAC;QACrC,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,QAAQ,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE;YACjD,MAAM,IAAI,GAAG;gBACX,MAAM,EAAE,WAAW;gBACnB,KAAK,EAAE,KAAK;gBACZ,GAAG,CAAC,UAAU,CAAC,cAAc,IAAI;oBAC/B,eAAe,EAAE,UAAU,CAAC,cAAc;iBAC3C,CAAC;gBACF,GAAG,CAAC,UAAU,CAAC,eAAe,IAAI;oBAChC,gBAAgB,EAAE,UAAU,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,cAAc,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACzE,MAAM,EAAE,WAAW;wBACnB,SAAS,EAAE,cAAc,CAAC,SAAS;wBACnC,KAAK,EAAE,GAAG;wBACV,YAAY,EAAE,cAAc,CAAC,cAAc;wBAC3C,UAAU,EAAE,cAAc,CAAC,YAAY;qBACxC,CAAC,CAAC;iBACJ,CAAC;gBACF,GAAG,CAAC,UAAU,CAAC,aAAa,IAAI;oBAC9B,SAAS,EAAE,UAAU,CAAC,aAAa;iBACpC,CAAC;gBACF,GAAG,CAAC,UAAU,CAAC,UAAU,IAAI;oBAC3B,SAAS,EAAE,UAAU,CAAC,UAAU,CAAC,MAAM;iBACxC,CAAC;aACH,CAAC;YACF,UAAU,IAAI,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,IAAI,CAAC,CAAC;YACnE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAC3B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,CAAC,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE,gBAAgB;YAC1B,KAAK;YACL,KAAK,EAAE;gBACL,aAAa,EAAE,UAAU;gBACzB,YAAY,EAAE,UAAU;aACzB;SACF,CAAC;QAEF,iDAAiD;QACjD,OAAO,4BAA4B,CAAC,cAAc,EAAE,cAAc,CAAC,eAAyB,CAAC,CAAC;IAChG,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;AAC1E,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,sBAAsB,GAAmB;IACpD,KAAK,EAAE;QACL,KAAK,EAAE,SAAS;QAChB,QAAQ,EAAE,IAAI;QACd,SAAS,EAAE,CAAC,KAAkB,EAAE,EAAE;YAChC,IAAI,OAAO,KAAK,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACpC,OAAO,KAAK,CAAC,KAAK,CAAC;YACrB,CAAC;YACD,OAAO,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC;KACF;CACF,CAAC;AAEF,qFAAqF;AACrF,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,KAAa,EAAW,EAAE;IAC5D,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,OAAO,KAAK,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC;AAC9C,CAAC,CAAC;AAEF,gGAAgG;AAChG,MAAM,CAAC,MAAM,uBAAuB,GAAwB,IAAI,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC;AAE7F,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,KAAa,EAAW,EAAE;IAC/D,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,KAAK,MAAM,MAAM,IAAI,uBAAuB,EAAE,CAAC;QAC7C,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,IAAI,CAAC;IAC1C,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAgBF,MAAM,8BAA8B,GAAG,YAAY,CAAC;AACpD,MAAM,8BAA8B,GAAG,WAAW,CAAC;AAEnD,MAAM,eAAe,GAAG,CAAC,MAAc,EAAE,YAAoB,EAAc,EAAE;IAC3E,oDAAoD;IACpD,IAAI,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAC9D,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC;QAC7D,OAAO,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAC;IAC5C,CAAC;IACD,OAAO,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC;AAClE,CAAC,CAAC;AAEF,MAAM,aAAa,GAAG,CAAC,GAAW,EAAE,YAAoB,EAAc,EAAE;IACtE,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,YAAY,CAAC;IAClD,OAAO,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,CAAC;AAClD,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,kBAAkB,GAAG,CAAC,CAAiD,EAAiC,EAAE;IAC9G,MAAM,IAAI,GAAsB,EAAE,CAAC;IACnC,IAAI,OAAO,CAAC,CAAC,YAAY,KAAK,QAAQ;QAAE,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC,CAAC,YAAY,GAAG,CAAC;IAChF,IAAI,OAAO,CAAC,CAAC,UAAU,KAAK,QAAQ;QAAE,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,UAAU,GAAG,CAAC;IAC1E,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;AACzD,CAAC,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,0BAA0B,GAAG,CAAC,MAAyB,EAA2B,EAAE;IACxF,MAAM,KAAK,GAAiB,EAAE,CAAC;IAE/B,MAAM,cAAc,GAAG,CAAC,CAAS,EAAE,EAAE;QACnC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC;IAEF,MAAM,kBAAkB,GAAG,CAAC,IAAgB,EAAE,EAAE;QAC9C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC;YACpB,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,8BAA8B,CAAC,CAAC,CAAC;QAC5E,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;YACvB,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,8BAA8B,CAAC,CAAC,CAAC;QACjF,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;gBACnB,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,8BAA8B,CAAC,CAAC;gBAC3E,IAAI,QAAQ;oBAAE,IAAI,CAAC,aAAa,GAAG,QAAQ,CAAC;gBAC5C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC;YACD,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;gBACtB,MAAM,IAAI,GAAG,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,8BAA8B,CAAC,CAAC;gBAChF,IAAI,QAAQ;oBAAE,IAAI,CAAC,aAAa,GAAG,QAAQ,CAAC;gBAC5C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;SAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;YAC1B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC7B,cAAc,CAAC,IAAI,CAAC,CAAC;YACvB,CAAC;iBAAM,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC5C,kBAAkB,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,CAAC;AACnB,CAAC,CAAC;AAEF,4JAA4J;AAC5J,MAAM,CAAC,MAAM,wBAAwB,GAAmB;IACtD,KAAK,EAAE;QACL,KAAK,EAAE,SAAS;QAChB,QAAQ,EAAE,IAAI;QACd,SAAS,EAAE,CAAC,MAAyB,EAAE,EAAE,CAAC,0BAA0B,CAAC,MAAM,CAAC;KAC7E;IACD,UAAU,EAAE;QACV,KAAK,EAAE,sBAAsB;QAC7B,QAAQ,EAAE,KAAK;KAChB;IACD,SAAS,EAAE;QACT,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,KAAK;KAChB;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,gBAAgB,EAAE,IAAI;KACvB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,iBAAiB;QACxB,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,+BAA+B;KAClD;IACD,sEAAsE;IACtE,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;QACb,gBAAgB,EAAE,IAAI;KACvB;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,mCAAmC,GAOX,CACnC,QAAQ,EACR,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,kBAAkB,EAClB,cAAc,EACd,EAAE;IACF,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,aAAa,GAAG,4BAA4B,CAAC,QAA+B,CAAC,CAAC;QACpF,IAAI,aAAa;YAAE,OAAO,aAAa,CAAC;IAC1C,CAAC;IAED,MAAM,KAAK,GAAI,cAAc,CAAC,KAAgB,IAAI,EAAE,CAAC;IAErD,IAAI,WAAW,IAAI,QAAQ,IAAI,QAAQ,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9F,MAAM,YAAY,GAAG,QAAQ,CAAC,aAAa,EAAE,gBAAgB,IAAI,CAAC,CAAC;QACnE,MAAM,WAAW,GAAG,QAAQ,CAAC,aAAa,EAAE,eAAe,IAAI,YAAY,CAAC;QAE5E,MAAM,IAAI,GAAwB;YAChC;gBACE,MAAM,EAAE,WAAW;gBACnB,SAAS,EAAE,QAAQ,CAAC,SAAS,CAAC,MAAM;gBACpC,KAAK,EAAE,CAAC;aACT;SACF,CAAC;QAEF,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,MAAM;YACd,IAAI;YACJ,QAAQ,EAAE,gBAAgB;YAC1B,KAAK;YACL,KAAK,EAAE;gBACL,aAAa,EAAE,YAAY;gBAC3B,YAAY,EAAE,WAAW;aAC1B;SACF,CAAC;QAEF,OAAO,4BAA4B,CAAC,cAAc,EAAE,cAAc,CAAC,eAAyB,CAAC,CAAC;IAChG,CAAC;IAED,OAAO,oCAAoC,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;AAC1E,CAAC,CAAC"}

package/dist/providers/google-vertex-ai/getBatchOutput.d.ts

@@ -0,0 +1,3 @@
+import { RequestHandler } from '../types.js';
+export declare const BatchOutputRequestHandler: RequestHandler;
+export declare const BatchOutputResponseTransform: (response: Response) => Promise<Response>;