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

package/dist/providers/google/imageGenerate.d.ts

@@ -0,0 +1,17 @@
+import { GoogleGeminiImageGenResponse, isGeminiImageModel } from '../google-vertex-ai/imageGenerate';
+import { ErrorResponse, ImageGenerateResponse, ProviderConfig } from '../types';
+import { GoogleErrorResponse } from './chatComplete';
+export { isGeminiImageModel };
+/** Request config for Gemini image generation (generateContent with responseModalities). */
+export declare const GoogleGeminiImageGenConfig: ProviderConfig;
+/** Request config for Imagen image generation (predict: instances + parameters). */
+export declare const GoogleImagenImageGenConfig: ProviderConfig;
+/** Imagen predict response: predictions with bytesBase64Encoded. */
+interface GoogleImagenImageGenResponse {
+    predictions?: Array<{
+        bytesBase64Encoded?: string;
+    }>;
+}
+/** Response transform: branches on candidates (Gemini) vs predictions (Imagen). */
+export declare const GoogleImageGenResponseTransform: (response: GoogleGeminiImageGenResponse | GoogleImagenImageGenResponse | GoogleErrorResponse, responseStatus: number) => ImageGenerateResponse | ErrorResponse;
+//# sourceMappingURL=imageGenerate.d.ts.map

package/dist/providers/google/imageGenerate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"imageGenerate.d.ts","sourceRoot":"","sources":["../../../src/providers/google/imageGenerate.ts"],"names":[],"mappings":"AASA,OAAO,EAEL,4BAA4B,EAC5B,kBAAkB,EAEnB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAEhF,OAAO,EAAE,mBAAmB,EAAgC,MAAM,gBAAgB,CAAC;AAEnF,OAAO,EAAE,kBAAkB,EAAE,CAAC;AAE9B,4FAA4F;AAC5F,eAAO,MAAM,0BAA0B,EAAE,cA2CxC,CAAC;AAoBF,oFAAoF;AACpF,eAAO,MAAM,0BAA0B,EAAE,cA4CxC,CAAC;AAEF,oEAAoE;AACpE,UAAU,4BAA4B;IACpC,WAAW,CAAC,EAAE,KAAK,CAAC;QAAE,kBAAkB,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACtD;AA4DD,mFAAmF;AACnF,eAAO,MAAM,+BAA+B,EAAE,CAC5C,QAAQ,EAAE,4BAA4B,GAAG,4BAA4B,GAAG,mBAAmB,EAC3F,cAAc,EAAE,MAAM,KACnB,qBAAqB,GAAG,aAQ5B,CAAC"}

package/dist/providers/google/imageGenerate.js

