phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 01:48:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix for the logger

Browse source
This commit is contained in:
r-bit-rry 2025-11-30 17:45:47 +02:00
parent c3c9edf981
commit dc4c7eaed7
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/llama_stack/providers/utils/inference/stream_utils.py
View file

@ -4,10 +4,11 @@
# This source code is licensed under the terms described in the LICENSE file in # This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree. # the root directory of this source tree.
import logging
from collections.abc import AsyncIterator 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]: 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: async for item in stream:
yield item yield item
except Exception as e: except Exception as e:
logger.error(f"Error in wrapped async stream: {e}") log.error(f"Error in wrapped async stream: {e}")
raise raise