File size: 1,061 Bytes
171e985 4f52ef2 171e985 89b9dc1 171e985 b409e9d 171e985 b409e9d 171e985 b409e9d 171e985 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
import logging #### todo add transformer logs
import streamlit as st
# Configure the logging settings for transformers
transformers_logger = logging.getLogger("transformers.file_utils")
transformers_logger.setLevel(logging.INFO) # Set the desired logging level
log_enabled = False
##################################### logger #################################
def log_response(response):
if log_enabled:
with st.chat_message("ai"):
st.markdown("Agent Response\n {}".format(response))
print(response)
# Custom logging handler to append log messages to the chat
class ChatHandler(logging.Handler):
def __init__(self):
super().__init__()
def emit(self, record):
log_message = self.format(record)
#with st.chat_message("ai"):
st.markdown(f"Log: {log_message}")
with st.chat_message("ai"):
st.markdown("Agent Response\n {}".format(record))
# Add the custom handler to the transformers_logger
chat_handler = ChatHandler()
transformers_logger.addHandler(chat_handler) |