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

linting

parent eb1ebbb7
No related branches found
No related tags found
No related merge requests found
...@@ -8,7 +8,6 @@ from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder ...@@ -8,7 +8,6 @@ from semantic_router.encoders import BaseEncoder, CohereEncoder, OpenAIEncoder
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):
...@@ -215,12 +214,12 @@ class TestRouteLayer: ...@@ -215,12 +214,12 @@ class TestRouteLayer:
route_layer = RouteLayer(encoder=openai_encoder, routes=routes) route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
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): def test_query_filter(self, openai_encoder, routes):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes) route_layer = RouteLayer(encoder=openai_encoder, routes=routes)
query_result = route_layer(text="Hello", route_filter=["Route 1"]).name query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
assert query_result in ["Route 1"] assert query_result in ["Route 1"]
# def test_query_filter_openai_index(self, openai_encoder, routes): # def test_query_filter_openai_index(self, openai_encoder, routes):
# route_layer = RouteLayer(encoder=openai_encoder, routes=routes, index=PineconeIndex) # route_layer = RouteLayer(encoder=openai_encoder, routes=routes, index=PineconeIndex)
# query_result = route_layer(text="Hello", route_filter=["Route 1"]).name # query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment