@langchain/google-genai 0.0.4 → 0.0.6

package/README.md

@@ -21,14 +21,14 @@ You can do so by adding appropriate field to your project's `package.json` like
     "langchain": "0.0.207"
   },
   "resolutions": {
-    "@langchain/core": "0.1.1"
+    "@langchain/core": "0.1.5"
   },
   "overrides": {
-    "@langchain/core": "0.1.1"
+    "@langchain/core": "0.1.5"
   },
   "pnpm": {
     "overrides": {
-      "@langchain/core": "0.1.1"
+      "@langchain/core": "0.1.5"
     }
   }
 }

package/dist/chat_models.cjs

@@ -163,7 +163,7 @@ class ChatGoogleGenerativeAI extends chat_models_1.BaseChatModel {
     _llmType() {
         return "googlegenerativeai";
     }
-    async _generate(messages, options, _runManager) {
+    async _generate(messages, options, runManager) {
         const prompt = (0, utils_js_1.convertBaseMessagesToContent)(messages, this._isMultimodalModel);
         const res = await this.caller.callWithOptions({ signal: options?.signal }, async () => {
             let output;
@@ -182,9 +182,11 @@ class ChatGoogleGenerativeAI extends chat_models_1.BaseChatModel {
             }
             return output;
         });
-        return (0, utils_js_1.mapGenerateContentResultToChatResult)(res.response);
+        const generationResult = (0, utils_js_1.mapGenerateContentResultToChatResult)(res.response);
+        await runManager?.handleLLMNewToken(generationResult.generations[0].text ?? "");
+        return generationResult;
     }
-    async *_streamResponseChunks(messages, options, _runManager) {
+    async *_streamResponseChunks(messages, options, runManager) {
         const prompt = (0, utils_js_1.convertBaseMessagesToContent)(messages, this._isMultimodalModel);
         const stream = await this.caller.callWithOptions({ signal: options?.signal }, async () => {
             const { stream } = await this.client.generateContentStream({
@@ -198,6 +200,7 @@ class ChatGoogleGenerativeAI extends chat_models_1.BaseChatModel {
                 continue;
             }
             yield chunk;
+            await runManager?.handleLLMNewToken(chunk.text ?? "");
         }
     }
 }

package/dist/chat_models.d.ts

@@ -125,6 +125,6 @@ export declare class ChatGoogleGenerativeAI extends BaseChatModel implements Goo
     constructor(fields?: GoogleGenerativeAIChatInput);
     _combineLLMOutput(): never[];
     _llmType(): string;
-    _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], _runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
-    _streamResponseChunks(messages: BaseMessage[], options: this["ParsedCallOptions"], _runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
+    _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
+    _streamResponseChunks(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
 }

package/dist/chat_models.js

@@ -160,7 +160,7 @@ export class ChatGoogleGenerativeAI extends BaseChatModel {
     _llmType() {
         return "googlegenerativeai";
     }
-    async _generate(messages, options, _runManager) {
+    async _generate(messages, options, runManager) {
         const prompt = convertBaseMessagesToContent(messages, this._isMultimodalModel);
         const res = await this.caller.callWithOptions({ signal: options?.signal }, async () => {
             let output;
@@ -179,9 +179,11 @@ export class ChatGoogleGenerativeAI extends BaseChatModel {
             }
             return output;
         });
-        return mapGenerateContentResultToChatResult(res.response);
+        const generationResult = mapGenerateContentResultToChatResult(res.response);
+        await runManager?.handleLLMNewToken(generationResult.generations[0].text ?? "");
+        return generationResult;
     }
-    async *_streamResponseChunks(messages, options, _runManager) {
+    async *_streamResponseChunks(messages, options, runManager) {
         const prompt = convertBaseMessagesToContent(messages, this._isMultimodalModel);
         const stream = await this.caller.callWithOptions({ signal: options?.signal }, async () => {
             const { stream } = await this.client.generateContentStream({
@@ -195,6 +197,7 @@ export class ChatGoogleGenerativeAI extends BaseChatModel {
                 continue;
             }
             yield chunk;
+            await runManager?.handleLLMNewToken(chunk.text ?? "");
         }
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/google-genai",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "description": "Sample integration for LangChain.js",
   "type": "module",
   "engines": {
@@ -23,7 +23,6 @@
     "lint:fix": "yarn lint --fix",
     "clean": "rm -rf dist/ && NODE_OPTIONS=--max-old-space-size=4096 node scripts/create-entrypoints.js pre",
     "prepack": "yarn build",
-    "release": "release-it --only-version --config .release-it.json",
     "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",
@@ -35,7 +34,7 @@
   "license": "MIT",
   "dependencies": {
     "@google/generative-ai": "^0.1.0",
-    "@langchain/core": "~0.1.2"
+    "@langchain/core": "~0.1.5"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
@@ -57,6 +56,7 @@
     "jest": "^29.5.0",
     "jest-environment-node": "^29.6.4",
     "prettier": "^2.8.3",
+    "release-it": "^15.10.1",
     "rollup": "^4.5.2",
     "ts-jest": "^29.1.0",
     "typescript": "<5.2.0"