Merge pull request #1508 from mokeyish/patch-2

Add milvusTextField configuration for Milvus
This commit is contained in:
Henry Heng 2024-01-10 16:31:23 +00:00 committed by GitHub
commit 1d8d97c7a1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 1 deletions

View File

@ -65,6 +65,14 @@ class Milvus_VectorStores implements INode {
name: 'milvusCollection', name: 'milvusCollection',
type: 'string' type: 'string'
}, },
{
label: 'Milvus Text Field',
name: 'milvusTextField',
type: 'string',
placeholder: 'langchain_text',
optional: true,
additionalParams: true
},
{ {
label: 'Milvus Filter', label: 'Milvus Filter',
name: 'milvusFilter', name: 'milvusFilter',
@ -150,6 +158,7 @@ class Milvus_VectorStores implements INode {
const address = nodeData.inputs?.milvusServerUrl as string const address = nodeData.inputs?.milvusServerUrl as string
const collectionName = nodeData.inputs?.milvusCollection as string const collectionName = nodeData.inputs?.milvusCollection as string
const milvusFilter = nodeData.inputs?.milvusFilter as string const milvusFilter = nodeData.inputs?.milvusFilter as string
const textField = nodeData.inputs?.milvusTextField as string
// embeddings // embeddings
const embeddings = nodeData.inputs?.embeddings as Embeddings const embeddings = nodeData.inputs?.embeddings as Embeddings
@ -169,7 +178,8 @@ class Milvus_VectorStores implements INode {
// init MilvusLibArgs // init MilvusLibArgs
const milVusArgs: MilvusLibArgs = { const milVusArgs: MilvusLibArgs = {
url: address, url: address,
collectionName: collectionName collectionName: collectionName,
textField: textField
} }
if (milvusUser) milVusArgs.username = milvusUser if (milvusUser) milVusArgs.username = milvusUser