diff --git a/src/backend/base/langflow/components/vectorstores/Cassandra.py b/src/backend/base/langflow/components/vectorstores/Cassandra.py index 79f389465..da6f23f71 100644 --- a/src/backend/base/langflow/components/vectorstores/Cassandra.py +++ b/src/backend/base/langflow/components/vectorstores/Cassandra.py @@ -1,13 +1,13 @@ from typing import List from langchain_community.vectorstores import Cassandra -from langchain.schema import BaseRetriever from langflow.custom import Component from langflow.schema import Data from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput, DropdownInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_core.retrievers import BaseRetriever class CassandraVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/Couchbase.py b/src/backend/base/langflow/components/vectorstores/Couchbase.py index c409a3986..7dce4c704 100644 --- a/src/backend/base/langflow/components/vectorstores/Couchbase.py +++ b/src/backend/base/langflow/components/vectorstores/Couchbase.py @@ -2,13 +2,13 @@ from datetime import timedelta from typing import List from langchain_community.vectorstores import CouchbaseVectorStore -from langchain.schema import BaseRetriever from langflow.custom import Component from langflow.schema import Data from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_core.retrievers import BaseRetriever class CouchbaseVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/FAISS.py b/src/backend/base/langflow/components/vectorstores/FAISS.py index 52dd11792..8492e36c8 100644 --- a/src/backend/base/langflow/components/vectorstores/FAISS.py +++ b/src/backend/base/langflow/components/vectorstores/FAISS.py @@ -1,8 +1,5 @@ from typing import List -from langchain.vectorstores import FAISS -from langchain.schema import BaseRetriever - from langflow.custom import Component from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput from langflow.schema import Data @@ -11,6 +8,8 @@ from langflow.field_typing import Text from langflow.helpers.data import docs_to_data from loguru import logger +from langchain_community.vectorstores import FAISS +from langchain_core.retrievers import BaseRetriever class FaissVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py b/src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py index 3d83c246f..f56145cc6 100644 --- a/src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py +++ b/src/backend/base/langflow/components/vectorstores/MongoDBAtlasVector.py @@ -1,13 +1,13 @@ from typing import List from langchain_community.vectorstores import MongoDBAtlasVectorSearch -from langchain.schema import BaseRetriever from langflow.custom import Component from langflow.schema import Data from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_core.retrievers import BaseRetriever class MongoVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/Pinecone.py b/src/backend/base/langflow/components/vectorstores/Pinecone.py index 736f83480..f029db76d 100644 --- a/src/backend/base/langflow/components/vectorstores/Pinecone.py +++ b/src/backend/base/langflow/components/vectorstores/Pinecone.py @@ -1,13 +1,12 @@ from typing import List -from langchain.vectorstores import Pinecone -from langchain.schema import BaseRetriever - from langflow.custom import Component from langflow.schema import Data from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput, DropdownInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_core.retrievers import BaseRetriever +from langchain_pinecone import Pinecone class PineconeVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/Qdrant.py b/src/backend/base/langflow/components/vectorstores/Qdrant.py index be43e4599..9d9ec7768 100644 --- a/src/backend/base/langflow/components/vectorstores/Qdrant.py +++ b/src/backend/base/langflow/components/vectorstores/Qdrant.py @@ -1,13 +1,12 @@ from typing import List -from langchain.vectorstores import Qdrant -from langchain.schema import BaseRetriever - from langflow.custom import Component from langflow.schema import Data from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput, DropdownInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_community.vectorstores import Qdrant +from langchain_core.retrievers import BaseRetriever class QdrantVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py b/src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py index 75c1e2e13..4c7fe458b 100644 --- a/src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py +++ b/src/backend/base/langflow/components/vectorstores/SupabaseVectorStore.py @@ -1,7 +1,6 @@ from typing import List from langchain_community.vectorstores import SupabaseVectorStore -from langchain.schema import BaseRetriever from supabase.client import Client, create_client from langflow.custom import Component @@ -9,6 +8,7 @@ from langflow.schema import Data from langflow.inputs import IntInput, StrInput, HandleInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_core.retrievers import BaseRetriever class SupabaseVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/Upstash.py b/src/backend/base/langflow/components/vectorstores/Upstash.py index 97bfed4fd..d74381a8a 100644 --- a/src/backend/base/langflow/components/vectorstores/Upstash.py +++ b/src/backend/base/langflow/components/vectorstores/Upstash.py @@ -1,13 +1,13 @@ from typing import List from langchain_community.vectorstores import UpstashVectorStore -from langchain.schema import BaseRetriever from langflow.custom import Component from langflow.schema import Data from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_core.retrievers import BaseRetriever class UpstashVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/Vectara.py b/src/backend/base/langflow/components/vectorstores/Vectara.py index 99b07ea89..cf9d9d3d2 100644 --- a/src/backend/base/langflow/components/vectorstores/Vectara.py +++ b/src/backend/base/langflow/components/vectorstores/Vectara.py @@ -1,14 +1,13 @@ from typing import List -from langchain.embeddings import FakeEmbeddings -from langchain.vectorstores import Vectara -from langchain.schema import BaseRetriever - from langflow.custom import Component from langflow.schema import Data from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_community.embeddings import FakeEmbeddings +from langchain_community.vectorstores import Vectara +from langchain_core.retrievers import BaseRetriever class VectaraVectorStoreComponent(Component): diff --git a/src/backend/base/langflow/components/vectorstores/Weaviate.py b/src/backend/base/langflow/components/vectorstores/Weaviate.py index f22a7e823..a890cca72 100644 --- a/src/backend/base/langflow/components/vectorstores/Weaviate.py +++ b/src/backend/base/langflow/components/vectorstores/Weaviate.py @@ -1,14 +1,14 @@ from typing import List import weaviate -from langchain.vectorstores import Weaviate -from langchain.schema import BaseRetriever from langflow.custom import Component from langflow.schema import Data from langflow.inputs import BoolInput, IntInput, StrInput, HandleInput from langflow.template import Output from langflow.helpers.data import docs_to_data +from langchain_community.vectorstores import Weaviate +from langchain_core.retrievers import BaseRetriever class WeaviateVectorStoreComponent(Component):