diff --git a/packages/components/credentials/CometApi.credential.ts b/packages/components/credentials/CometApi.credential.ts new file mode 100644 index 000000000..58ec60610 --- /dev/null +++ b/packages/components/credentials/CometApi.credential.ts @@ -0,0 +1,23 @@ +import { INodeCredential, INodeParams } from '../src/Interface' + +class CometApi implements INodeCredential { + label: string + name: string + version: number + inputs: INodeParams[] + + constructor() { + this.label = 'Comet API' + this.name = 'cometApi' + this.version = 1.0 + this.inputs = [ + { + label: 'Comet API Key', + name: 'cometApiKey', + type: 'password' + } + ] + } +} + +module.exports = { credClass: CometApi } diff --git a/packages/components/nodes/chatmodels/ChatCometAPI/ChatCometAPI.ts b/packages/components/nodes/chatmodels/ChatCometAPI/ChatCometAPI.ts new file mode 100644 index 000000000..295c5e7ce --- /dev/null +++ b/packages/components/nodes/chatmodels/ChatCometAPI/ChatCometAPI.ts @@ -0,0 +1,176 @@ +import { BaseCache } from '@langchain/core/caches' +import { ChatOpenAI, ChatOpenAIFields } from '@langchain/openai' +import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' +import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils' + +class ChatCometAPI_ChatModels implements INode { + readonly baseURL: string = 'https://api.cometapi.com/v1' + label: string + name: string + version: number + type: string + icon: string + category: string + description: string + baseClasses: string[] + credential: INodeParams + inputs: INodeParams[] + + constructor() { + this.label = 'ChatCometAPI' + this.name = 'chatCometAPI' + this.version = 1.0 + this.type = 'ChatCometAPI' + this.icon = 'cometapi.svg' + this.category = 'Chat Models' + this.description = 'Wrapper around CometAPI large language models that use the Chat endpoint' + this.baseClasses = [this.type, ...getBaseClasses(ChatOpenAI)] + this.credential = { + label: 'Connect Credential', + name: 'credential', + type: 'credential', + credentialNames: ['cometApi'] + } + this.inputs = [ + { + label: 'Cache', + name: 'cache', + type: 'BaseCache', + optional: true + }, + { + label: 'Model Name', + name: 'modelName', + type: 'string', + default: 'gpt-5-mini', + description: 'Enter the model name (e.g., gpt-5-mini, claude-sonnet-4-20250514, gemini-2.0-flash)' + }, + { + label: 'Temperature', + name: 'temperature', + type: 'number', + step: 0.1, + default: 0.7, + optional: true + }, + { + label: 'Streaming', + name: 'streaming', + type: 'boolean', + default: true, + optional: true, + additionalParams: true + }, + { + label: 'Max Tokens', + name: 'maxTokens', + type: 'number', + step: 1, + optional: true, + additionalParams: true + }, + { + label: 'Top Probability', + name: 'topP', + type: 'number', + step: 0.1, + optional: true, + additionalParams: true + }, + { + label: 'Frequency Penalty', + name: 'frequencyPenalty', + type: 'number', + step: 0.1, + optional: true, + additionalParams: true + }, + { + label: 'Presence Penalty', + name: 'presencePenalty', + type: 'number', + step: 0.1, + optional: true, + additionalParams: true + }, + { + label: 'Base Options', + name: 'baseOptions', + type: 'json', + optional: true, + additionalParams: true, + description: 'Additional options to pass to the CometAPI client. This should be a JSON object.' + } + ] + } + + async init(nodeData: INodeData, _: string, options: ICommonObject): Promise { + const temperature = nodeData.inputs?.temperature as string + const modelName = nodeData.inputs?.modelName as string + const maxTokens = nodeData.inputs?.maxTokens as string + const topP = nodeData.inputs?.topP as string + const frequencyPenalty = nodeData.inputs?.frequencyPenalty as string + const presencePenalty = nodeData.inputs?.presencePenalty as string + const streaming = nodeData.inputs?.streaming as boolean + const baseOptions = nodeData.inputs?.baseOptions + + if (nodeData.inputs?.credentialId) { + nodeData.credential = nodeData.inputs?.credentialId + } + const credentialData = await getCredentialData(nodeData.credential ?? '', options) + const openAIApiKey = getCredentialParam('cometApiKey', credentialData, nodeData) + + // Custom error handling for missing API key + if (!openAIApiKey || openAIApiKey.trim() === '') { + throw new Error( + 'CometAPI API Key is missing or empty. Please provide a valid CometAPI API key in the credential configuration.' + ) + } + + // Custom error handling for missing model name + if (!modelName || modelName.trim() === '') { + throw new Error('Model Name is required. Please enter a valid model name (e.g., gpt-5-mini, claude-sonnet-4-20250514).') + } + + const cache = nodeData.inputs?.cache as BaseCache + + const obj: ChatOpenAIFields = { + temperature: parseFloat(temperature), + modelName, + openAIApiKey, + apiKey: openAIApiKey, + streaming: streaming ?? true + } + + if (maxTokens) obj.maxTokens = parseInt(maxTokens, 10) + if (topP) obj.topP = parseFloat(topP) + if (frequencyPenalty) obj.frequencyPenalty = parseFloat(frequencyPenalty) + if (presencePenalty) obj.presencePenalty = parseFloat(presencePenalty) + if (cache) obj.cache = cache + + let parsedBaseOptions: any | undefined = undefined + + if (baseOptions) { + try { + parsedBaseOptions = typeof baseOptions === 'object' ? baseOptions : JSON.parse(baseOptions) + if (parsedBaseOptions.baseURL) { + console.warn("The 'baseURL' parameter is not allowed when using the ChatCometAPI node.") + parsedBaseOptions.baseURL = undefined + } + } catch (exception) { + throw new Error('Invalid JSON in the BaseOptions: ' + exception) + } + } + + const model = new ChatOpenAI({ + ...obj, + configuration: { + baseURL: this.baseURL, + ...parsedBaseOptions + } + }) + return model + } +} + +module.exports = { nodeClass: ChatCometAPI_ChatModels } diff --git a/packages/components/nodes/chatmodels/ChatCometAPI/cometapi.svg b/packages/components/nodes/chatmodels/ChatCometAPI/cometapi.svg new file mode 100644 index 000000000..9f1d803d4 --- /dev/null +++ b/packages/components/nodes/chatmodels/ChatCometAPI/cometapi.svg @@ -0,0 +1,7 @@ + + + + + + + diff --git a/packages/server/src/utils/index.ts b/packages/server/src/utils/index.ts index 8ac28593b..9451a49b6 100644 --- a/packages/server/src/utils/index.ts +++ b/packages/server/src/utils/index.ts @@ -1491,7 +1491,8 @@ export const isFlowValidForStream = (reactFlowNodes: IReactFlowNode[], endingNod 'chatTogetherAI_LlamaIndex', 'chatFireworks', 'ChatSambanova', - 'chatBaiduWenxin' + 'chatBaiduWenxin', + 'chatCometAPI' ], LLMs: ['azureOpenAI', 'openAI', 'ollama'] }