@illuma-ai/agents 1.0.90 → 1.0.94

package/src/llm/google/index.js

@@ -0,0 +1,215 @@
+/* eslint-disable @typescript-eslint/ban-ts-comment */
+import { AIMessageChunk } from '@langchain/core/messages';
+import { ChatGenerationChunk } from '@langchain/core/outputs';
+import { ChatGoogleGenerativeAI } from '@langchain/google-genai';
+import { getEnvironmentVariable } from '@langchain/core/utils/env';
+import { GoogleGenerativeAI as GenerativeAI } from '@google/generative-ai';
+import { convertResponseContentToChatGenerationChunk, convertBaseMessagesToContent, mapGenerateContentResultToChatResult, } from './utils/common';
+export class CustomChatGoogleGenerativeAI extends ChatGoogleGenerativeAI {
+    thinkingConfig;
+    /**
+     * Override to add gemini-3 model support for multimodal and function calling thought signatures
+     */
+    get _isMultimodalModel() {
+        return (this.model.startsWith('gemini-1.5') ||
+            this.model.startsWith('gemini-2') ||
+            (this.model.startsWith('gemma-3-') &&
+                !this.model.startsWith('gemma-3-1b')) ||
+            this.model.startsWith('gemini-3'));
+    }
+    constructor(fields) {
+        super(fields);
+        this.model = fields.model.replace(/^models\//, '');
+        this.maxOutputTokens = fields.maxOutputTokens ?? this.maxOutputTokens;
+        if (this.maxOutputTokens != null && this.maxOutputTokens < 0) {
+            throw new Error('`maxOutputTokens` must be a positive integer');
+        }
+        this.temperature = fields.temperature ?? this.temperature;
+        if (this.temperature != null &&
+            (this.temperature < 0 || this.temperature > 2)) {
+            throw new Error('`temperature` must be in the range of [0.0,2.0]');
+        }
+        this.topP = fields.topP ?? this.topP;
+        if (this.topP != null && this.topP < 0) {
+            throw new Error('`topP` must be a positive integer');
+        }
+        if (this.topP != null && this.topP > 1) {
+            throw new Error('`topP` must be below 1.');
+        }
+        this.topK = fields.topK ?? this.topK;
+        if (this.topK != null && this.topK < 0) {
+            throw new Error('`topK` must be a positive integer');
+        }
+        this.stopSequences = fields.stopSequences ?? this.stopSequences;
+        this.apiKey = fields.apiKey ?? getEnvironmentVariable('GOOGLE_API_KEY');
+        if (this.apiKey == null || this.apiKey === '') {
+            throw new Error('Please set an API key for Google GenerativeAI ' +
+                'in the environment variable GOOGLE_API_KEY ' +
+                'or in the `apiKey` field of the ' +
+                'ChatGoogleGenerativeAI constructor');
+        }
+        this.safetySettings = fields.safetySettings ?? this.safetySettings;
+        if (this.safetySettings && this.safetySettings.length > 0) {
+            const safetySettingsSet = new Set(this.safetySettings.map((s) => s.category));
+            if (safetySettingsSet.size !== this.safetySettings.length) {
+                throw new Error('The categories in `safetySettings` array must be unique');
+            }
+        }
+        this.thinkingConfig = fields.thinkingConfig ?? this.thinkingConfig;
+        this.streaming = fields.streaming ?? this.streaming;
+        this.json = fields.json;
+        // @ts-ignore - Accessing private property from parent class
+        this.client = new GenerativeAI(this.apiKey).getGenerativeModel({
+            model: this.model,
+            safetySettings: this.safetySettings,
+            generationConfig: {
+                stopSequences: this.stopSequences,
+                maxOutputTokens: this.maxOutputTokens,
+                temperature: this.temperature,
+                topP: this.topP,
+                topK: this.topK,
+                ...(this.json != null
+                    ? { responseMimeType: 'application/json' }
+                    : {}),
+            },
+        }, {
+            apiVersion: fields.apiVersion,
+            baseUrl: fields.baseUrl,
+            customHeaders: fields.customHeaders,
+        });
+        this.streamUsage = fields.streamUsage ?? this.streamUsage;
+    }
+    static lc_name() {
+        return 'IllumaGoogleGenerativeAI';
+    }
+    /**
+     * Helper function to convert Gemini API usage metadata to LangChain format
+     * Includes support for cached tokens and tier-based tracking for gemini-3-pro-preview
+     */
+    _convertToUsageMetadata(usageMetadata, model) {
+        if (!usageMetadata) {
+            return undefined;
+        }
+        const output = {
+            input_tokens: usageMetadata.promptTokenCount ?? 0,
+            output_tokens: (usageMetadata.candidatesTokenCount ?? 0) +
+                (usageMetadata.thoughtsTokenCount ?? 0),
+            total_tokens: usageMetadata.totalTokenCount ?? 0,
+        };
+        if (usageMetadata.cachedContentTokenCount) {
+            output.input_token_details ??= {};
+            output.input_token_details.cache_read =
+                usageMetadata.cachedContentTokenCount;
+        }
+        // gemini-3-pro-preview has bracket based tracking of tokens per request
+        if (model === 'gemini-3-pro-preview') {
+            const over200k = Math.max(0, (usageMetadata.promptTokenCount ?? 0) - 200000);
+            const cachedOver200k = Math.max(0, (usageMetadata.cachedContentTokenCount ?? 0) - 200000);
+            if (over200k) {
+                output.input_token_details = {
+                    ...output.input_token_details,
+                    over_200k: over200k,
+                };
+            }
+            if (cachedOver200k) {
+                output.input_token_details = {
+                    ...output.input_token_details,
+                    cache_read_over_200k: cachedOver200k,
+                };
+            }
+        }
+        return output;
+    }
+    invocationParams(options) {
+        const params = super.invocationParams(options);
+        if (this.thinkingConfig) {
+            /** @ts-ignore */
+            this.client.generationConfig = {
+                /** @ts-ignore */
+                ...this.client.generationConfig,
+                /** @ts-ignore */
+                thinkingConfig: this.thinkingConfig,
+            };
+        }
+        return params;
+    }
+    async _generate(messages, options, runManager) {
+        const prompt = convertBaseMessagesToContent(messages, this._isMultimodalModel, this.useSystemInstruction, this.model);
+        let actualPrompt = prompt;
+        if (prompt?.[0].role === 'system') {
+            const [systemInstruction] = prompt;
+            /** @ts-ignore */
+            this.client.systemInstruction = systemInstruction;
+            actualPrompt = prompt.slice(1);
+        }
+        const parameters = this.invocationParams(options);
+        const request = {
+            ...parameters,
+            contents: actualPrompt,
+        };
+        const res = await this.caller.callWithOptions({ signal: options.signal }, async () => 
+        /** @ts-ignore */
+        this.client.generateContent(request));
+        const response = res.response;
+        const usageMetadata = this._convertToUsageMetadata(
+        /** @ts-ignore */
+        response.usageMetadata, this.model);
+        /** @ts-ignore */
+        const generationResult = mapGenerateContentResultToChatResult(response, {
+            usageMetadata,
+        });
+        await runManager?.handleLLMNewToken(generationResult.generations[0].text || '', undefined, undefined, undefined, undefined, undefined);
+        return generationResult;
+    }
+    async *_streamResponseChunks(messages, options, runManager) {
+        const prompt = convertBaseMessagesToContent(messages, this._isMultimodalModel, this.useSystemInstruction, this.model);
+        let actualPrompt = prompt;
+        if (prompt?.[0].role === 'system') {
+            const [systemInstruction] = prompt;
+            /** @ts-ignore */
+            this.client.systemInstruction = systemInstruction;
+            actualPrompt = prompt.slice(1);
+        }
+        const parameters = this.invocationParams(options);
+        const request = {
+            ...parameters,
+            contents: actualPrompt,
+        };
+        const stream = await this.caller.callWithOptions({ signal: options.signal }, async () => {
+            /** @ts-ignore */
+            const { stream } = await this.client.generateContentStream(request);
+            return stream;
+        });
+        let index = 0;
+        let lastUsageMetadata;
+        for await (const response of stream) {
+            if ('usageMetadata' in response &&
+                this.streamUsage !== false &&
+                options.streamUsage !== false) {
+                lastUsageMetadata = this._convertToUsageMetadata(response.usageMetadata, this.model);
+            }
+            const chunk = convertResponseContentToChatGenerationChunk(response, {
+                usageMetadata: undefined,
+                index,
+            });
+            index += 1;
+            if (!chunk) {
+                continue;
+            }
+            yield chunk;
+            await runManager?.handleLLMNewToken(chunk.text || '', undefined, undefined, undefined, undefined, { chunk });
+        }
+        if (lastUsageMetadata) {
+            const finalChunk = new ChatGenerationChunk({
+                text: '',
+                message: new AIMessageChunk({
+                    content: '',
+                    usage_metadata: lastUsageMetadata,
+                }),
+            });
+            yield finalChunk;
+            await runManager?.handleLLMNewToken(finalChunk.text || '', undefined, undefined, undefined, undefined, { chunk: finalChunk });
+        }
+    }
+}
+//# sourceMappingURL=index.js.map

package/src/llm/google/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["index.ts"],"names":[],"mappings":"AAAA,sDAAsD;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAC;AACnE,OAAO,EAAE,kBAAkB,IAAI,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAS3E,OAAO,EACL,2CAA2C,EAC3C,4BAA4B,EAC5B,oCAAoC,GACrC,MAAM,gBAAgB,CAAC;AAExB,MAAM,OAAO,4BAA6B,SAAQ,sBAAsB;IACtE,cAAc,CAAwB;IAEtC;;OAEG;IACH,IAAI,kBAAkB;QACpB,OAAO,CACL,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC;YACnC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;YACjC,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;gBAChC,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;YACvC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAClC,CAAC;IACJ,CAAC;IAED,YAAY,MAA2B;QACrC,KAAK,CAAC,MAAM,CAAC,CAAC;QAEd,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAEnD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC;QAEtE,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,IAAI,IAAI,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;YAC7D,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;QAC1D,IACE,IAAI,CAAC,WAAW,IAAI,IAAI;YACxB,CAAC,IAAI,CAAC,WAAW,GAAG,CAAC,IAAI,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,EAC9C,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACrC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACrC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC;QAEhE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,sBAAsB,CAAC,gBAAgB,CAAC,CAAC;QACxE,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,EAAE,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CACb,gDAAgD;gBAC9C,6CAA6C;gBAC7C,kCAAkC;gBAClC,oCAAoC,CACvC,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,cAAc,CAAC;QACnE,IAAI,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1D,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAC/B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAC3C,CAAC;YACF,IAAI,iBAAiB,CAAC,IAAI,KAAK,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;gBAC1D,MAAM,IAAI,KAAK,CACb,yDAAyD,CAC1D,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,cAAc,CAAC;QAEnE,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QACpD,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QAExB,4DAA4D;QAC5D,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAC5D;YACE,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,cAAc,EAAE,IAAI,CAAC,cAAiC;YACtD,gBAAgB,EAAE;gBAChB,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,eAAe,EAAE,IAAI,CAAC,eAAe;gBACrC,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI;oBACnB,CAAC,CAAC,EAAE,gBAAgB,EAAE,kBAAkB,EAAE;oBAC1C,CAAC,CAAC,EAAE,CAAC;aACR;SACF,EACD;YACE,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,aAAa,EAAE,MAAM,CAAC,aAAa;SACpC,CACF,CAAC;QACF,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,CAAC;IAC5D,CAAC;IAED,MAAM,CAAC,OAAO;QACZ,OAAO,0BAA0B,CAAC;IACpC,CAAC;IAED;;;OAGG;IACK,uBAAuB,CAC7B,aAAiD,EACjD,KAAa;QAEb,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,MAAM,GAAkB;YAC5B,YAAY,EAAE,aAAa,CAAC,gBAAgB,IAAI,CAAC;YACjD,aAAa,EACX,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC,CAAC;gBACzC,CAAC,aAAa,CAAC,kBAAkB,IAAI,CAAC,CAAC;YACzC,YAAY,EAAE,aAAa,CAAC,eAAe,IAAI,CAAC;SACjD,CAAC;QAEF,IAAI,aAAa,CAAC,uBAAuB,EAAE,CAAC;YAC1C,MAAM,CAAC,mBAAmB,KAAK,EAAE,CAAC;YAClC,MAAM,CAAC,mBAAmB,CAAC,UAAU;gBACnC,aAAa,CAAC,uBAAuB,CAAC;QAC1C,CAAC;QAED,wEAAwE;QACxE,IAAI,KAAK,KAAK,sBAAsB,EAAE,CAAC;YACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CACvB,CAAC,EACD,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC,CAAC,GAAG,MAAM,CAC/C,CAAC;YACF,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAC7B,CAAC,EACD,CAAC,aAAa,CAAC,uBAAuB,IAAI,CAAC,CAAC,GAAG,MAAM,CACtD,CAAC;YACF,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,CAAC,mBAAmB,GAAG;oBAC3B,GAAG,MAAM,CAAC,mBAAmB;oBAC7B,SAAS,EAAE,QAAQ;iBACC,CAAC;YACzB,CAAC;YACD,IAAI,cAAc,EAAE,CAAC;gBACnB,MAAM,CAAC,mBAAmB,GAAG;oBAC3B,GAAG,MAAM,CAAC,mBAAmB;oBAC7B,oBAAoB,EAAE,cAAc;iBAChB,CAAC;YACzB,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,gBAAgB,CACd,OAAmC;QAEnC,MAAM,MAAM,GAAG,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,iBAAiB;YACjB,IAAI,CAAC,MAAM,CAAC,gBAAgB,GAAG;gBAC7B,iBAAiB;gBACjB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;gBAC/B,iBAAiB;gBACjB,cAAc,EAAE,IAAI,CAAC,cAAc;aACpC,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,SAAS,CACb,QAAuB,EACvB,OAAkC,EAClC,UAAqC;QAErC,MAAM,MAAM,GAAG,4BAA4B,CACzC,QAAQ,EACR,IAAI,CAAC,kBAAkB,EACvB,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,KAAK,CACX,CAAC;QACF,IAAI,YAAY,GAAG,MAAM,CAAC;QAC1B,IAAI,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,MAAM,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC;YACnC,iBAAiB;YACjB,IAAI,CAAC,MAAM,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;YAClD,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG;YACd,GAAG,UAAU;YACb,QAAQ,EAAE,YAAY;SACvB,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAC3C,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,EAC1B,KAAK,IAAI,EAAE;QACT,iBAAiB;QACjB,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CACvC,CAAC;QAEF,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;QAC9B,MAAM,aAAa,GAAG,IAAI,CAAC,uBAAuB;QAChD,iBAAiB;QACjB,QAAQ,CAAC,aAAa,EACtB,IAAI,CAAC,KAAK,CACX,CAAC;QAEF,iBAAiB;QACjB,MAAM,gBAAgB,GAAG,oCAAoC,CAAC,QAAQ,EAAE;YACtE,aAAa;SACd,CAAC,CAAC;QAEH,MAAM,UAAU,EAAE,iBAAiB,CACjC,gBAAgB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,EAC1C,SAAS,EACT,SAAS,EACT,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;QACF,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,CAAC,qBAAqB,CAC1B,QAAuB,EACvB,OAAkC,EAClC,UAAqC;QAErC,MAAM,MAAM,GAAG,4BAA4B,CACzC,QAAQ,EACR,IAAI,CAAC,kBAAkB,EACvB,IAAI,CAAC,oBAAoB,EACzB,IAAI,CAAC,KAAK,CACX,CAAC;QACF,IAAI,YAAY,GAAG,MAAM,CAAC;QAC1B,IAAI,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAClC,MAAM,CAAC,iBAAiB,CAAC,GAAG,MAAM,CAAC;YACnC,iBAAiB;YACjB,IAAI,CAAC,MAAM,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;YAClD,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG;YACd,GAAG,UAAU;YACb,QAAQ,EAAE,YAAY;SACvB,CAAC;QACF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAC9C,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,EAC1B,KAAK,IAAI,EAAE;YACT,iBAAiB;YACjB,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;YACpE,OAAO,MAAM,CAAC;QAChB,CAAC,CACF,CAAC;QAEF,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,iBAA4C,CAAC;QACjD,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;YACpC,IACE,eAAe,IAAI,QAAQ;gBAC3B,IAAI,CAAC,WAAW,KAAK,KAAK;gBAC1B,OAAO,CAAC,WAAW,KAAK,KAAK,EAC7B,CAAC;gBACD,iBAAiB,GAAG,IAAI,CAAC,uBAAuB,CAC9C,QAAQ,CAAC,aAAmD,EAC5D,IAAI,CAAC,KAAK,CACX,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,GAAG,2CAA2C,CAAC,QAAQ,EAAE;gBAClE,aAAa,EAAE,SAAS;gBACxB,KAAK;aACN,CAAC,CAAC;YACH,KAAK,IAAI,CAAC,CAAC;YACX,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,SAAS;YACX,CAAC;YAED,MAAM,KAAK,CAAC;YACZ,MAAM,UAAU,EAAE,iBAAiB,CACjC,KAAK,CAAC,IAAI,IAAI,EAAE,EAChB,SAAS,EACT,SAAS,EACT,SAAS,EACT,SAAS,EACT,EAAE,KAAK,EAAE,CACV,CAAC;QACJ,CAAC;QAED,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,UAAU,GAAG,IAAI,mBAAmB,CAAC;gBACzC,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,IAAI,cAAc,CAAC;oBAC1B,OAAO,EAAE,EAAE;oBACX,cAAc,EAAE,iBAAiB;iBAClC,CAAC;aACH,CAAC,CAAC;YACH,MAAM,UAAU,CAAC;YACjB,MAAM,UAAU,EAAE,iBAAiB,CACjC,UAAU,CAAC,IAAI,IAAI,EAAE,EACrB,SAAS,EACT,SAAS,EACT,SAAS,EACT,SAAS,EACT,EAAE,KAAK,EAAE,UAAU,EAAE,CACtB,CAAC;QACJ,CAAC;IACH,CAAC;CACF"}

package/src/llm/google/index.ts

@@ -10,9 +10,8 @@ import type {
 } from '@google/generative-ai';
 import type { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
 import type { BaseMessage, UsageMetadata } from '@langchain/core/messages';
-import type { GeminiGenerationConfig } from '@langchain/google-common';
 import type { GeminiApiUsageMetadata, InputTokenDetails } from './types';
-import type { GoogleClientOptions } from '@/types';
+import type { GoogleClientOptions, GoogleThinkingConfig } from '@/types';
 import {
   convertResponseContentToChatGenerationChunk,
   convertBaseMessagesToContent,
@@ -20,7 +19,7 @@ import {
 } from './utils/common';
 
 export class CustomChatGoogleGenerativeAI extends ChatGoogleGenerativeAI {
-  thinkingConfig?: GeminiGenerationConfig['thinkingConfig'];
+  thinkingConfig?: GoogleThinkingConfig;
 
   /**
    * Override to add gemini-3 model support for multimodal and function calling thought signatures

package/src/llm/google/types.js

@@ -0,0 +1,12 @@
+/** Enum for content modality types */
+var Modality;
+(function (Modality) {
+    Modality["MODALITY_UNSPECIFIED"] = "MODALITY_UNSPECIFIED";
+    Modality["TEXT"] = "TEXT";
+    Modality["IMAGE"] = "IMAGE";
+    Modality["VIDEO"] = "VIDEO";
+    Modality["AUDIO"] = "AUDIO";
+    Modality["DOCUMENT"] = "DOCUMENT";
+})(Modality || (Modality = {}));
+export {};
+//# sourceMappingURL=types.js.map

package/src/llm/google/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["types.ts"],"names":[],"mappings":"AAmBA,sCAAsC;AACtC,IAAK,QAOJ;AAPD,WAAK,QAAQ;IACX,yDAA6C,CAAA;IAC7C,yBAAa,CAAA;IACb,2BAAe,CAAA;IACf,2BAAe,CAAA;IACf,2BAAe,CAAA;IACf,iCAAqB,CAAA;AACvB,CAAC,EAPI,QAAQ,KAAR,QAAQ,QAOZ"}