WIP

Author: rbs333

redisvl/extensions/router/schema.py

@@ -16,7 +16,7 @@ class Route(BaseModel):
     """List of reference phrases for the route."""
     metadata: Dict[str, str] = Field(default={})
     """Metadata associated with the route."""
-    distance_threshold: Optional[float] = Field(default=None)
+    distance_threshold: Optional[float] = Field(default=0.5)
     """Distance threshold for matching the route."""
 
     @validator("name")
@@ -63,7 +63,7 @@ class DistanceAggregationMethod(Enum):
 class RoutingConfig(BaseModel):
     """Configuration for routing behavior."""
 
-    distance_threshold: float = Field(default=0.5)
+    # distance_threshold: float = Field(default=0.5)
     """The threshold for semantic distance."""
     max_k: int = Field(default=1)
     """The maximum number of top matches to return."""

redisvl/extensions/router/semantic.py

@@ -302,12 +302,17 @@ def _classify_multi_route(
         aggregation_method: DistanceAggregationMethod,
     ) -> List[RouteMatch]:
         """Classify to multiple routes, up to max_k (int), using a vector."""
+
+        # Set range query distance threshold to get all possible results to be further filtered
+        distance_threshold = max(route.distance_threshold for route in self.routes)
+
         vector_range_query = RangeQuery(
             vector=vector,
             vector_field_name=ROUTE_VECTOR_FIELD_NAME,
             distance_threshold=distance_threshold,
             return_fields=["route_name"],
         )
+
         aggregate_request = self._build_aggregate_request(
             vector_range_query, aggregation_method, max_k
         )
@@ -368,7 +373,9 @@ def __call__(
         self,
         statement: Optional[str] = None,
         vector: Optional[List[float]] = None,
-        distance_threshold: Optional[float] = None,
+        distance_threshold: Optional[
+            float
+        ] = None,  # TODO: does this get removed if route becomes the owner of a distance threshold? Or do we apply to all?
         aggregation_method: Optional[DistanceAggregationMethod] = None,
     ) -> RouteMatch:
         """Query the semantic router with a given statement or vector.