From 772e23e29e0501f16a22457a3a0d0a646b05aefe Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Fri, 8 Nov 2024 14:15:31 -0800 Subject: [PATCH] register singature fix --- llama_stack/distribution/routers/routing_tables.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py index 7c231deb5..bd34ad7af 100644 --- a/llama_stack/distribution/routers/routing_tables.py +++ b/llama_stack/distribution/routers/routing_tables.py @@ -209,7 +209,7 @@ class ModelsRoutingTable(CommonRoutingTableImpl, Models): provider_id: Optional[str] = None, llama_model: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None, - ) -> None: + ) -> Model: if provider_model_id is None: provider_model_id = model_id if provider_id is None: @@ -232,6 +232,7 @@ class ModelsRoutingTable(CommonRoutingTableImpl, Models): metadata=metadata, ) await self.register_object(model) + return model class ShieldsRoutingTable(CommonRoutingTableImpl, Shields):