@langchain/google-genai 0.0.14 → 0.0.15

package/dist/chat_models.cjs

@@ -108,18 +108,6 @@ class ChatGoogleGenerativeAI extends chat_models_1.BaseChatModel {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "apiVersion", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "v1"
-        });
-        Object.defineProperty(this, "baseUrl", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "https://generativeai.googleapis.com"
-        });
         Object.defineProperty(this, "streaming", {
             enumerable: true,
             configurable: true,
@@ -184,10 +172,21 @@ class ChatGoogleGenerativeAI extends chat_models_1.BaseChatModel {
                 topK: this.topK,
             },
         }, {
-            apiVersion: this.apiVersion,
-            baseUrl: this.baseUrl,
+            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;
@@ -139,12 +139,11 @@ export declare class ChatGoogleGenerativeAI extends BaseChatModel implements Goo
     stopSequences: string[];
     safetySettings?: SafetySetting[];
     apiKey?: string;
-    apiVersion?: string;
-    baseUrl?: string;
     streaming: boolean;
     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

@@ -105,18 +105,6 @@ export class ChatGoogleGenerativeAI extends BaseChatModel {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "apiVersion", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "v1"
-        });
-        Object.defineProperty(this, "baseUrl", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "https://generativeai.googleapis.com"
-        });
         Object.defineProperty(this, "streaming", {
             enumerable: true,
             configurable: true,
@@ -181,10 +169,21 @@ export class ChatGoogleGenerativeAI extends BaseChatModel {
                 topK: this.topK,
             },
         }, {
-            apiVersion: this.apiVersion,
-            baseUrl: this.baseUrl,
+            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.14",
+  "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",