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

lint fix

Browse Source
This commit is contained in:
Yongtae
2023-08-03 00:41:10 +09:00
parent fdb4450755
commit 7aaeab12c7
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/chatmodels/GoogleVertexAI/GoogleVertexAI.ts
+4 -6
View File
@@ -44,7 +44,7 @@ class GoogleVertexAI_ChatModels implements INode {
{ {
label: 'codechat-bison@001', label: 'codechat-bison@001',
name: 'codechat-bison@001' name: 'codechat-bison@001'
}, }
], ],
default: 'chat-bison', default: 'chat-bison',
optional: true optional: true
@@ -72,11 +72,11 @@ class GoogleVertexAI_ChatModels implements INode {
step: 0.1, step: 0.1,
optional: true, optional: true,
additionalParams: true additionalParams: true
}, }
] ]
} }
async init(nodeData: INodeData, _: string,): Promise<any> { async init(nodeData: INodeData, _: string): Promise<any> {
const temperature = nodeData.inputs?.temperature as string const temperature = nodeData.inputs?.temperature as string
const model = nodeData.inputs?.modelName as string const model = nodeData.inputs?.modelName as string
const maxOutputTokens = nodeData.inputs?.maxOutputTokens as string const maxOutputTokens = nodeData.inputs?.maxOutputTokens as string
@@ -84,14 +84,12 @@ class GoogleVertexAI_ChatModels implements INode {
const obj: Partial<GoogleVertexAIChatInput> = { const obj: Partial<GoogleVertexAIChatInput> = {
temperature: parseFloat(temperature), temperature: parseFloat(temperature),
model, model
} }
if (maxOutputTokens) obj.maxOutputTokens = parseInt(maxOutputTokens, 10) if (maxOutputTokens) obj.maxOutputTokens = parseInt(maxOutputTokens, 10)
if (topP) obj.topP = parseFloat(topP) if (topP) obj.topP = parseFloat(topP)
const chat_model = new ChatGoogleVertexAI(obj) const chat_model = new ChatGoogleVertexAI(obj)
return chat_model return chat_model
} }