src/agents/sqlspideragent.py

from autogen import ConversableAgent
import streamlit as st


class TrackableSpiderConversableAgent(ConversableAgent):
    def _process_received_message(self, message, sender, silent):
        with st.chat_message(sender.name):
            st.write(message)
        return super()._process_received_message(message, sender, silent)