diff --git a/packages/components/nodes/chatmodels/ChatOpenAI/ChatOpenAI_LlamaIndex.ts b/packages/components/nodes/chatmodels/ChatOpenAI/ChatOpenAI_LlamaIndex.ts index 8b3567a6..4c0ac005 100644 --- a/packages/components/nodes/chatmodels/ChatOpenAI/ChatOpenAI_LlamaIndex.ts +++ b/packages/components/nodes/chatmodels/ChatOpenAI/ChatOpenAI_LlamaIndex.ts @@ -124,6 +124,13 @@ class ChatOpenAI_LlamaIndex_LLMs implements INode { step: 1, optional: true, additionalParams: true + }, + { + label: 'BasePath', + name: 'basepath', + type: 'string', + optional: true, + additionalParams: true } ] } @@ -134,6 +141,7 @@ class ChatOpenAI_LlamaIndex_LLMs implements INode { const maxTokens = nodeData.inputs?.maxTokens as string const topP = nodeData.inputs?.topP as string const timeout = nodeData.inputs?.timeout as string + const basePath = nodeData.inputs?.basepath as string const credentialData = await getCredentialData(nodeData.credential ?? '', options) const openAIApiKey = getCredentialParam('openAIApiKey', credentialData, nodeData) @@ -144,6 +152,12 @@ class ChatOpenAI_LlamaIndex_LLMs implements INode { apiKey: openAIApiKey } + if (basePath) { + obj.additionalSessionOptions = { + baseURL: basePath + } + } + if (maxTokens) obj.maxTokens = parseInt(maxTokens, 10) if (topP) obj.topP = parseFloat(topP) if (timeout) obj.timeout = parseInt(timeout, 10) diff --git a/packages/components/nodes/embeddings/OpenAIEmbedding/OpenAIEmbedding_LlamaIndex.ts b/packages/components/nodes/embeddings/OpenAIEmbedding/OpenAIEmbedding_LlamaIndex.ts index 960197fe..175e426d 100644 --- a/packages/components/nodes/embeddings/OpenAIEmbedding/OpenAIEmbedding_LlamaIndex.ts +++ b/packages/components/nodes/embeddings/OpenAIEmbedding/OpenAIEmbedding_LlamaIndex.ts @@ -73,6 +73,7 @@ class OpenAIEmbedding_LlamaIndex_Embeddings implements INode { async init(nodeData: INodeData, _: string, options: ICommonObject): Promise { const timeout = nodeData.inputs?.timeout as string const modelName = nodeData.inputs?.modelName as string + const basePath = nodeData.inputs?.basepath as string const credentialData = await getCredentialData(nodeData.credential ?? '', options) const openAIApiKey = getCredentialParam('openAIApiKey', credentialData, nodeData) @@ -82,7 +83,11 @@ class OpenAIEmbedding_LlamaIndex_Embeddings implements INode { model: modelName } if (timeout) obj.timeout = parseInt(timeout, 10) - + if (basePath) { + obj.additionalSessionOptions = { + baseURL: basePath + } + } const model = new OpenAIEmbedding(obj) return model }