From 38a9c119df73369d251c5cf50eda6cb4eda82f61 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Tue, 22 Jul 2025 10:13:27 -0700
Subject: [PATCH] must override get_provider_impl also

---
 llama_stack/distribution/routing_tables/models.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/llama_stack/distribution/routing_tables/models.py b/llama_stack/distribution/routing_tables/models.py
index 2f3ce8193..f2787b308 100644
--- a/llama_stack/distribution/routing_tables/models.py
+++ b/llama_stack/distribution/routing_tables/models.py
@@ -38,6 +38,10 @@ class ModelsRoutingTable(CommonRoutingTableImpl, Models):
     async def get_model(self, model_id: str) -> Model:
         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(
         self,
         model_id: str,