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

add modelName selection into Azure

Browse Source
This commit is contained in:
chungyau97
2023-05-12 08:43:00 +07:00
parent bb3e6e3bf8
commit b0e9387087
3 changed files with 85 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/llms/Azure OpenAI/AzureOpenAI.ts
+31
View File
@@ -26,6 +26,35 @@ class AzureOpenAI_LLMs implements INode {
name: 'azureOpenAIApiKey',
type: 'password'
},
{
label: 'Model Name',
name: 'modelName',
type: 'options',
options: [
{
label: 'text-davinci-003',
name: 'text-davinci-003'
},
{
label: 'text-davinci-002',
name: 'text-davinci-002'
},
{
label: 'text-curie-001',
name: 'text-curie-001'
},
{
label: 'text-babbage-001',
name: 'text-babbage-001'
},
{
label: 'text-ada-001',
name: 'text-ada-001'
}
],
default: 'text-davinci-003',
optional: true
},
{
label: 'Temperature',
name: 'temperature',
@@ -73,6 +102,7 @@ class AzureOpenAI_LLMs implements INode {
async init(nodeData: INodeData): Promise<any> {
const azureOpenAIApiKey = nodeData.inputs?.azureOpenAIApiKey as string
const temperature = nodeData.inputs?.temperature as string
const modelName = nodeData.inputs?.modelName as string
const azureOpenAIApiInstanceName = nodeData.inputs?.azureOpenAIApiInstanceName as string
const azureOpenAIApiDeploymentName = nodeData.inputs?.azureOpenAIApiDeploymentName as string
const azureOpenAIApiVersion = nodeData.inputs?.azureOpenAIApiVersion as string
@@ -81,6 +111,7 @@ class AzureOpenAI_LLMs implements INode {
const obj: Partial<AzureOpenAIInput> & Partial<OpenAIInput> = {
temperature: parseInt(temperature, 10),
modelName,
azureOpenAIApiKey,
azureOpenAIApiInstanceName,
azureOpenAIApiDeploymentName,