diff --git a/llama_stack/providers/utils/inference/openai_mixin.py b/llama_stack/providers/utils/inference/openai_mixin.py index acca73800..9137013ee 100644 --- a/llama_stack/providers/utils/inference/openai_mixin.py +++ b/llama_stack/providers/utils/inference/openai_mixin.py @@ -132,7 +132,10 @@ class OpenAIMixin(NeedsRequestProviderData, ABC, BaseModel): :return: An iterable of model IDs or None if not implemented """ - return [m.id async for m in self.client.models.list()] + client = self.client + async with client: + model_ids = [m.id async for m in client.models.list()] + return model_ids async def initialize(self) -> None: """