From b6c2aeb179bb1e90ddf0d5d9644e7b2c02991869 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sat, 27 Apr 2024 18:52:00 +0800
Subject: [PATCH] fixes
---
semantic_router/index/pinecone.py | 4 ++--
tests/unit/test_layer.py | 2 --
2 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/semantic_router/index/pinecone.py b/semantic_router/index/pinecone.py
index 148c9070..31fbfac8 100644
--- a/semantic_router/index/pinecone.py
+++ b/semantic_router/index/pinecone.py
@@ -122,7 +122,7 @@ class PineconeIndex(BaseIndex):
time.sleep(0.5)
elif index_exists:
# if the index exists we just return it
- index = self.client.Index(self.index_name, namespace=self.namespace)
+ index = self.client.Index(self.index_name)
# grab the dimensions from the index
self.dimensions = index.describe_index_stats()["dimension"]
elif force_create and not dimensions_given:
@@ -234,7 +234,7 @@ class PineconeIndex(BaseIndex):
def delete(self, route_name: str):
route_vec_ids = self._get_route_ids(route_name=route_name)
if self.index is not None:
- self.index.delete(ids=route_vec_ids)
+ self.index.delete(ids=route_vec_ids, namespace=self.namespace)
else:
raise ValueError("Index is None, could not delete.")
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 1cee98ab..7607d37c 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -281,9 +281,7 @@ class TestRouteLayer:
encoder=openai_encoder, routes=routes, index=pineconeindex
)
time.sleep(5) # allow for index to be populated
- print(routes)
query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
- print(query_result)
try:
route_layer(text="Hello", route_filter=["Route 8"]).name
--
GitLab