From c53a9d07142fdc6ed4c6cc50a1d31e11ff91b4ef Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 21 Mar 2024 13:23:06 -0300 Subject: [PATCH] Refactor MongoDBAtlasVectorSearch imports --- .../components/vectorstores/MongoDBAtlasVectorSearch.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/vectorstores/MongoDBAtlasVectorSearch.py b/src/backend/langflow/components/vectorstores/MongoDBAtlasVectorSearch.py index 7f365914e..10b69ce15 100644 --- a/src/backend/langflow/components/vectorstores/MongoDBAtlasVectorSearch.py +++ b/src/backend/langflow/components/vectorstores/MongoDBAtlasVectorSearch.py @@ -1,8 +1,7 @@ from typing import List, Optional from langflow.components.vectorstores.base.model import LCVectorStoreComponent -from langflow.components.vectorstores.MongoDBAtlasVector import \ - MongoDBAtlasComponent +from langflow.components.vectorstores.MongoDBAtlasVector import MongoDBAtlasComponent from langflow.field_typing import Embeddings, NestedDict, Text from langflow.schema import Record @@ -37,9 +36,8 @@ class MongoDBAtlasSearchComponent(MongoDBAtlasComponent, LCVectorStoreComponent) mongodb_atlas_cluster_uri: str = "", search_kwargs: Optional[NestedDict] = None, ) -> List[Record]: - search_kwargs = search_kwargs or {} - vector_store = super().build( + vector_store = super().build( connection_string=mongodb_atlas_cluster_uri, namespace=f"{db_name}.{collection_name}", embedding=embedding,