fix pinecone (#2414)

This commit is contained in:
Dev Khant
2025-03-20 23:47:09 +05:30
committed by GitHub
parent 8e6a08aa83
commit 3cc7013fde
2 changed files with 21 additions and 14 deletions

View File

@@ -199,12 +199,13 @@ class PineconeDB(VectorStoreBase):
return pinecone_filter
def search(self, query: List[float], limit: int = 5, filters: Optional[Dict] = None) -> List[OutputData]:
def search(self, query: str, vectors: List[float], limit: int = 5, filters: Optional[Dict] = None) -> List[OutputData]:
"""
Search for similar vectors.
Args:
query (list): Query vector.
query (str): Query.
vectors (list): List of vectors to search.
limit (int, optional): Number of results to return. Defaults to 5.
filters (dict, optional): Filters to apply to the search. Defaults to None.
@@ -214,7 +215,7 @@ class PineconeDB(VectorStoreBase):
filter_dict = self._create_filter(filters) if filters else None
query_params = {
"vector": query,
"vector": vectors,
"top_k": limit,
"include_metadata": True,
"include_values": False,