Feat/deepseek chat node (#3732)
* Feat: Add deepseek models to components configuration * Feat: Implement Deepseek API integration with chat models and add SVG icon * Refactor: Remove image input options and add missing baseOptions in deepseek chat node
This commit is contained in:
Nguyễn Đức Hùng
GitHub
parent
0381a99c4d
commit
93f3a5d98a
|
|
@ -0,0 +1,23 @@
|
|||
import { INodeCredential, INodeParams } from '../src/Interface'
|
||||
|
||||
class DeepseekApi implements INodeCredential {
|
||||
label: string
|
||||
name: string
|
||||
version: number
|
||||
inputs: INodeParams[]
|
||||
|
||||
constructor() {
|
||||
this.label = 'DeepseekAI API'
|
||||
this.name = 'deepseekApi'
|
||||
this.version = 1.0
|
||||
this.inputs = [
|
||||
{
|
||||
label: 'DeepseekAI API Key',
|
||||
name: 'deepseekApiKey',
|
||||
type: 'password'
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = { credClass: DeepseekApi }
|
||||
|
|
@ -572,6 +572,19 @@
|
|||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "deepseek",
|
||||
"models": [
|
||||
{
|
||||
"label": "deepseek-chat",
|
||||
"name": "deepseek-chat"
|
||||
},
|
||||
{
|
||||
"label": "deepseek-coder",
|
||||
"name": "deepseek-coder"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"name": "chatOpenAI",
|
||||
"models": [
|
||||
|
|
|
|||
|
|
@ -0,0 +1,197 @@
|
|||
import { BaseCache } from '@langchain/core/caches'
|
||||
import { BaseChatModelParams } from '@langchain/core/language_models/chat_models'
|
||||
import { ChatOpenAI, LegacyOpenAIInput, OpenAIChatInput } from '@langchain/openai'
|
||||
import type { ClientOptions } from 'openai'
|
||||
import { ICommonObject, INode, INodeData, INodeOptionsValue, INodeParams } from '../../../src/Interface'
|
||||
import { getModels, MODEL_TYPE } from '../../../src/modelLoader'
|
||||
import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils'
|
||||
|
||||
class Deepseek_ChatModels implements INode {
|
||||
readonly baseURL: string = 'https://api.deepseek.com'
|
||||
label: string
|
||||
name: string
|
||||
version: number
|
||||
type: string
|
||||
icon: string
|
||||
category: string
|
||||
description: string
|
||||
baseClasses: string[]
|
||||
credential: INodeParams
|
||||
inputs: INodeParams[]
|
||||
|
||||
constructor() {
|
||||
this.label = 'ChatDeepseek'
|
||||
this.name = 'chatDeepseek'
|
||||
this.version = 1.0
|
||||
this.type = 'chatDeepseek'
|
||||
this.icon = 'deepseek.svg'
|
||||
this.category = 'Chat Models'
|
||||
this.description = 'Wrapper around Deepseek large language models that use the Chat endpoint'
|
||||
this.baseClasses = [this.type, ...getBaseClasses(ChatOpenAI)]
|
||||
this.credential = {
|
||||
label: 'Connect Credential',
|
||||
name: 'credential',
|
||||
type: 'credential',
|
||||
credentialNames: ['deepseekApi']
|
||||
}
|
||||
this.inputs = [
|
||||
{
|
||||
label: 'Cache',
|
||||
name: 'cache',
|
||||
type: 'BaseCache',
|
||||
optional: true
|
||||
},
|
||||
{
|
||||
label: 'Model Name',
|
||||
name: 'modelName',
|
||||
type: 'asyncOptions',
|
||||
loadMethod: 'listModels',
|
||||
default: 'deepseek-chat'
|
||||
},
|
||||
{
|
||||
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: 'Timeout',
|
||||
name: 'timeout',
|
||||
type: 'number',
|
||||
step: 1,
|
||||
optional: true,
|
||||
additionalParams: true
|
||||
},
|
||||
{
|
||||
label: 'Stop Sequence',
|
||||
name: 'stopSequence',
|
||||
type: 'string',
|
||||
rows: 4,
|
||||
optional: true,
|
||||
description: 'List of stop words to use when generating. Use comma to separate multiple stop words.',
|
||||
additionalParams: true
|
||||
},
|
||||
{
|
||||
label: 'Base Options',
|
||||
name: 'baseOptions',
|
||||
type: 'json',
|
||||
optional: true,
|
||||
additionalParams: true,
|
||||
description: 'Additional options to pass to the Deepseek client. This should be a JSON object.'
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
//@ts-ignore
|
||||
loadMethods = {
|
||||
async listModels(): Promise<INodeOptionsValue[]> {
|
||||
return await getModels(MODEL_TYPE.CHAT, 'deepseek')
|
||||
}
|
||||
}
|
||||
|
||||
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
|
||||
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 timeout = nodeData.inputs?.timeout as string
|
||||
const stopSequence = nodeData.inputs?.stopSequence 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('deepseekApiKey', credentialData, nodeData)
|
||||
|
||||
const cache = nodeData.inputs?.cache as BaseCache
|
||||
|
||||
const obj: Partial<OpenAIChatInput> & BaseChatModelParams & { configuration?: ClientOptions & LegacyOpenAIInput } = {
|
||||
temperature: parseFloat(temperature),
|
||||
modelName,
|
||||
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 (timeout) obj.timeout = parseInt(timeout, 10)
|
||||
if (cache) obj.cache = cache
|
||||
if (stopSequence) {
|
||||
const stopSequenceArray = stopSequence.split(',').map((item) => item.trim())
|
||||
obj.stop = stopSequenceArray
|
||||
}
|
||||
|
||||
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 ChatDeepseek 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: Deepseek_ChatModels }
|
||||
|
|
@ -0,0 +1,30 @@
|
|||
<?xml version="1.0" standalone="no"?>
|
||||
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 20010904//EN"
|
||||
"http://www.w3.org/TR/2001/REC-SVG-20010904/DTD/svg10.dtd">
|
||||
<svg version="1.0" xmlns="http://www.w3.org/2000/svg"
|
||||
width="225.000000pt" height="225.000000pt" viewBox="0 0 225.000000 225.000000"
|
||||
preserveAspectRatio="xMidYMid meet">
|
||||
|
||||
<g transform="translate(0.000000,225.000000) scale(0.100000,-0.100000)"
|
||||
fill="#000000" stroke="none">
|
||||
<path d="M1527 1893 c-18 -39 -22 -66 -22 -143 0 -125 30 -194 118 -276 l61
|
||||
-56 -13 -47 c-7 -25 -17 -50 -22 -56 -21 -22 -123 49 -264 185 -78 74 -164
|
||||
151 -192 170 -75 50 -91 96 -53 150 11 16 29 32 40 35 24 8 26 30 4 39 -43 16
|
||||
-140 5 -239 -28 -101 -34 -104 -35 -231 -29 -147 7 -261 -11 -351 -53 -233
|
||||
-110 -377 -369 -360 -644 23 -365 263 -673 617 -792 92 -31 100 -32 265 -32
|
||||
133 0 184 3 235 18 83 23 174 66 232 110 l45 35 49 -16 c69 -21 247 -22 297 0
|
||||
31 13 37 21 37 46 0 26 -6 32 -54 54 -30 14 -71 31 -93 38 -68 23 -68 26 -5
|
||||
98 32 36 73 87 91 113 86 127 155 330 168 489 l6 77 56 11 c107 21 209 99 256
|
||||
196 33 68 54 188 36 209 -19 23 -34 20 -67 -15 -41 -44 -99 -68 -164 -69 -57
|
||||
0 -125 -25 -149 -53 -19 -23 -28 -21 -35 7 -10 39 -55 85 -110 113 -70 35 -93
|
||||
58 -111 114 -22 66 -48 67 -78 2z m-1139 -549 c180 -37 351 -133 478 -266 38
|
||||
-40 114 -138 169 -217 103 -148 211 -273 284 -326 22 -16 41 -34 41 -39 0 -13
|
||||
-148 1 -199 19 -24 8 -83 44 -130 79 -139 104 -217 149 -259 150 -33 1 -37 -2
|
||||
-40 -26 -2 -14 8 -49 22 -77 30 -58 26 -77 -19 -90 -88 -24 -196 30 -336 169
|
||||
-84 84 -101 107 -147 200 -59 121 -89 222 -98 330 -5 58 -3 78 7 87 31 25 124
|
||||
28 227 7z m920 -73 c65 -33 178 -175 184 -231 6 -48 -88 -67 -148 -29 -45 27
|
||||
-54 47 -54 113 0 63 -22 96 -64 96 -36 0 -56 10 -56 29 0 44 73 55 138 22z
|
||||
m-78 -106 c10 -12 10 -18 0 -30 -25 -30 -61 -7 -46 30 3 8 12 15 19 15 8 0 20
|
||||
-7 27 -15z"/>
|
||||
</g>
|
||||
</svg>
|
||||
|
After Width: | Height: | Size: 1.8 KiB |
Loading…
Reference in New Issue