tweaks

Author: tylerhutcherson

docs/_static/js/sidebar.js

@@ -11,6 +11,7 @@ const toc = [
         { title: "Vectorizers", path: "/user_guide/vectorizers_04.html" },
         { title: "Rerankers", path: "/user_guide/rerankers_06.html" },
         { title: "Semantic Caching", path: "/user_guide/llmcache_03.html" },
+        { title: "Semantic Routing", path: "/user_guide/semantic_router_08.html" },
     ]},
     { header: "API", toc: [
         { title: "Schema", path: "/api/schema.html"},

docs/user_guide/index.md

@@ -18,5 +18,6 @@ vectorizers_04
 hash_vs_json_05
 rerankers_06
 session_manager_07
+semantic_router_08
 ```
 

docs/user_guide/semantic_router_08.ipynb

@@ -90,7 +90,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "14:09:10 redisvl.index.index INFO   Index already exists, overwriting.\n"
+      "14:13:26 redisvl.index.index INFO   Index already exists, overwriting.\n"
      ]
     }
    ],
@@ -136,16 +136,6 @@
    "execution_count": 4,
    "metadata": {},
    "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "huggingface/tokenizers: The current process just got forked, after parallelism has already been used. Disabling parallelism to avoid deadlocks...\n",
-      "To disable this warning, you can either:\n",
-      "\t- Avoid using `tokenizers` before the fork if possible\n",
-      "\t- Explicitly set the environment variable TOKENIZERS_PARALLELISM=(true | false)\n"
-     ]
-    },
     {
      "name": "stdout",
      "output_type": "stream",
@@ -343,28 +333,15 @@
    "cell_type": "code",
    "execution_count": 11,
    "metadata": {},
-   "outputs": [
-    {
-     "ename": "AttributeError",
-     "evalue": "'SearchIndex' object has no attribute 'clear'",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
-      "Cell \u001b[0;32mIn[11], line 2\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[38;5;66;03m# Use clear to flush all routes from the index\u001b[39;00m\n\u001b[0;32m----> 2\u001b[0m \u001b[43mrouter\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mclear\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n",
-      "File \u001b[0;32m~/AppliedAI/redis-vl-python/redisvl/extensions/router/semantic.py:437\u001b[0m, in \u001b[0;36mSemanticRouter.clear\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m    436\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m \u001b[38;5;21mclear\u001b[39m(\u001b[38;5;28mself\u001b[39m):\n\u001b[0;32m--> 437\u001b[0m     \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43m_index\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mclear\u001b[49m()\n",
-      "\u001b[0;31mAttributeError\u001b[0m: 'SearchIndex' object has no attribute 'clear'"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "# Use clear to flush all routes from the index\n",
     "router.clear()"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 12,
    "metadata": {},
    "outputs": [],
    "source": [

redisvl/extensions/router/semantic.py

@@ -336,7 +336,9 @@ def _classify_many(
                 )
             raise e
 
-    def _pass_threshold(self, route_match: Optional[RouteMatch], distance_threshold: float) -> bool:
+    def _pass_threshold(
+        self, route_match: Optional[RouteMatch], distance_threshold: float
+    ) -> bool:
         """Check if a route match passes the distance threshold.
 
         Args:
@@ -348,7 +350,9 @@ def _pass_threshold(self, route_match: Optional[RouteMatch], distance_threshold:
         """
         if route_match:
             if route_match.distance is not None and route_match.route is not None:
-                _distance_threshold = route_match.route.distance_threshold or distance_threshold
+                _distance_threshold = (
+                    route_match.route.distance_threshold or distance_threshold
+                )
                 if _distance_threshold:
                     return route_match.distance <= _distance_threshold
         return False
@@ -358,7 +362,7 @@ def __call__(
         statement: Optional[str] = None,
         vector: Optional[List[float]] = None,
         distance_threshold: Optional[float] = None,
-        aggregation_method: Optional[DistanceAggregationMethod] = None
+        aggregation_method: Optional[DistanceAggregationMethod] = None,
     ) -> RouteMatch:
         """Query the semantic router with a given statement or vector.
 
@@ -379,10 +383,10 @@ def __call__(
         distance_threshold = (
             distance_threshold or self.routing_config.distance_threshold
         )
-        aggregation_method = aggregation_method or self.routing_config.aggregation_method
-        route_matches = self._classify(
-            vector, distance_threshold, aggregation_method
+        aggregation_method = (
+            aggregation_method or self.routing_config.aggregation_method
         )
+        route_matches = self._classify(vector, distance_threshold, aggregation_method)
         route_match = route_matches[0] if route_matches else None
 
         if route_match and self._pass_threshold(route_match, distance_threshold):
@@ -396,7 +400,7 @@ def route_many(
         vector: Optional[List[float]] = None,
         max_k: Optional[int] = None,
         distance_threshold: Optional[float] = None,
-        aggregation_method: Optional[DistanceAggregationMethod] = None
+        aggregation_method: Optional[DistanceAggregationMethod] = None,
     ) -> List[RouteMatch]:
         """Query the semantic router with a given statement or vector for multiple matches.
 
@@ -419,7 +423,9 @@ def route_many(
             distance_threshold or self.routing_config.distance_threshold
         )
         max_k = max_k or self.routing_config.max_k
-        aggregation_method = aggregation_method or self.routing_config.aggregation_method
+        aggregation_method = (
+            aggregation_method or self.routing_config.aggregation_method
+        )
         route_matches = self._classify_many(
             vector, max_k, distance_threshold, aggregation_method
         )

tests/integration/test_semantic_router.py

@@ -33,7 +33,7 @@ def semantic_router(client, routes):
         overwrite=False,
     )
     yield router
-    router._index.delete(drop=True)
+    router.delete()
 
 
 def test_initialize_router(semantic_router):