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

persist registered objects with distribution

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-01 08:31:17 -07:00 • committed by Dinesh Yeduguru
parent ac93dd89cf
commit 8fe22230b8
9 changed files with 85 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -14,6 +14,7 @@ from llama_stack.apis.memory_banks import * # noqa: F403
from llama_stack.apis.datasets import * # noqa: F403
from llama_stack.distribution.datatypes import * # noqa: F403
import llama_stack.distribution.store as distribution_store
def get_impl_api(p: Any) -> Api:
@ -170,8 +171,7 @@ class CommonRoutingTableImpl(RoutingTable):
if obj.identifier not in self.registry:
self.registry[obj.identifier] = []
self.registry[obj.identifier].append(obj)
# TODO: persist this to a store
await distribution_store.REGISTRY.register(obj)
class ModelsRoutingTable(CommonRoutingTableImpl, Models):