diff --git a/llama_stack/distribution/routers/routing_tables.py b/llama_stack/distribution/routers/routing_tables.py index 1aba4884b..7b369df2c 100644 --- a/llama_stack/distribution/routers/routing_tables.py +++ b/llama_stack/distribution/routers/routing_tables.py @@ -32,6 +32,10 @@ async def register_object_with_provider(obj: RoutableObject, p: Any) -> None: api = get_impl_api(p) if obj.provider_id == "remote": + # TODO: this is broken right now because we use the generic + # { identifier, provider_id, provider_resource_id } tuple here + # but the APIs expect things like ModelInput, ShieldInput, etc. + # if this is just a passthrough, we want to let the remote # end actually do the registration with the correct provider obj = obj.model_copy(deep=True) diff --git a/llama_stack/providers/inline/safety/llama_guard/llama_guard.py b/llama_stack/providers/inline/safety/llama_guard/llama_guard.py index 9c3ec7750..12d012b16 100644 --- a/llama_stack/providers/inline/safety/llama_guard/llama_guard.py +++ b/llama_stack/providers/inline/safety/llama_guard/llama_guard.py @@ -128,7 +128,6 @@ class LlamaGuardSafetyImpl(Safety, ShieldsProtocolPrivate): pass async def register_shield(self, shield: Shield) -> None: - print(f"Registering shield {shield}") if shield.shield_type != ShieldType.llama_guard: raise ValueError(f"Unsupported shield type: {shield.shield_type}")