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

fix: single utt tests

parent 8f0f65b0
No related branches found
No related tags found
No related merge requests found
...@@ -268,7 +268,7 @@ class TestSemanticRouter: ...@@ -268,7 +268,7 @@ class TestSemanticRouter:
index=index, index=index,
auto_sync="local", auto_sync="local",
) )
route_layer.add(routes=[route_single_utterance]) 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
...@@ -286,7 +286,7 @@ class TestSemanticRouter: ...@@ -286,7 +286,7 @@ class TestSemanticRouter:
) )
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
route_layer.add(routes=[route_single_utterance]) 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
_ = route_layer("Hello") _ = route_layer("Hello")
assert len(route_layer.index.get_utterances()) == 1 assert len(route_layer.index.get_utterances()) == 1
......
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