Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add bathSize into AzureOpenAIEmbedding

This commit is contained in:
chungyau97 2023-05-29 22:00:01 +07:00
parent 5fa3ad4033
commit d5da8626b2
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
packages/components/nodes/embeddings/AzureOpenAIEmbedding/AzureOpenAIEmbedding.ts
View File

@ -55,6 +55,14 @@ class AzureOpenAIEmbedding_Embeddings implements INode {
],
default: '2023-03-15-preview'
},
{
label: 'Batch Size',
name: 'batchSize',
type: 'number',
default: '1',
optional: true,
additionalParams: true
},
{
label: 'Timeout',
name: 'timeout',
@ -70,6 +78,7 @@ class AzureOpenAIEmbedding_Embeddings implements INode {
const azureOpenAIApiInstanceName = nodeData.inputs?.azureOpenAIApiInstanceName as string
const azureOpenAIApiDeploymentName = nodeData.inputs?.azureOpenAIApiDeploymentName as string
const azureOpenAIApiVersion = nodeData.inputs?.azureOpenAIApiVersion as string
const batchSize = nodeData.inputs?.batchSize as string
const timeout = nodeData.inputs?.timeout as string
const obj: Partial<OpenAIEmbeddingsParams> & Partial<AzureOpenAIInput> = {
@ -79,6 +88,7 @@ class AzureOpenAIEmbedding_Embeddings implements INode {
azureOpenAIApiVersion
}
if (batchSize) obj.batchSize = parseInt(batchSize, 10)
if (timeout) obj.timeout = parseInt(timeout, 10)
const model = new OpenAIEmbeddings(obj)