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: except PyMongoError as e:
logger.error(f"Error inserting data: {e}") logger.error(f"Error inserting data: {e}")
def search( def search(self, query: str, vectors: List[float], limit=5, filters: Optional[Dict] = None) -> List[OutputData]:
self, query: str, query_vector: List[float], limit=5, filters: Optional[Dict] = None
) -> List[OutputData]:
""" """
Search for similar vectors using the vector search index. Search for similar vectors using the vector search index.
Args: Args:
query (str): Query string 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. limit (int, optional): Number of results to return. Defaults to 5.
filters (Dict, optional): Filters to apply to the search. filters (Dict, optional): Filters to apply to the search.
@@ -141,7 +139,7 @@ class MongoDB(VectorStoreBase):
"index": self.index_name, "index": self.index_name,
"limit": limit, "limit": limit,
"numCandidates": limit, "numCandidates": limit,
"queryVector": query_vector, "queryVector": vectors,
"path": "embedding", "path": "embedding",
} }
}, },