From eda1197a92a3c88e8538d9476d186a505720ffe3 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Wed, 12 Mar 2025 13:00:14 -0700 Subject: [PATCH] fix: fix build error in context.py --- llama_stack/distribution/utils/context.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/distribution/utils/context.py b/llama_stack/distribution/utils/context.py index 107ce7127..2f32afba2 100644 --- a/llama_stack/distribution/utils/context.py +++ b/llama_stack/distribution/utils/context.py @@ -19,7 +19,7 @@ def preserve_contexts_async_generator( and we need to preserve the context across the event loop boundary. """ - async def wrapper(): + async def wrapper() -> AsyncGenerator[T, None]: while True: try: item = await gen.__anext__()