phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-12 13:00:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update

Browse source
This commit is contained in:
Xi Yan 2025-02-12 20:42:01 -08:00
parent 8759196e29
commit e07776fff6
1 changed files with 16 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

20
llama_stack/distribution/routers/routing_tables.py
View file

@ -4,6 +4,7 @@
# This source code is licensed under the terms described in the LICENSE file in # This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree. # the root directory of this source tree.
import logging
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
from pydantic import TypeAdapter from pydantic import TypeAdapter
@ -39,6 +40,8 @@ from llama_stack.distribution.datatypes import (
from llama_stack.distribution.store import DistributionRegistry from llama_stack.distribution.store import DistributionRegistry
from llama_stack.providers.datatypes import Api, RoutingTable from llama_stack.providers.datatypes import Api, RoutingTable
logger = logging.getLogger(__name__)
def get_impl_api(p: Any) -> Api: def get_impl_api(p: Any) -> Api:
return p.__provider_spec__.api return p.__provider_spec__.api
@ -466,16 +469,18 @@ class BenchmarksRoutingTable(CommonRoutingTableImpl, Benchmarks):
) )
await self.register_object(benchmark) await self.register_object(benchmark)
async def DEPRECATED_list_benchmarks(self) -> ListBenchmarksResponse: async def DEPRECATED_list_eval_tasks(self) -> ListBenchmarksResponse:
logger.warning("DEPRECATED: Use /eval/benchmarks instead")
raise DeprecationWarning("Use /eval/tasks instead") raise DeprecationWarning("Use /eval/tasks instead")
async def DEPRECATED_get_benchmark( async def DEPRECATED_get_eval_task(
self, self,
benchmark_id: str, benchmark_id: str,
) -> Optional[Benchmark]: ) -> Optional[Benchmark]:
logger.warning("DEPRECATED: Use /eval/benchmarks instead")
raise DeprecationWarning("Use /eval/tasks instead") raise DeprecationWarning("Use /eval/tasks instead")
async def DEPRECATED_register_benchmark( async def DEPRECATED_register_eval_task(
self, self,
benchmark_id: str, benchmark_id: str,
dataset_id: str, dataset_id: str,
@ -484,7 +489,14 @@ class BenchmarksRoutingTable(CommonRoutingTableImpl, Benchmarks):
provider_id: Optional[str] = None, provider_id: Optional[str] = None,
metadata: Optional[Dict[str, Any]] = None, metadata: Optional[Dict[str, Any]] = None,
) -> None: ) -> None:
raise DeprecationWarning("Use /eval/tasks instead") logger.warning("DEPRECATED: Use /eval/benchmarks instead")
self.register_benchmark(
benchmark_id=benchmark_id,
dataset_id=dataset_id,
scoring_functions=scoring_functions,
metadata=metadata,
provider_benchmark_id=provider_benchmark_id,
)
class ToolGroupsRoutingTable(CommonRoutingTableImpl, ToolGroups): class ToolGroupsRoutingTable(CommonRoutingTableImpl, ToolGroups):