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

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

@@ -0,0 +1,190 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { GoogleBatchRecord } from './types';
+export declare const GoogleRetrieveBatchResponseTransform: (response: GoogleBatchRecord, status: number) => GoogleBatchRecord | {
+    errors?: {
+        object: string;
+        data: {
+            code: string;
+            message: string;
+        }[];
+    } | undefined;
+    request_counts: {
+        total: number;
+        completed: string | undefined;
+        failed: string | undefined;
+    };
+    failed_at: number;
+    completed_at?: undefined;
+    cancelled_at?: undefined;
+    in_progress_at: number;
+    id: string | undefined;
+    object: string;
+    endpoint: import("../../globals").BatchEndpoints;
+    input_file_id: string;
+    completion_window: null;
+    status: string;
+    output_file_id: string;
+    error_file_id: string;
+    created_at: number;
+} | {
+    errors?: {
+        object: string;
+        data: {
+            code: string;
+            message: string;
+        }[];
+    } | undefined;
+    request_counts: {
+        total: number;
+        completed: string | undefined;
+        failed: string | undefined;
+    };
+    completed_at: number;
+    failed_at: number;
+    cancelled_at?: undefined;
+    in_progress_at: number;
+    id: string | undefined;
+    object: string;
+    endpoint: import("../../globals").BatchEndpoints;
+    input_file_id: string;
+    completion_window: null;
+    status: string;
+    output_file_id: string;
+    error_file_id: string;
+    created_at: number;
+} | {
+    errors?: {
+        object: string;
+        data: {
+            code: string;
+            message: string;
+        }[];
+    } | undefined;
+    request_counts: {
+        total: number;
+        completed: string | undefined;
+        failed: string | undefined;
+    };
+    cancelled_at: number;
+    failed_at: number;
+    completed_at?: undefined;
+    in_progress_at: number;
+    id: string | undefined;
+    object: string;
+    endpoint: import("../../globals").BatchEndpoints;
+    input_file_id: string;
+    completion_window: null;
+    status: string;
+    output_file_id: string;
+    error_file_id: string;
+    created_at: number;
+} | {
+    errors?: {
+        object: string;
+        data: {
+            code: string;
+            message: string;
+        }[];
+    } | undefined;
+    request_counts: {
+        total: number;
+        completed: string | undefined;
+        failed: string | undefined;
+    };
+    completed_at: number;
+    failed_at?: undefined;
+    cancelled_at?: undefined;
+    in_progress_at: number;
+    id: string | undefined;
+    object: string;
+    endpoint: import("../../globals").BatchEndpoints;
+    input_file_id: string;
+    completion_window: null;
+    status: string;
+    output_file_id: string;
+    error_file_id: string;
+    created_at: number;
+} | {
+    errors?: {
+        object: string;
+        data: {
+            code: string;
+            message: string;
+        }[];
+    } | undefined;
+    request_counts: {
+        total: number;
+        completed: string | undefined;
+        failed: string | undefined;
+    };
+    cancelled_at: number;
+    failed_at?: undefined;
+    completed_at: number;
+    in_progress_at: number;
+    id: string | undefined;
+    object: string;
+    endpoint: import("../../globals").BatchEndpoints;
+    input_file_id: string;
+    completion_window: null;
+    status: string;
+    output_file_id: string;
+    error_file_id: string;
+    created_at: number;
+} | {
+    errors?: {
+        object: string;
+        data: {
+            code: string;
+            message: string;
+        }[];
+    } | undefined;
+    request_counts: {
+        total: number;
+        completed: string | undefined;
+        failed: string | undefined;
+    };
+    cancelled_at: number;
+    failed_at?: undefined;
+    completed_at?: undefined;
+    in_progress_at: number;
+    id: string | undefined;
+    object: string;
+    endpoint: import("../../globals").BatchEndpoints;
+    input_file_id: string;
+    completion_window: null;
+    status: string;
+    output_file_id: string;
+    error_file_id: string;
+    created_at: number;
+} | {
+    errors?: {
+        object: string;
+        data: {
+            code: string;
+            message: string;
+        }[];
+    } | undefined;
+    request_counts: {
+        total: number;
+        completed: string | undefined;
+        failed: string | undefined;
+    };
+    failed_at?: undefined;
+    completed_at?: undefined;
+    cancelled_at?: undefined;
+    in_progress_at: number;
+    id: string | undefined;
+    object: string;
+    endpoint: import("../../globals").BatchEndpoints;
+    input_file_id: string;
+    completion_window: null;
+    status: string;
+    output_file_id: string;
+    error_file_id: string;
+    created_at: number;
+};
+//# sourceMappingURL=retrieveBatch.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveBatch.d.ts","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/retrieveBatch.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAG5C,eAAO,MAAM,oCAAoC,GAAI,UAAU,iBAAiB,EAAE,QAAQ,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAM/F,CAAC"}

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

