From 4286b5461ff9c9f2d1a2d271b5db932a816f3a82 Mon Sep 17 00:00:00 2001 From: tolgadevAI <164843802+tolgadevAI@users.noreply.github.com> Date: Tue, 27 Aug 2024 15:29:07 +0300 Subject: [PATCH] lint --- semantic_router/layer.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/semantic_router/layer.py b/semantic_router/layer.py index aa4a8ef6..4f478ea0 100644 --- a/semantic_router/layer.py +++ b/semantic_router/layer.py @@ -488,16 +488,16 @@ class RouteLayer: route_names = [] all_embeddings = [] - all_utterances = [] + all_utterances: List[str] = [] all_function_schemas = [] for route in routes: logger.info(f"Adding `{route.name}` route") route_embeddings = self.encoder(route.utterances) - + # Set score_threshold if not already set route.score_threshold = route.score_threshold or self.score_threshold - + # Prepare data for batch insertion route_names.extend([route.name] * len(route.utterances)) all_embeddings.extend(route_embeddings) @@ -520,7 +520,6 @@ class RouteLayer: logger.error(f"Failed to add routes to the index: {e}") raise Exception("Indexing error occurred") from e - def _add_and_sync_routes(self, routes: List[Route]): # create embeddings for all routes and sync at startup with remote ones based on sync setting local_route_names, local_utterances, local_function_schemas = ( -- GitLab