@@ -0,0 +1,181 @@
+/**
+ * Google AI Studio image generation.
+ * Gemini models use generateContent with responseModalities; Imagen models use predict.
+ * Gateway params (n, size, aspect_ratio, etc.) are mapped to the respective API formats.
+ * Tools (e.g. google_search for grounding) are supported for Gemini image models only.
+ * Gateway accepts tools in snake_case as per Gemini API, e.g. tools: [{ google_search: {} }].
+ */
+import { GOOGLE_GEMINI } from '../../globals.js';
+import { buildGeminiImageGenGenerationConfig, isGeminiImageModel, mapGeminiSizeToAspectRatio, } from '../google-vertex-ai/imageGenerate.js';
+import { generateInvalidProviderResponseError } from '../utils.js';
+import { GoogleErrorResponseTransform } from './chatComplete.js';
+export { isGeminiImageModel };
+/** Request config for Gemini image generation (generateContent with responseModalities). */
+export const GoogleGeminiImageGenConfig = {
+    prompt: {
+        param: 'contents',
+        required: true,
+        transform: (params) => {
+            const promptText = Array.isArray(params.prompt)
+                ? params.prompt.join(' ')
+                : params.prompt;
+            return [{ parts: [{ text: promptText }] }];
+        },
+    },
+    n: {
+        param: 'generationConfig',
+        transform: buildGeminiImageGenGenerationConfig,
+    },
+    candidate_count: {
+        param: 'generationConfig',
+        transform: buildGeminiImageGenGenerationConfig,
+    },
+    size: {
+        param: 'generationConfig',
+        transform: buildGeminiImageGenGenerationConfig,
+    },
+    aspect_ratio: {
+        param: 'generationConfig',
+        transform: buildGeminiImageGenGenerationConfig,
+    },
+    image_size: {
+        param: 'generationConfig',
+        transform: buildGeminiImageGenGenerationConfig,
+    },
+    response_modalities: {
+        param: 'generationConfig',
+        transform: buildGeminiImageGenGenerationConfig,
+    },
+    tools: {
+        param: 'tools',
+    },
+    // Gemini has no end-user identifier field and rejects unknown fields.
+    user: {
+        param: 'user',
+        removeFromParams: true,
+    },
+};
+/** Builds Imagen predict parameters from gateway params. Docs define the contract; we map names and pass through. */
+function transformImagenParams(params) {
+    const aspectRatio = params['aspect_ratio']
+        ? params['aspect_ratio']
+        : mapGeminiSizeToAspectRatio(params['size']);
+    const result = {
+        sampleCount: params['n'] ?? 1,
+        aspectRatio,
+    };
+    if (params['image_size'])
+        result['imageSize'] = params['image_size'];
+    if (params['person_generation'])
+        result['personGeneration'] = params['person_generation'];
+    return result;
+}
+/** Request config for Imagen image generation (predict: instances + parameters). */
+export const GoogleImagenImageGenConfig = {
+    tools: {
+        param: 'tools',
+        transform: () => undefined,
+        removeFromParams: true,
+    },
+    prompt: {
+        param: 'instances',
+        required: true,
+        transform: (params) => {
+            const p = params.prompt;
+            if (Array.isArray(p)) {
+                return p.map(text => ({ prompt: text }));
+            }
+            return [{ prompt: p }];
+        },
+    },
+    n: {
+        param: 'parameters',
+        required: true,
+        default: 1,
+        transform: transformImagenParams,
+    },
+    size: {
+        param: 'parameters',
+        transform: transformImagenParams,
+    },
+    aspect_ratio: {
+        param: 'parameters',
+        transform: transformImagenParams,
+    },
+    image_size: {
+        param: 'parameters',
+        transform: transformImagenParams,
+    },
+    person_generation: {
+        param: 'parameters',
+        transform: transformImagenParams,
+    },
+    // Imagen has no end-user identifier field and rejects unknown fields.
+    user: {
+        param: 'user',
+        removeFromParams: true,
+    },
+};
+/** Maps Gemini usageMetadata to gateway usage (input/output/total tokens). */
+function transformGeminiImageUsage(usageMetadata) {
+    if (!usageMetadata)
+        return undefined;
+    return {
+        input_tokens: usageMetadata.promptTokenCount ?? 0,
+        output_tokens: usageMetadata.candidatesTokenCount ?? 0,
+        total_tokens: usageMetadata.totalTokenCount ?? 0,
+    };
+}
+/** Converts Gemini candidates + inlineData to gateway image response; includes usage when present. */
+function transformGeminiResponse(response, responseStatus) {
+    if (responseStatus !== 200) {
+        const err = GoogleErrorResponseTransform(response, GOOGLE_GEMINI);
+        if (err)
+            return err;
+    }
+    if ('candidates' in response && Array.isArray(response.candidates)) {
+        const data = [];
+        for (const c of response.candidates) {
+            for (const part of c.content?.parts ?? []) {
+                if (part.inlineData?.data)
+                    data.push({ b64_json: part.inlineData.data });
+            }
+        }
+        const result = {
+            created: Math.floor(Date.now() / 1000),
+            data,
+            provider: GOOGLE_GEMINI,
+        };
+        if (response.usageMetadata)
+            result.usage = transformGeminiImageUsage(response.usageMetadata);
+        return result;
+    }
+    return generateInvalidProviderResponseError(response, GOOGLE_GEMINI);
+}
+/** Converts Imagen predictions to gateway image response. */
+function transformImagenResponse(response, responseStatus) {
+    if (responseStatus !== 200) {
+        const err = GoogleErrorResponseTransform(response, GOOGLE_GEMINI);
+        if (err)
+            return err;
+    }
+    if ('predictions' in response && Array.isArray(response.predictions)) {
+        return {
+            created: Math.floor(Date.now() / 1000),
+            data: response.predictions.map(p => ({ b64_json: p.bytesBase64Encoded })),
+            provider: GOOGLE_GEMINI,
+        };
+    }
+    return generateInvalidProviderResponseError(response, GOOGLE_GEMINI);
+}
+/** Response transform: branches on candidates (Gemini) vs predictions (Imagen). */
+export const GoogleImageGenResponseTransform = (response, responseStatus) => {
+    if ('candidates' in response && Array.isArray(response.candidates)) {
+        return transformGeminiResponse(response, responseStatus);
+    }
+    if ('predictions' in response) {
+        return transformImagenResponse(response, responseStatus);
+    }
+    return generateInvalidProviderResponseError(response, GOOGLE_GEMINI);
+};
+//# sourceMappingURL=imageGenerate.js.map

