fix: temperature convert to float

This commit is contained in:
Jeffrey-Wang
2023-06-29 21:23:21 +08:00
parent 7a8db1dff3
commit 184a783847
8 changed files with 8 additions and 8 deletions
@@ -179,7 +179,7 @@ class AzureOpenAI_LLMs implements INode {
const streaming = nodeData.inputs?.streaming as boolean
const obj: Partial<AzureOpenAIInput> & Partial<OpenAIInput> = {
temperature: parseInt(temperature, 10),
temperature: parseFloat(temperature),
modelName,
azureOpenAIApiKey,
azureOpenAIApiInstanceName,
@@ -87,7 +87,7 @@ class Cohere_LLMs implements INode {
if (maxTokens) obj.maxTokens = parseInt(maxTokens, 10)
if (modelName) obj.model = modelName
if (temperature) obj.temperature = parseInt(temperature, 10)
if (temperature) obj.temperature = parseFloat(temperature)
const model = new Cohere(obj)
return model
@@ -89,7 +89,7 @@ class HuggingFaceInference_LLMs implements INode {
apiKey
}
if (temperature) obj.temperature = parseInt(temperature, 10)
if (temperature) obj.temperature = parseFloat(temperature)
if (maxTokens) obj.maxTokens = parseInt(maxTokens, 10)
if (topP) obj.topP = parseInt(topP, 10)
if (hfTopK) obj.topK = parseInt(hfTopK, 10)
@@ -132,7 +132,7 @@ class OpenAI_LLMs implements INode {
const basePath = nodeData.inputs?.basepath as string
const obj: Partial<OpenAIInput> & { openAIApiKey?: string } = {
temperature: parseInt(temperature, 10),
temperature: parseFloat(temperature),
modelName,
openAIApiKey,
streaming: streaming ?? true