phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-18 15:27:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

undo some unintentional changes, clean up some stuff

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-12 19:47:46 -08:00
parent 0121114a5d
commit 4f3b009980
3 changed files with 20 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -38,15 +38,15 @@ async def register_object_with_provider(obj: RoutableObject, p: Any) -> Routable
if api == Api.inference:
return await p.register_model(obj)
elif api == Api.safety:
await p.register_shield(**obj.model_dump())
await p.register_shield(obj)
elif api == Api.memory:
await p.register_memory_bank(**obj.model_dump())
await p.register_memory_bank(obj)
elif api == Api.datasetio:
await p.register_dataset(**obj.model_dump())
await p.register_dataset(obj)
elif api == Api.scoring:
await p.register_scoring_function(**obj.model_dump())
await p.register_scoring_function(obj)
elif api == Api.eval:
await p.register_eval_task(**obj.model_dump())
await p.register_eval_task(obj)
else:
raise ValueError(f"Unknown API {api} for registering object with provider")
@ -95,7 +95,6 @@ class CommonRoutingTableImpl(RoutingTable):
p.scoring_function_store = self
scoring_functions = await p.list_scoring_functions()
await add_objects(scoring_functions, pid, ScoringFn)
elif api == Api.eval:
p.eval_task_store = self