package/dist/providers/google/imageGenerate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"imageGenerate.js","sourceRoot":"","sources":["../../../src/providers/google/imageGenerate.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAE9C,OAAO,EACL,mCAAmC,EAEnC,kBAAkB,EAClB,0BAA0B,GAC3B,MAAM,mCAAmC,CAAC;AAE3C,OAAO,EAAE,oCAAoC,EAAE,MAAM,UAAU,CAAC;AAChE,OAAO,EAAuB,4BAA4B,EAAE,MAAM,gBAAgB,CAAC;AAEnF,OAAO,EAAE,kBAAkB,EAAE,CAAC;AAE9B,4FAA4F;AAC5F,MAAM,CAAC,MAAM,0BAA0B,GAAmB;IACxD,MAAM,EAAE;QACN,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,IAAI;QACd,SAAS,EAAE,CAAC,MAAc,EAA6C,EAAE;YACvE,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC;gBAC7C,CAAC,CAAE,MAAM,CAAC,MAAmB,CAAC,IAAI,CAAC,GAAG,CAAC;gBACvC,CAAC,CAAE,MAAM,CAAC,MAAiB,CAAC;YAC9B,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7C,CAAC;KACF;IACD,CAAC,EAAE;QACD,KAAK,EAAE,kBAAkB;QACzB,SAAS,EAAE,mCAAmC;KAC/C;IACD,eAAe,EAAE;QACf,KAAK,EAAE,kBAAkB;QACzB,SAAS,EAAE,mCAAmC;KAC/C;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,kBAAkB;QACzB,SAAS,EAAE,mCAAmC;KAC/C;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,kBAAkB;QACzB,SAAS,EAAE,mCAAmC;KAC/C;IACD,UAAU,EAAE;QACV,KAAK,EAAE,kBAAkB;QACzB,SAAS,EAAE,mCAAmC;KAC/C;IACD,mBAAmB,EAAE;QACnB,KAAK,EAAE,kBAAkB;QACzB,SAAS,EAAE,mCAAmC;KAC/C;IACD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;KACf;IACD,sEAAsE;IACtE,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;QACb,gBAAgB,EAAE,IAAI;KACvB;CACF,CAAC;AAEF,qHAAqH;AACrH,SAAS,qBAAqB,CAAC,MAA+B;IAC5D,MAAM,WAAW,GAAG,MAAM,CAAC,cAAc,CAAC;QACxC,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC;QACxB,CAAC,CAAC,0BAA0B,CAAC,MAAM,CAAC,MAAM,CAAuB,CAAC,CAAC;IACrE,MAAM,MAAM,GAA4B;QACtC,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;QAC7B,WAAW;KACZ,CAAC;IACF,IAAI,MAAM,CAAC,YAAY,CAAC;QAAE,MAAM,CAAC,WAAW,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;IACrE,IAAI,MAAM,CAAC,mBAAmB,CAAC;QAAE,MAAM,CAAC,kBAAkB,CAAC,GAAG,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAC1F,OAAO,MAAM,CAAC;AAChB,CAAC;AAMD,oFAAoF;AACpF,MAAM,CAAC,MAAM,0BAA0B,GAAmB;IACxD,KAAK,EAAE;QACL,KAAK,EAAE,OAAO;QACd,SAAS,EAAE,GAAG,EAAE,CAAC,SAAS;QAC1B,gBAAgB,EAAE,IAAI;KACvB;IACD,MAAM,EAAE;QACN,KAAK,EAAE,WAAW;QAClB,QAAQ,EAAE,IAAI;QACd,SAAS,EAAE,CAAC,MAAc,EAAoB,EAAE;YAC9C,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC;YACxB,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;gBACrB,OAAQ,CAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACzD,CAAC;YACD,OAAO,CAAC,EAAE,MAAM,EAAE,CAAW,EAAE,CAAC,CAAC;QACnC,CAAC;KACF;IACD,CAAC,EAAE;QACD,KAAK,EAAE,YAAY;QACnB,QAAQ,EAAE,IAAI;QACd,OAAO,EAAE,CAAC;QACV,SAAS,EAAE,qBAAqB;KACjC;IACD,IAAI,EAAE;QACJ,KAAK,EAAE,YAAY;QACnB,SAAS,EAAE,qBAAqB;KACjC;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,YAAY;QACnB,SAAS,EAAE,qBAAqB;KACjC;IACD,UAAU,EAAE;QACV,KAAK,EAAE,YAAY;QACnB,SAAS,EAAE,qBAAqB;KACjC;IACD,iBAAiB,EAAE;QACjB,KAAK,EAAE,YAAY;QACnB,SAAS,EAAE,qBAAqB;KACjC;IACD,sEAAsE;IACtE,IAAI,EAAE;QACJ,KAAK,EAAE,MAAM;QACb,gBAAgB,EAAE,IAAI;KACvB;CACF,CAAC;AAOF,8EAA8E;AAC9E,SAAS,yBAAyB,CAChC,aAA4D;IAE5D,IAAI,CAAC,aAAa;QAAE,OAAO,SAAS,CAAC;IACrC,OAAO;QACL,YAAY,EAAE,aAAa,CAAC,gBAAgB,IAAI,CAAC;QACjD,aAAa,EAAE,aAAa,CAAC,oBAAoB,IAAI,CAAC;QACtD,YAAY,EAAE,aAAa,CAAC,eAAe,IAAI,CAAC;KACjD,CAAC;AACJ,CAAC;AAED,sGAAsG;AACtG,SAAS,uBAAuB,CAC9B,QAAsC,EACtC,cAAsB;IAEtB,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,4BAA4B,CAAC,QAA+B,EAAE,aAAa,CAAC,CAAC;QACzF,IAAI,GAAG;YAAE,OAAO,GAAG,CAAC;IACtB,CAAC;IACD,IAAI,YAAY,IAAI,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QACnE,MAAM,IAAI,GAAiC,EAAE,CAAC;QAC9C,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC;YACpC,KAAK,MAAM,IAAI,IAAI,CAAC,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE,EAAE,CAAC;gBAC1C,IAAI,IAAI,CAAC,UAAU,EAAE,IAAI;oBAAE,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3E,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAqF;YAC/F,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACtC,IAAI;YACJ,QAAQ,EAAE,aAAa;SACxB,CAAC;QACF,IAAI,QAAQ,CAAC,aAAa;YAAE,MAAM,CAAC,KAAK,GAAG,yBAAyB,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QAC7F,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,oCAAoC,CAAC,QAAmC,EAAE,aAAa,CAAC,CAAC;AAClG,CAAC;AAED,6DAA6D;AAC7D,SAAS,uBAAuB,CAC9B,QAAsC,EACtC,cAAsB;IAEtB,IAAI,cAAc,KAAK,GAAG,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,4BAA4B,CAAC,QAA+B,EAAE,aAAa,CAAC,CAAC;QACzF,IAAI,GAAG;YAAE,OAAO,GAAG,CAAC;IACtB,CAAC;IACD,IAAI,aAAa,IAAI,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;QACrE,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;YACtC,IAAI,EAAE,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC;YACzE,QAAQ,EAAE,aAAa;SACxB,CAAC;IACJ,CAAC;IACD,OAAO,oCAAoC,CAAC,QAAmC,EAAE,aAAa,CAAC,CAAC;AAClG,CAAC;AAED,mFAAmF;AACnF,MAAM,CAAC,MAAM,+BAA+B,GAGC,CAAC,QAAQ,EAAE,cAAc,EAAE,EAAE;IACxE,IAAI,YAAY,IAAI,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QACnE,OAAO,uBAAuB,CAAC,QAAwC,EAAE,cAAc,CAAC,CAAC;IAC3F,CAAC;IACD,IAAI,aAAa,IAAI,QAAQ,EAAE,CAAC;QAC9B,OAAO,uBAAuB,CAAC,QAAwC,EAAE,cAAc,CAAC,CAAC;IAC3F,CAAC;IACD,OAAO,oCAAoC,CAAC,QAAmC,EAAE,aAAa,CAAC,CAAC;AAClG,CAAC,CAAC"}

package/dist/providers/google/index.d.ts

@@ -0,0 +1,9 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { ProviderConfigs } from '../types';
+declare const GoogleConfig: ProviderConfigs;
+export default GoogleConfig;
+//# sourceMappingURL=index.d.ts.map

package/dist/providers/google/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/google/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAsB3C,QAAA,MAAM,YAAY,EAAE,eAmCnB,CAAC;AAEF,eAAe,YAAY,CAAC"}

package/dist/providers/google/index.js

@@ -0,0 +1,44 @@
+import GoogleApiConfig from './api.js';
+import { GoogleChatCompleteConfig, GoogleChatCompleteResponseTransform, GoogleChatCompleteStreamChunkTransform, } from './chatComplete.js';
+import { GoogleCreateSpeechConfig, GoogleCreateSpeechRequestHandler, GoogleCreateSpeechResponseTransform, } from './createSpeech.js';
+import { GoogleEmbedConfig, GoogleEmbedResponseTransform } from './embed.js';
+import { GoogleImageEditRequestHandler, GoogleImageEditResponseTransform } from './imageEdit.js';
+import { GoogleGeminiImageGenConfig, GoogleImageGenResponseTransform, GoogleImagenImageGenConfig, isGeminiImageModel, } from './imageGenerate.js';
+import { GoogleChatCompleteRequestTransform } from './responseFormatHandlers.js';
+const GoogleConfig = {
+    api: GoogleApiConfig,
+    requestHandlers: {
+        imageEdit: GoogleImageEditRequestHandler,
+        createSpeech: GoogleCreateSpeechRequestHandler,
+    },
+    getConfig: ({ params }) => {
+        const model = params?.model;
+        const imageGenerate = model !== undefined && model !== ''
+            ? isGeminiImageModel(model)
+                ? GoogleGeminiImageGenConfig
+                : GoogleImagenImageGenConfig
+            : undefined;
+        const imageEdit = model !== undefined && model !== '' && isGeminiImageModel(model) ? {} : undefined;
+        return {
+            chatComplete: GoogleChatCompleteConfig,
+            generateContent: {},
+            embed: GoogleEmbedConfig,
+            imageGenerate,
+            imageEdit,
+            createSpeech: GoogleCreateSpeechConfig,
+            requestTransforms: {
+                chatComplete: GoogleChatCompleteRequestTransform,
+            },
+            responseTransforms: {
+                chatComplete: GoogleChatCompleteResponseTransform,
+                'stream-chatComplete': GoogleChatCompleteStreamChunkTransform,
+                embed: GoogleEmbedResponseTransform,
+                imageGenerate: GoogleImageGenResponseTransform,
+                imageEdit: GoogleImageEditResponseTransform,
+                createSpeech: GoogleCreateSpeechResponseTransform,
+            },
+        };
+    },
+};
+export default GoogleConfig;
+//# sourceMappingURL=index.js.map

