From 0262f97b2bc6579937ac384a7c5a6f472cce9fe9 Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Mon, 11 Mar 2024 17:17:03 +0400
Subject: [PATCH] Linting.
---
semantic_router/layer.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/semantic_router/layer.py b/semantic_router/layer.py
index 5acf92e0..1ab1db73 100644
--- a/semantic_router/layer.py
+++ b/semantic_router/layer.py
@@ -230,7 +230,7 @@ class RouteLayer:
route, top_class_scores = self._retrieve_top_route(vector)
passed = self._check_threshold(top_class_scores, route)
- if passed:
+ if passed and route is not None:
if route.function_schema and text is None:
raise ValueError(
"Route has a function schema, but no text was provided."
@@ -512,7 +512,7 @@ class RouteLayer:
"""
route, scores = self._retrieve_top_route(vector)
passed = self._check_threshold(scores, route)
- if passed:
+ if passed and route is not None:
return RouteChoice(
name=route.name, function_call=None, similarity_score=None, trigger=None
)
--
GitLab