Skip to content

Commit 6e4cb10

Browse files
backward compatibility
1 parent 941cb0e commit 6e4cb10

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

lagent/agents/agent.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)