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

fix routing table

Browse source
This commit is contained in:
Xi Yan 2025-03-18 20:43:27 -07:00
parent 9a660a934b
commit 919955a2bf
1 changed files with 7 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -481,15 +481,12 @@ class BenchmarksRoutingTable(CommonRoutingTableImpl, Benchmarks):
) -> Benchmark:
if metadata is None:
metadata = {}
if provider_id is None:
if len(self.impls_by_provider_id) == 1:
provider_id = list(self.impls_by_provider_id.keys())[0]
else:
raise ValueError(
"No provider specified and multiple providers available. Please specify a provider_id."
)
if provider_benchmark_id is None:
provider_benchmark_id = benchmark_id
# TODO (xiyan): we will need a way to infer provider_id for evaluation
# keep it as meta-reference for now
if len(self.impls_by_provider_id) == 0:
raise ValueError("No evaluation providers available. Please configure an evaluation provider.")
provider_id = list(self.impls_by_provider_id.keys())[0]
benchmark = Benchmark(
identifier=benchmark_id,
@ -497,7 +494,7 @@ class BenchmarksRoutingTable(CommonRoutingTableImpl, Benchmarks):
grader_ids=grader_ids,
metadata=metadata,
provider_id=provider_id,
provider_resource_id=provider_benchmark_id,
provider_resource_id=benchmark_id,
)
await self.register_object(benchmark)
return benchmark