@@ -0,0 +1,14 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+// GoogleToOpenAIBatch
+import { GoogleToOpenAIBatch } from './utils.js';
+export const GoogleRetrieveBatchResponseTransform = (response, status) => {
+    if (status !== 200) {
+        return response;
+    }
+    return GoogleToOpenAIBatch(response);
+};
+//# sourceMappingURL=retrieveBatch.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveBatch.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/retrieveBatch.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,sBAAsB;AAGtB,OAAO,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAE9C,MAAM,CAAC,MAAM,oCAAoC,GAAG,CAAC,QAA2B,EAAE,MAAc,EAAE,EAAE;IAClG,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;QACnB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,OAAO,mBAAmB,CAAC,QAAQ,CAAC,CAAC;AACvC,CAAC,CAAC"}

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

@@ -0,0 +1,4 @@
+import { RequestHandler } from '../types';
+export declare const GoogleRetrieveFileRequestHandler: RequestHandler;
+export declare const GoogleRetrieveFileResponseTransform: (response: Response) => Response;
+//# sourceMappingURL=retrieveFile.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveFile.d.ts","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/retrieveFile.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAI1C,eAAO,MAAM,gCAAgC,EAAE,cA4C9C,CAAC;AAEF,eAAO,MAAM,mCAAmC,GAAI,UAAU,QAAQ,aAErE,CAAC"}

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

@@ -0,0 +1,52 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * But needs some custom changes for TFY
+ * 1. Added object: file to the response
+ * ================================
+ */
+import { safeFetch } from '../../http/index.js';
+import { sdkRequestUrl } from '../../utils/sdkRequestUrl.js';
+import { providerApiArgsFromHandler } from '../requestHandlerApiArgs.js';
+import GoogleApiConfig from './api.js';
+import { getBucketAndFile } from './utils.js';
+export const GoogleRetrieveFileRequestHandler = async (args) => {
+    const gatewayUrl = sdkRequestUrl(args.request);
+    const fileId = args.request.path.split('/').pop();
+    const { bucket, file } = getBucketAndFile(fileId ?? '');
+    const headers = await GoogleApiConfig.headers({
+        ...providerApiArgsFromHandler(args, { endpoint: 'retrieveFile', method: 'HEAD' }),
+        transformedRequestUrl: gatewayUrl,
+        transformedRequestBody: {},
+    });
+    const url = `https://storage.googleapis.com/${bucket}/${file}`;
+    const response = await safeFetch(url, { ssrf: 'guard', headers, method: 'HEAD', signal: args.signal });
+    if (response.status !== 200) {
+        throw new Error('File not found');
+    }
+    const responseHeaders = response.headers;
+    const bytes = responseHeaders.get('Content-Length');
+    const updatedAt = responseHeaders.get('Last-Modified');
+    const createdAt = responseHeaders.get('Date');
+    const id = fileId;
+    const filename = file.split('/').pop();
+    const purpose = null;
+    const status = 'processed';
+    const fileObject = {
+        bytes: Number.parseInt(bytes ?? '0'),
+        updatedAt: new Date(updatedAt ?? '').getTime(),
+        createdAt: new Date(createdAt ?? '').getTime(),
+        object: 'file',
+        id,
+        filename,
+        purpose,
+        status,
+    };
+    return new Response(JSON.stringify(fileObject), {
+        headers: { 'Content-Type': 'application/json' },
+    });
+};
+export const GoogleRetrieveFileResponseTransform = (response) => {
+    return response;
+};
+//# sourceMappingURL=retrieveFile.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveFile.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/retrieveFile.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AACvC,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AAEtE,OAAO,eAAe,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAE3C,MAAM,CAAC,MAAM,gCAAgC,GAAmB,KAAK,EAAC,IAAI,EAAC,EAAE;IAC3E,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;IAElD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,gBAAgB,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;IAExD,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC;QAC5C,GAAG,0BAA0B,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;QACjF,qBAAqB,EAAE,UAAU;QACjC,sBAAsB,EAAE,EAAE;KAC3B,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,kCAAkC,MAAM,IAAI,IAAI,EAAE,CAAC;IAE/D,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IAEvG,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;QAC5B,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,eAAe,GAAG,QAAQ,CAAC,OAAO,CAAC;IAEzC,MAAM,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACpD,MAAM,SAAS,GAAG,eAAe,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;IACvD,MAAM,SAAS,GAAG,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9C,MAAM,EAAE,GAAG,MAAM,CAAC;IAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;IACvC,MAAM,OAAO,GAAG,IAAI,CAAC;IACrB,MAAM,MAAM,GAAG,WAAW,CAAC;IAE3B,MAAM,UAAU,GAAG;QACjB,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK,IAAI,GAAG,CAAC;QACpC,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE;QAC9C,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE;QAC9C,MAAM,EAAE,MAAM;QACd,EAAE;QACF,QAAQ;QACR,OAAO;QACP,MAAM;KACP,CAAC;IAEF,OAAO,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE;QAC9C,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;KAChD,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,mCAAmC,GAAG,CAAC,QAAkB,EAAE,EAAE;IACxE,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC"}

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

@@ -0,0 +1,2 @@
+export declare const GoogleRetrieveFileContentResponseTransform: (response: Response, status: number) => Response | import("../types").ErrorResponse;
+//# sourceMappingURL=retrieveFileContent.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveFileContent.d.ts","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/retrieveFileContent.ts"],"names":[],"mappings":"AAOA,eAAO,MAAM,0CAA0C,GAAI,UAAU,QAAQ,EAAE,QAAQ,MAAM,gDAK5F,CAAC"}

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

@@ -0,0 +1,13 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { GoogleErrorResponseTransform } from './utils.js';
+export const GoogleRetrieveFileContentResponseTransform = (response, status) => {
+    if (status !== 200) {
+        return GoogleErrorResponseTransform(response) || response;
+    }
+    return response;
+};
+//# sourceMappingURL=retrieveFileContent.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveFileContent.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/retrieveFileContent.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAEvD,MAAM,CAAC,MAAM,0CAA0C,GAAG,CAAC,QAAkB,EAAE,MAAc,EAAE,EAAE;IAC/F,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;QACnB,OAAO,4BAA4B,CAAC,QAAe,CAAC,IAAI,QAAQ,CAAC;IACnE,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC,CAAC"}

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

@@ -0,0 +1,20 @@
+export declare const GoogleFinetuneRetrieveResponseTransform: (response: Response, status: number) => Response | {
+    suffix?: string | undefined;
+    validation_file?: string | undefined;
+    hyperparameters: {
+        batch_size: number;
+        learning_rate_multiplier: number;
+        n_epochs: number;
+    };
+    model: string;
+    trained_tokens: number | undefined;
+    training_file: string;
+    finished_at?: number | undefined;
+    id: string | undefined;
+    object: string;
+    status: string;
+    created_at: number;
+    error: string;
+    fine_tuned_model: string | undefined;
+};
+//# sourceMappingURL=retrieveFinetune.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveFinetune.d.ts","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/retrieveFinetune.ts"],"names":[],"mappings":"AAQA,eAAO,MAAM,uCAAuC,GAAI,UAAU,QAAQ,EAAE,QAAQ,MAAM;;;;;;;;;;;;;;;;;;CAMzF,CAAC"}

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

@@ -0,0 +1,8 @@
+import { GoogleToOpenAIFinetune } from './utils.js';
+export const GoogleFinetuneRetrieveResponseTransform = (response, status) => {
+    if (status !== 200) {
+        return response;
+    }
+    return GoogleToOpenAIFinetune(response);
+};
+//# sourceMappingURL=retrieveFinetune.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieveFinetune.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/retrieveFinetune.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAEjD,MAAM,CAAC,MAAM,uCAAuC,GAAG,CAAC,QAAkB,EAAE,MAAc,EAAE,EAAE;IAC5F,IAAI,MAAM,KAAK,GAAG,EAAE,CAAC;QACnB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,OAAO,sBAAsB,CAAC,QAA2C,CAAC,CAAC;AAC7E,CAAC,CAAC"}

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

