From 3fa78fc491c818e616c6b7390e5731c1a6ee3c75 Mon Sep 17 00:00:00 2001 From: Ismail Ashraq <ismailashraq@Ismails-MacBook-Pro.local> Date: Tue, 16 Jul 2024 15:10:38 +0800 Subject: [PATCH] more lint fixes --- semantic_router/index/local.py | 2 +- semantic_router/index/pinecone.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/semantic_router/index/local.py b/semantic_router/index/local.py index 1116ffe4..5426ec76 100644 --- a/semantic_router/index/local.py +++ b/semantic_router/index/local.py @@ -98,7 +98,7 @@ class LocalIndex(BaseIndex): scores, idx = top_scores(sim, top_k) route_names = [self.routes[i] for i in idx] return scores, route_names - + async def aquery( self, vector: np.ndarray, diff --git a/semantic_router/index/pinecone.py b/semantic_router/index/pinecone.py index 732e1f35..c04cdc9a 100644 --- a/semantic_router/index/pinecone.py +++ b/semantic_router/index/pinecone.py @@ -475,7 +475,7 @@ class PineconeIndex(BaseIndex): filter_query = None results = self.index.query( vector=[query_vector_list], - sparse_vector=kwargs.get('sparse_vector', None), + sparse_vector=kwargs.get("sparse_vector", None), top_k=top_k, filter=filter_query, include_metadata=True, @@ -501,7 +501,7 @@ class PineconeIndex(BaseIndex): filter_query = None results = await self._async_query( vector=query_vector_list, - sparse_vector=kwargs.get('sparse_vector', None), + sparse_vector=kwargs.get("sparse_vector", None), namespace=self.namespace or "", filter=filter_query, top_k=top_k, -- GitLab