Refactor vector search components to use updated vector store model

src/backend/base/langflow/components/vectorsearch/AstraDBSearch.py

@@ -1,7 +1,7 @@
 from typing import List, Optional
 
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.AstraDB import AstraVectorStoreComponent
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
 from langflow.field_typing import Embeddings, Text
 from langflow.schema import Data
 

src/backend/base/langflow/components/vectorsearch/CassandraSearch.py

@@ -2,7 +2,7 @@ from typing import Any, List, Optional, Tuple
 
 from langchain_community.utilities.cassandra import SetupMode
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.Cassandra import CassandraVectorStoreComponent
 from langflow.field_typing import Embeddings, Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorsearch/ChromaSearch.py

@@ -3,7 +3,9 @@ from typing import List, Optional
 import chromadb
 from chromadb.config import Settings
 from langchain_chroma import Chroma
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+
+
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.field_typing import Embeddings, Text
 from langflow.schema import Data
 

src/backend/base/langflow/components/vectorsearch/CouchbaseSearch.py

@@ -1,6 +1,6 @@
 from typing import List
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.Couchbase import CouchbaseVectorStoreComponent
 from langflow.field_typing import Embeddings, Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorsearch/FAISSSearch.py

@@ -2,7 +2,7 @@ from typing import List
 
 from langchain_community.vectorstores.faiss import FAISS
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.field_typing import Embeddings, Text
 from langflow.schema import Data
 

src/backend/base/langflow/components/vectorsearch/MongoDBAtlasVectorSearch.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.MongoDBAtlasVector import MongoVectorStoreComponent
 from langflow.field_typing import Embeddings, NestedDict, Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorsearch/PineconeSearch.py

@@ -2,7 +2,7 @@ from typing import List, Optional
 
 from langchain_pinecone._utilities import DistanceStrategy
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.Pinecone import PineconeVectorStoreComponent
 from langflow.field_typing import Embeddings, Text
 from langflow.field_typing.constants import NestedDict

src/backend/base/langflow/components/vectorsearch/QdrantSearch.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.Qdrant import QdrantVectorStoreComponent
 from langflow.field_typing import Embeddings, NestedDict, Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorsearch/RedisSearch.py

@@ -2,7 +2,7 @@ from typing import List, Optional
 
 from langchain_core.embeddings import Embeddings
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.Redis import RedisVectorStoreComponent
 from langflow.field_typing import Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorsearch/SupabaseVectorStoreSearch.py

@@ -3,7 +3,7 @@ from typing import List
 from langchain_community.vectorstores.supabase import SupabaseVectorStore
 from supabase.client import Client, create_client
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.field_typing import Embeddings, Text
 from langflow.schema import Data
 

src/backend/base/langflow/components/vectorsearch/UpstashSearch.py

@@ -2,7 +2,7 @@ from typing import List, Optional
 
 from langchain_core.embeddings import Embeddings
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.Upstash import UpstashVectorStoreComponent
 from langflow.field_typing import Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorsearch/VectaraSearch.py

@@ -2,7 +2,7 @@ from typing import List
 
 from langchain_community.vectorstores.vectara import Vectara
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.Vectara import VectaraVectorStoreComponent
 from langflow.field_typing import Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorsearch/WeaviateSearch.py

@@ -2,7 +2,7 @@ from typing import List, Optional
 
 from langchain_core.embeddings import Embeddings
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.Weaviate import WeaviateVectorStoreComponent
 from langflow.field_typing import Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorsearch/pgvectorSearch.py

@@ -2,7 +2,7 @@ from typing import List
 
 from langchain_core.embeddings import Embeddings
 
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.components.vectorstores.pgvector import PGVectorStoreComponent
 from langflow.field_typing import Text
 from langflow.schema import Data

src/backend/base/langflow/components/vectorstores/Chroma.py

@@ -5,9 +5,10 @@ from chromadb.config import Settings
 from langchain_chroma.vectorstores import Chroma
 from loguru import logger
 
+from langflow.base.vectorstores.model import LCVectorStoreComponent
 from langflow.base.vectorstores.utils import chroma_collection_to_data
-from langflow.components.vectorstores.base.model import LCVectorStoreComponent
+from langflow.inputs import BoolInput, DataInput, DropdownInput, HandleInput, IntInput, MessageInput, StrInput
-from langflow.inputs import BoolInput, DropdownInput, HandleInput, IntInput, StrInput, MessageInput, DataInput
+                             IntInput, MessageInput, StrInput)
 from langflow.schema import Data
 
 if TYPE_CHECKING:

src/backend/base/langflow/components/vectorstores/base/__init__.py

@@ -1,3 +0,0 @@
-from .model import LCVectorStoreComponent
-
-__all__ = ["LCVectorStoreComponent"]