replace query_vector args in search method of mongodb vector_stores (#3379)
This commit is contained in:
@@ -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",
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|||||||
Reference in New Issue
Block a user