diff --git a/packages/components/credentials/AlibabaApi.credential.ts b/packages/components/credentials/AlibabaApi.credential.ts new file mode 100644 index 00000000000..7ebfd148fc1 --- /dev/null +++ b/packages/components/credentials/AlibabaApi.credential.ts @@ -0,0 +1,23 @@ +import { INodeParams, INodeCredential } from '../src/Interface' + +class AlibabaApi implements INodeCredential { + label: string + name: string + version: number + inputs: INodeParams[] + + constructor() { + this.label = 'Alibaba API' + this.name = 'AlibabaApi' + this.version = 1.0 + this.inputs = [ + { + label: 'Alibaba Api Key', + name: 'alibabaApiKey', + type: 'password' + } + ] + } +} + +module.exports = { credClass: AlibabaApi } diff --git a/packages/components/models.json b/packages/components/models.json index 83bcf615e77..e6ac76618c9 100644 --- a/packages/components/models.json +++ b/packages/components/models.json @@ -411,6 +411,15 @@ } ] }, + { + "name": "chatAlibabaTongyi", + "models": [ + { + "label": "qwen-plus", + "name": "qwen-plus" + } + ] + }, { "name": "chatGoogleVertexAI", "models": [ diff --git a/packages/components/nodes/chatmodels/ChatAlibabaTongyi/ChatAlibabaTongyi.ts b/packages/components/nodes/chatmodels/ChatAlibabaTongyi/ChatAlibabaTongyi.ts new file mode 100644 index 00000000000..0425d76feee --- /dev/null +++ b/packages/components/nodes/chatmodels/ChatAlibabaTongyi/ChatAlibabaTongyi.ts @@ -0,0 +1,79 @@ +import { BaseCache } from '@langchain/core/caches' +import { ChatAlibabaTongyi } from '@langchain/community/chat_models/alibaba_tongyi' +import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' +import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils' +import { BaseChatModelParams } from '@langchain/core/language_models/chat_models' + +class ChatAlibabaTongyi_ChatModels implements INode { + label: string + name: string + version: number + type: string + icon: string + category: string + description: string + baseClasses: string[] + credential: INodeParams + inputs: INodeParams[] + + constructor() { + this.label = 'ChatAlibabaTongyi' + this.name = 'chatAlibabaTongyi' + this.version = 1.0 + this.type = 'ChatAlibabaTongyi' + this.icon = 'alibaba-svgrepo-com.svg' + this.category = 'Chat Models' + this.description = 'Wrapper around Alibaba Tongyi Chat Endpoints' + this.baseClasses = [this.type, ...getBaseClasses(ChatAlibabaTongyi)] + this.credential = { + label: 'Connect Credential', + name: 'credential', + type: 'credential', + credentialNames: ['AlibabaApi'] + } + this.inputs = [ + { + label: 'Cache', + name: 'cache', + type: 'BaseCache', + optional: true + }, + { + label: 'Model', + name: 'modelName', + type: 'string', + placeholder: 'qwen-plus' + }, + { + label: 'Temperature', + name: 'temperature', + type: 'number', + step: 0.1, + default: 0.9, + optional: true + } + ] + } + + async init(nodeData: INodeData, _: string, options: ICommonObject): Promise { + const cache = nodeData.inputs?.cache as BaseCache + const temperature = nodeData.inputs?.temperature as string + const modelName = nodeData.inputs?.modelName as string + + const credentialData = await getCredentialData(nodeData.credential ?? '', options) + const alibabaApiKey = getCredentialParam('alibabaApiKey', credentialData, nodeData) + + const obj: Partial & BaseChatModelParams = { + streaming: true, + alibabaApiKey, + model: modelName, + temperature: temperature ? parseFloat(temperature) : undefined + } + if (cache) obj.cache = cache + + const model = new ChatAlibabaTongyi(obj) + return model + } +} + +module.exports = { nodeClass: ChatAlibabaTongyi_ChatModels } diff --git a/packages/components/nodes/chatmodels/ChatAlibabaTongyi/alibaba-svgrepo-com.svg b/packages/components/nodes/chatmodels/ChatAlibabaTongyi/alibaba-svgrepo-com.svg new file mode 100644 index 00000000000..3ba6d875e60 --- /dev/null +++ b/packages/components/nodes/chatmodels/ChatAlibabaTongyi/alibaba-svgrepo-com.svg @@ -0,0 +1,8 @@ + + + + + + + + \ No newline at end of file diff --git a/packages/server/src/utils/index.ts b/packages/server/src/utils/index.ts index 172177e1224..f2fb5f2467b 100644 --- a/packages/server/src/utils/index.ts +++ b/packages/server/src/utils/index.ts @@ -1192,6 +1192,7 @@ export const isFlowValidForStream = (reactFlowNodes: IReactFlowNode[], endingNod 'awsChatBedrock', 'chatMistralAI', 'chatMistral_LlamaIndex', + 'chatAlibabaTongyi', 'groqChat', 'chatGroq_LlamaIndex', 'chatCohere',