This commit is contained in:
Yongtae 2023-08-03 00:04:22 +09:00
parent a26f76c705
commit 45069e10f2
1 changed files with 6 additions and 6 deletions

View File

@ -52,9 +52,9 @@ class GoogleVertexAI_LLMs implements INode {
{ {
label: 'code-gecko@001', label: 'code-gecko@001',
name: 'code-gecko@001' name: 'code-gecko@001'
}, }
], ],
default: 'text-bison', default: 'text-bison'
}, },
{ {
label: 'Temperature', label: 'Temperature',
@ -79,7 +79,7 @@ class GoogleVertexAI_LLMs implements INode {
step: 0.1, step: 0.1,
optional: true, optional: true,
additionalParams: true additionalParams: true
}, }
] ]
} }
@ -91,15 +91,15 @@ class GoogleVertexAI_LLMs implements INode {
const obj: Partial<GoogleVertexAITextInput> = { const obj: Partial<GoogleVertexAITextInput> = {
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 llm_model = new GoogleVertexAI(obj,) const llm_model = new GoogleVertexAI(obj)
return llm_model return llm_model
} }
} }
module.exports = { nodeClass: GoogleVertexAI_LLMs } module.exports = { nodeClass: GoogleVertexAI_LLMs }