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

fix test

Browse source
This commit is contained in:
Ashwin Bharambe 2025-03-02 18:49:18 -08:00
parent 82d9d43b5d
commit 027f0862fd
2 changed files with 11 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/logcat.py
View file

@ -161,7 +161,10 @@ def init(default_level: int = logging.INFO) -> None:
def _should_log(level: int, category: str) -> bool:
category_level = _category_levels.get(category.lower(), _default_level)
category = category.lower()
if category not in _category_levels:
return False
category_level = _category_levels[category]
return level >= category_level

22
tests/test_logcat.py
View file

@ -16,15 +16,8 @@ class TestLogcat(unittest.TestCase):
def setUp(self):
self.original_env = os.environ.get("LLAMA_STACK_LOGGING")
self.original_handlers = list(logcat._logger.handlers)
self.original_level = logcat._logger.level
self.log_output = io.StringIO()
self.handler = logging.StreamHandler(self.log_output)
self.handler.setFormatter(logging.Formatter("[%(category)s] %(message)s"))
logcat._logger.handlers.clear()
logcat._logger.addHandler(self.handler)
self._init_logcat()
def tearDown(self):
if self.original_env is not None:
@ -32,12 +25,12 @@ class TestLogcat(unittest.TestCase):
else:
os.environ.pop("LLAMA_STACK_LOGGING", None)
def _init_logcat(self):
logcat.init(default_level=logging.DEBUG)
self.handler = logging.StreamHandler(self.log_output)
self.handler.setFormatter(logging.Formatter("[%(category)s] %(message)s"))
logcat._logger.handlers.clear()
for handler in self.original_handlers:
logcat._logger.addHandler(handler)
logcat._logger.setLevel(self.original_level)
logcat._initialize()
logcat._logger.addHandler(self.handler)
def test_basic_logging(self):
logcat.info("server", "Info message")
@ -62,8 +55,7 @@ class TestLogcat(unittest.TestCase):
def test_env_var_control(self):
os.environ["LLAMA_STACK_LOGGING"] = "server=debug;inference=warning"
logcat._initialize()
self._init_logcat()
# These should be visible based on the environment settings
logcat.debug("server", "Server debug message")