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

add env var

parent 89029f01
Branches
Tags
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment