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

fix: test for single utt

parent ec78a75b
No related branches found
No related tags found
No related merge requests found
...@@ -258,16 +258,16 @@ class TestSemanticRouter: ...@@ -258,16 +258,16 @@ class TestSemanticRouter:
) )
assert route_layer.score_threshold == openai_encoder.score_threshold assert route_layer.score_threshold == openai_encoder.score_threshold
def test_initialization_single_utterance( def test_add_single_utterance(
self, route_single_utterance, openai_encoder, index_cls self, route_single_utterance, openai_encoder, index_cls
): ):
index = init_index(index_cls) index = init_index(index_cls)
route_layer = SemanticRouter( route_layer = SemanticRouter(
encoder=openai_encoder, encoder=openai_encoder,
routes=route_single_utterance,
index=index, index=index,
auto_sync="local", auto_sync="local",
) )
route_layer.add(routes=[route_single_utterance])
assert route_layer.score_threshold == openai_encoder.score_threshold assert route_layer.score_threshold == openai_encoder.score_threshold
if index_cls is PineconeIndex: if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be updated time.sleep(PINECONE_SLEEP) # allow for index to be updated
......
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