farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-25 23:01:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Feature: Add BasePath parameter setting support to ChatOpenAI LlamaIndex (#2132)

Browse Source 
* ChatOpenAI LlamaIndex support BasePath Param

* Add BasePath parameter setting support to ChatOpenAI LlamaIndex

* Add BasePath parameter setting support to ChatOpenAI LlamaIndex
This commit is contained in:
louyongjiu
2024-04-09 21:35:23 +08:00
committed by GitHub
parent 19bb23440a
commit 827de07e94
2 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/chatmodels/ChatOpenAI/ChatOpenAI_LlamaIndex.ts
+14
View File
@@ -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)
packages/components/nodes/embeddings/OpenAIEmbedding/OpenAIEmbedding_LlamaIndex.ts
+6 -1
View File
@@ -73,6 +73,7 @@ class OpenAIEmbedding_LlamaIndex_Embeddings implements INode {
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
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
}