diff --git a/src/llama_stack/core/resolver.py b/src/llama_stack/core/resolver.py index 15720df95..dbab81f23 100644 --- a/src/llama_stack/core/resolver.py +++ b/src/llama_stack/core/resolver.py @@ -374,6 +374,9 @@ async def instantiate_provider( method = "get_adapter_impl" args = [config, deps] + if "policy" in inspect.signature(getattr(module, method)).parameters: + args.append(policy) + elif isinstance(provider_spec, AutoRoutedProviderSpec): method = "get_auto_router_impl"