diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py index be10e3b6033a6bd8d7f6bf3c52dcc5893790d1fc..067e78a9be0f5b17e555ab3512c24eac88d15e86 100644 --- a/tests/unit/test_router.py +++ b/tests/unit/test_router.py @@ -272,6 +272,23 @@ class TestSemanticRouter: assert route_layer.score_threshold == openai_encoder.score_threshold if index_cls is PineconeIndex: time.sleep(PINECONE_SLEEP) # allow for index to be updated + _ = route_layer("Hello") + assert len(route_layer.index.get_utterances()) == 6 + + def test_init_and_add_single_utterance( + self, route_single_utterance, openai_encoder, index_cls + ): + index = init_index(index_cls) + route_layer = SemanticRouter( + encoder=openai_encoder, + index=index, + auto_sync="local", + ) + if index_cls is PineconeIndex: + time.sleep(PINECONE_SLEEP) # allow for index to be updated + route_layer.add(routes=[route_single_utterance]) + assert route_layer.score_threshold == openai_encoder.score_threshold + _ = route_layer("Hello") assert len(route_layer.index.get_utterances()) == 1 def test_delete_index(self, openai_encoder, routes, index_cls):