feat(router): add LLM routing with cost optimization and pretrained configs

redisvl/extensions/router/__init__.py

@@ -1,4 +1,25 @@
-from redisvl.extensions.router.schema import Route, RoutingConfig
-from redisvl.extensions.router.semantic import SemanticRouter
+from redisvl.extensions.router.schema import (
+    DistanceAggregationMethod,
+    PretrainedReference,
+    PretrainedRoute,
+    PretrainedRouterConfig,
+    Route,
+    RouteMatch,
+    RoutingConfig,
+)
+from redisvl.extensions.router.semantic import AsyncSemanticRouter, SemanticRouter
 
-__all__ = ["SemanticRouter", "Route", "RoutingConfig"]
+__all__ = [
+    # Main router classes
+    "SemanticRouter",
+    "AsyncSemanticRouter",
+    # Schema classes
+    "Route",
+    "RouteMatch",
+    "RoutingConfig",
+    "DistanceAggregationMethod",
+    # Pretrained classes
+    "PretrainedReference",
+    "PretrainedRoute",
+    "PretrainedRouterConfig",
+]

redisvl/extensions/router/pretrained/__init__.py

@@ -0,0 +1,30 @@
+"""Pretrained router configurations.
+
+This module provides pre-built router configurations that can be loaded
+without needing to re-embed references.
+"""
+
+from pathlib import Path
+
+PRETRAINED_DIR = Path(__file__).parent
+
+
+def get_pretrained_path(name: str) -> Path:
+    """Get path to a pretrained configuration.
+
+    Args:
+        name: Pretrained config name (e.g., "default", "cost_optimized")
+
+    Returns:
+        Path to the pretrained JSON file.
+    """
+    path = PRETRAINED_DIR / f"{name}.json"
+    if not path.exists():
+        available = [f.stem for f in PRETRAINED_DIR.glob("*.json")]
+        raise ValueError(
+            f"Pretrained config '{name}' not found. " f"Available: {available}"
+        )
+    return path
+
+
+__all__ = ["get_pretrained_path", "PRETRAINED_DIR"]