From 16d103842aa3e4946aec602874f16711fe101d43 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Tue, 10 Dec 2024 08:47:32 -0800 Subject: [PATCH] Revert "await end_trace in libcli" This reverts commit 7615da78b8a60c908584acfc305428d737c000e0. --- 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 9265bb560..45382c417 100644 --- a/llama_stack/distribution/library_client.py +++ b/llama_stack/distribution/library_client.py @@ -295,7 +295,7 @@ class AsyncLlamaStackAsLibraryClient(AsyncLlamaStackClient): body = self._convert_body(path, body) return convert_pydantic_to_json_value(await func(**body), cast_to) finally: - await end_trace() + end_trace() async def _call_streaming(self, path: str, body: dict = None, cast_to: Any = None): await start_trace(path, {"__location__": "library_client"}) @@ -308,7 +308,7 @@ class AsyncLlamaStackAsLibraryClient(AsyncLlamaStackClient): async for chunk in await func(**body): yield convert_pydantic_to_json_value(chunk, cast_to) finally: - await end_trace() + end_trace() def _convert_body(self, path: str, body: Optional[dict] = None) -> dict: if not body: