From dc4c7eaed7de77aed79d72cfe6caf5cee22943e7 Mon Sep 17 00:00:00 2001 From: r-bit-rry Date: Sun, 30 Nov 2025 17:45:47 +0200 Subject: [PATCH] fix for the logger --- src/llama_stack/providers/utils/inference/stream_utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/llama_stack/providers/utils/inference/stream_utils.py b/src/llama_stack/providers/utils/inference/stream_utils.py index 3a9292893..a15e29d08 100644 --- a/src/llama_stack/providers/utils/inference/stream_utils.py +++ b/src/llama_stack/providers/utils/inference/stream_utils.py @@ -4,10 +4,11 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. -import logging from collections.abc import AsyncIterator -logger = logging.getLogger(__name__) +from llama_stack.log import get_logger + +log = get_logger(name=__name__, category="providers::utils") async def wrap_async_stream[T](stream: AsyncIterator[T]) -> AsyncIterator[T]: @@ -18,5 +19,5 @@ async def wrap_async_stream[T](stream: AsyncIterator[T]) -> AsyncIterator[T]: async for item in stream: yield item except Exception as e: - logger.error(f"Error in wrapped async stream: {e}") + log.error(f"Error in wrapped async stream: {e}") raise