fix: recover run agents after restart

2026-08-15 22:37:14 -05:00
parent 93d6172072
commit a5a3361e7f
11 changed files with 731 additions and 79 deletions
@@ -2008,21 +2008,25 @@ impl AIConversation {
            .sum()
    }

-    pub fn contains_action(&self, action_id: &AIAgentActionId) -> bool {
-        self.task_store.tasks().any(|task| {
-            task.exchanges()
-            .any(|exchange| {
-                let Some(output) = exchange.output_status.output()
-                else {
-                    return false;
-                };
-                output.get().messages.iter().any(|step| {
-                    matches!(step, AIAgentOutputMessage{ message: AIAgentOutputMessageType::Action(AIAgentAction { id, .. }), .. } if id == action_id)
+    pub fn action(&self, action_id: &AIAgentActionId) -> Option<AIAgentAction> {
+        self.task_store.tasks().find_map(|task| {
+            task.exchanges().find_map(|exchange| {
+                let output = exchange.output_status.output()?;
+                output.get().messages.iter().find_map(|step| match step {
+                    AIAgentOutputMessage {
+                        message: AIAgentOutputMessageType::Action(action),
+                        ..
+                    } if &action.id == action_id => Some(action.clone()),
+                    AIAgentOutputMessage { .. } => None,
                })
            })
        })
    }

+    pub fn contains_action(&self, action_id: &AIAgentActionId) -> bool {
+        self.action(action_id).is_some()
+    }
+
    /// Returns the exchange ID that contains the given action ID, if any.
    pub fn exchange_id_for_action(&self, action_id: &AIAgentActionId) -> Option<AIAgentExchangeId> {
        for task in self.task_store.tasks() {