Skip to content

Commit 534d9ee

Browse files
committed
add env var
1 parent 89029f0 commit 534d9ee

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/unit/test_layer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -120,9 +120,9 @@ def test_initialization_different_encoders(self, cohere_encoder, openai_encoder)
120120
assert route_layer_openai.score_threshold == 0.82
121121

122122
def test_initialization_no_encoder(self, openai_encoder):
123-
with patch("semantic_router.encoders.OpenAIEncoder") as _:
124-
route_layer_none = RouteLayer(encoder=None)
125-
assert route_layer_none.score_threshold == openai_encoder.score_threshold
123+
os.environ["OPENAI_API_KEY"] = "test_api_key"
124+
route_layer_none = RouteLayer(encoder=None)
125+
assert route_layer_none.score_threshold == openai_encoder.score_threshold
126126

127127
def test_initialization_dynamic_route(self, cohere_encoder, openai_encoder):
128128
route_layer_cohere = RouteLayer(encoder=cohere_encoder)

0 commit comments

Comments
 (0)