farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 21:00:58 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support getNumTokens for ChatGoogleGenerativeAI (#4149)

Browse Source
This commit is contained in:
Hans
2025-04-14 23:21:15 +08:00
committed by GitHub
parent c318fc57e9
commit 654bd48849
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/FlowiseChatGoogleGenerativeAI.ts
+6
View File
@@ -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( async _generateNonStreaming(
prompt: Content[], prompt: Content[],
options: this['ParsedCallOptions'], options: this['ParsedCallOptions'],