@langchain/google-common 0.1.5 → 0.1.7

package/dist/chat_models.cjs

@@ -53,10 +53,33 @@ class ChatConnection extends connection_js_1.AbstractGoogleLLMConnection {
         }
         return true;
     }
+    computeGoogleSearchToolAdjustmentFromModel() {
+        if (this.modelName.startsWith("gemini-1.0")) {
+            return "googleSearchRetrieval";
+        }
+        else if (this.modelName.startsWith("gemini-1.5")) {
+            return "googleSearchRetrieval";
+        }
+        else {
+            return "googleSearch";
+        }
+    }
+    computeGoogleSearchToolAdjustment(apiConfig) {
+        const adj = apiConfig.googleSearchToolAdjustment;
+        if (adj === undefined || adj === true) {
+            return this.computeGoogleSearchToolAdjustmentFromModel();
+        }
+        else {
+            return adj;
+        }
+    }
     buildGeminiAPI() {
+        const apiConfig = this.apiConfig ?? {};
+        const googleSearchToolAdjustment = this.computeGoogleSearchToolAdjustment(apiConfig);
         const geminiConfig = {
             useSystemInstruction: this.useSystemInstruction,
-            ...this.apiConfig,
+            googleSearchToolAdjustment,
+            ...apiConfig,
         };
         return (0, gemini_js_1.getGeminiAPI)(geminiConfig);
     }
@@ -128,12 +151,36 @@ class ChatGoogleBase extends chat_models_1.BaseChatModel {
             writable: true,
             value: 40
         });
+        Object.defineProperty(this, "presencePenalty", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "frequencyPenalty", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "stopSequences", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: []
         });
+        Object.defineProperty(this, "logprobs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "topLogprobs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0
+        });
         Object.defineProperty(this, "safetySettings", {
             enumerable: true,
             configurable: true,
@@ -199,7 +246,13 @@ class ChatGoogleBase extends chat_models_1.BaseChatModel {
         return new auth_js_1.ApiKeyGoogleAuth(apiKey);
     }
     buildApiKey(fields) {
-        return fields?.apiKey ?? (0, env_1.getEnvironmentVariable)("GOOGLE_API_KEY");
+        if (fields?.platformType !== "gcp") {
+            return fields?.apiKey ?? (0, env_1.getEnvironmentVariable)("GOOGLE_API_KEY");
+        }
+        else {
+            // GCP doesn't support API Keys
+            return undefined;
+        }
     }
     buildClient(fields) {
         const apiKey = this.buildApiKey(fields);

package/dist/chat_models.d.ts

@@ -7,15 +7,17 @@ import { BaseLanguageModelInput, StructuredOutputMethodOptions } from "@langchai
 import type { z } from "zod";
 import { Runnable } from "@langchain/core/runnables";
 import { AsyncCaller } from "@langchain/core/utils/async_caller";
-import { GoogleAIBaseLLMInput, GoogleAIModelParams, GoogleAISafetySetting, GoogleConnectionParams, GooglePlatformType, GoogleAIBaseLanguageModelCallOptions, GoogleAIAPI, GoogleAIAPIParams } from "./types.js";
+import { GoogleAIBaseLLMInput, GoogleAIModelParams, GoogleAISafetySetting, GoogleConnectionParams, GooglePlatformType, GoogleAIBaseLanguageModelCallOptions, GoogleAIAPI, GoogleAIAPIParams, GoogleSearchToolSetting } from "./types.js";
 import { AbstractGoogleLLMConnection } from "./connection.js";
 import { GoogleAbstractedClient } from "./auth.js";
-import type { GoogleBaseLLMInput, GoogleAISafetyHandler, GoogleAISafetyParams, GoogleAIToolType } from "./types.js";
+import type { GoogleBaseLLMInput, GoogleAISafetyHandler, GoogleAISafetyParams, GoogleAIToolType, GeminiAPIConfig } from "./types.js";
 export declare class ChatConnection<AuthOptions> extends AbstractGoogleLLMConnection<BaseMessage[], AuthOptions> {
     convertSystemMessageToHumanContent: boolean | undefined;
     constructor(fields: GoogleAIBaseLLMInput<AuthOptions> | undefined, caller: AsyncCaller, client: GoogleAbstractedClient, streaming: boolean);
     get useSystemInstruction(): boolean;
     get computeUseSystemInstruction(): boolean;
+    computeGoogleSearchToolAdjustmentFromModel(): Exclude<GoogleSearchToolSetting, boolean>;
+    computeGoogleSearchToolAdjustment(apiConfig: GeminiAPIConfig): Exclude<GoogleSearchToolSetting, true>;
     buildGeminiAPI(): GoogleAIAPI;
     get api(): GoogleAIAPI;
 }
@@ -39,7 +41,11 @@ export declare abstract class ChatGoogleBase<AuthOptions> extends BaseChatModel<
     maxOutputTokens: number;
     topP: number;
     topK: number;
+    presencePenalty: number;
+    frequencyPenalty: number;
     stopSequences: string[];
+    logprobs: boolean;
+    topLogprobs: number;
     safetySettings: GoogleAISafetySetting[];
     convertSystemMessageToHumanContent: boolean | undefined;
     safetyHandler: GoogleAISafetyHandler;

package/dist/chat_models.js

@@ -50,10 +50,33 @@ export class ChatConnection extends AbstractGoogleLLMConnection {
         }
         return true;
     }
+    computeGoogleSearchToolAdjustmentFromModel() {
+        if (this.modelName.startsWith("gemini-1.0")) {
+            return "googleSearchRetrieval";
+        }
+        else if (this.modelName.startsWith("gemini-1.5")) {
+            return "googleSearchRetrieval";
+        }
+        else {
+            return "googleSearch";
+        }
+    }
+    computeGoogleSearchToolAdjustment(apiConfig) {
+        const adj = apiConfig.googleSearchToolAdjustment;
+        if (adj === undefined || adj === true) {
+            return this.computeGoogleSearchToolAdjustmentFromModel();
+        }
+        else {
+            return adj;
+        }
+    }
     buildGeminiAPI() {
+        const apiConfig = this.apiConfig ?? {};
+        const googleSearchToolAdjustment = this.computeGoogleSearchToolAdjustment(apiConfig);
         const geminiConfig = {
             useSystemInstruction: this.useSystemInstruction,
-            ...this.apiConfig,
+            googleSearchToolAdjustment,
+            ...apiConfig,
         };
         return getGeminiAPI(geminiConfig);
     }
@@ -124,12 +147,36 @@ export class ChatGoogleBase extends BaseChatModel {
             writable: true,
             value: 40
         });
+        Object.defineProperty(this, "presencePenalty", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "frequencyPenalty", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "stopSequences", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: []
         });
