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

Merge pull request #215 from FlowiseAI/bugfix/AzureOpenAIEmbeddingError

Browse Source 
bugfix/AzureOpenAIEmbeddingError
This commit is contained in:
Henry Heng
2023-05-29 17:00:34 +01:00
committed by GitHub
parent 5fa3ad4033 d5da8626b2
commit 850921aa68
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/embeddings/AzureOpenAIEmbedding/AzureOpenAIEmbedding.ts
+10
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)