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

correct event loop management

Browse source
This commit is contained in:
Xi Yan 2025-01-14 10:39:44 -08:00
parent ba85634758
commit ee2d4a0b89
1 changed files with 5 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/distribution/library_client.py
View file

@ -144,15 +144,12 @@ class LlamaStackAsLibraryClient(LlamaStackClient):
print(f"Removed handler {handler.__class__.__name__} from root logger")
def request(self, *args, **kwargs):
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
async_response = loop.run_until_complete(
self.async_client.request(*args, **kwargs)
)
async_response = asyncio.run(self.async_client.request(*args, **kwargs))
if kwargs.get("stream"):
# NOTE: We are using AsyncLlamaStackClient under the hood
# We need to convert the AsyncStream from async client into
# SyncStream return type for streaming
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
def sync_generator():
try:
while True:
@ -165,7 +162,6 @@ class LlamaStackAsLibraryClient(LlamaStackClient):
return sync_generator()
else:
loop.close()
return async_response