| import { INode, INodeData, INodeParams } from '../../../src/Interface' | |
| import { getBaseClasses } from '../../../src/utils' | |
| import { AnthropicInput, ChatAnthropic } from 'langchain/chat_models/anthropic' | |
| class ChatAnthropic_ChatModels implements INode { | |
| label: string | |
| name: string | |
| type: string | |
| icon: string | |
| category: string | |
| description: string | |
| baseClasses: string[] | |
| inputs: INodeParams[] | |
| constructor() { | |
| this.label = 'ChatAnthropic' | |
| this.name = 'chatAnthropic' | |
| this.type = 'ChatAnthropic' | |
| this.icon = 'chatAnthropic.png' | |
| this.category = 'Chat Models' | |
| this.description = 'Wrapper around ChatAnthropic large language models that use the Chat endpoint' | |
| this.baseClasses = [this.type, ...getBaseClasses(ChatAnthropic)] | |
| this.inputs = [ | |
| { | |
| label: 'ChatAnthropic Api Key', | |
| name: 'anthropicApiKey', | |
| type: 'password' | |
| }, | |
| { | |
| label: 'Model Name', | |
| name: 'modelName', | |
| type: 'options', | |
| options: [ | |
| { | |
| label: 'claude-v1', | |
| name: 'claude-v1' | |
| }, | |
| { | |
| label: 'claude-v1-100k', | |
| name: 'claude-v1-100k' | |
| }, | |
| { | |
| label: 'claude-v1.0', | |
| name: 'claude-v1.0' | |
| }, | |
| { | |
| label: 'claude-v1.2', | |
| name: 'claude-v1.2' | |
| }, | |
| { | |
| label: 'claude-v1.3', | |
| name: 'claude-v1.3' | |
| }, | |
| { | |
| label: 'claude-v1.3-100k', | |
| name: 'claude-v1.3-100k' | |
| }, | |
| { | |
| label: 'claude-instant-v1', | |
| name: 'claude-instant-v1' | |
| }, | |
| { | |
| label: 'claude-instant-v1-100k', | |
| name: 'claude-instant-v1-100k' | |
| }, | |
| { | |
| label: 'claude-instant-v1.0', | |
| name: 'claude-instant-v1.0' | |
| }, | |
| { | |
| label: 'claude-instant-v1.1', | |
| name: 'claude-instant-v1.1' | |
| }, | |
| { | |
| label: 'claude-instant-v1.1-100k', | |
| name: 'claude-instant-v1.1-100k' | |
| } | |
| ], | |
| default: 'claude-v1', | |
| optional: true | |
| }, | |
| { | |
| label: 'Temperature', | |
| name: 'temperature', | |
| type: 'number', | |
| default: 0.9, | |
| optional: true | |
| }, | |
| { | |
| label: 'Max Tokens', | |
| name: 'maxTokensToSample', | |
| type: 'number', | |
| optional: true, | |
| additionalParams: true | |
| }, | |
| { | |
| label: 'Top P', | |
| name: 'topP', | |
| type: 'number', | |
| optional: true, | |
| additionalParams: true | |
| }, | |
| { | |
| label: 'Top K', | |
| name: 'topK', | |
| type: 'number', | |
| optional: true, | |
| additionalParams: true | |
| } | |
| ] | |
| } | |
| async init(nodeData: INodeData): Promise<any> { | |
| const temperature = nodeData.inputs?.temperature as string | |
| const modelName = nodeData.inputs?.modelName as string | |
| const anthropicApiKey = nodeData.inputs?.anthropicApiKey as string | |
| const maxTokensToSample = nodeData.inputs?.maxTokensToSample as string | |
| const topP = nodeData.inputs?.topP as string | |
| const topK = nodeData.inputs?.topK as string | |
| const streaming = nodeData.inputs?.streaming as boolean | |
| const obj: Partial<AnthropicInput> & { anthropicApiKey?: string } = { | |
| temperature: parseInt(temperature, 10), | |
| modelName, | |
| anthropicApiKey, | |
| streaming: streaming ?? true | |
| } | |
| if (maxTokensToSample) obj.maxTokensToSample = parseInt(maxTokensToSample, 10) | |
| if (topP) obj.topP = parseInt(topP, 10) | |
| if (topK) obj.topK = parseInt(topK, 10) | |
| const model = new ChatAnthropic(obj) | |
| return model | |
| } | |
| } | |
| module.exports = { nodeClass: ChatAnthropic_ChatModels } | |