From 0713c2eb48585803062e59f85aa20c6840f65a57 Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Tue, 2 Apr 2024 15:03:00 +0800
Subject: [PATCH] add os.environ
---
tests/unit/test_layer.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 8af6e1cc..59830da1 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -258,7 +258,8 @@ class TestRouteLayer:
assert query_result in ["Route 1"]
def test_query_filter_pinecone(self, openai_encoder, routes, index_cls):
- pineconeindex = PineconeIndex()
+ pinecone_api_key = os.environ["PINECONE_API_KEY"]
+ pineconeindex = PineconeIndex(api_key=pinecone_api_key)
route_layer = RouteLayer(
encoder=openai_encoder, routes=routes, index=pineconeindex
)
--
GitLab