replace query_vector args in search method of mongodb vector_stores (#3379)

This commit is contained in:
Rupam Jana
2025-08-29 19:05:25 +05:30
committed by GitHub
parent 84af5ad265
commit 458d7ab8a3
+3 -5
View File
@@ -111,15 +111,13 @@ class MongoDB(VectorStoreBase):
except PyMongoError as e:
logger.error(f"Error inserting data: {e}")
def search(
self, query: str, query_vector: List[float], limit=5, filters: Optional[Dict] = None
) -> List[OutputData]:
def search(self, query: str, vectors: List[float], limit=5, filters: Optional[Dict] = None) -> List[OutputData]:
"""
Search for similar vectors using the vector search index.
Args:
query (str): Query string
query_vector (List[float]): Query vector.
vectors (List[float]): Query vector.
limit (int, optional): Number of results to return. Defaults to 5.
filters (Dict, optional): Filters to apply to the search.
@@ -141,7 +139,7 @@ class MongoDB(VectorStoreBase):
"index": self.index_name,
"limit": limit,
"numCandidates": limit,
"queryVector": query_vector,
"queryVector": vectors,
"path": "embedding",
}
},