Add vector search components and vector store search components

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-28 20:16:00 -03:00
commit a3924d657c
16 changed files with 33 additions and 23 deletions

View file

@ -0,0 +1,25 @@
from .AstraDBSearch import AstraDBSearchComponent
from .ChromaSearch import ChromaSearchComponent
from .FAISSSearch import FAISSSearchComponent
from .MongoDBAtlasVectorSearch import MongoDBAtlasSearchComponent
from .PineconeSearch import PineconeSearchComponent
from .QdrantSearch import QdrantSearchComponent
from .RedisSearch import RedisSearchComponent
from .SupabaseVectorStoreSearch import SupabaseSearchComponent
from .VectaraSearch import VectaraSearchComponent
from .WeaviateSearch import WeaviateSearchVectorStore
from .pgvectorSearch import PGVectorSearchComponent
__all__ = [
"AstraDBSearchComponent",
"ChromaSearchComponent",
"FAISSSearchComponent",
"MongoDBAtlasSearchComponent",
"PineconeSearchComponent",
"QdrantSearchComponent",
"RedisSearchComponent",
"SupabaseSearchComponent",
"VectaraSearchComponent",
"WeaviateSearchVectorStore",
"PGVectorSearchComponent",
]

View file

@ -1,44 +1,26 @@
from .AstraDB import AstraDBVectorStoreComponent
from .Chroma import ChromaComponent from .Chroma import ChromaComponent
from .ChromaSearch import ChromaSearchComponent
from .FAISS import FAISSComponent from .FAISS import FAISSComponent
from .FAISSSearch import FAISSSearchComponent
from .MongoDBAtlasVector import MongoDBAtlasComponent from .MongoDBAtlasVector import MongoDBAtlasComponent
from .MongoDBAtlasVectorSearch import MongoDBAtlasSearchComponent
from .Pinecone import PineconeComponent from .Pinecone import PineconeComponent
from .PineconeSearch import PineconeSearchComponent
from .Qdrant import QdrantComponent from .Qdrant import QdrantComponent
from .QdrantSearch import QdrantSearchComponent
from .Redis import RedisComponent from .Redis import RedisComponent
from .RedisSearch import RedisSearchComponent
from .SupabaseVectorStore import SupabaseComponent from .SupabaseVectorStore import SupabaseComponent
from .SupabaseVectorStoreSearch import SupabaseSearchComponent
from .Vectara import VectaraComponent from .Vectara import VectaraComponent
from .VectaraSearch import VectaraSearchComponent
from .Weaviate import WeaviateVectorStoreComponent from .Weaviate import WeaviateVectorStoreComponent
from .WeaviateSearch import WeaviateSearchVectorStore
from .pgvector import PGVectorComponent from .pgvector import PGVectorComponent
from .pgvectorSearch import PGVectorSearchComponent
__all__ = [ __all__ = [
"AstraDBVectorStoreComponent",
"ChromaComponent", "ChromaComponent",
"ChromaSearchComponent",
"FAISSComponent", "FAISSComponent",
"FAISSSearchComponent",
"MongoDBAtlasComponent", "MongoDBAtlasComponent",
"MongoDBAtlasSearchComponent",
"PineconeComponent", "PineconeComponent",
"PineconeSearchComponent",
"QdrantComponent", "QdrantComponent",
"QdrantSearchComponent",
"RedisComponent", "RedisComponent",
"RedisSearchComponent",
"SupabaseComponent", "SupabaseComponent",
"SupabaseSearchComponent",
"VectaraComponent", "VectaraComponent",
"VectaraSearchComponent",
"WeaviateVectorStoreComponent", "WeaviateVectorStoreComponent",
"WeaviateSearchVectorStore",
"base", "base",
"PGVectorComponent", "PGVectorComponent",
"PGVectorSearchComponent",
] ]

View file

@ -732,12 +732,12 @@ export const PRIORITY_SIDEBAR_ORDER = [
"saved_components", "saved_components",
"inputs", "inputs",
"outputs", "outputs",
"prompts",
"data", "data",
"prompt", "prompt",
"models", "models",
"vectorstores",
"vectorsearch",
"helpers", "helpers",
"experimental",
]; ];
export const NATIVE_CATEGORIES = [ export const NATIVE_CATEGORIES = [
"inputs", "inputs",

View file

@ -236,6 +236,7 @@ export const nodeColors: { [char: string]: string } = {
embeddings: "#42BAA7", embeddings: "#42BAA7",
documentloaders: "#7AAE42", documentloaders: "#7AAE42",
vectorstores: "#AA8742", vectorstores: "#AA8742",
vectorsearch: "#AA8742",
textsplitters: "#B47CB5", textsplitters: "#B47CB5",
toolkits: "#DB2C2C", toolkits: "#DB2C2C",
wrappers: "#E6277A", wrappers: "#E6277A",
@ -267,6 +268,7 @@ export const nodeNames: { [char: string]: string } = {
embeddings: "Embeddings", embeddings: "Embeddings",
documentloaders: "Loaders", documentloaders: "Loaders",
vectorstores: "Vector Stores", vectorstores: "Vector Stores",
vectorsearch: "Vector Search",
toolkits: "Toolkits", toolkits: "Toolkits",
wrappers: "Wrappers", wrappers: "Wrappers",
textsplitters: "Text Splitters", textsplitters: "Text Splitters",
@ -369,6 +371,7 @@ export const nodeIconsLucide: iconsType = {
saved_components: GradientSave, saved_components: GradientSave,
documentloaders: Paperclip, documentloaders: Paperclip,
vectorstores: Layers, vectorstores: Layers,
vectorsearch: Search,
toolkits: Package2, toolkits: Package2,
textsplitters: Scissors, textsplitters: Scissors,
wrappers: Gift, wrappers: Gift,