Skip to content
Snippets Groups Projects
Commit 697dbaff authored by zahid-syed's avatar zahid-syed
Browse files

added filter test again after merge

parent 23b7363a
No related branches found
No related tags found
No related merge requests found
...@@ -11,7 +11,6 @@ from semantic_router.index.qdrant import QdrantIndex ...@@ -11,7 +11,6 @@ from semantic_router.index.qdrant import QdrantIndex
from semantic_router.layer import LayerConfig, RouteLayer from semantic_router.layer import LayerConfig, RouteLayer
from semantic_router.llms.base import BaseLLM from semantic_router.llms.base import BaseLLM
from semantic_router.route import Route from semantic_router.route import Route
from semantic_router.index.pinecone import PineconeIndex
def mock_encoder_call(utterances): def mock_encoder_call(utterances):
...@@ -245,6 +244,12 @@ class TestRouteLayer: ...@@ -245,6 +244,12 @@ class TestRouteLayer:
query_result = route_layer(text="Hello").name query_result = route_layer(text="Hello").name
assert query_result in ["Route 1", "Route 2"] 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): def test_query_with_no_index(self, openai_encoder, index_cls):
route_layer = RouteLayer(encoder=openai_encoder, index=index_cls()) route_layer = RouteLayer(encoder=openai_encoder, index=index_cls())
with pytest.raises(ValueError): with pytest.raises(ValueError):
......
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