phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 19:04:19 +00:00
Code Issues Projects Releases Packages Wiki 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 sys
import traceback
import warnings
from contextlib import asynccontextmanager
from ssl import SSLError
@ -45,6 +46,16 @@ from llama_stack.distribution.stack import construct_stack
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:
if isinstance(data, BaseModel):
data = data.model_dump_json()