@@ -73,7 +73,10 @@ def __call__(self, *message: AgentMessage, session_id=0, **kwargs) -> AgentMessa
7373 self .update_memory (message , session_id = session_id )
7474 response_message = self .forward (* message , session_id = session_id , ** kwargs )
7575 if not isinstance (response_message , AgentMessage ):
76- response_message = AgentMessage .from_model_response (response_message , self .name )
76+ if isinstance (response_message , str ):
77+ response_message = AgentMessage (sender = self .name , content = response_message )
78+ else :
79+ response_message = AgentMessage .from_model_response (response_message , self .name )
7780 self .update_memory (response_message , session_id = session_id )
7881 response_message = copy .deepcopy (response_message )
7982 for hook in self ._hooks .values ():
@@ -208,7 +211,10 @@ async def __call__(self, *message: AgentMessage, session_id=0, **kwargs) -> Agen
208211 self .update_memory (message , session_id = session_id )
209212 response_message = await self .forward (* message , session_id = session_id , ** kwargs )
210213 if not isinstance (response_message , AgentMessage ):
211- response_message = AgentMessage .from_model_response (response_message , self .name )
214+ if isinstance (response_message , str ):
215+ response_message = AgentMessage (sender = self .name , content = response_message )
216+ else :
217+ response_message = AgentMessage .from_model_response (response_message , self .name )
212218 self .update_memory (response_message , session_id = session_id )
213219 response_message = copy .deepcopy (response_message )
214220 for hook in self ._hooks .values ():
0 commit comments