From 92f72b27a5a9df83c8eea222c3b288ba4bd710ed Mon Sep 17 00:00:00 2001
From: Kerem Yilmaz <kerem@skyvern.com>
Date: Wed, 5 Jun 2024 15:41:08 -0700
Subject: [PATCH] fix element_id type for action and web action + add unit
 tests for actions (#424)

---
 skyvern/webeye/actions/actions.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/skyvern/webeye/actions/actions.py b/skyvern/webeye/actions/actions.py
index 119fe0f9..d1aadd7c 100644
--- a/skyvern/webeye/actions/actions.py
+++ b/skyvern/webeye/actions/actions.py
@@ -1,5 +1,5 @@
 from enum import StrEnum
-from typing import Any, Dict
+from typing import Annotated, Any, Dict
 
 import structlog
 from deprecation import deprecated
@@ -47,7 +47,7 @@ class Action(BaseModel):
     action_type: ActionType
     description: str | None = None
     reasoning: str | None = None
-    element_id: int | str | None = None
+    element_id: Annotated[str, Field(coerce_numbers_to_str=True)] | None = None
 
     # DecisiveAction (CompleteAction, TerminateAction) fields
     errors: list[UserDefinedError] | None = None
@@ -64,7 +64,7 @@ class Action(BaseModel):
 
 
 class WebAction(Action):
-    element_id: int | str
+    element_id: Annotated[str, Field(coerce_numbers_to_str=True)]
 
 
 class DecisiveAction(Action):