From b6e2b19c3be6fa61b4dbdcae10807235e547d378 Mon Sep 17 00:00:00 2001
From: Ismail Ashraq <ismailashraq@Ismails-MacBook-Pro.local>
Date: Tue, 16 Jul 2024 14:52:11 +0800
Subject: [PATCH] fix lint errors
---
semantic_router/index/pinecone.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/semantic_router/index/pinecone.py b/semantic_router/index/pinecone.py
index dfc5aaef..732e1f35 100644
--- a/semantic_router/index/pinecone.py
+++ b/semantic_router/index/pinecone.py
@@ -462,9 +462,9 @@ class PineconeIndex(BaseIndex):
def query(
self,
vector: np.ndarray,
- sparse_vector: Optional[dict] = None,
top_k: int = 5,
route_filter: Optional[List[str]] = None,
+ **kwargs: Any,
) -> Tuple[np.ndarray, List[str]]:
if self.index is None:
raise ValueError("Index is not populated.")
@@ -475,7 +475,7 @@ class PineconeIndex(BaseIndex):
filter_query = None
results = self.index.query(
vector=[query_vector_list],
- sparse_vector=sparse_vector,
+ sparse_vector=kwargs.get('sparse_vector', None),
top_k=top_k,
filter=filter_query,
include_metadata=True,
@@ -488,9 +488,9 @@ class PineconeIndex(BaseIndex):
async def aquery(
self,
vector: np.ndarray,
- sparse_vector: Optional[dict] = None,
top_k: int = 5,
route_filter: Optional[List[str]] = None,
+ **kwargs: Any,
) -> Tuple[np.ndarray, List[str]]:
if self.async_client is None or self.host is None:
raise ValueError("Async client or host are not initialized.")
@@ -501,7 +501,7 @@ class PineconeIndex(BaseIndex):
filter_query = None
results = await self._async_query(
vector=query_vector_list,
- sparse_vector=sparse_vector,
+ sparse_vector=kwargs.get('sparse_vector', None),
namespace=self.namespace or "",
filter=filter_query,
top_k=top_k,
--
GitLab