From c1ad0ed391c4af7af06560c8672cecca8002d1db Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 24 Oct 2024 13:11:15 -0700 Subject: [PATCH] get rid of org id, task id, step id, step order and action order from the action history (#1050) --- skyvern/forge/agent.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py index fd2d3db7..94665177 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -1217,7 +1217,18 @@ class ForgeAgent: return json.dumps( [ { - "action": action.model_dump(exclude_none=True, exclude={"text", "confidence_float"}), + "action": action.model_dump( + exclude_none=True, + exclude={ + "text", + "confidence_float", + "organization_id", + "task_id", + "step_id", + "step_order", + "action_order", + }, + ), "results": [ result.model_dump( exclude_none=True,