mirror of
https://github.com/farcasclaudiu/Flowise.git
synced 2026-06-29 07:01:04 +03:00
add additional config
This commit is contained in:
@@ -23,7 +23,7 @@ class Qdrant_Existing_VectorStores implements INode {
|
|||||||
constructor() {
|
constructor() {
|
||||||
this.label = 'Qdrant Load Existing Index'
|
this.label = 'Qdrant Load Existing Index'
|
||||||
this.name = 'qdrantExistingIndex'
|
this.name = 'qdrantExistingIndex'
|
||||||
this.version = 1.0
|
this.version = 2.0
|
||||||
this.type = 'Qdrant'
|
this.type = 'Qdrant'
|
||||||
this.icon = 'qdrant.png'
|
this.icon = 'qdrant.png'
|
||||||
this.category = 'Vector Stores'
|
this.category = 'Vector Stores'
|
||||||
@@ -55,8 +55,39 @@ class Qdrant_Existing_VectorStores implements INode {
|
|||||||
type: 'string'
|
type: 'string'
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
label: 'Qdrant Collection Cofiguration',
|
label: 'Vector Dimension',
|
||||||
|
name: 'qdrantVectorDimension',
|
||||||
|
type: 'number',
|
||||||
|
default: 1536,
|
||||||
|
additionalParams: true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
label: 'Similarity',
|
||||||
|
name: 'qdrantSimilarity',
|
||||||
|
description: 'Similarity measure used in Qdrant.',
|
||||||
|
type: 'options',
|
||||||
|
default: 'Cosine',
|
||||||
|
options: [
|
||||||
|
{
|
||||||
|
label: 'Cosine',
|
||||||
|
name: 'Cosine'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
label: 'Euclid',
|
||||||
|
name: 'Euclid'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
label: 'Dot',
|
||||||
|
name: 'Dot'
|
||||||
|
}
|
||||||
|
],
|
||||||
|
additionalParams: true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
label: 'Additional Collection Cofiguration',
|
||||||
name: 'qdrantCollectionConfiguration',
|
name: 'qdrantCollectionConfiguration',
|
||||||
|
description:
|
||||||
|
'Refer to <a target="_blank" href="https://qdrant.tech/documentation/concepts/collections">collection docs</a> for more reference',
|
||||||
type: 'json',
|
type: 'json',
|
||||||
optional: true,
|
optional: true,
|
||||||
additionalParams: true
|
additionalParams: true
|
||||||
@@ -98,6 +129,8 @@ class Qdrant_Existing_VectorStores implements INode {
|
|||||||
const collectionName = nodeData.inputs?.qdrantCollection as string
|
const collectionName = nodeData.inputs?.qdrantCollection as string
|
||||||
let qdrantCollectionConfiguration = nodeData.inputs?.qdrantCollectionConfiguration
|
let qdrantCollectionConfiguration = nodeData.inputs?.qdrantCollectionConfiguration
|
||||||
const embeddings = nodeData.inputs?.embeddings as Embeddings
|
const embeddings = nodeData.inputs?.embeddings as Embeddings
|
||||||
|
const qdrantSimilarity = nodeData.inputs?.qdrantSimilarity
|
||||||
|
const qdrantVectorDimension = nodeData.inputs?.qdrantVectorDimension
|
||||||
const output = nodeData.outputs?.output as string
|
const output = nodeData.outputs?.output as string
|
||||||
const topK = nodeData.inputs?.topK as string
|
const topK = nodeData.inputs?.topK as string
|
||||||
let queryFilter = nodeData.inputs?.queryFilter
|
let queryFilter = nodeData.inputs?.queryFilter
|
||||||
@@ -126,7 +159,14 @@ class Qdrant_Existing_VectorStores implements INode {
|
|||||||
typeof qdrantCollectionConfiguration === 'object'
|
typeof qdrantCollectionConfiguration === 'object'
|
||||||
? qdrantCollectionConfiguration
|
? qdrantCollectionConfiguration
|
||||||
: JSON.parse(qdrantCollectionConfiguration)
|
: JSON.parse(qdrantCollectionConfiguration)
|
||||||
dbConfig.collectionConfig = qdrantCollectionConfiguration
|
dbConfig.collectionConfig = {
|
||||||
|
...qdrantCollectionConfiguration,
|
||||||
|
vectors: {
|
||||||
|
...qdrantCollectionConfiguration.vectors,
|
||||||
|
size: qdrantVectorDimension ? parseInt(qdrantVectorDimension, 10) : 1536,
|
||||||
|
distance: qdrantSimilarity ?? 'Cosine'
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (queryFilter) {
|
if (queryFilter) {
|
||||||
|
|||||||
@@ -25,7 +25,7 @@ class QdrantUpsert_VectorStores implements INode {
|
|||||||
constructor() {
|
constructor() {
|
||||||
this.label = 'Qdrant Upsert Document'
|
this.label = 'Qdrant Upsert Document'
|
||||||
this.name = 'qdrantUpsert'
|
this.name = 'qdrantUpsert'
|
||||||
this.version = 1.0
|
this.version = 2.0
|
||||||
this.type = 'Qdrant'
|
this.type = 'Qdrant'
|
||||||
this.icon = 'qdrant.png'
|
this.icon = 'qdrant.png'
|
||||||
this.category = 'Vector Stores'
|
this.category = 'Vector Stores'
|
||||||
@@ -62,6 +62,35 @@ class QdrantUpsert_VectorStores implements INode {
|
|||||||
name: 'qdrantCollection',
|
name: 'qdrantCollection',
|
||||||
type: 'string'
|
type: 'string'
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
label: 'Vector Dimension',
|
||||||
|
name: 'qdrantVectorDimension',
|
||||||
|
type: 'number',
|
||||||
|
default: 1536,
|
||||||
|
additionalParams: true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
label: 'Similarity',
|
||||||
|
name: 'qdrantSimilarity',
|
||||||
|
description: 'Similarity measure used in Qdrant.',
|
||||||
|
type: 'options',
|
||||||
|
default: 'Cosine',
|
||||||
|
options: [
|
||||||
|
{
|
||||||
|
label: 'Cosine',
|
||||||
|
name: 'Cosine'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
label: 'Euclid',
|
||||||
|
name: 'Euclid'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
label: 'Dot',
|
||||||
|
name: 'Dot'
|
||||||
|
}
|
||||||
|
],
|
||||||
|
additionalParams: true
|
||||||
|
},
|
||||||
{
|
{
|
||||||
label: 'Top K',
|
label: 'Top K',
|
||||||
name: 'topK',
|
name: 'topK',
|
||||||
@@ -99,6 +128,9 @@ class QdrantUpsert_VectorStores implements INode {
|
|||||||
const collectionName = nodeData.inputs?.qdrantCollection as string
|
const collectionName = nodeData.inputs?.qdrantCollection as string
|
||||||
const docs = nodeData.inputs?.document as Document[]
|
const docs = nodeData.inputs?.document as Document[]
|
||||||
const embeddings = nodeData.inputs?.embeddings as Embeddings
|
const embeddings = nodeData.inputs?.embeddings as Embeddings
|
||||||
|
const qdrantSimilarity = nodeData.inputs?.qdrantSimilarity
|
||||||
|
const qdrantVectorDimension = nodeData.inputs?.qdrantVectorDimension
|
||||||
|
|
||||||
const output = nodeData.outputs?.output as string
|
const output = nodeData.outputs?.output as string
|
||||||
const topK = nodeData.inputs?.topK as string
|
const topK = nodeData.inputs?.topK as string
|
||||||
const k = topK ? parseFloat(topK) : 4
|
const k = topK ? parseFloat(topK) : 4
|
||||||
@@ -121,7 +153,13 @@ class QdrantUpsert_VectorStores implements INode {
|
|||||||
const dbConfig: QdrantLibArgs = {
|
const dbConfig: QdrantLibArgs = {
|
||||||
client,
|
client,
|
||||||
url: qdrantServerUrl,
|
url: qdrantServerUrl,
|
||||||
collectionName
|
collectionName,
|
||||||
|
collectionConfig: {
|
||||||
|
vectors: {
|
||||||
|
size: qdrantVectorDimension ? parseInt(qdrantVectorDimension, 10) : 1536,
|
||||||
|
distance: qdrantSimilarity ?? 'Cosine'
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const retrieverConfig: RetrieverConfig = {
|
const retrieverConfig: RetrieverConfig = {
|
||||||
|
|||||||
Reference in New Issue
Block a user