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