From a2982b3310db065ee5b271f7b04adf7f00e98d0b Mon Sep 17 00:00:00 2001 From: Nathan Weinberg Date: Mon, 17 Mar 2025 13:32:47 -0400 Subject: [PATCH] fix: add shutdown method for ProviderImpl Signed-off-by: Nathan Weinberg --- llama_stack/distribution/providers.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/llama_stack/distribution/providers.py b/llama_stack/distribution/providers.py index fb2476767..cf9b0b975 100644 --- a/llama_stack/distribution/providers.py +++ b/llama_stack/distribution/providers.py @@ -8,10 +8,13 @@ from pydantic import BaseModel from llama_stack.apis.providers import ListProvidersResponse, ProviderInfo, Providers +from llama_stack.log import get_logger from .datatypes import StackRunConfig from .stack import redact_sensitive_fields +logger = get_logger(name=__name__, category="core") + class ProviderImplConfig(BaseModel): run_config: StackRunConfig @@ -31,6 +34,10 @@ class ProviderImpl(Providers): async def initialize(self) -> None: pass + async def shutdown(self) -> None: + logger.debug("ProviderImpl.shutdown") + pass + async def list_providers(self) -> ListProvidersResponse: run_config = self.config.run_config safe_config = StackRunConfig(**redact_sensitive_fields(run_config.model_dump()))