From a2f363f8540372d5df8e18a00aa0af3cb1eaa64f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Pedersen?= <andrped94@gmail.com>
Date: Wed, 13 Mar 2024 18:40:57 +0100
Subject: [PATCH] Added top_k check in class __init__
---
semantic_router/hybrid_layer.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/semantic_router/hybrid_layer.py b/semantic_router/hybrid_layer.py
index 4a9a368d..9791786f 100644
--- a/semantic_router/hybrid_layer.py
+++ b/semantic_router/hybrid_layer.py
@@ -37,6 +37,8 @@ class HybridRouteLayer:
self.alpha = alpha
self.top_k = top_k
+ if self.top_k < 1:
+ raise ValueError(f"top_k needs to be >= 1, but was: {self.top_k}.")
self.routes = routes
if isinstance(self.sparse_encoder, TfidfEncoder) and hasattr(
self.sparse_encoder, "fit"
--
GitLab