n8n-nodes-vercel-ai-sdk-universal-temp 0.1.68 → 0.1.70

package/dist/nodes/UniversalAI/UniversalAI.node.js

@@ -42,135 +42,7 @@ const ai_1 = require("ai");
 const zod_1 = require("zod");
 const ajv_1 = __importDefault(require("ajv"));
 const descriptions_1 = require("../shared/descriptions");
-class Cache {
-    constructor(maxSize = 100, ttl = 5 * 60 * 1000) {
-        this.cache = new Map();
-        this.totalHits = 0;
-        this.totalMisses = 0;
-        this.totalEvictions = 0;
-        this.maxSize = maxSize;
-        this.ttl = ttl;
-    }
-    get(key) {
-        const item = this.cache.get(key);
-        if (!item) {
-            this.totalMisses++;
-            return undefined;
-        }
-        const now = Date.now();
-        if (item.expiresAt && now > item.expiresAt) {
-            this.cache.delete(key);
-            this.totalEvictions++;
-            this.totalMisses++;
-            return undefined;
-        }
-        item.hits++;
-        this.totalHits++;
-        return item.value;
-    }
-    set(key, value, customTTL) {
-        const now = Date.now();
-        const expiresAt = customTTL ? now + customTTL : (this.ttl > 0 ? now + this.ttl : undefined);
-        if (this.cache.size >= this.maxSize) {
-            let oldestKey;
-            let oldestTime = now;
-            for (const [k, v] of this.cache.entries()) {
-                if (v.timestamp < oldestTime) {
-                    oldestTime = v.timestamp;
-                    oldestKey = k;
-                }
-            }
-            if (oldestKey) {
-                this.cache.delete(oldestKey);
-                this.totalEvictions++;
-            }
-        }
-        this.cache.set(key, { value, timestamp: now, hits: 0, expiresAt });
-    }
-    delete(key) {
-        return this.cache.delete(key);
-    }
-    clear() {
-        this.cache.clear();
-        this.totalHits = 0;
-        this.totalMisses = 0;
-        this.totalEvictions = 0;
-    }
-    getStats() {
-        return {
-            size: this.cache.size,
-            maxSize: this.maxSize,
-            hitRate: this.totalHits / (this.totalHits + this.totalMisses) || 0,
-            totalHits: this.totalHits,
-            totalMisses: this.totalMisses,
-            totalEvictions: this.totalEvictions,
-            ttl: this.ttl,
-        };
-    }
-}
-const modelCache = new Cache(50);
-const providerCache = new Cache(20);
-const schemaCache = new Cache(30);
-const googleCacheClients = new Cache(10, 60 * 60 * 1000);
-const googleCachedContexts = new Cache(50, 55 * 60 * 1000);
-async function getGoogleCacheManager(apiKey) {
-    let client = googleCacheClients.get(apiKey);
-    if (!client) {
-        const { GoogleGenAI } = await Promise.resolve().then(() => __importStar(require('@google/genai')));
-        client = new GoogleGenAI({ apiKey });
-        googleCacheClients.set(apiKey, client);
-    }
-    return client;
-}
-async function createGoogleCache(exec, index, apiKey, cacheableContent, modelId, thinkingBudgetValue, includeThoughts, tools, ttlSeconds) {
-    try {
-        if (!cacheableContent || cacheableContent.length === 0) {
-            return null;
-        }
-        const googleCacheManager = await getGoogleCacheManager(apiKey);
-        const cacheKeyData = {
-            content: JSON.stringify(cacheableContent),
-            tools: tools ? Object.keys(tools).sort() : [],
-            model: modelId,
-        };
-        const cacheKey = JSON.stringify(cacheKeyData, Object.keys(cacheKeyData).sort());
-        const existingCache = googleCachedContexts.get(cacheKey);
-        if (existingCache) {
-            console.log(`UniversalAI: Reusing existing cache: ${existingCache.name}`);
-            return existingCache.name;
-        }
-        const ttl = ttlSeconds || 3600;
-        const displayName = `universal_ai_cache_${Date.now()}`;
-        const cacheConfig = {
-            model: modelId,
-            config: {
-                displayName,
-                ttl: `${ttl}s`,
-                contents: cacheableContent,
-            },
-        };
-        if (tools && Object.keys(tools).length > 0) {
-            cacheConfig.config.tools = Object.values(tools);
-        }
-        console.log(`UniversalAI: Creating new cache...`);
-        const result = await googleCacheManager.caches.create(cacheConfig);
-        const cachedContentName = result === null || result === void 0 ? void 0 : result.name;
-        if (!cachedContentName) {
-            throw new Error('Cache creation returned no name identifier');
-        }
-        console.log(`UniversalAI: Cache created successfully: ${cachedContentName}`);
-        googleCachedContexts.set(cacheKey, { name: cachedContentName }, (ttl - 300) * 1000);
-        return cachedContentName;
-    }
-    catch (error) {
-        const errorMessage = error instanceof Error ? error.message : String(error);
-        console.error('UniversalAI: Google cache creation failed:', errorMessage);
-        if (errorMessage.includes('INVALID_ARGUMENT') || errorMessage.includes('content is too small')) {
-            console.warn('UniversalAI: Content may be below minimum token requirement (1,024 tokens (~4800 symbols) for 2.5 Flash and 2,048 tokens (~9600 symbols) for 2.5 Pro)');
-        }
-        return null;
-    }
-}
+const helpers_1 = require("../shared/helpers");
 const messageSchema = zod_1.z.object({
     role: zod_1.z.enum(['system', 'user', 'assistant']),
     content: zod_1.z.any(),
@@ -308,78 +180,75 @@ async function buildMessageWithAttachments(role, content, attachments, itemBinar
     if (content) {
         parts.push({ type: 'text', text: content });
     }
-    const MAX_CONCURRENT_ATTACHMENTS = 3;
-    const processedAttachments = [];
-    for (let i = 0; i < attachments.length; i += MAX_CONCURRENT_ATTACHMENTS) {
-        const batch = attachments.slice(i, i + MAX_CONCURRENT_ATTACHMENTS);
-        const batchPromises = batch.map(attachment => processAttachment(attachment, itemBinary, exec, itemIndex, shouldCacheMessage, cacheableContent));
-        const batchResults = await Promise.all(batchPromises);
-        processedAttachments.push(...batchResults);
-    }
-    for (const attachment of processedAttachments) {
-        if (attachment) {
-            parts.push(attachment);
+    for (const attachment of attachments) {
+        const filePart = await processAttachment(attachment, itemBinary, exec, itemIndex, shouldCacheMessage, cacheableContent);
+        if (filePart) {
+            parts.push(filePart);
         }
     }
     return parts.length > 0 ? { role, content: parts } : null;
 }
 async function processAttachment(attachment, itemBinary, exec, itemIndex, shouldCacheMessage, cacheableContent) {
-    const fileContentInput = attachment.fileContent;
-    if (!fileContentInput || typeof fileContentInput !== 'string')
+    const fileResult = await processFileData(attachment.fileContent, itemBinary, attachment, cacheableContent, exec, itemIndex);
+    if (!fileResult)
         return null;
-    let mimeType = getMimeType(attachment);
-    let fileData;
-    if (isUrl(fileContentInput)) {
-        fileData = fileContentInput;
-        if (attachment.cacheAttachment === true) {
-            cacheableContent.push({
-                role: 'user',
-                parts: [{ fileUri: fileContentInput, mimeType: mimeType || 'application/octet-stream' }],
-            });
-            return null;
-        }
-    }
-    else {
-        const result = await getBinaryData(fileContentInput, itemBinary, exec, itemIndex);
-        fileData = result.data;
-        if (!mimeType && result.mimeType) {
-            mimeType = result.mimeType;
-        }
-    }
-    if (!fileData || (Buffer.isBuffer(fileData) && fileData.length === 0)) {
-        return null;
-    }
     return {
         type: 'file',
-        data: fileData,
-        mediaType: mimeType || 'application/octet-stream',
+        data: fileResult.data,
+        mediaType: fileResult.mimeType,
     };
 }
-function getMimeType(attachment) {
-    return attachment.mimeType === 'other'
-        ? attachment.mimeTypeOther
-        : attachment.mimeType;
-}
-async function getBinaryData(fileContentInput, itemBinary, exec, itemIndex) {
-    if (itemBinary === null || itemBinary === void 0 ? void 0 : itemBinary[fileContentInput]) {
-        const binaryData = itemBinary[fileContentInput];
-        return {
-            data: Buffer.from(binaryData.data, 'base64'),
-            mimeType: binaryData.mimeType,
-        };
+async function processFileData(fileContentInput, itemBinary, attachment, cacheableContent, exec, itemIndex) {
+    if (!fileContentInput || typeof fileContentInput !== 'string') {
+        return null;
     }
+    let mimeType = attachment.mimeType === 'other' ? attachment.mimeTypeOther : attachment.mimeType;
+    let fileData;
+    let shouldCache = attachment.cacheAttachment === true;
     try {
-        if (isLikelyBase64(fileContentInput)) {
-            const buffer = Buffer.from(fileContentInput, 'base64');
-            if (buffer.length > 0 && buffer.length < 50 * 1024 * 1024) {
-                return { data: buffer, mimeType: undefined };
+        if (isUrl(fileContentInput)) {
+            fileData = fileContentInput;
+            if (shouldCache) {
+                cacheableContent.push({
+                    role: 'user',
+                    parts: [{ fileUri: fileContentInput, mimeType: mimeType || 'application/octet-stream' }],
+                });
+                return null;
+            }
+        }
+        else if (fileContentInput.startsWith('data:')) {
+            const dataUriMatch = fileContentInput.match(/^data:([^;]+);base64,(.+)$/);
+            if (!dataUriMatch)
+                return null;
+            const [, extractedMimeType, base64Data] = dataUriMatch;
+            fileData = Buffer.from(base64Data, 'base64');
+            if (fileData.length > 20 * 1024 * 1024) {
+                throw new Error('Base64 data exceeds 20MB limit');
+            }
+            mimeType = mimeType || extractedMimeType;
+        }
+        else if (isLikelyBase64(fileContentInput)) {
+            fileData = Buffer.from(fileContentInput, 'base64');
+            if (fileData.length > 20 * 1024 * 1024) {
+                throw new Error('Base64 data exceeds 20MB limit');
             }
         }
+        else if (itemBinary === null || itemBinary === void 0 ? void 0 : itemBinary[fileContentInput]) {
+            const binaryData = itemBinary[fileContentInput];
+            fileData = Buffer.from(binaryData.data, 'base64');
+            mimeType = mimeType || binaryData.mimeType;
+        }
+        else {
+            return null;
+        }
+        if (!fileData || (Buffer.isBuffer(fileData) && fileData.length === 0)) {
+            return null;
+        }
+        return { data: fileData, mimeType: mimeType || 'application/octet-stream', shouldCache };
     }
     catch (error) {
         throw new n8n_workflow_1.NodeOperationError(exec.getNode(), `Invalid file content for attachment: ${error.message}`);
     }
-    return { data: null, mimeType: undefined };
 }
 function formatTextResult(result, includeRequestBody, provider) {
     var _a, _b, _c, _d, _e;
@@ -511,52 +380,9 @@ function formatResponse(result) {
         headers: (_d = result.response) === null || _d === void 0 ? void 0 : _d.headers,
     };
 }
-async function getProvider(provider, apiKey, baseURL, customHeaders) {
-    const headersKey = customHeaders
-        ? JSON.stringify(Object.keys(customHeaders)
-            .sort()
-            .map((key) => [key, customHeaders[key]]))
-        : '';
-    const cacheKey = `${provider}:${apiKey}:${baseURL || ''}:${headersKey}`;
-    const cached = providerCache.get(cacheKey);
-    if (cached)
-        return cached;
-    let providerInstance;
-    try {
-        switch (provider) {
-            case 'google':
-                const { createGoogleGenerativeAI } = await Promise.resolve().then(() => __importStar(require('@ai-sdk/google')));
-                providerInstance = createGoogleGenerativeAI({
-                    apiKey,
-                    ...(baseURL && { baseURL }),
-                    ...(customHeaders && Object.keys(customHeaders).length > 0 && { headers: customHeaders }),
-                });
-                break;
-            case 'deepseek':
-                const { createDeepSeek } = await Promise.resolve().then(() => __importStar(require('@ai-sdk/deepseek')));
-                providerInstance = createDeepSeek({ apiKey, ...(baseURL && { baseURL }) });
-                break;
-            case 'groq':
-                const { createGroq } = await Promise.resolve().then(() => __importStar(require('@ai-sdk/groq')));
-                providerInstance = createGroq({ apiKey, ...(baseURL && { baseURL }) });
-                break;
-            case 'openrouter':
-                const { createOpenRouter } = await Promise.resolve().then(() => __importStar(require('@openrouter/ai-sdk-provider')));
-                providerInstance = createOpenRouter({ apiKey, ...(baseURL && { baseURL }) });
-                break;
-            default:
-                throw new Error(`Unsupported provider: ${provider}`);
-        }
-        providerCache.set(cacheKey, providerInstance);
-        return providerInstance;
-    }
-    catch (error) {
-        throw new Error(`Failed to initialize ${provider} provider: ${error.message}`);
-    }
-}
 function parseAndValidateSchema(rawSchema, exec) {
     const cacheKey = `schema:${Buffer.from(rawSchema).toString('base64').substring(0, 50)}`;
-    const cached = schemaCache.get(cacheKey);
+    const cached = helpers_1.schemaCache.get(cacheKey);
     if (cached)
         return cached;
     let parsedSchema;
@@ -569,7 +395,7 @@ function parseAndValidateSchema(rawSchema, exec) {
     if (!ajv.validateSchema(parsedSchema)) {
         throw new n8n_workflow_1.NodeOperationError(exec.getNode(), `Invalid JSON Schema: ${ajv.errorsText(ajv.errors)}`);
     }
-    schemaCache.set(cacheKey, parsedSchema);
+    helpers_1.schemaCache.set(cacheKey, parsedSchema);
     return parsedSchema;
 }
 function parseStopSequences(stopSequencesStr) {
@@ -593,17 +419,17 @@ class UniversalAI {
                 async getModels() {
                     const provider = this.getCurrentNodeParameter('provider');
                     const cacheKey = `models:${provider}`;
-                    const cached = modelCache.get(cacheKey);
+                    const cached = helpers_1.modelCache.get(cacheKey);
                     if (cached)
                         return cached;
-                    const { OPENROUTER_MODELS, GOOGLE_GEMINI_MODELS, DEEPSEEK_MODELS, GROQ_MODELS } = await Promise.resolve().then(() => __importStar(require('./model-lists')));
+                    const { OPENROUTER_MODELS, GOOGLE_GEMINI_MODELS, DEEPSEEK_MODELS, GROQ_MODELS } = await Promise.resolve().then(() => __importStar(require('../shared//model-lists')));
                     const models = {
                         google: GOOGLE_GEMINI_MODELS,
                         deepseek: DEEPSEEK_MODELS,
                         groq: GROQ_MODELS,
                         openrouter: OPENROUTER_MODELS,
                     }[provider] || [];
-                    modelCache.set(cacheKey, models);
+                    helpers_1.modelCache.set(cacheKey, models);
                     return models;
                 },
             },
@@ -613,25 +439,13 @@ class UniversalAI {
         const items = this.getInputData();
         const returnData = [];
         const provider = this.getNodeParameter('provider', 0);
-        const credentialType = {
-            google: 'googleGenerativeAIApi',
-            deepseek: 'deepSeekApi',
-            groq: 'groqApi',
-            openrouter: 'openRouterApi',
-        }[provider];
-        if (!credentialType) {
-            throw new n8n_workflow_1.NodeOperationError(this.getNode(), `Unsupported provider: ${provider}`);
-        }
-        const credentials = await this.getCredentials(credentialType);
-        if (!(credentials === null || credentials === void 0 ? void 0 : credentials.apiKey)) {
-            throw new n8n_workflow_1.NodeOperationError(this.getNode(), 'No API key provided in credentials');
-        }
+        const { apiKey, baseUrl } = await (0, helpers_1.getCredentials)(this, provider);
         const customHeaders = provider === 'google' ? getGoogleCustomHeaders(this, 0) : undefined;
-        const aiProvider = await getProvider(provider, credentials.apiKey, credentials.baseUrl, customHeaders);
+        const aiProvider = await (0, helpers_1.getProvider)(this, provider, apiKey, baseUrl, customHeaders);
         for (let i = 0; i < items.length; i++) {
             if (this.continueOnFail()) {
                 try {
-                    const result = await processItem(this, i, provider, aiProvider, credentials.apiKey);
+                    const result = await processItem(this, i, provider, aiProvider, apiKey);
                     returnData.push(...result);
                 }
                 catch (error) {
@@ -643,7 +457,7 @@ class UniversalAI {
                 }
             }
             else {
-                const result = await processItem(this, i, provider, aiProvider, credentials.apiKey);
+                const result = await processItem(this, i, provider, aiProvider, apiKey);
                 returnData.push(...result);
             }
         }
@@ -683,21 +497,6 @@ function getModelSettings(exec, index, provider, operation, options) {
     }
     return settings;
 }
-function buildGoogleProviderOptions(exec, index, cachedContentName, thinkingBudgetOverride, includeThoughtsOverride) {
-    const thinkingBudgetValue = thinkingBudgetOverride !== null && thinkingBudgetOverride !== void 0 ? thinkingBudgetOverride : Number(exec.getNodeParameter('thinkingBudget', index, -1));
-    const includeThoughts = includeThoughtsOverride !== null && includeThoughtsOverride !== void 0 ? includeThoughtsOverride : exec.getNodeParameter('includeThoughts', index, false);
-    const options = {};
-    if (!Number.isNaN(thinkingBudgetValue) && thinkingBudgetValue > -1) {
-        options.thinkingConfig = {
-            thinkingBudget: Math.max(0, thinkingBudgetValue),
-            includeThoughts,
-        };
-    }
-    if (cachedContentName) {
-        options.cachedContent = cachedContentName;
-    }
-    return Object.keys(options).length > 0 ? options : undefined;
-}
 function getGoogleCustomHeaders(exec, index) {
     var _a, _b;
     const headersCollection = exec.getNodeParameter('customHeaders', index, {});
@@ -719,30 +518,9 @@ function getGoogleCustomHeaders(exec, index) {
 async function generateTextOperation(exec, index, provider, aiProvider, model, modelSettings, input, options, apiKey) {
     const enableStreaming = exec.getNodeParameter('enableStreaming', index, false);
     const includeRequestBody = options.includeRequestBody;
-    const tools = provider === 'google' ? await buildGoogleTools(exec, index) : undefined;
-    let googleProviderOptions;
-    let thinkingBudgetValue;
-    let includeThoughts;
-    if (provider === 'google') {
-        thinkingBudgetValue = Number(exec.getNodeParameter('thinkingBudget', index, -1));
-        includeThoughts = exec.getNodeParameter('includeThoughts', index, false);
-    }
-    if (provider === 'google' &&
-        input.cacheableContent &&
-        input.cacheableContent.length > 0) {
-        const cacheTTL = exec.getNodeParameter('cacheTTL', index, 3600);
-        const cachedContentName = await createGoogleCache(exec, index, apiKey, input.cacheableContent, model, thinkingBudgetValue !== null && thinkingBudgetValue !== void 0 ? thinkingBudgetValue : -1, includeThoughts !== null && includeThoughts !== void 0 ? includeThoughts : false, tools, cacheTTL);
-        if (cachedContentName) {
-            googleProviderOptions = buildGoogleProviderOptions(exec, index, cachedContentName, thinkingBudgetValue, includeThoughts);
-        }
-        else {
-            googleProviderOptions = buildGoogleProviderOptions(exec, index, undefined, thinkingBudgetValue, includeThoughts);
-            console.log('UniversalAI: Cache creation failed, falling back to regular content');
-        }
-    }
-    else if (provider === 'google') {
-        googleProviderOptions = buildGoogleProviderOptions(exec, index, undefined, thinkingBudgetValue, includeThoughts);
-    }
+    const googleProviderOptions = provider === 'google'
+        ? await (0, helpers_1.setupGoogleProviderOptions)(exec, index, input, model, apiKey)
+        : undefined;
     const params = {
         model: aiProvider(model, modelSettings),
     };
@@ -752,50 +530,13 @@ async function generateTextOperation(exec, index, provider, aiProvider, model, m
         params.system = input.system;
     if (input.messages)
         params.messages = input.messages;
-    if (tools && Object.keys(tools).length > 0) {
-        params.tools = tools;
-    }
     if (provider === 'google' && googleProviderOptions) {
         params.providerOptions = {
             google: googleProviderOptions,
         };
     }
-    if (provider === 'google' &&
-        input.cacheableContent &&
-        input.cacheableContent.length > 0 &&
-        !(googleProviderOptions === null || googleProviderOptions === void 0 ? void 0 : googleProviderOptions.cachedContent)) {
-        if (input.messages && input.messages.length > 0) {
-            const fallbackMessages = input.cacheableContent.map(content => {
-                const role = content.role === 'model' ? 'assistant' : (content.role === 'user' ? 'user' : 'system');
-                const textParts = content.parts
-                    .filter(part => 'text' in part)
-                    .map(part => part.text);
-                return {
-                    role: role,
-                    content: textParts.join('\n')
-                };
-            });
-            params.messages = [...fallbackMessages, ...(params.messages || [])];
-        }
-        else {
-            for (const content of input.cacheableContent) {
-                const textParts = content.parts
-                    .filter(part => 'text' in part)
-                    .map(part => part.text)
-                    .join('\n');
-                if (textParts.trim()) {
-                    if (!params.system) {
-                        params.system = textParts.trim();
-                    }
-                    else if (!params.prompt) {
-                        params.prompt = textParts.trim();
-                    }
-                    else {
-                        params.prompt = params.prompt + '\n\n' + textParts.trim();
-                    }
-                }
-            }
-        }
+    if (provider === 'google' && !(googleProviderOptions === null || googleProviderOptions === void 0 ? void 0 : googleProviderOptions.cachedContent)) {
+        (0, helpers_1.restoreCacheableContent)(params, input);
     }
     const textNumericKeys = [
         'maxTokens',
@@ -814,25 +555,6 @@ async function generateTextOperation(exec, index, provider, aiProvider, model, m
     const formattedResult = formatTextResult(result, includeRequestBody, provider);
     return [{ json: formattedResult }];
 }
-async function buildGoogleTools(exec, index) {
-    const googleTools = exec.getNodeParameter('googleTools', index, []);
-    if (!googleTools || googleTools.length === 0) {
-        return undefined;
-    }
-    const tools = {};
-    const { google } = await Promise.resolve().then(() => __importStar(require('@ai-sdk/google')));
-    const toolSet = new Set(googleTools);
-    if (toolSet.has('google_search')) {
-        tools.google_search = google.tools.googleSearch({});
-    }
-    if (toolSet.has('url_context')) {
-        tools.url_context = google.tools.urlContext({});
-    }
-    if (toolSet.has('code_execution')) {
-        tools.code_execution = google.tools.codeExecution({});
-    }
-    return tools;
-}
 async function handleStreaming(params, provider, includeRequestBody) {
     const stream = await (0, ai_1.streamText)(params);
     const chunks = [];
@@ -877,29 +599,9 @@ async function generateObjectOperation(exec, index, provider, aiProvider, model,
     const schemaDescription = exec.getNodeParameter('schemaDescription', index, '');
     const rawSchema = exec.getNodeParameter('schema', index);
     const parsedSchema = parseAndValidateSchema(rawSchema, exec);
-    let googleProviderOptions;
-    let thinkingBudgetValue;
-    let includeThoughts;
-    if (provider === 'google') {
-        thinkingBudgetValue = Number(exec.getNodeParameter('thinkingBudget', index, -1));
-        includeThoughts = exec.getNodeParameter('includeThoughts', index, false);
-    }
-    if (provider === 'google' &&
-        input.cacheableContent &&
-        input.cacheableContent.length > 0) {
-        const cacheTTL = exec.getNodeParameter('cacheTTL', index, 3600);
-        const cachedContentName = await createGoogleCache(exec, index, apiKey, input.cacheableContent, model, thinkingBudgetValue !== null && thinkingBudgetValue !== void 0 ? thinkingBudgetValue : -1, includeThoughts !== null && includeThoughts !== void 0 ? includeThoughts : false, undefined, cacheTTL);
-        if (cachedContentName) {
-            googleProviderOptions = buildGoogleProviderOptions(exec, index, cachedContentName, thinkingBudgetValue, includeThoughts);
-        }
-        else {
-            googleProviderOptions = buildGoogleProviderOptions(exec, index, undefined, thinkingBudgetValue, includeThoughts);
-            console.log('UniversalAI: Cache creation failed, falling back to regular content');
-        }
-    }
-    else if (provider === 'google') {
-        googleProviderOptions = buildGoogleProviderOptions(exec, index, undefined, thinkingBudgetValue, includeThoughts);
-    }
+    const googleProviderOptions = provider === 'google'
+        ? await (0, helpers_1.setupGoogleProviderOptions)(exec, index, input, model, apiKey)
+        : undefined;
     const params = {
         model: aiProvider(model, modelSettings),
         schema: (0, ai_1.jsonSchema)(parsedSchema),
@@ -917,42 +619,8 @@ async function generateObjectOperation(exec, index, provider, aiProvider, model,
             google: googleProviderOptions,
         };
     }
-    if (provider === 'google' &&
-        input.cacheableContent &&
-        input.cacheableContent.length > 0 &&
-        !(googleProviderOptions === null || googleProviderOptions === void 0 ? void 0 : googleProviderOptions.cachedContent)) {
-        if (input.messages && input.messages.length > 0) {
-            const fallbackMessages = input.cacheableContent.map(content => {
-                const role = content.role === 'model' ? 'assistant' : (content.role === 'user' ? 'user' : 'system');
-                const textParts = content.parts
-                    .filter(part => 'text' in part)
-                    .map(part => part.text);
-                return {
-                    role: role,
-                    content: textParts.join('\n')
-                };
-            });
-            params.messages = [...fallbackMessages, ...(params.messages || [])];
-        }
-        else {
-            for (const content of input.cacheableContent) {
-                const textParts = content.parts
-                    .filter(part => 'text' in part)
-                    .map(part => part.text)
-                    .join('\n');
-                if (textParts.trim()) {
-                    if (!params.system) {
-                        params.system = textParts.trim();
-                    }
-                    else if (!params.prompt) {
-                        params.prompt = textParts.trim();
-                    }
-                    else {
-                        params.prompt = params.prompt + '\n\n' + textParts.trim();
-                    }
-                }
-            }
-        }
+    if (provider === 'google' && !(googleProviderOptions === null || googleProviderOptions === void 0 ? void 0 : googleProviderOptions.cachedContent)) {
+        (0, helpers_1.restoreCacheableContent)(params, input);
     }
     const objectNumericKeys = [
         'temperature',