Skip to content

Commit 7679e26

Browse files
committed
fixes
1 parent aede7ae commit 7679e26

File tree

3 files changed

+32
-110
lines changed

3 files changed

+32
-110
lines changed

poetry.lock

Lines changed: 29 additions & 108 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

semantic_router/hybrid_layer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
from semantic_router.encoders import (
77
BaseEncoder,
8+
BM25Encoder,
89
TfidfEncoder,
910
)
1011
from semantic_router.route import Route
@@ -24,7 +25,7 @@ def __init__(
2425
routes: list[Route] = [],
2526
alpha: float = 0.3,
2627
):
27-
self.encoder = dense_encoder
28+
self.encoder = encoder
2829
self.score_threshold = self.encoder.score_threshold
2930

3031
if sparse_encoder is None:

tests/unit/test_hybrid_layer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def test_failover_score_threshold(self, base_encoder):
164164

165165
def test_add_route_tfidf(self, cohere_encoder, tfidf_encoder, routes):
166166
hybrid_route_layer = HybridRouteLayer(
167-
dense_encoder=cohere_encoder,
167+
encoder=cohere_encoder,
168168
sparse_encoder=tfidf_encoder,
169169
routes=routes[:-1],
170170
)

0 commit comments

Comments
 (0)