Spaces:
Sleeping
Sleeping
File size: 1,013 Bytes
10231de 40fd4e5 10231de 3328745 10231de 40fd4e5 ac85c1d 40fd4e5 |
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 |
from autogen import AssistantAgent
import streamlit as st
from autogen.agentchat.contrib.llamaindex_conversable_agent import LLamaIndexConversableAgent
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 TrackableLLamaIndexConversableAgent(LLamaIndexConversableAgent):
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)
else:
with st.chat_message("ai"):
st.write(message['content'])
return super()._process_received_message(message, sender, silent)
|