From d1b44c1251d387d06477abc6f9d327f0d7f9ea26 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Wed, 19 Mar 2025 09:55:18 -0700
Subject: [PATCH] benchmark->evaluation routing?

---
 llama_stack/distribution/distribution.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/llama_stack/distribution/distribution.py b/llama_stack/distribution/distribution.py
index 7e7237403..a710c49b1 100644
--- a/llama_stack/distribution/distribution.py
+++ b/llama_stack/distribution/distribution.py
@@ -43,13 +43,21 @@ def builtin_automatically_routed_apis() -> List[AutoRoutedApiInfo]:
             routing_table_api=Api.tool_groups,
             router_api=Api.tool_runtime,
         ),
+        AutoRoutedApiInfo(
+            routing_table_api=Api.benchmarks,
+            router_api=Api.evaluation,
+        ),
     ]
 
 
 def providable_apis() -> List[Api]:
-    routing_table_apis = {x.routing_table_api for x in builtin_automatically_routed_apis()}
+    routing_table_apis = {
+        x.routing_table_api for x in builtin_automatically_routed_apis()
+    }
     return [
-        api for api in Api if api not in routing_table_apis and api not in [Api.inspect, Api.providers, Api.benchmarks]
+        api
+        for api in Api
+        if api not in routing_table_apis and api not in [Api.inspect, Api.providers]
     ]