Update logger.py
Browse files
logger.py
CHANGED
@@ -7,7 +7,9 @@ transformers_logger.setLevel(logging.INFO) # Set the desired logging level
|
|
7 |
log_enabled = False
|
8 |
|
9 |
|
|
|
10 |
|
|
|
11 |
|
12 |
|
13 |
def log_response(response):
|
@@ -26,6 +28,8 @@ class ChatHandler(logging.Handler):
|
|
26 |
log_message = self.format(record)
|
27 |
#with st.chat_message("ai"):
|
28 |
st.markdown(f"Log: {log_message}")
|
|
|
|
|
29 |
|
30 |
# Add the custom handler to the transformers_logger
|
31 |
chat_handler = ChatHandler()
|
|
|
7 |
log_enabled = False
|
8 |
|
9 |
|
10 |
+
##################################### logger #################################
|
11 |
|
12 |
+
|
13 |
|
14 |
|
15 |
def log_response(response):
|
|
|
28 |
log_message = self.format(record)
|
29 |
#with st.chat_message("ai"):
|
30 |
st.markdown(f"Log: {log_message}")
|
31 |
+
with st.chat_message("ai"):
|
32 |
+
st.markdown("Agent Response\n {}".format(record))
|
33 |
|
34 |
# Add the custom handler to the transformers_logger
|
35 |
chat_handler = ChatHandler()
|