fix cheshire-cat-ai#879; more flexible

core/cat/agents/__init__.py

@@ -0,0 +1 @@
+from .base_agent import BaseAgent, AgentOutput

core/cat/agents/main_agent.py

@@ -51,8 +51,10 @@ async def execute(self, stray) -> AgentOutput:
         fast_reply = self.mad_hatter.execute_hook(
             "agent_fast_reply", fast_reply, cat=stray
         )
-        if len(fast_reply.keys()) > 0:
+        if isinstance(fast_reply, AgentOutput):
             return fast_reply
+        if isinstance(fast_reply, dict) and "output" in fast_reply:
+            return AgentOutput(**fast_reply)
 
         # obtain prompt parts from plugins
         prompt_prefix = self.mad_hatter.execute_hook(

core/cat/looking_glass/stray_cat.py

@@ -429,7 +429,9 @@ async def __call__(self, message_dict):
 
         # why this response?
         why = self.__build_why()
+        # TODO: should these assignations be included in self.__build_why ?
         why.intermediate_steps = agent_output.intermediate_steps
+        why.agent_output = agent_output.model_dump()
 
         # prepare final cat message
         final_output = CatMessage(