Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit 0b4f159

Browse files
committed
feat(log): Add env-based LOG_LEVEL config to CLI
Branch: GraniteCodeSupport Signed-off-by: Gabe Goodhart <[email protected]>
1 parent 38a649a commit 0b4f159

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

torchchat/cli/cli.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,15 @@
1717
allowable_params_table,
1818
)
1919

20-
logging.basicConfig(level=logging.INFO, format="%(message)s")
20+
_log_level_env = os.getenv("LOG_LEVEL", "INFO")
21+
try:
22+
_log_level = getattr(logging, _log_level_env.upper())
23+
except AttributeError:
24+
print(f"Invalid log level: {_log_level_env}", file=sys.stderr)
25+
_log_level = logging.INFO
26+
27+
28+
logging.basicConfig(level=_log_level, format="%(message)s")
2129
logger = logging.getLogger(__name__)
2230

2331
default_device = os.getenv("TORCHCHAT_DEVICE", "fast")

0 commit comments

Comments
 (0)