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

must override get_provider_impl also

Browse source
This commit is contained in:
Ashwin Bharambe 2025-07-22 10:13:27 -07:00
parent 50d16dc707
commit 38a9c119df
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -38,6 +38,10 @@ class ModelsRoutingTable(CommonRoutingTableImpl, Models):
async def get_model(self, model_id: str) -> Model: async def get_model(self, model_id: str) -> Model:
return await lookup_model(self, model_id) return await lookup_model(self, model_id)
async def get_provider_impl(self, model_id: str) -> Any:
model = await lookup_model(self, model_id)
return self.impls_by_provider_id[model.provider_id]
async def register_model( async def register_model(
self, self,
model_id: str, model_id: str,