error handling

microsoft · Dec 12, 2024 · 73249cf · 73249cf
1 parent 07a5603
commit 73249cf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ufo/agents/processors/follower_agent_processor.py b/ufo/agents/processors/follower_agent_processor.py
@@ -28,6 +28,8 @@ def __init__(self, agent: "FollowerAgent", context: Context) -> None:
         super().__init__(agent, context)
         self.subtask = self.context.get(ContextNames.REQUEST)
 
+    @AppAgentProcessor.exception_capture
+    @AppAgentProcessor.method_timer
     def get_prompt_message(self) -> None:
         """
         Get the prompt message for the AppAgent in the follower mode. It may accept additional prompts as input.