diff --git a/semantic_router/layer.py b/semantic_router/layer.py index aa4a8ef6623fe0857cfc10f3677252a47e3f3d2f..4f478ea0fddaaa2b3d1c20ab54650dcaa2deb013 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 = (