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

add env var

parent 89029f01
No related branches found
No related tags found
No related merge requests found
...@@ -120,9 +120,9 @@ class TestRouteLayer: ...@@ -120,9 +120,9 @@ class TestRouteLayer:
assert route_layer_openai.score_threshold == 0.82 assert route_layer_openai.score_threshold == 0.82
def test_initialization_no_encoder(self, openai_encoder): def test_initialization_no_encoder(self, openai_encoder):
with patch("semantic_router.encoders.OpenAIEncoder") as _: os.environ["OPENAI_API_KEY"] = "test_api_key"
route_layer_none = RouteLayer(encoder=None) route_layer_none = RouteLayer(encoder=None)
assert route_layer_none.score_threshold == openai_encoder.score_threshold assert route_layer_none.score_threshold == openai_encoder.score_threshold
def test_initialization_dynamic_route(self, cohere_encoder, openai_encoder): def test_initialization_dynamic_route(self, cohere_encoder, openai_encoder):
route_layer_cohere = RouteLayer(encoder=cohere_encoder) route_layer_cohere = RouteLayer(encoder=cohere_encoder)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment