Skip to content

Commit 5d966e2

Browse files
committed
Get Python chatbot log level from env var
1 parent 7cc0914 commit 5d966e2

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

e2e_tests/python/main.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import boto3
33
import json
44
import logging
5+
import os
56
from typing import Any
67

78
from chat_session import ChatSession
@@ -11,7 +12,8 @@
1112

1213
# Configure logging
1314
logging.basicConfig(
14-
level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
15+
level=getattr(logging, os.getenv("LOG_LEVEL", "INFO").upper()),
16+
format="%(asctime)s - %(levelname)s - %(message)s",
1517
)
1618
logging.getLogger("aiobotocore").setLevel(logging.CRITICAL)
1719
logging.getLogger("boto3").setLevel(logging.CRITICAL)

examples/chatbots/python/main.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import boto3
33
import json
44
import logging
5+
import os
56
from typing import Any
67

78
from chat_session import ChatSession
@@ -11,7 +12,8 @@
1112

1213
# Configure logging
1314
logging.basicConfig(
14-
level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
15+
level=getattr(logging, os.getenv("LOG_LEVEL", "INFO").upper()),
16+
format="%(asctime)s - %(levelname)s - %(message)s",
1517
)
1618
logging.getLogger("aiobotocore").setLevel(logging.CRITICAL)
1719
logging.getLogger("boto3").setLevel(logging.CRITICAL)

0 commit comments

Comments
 (0)