@langchain/google-common 0.0.10 → 0.0.12

package/dist/chat_models.cjs

@@ -15,10 +15,73 @@ const auth_js_1 = require("./auth.cjs");
 const failed_handler_js_1 = require("./utils/failed_handler.cjs");
 const zod_to_gemini_parameters_js_1 = require("./utils/zod_to_gemini_parameters.cjs");
 class ChatConnection extends connection_js_1.AbstractGoogleLLMConnection {
+    constructor(fields, caller, client, streaming) {
+        super(fields, caller, client, streaming);
+        Object.defineProperty(this, "convertSystemMessageToHumanContent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.convertSystemMessageToHumanContent =
+            fields?.convertSystemMessageToHumanContent;
+    }
+    get useSystemInstruction() {
+        return typeof this.convertSystemMessageToHumanContent === "boolean"
+            ? !this.convertSystemMessageToHumanContent
+            : this.computeUseSystemInstruction;
+    }
+    get computeUseSystemInstruction() {
+        // This works on models from April 2024 and later
+        //   Vertex AI: gemini-1.5-pro and gemini-1.0-002 and later
+        //   AI Studio: gemini-1.5-pro-latest
+        if (this.modelFamily === "palm") {
+            return false;
+        }
+        else if (this.modelName === "gemini-1.0-pro-001") {
+            return false;
+        }
+        else if (this.modelName.startsWith("gemini-pro-vision")) {
+            return false;
+        }
+        else if (this.modelName.startsWith("gemini-1.0-pro-vision")) {
+            return false;
+        }
+        else if (this.modelName === "gemini-pro" && this.platform === "gai") {
+            // on AI Studio gemini-pro is still pointing at gemini-1.0-pro-001
+            return false;
+        }
+        return true;
+    }
     formatContents(input, _parameters) {
         return input
-            .map((msg, i) => (0, gemini_js_1.baseMessageToContent)(msg, input[i - 1]))
-            .reduce((acc, cur) => [...acc, ...cur]);
+            .map((msg, i) => (0, gemini_js_1.baseMessageToContent)(msg, input[i - 1], this.useSystemInstruction))
+            .reduce((acc, cur) => {
+            // Filter out the system content, since those don't belong
+            // in the actual content.
+            const hasNoSystem = cur.every((content) => content.role !== "system");
+            return hasNoSystem ? [...acc, ...cur] : acc;
+        }, []);
+    }
+    formatSystemInstruction(input, _parameters) {
+        if (!this.useSystemInstruction) {
+            return {};
+        }
+        let ret = {};
+        input.forEach((message, index) => {
+            if (message._getType() === "system") {
+                // For system types, we only want it if it is the first message,
+                // if it appears anywhere else, it should be an error.
+                if (index === 0) {
+                    // eslint-disable-next-line prefer-destructuring
+                    ret = (0, gemini_js_1.baseMessageToContent)(message, undefined, true)[0];
+                }
+                else {
+                    throw new Error("System messages are only permitted as the first passed message.");
+                }
+            }
+        });
+        return ret;
     }
 }
 function convertToGeminiTools(structuredTools) {
@@ -54,11 +117,12 @@ class ChatGoogleBase extends chat_models_1.BaseChatModel {
             writable: true,
             value: true
         });
+        // Set based on modelName
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: "gemini-pro"
+            value: void 0
         });
         Object.defineProperty(this, "modelName", {
             enumerable: true,
@@ -102,6 +166,13 @@ class ChatGoogleBase extends chat_models_1.BaseChatModel {
             writable: true,
             value: []
         });
+        // May intentionally be undefined, meaning to compute this.
+        Object.defineProperty(this, "convertSystemMessageToHumanContent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "safetyHandler", {
             enumerable: true,
             configurable: true,

package/dist/chat_models.d.ts

@@ -6,13 +6,19 @@ import { AIMessageChunk } from "@langchain/core/messages";
 import { BaseLanguageModelInput, StructuredOutputMethodOptions } from "@langchain/core/language_models/base";
 import type { z } from "zod";
 import { Runnable, RunnableInterface } from "@langchain/core/runnables";
+import { AsyncCaller } from "@langchain/core/utils/async_caller";
 import { StructuredToolInterface } from "@langchain/core/tools";
 import { GoogleAIBaseLLMInput, GoogleAIModelParams, GoogleAISafetySetting, GoogleConnectionParams, GooglePlatformType, GeminiContent, GoogleAIBaseLanguageModelCallOptions } from "./types.js";
 import { AbstractGoogleLLMConnection } from "./connection.js";
 import { GoogleAbstractedClient } from "./auth.js";
 import type { GoogleBaseLLMInput, GoogleAISafetyHandler, GoogleAISafetyParams } from "./types.js";
 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;
     formatContents(input: BaseMessage[], _parameters: GoogleAIModelParams): GeminiContent[];
+    formatSystemInstruction(input: BaseMessage[], _parameters: GoogleAIModelParams): GeminiContent;
 }
 /**
  * Input to chat model class.
@@ -33,6 +39,7 @@ export declare abstract class ChatGoogleBase<AuthOptions> extends BaseChatModel<
     topK: number;
     stopSequences: string[];
     safetySettings: GoogleAISafetySetting[];
+    convertSystemMessageToHumanContent: boolean | undefined;
     safetyHandler: GoogleAISafetyHandler;
     protected connection: ChatConnection<AuthOptions>;
     protected streamedConnection: ChatConnection<AuthOptions>;

package/dist/chat_models.js

@@ -12,10 +12,73 @@ import { ApiKeyGoogleAuth } from "./auth.js";
 import { ensureParams } from "./utils/failed_handler.js";
 import { zodToGeminiParameters } from "./utils/zod_to_gemini_parameters.js";
 class ChatConnection extends AbstractGoogleLLMConnection {
+    constructor(fields, caller, client, streaming) {
+        super(fields, caller, client, streaming);
+        Object.defineProperty(this, "convertSystemMessageToHumanContent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.convertSystemMessageToHumanContent =
+            fields?.convertSystemMessageToHumanContent;
+    }
+    get useSystemInstruction() {
+        return typeof this.convertSystemMessageToHumanContent === "boolean"
+            ? !this.convertSystemMessageToHumanContent
+            : this.computeUseSystemInstruction;
+    }
+    get computeUseSystemInstruction() {
+        // This works on models from April 2024 and later
+        //   Vertex AI: gemini-1.5-pro and gemini-1.0-002 and later
+        //   AI Studio: gemini-1.5-pro-latest
+        if (this.modelFamily === "palm") {
+            return false;
+        }
+        else if (this.modelName === "gemini-1.0-pro-001") {
+            return false;
+        }
+        else if (this.modelName.startsWith("gemini-pro-vision")) {
+            return false;
+        }
+        else if (this.modelName.startsWith("gemini-1.0-pro-vision")) {
+            return false;
+        }
+        else if (this.modelName === "gemini-pro" && this.platform === "gai") {
+            // on AI Studio gemini-pro is still pointing at gemini-1.0-pro-001
+            return false;
+        }
+        return true;
+    }
     formatContents(input, _parameters) {
         return input
-            .map((msg, i) => baseMessageToContent(msg, input[i - 1]))
-            .reduce((acc, cur) => [...acc, ...cur]);
+            .map((msg, i) => baseMessageToContent(msg, input[i - 1], this.useSystemInstruction))
+            .reduce((acc, cur) => {
+            // Filter out the system content, since those don't belong
+            // in the actual content.
+            const hasNoSystem = cur.every((content) => content.role !== "system");
+            return hasNoSystem ? [...acc, ...cur] : acc;
+        }, []);
+    }
+    formatSystemInstruction(input, _parameters) {
+        if (!this.useSystemInstruction) {
+            return {};
+        }
+        let ret = {};
+        input.forEach((message, index) => {
+            if (message._getType() === "system") {
+                // For system types, we only want it if it is the first message,
+                // if it appears anywhere else, it should be an error.
+                if (index === 0) {
+                    // eslint-disable-next-line prefer-destructuring
+                    ret = baseMessageToContent(message, undefined, true)[0];
+                }
+                else {
+                    throw new Error("System messages are only permitted as the first passed message.");
+                }
+            }
+        });
+        return ret;
     }
 }
 function convertToGeminiTools(structuredTools) {
@@ -51,11 +114,12 @@ export class ChatGoogleBase extends BaseChatModel {
             writable: true,
             value: true
         });
+        // Set based on modelName
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: "gemini-pro"
+            value: void 0
         });
         Object.defineProperty(this, "modelName", {
             enumerable: true,
@@ -99,6 +163,13 @@ export class ChatGoogleBase extends BaseChatModel {
             writable: true,
             value: []
         });
+        // May intentionally be undefined, meaning to compute this.
+        Object.defineProperty(this, "convertSystemMessageToHumanContent", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "safetyHandler", {
             enumerable: true,
             configurable: true,

package/dist/connection.cjs

@@ -28,21 +28,26 @@ class GoogleConnection {
         this.streaming = streaming ?? false;
     }
     async _clientInfoHeaders() {
-        const clientLibraryVersion = await this._clientLibraryVersion();
+        const { userAgent, clientLibraryVersion } = await this._getClientInfo();
         return {
-            "User-Agent": clientLibraryVersion,
+            "User-Agent": userAgent,
+            "Client-Info": clientLibraryVersion,
         };
     }
-    async _clientLibraryVersion() {
+    async _getClientInfo() {
         const env = await (0, env_1.getRuntimeEnvironment)();
         const langchain = env?.library ?? "langchain-js";
-        const langchainVersion = env?.libraryVersion ?? "0";
+        // TODO: Add an API for getting the current LangChain version
+        const langchainVersion = "0";
         const moduleName = await this._moduleName();
-        let ret = `${langchain}/${langchainVersion}`;
+        let clientLibraryVersion = `${langchain}/${langchainVersion}`;
         if (moduleName && moduleName.length) {
-            ret = `${ret}-${moduleName}`;
+            clientLibraryVersion = `${clientLibraryVersion}-${moduleName}`;
         }
-        return ret;
+        return {
+            userAgent: clientLibraryVersion,
+            clientLibraryVersion: `${langchainVersion}-${moduleName}`,
+        };
     }
     async _moduleName() {
         return this.constructor.name;
@@ -208,11 +213,15 @@ class AbstractGoogleLLMConnection extends GoogleAIConnection {
             topP: parameters.topP,
             maxOutputTokens: parameters.maxOutputTokens,
             stopSequences: parameters.stopSequences,
+            responseMimeType: parameters.responseMimeType,
         };
     }
     formatSafetySettings(_input, parameters) {
         return parameters.safetySettings ?? [];
     }
+    formatSystemInstruction(_input, _parameters) {
+        return {};
+    }
     // Borrowed from the OpenAI invocation params test
     isStructuredToolArray(tools) {
         return (tools !== undefined &&
@@ -253,6 +262,7 @@ class AbstractGoogleLLMConnection extends GoogleAIConnection {
         const generationConfig = this.formatGenerationConfig(input, parameters);
         const tools = this.formatTools(input, parameters);
         const safetySettings = this.formatSafetySettings(input, parameters);
+        const systemInstruction = this.formatSystemInstruction(input, parameters);
         const ret = {
             contents,
             generationConfig,
@@ -263,6 +273,11 @@ class AbstractGoogleLLMConnection extends GoogleAIConnection {
         if (safetySettings && safetySettings.length) {
             ret.safetySettings = safetySettings;
         }
+        if (systemInstruction?.role &&
+            systemInstruction?.parts &&
+            systemInstruction?.parts?.length) {
+            ret.systemInstruction = systemInstruction;
+        }
         return ret;
     }
 }

package/dist/connection.d.ts

@@ -11,7 +11,10 @@ export declare abstract class GoogleConnection<CallOptions extends AsyncCallerCa
     abstract buildUrl(): Promise<string>;
     abstract buildMethod(): GoogleAbstractedClientOpsMethod;
     _clientInfoHeaders(): Promise<Record<string, string>>;
-    _clientLibraryVersion(): Promise<string>;
+    _getClientInfo(): Promise<{
+        userAgent: string;
+        clientLibraryVersion: string;
+    }>;
     _moduleName(): Promise<string>;
     _request(data: unknown | undefined, options: CallOptions): Promise<ResponseType>;
 }
@@ -45,6 +48,7 @@ export declare abstract class AbstractGoogleLLMConnection<MessageType, AuthOptio
     abstract formatContents(input: MessageType, parameters: GoogleAIModelRequestParams): GeminiContent[];
     formatGenerationConfig(_input: MessageType, parameters: GoogleAIModelRequestParams): GeminiGenerationConfig;
     formatSafetySettings(_input: MessageType, parameters: GoogleAIModelRequestParams): GeminiSafetySetting[];
+    formatSystemInstruction(_input: MessageType, _parameters: GoogleAIModelRequestParams): GeminiContent;
     isStructuredToolArray(tools?: unknown[]): tools is StructuredToolInterface[];
     structuredToolToFunctionDeclaration(tool: StructuredToolInterface): GeminiFunctionDeclaration;
     structuredToolsToGeminiTools(tools: StructuredToolInterface[]): GeminiTool[];

package/dist/connection.js

@@ -25,21 +25,26 @@ export class GoogleConnection {
         this.streaming = streaming ?? false;
     }
     async _clientInfoHeaders() {
-        const clientLibraryVersion = await this._clientLibraryVersion();
+        const { userAgent, clientLibraryVersion } = await this._getClientInfo();
         return {
-            "User-Agent": clientLibraryVersion,
+            "User-Agent": userAgent,
+            "Client-Info": clientLibraryVersion,
         };
     }
-    async _clientLibraryVersion() {
+    async _getClientInfo() {
         const env = await getRuntimeEnvironment();
         const langchain = env?.library ?? "langchain-js";
-        const langchainVersion = env?.libraryVersion ?? "0";
+        // TODO: Add an API for getting the current LangChain version
+        const langchainVersion = "0";
         const moduleName = await this._moduleName();
-        let ret = `${langchain}/${langchainVersion}`;
+        let clientLibraryVersion = `${langchain}/${langchainVersion}`;
         if (moduleName && moduleName.length) {
-            ret = `${ret}-${moduleName}`;
+            clientLibraryVersion = `${clientLibraryVersion}-${moduleName}`;
         }
-        return ret;
+        return {
+            userAgent: clientLibraryVersion,
+            clientLibraryVersion: `${langchainVersion}-${moduleName}`,
+        };
     }
     async _moduleName() {
         return this.constructor.name;
@@ -202,11 +207,15 @@ export class AbstractGoogleLLMConnection extends GoogleAIConnection {
             topP: parameters.topP,
             maxOutputTokens: parameters.maxOutputTokens,
             stopSequences: parameters.stopSequences,
+            responseMimeType: parameters.responseMimeType,
         };
     }
     formatSafetySettings(_input, parameters) {
         return parameters.safetySettings ?? [];
     }
+    formatSystemInstruction(_input, _parameters) {
+        return {};
+    }
     // Borrowed from the OpenAI invocation params test
     isStructuredToolArray(tools) {
         return (tools !== undefined &&
@@ -247,6 +256,7 @@ export class AbstractGoogleLLMConnection extends GoogleAIConnection {
         const generationConfig = this.formatGenerationConfig(input, parameters);
         const tools = this.formatTools(input, parameters);
         const safetySettings = this.formatSafetySettings(input, parameters);
+        const systemInstruction = this.formatSystemInstruction(input, parameters);
         const ret = {
             contents,
             generationConfig,
@@ -257,6 +267,11 @@ export class AbstractGoogleLLMConnection extends GoogleAIConnection {
         if (safetySettings && safetySettings.length) {
             ret.safetySettings = safetySettings;
         }
+        if (systemInstruction?.role &&
+            systemInstruction?.parts &&
+            systemInstruction?.parts?.length) {
+            ret.systemInstruction = systemInstruction;
+        }
         return ret;
     }
 }

package/dist/llms.cjs

@@ -107,6 +107,12 @@ class GoogleBaseLLM extends llms_1.LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "responseMimeType", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "text/plain"
+        });
         Object.defineProperty(this, "connection", {
             enumerable: true,
             configurable: true,

package/dist/llms.d.ts

@@ -3,7 +3,7 @@ import { LLM } from "@langchain/core/language_models/llms";
 import { type BaseLanguageModelCallOptions, BaseLanguageModelInput } from "@langchain/core/language_models/base";
 import { BaseMessage, MessageContent } from "@langchain/core/messages";
 import { AbstractGoogleLLMConnection } from "./connection.js";
-import { GoogleAIBaseLLMInput, GoogleAIModelParams, GoogleAISafetySetting, GooglePlatformType, GeminiContent } from "./types.js";
+import { GoogleAIBaseLLMInput, GoogleAIModelParams, GoogleAISafetySetting, GooglePlatformType, GeminiContent, GoogleAIResponseMimeType } from "./types.js";
 import { GoogleAbstractedClient } from "./auth.js";
 import { ChatGoogleBase } from "./chat_models.js";
 import type { GoogleBaseLLMInput, GoogleAISafetyHandler } from "./types.js";
@@ -27,6 +27,7 @@ export declare abstract class GoogleBaseLLM<AuthOptions> extends LLM<BaseLanguag
     stopSequences: string[];
     safetySettings: GoogleAISafetySetting[];
     safetyHandler: GoogleAISafetyHandler;
+    responseMimeType: GoogleAIResponseMimeType;
     protected connection: GoogleLLMConnection<AuthOptions>;
     protected streamedConnection: GoogleLLMConnection<AuthOptions>;
     constructor(fields?: GoogleBaseLLMInput<AuthOptions>);

package/dist/llms.js

@@ -104,6 +104,12 @@ export class GoogleBaseLLM extends LLM {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "responseMimeType", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "text/plain"
+        });
         Object.defineProperty(this, "connection", {
             enumerable: true,
             configurable: true,

package/dist/types.d.ts

@@ -37,6 +37,7 @@ export interface GoogleAISafetySetting {
     category: string;
     threshold: string;
 }
+export type GoogleAIResponseMimeType = "text/plain" | "application/json";
 export interface GoogleAIModelParams {
     /** Model to use */
     model?: string;
@@ -73,6 +74,17 @@ export interface GoogleAIModelParams {
     topK?: number;
     stopSequences?: string[];
     safetySettings?: GoogleAISafetySetting[];
+    convertSystemMessageToHumanContent?: boolean;
+    /**
+     * Available for `gemini-1.5-pro`.
+     * The output format of the generated candidate text.
+     * Supported MIME types:
+     *  - `text/plain`: Text output.
+     *  - `application/json`: JSON response in the candidates.
+     *
+     * @default "text/plain"
+     */
+    responseMimeType?: GoogleAIResponseMimeType;
 }
 /**
  * The params which can be passed to the API at request time.
@@ -128,7 +140,7 @@ export type GeminiSafetyRating = {
     category: string;
     probability: string;
 } & Record<string, unknown>;
-export type GeminiRole = "user" | "model" | "function";
+export type GeminiRole = "system" | "user" | "model" | "function";
 export interface GeminiContent {
     parts: GeminiPart[];
     role: GeminiRole;
@@ -159,9 +171,11 @@ export interface GeminiGenerationConfig {
     temperature?: number;
     topP?: number;
     topK?: number;
+    responseMimeType?: GoogleAIResponseMimeType;
 }
 export interface GeminiRequest {
     contents?: GeminiContent[];
+    systemInstruction?: GeminiContent;
     tools?: GeminiTool[];
     safetySettings?: GeminiSafetySetting[];
     generationConfig?: GeminiGenerationConfig;
@@ -202,4 +216,12 @@ export interface GoogleAISafetyHandler {
 export interface GoogleAISafetyParams {
     safetyHandler?: GoogleAISafetyHandler;
 }
+export type GeminiJsonSchema = Record<string, unknown> & {
+    properties?: Record<string, GeminiJsonSchema>;
+    type: GeminiFunctionSchemaType;
+};
+export interface GeminiJsonSchemaDirty extends GeminiJsonSchema {
+    properties?: Record<string, GeminiJsonSchemaDirty>;
+    additionalProperties?: boolean;
+}
 export {};

package/dist/utils/common.cjs

@@ -24,6 +24,14 @@ function copyAIModelParamsInto(params, options, target) {
         options?.stopSequences ?? params?.stopSequences ?? target.stopSequences;
     ret.safetySettings =
         options?.safetySettings ?? params?.safetySettings ?? target.safetySettings;
+    ret.convertSystemMessageToHumanContent =
+        options?.convertSystemMessageToHumanContent ??
+            params?.convertSystemMessageToHumanContent ??
+            target?.convertSystemMessageToHumanContent;
+    ret.responseMimeType =
+        options?.responseMimeType ??
+            params?.responseMimeType ??
+            target?.responseMimeType;
     ret.tools = options?.tools;
     // Ensure tools are formatted properly for Gemini
     const geminiTools = options?.tools

package/dist/utils/common.js

@@ -20,6 +20,14 @@ export function copyAIModelParamsInto(params, options, target) {
         options?.stopSequences ?? params?.stopSequences ?? target.stopSequences;
     ret.safetySettings =
         options?.safetySettings ?? params?.safetySettings ?? target.safetySettings;
+    ret.convertSystemMessageToHumanContent =
+        options?.convertSystemMessageToHumanContent ??
+            params?.convertSystemMessageToHumanContent ??
+            target?.convertSystemMessageToHumanContent;
+    ret.responseMimeType =
+        options?.responseMimeType ??
+            params?.responseMimeType ??
+            target?.responseMimeType;
     ret.tools = options?.tools;
     // Ensure tools are formatted properly for Gemini
     const geminiTools = options?.tools

package/dist/utils/gemini.cjs

@@ -158,11 +158,13 @@ function roleMessageToContent(role, message) {
         },
     ];
 }
-function systemMessageToContent(message) {
-    return [
-        ...roleMessageToContent("user", message),
-        ...roleMessageToContent("model", new messages_1.AIMessage("Ok")),
-    ];
+function systemMessageToContent(message, useSystemInstruction) {
+    return useSystemInstruction
+        ? roleMessageToContent("system", message)
+        : [
+            ...roleMessageToContent("user", message),
+            ...roleMessageToContent("model", new messages_1.AIMessage("Ok")),
+        ];
 }
 function toolMessageToContent(message, prevMessage) {
     const contentStr = typeof message.content === "string"
@@ -211,11 +213,11 @@ function toolMessageToContent(message, prevMessage) {
         ];
     }
 }
-function baseMessageToContent(message, prevMessage) {
+function baseMessageToContent(message, prevMessage, useSystemInstruction) {
     const type = message._getType();
     switch (type) {
         case "system":
-            return systemMessageToContent(message);
+            return systemMessageToContent(message, useSystemInstruction);
         case "human":
             return roleMessageToContent("user", message);
         case "ai":

package/dist/utils/gemini.d.ts

@@ -2,7 +2,7 @@ import { AIMessageFields, BaseMessage, BaseMessageChunk, BaseMessageFields, Mess
 import { ChatGeneration, ChatGenerationChunk, ChatResult, Generation } from "@langchain/core/outputs";
 import type { GoogleLLMResponse, GoogleAIModelParams, GeminiPart, GeminiContent, GenerateContentResponseData, GoogleAISafetyHandler } from "../types.js";
 export declare function messageContentToParts(content: MessageContent): GeminiPart[];
-export declare function baseMessageToContent(message: BaseMessage, prevMessage?: BaseMessage): GeminiContent[];
+export declare function baseMessageToContent(message: BaseMessage, prevMessage: BaseMessage | undefined, useSystemInstruction: boolean): GeminiContent[];
 export declare function partsToMessageContent(parts: GeminiPart[]): MessageContent;
 interface FunctionCall {
     name: string;

package/dist/utils/gemini.js

@@ -154,11 +154,13 @@ function roleMessageToContent(role, message) {
         },
     ];
 }
-function systemMessageToContent(message) {
-    return [
-        ...roleMessageToContent("user", message),
-        ...roleMessageToContent("model", new AIMessage("Ok")),
-    ];
+function systemMessageToContent(message, useSystemInstruction) {
+    return useSystemInstruction
+        ? roleMessageToContent("system", message)
+        : [
+            ...roleMessageToContent("user", message),
+            ...roleMessageToContent("model", new AIMessage("Ok")),
+        ];
 }
 function toolMessageToContent(message, prevMessage) {
     const contentStr = typeof message.content === "string"
@@ -207,11 +209,11 @@ function toolMessageToContent(message, prevMessage) {
         ];
     }
 }
-export function baseMessageToContent(message, prevMessage) {
+export function baseMessageToContent(message, prevMessage, useSystemInstruction) {
     const type = message._getType();
     switch (type) {
         case "system":
-            return systemMessageToContent(message);
+            return systemMessageToContent(message, useSystemInstruction);
         case "human":
             return roleMessageToContent("user", message);
         case "ai":

package/dist/utils/zod_to_gemini_parameters.cjs

@@ -3,14 +3,35 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.zodToGeminiParameters = void 0;
 const zod_to_json_schema_1 = require("zod-to-json-schema");
+function removeAdditionalProperties(schema) {
+    const updatedSchema = { ...schema };
+    if (Object.hasOwn(updatedSchema, "additionalProperties")) {
+        delete updatedSchema.additionalProperties;
+    }
+    if (updatedSchema.properties) {
+        const keys = Object.keys(updatedSchema.properties);
+        removeProperties(updatedSchema.properties, keys, 0);
+    }
+    return updatedSchema;
+}
+function removeProperties(properties, keys, index) {
+    if (index >= keys.length) {
+        return;
+    }
+    const key = keys[index];
+    // eslint-disable-next-line no-param-reassign
+    properties[key] = removeAdditionalProperties(properties[key]);
+    removeProperties(properties, keys, index + 1);
+}
 function zodToGeminiParameters(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 zodObj) {
     // Gemini doesn't accept either the $schema or additionalProperties
     // attributes, so we need to explicitly remove them.
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const jsonSchema = (0, zod_to_json_schema_1.zodToJsonSchema)(zodObj);
-    const { $schema, additionalProperties, ...rest } = jsonSchema;
+    // const jsonSchema = zodToJsonSchema(zodObj) as any;
+    const jsonSchema = removeAdditionalProperties((0, zod_to_json_schema_1.zodToJsonSchema)(zodObj));
+    const { $schema, ...rest } = jsonSchema;
     return rest;
 }
 exports.zodToGeminiParameters = zodToGeminiParameters;

package/dist/utils/zod_to_gemini_parameters.js

@@ -1,12 +1,33 @@
 /* eslint-disable @typescript-eslint/no-unused-vars */
 import { zodToJsonSchema } from "zod-to-json-schema";
+function removeAdditionalProperties(schema) {
+    const updatedSchema = { ...schema };
+    if (Object.hasOwn(updatedSchema, "additionalProperties")) {
+        delete updatedSchema.additionalProperties;
+    }
+    if (updatedSchema.properties) {
+        const keys = Object.keys(updatedSchema.properties);
+        removeProperties(updatedSchema.properties, keys, 0);
+    }
+    return updatedSchema;
+}
+function removeProperties(properties, keys, index) {
+    if (index >= keys.length) {
+        return;
+    }
+    const key = keys[index];
+    // eslint-disable-next-line no-param-reassign
+    properties[key] = removeAdditionalProperties(properties[key]);
+    removeProperties(properties, keys, index + 1);
+}
 export function zodToGeminiParameters(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 zodObj) {
     // Gemini doesn't accept either the $schema or additionalProperties
     // attributes, so we need to explicitly remove them.
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const jsonSchema = zodToJsonSchema(zodObj);
-    const { $schema, additionalProperties, ...rest } = jsonSchema;
+    // const jsonSchema = zodToJsonSchema(zodObj) as any;
+    const jsonSchema = removeAdditionalProperties(zodToJsonSchema(zodObj));
+    const { $schema, ...rest } = jsonSchema;
     return rest;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/google-common",
-  "version": "0.0.10",
+  "version": "0.0.12",
   "description": "Core types and classes for Google services.",
   "type": "module",
   "engines": {