@langchain/google-genai 0.0.13 → 0.0.15

package/dist/chat_models.cjs

@@ -171,8 +171,22 @@ class ChatGoogleGenerativeAI extends chat_models_1.BaseChatModel {
                 topP: this.topP,
                 topK: this.topK,
             },
+        }, {
+            apiVersion: fields?.apiVersion,
+            baseUrl: fields?.baseUrl,
         });
     }
+    getLsParams(options) {
+        const params = this.invocationParams(options);
+        return {
+            ls_provider: "google_genai",
+            ls_model_name: this.model,
+            ls_model_type: "chat",
+            ls_temperature: params.temperature ?? undefined,
+            ls_max_tokens: params.maxOutputTokens ?? undefined,
+            ls_stop: options.stop,
+        };
+    }
     _combineLLMOutput() {
         return [];
     }

package/dist/chat_models.d.ts

@@ -2,7 +2,7 @@ import type { SafetySetting } from "@google/generative-ai";
 import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
 import { BaseMessage } from "@langchain/core/messages";
 import { ChatGenerationChunk, ChatResult } from "@langchain/core/outputs";
-import { BaseChatModel, type BaseChatModelParams } from "@langchain/core/language_models/chat_models";
+import { BaseChatModel, LangSmithParams, type BaseChatModelParams } from "@langchain/core/language_models/chat_models";
 export type BaseMessageExamplePair = {
     input: BaseMessage;
     output: BaseMessage;
@@ -84,6 +84,14 @@ export interface GoogleGenerativeAIChatInput extends BaseChatModelParams {
      * Google API key to use
      */
     apiKey?: string;
+    /**
+     * Google API version to use
+     */
+    apiVersion?: string;
+    /**
+     * Google API base URL to use
+     */
+    baseUrl?: string;
     /** Whether to stream the results or not */
     streaming?: boolean;
 }
@@ -135,6 +143,7 @@ export declare class ChatGoogleGenerativeAI extends BaseChatModel implements Goo
     private client;
     get _isMultimodalModel(): boolean;
     constructor(fields?: GoogleGenerativeAIChatInput);
+    protected getLsParams(options: this["ParsedCallOptions"]): LangSmithParams;
     _combineLLMOutput(): never[];
     _llmType(): string;
     _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;

package/dist/chat_models.js

@@ -168,8 +168,22 @@ export class ChatGoogleGenerativeAI extends BaseChatModel {
                 topP: this.topP,
                 topK: this.topK,
             },
+        }, {
+            apiVersion: fields?.apiVersion,
+            baseUrl: fields?.baseUrl,
         });
     }
+    getLsParams(options) {
+        const params = this.invocationParams(options);
+        return {
+            ls_provider: "google_genai",
+            ls_model_name: this.model,
+            ls_model_type: "chat",
+            ls_temperature: params.temperature ?? undefined,
+            ls_max_tokens: params.maxOutputTokens ?? undefined,
+            ls_stop: options.stop,
+        };
+    }
     _combineLLMOutput() {
         return [];
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/google-genai",
-  "version": "0.0.13",
+  "version": "0.0.15",
   "description": "Sample integration for LangChain.js",
   "type": "module",
   "engines": {
@@ -26,10 +26,10 @@
     "lint:fix": "yarn lint:eslint --fix && yarn lint:dpdm",
     "clean": "rm -rf dist/ && NODE_OPTIONS=--max-old-space-size=4096 yarn lc-build --config ./langchain.config.js --create-entrypoints --pre",
     "prepack": "yarn build",
-    "test": "NODE_OPTIONS=--experimental-vm-modules jest --testPathIgnorePatterns=\\.int\\.test.ts --testTimeout 30000 --maxWorkers=50%",
-    "test:watch": "NODE_OPTIONS=--experimental-vm-modules jest --watch --testPathIgnorePatterns=\\.int\\.test.ts",
-    "test:single": "NODE_OPTIONS=--experimental-vm-modules yarn run jest --config jest.config.cjs --testTimeout 100000",
-    "test:int": "NODE_OPTIONS=--experimental-vm-modules jest --testPathPattern=\\.int\\.test.ts --testTimeout 100000 --maxWorkers=50%",
+    "test": "yarn build:deps && NODE_OPTIONS=--experimental-vm-modules jest --testPathIgnorePatterns=\\.int\\.test.ts --testTimeout 30000 --maxWorkers=50%",
+    "test:watch": "yarn build:deps && NODE_OPTIONS=--experimental-vm-modules jest --watch --testPathIgnorePatterns=\\.int\\.test.ts",
+    "test:single": "yarn build:deps && NODE_OPTIONS=--experimental-vm-modules yarn run jest --config jest.config.cjs --testTimeout 100000",
+    "test:int": "yarn build:deps && NODE_OPTIONS=--experimental-vm-modules jest --testPathPattern=\\.int\\.test.ts --testTimeout 100000 --maxWorkers=50%",
     "format": "prettier --config .prettierrc --write \"src\"",
     "format:check": "prettier --config .prettierrc --check \"src\"",
     "move-cjs-to-dist": "yarn lc-build --config ./langchain.config.js --move-cjs-dist",
@@ -40,7 +40,7 @@
   "license": "MIT",
   "dependencies": {
     "@google/generative-ai": "^0.7.0",
-    "@langchain/core": "<0.3.0 || >0.1.0"
+    "@langchain/core": ">0.1.5  <0.3.0"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",