phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

better detection for jupyter

Browse source
This commit is contained in:
Ashwin Bharambe 2024-12-09 09:37:52 -08:00
parent 5335393fe3
commit a2170353af
1 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

17
llama_stack/distribution/library_client.py
View file

@ -33,16 +33,17 @@ from llama_stack.distribution.stack import (
T = TypeVar("T")
def is_jupyter():
"""Check if we're running in a Jupyter notebook"""
def in_notebook():
try:
shell = get_ipython().__class__.__name__ # type: ignore
if shell == "ZMQInteractiveShell": # Jupyter notebook or qtconsole
return True
else:
from IPython import get_ipython
if "IPKernelApp" not in get_ipython().config: # pragma: no cover
return False
except NameError: # Probably standard Python interpreter
except ImportError:
return False
except AttributeError:
return False
return True
def stream_across_asyncio_run_boundary(
@ -115,7 +116,7 @@ class LlamaStackAsLibraryClient(LlamaStackClient):
self.pool_executor = ThreadPoolExecutor(max_workers=4)
def initialize(self):
if is_jupyter():
if in_notebook():
import nest_asyncio
nest_asyncio.apply()