From 697dbaffd0a8eccf56c90a71b8a9a029c6403218 Mon Sep 17 00:00:00 2001
From: zahid-syed <zahid.s2618@gmail.com>
Date: Wed, 20 Mar 2024 15:08:14 -0400
Subject: [PATCH] added filter test again after merge
---
tests/unit/test_layer.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 2886a46e..cd32b35e 100644
--- a/tests/unit/test_layer.py
+++ b/tests/unit/test_layer.py
@@ -11,7 +11,6 @@ from semantic_router.index.qdrant import QdrantIndex
from semantic_router.layer import LayerConfig, RouteLayer
from semantic_router.llms.base import BaseLLM
from semantic_router.route import Route
-from semantic_router.index.pinecone import PineconeIndex
def mock_encoder_call(utterances):
@@ -245,6 +244,12 @@ class TestRouteLayer:
query_result = route_layer(text="Hello").name
assert query_result in ["Route 1", "Route 2"]
+
+ def test_query_filter(self, openai_encoder, routes, index_cls):
+ route_layer = RouteLayer(encoder=openai_encoder, routes=routes, index=index_cls())
+ query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
+ assert query_result in ["Route 1"]
+
def test_query_with_no_index(self, openai_encoder, index_cls):
route_layer = RouteLayer(encoder=openai_encoder, index=index_cls())
with pytest.raises(ValueError):
--
GitLab