jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor MongoDBAtlasVectorSearch imports

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-21 13:23:06 -03:00
commit c53a9d0714
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/components/vectorstores/MongoDBAtlasVectorSearch.py
View file

@ -1,8 +1,7 @@
from typing import List, Optional from typing import List, Optional
from langflow.components.vectorstores.base.model import LCVectorStoreComponent from langflow.components.vectorstores.base.model import LCVectorStoreComponent
from langflow.components.vectorstores.MongoDBAtlasVector import \ from langflow.components.vectorstores.MongoDBAtlasVector import MongoDBAtlasComponent
MongoDBAtlasComponent
from langflow.field_typing import Embeddings, NestedDict, Text from langflow.field_typing import Embeddings, NestedDict, Text
from langflow.schema import Record from langflow.schema import Record
@ -37,7 +36,6 @@ class MongoDBAtlasSearchComponent(MongoDBAtlasComponent, LCVectorStoreComponent)
mongodb_atlas_cluster_uri: str = "", mongodb_atlas_cluster_uri: str = "",
search_kwargs: Optional[NestedDict] = None, search_kwargs: Optional[NestedDict] = None,
) -> List[Record]: ) -> List[Record]:
search_kwargs = search_kwargs or {} search_kwargs = search_kwargs or {}
vector_store = super().build( vector_store = super().build(
connection_string=mongodb_atlas_cluster_uri, connection_string=mongodb_atlas_cluster_uri,