@@ -0,0 +1,15 @@
+/**
+ * ================================
+ * Inherited upstream file
+ * ================================
+ */
+import { Params } from '../../types/requestBody';
+import { GoogleEmbedParams } from './embed';
+import { EmbedInstancesData } from './types';
+/**
+ * @see https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/gemini#request_body
+ */
+export declare function transformGenerationConfig(params: Params): Record<string, unknown>;
+export declare function transformEmbeddingsParameters(params: GoogleEmbedParams): Record<string, unknown>;
+export declare function transformEmbeddingInputs(params: GoogleEmbedParams): EmbedInstancesData[];
+//# sourceMappingURL=transformGenerationConfig.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"transformGenerationConfig.d.ts","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/transformGenerationConfig.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEjD,OAAO,EAAE,iBAAiB,EAA0B,MAAM,SAAS,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAS7C;;GAEG;AACH,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,MAAM,2BAyEvD;AAED,wBAAgB,6BAA6B,CAAC,MAAM,EAAE,iBAAiB,2BAetE;AAED,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,iBAAiB,wBAuCjE"}

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

@@ -0,0 +1,136 @@
+import { calculateBudgetTokens } from '../../utils/reasoningUtils.js';
+import { isMultimodalEmbedModel } from './embed.js';
+import { budgetTokensToThinkingLevel, buildGeminiThinkingConfig, recursivelyDeleteUnsupportedParameters, supportsThinking, supportsThinkingLevel, transformGeminiToolParameters, } from './utils.js';
+/**
+ * @see https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/gemini#request_body
+ */
+export function transformGenerationConfig(params) {
+    const generationConfig = {};
+    if (params['temperature'] != null) {
+        generationConfig['temperature'] = params['temperature'];
+    }
+    if (params['top_p'] != null) {
+        generationConfig['topP'] = params['top_p'];
+    }
+    if (params['top_k'] != null) {
+        generationConfig['topK'] = params['top_k'];
+    }
+    if (params['max_tokens'] != null) {
+        generationConfig['maxOutputTokens'] = params['max_tokens'];
+    }
+    if (params['max_completion_tokens'] != null) {
+        generationConfig['maxOutputTokens'] = params['max_completion_tokens'];
+    }
+    if (params['stop']) {
+        generationConfig['stopSequences'] = params['stop'];
+    }
+    if (params?.response_format?.type === 'json_object') {
+        generationConfig['responseMimeType'] = 'application/json';
+    }
+    if (params['logprobs']) {
+        generationConfig['responseLogprobs'] = params['logprobs'];
+    }
+    if (params['top_logprobs']) {
+        generationConfig['logprobs'] = params['top_logprobs']; // range 1-5, openai supports 1-20
+    }
+    if (params['seed']) {
+        generationConfig['seed'] = params['seed'];
+    }
+    if (params?.response_format?.type === 'json_schema') {
+        generationConfig['responseMimeType'] = 'application/json';
+        const schema = params?.response_format?.json_schema?.schema ?? params?.response_format?.json_schema;
+        recursivelyDeleteUnsupportedParameters(schema);
+        generationConfig['responseSchema'] = transformGeminiToolParameters(schema);
+    }
+    const model = params.model ?? '';
+    if (supportsThinking(model)) {
+        const thinkingLevelSupported = supportsThinkingLevel(model);
+        if (params?.thinking) {
+            const { budget_tokens, type } = params.thinking;
+            const enabled = type === 'enabled' && (budget_tokens === undefined || budget_tokens > 0);
+            if (thinkingLevelSupported) {
+                generationConfig['thinkingConfig'] = {
+                    ...(enabled && { thinkingLevel: budgetTokensToThinkingLevel(budget_tokens, model) }),
+                    includeThoughts: enabled,
+                };
+            }
+            else {
+                generationConfig['thinkingConfig'] = {
+                    thinkingBudget: budget_tokens,
+                    includeThoughts: enabled,
+                };
+            }
+        }
+        else if (params?.reasoning_effort) {
+            if (thinkingLevelSupported) {
+                generationConfig['thinkingConfig'] = buildGeminiThinkingConfig(params.reasoning_effort, model);
+            }
+            else {
+                const maxTokens = params.max_completion_tokens || params.max_tokens || 2048;
+                const budgetTokens = calculateBudgetTokens(maxTokens, params.reasoning_effort);
+                generationConfig['thinkingConfig'] = {
+                    thinkingBudget: budgetTokens,
+                    includeThoughts: params.reasoning_effort !== 'none',
+                };
+            }
+        }
+    }
+    return generationConfig;
+}
+export function transformEmbeddingsParameters(params) {
+    let embeddingsParameters = {};
+    if (params['parameters'] && typeof params['parameters'] === 'object') {
+        embeddingsParameters = { ...params['parameters'] };
+    }
+    if (params['dimensions']) {
+        // for multimodal embeddings, the parameter is dimension
+        if (Array.isArray(params.input) && typeof params.input[0] === 'object') {
+            embeddingsParameters['dimension'] = params['dimensions'];
+        }
+        else {
+            embeddingsParameters['outputDimensionality'] = params['dimensions'];
+        }
+    }
+    return embeddingsParameters;
+}
+export function transformEmbeddingInputs(params) {
+    const instances = [];
+    // Multimodal embed models (e.g. `multimodalembedding-001`) expect `{text}` on
+    // `:predict` instances; classic text-embedding models expect `{content, task_type}`.
+    const isMultimodal = isMultimodalEmbedModel(params.model ?? '');
+    const wrapStringInput = (s) => isMultimodal ? { text: s } : { content: s, task_type: params.task_type };
+    if (Array.isArray(params.input)) {
+        params.input.forEach(input => {
+            if (typeof input === 'string') {
+                instances.push(wrapStringInput(input));
+            }
+            else if (typeof input === 'object') {
+                instances.push({
+                    text: input.text,
+                    ...(input.image && {
+                        image: {
+                            gcsUri: input.image.url,
+                            bytesBase64Encoded: input.image.base64,
+                        },
+                    }),
+                    ...(input.video && {
+                        video: {
+                            gcsUri: input.video.url,
+                            bytesBase64Encoded: input.video.base64,
+                            videoSegmentConfig: {
+                                startOffsetSec: input.video.start_offset,
+                                endOffsetSec: input.video.end_offset,
+                                intervalSec: input.video.interval,
+                            },
+                        },
+                    }),
+                });
+            }
+        });
+    }
+    else if (typeof params.input === 'string') {
+        instances.push(wrapStringInput(params.input));
+    }
+    return instances;
+}
+//# sourceMappingURL=transformGenerationConfig.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"transformGenerationConfig.js","sourceRoot":"","sources":["../../../src/providers/google-vertex-ai/transformGenerationConfig.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAqB,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAEpE,OAAO,EACL,2BAA2B,EAC3B,yBAAyB,EACzB,sCAAsC,EACtC,gBAAgB,EAChB,qBAAqB,EACrB,6BAA6B,GAC9B,MAAM,SAAS,CAAC;AACjB;;GAEG;AACH,MAAM,UAAU,yBAAyB,CAAC,MAAc;IACtD,MAAM,gBAAgB,GAA4B,EAAE,CAAC;IACrD,IAAI,MAAM,CAAC,aAAa,CAAC,IAAI,IAAI,EAAE,CAAC;QAClC,gBAAgB,CAAC,aAAa,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;IAC1D,CAAC;IACD,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;QAC5B,gBAAgB,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;IAC7C,CAAC;IACD,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC;QAC5B,gBAAgB,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;IAC7C,CAAC;IACD,IAAI,MAAM,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAAC;QACjC,gBAAgB,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;IAC7D,CAAC;IACD,IAAI,MAAM,CAAC,uBAAuB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC5C,gBAAgB,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC,uBAAuB,CAAC,CAAC;IACxE,CAAC;IACD,IAAI,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QACnB,gBAAgB,CAAC,eAAe,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IACrD,CAAC;IACD,IAAI,MAAM,EAAE,eAAe,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;QACpD,gBAAgB,CAAC,kBAAkB,CAAC,GAAG,kBAAkB,CAAC;IAC5D,CAAC;IACD,IAAI,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;QACvB,gBAAgB,CAAC,kBAAkB,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;IAC5D,CAAC;IACD,IAAI,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;QAC3B,gBAAgB,CAAC,UAAU,CAAC,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,kCAAkC;IAC3F,CAAC;IACD,IAAI,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QACnB,gBAAgB,CAAC,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,MAAM,EAAE,eAAe,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;QACpD,gBAAgB,CAAC,kBAAkB,CAAC,GAAG,kBAAkB,CAAC;QAC1D,MAAM,MAAM,GAAG,MAAM,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,IAAI,MAAM,EAAE,eAAe,EAAE,WAAW,CAAC;QACpG,sCAAsC,CAAC,MAAM,CAAC,CAAC;QAC/C,gBAAgB,CAAC,gBAAgB,CAAC,GAAG,6BAA6B,CAAC,MAAM,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;IAEjC,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,sBAAsB,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAE5D,IAAI,MAAM,EAAE,QAAQ,EAAE,CAAC;YACrB,MAAM,EAAE,aAAa,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC;YAChD,MAAM,OAAO,GAAG,IAAI,KAAK,SAAS,IAAI,CAAC,aAAa,KAAK,SAAS,IAAI,aAAa,GAAG,CAAC,CAAC,CAAC;YACzF,IAAI,sBAAsB,EAAE,CAAC;gBAC3B,gBAAgB,CAAC,gBAAgB,CAAC,GAAG;oBACnC,GAAG,CAAC,OAAO,IAAI,EAAE,aAAa,EAAE,2BAA2B,CAAC,aAAa,EAAE,KAAK,CAAC,EAAE,CAAC;oBACpF,eAAe,EAAE,OAAO;iBACzB,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,gBAAgB,CAAC,gBAAgB,CAAC,GAAG;oBACnC,cAAc,EAAE,aAAa;oBAC7B,eAAe,EAAE,OAAO;iBACzB,CAAC;YACJ,CAAC;QACH,CAAC;aAAM,IAAI,MAAM,EAAE,gBAAgB,EAAE,CAAC;YACpC,IAAI,sBAAsB,EAAE,CAAC;gBAC3B,gBAAgB,CAAC,gBAAgB,CAAC,GAAG,yBAAyB,CAAC,MAAM,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;YACjG,CAAC;iBAAM,CAAC;gBACN,MAAM,SAAS,GAAG,MAAM,CAAC,qBAAqB,IAAI,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC;gBAC5E,MAAM,YAAY,GAAG,qBAAqB,CAAC,SAAS,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;gBAC/E,gBAAgB,CAAC,gBAAgB,CAAC,GAAG;oBACnC,cAAc,EAAE,YAAY;oBAC5B,eAAe,EAAE,MAAM,CAAC,gBAAgB,KAAK,MAAM;iBACpD,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,MAAyB;IACrE,IAAI,oBAAoB,GAA4B,EAAE,CAAC;IACvD,IAAI,MAAM,CAAC,YAAY,CAAC,IAAI,OAAO,MAAM,CAAC,YAAY,CAAC,KAAK,QAAQ,EAAE,CAAC;QACrE,oBAAoB,GAAG,EAAE,GAAG,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;IACrD,CAAC;IACD,IAAI,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;QACzB,wDAAwD;QACxD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;YACvE,oBAAoB,CAAC,WAAW,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;QAC3D,CAAC;aAAM,CAAC;YACN,oBAAoB,CAAC,sBAAsB,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAED,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAyB;IAChE,MAAM,SAAS,GAAyB,EAAE,CAAC;IAC3C,8EAA8E;IAC9E,qFAAqF;IACrF,MAAM,YAAY,GAAG,sBAAsB,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAChE,MAAM,eAAe,GAAG,CAAC,CAAS,EAAsB,EAAE,CACxD,YAAY,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC;IAE3E,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QAChC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;YAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;YACzC,CAAC;iBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI;wBACjB,KAAK,EAAE;4BACL,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG;4BACvB,kBAAkB,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM;yBACvC;qBACF,CAAC;oBACF,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI;wBACjB,KAAK,EAAE;4BACL,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG;4BACvB,kBAAkB,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM;4BACtC,kBAAkB,EAAE;gCAClB,cAAc,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY;gCACxC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,UAAU;gCACpC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ;6BAClC;yBACF;qBACF,CAAC;iBACH,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;SAAM,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5C,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}