package/dist/providers/google/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/google/index.ts"],"names":[],"mappings":"AAMA,OAAO,eAAe,MAAM,OAAO,CAAC;AACpC,OAAO,EACL,wBAAwB,EACxB,mCAAmC,EACnC,sCAAsC,GACvC,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACL,wBAAwB,EACxB,gCAAgC,EAChC,mCAAmC,GACpC,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAC1E,OAAO,EAAE,6BAA6B,EAAE,gCAAgC,EAAE,MAAM,aAAa,CAAC;AAC9F,OAAO,EACL,0BAA0B,EAC1B,+BAA+B,EAC/B,0BAA0B,EAC1B,kBAAkB,GACnB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,kCAAkC,EAAE,MAAM,0BAA0B,CAAC;AAE9E,MAAM,YAAY,GAAoB;IACpC,GAAG,EAAE,eAAe;IACpB,eAAe,EAAE;QACf,SAAS,EAAE,6BAA6B;QACxC,YAAY,EAAE,gCAAgC;KAC/C;IACD,SAAS,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;QACxB,MAAM,KAAK,GAAG,MAAM,EAAE,KAA2B,CAAC;QAClD,MAAM,aAAa,GACjB,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,EAAE;YACjC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC;gBACzB,CAAC,CAAC,0BAA0B;gBAC5B,CAAC,CAAC,0BAA0B;YAC9B,CAAC,CAAC,SAAS,CAAC;QAChB,MAAM,SAAS,GAAG,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,EAAE,IAAI,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QACpG,OAAO;YACL,YAAY,EAAE,wBAAwB;YACtC,eAAe,EAAE,EAAE;YACnB,KAAK,EAAE,iBAAiB;YACxB,aAAa;YACb,SAAS;YACT,YAAY,EAAE,wBAAwB;YACtC,iBAAiB,EAAE;gBACjB,YAAY,EAAE,kCAAkC;aACjD;YACD,kBAAkB,EAAE;gBAClB,YAAY,EAAE,mCAAmC;gBACjD,qBAAqB,EAAE,sCAAsC;gBAC7D,KAAK,EAAE,4BAA4B;gBACnC,aAAa,EAAE,+BAA+B;gBAC9C,SAAS,EAAE,gCAAgC;gBAC3C,YAAY,EAAE,mCAAmC;aAClD;SACF,CAAC;IACJ,CAAC;CACF,CAAC;AAEF,eAAe,YAAY,CAAC"}

