From 7615da78b8a60c908584acfc305428d737c000e0 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Mon, 9 Dec 2024 15:54:42 -0800 Subject: [PATCH] await end_trace in libcli --- llama_stack/distribution/library_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/distribution/library_client.py b/llama_stack/distribution/library_client.py index 3a87f0c97..08c8e2b5d 100644 --- a/llama_stack/distribution/library_client.py +++ b/llama_stack/distribution/library_client.py @@ -253,7 +253,7 @@ class AsyncLlamaStackAsLibraryClient(AsyncLlamaStackClient): body = self._convert_body(path, body) return await func(**body) finally: - end_trace() + await end_trace() async def _call_streaming(self, path: str, method: str, body: dict = None): await start_trace(path, {"__location__": "library_client"}) @@ -266,7 +266,7 @@ class AsyncLlamaStackAsLibraryClient(AsyncLlamaStackClient): async for chunk in await func(**body): yield chunk finally: - end_trace() + await end_trace() def _convert_body(self, path: str, body: Optional[dict] = None) -> dict: if not body: