Spaces:
Sleeping
Sleeping
File size: 719 Bytes
9f396ec 6617b51 9f396ec 6617b51 9f396ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from autogen import AssistantAgent
import streamlit as st
import base64
from io import BytesIO
class TrackableAssistantAgent(AssistantAgent):
def _process_received_message(self, message, sender, silent):
if message and type(message)== str and sender.name =="Userproxy":
with st.chat_message("user"):
st.write(message)
return super()._process_received_message(message, sender, silent)
class TrackableImageAssistantAgent(AssistantAgent):
def _process_received_message(self, message, sender, silent):
# with st.chat_message('ai'):
# st.image('./imagegen/response.jpeg')
return super()._process_received_message(message, sender, silent)
|