Support getNumTokens for ChatGoogleGenerativeAI (#4149)

hans00 · web-flow · commit 654bd48849bc · 2025-04-14T23:21:15.000+08:00
diff --git a/packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/FlowiseChatGoogleGenerativeAI.ts b/packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/FlowiseChatGoogleGenerativeAI.ts
@@ -209,6 +209,12 @@ class LangchainChatGoogleGenerativeAI
         }
     }
 
+    async getNumTokens(prompt: BaseMessage[]) {
+        const contents = convertBaseMessagesToContent(prompt, this._isMultimodalModel)
+        const { totalTokens } = await this.client.countTokens({ contents })
+        return totalTokens
+    }
+
     async _generateNonStreaming(
         prompt: Content[],
         options: this['ParsedCallOptions'],

Original file line number	Diff line number	Diff line change
`@@ -209,6 +209,12 @@ class LangchainChatGoogleGenerativeAI`
`209`	`209`	`}`
`210`	`210`	`}`
`211`	`211`
	`212`	`+ async getNumTokens(prompt: BaseMessage[]) {`
	`213`	`+ const contents = convertBaseMessagesToContent(prompt, this._isMultimodalModel)`
	`214`	`+ const { totalTokens } = await this.client.countTokens({ contents })`
	`215`	`+ return totalTokens`
	`216`	`+ }`
	`217`	`+`
`212`	`218`	`async _generateNonStreaming(`
`213`	`219`	`prompt: Content[],`
`214`	`220`	`options: this['ParsedCallOptions'],`