"""Inference module for query routing.

Provides two routing pipelines:

1. Router (MLP-based): Fast, simple routing using a 4-layer feedforward network
2. GraphRouter (dual-encoder): Rich query-LLM compatibility modeling using
   learned embeddings

Both pipelines use the same text embedding model for query encoding.
"""

from partner_inference_harness.inference.graph_router import (
    GraphRouter,
    GraphRoutingResult,
    create_graph_router,
)
from partner_inference_harness.inference.router import (
    Router,
    RoutingResult,
    create_router,
)

__all__ = [
    # MLP Router
    "Router",
    "RoutingResult",
    "create_router",
    # Graph Router
    "GraphRouter",
    "GraphRoutingResult",
    "create_graph_router",
]