+        Object.defineProperty(this, "logprobs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "topLogprobs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 0
+        });
         Object.defineProperty(this, "safetySettings", {
             enumerable: true,
             configurable: true,
@@ -195,7 +242,13 @@ export class ChatGoogleBase extends BaseChatModel {
         return new ApiKeyGoogleAuth(apiKey);
     }
     buildApiKey(fields) {
-        return fields?.apiKey ?? getEnvironmentVariable("GOOGLE_API_KEY");
+        if (fields?.platformType !== "gcp") {
+            return fields?.apiKey ?? getEnvironmentVariable("GOOGLE_API_KEY");
+        }
+        else {
+            // GCP doesn't support API Keys
+            return undefined;
+        }
     }
     buildClient(fields) {
         const apiKey = this.buildApiKey(fields);

package/dist/index.cjs

@@ -17,6 +17,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./chat_models.cjs"), exports);
 __exportStar(require("./llms.cjs"), exports);
 __exportStar(require("./embeddings.cjs"), exports);
+__exportStar(require("./output_parsers.cjs"), exports);
 __exportStar(require("./auth.cjs"), exports);
 __exportStar(require("./connection.cjs"), exports);
 __exportStar(require("./types.cjs"), exports);

package/dist/index.d.ts

@@ -1,6 +1,7 @@
 export * from "./chat_models.js";
 export * from "./llms.js";
 export * from "./embeddings.js";
+export * from "./output_parsers.js";
 export * from "./auth.js";
 export * from "./connection.js";
 export * from "./types.js";

package/dist/index.js

@@ -1,6 +1,7 @@
 export * from "./chat_models.js";
 export * from "./llms.js";
 export * from "./embeddings.js";
+export * from "./output_parsers.js";
 export * from "./auth.js";
 export * from "./connection.js";
 export * from "./types.js";

package/dist/output_parsers.cjs

@@ -0,0 +1,175 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MarkdownGoogleSearchOutputParser = exports.SimpleGoogleSearchOutputParser = exports.BaseGoogleSearchOutputParser = void 0;
+const output_parsers_1 = require("@langchain/core/output_parsers");
+class BaseGoogleSearchOutputParser extends output_parsers_1.BaseLLMOutputParser {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["google_common", "output_parsers"]
+        });
+    }
+    generationToGroundingInfo(generation) {
+        if ("message" in generation) {
+            const responseMetadata = generation?.message?.response_metadata;
+            const metadata = responseMetadata?.groundingMetadata;
+            const supports = responseMetadata?.groundingSupport ?? metadata?.groundingSupports ?? [];
+            if (metadata) {
+                return {
+                    metadata,
+                    supports,
+                };
+            }
+        }
+        return undefined;
+    }
+    generationsToGroundingInfo(generations) {
+        for (const generation of generations) {
+            const info = this.generationToGroundingInfo(generation);
+            if (info !== undefined) {
+                return info;
+            }
+        }
+        return undefined;
+    }
+    generationToString(generation) {
+        if ("message" in generation) {
+            const content = generation?.message?.content;
+            if (typeof content === "string") {
+                return content;
+            }
+            else {
+                return content
+                    .map((c) => {
+                    if (c?.type === "text") {
+                        return c?.text ?? "";
+                    }
+                    else {
+                        return "";
+                    }
+                })
+                    .reduce((previousValue, currentValue) => `${previousValue}${currentValue}`);
+            }
+        }
+        return generation.text;
+    }
+    generationsToString(generations) {
+        return generations
+            .map((generation) => this.generationToString(generation))
+            .reduce((previousValue, currentValue) => `${previousValue}${currentValue}`);
+    }
+    annotateSegment(text, grounding, support, index) {
+        const start = support.segment.startIndex ?? 0;
+        const end = support.segment.endIndex;
+        const textBefore = text.substring(0, start);
+        const textSegment = text.substring(start, end);
+        const textAfter = text.substring(end);
+        const textPrefix = this.segmentPrefix(grounding, support, index) ?? "";
+        const textSuffix = this.segmentSuffix(grounding, support, index) ?? "";
+        return `${textBefore}${textPrefix}${textSegment}${textSuffix}${textAfter}`;
+    }
+    annotateTextSegments(text, grounding) {
+        // Go through each support info in reverse, since the segment info
+        // is sorted, and we won't need to adjust string indexes this way.
+        let ret = text;
+        for (let co = grounding.supports.length - 1; co >= 0; co -= 1) {
+            const support = grounding.supports[co];
+            ret = this.annotateSegment(ret, grounding, support, co);
+        }
+        return ret;
+    }
+    /**
+     * Google requires us to
+     * "Display the Search Suggestion exactly as provided without any modifications"
+     * So this will typically be called from the textSuffix() method to get
+     * a string that renders HTML.
+     * See https://ai.google.dev/gemini-api/docs/grounding/search-suggestions
+     * @param grounding
+     */
+    searchSuggestion(grounding) {
+        return grounding?.metadata?.searchEntryPoint?.renderedContent ?? "";
+    }
+    annotateText(text, grounding) {
+        const prefix = this.textPrefix(text, grounding) ?? "";
+        const suffix = this.textSuffix(text, grounding) ?? "";
+        const body = this.annotateTextSegments(text, grounding);
+        return `${prefix}${body}${suffix}`;
+    }
+    async parseResult(generations, _callbacks) {
+        const text = this.generationsToString(generations);
+        const grounding = this.generationsToGroundingInfo(generations);
+        if (!grounding) {
+            return text;
+        }
+        return this.annotateText(text, grounding);
+    }
+}
+exports.BaseGoogleSearchOutputParser = BaseGoogleSearchOutputParser;
+class SimpleGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {
+    segmentPrefix(_grounding, _support, _index) {
+        return undefined;
+    }
+    segmentSuffix(_grounding, support, _index) {
+        const indices = support.groundingChunkIndices.map((i) => i + 1);
+        return ` [${indices.join(", ")}]`;
+    }
+    textPrefix(_text, _grounding) {
+        return "Google Says:\n";
+    }
+    chunkToString(chunk, index) {
+        const info = chunk.retrievedContext ?? chunk.web;
+        return `${index + 1}. ${info.title} - ${info.uri}`;
+    }
+    textSuffix(_text, grounding) {
+        let ret = "\n";
+        const chunks = grounding?.metadata?.groundingChunks ?? [];
+        chunks.forEach((chunk, index) => {
+            ret = `${ret}${this.chunkToString(chunk, index)}\n`;
+        });
+        return ret;
+    }
+}
+exports.SimpleGoogleSearchOutputParser = SimpleGoogleSearchOutputParser;
+class MarkdownGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {
+    segmentPrefix(_grounding, _support, _index) {
+        return undefined;
+    }
+    chunkLink(grounding, index) {
+        const chunk = grounding.metadata.groundingChunks[index];
+        const url = chunk.retrievedContext?.uri ?? chunk.web?.uri;
+        const num = index + 1;
+        return `[[${num}](${url})]`;
+    }
+    segmentSuffix(grounding, support, _index) {
+        let ret = "";
+        support.groundingChunkIndices.forEach((chunkIndex) => {
+            const link = this.chunkLink(grounding, chunkIndex);
+            ret = `${ret}${link}`;
+        });
+        return ret;
+    }
+    textPrefix(_text, _grounding) {
+        return undefined;
+    }
+    chunkSuffixLink(chunk, index) {
+        const num = index + 1;
+        const info = chunk.retrievedContext ?? chunk.web;
+        const url = info.uri;
+        const site = info.title;
+        return `${num}. [${site}](${url})`;
+    }
+    textSuffix(_text, grounding) {
+        let ret = "\n**Search Sources**\n";
+        const chunks = grounding.metadata.groundingChunks;
+        chunks.forEach((chunk, index) => {
+            ret = `${ret}${this.chunkSuffixLink(chunk, index)}\n`;
+        });
+        const search = this.searchSuggestion(grounding);
+        ret = `${ret}\n${search}`;
+        return ret;
+    }
+}
+exports.MarkdownGoogleSearchOutputParser = MarkdownGoogleSearchOutputParser;