package/dist/providers/google/responseFormatHandlers.d.ts

@@ -0,0 +1,14 @@
+/**
+ * ================================
+ * NOTE: TrueFoundry managaed file: Custom response format handlers for Google/Gemini
+ * ================================
+ */
+import { Params } from '../../types/requestBody';
+/**
+ * Custom native handler for Google/Gemini response formats (shared between Google and Google Vertex AI)
+ * Uses Google's native responseMimeType and responseSchema capabilities
+ */
+export declare const googleNativeResponseFormatHandler: (requestBody: Params) => Params;
+export declare const GoogleChatCompleteRequestTransform: (requestBody: Params) => Params;
+export declare const GoogleVertexChatCompleteRequestTransform: (requestBody: Params) => Params;
+//# sourceMappingURL=responseFormatHandlers.d.ts.map

package/dist/providers/google/responseFormatHandlers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"responseFormatHandlers.d.ts","sourceRoot":"","sources":["../../../src/providers/google/responseFormatHandlers.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAIjD;;;GAGG;AACH,eAAO,MAAM,iCAAiC,GAAI,aAAa,MAAM,KAAG,MAYvE,CAAC;AAeF,eAAO,MAAM,kCAAkC,gBATgB,MAAM,KAAG,MASkB,CAAC;AAG3F,eAAO,MAAM,wCAAwC,gBAZU,MAAM,KAAG,MAYwB,CAAC"}

package/dist/providers/google/responseFormatHandlers.js

@@ -0,0 +1,30 @@
+import { transformGenerationConfig } from '../google-vertex-ai/transformGenerationConfig.js';
+import { ResponseFormatSupport, universalResponseFormatRequestTransform } from '../shared/responseFormatUtils.js';
+/**
+ * Custom native handler for Google/Gemini response formats (shared between Google and Google Vertex AI)
+ * Uses Google's native responseMimeType and responseSchema capabilities
+ */
+export const googleNativeResponseFormatHandler = (requestBody) => {
+    // Apply Google's native response format handling via transformGenerationConfig
+    const generationConfig = transformGenerationConfig(requestBody);
+    // Integrate the generationConfig back into the request
+    return {
+        ...requestBody,
+        generationConfig: {
+            ...requestBody.generationConfig,
+            ...generationConfig,
+        },
+    };
+};
+/**
+ * Shared Google request transform using universal response format logic
+ * Google has partial support: native when no tool conflicts, tool conversion otherwise
+ */
+const sharedGoogleChatCompleteRequestTransform = (requestBody) => {
+    return universalResponseFormatRequestTransform(requestBody, ResponseFormatSupport.PARTIAL, googleNativeResponseFormatHandler);
+};
+// Export for Google provider
+export const GoogleChatCompleteRequestTransform = sharedGoogleChatCompleteRequestTransform;
+// Export for Google Vertex AI provider (same function, different name for clarity)
+export const GoogleVertexChatCompleteRequestTransform = sharedGoogleChatCompleteRequestTransform;
+//# sourceMappingURL=responseFormatHandlers.js.map

