Skip to content
Snippets Groups Projects
Unverified Commit 9f16810c authored by James Briggs's avatar James Briggs
Browse files

add delete and increase sleep

parent c4dd453a
Branches
Tags
No related merge requests found
......@@ -280,7 +280,7 @@ class TestRouteLayer:
route_layer = RouteLayer(
encoder=openai_encoder, routes=routes, index=pineconeindex
)
time.sleep(5) # allow for index to be populated
time.sleep(10) # allow for index to be populated
query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
try:
......@@ -288,6 +288,9 @@ class TestRouteLayer:
except ValueError:
assert True
# delete index
pineconeindex.delete_index()
assert query_result in ["Route 1"]
def test_namespace_pinecone_index(self, openai_encoder, routes, index_cls):
......@@ -296,7 +299,7 @@ class TestRouteLayer:
route_layer = RouteLayer(
encoder=openai_encoder, routes=routes, index=pineconeindex
)
time.sleep(5) # allow for index to be populated
time.sleep(10) # allow for index to be populated
query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
try:
......@@ -304,6 +307,9 @@ class TestRouteLayer:
except ValueError:
assert True
# delete index
pineconeindex.delete_index()
assert query_result in ["Route 1"]
def test_query_with_no_index(self, openai_encoder, index_cls):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment