From 534d9eef02acdc085b693c38fc32683c77ea22ed Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 7 Jan 2024 16:59:27 +0100
Subject: [PATCH] add env var
---
tests/unit/test_layer.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index cdf86bbc..6e2ac317 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -120,9 +120,9 @@ class TestRouteLayer:
assert route_layer_openai.score_threshold == 0.82
def test_initialization_no_encoder(self, openai_encoder):
- with patch("semantic_router.encoders.OpenAIEncoder") as _:
- route_layer_none = RouteLayer(encoder=None)
- assert route_layer_none.score_threshold == openai_encoder.score_threshold
+ os.environ["OPENAI_API_KEY"] = "test_api_key"
+ route_layer_none = RouteLayer(encoder=None)
+ assert route_layer_none.score_threshold == openai_encoder.score_threshold
def test_initialization_dynamic_route(self, cohere_encoder, openai_encoder):
route_layer_cohere = RouteLayer(encoder=cohere_encoder)
--
GitLab