package/dist/providers/google/responseFormatHandlers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"responseFormatHandlers.js","sourceRoot":"","sources":["../../../src/providers/google/responseFormatHandlers.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,yBAAyB,EAAE,MAAM,+CAA+C,CAAC;AAC1F,OAAO,EAAE,qBAAqB,EAAE,uCAAuC,EAAE,MAAM,+BAA+B,CAAC;AAE/G;;;GAGG;AACH,MAAM,CAAC,MAAM,iCAAiC,GAAG,CAAC,WAAmB,EAAU,EAAE;IAC/E,+EAA+E;IAC/E,MAAM,gBAAgB,GAAG,yBAAyB,CAAC,WAAW,CAAC,CAAC;IAEhE,uDAAuD;IACvD,OAAO;QACL,GAAG,WAAW;QACd,gBAAgB,EAAE;YAChB,GAAG,WAAW,CAAC,gBAAgB;YAC/B,GAAG,gBAAgB;SACpB;KACF,CAAC;AACJ,CAAC,CAAC;AAEF;;;GAGG;AACH,MAAM,wCAAwC,GAAG,CAAC,WAAmB,EAAU,EAAE;IAC/E,OAAO,uCAAuC,CAC5C,WAAW,EACX,qBAAqB,CAAC,OAAO,EAC7B,iCAAiC,CAClC,CAAC;AACJ,CAAC,CAAC;AAEF,6BAA6B;AAC7B,MAAM,CAAC,MAAM,kCAAkC,GAAG,wCAAwC,CAAC;AAE3F,mFAAmF;AACnF,MAAM,CAAC,MAAM,wCAAwC,GAAG,wCAAwC,CAAC"}

package/dist/providers/google/types.d.ts

@@ -0,0 +1,49 @@
+/** Streaming state for Google/Gemini chat completions. */
+export interface GoogleChatStreamState {
+    containsChainOfThoughtMessage?: boolean;
+    /** Running tool-call index across all streaming chunks */
+    cumulativeToolCallIndex?: number;
+}
+/** Delta message shape emitted per SSE chunk in Google/Gemini streaming. */
+export type GoogleStreamDeltaMessage = {
+    role: string;
+    content: string;
+    reasoning_content?: string;
+    thinking_blocks?: Array<{
+        type: 'thinking';
+        thinking: string;
+        signature?: string;
+    }>;
+} | {
+    role: string;
+    tool_calls: Array<{
+        index: number;
+        id: string;
+        type: string;
+        function: {
+            name: string;
+            arguments: string;
+        };
+        provider_specific_fields?: {
+            thought_signature: string;
+        };
+    }>;
+} | {
+    role: string;
+    content_blocks: unknown[];
+};
+export declare enum GOOGLE_GENERATE_CONTENT_FINISH_REASON {
+    FINISH_REASON_UNSPECIFIED = "FINISH_REASON_UNSPECIFIED",
+    STOP = "STOP",
+    MAX_TOKENS = "MAX_TOKENS",
+    SAFETY = "SAFETY",
+    RECITATION = "RECITATION",
+    LANGUAGE = "LANGUAGE",
+    OTHER = "OTHER",
+    BLOCKLIST = "BLOCKLIST",
+    PROHIBITED_CONTENT = "PROHIBITED_CONTENT",
+    SPII = "SPII",
+    MALFORMED_FUNCTION_CALL = "MALFORMED_FUNCTION_CALL",
+    IMAGE_SAFETY = "IMAGE_SAFETY"
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/providers/google/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/providers/google/types.ts"],"names":[],"mappings":"AAAA,0DAA0D;AAC1D,MAAM,WAAW,qBAAqB;IACpC,6BAA6B,CAAC,EAAE,OAAO,CAAC;IACxC,0DAA0D;IAC1D,uBAAuB,CAAC,EAAE,MAAM,CAAC;CAClC;AAED,4EAA4E;AAC5E,MAAM,MAAM,wBAAwB,GAChC;IACE,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,eAAe,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACrF,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,KAAK,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAA;SAAE,CAAC;QAC9C,wBAAwB,CAAC,EAAE;YAAE,iBAAiB,EAAE,MAAM,CAAA;SAAE,CAAC;KAC1D,CAAC,CAAC;CACJ,GACD;IACE,IAAI,EAAE,MAAM,CAAC;IACb,cAAc,EAAE,OAAO,EAAE,CAAC;CAC3B,CAAC;AAEN,oBAAY,qCAAqC;IAC/C,yBAAyB,8BAA8B;IACvD,IAAI,SAAS;IACb,UAAU,eAAe;IACzB,MAAM,WAAW;IACjB,UAAU,eAAe;IACzB,QAAQ,aAAa;IACrB,KAAK,UAAU;IACf,SAAS,cAAc;IACvB,kBAAkB,uBAAuB;IACzC,IAAI,SAAS;IACb,uBAAuB,4BAA4B;IACnD,YAAY,iBAAiB;CAC9B"}

package/dist/providers/google/types.js

@@ -0,0 +1,16 @@
+export var GOOGLE_GENERATE_CONTENT_FINISH_REASON;
+(function (GOOGLE_GENERATE_CONTENT_FINISH_REASON) {
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["FINISH_REASON_UNSPECIFIED"] = "FINISH_REASON_UNSPECIFIED";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["STOP"] = "STOP";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["MAX_TOKENS"] = "MAX_TOKENS";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["SAFETY"] = "SAFETY";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["RECITATION"] = "RECITATION";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["LANGUAGE"] = "LANGUAGE";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["OTHER"] = "OTHER";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["BLOCKLIST"] = "BLOCKLIST";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["PROHIBITED_CONTENT"] = "PROHIBITED_CONTENT";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["SPII"] = "SPII";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["MALFORMED_FUNCTION_CALL"] = "MALFORMED_FUNCTION_CALL";
+    GOOGLE_GENERATE_CONTENT_FINISH_REASON["IMAGE_SAFETY"] = "IMAGE_SAFETY";
+})(GOOGLE_GENERATE_CONTENT_FINISH_REASON || (GOOGLE_GENERATE_CONTENT_FINISH_REASON = {}));
+//# sourceMappingURL=types.js.map

package/dist/providers/google/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/providers/google/types.ts"],"names":[],"mappings":"AA8BA,MAAM,CAAN,IAAY,qCAaX;AAbD,WAAY,qCAAqC;IAC/C,gGAAuD,CAAA;IACvD,sDAAa,CAAA;IACb,kEAAyB,CAAA;IACzB,0DAAiB,CAAA;IACjB,kEAAyB,CAAA;IACzB,8DAAqB,CAAA;IACrB,wDAAe,CAAA;IACf,gEAAuB,CAAA;IACvB,kFAAyC,CAAA;IACzC,sDAAa,CAAA;IACb,4FAAmD,CAAA;IACnD,sEAA6B,CAAA;AAC/B,CAAC,EAbW,qCAAqC,KAArC,qCAAqC,QAahD"}

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

@@ -0,0 +1,15 @@
+import type { ProviderOptions } from '../../types/contracts';
+import type { ProviderHeaderOptions } from '../../types/options';
+import { ProviderAPIConfig } from '../types';
+export declare function resolveVertexFields(providerOptions: ProviderOptions, headerOptions: ProviderHeaderOptions): {
+    vertexProjectId: string;
+    vertexRegion: string;
+    vertexBaseUrl: string | undefined;
+    vertexServiceAccountJson: Record<string, any> | undefined;
+    vertexExternalAccountJson: Record<string, any> | undefined;
+    vertexProviderAccountFqn: string;
+    vertexModelName: string | undefined;
+};
+export declare const GoogleApiConfig: ProviderAPIConfig;
+export default GoogleApiConfig;
+//# sourceMappingURL=api.d.ts.map

package/dist/providers/google-vertex-ai/api.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.d.ts","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/api.ts"],"names":[],"mappings":"AAeA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAMjE,OAAO,EAAY,iBAAiB,EAAE,MAAM,UAAU,CAAC;AA0DvD,wBAAgB,mBAAmB,CAAC,eAAe,EAAE,eAAe,EAAE,aAAa,EAAE,qBAAqB;;;;;;;;EA8BzG;AAyBD,eAAO,MAAM,eAAe,EAAE,iBAwR7B,CAAC;AAEF,eAAe,eAAe,CAAC"}