From 2d1cebad88a95e7bc1576779ef1f7da4a078ca8e Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 13 Mar 2025 17:14:45 -0700 Subject: [PATCH] fix --- llama_stack/distribution/providers.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/llama_stack/distribution/providers.py b/llama_stack/distribution/providers.py index 82b86be1d..ecfc3184a 100644 --- a/llama_stack/distribution/providers.py +++ b/llama_stack/distribution/providers.py @@ -4,14 +4,11 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. +from typing import Optional + from pydantic import BaseModel -from llama_stack.apis.providers import ( - GetProviderResponse, - ListProvidersResponse, - ProviderInfo, - Providers, -) +from llama_stack.apis.providers import ListProvidersResponse, ProviderInfo, Providers from .datatypes import StackRunConfig from .stack import redact_sensitive_fields @@ -52,7 +49,7 @@ class ProviderImpl(Providers): return ListProvidersResponse(data=ret) - async def inspect_provider(self, provider_id: str) -> GetProviderResponse: + async def inspect_provider(self, provider_id: str) -> Optional[ProviderInfo]: run_config = self.config.run_config safe_config = StackRunConfig(**redact_sensitive_fields(run_config.model_dump())) for api, providers in safe_config.providers.items():