phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 10:42:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

The first draft of the Preprocessing API.

Browse source
This commit is contained in:
ilya-kolchinsky 2025-03-03 13:32:17 +01:00
parent 7f9b767277
commit aa1b670d5c
18 changed files with 327 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/distribution/routers/__init__.py
View file

@ -14,6 +14,7 @@ from .routing_tables import (
BenchmarksRoutingTable,
DatasetsRoutingTable,
ModelsRoutingTable,
PreprocessorsRoutingTable,
ScoringFunctionsRoutingTable,
ShieldsRoutingTable,
ToolGroupsRoutingTable,
@ -35,6 +36,7 @@ async def get_routing_table_impl(
"scoring_functions": ScoringFunctionsRoutingTable,
"benchmarks": BenchmarksRoutingTable,
"tool_groups": ToolGroupsRoutingTable,
"preprocessors": PreprocessorsRoutingTable,
}
if api.value not in api_to_tables:
@ -50,6 +52,7 @@ async def get_auto_router_impl(api: Api, routing_table: RoutingTable, _deps) ->
DatasetIORouter,
EvalRouter,
InferenceRouter,
PreprocessingRouter,
SafetyRouter,
ScoringRouter,
ToolRuntimeRouter,
@ -64,6 +67,7 @@ async def get_auto_router_impl(api: Api, routing_table: RoutingTable, _deps) ->
"scoring": ScoringRouter,
"eval": EvalRouter,
"tool_runtime": ToolRuntimeRouter,
"preprocessing": PreprocessingRouter,
}
if api.value not in api_to_routers:
raise ValueError(f"API {api.value} not found in router map")