phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

allow seeing warnings with traces optionally

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-13 12:27:19 -08:00
parent 96e7ef646f
commit 7f6ac2fbd7
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
llama_stack/distribution/server/server.py
View file

@ -13,6 +13,7 @@ import re
import signal import signal
import sys import sys
import traceback import traceback
import warnings
from contextlib import asynccontextmanager from contextlib import asynccontextmanager
from ssl import SSLError from ssl import SSLError
@ -45,6 +46,16 @@ from llama_stack.distribution.stack import construct_stack
from .endpoints import get_all_api_endpoints from .endpoints import get_all_api_endpoints
def warn_with_traceback(message, category, filename, lineno, file=None, line=None):
log = file if hasattr(file, "write") else sys.stderr
traceback.print_stack(file=log)
log.write(warnings.formatwarning(message, category, filename, lineno, line))
if os.environ.get("LLAMA_STACK_TRACE_WARNINGS"):
warnings.showwarning = warn_with_traceback
def create_sse_event(data: Any) -> str: def create_sse_event(data: Any) -> str:
if isinstance(data, BaseModel): if isinstance(data, BaseModel):
data = data.model_dump_json() data = data.model_dump_json()