fix precommit

This commit is contained in:
Xi Yan 2025-03-23 16:42:50 -07:00
parent 81bc051411
commit 97e7717c9b
3 changed files with 71 additions and 3 deletions

View file

@ -46,6 +46,7 @@ async def get_routing_table_impl(
async def get_auto_router_impl(api: Api, routing_table: RoutingTable, deps: Dict[str, Any]) -> Any:
from .routers import (
DatasetIORouter,
EvaluationRouter,
InferenceRouter,
SafetyRouter,
ToolRuntimeRouter,
@ -58,6 +59,7 @@ async def get_auto_router_impl(api: Api, routing_table: RoutingTable, deps: Dict
"safety": SafetyRouter,
"datasetio": DatasetIORouter,
"tool_runtime": ToolRuntimeRouter,
"evaluation": EvaluationRouter,
}
api_to_deps = {
"inference": {"telemetry": Api.telemetry},