diff --git a/packages/components/nodes/llms/Azure OpenAI/Azure.svg b/packages/components/nodes/llms/Azure OpenAI/Azure.svg new file mode 100644 index 00000000..51eb6253 --- /dev/null +++ b/packages/components/nodes/llms/Azure OpenAI/Azure.svg @@ -0,0 +1,5 @@ + + + + + \ No newline at end of file diff --git a/packages/components/nodes/llms/Azure OpenAI/AzureOpenAI.ts b/packages/components/nodes/llms/Azure OpenAI/AzureOpenAI.ts new file mode 100644 index 00000000..3faa884d --- /dev/null +++ b/packages/components/nodes/llms/Azure OpenAI/AzureOpenAI.ts @@ -0,0 +1,98 @@ +import { INode, INodeData, INodeParams } from '../../../src/Interface' +import { getBaseClasses } from '../../../src/utils' +import { AzureOpenAIInput, OpenAI, OpenAIInput } from 'langchain/llms/openai' + +class AzureOpenAI_LLMs implements INode { + label: string + name: string + type: string + icon: string + category: string + description: string + baseClasses: string[] + inputs: INodeParams[] + + constructor() { + this.label = 'Azure OpenAI' + this.name = 'azureOpenAI' + this.type = 'AzureOpenAI' + this.icon = 'Azure.svg' + this.category = 'LLMs' + this.description = 'Wrapper around Azure OpenAI large language models' + this.baseClasses = [this.type, ...getBaseClasses(OpenAI)] + this.inputs = [ + { + label: 'Azure OpenAI Api Key', + name: 'azureOpenAIApiKey', + type: 'password' + }, + { + label: 'Temperature', + name: 'temperature', + type: 'number', + default: 0.9, + optional: true + }, + { + label: 'Azure OpenAI Api Instance Name', + name: 'azureOpenAIApiInstanceName', + type: 'string', + placeholder: 'YOUR-INSTANCE-NAME' + }, + { + label: 'Azure OpenAI Api Deployment Name', + name: 'azureOpenAIApiDeploymentName', + type: 'string', + placeholder: 'YOUR-DEPLOYMENT-NAME' + }, + { + label: 'Azure OpenAI Api Version', + name: 'azureOpenAIApiVersion', + type: 'string', + placeholder: 'YOUR-API-VERSION' + }, + { + label: 'Azure OpenAI Api Embeddings Deployment Name', + name: 'azureOpenAIApiEmbeddingsDeploymentName', + type: 'string', + placeholder: 'YOUR-EMBEDDINGS-NAME', + optional: true, + additionalParams: true + }, + { + label: 'Azure OpenAI Api Completions Deployment Name', + name: 'azureOpenAIApiCompletionsDeploymentName', + type: 'string', + placeholder: 'YOUR-COMPLETIONS-NAME', + optional: true, + additionalParams: true + } + ] + } + + async init(nodeData: INodeData): Promise { + const azureOpenAIApiKey = nodeData.inputs?.azureOpenAIApiKey as string + const temperature = nodeData.inputs?.temperature as string + const azureOpenAIApiInstanceName = nodeData.inputs?.azureOpenAIApiInstanceName as string + const azureOpenAIApiDeploymentName = nodeData.inputs?.azureOpenAIApiDeploymentName as string + const azureOpenAIApiVersion = nodeData.inputs?.azureOpenAIApiVersion as string + const azureOpenAIApiEmbeddingsDeploymentName = nodeData.inputs?.azureOpenAIApiEmbeddingsDeploymentName as string + const azureOpenAIApiCompletionsDeploymentName = nodeData.inputs?.azureOpenAIApiCompletionsDeploymentName as string + + const obj: Partial & Partial = { + temperature: parseInt(temperature, 10), + azureOpenAIApiKey, + azureOpenAIApiInstanceName, + azureOpenAIApiDeploymentName, + azureOpenAIApiVersion + } + + if (azureOpenAIApiEmbeddingsDeploymentName) obj.azureOpenAIApiEmbeddingsDeploymentName = azureOpenAIApiEmbeddingsDeploymentName + if (azureOpenAIApiCompletionsDeploymentName) obj.azureOpenAIApiCompletionsDeploymentName = azureOpenAIApiCompletionsDeploymentName + + const model = new OpenAI(obj) + return model + } +} + +module.exports = { nodeClass: AzureOpenAI_LLMs }