From ea2832e2bad4b0c8f563507084f8d95435cdd67c Mon Sep 17 00:00:00 2001 From: Stable Genius <259448942+stablegenius49@users.noreply.github.com> Date: Mon, 2 Mar 2026 09:01:25 -0800 Subject: [PATCH 1/2] feat(embeddings): add LM Studio embeddings node --- .../credentials/LMStudioApi.credential.ts | 23 ++++ .../LMStudioEmbedding/LMStudioEmbedding.ts | 114 ++++++++++++++++++ .../embeddings/LMStudioEmbedding/lmstudio.svg | 15 +++ 3 files changed, 152 insertions(+) create mode 100644 packages/components/credentials/LMStudioApi.credential.ts create mode 100644 packages/components/nodes/embeddings/LMStudioEmbedding/LMStudioEmbedding.ts create mode 100644 packages/components/nodes/embeddings/LMStudioEmbedding/lmstudio.svg diff --git a/packages/components/credentials/LMStudioApi.credential.ts b/packages/components/credentials/LMStudioApi.credential.ts new file mode 100644 index 00000000000..fa1c074e038 --- /dev/null +++ b/packages/components/credentials/LMStudioApi.credential.ts @@ -0,0 +1,23 @@ +import { INodeCredential, INodeParams } from '../src/Interface' + +class LMStudioApi implements INodeCredential { + label: string + name: string + version: number + inputs: INodeParams[] + + constructor() { + this.label = 'LM Studio API' + this.name = 'lmStudioApi' + this.version = 1.0 + this.inputs = [ + { + label: 'LM Studio API Key', + name: 'lmStudioApiKey', + type: 'password' + } + ] + } +} + +module.exports = { credClass: LMStudioApi } diff --git a/packages/components/nodes/embeddings/LMStudioEmbedding/LMStudioEmbedding.ts b/packages/components/nodes/embeddings/LMStudioEmbedding/LMStudioEmbedding.ts new file mode 100644 index 00000000000..0081ad9acf5 --- /dev/null +++ b/packages/components/nodes/embeddings/LMStudioEmbedding/LMStudioEmbedding.ts @@ -0,0 +1,114 @@ +import { ClientOptions, OpenAIEmbeddings, OpenAIEmbeddingsParams } from '@langchain/openai' +import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' +import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils' + +class LMStudioEmbedding_Embeddings 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 = 'LM Studio Embeddings' + this.name = 'lmStudioEmbeddings' + this.version = 1.0 + this.type = 'LMStudioEmbeddings' + this.icon = 'lmstudio.svg' + this.category = 'Embeddings' + this.description = "Generate embeddings using LM Studio's OpenAI-compatible API" + this.baseClasses = [this.type, ...getBaseClasses(OpenAIEmbeddings)] + this.credential = { + label: 'Connect Credential', + name: 'credential', + type: 'credential', + credentialNames: ['lmStudioApi'], + optional: true + } + this.inputs = [ + { + label: 'Base URL', + name: 'baseUrl', + type: 'string', + default: 'http://localhost:1234/v1' + }, + { + label: 'Model Name', + name: 'modelName', + type: 'string', + default: 'text-embedding-nomic-embed-text-v1.5' + }, + { + label: 'Strip New Lines', + name: 'stripNewLines', + type: 'boolean', + optional: true, + additionalParams: true + }, + { + label: 'Batch Size', + name: 'batchSize', + type: 'number', + optional: true, + additionalParams: true + }, + { + label: 'Timeout', + name: 'timeout', + type: 'number', + optional: true, + additionalParams: true + }, + { + label: 'Dimensions', + name: 'dimensions', + type: 'number', + optional: true, + additionalParams: true + } + ] + } + + async init(nodeData: INodeData, _: string, options: ICommonObject): Promise { + const baseUrl = nodeData.inputs?.baseUrl as string + const modelName = nodeData.inputs?.modelName as string + const stripNewLines = nodeData.inputs?.stripNewLines as boolean + const batchSize = nodeData.inputs?.batchSize as string + const timeout = nodeData.inputs?.timeout as string + const dimensions = nodeData.inputs?.dimensions as string + + if (nodeData.inputs?.credentialId) { + nodeData.credential = nodeData.inputs?.credentialId + } + + const credentialData = await getCredentialData(nodeData.credential ?? '', options) + const lmStudioApiKey = getCredentialParam('lmStudioApiKey', credentialData, nodeData) + + const obj: Partial & { openAIApiKey?: string; configuration?: ClientOptions } = { + modelName, + openAIApiKey: 'sk-' + } + + if (lmStudioApiKey) obj.openAIApiKey = lmStudioApiKey + if (stripNewLines !== undefined) obj.stripNewLines = stripNewLines + if (batchSize) obj.batchSize = parseInt(batchSize, 10) + if (timeout) obj.timeout = parseInt(timeout, 10) + if (dimensions) obj.dimensions = parseInt(dimensions, 10) + + if (baseUrl) { + obj.configuration = { + baseURL: baseUrl + } + } + + const model = new OpenAIEmbeddings(obj) + return model + } +} + +module.exports = { nodeClass: LMStudioEmbedding_Embeddings } diff --git a/packages/components/nodes/embeddings/LMStudioEmbedding/lmstudio.svg b/packages/components/nodes/embeddings/LMStudioEmbedding/lmstudio.svg new file mode 100644 index 00000000000..5364dd30bd4 --- /dev/null +++ b/packages/components/nodes/embeddings/LMStudioEmbedding/lmstudio.svg @@ -0,0 +1,15 @@ + + + + + + + + + + + + + + + From b4e6b1dbef222118740527926f91c8853443bf77 Mon Sep 17 00:00:00 2001 From: Stable Genius <259448942+stablegenius49@users.noreply.github.com> Date: Tue, 3 Mar 2026 00:43:20 -0800 Subject: [PATCH 2/2] fix(embeddings): safely parse optional LM Studio numeric params --- .../LMStudioEmbedding/LMStudioEmbedding.ts | 24 ++++++++++++++----- 1 file changed, 18 insertions(+), 6 deletions(-) diff --git a/packages/components/nodes/embeddings/LMStudioEmbedding/LMStudioEmbedding.ts b/packages/components/nodes/embeddings/LMStudioEmbedding/LMStudioEmbedding.ts index 0081ad9acf5..2a8b2f01b1d 100644 --- a/packages/components/nodes/embeddings/LMStudioEmbedding/LMStudioEmbedding.ts +++ b/packages/components/nodes/embeddings/LMStudioEmbedding/LMStudioEmbedding.ts @@ -2,6 +2,13 @@ import { ClientOptions, OpenAIEmbeddings, OpenAIEmbeddingsParams } from '@langch import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface' import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils' +const parseOptionalInt = (value?: string | number): number | undefined => { + if (value === undefined || value === null || value === '') return undefined + + const parsedValue = Number.parseInt(String(value), 10) + return Number.isNaN(parsedValue) ? undefined : parsedValue +} + class LMStudioEmbedding_Embeddings implements INode { label: string name: string @@ -78,9 +85,9 @@ class LMStudioEmbedding_Embeddings implements INode { const baseUrl = nodeData.inputs?.baseUrl as string const modelName = nodeData.inputs?.modelName as string const stripNewLines = nodeData.inputs?.stripNewLines as boolean - const batchSize = nodeData.inputs?.batchSize as string - const timeout = nodeData.inputs?.timeout as string - const dimensions = nodeData.inputs?.dimensions as string + const batchSize = nodeData.inputs?.batchSize as string | number + const timeout = nodeData.inputs?.timeout as string | number + const dimensions = nodeData.inputs?.dimensions as string | number if (nodeData.inputs?.credentialId) { nodeData.credential = nodeData.inputs?.credentialId @@ -96,9 +103,14 @@ class LMStudioEmbedding_Embeddings implements INode { if (lmStudioApiKey) obj.openAIApiKey = lmStudioApiKey if (stripNewLines !== undefined) obj.stripNewLines = stripNewLines - if (batchSize) obj.batchSize = parseInt(batchSize, 10) - if (timeout) obj.timeout = parseInt(timeout, 10) - if (dimensions) obj.dimensions = parseInt(dimensions, 10) + + const parsedBatchSize = parseOptionalInt(batchSize) + const parsedTimeout = parseOptionalInt(timeout) + const parsedDimensions = parseOptionalInt(dimensions) + + if (parsedBatchSize !== undefined) obj.batchSize = parsedBatchSize + if (parsedTimeout !== undefined) obj.timeout = parsedTimeout + if (parsedDimensions !== undefined) obj.dimensions = parsedDimensions if (baseUrl) { obj.configuration = {