From a89724ded19a30c23500ee1e62ded5f1a3376815 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 15 Dec 2024 14:19:43 +0400
Subject: [PATCH] fix: test for single utt
---
tests/unit/test_router.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index 0dece3d7..c7dec080 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -258,16 +258,16 @@ class TestSemanticRouter:
)
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
):
index = init_index(index_cls)
route_layer = SemanticRouter(
encoder=openai_encoder,
- routes=route_single_utterance,
index=index,
auto_sync="local",
)
+ route_layer.add(routes=[route_single_utterance])
assert route_layer.score_threshold == openai_encoder.score_threshold
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be updated
--
GitLab