package/dist/output_parsers.d.ts

@@ -0,0 +1,49 @@
+import { BaseLLMOutputParser } from "@langchain/core/output_parsers";
+import { Callbacks } from "@langchain/core/callbacks/manager";
+import { ChatGeneration, Generation } from "@langchain/core/outputs";
+import { GeminiGroundingChunk, GeminiGroundingMetadata, GeminiGroundingSupport } from "./types.js";
+type Generations = Generation[] | ChatGeneration[];
+type GroundingInfo = {
+    metadata: GeminiGroundingMetadata;
+    supports: GeminiGroundingSupport[];
+};
+export declare abstract class BaseGoogleSearchOutputParser extends BaseLLMOutputParser<string> {
+    lc_namespace: string[];
+    protected generationToGroundingInfo(generation: Generation | ChatGeneration): GroundingInfo | undefined;
+    protected generationsToGroundingInfo(generations: Generations): GroundingInfo | undefined;
+    protected generationToString(generation: Generation | ChatGeneration): string;
+    protected generationsToString(generations: Generations): string;
+    protected abstract segmentPrefix(grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string | undefined;
+    protected abstract segmentSuffix(grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string | undefined;
+    protected annotateSegment(text: string, grounding: GroundingInfo, support: GeminiGroundingSupport, index: number): string;
+    protected annotateTextSegments(text: string, grounding: GroundingInfo): string;
+    protected abstract textPrefix(text: string, grounding: GroundingInfo): string | undefined;
+    protected abstract textSuffix(text: string, grounding: GroundingInfo): string | undefined;
+    /**
+     * Google requires us to
+     * "Display the Search Suggestion exactly as provided without any modifications"
+     * So this will typically be called from the textSuffix() method to get
+     * a string that renders HTML.
+     * See https://ai.google.dev/gemini-api/docs/grounding/search-suggestions
+     * @param grounding
+     */
+    protected searchSuggestion(grounding: GroundingInfo): string;
+    protected annotateText(text: string, grounding: GroundingInfo): string;
+    parseResult(generations: Generations, _callbacks?: Callbacks): Promise<string>;
+}
+export declare class SimpleGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {
+    protected segmentPrefix(_grounding: GroundingInfo, _support: GeminiGroundingSupport, _index: number): string | undefined;
+    protected segmentSuffix(_grounding: GroundingInfo, support: GeminiGroundingSupport, _index: number): string | undefined;
+    protected textPrefix(_text: string, _grounding: GroundingInfo): string;
+    protected chunkToString(chunk: GeminiGroundingChunk, index: number): string;
+    protected textSuffix(_text: string, grounding: GroundingInfo): string;
+}
+export declare class MarkdownGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {
+    protected segmentPrefix(_grounding: GroundingInfo, _support: GeminiGroundingSupport, _index: number): string | undefined;
+    protected chunkLink(grounding: GroundingInfo, index: number): string;
+    protected segmentSuffix(grounding: GroundingInfo, support: GeminiGroundingSupport, _index: number): string | undefined;
+    protected textPrefix(_text: string, _grounding: GroundingInfo): string | undefined;
+    protected chunkSuffixLink(chunk: GeminiGroundingChunk, index: number): string;
+    protected textSuffix(_text: string, grounding: GroundingInfo): string | undefined;
+}
+export {};

package/dist/output_parsers.js

@@ -0,0 +1,169 @@
+import { BaseLLMOutputParser } from "@langchain/core/output_parsers";
+export class BaseGoogleSearchOutputParser extends BaseLLMOutputParser {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["google_common", "output_parsers"]
+        });
+    }
+    generationToGroundingInfo(generation) {
+        if ("message" in generation) {
+            const responseMetadata = generation?.message?.response_metadata;
+            const metadata = responseMetadata?.groundingMetadata;
+            const supports = responseMetadata?.groundingSupport ?? metadata?.groundingSupports ?? [];
+            if (metadata) {
+                return {
+                    metadata,
+                    supports,
+                };
+            }
+        }
+        return undefined;
+    }
+    generationsToGroundingInfo(generations) {
+        for (const generation of generations) {
+            const info = this.generationToGroundingInfo(generation);
+            if (info !== undefined) {
+                return info;
+            }
+        }
+        return undefined;
+    }
+    generationToString(generation) {
+        if ("message" in generation) {
+            const content = generation?.message?.content;
+            if (typeof content === "string") {
+                return content;
+            }
+            else {
+                return content
+                    .map((c) => {
+                    if (c?.type === "text") {
+                        return c?.text ?? "";
+                    }
+                    else {
+                        return "";
+                    }
+                })
+                    .reduce((previousValue, currentValue) => `${previousValue}${currentValue}`);
+            }
+        }
+        return generation.text;
+    }
+    generationsToString(generations) {
+        return generations
+            .map((generation) => this.generationToString(generation))
+            .reduce((previousValue, currentValue) => `${previousValue}${currentValue}`);
+    }
+    annotateSegment(text, grounding, support, index) {
+        const start = support.segment.startIndex ?? 0;
+        const end = support.segment.endIndex;
+        const textBefore = text.substring(0, start);
+        const textSegment = text.substring(start, end);
+        const textAfter = text.substring(end);
+        const textPrefix = this.segmentPrefix(grounding, support, index) ?? "";
+        const textSuffix = this.segmentSuffix(grounding, support, index) ?? "";
+        return `${textBefore}${textPrefix}${textSegment}${textSuffix}${textAfter}`;
+    }
+    annotateTextSegments(text, grounding) {
+        // Go through each support info in reverse, since the segment info
+        // is sorted, and we won't need to adjust string indexes this way.
+        let ret = text;
+        for (let co = grounding.supports.length - 1; co >= 0; co -= 1) {
+            const support = grounding.supports[co];
+            ret = this.annotateSegment(ret, grounding, support, co);
+        }
+        return ret;
+    }
+    /**
+     * Google requires us to
+     * "Display the Search Suggestion exactly as provided without any modifications"
+     * So this will typically be called from the textSuffix() method to get
+     * a string that renders HTML.
+     * See https://ai.google.dev/gemini-api/docs/grounding/search-suggestions
+     * @param grounding
+     */
+    searchSuggestion(grounding) {
+        return grounding?.metadata?.searchEntryPoint?.renderedContent ?? "";
+    }
+    annotateText(text, grounding) {
+        const prefix = this.textPrefix(text, grounding) ?? "";
+        const suffix = this.textSuffix(text, grounding) ?? "";
+        const body = this.annotateTextSegments(text, grounding);
+        return `${prefix}${body}${suffix}`;
+    }
+    async parseResult(generations, _callbacks) {
+        const text = this.generationsToString(generations);
+        const grounding = this.generationsToGroundingInfo(generations);
+        if (!grounding) {
+            return text;
+        }
+        return this.annotateText(text, grounding);
+    }
+}
+export class SimpleGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {
+    segmentPrefix(_grounding, _support, _index) {
+        return undefined;
+    }
+    segmentSuffix(_grounding, support, _index) {
+        const indices = support.groundingChunkIndices.map((i) => i + 1);
+        return ` [${indices.join(", ")}]`;
+    }
+    textPrefix(_text, _grounding) {
+        return "Google Says:\n";
+    }
+    chunkToString(chunk, index) {
+        const info = chunk.retrievedContext ?? chunk.web;
+        return `${index + 1}. ${info.title} - ${info.uri}`;
+    }
+    textSuffix(_text, grounding) {
+        let ret = "\n";
+        const chunks = grounding?.metadata?.groundingChunks ?? [];
+        chunks.forEach((chunk, index) => {
+            ret = `${ret}${this.chunkToString(chunk, index)}\n`;
+        });
+        return ret;
+    }
+}
+export class MarkdownGoogleSearchOutputParser extends BaseGoogleSearchOutputParser {
+    segmentPrefix(_grounding, _support, _index) {
+        return undefined;
+    }
+    chunkLink(grounding, index) {
+        const chunk = grounding.metadata.groundingChunks[index];
+        const url = chunk.retrievedContext?.uri ?? chunk.web?.uri;
+        const num = index + 1;
+        return `[[${num}](${url})]`;
+    }
+    segmentSuffix(grounding, support, _index) {
+        let ret = "";
+        support.groundingChunkIndices.forEach((chunkIndex) => {
+            const link = this.chunkLink(grounding, chunkIndex);
+            ret = `${ret}${link}`;
+        });
+        return ret;
+    }
+    textPrefix(_text, _grounding) {
+        return undefined;
+    }
+    chunkSuffixLink(chunk, index) {
+        const num = index + 1;
+        const info = chunk.retrievedContext ?? chunk.web;
+        const url = info.uri;
+        const site = info.title;
+        return `${num}. [${site}](${url})`;
+    }
+    textSuffix(_text, grounding) {
+        let ret = "\n**Search Sources**\n";
+        const chunks = grounding.metadata.groundingChunks;
+        chunks.forEach((chunk, index) => {
+            ret = `${ret}${this.chunkSuffixLink(chunk, index)}\n`;
+        });
+        const search = this.searchSuggestion(grounding);
+        ret = `${ret}\n${search}`;
+        return ret;
+    }
+}

package/dist/types.cjs

@@ -14,7 +14,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GoogleAISafetyMethod = exports.GoogleAISafetyThreshold = exports.GoogleAISafetyCategory = void 0;
+exports.GeminiToolAttributes = exports.GeminiSearchToolAttributes = exports.GoogleAISafetyMethod = exports.GoogleAISafetyThreshold = exports.GoogleAISafetyCategory = void 0;
 __exportStar(require("./types-anthropic.cjs"), exports);
 exports.GoogleAISafetyCategory = {
     Harassment: "HARM_CATEGORY_HARASSMENT",
@@ -54,3 +54,12 @@ exports.GoogleAISafetyMethod = {
     Severity: "SEVERITY",
     Probability: "PROBABILITY",
 };
+exports.GeminiSearchToolAttributes = [
+    "googleSearchRetrieval",
+    "googleSearch",
+];
+exports.GeminiToolAttributes = [
+    "functionDeclaration",
+    "retrieval",
+    ...exports.GeminiSearchToolAttributes,
+];