From 2b176df6f9ef71a65dc2871e893c6bb066edc315 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Mon, 28 Oct 2024 13:11:58 -0700 Subject: [PATCH] Fix action history (#1076) --- skyvern/forge/agent.py | 23 +++++------------------ 1 file changed, 5 insertions(+), 18 deletions(-) diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py index a864f308..7f5528ae 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -1219,28 +1219,15 @@ class ForgeAgent: { "action": action.model_dump( exclude_none=True, - exclude={ - "text", - "confidence_float", - "organization_id", - "task_id", - "step_id", - "step_order", - "action_order", - "skyvern_element_hash", - "skyvern_element_data", - }, + include={"action_type", "element_id", "status", "reasoning", "option"}, ), "results": [ result.model_dump( exclude_none=True, - exclude={ - "javascript_triggered", - "interacted_with_sibling", - "interacted_with_parent", - "step_retry_number", - "step_order", - "stop_execution_on_failure", + include={ + "success", + "exception_type", + "exception_message", }, ) for result in results