fix element_id type for action and web action + add unit tests for actions (#424)

This commit is contained in:
Kerem Yilmaz
2024-06-05 15:41:08 -07:00
committed by GitHub
parent d89c948b76
commit 92f72b27a5

View File

@@ -1,5 +1,5 @@
from enum import StrEnum from enum import StrEnum
from typing import Any, Dict from typing import Annotated, Any, Dict
import structlog import structlog
from deprecation import deprecated from deprecation import deprecated
@@ -47,7 +47,7 @@ class Action(BaseModel):
action_type: ActionType action_type: ActionType
description: str | None = None description: str | None = None
reasoning: 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 # DecisiveAction (CompleteAction, TerminateAction) fields
errors: list[UserDefinedError] | None = None errors: list[UserDefinedError] | None = None
@@ -64,7 +64,7 @@ class Action(BaseModel):
class WebAction(Action): class WebAction(Action):
element_id: int | str element_id: Annotated[str, Field(coerce_numbers_to_str=True)]
class DecisiveAction(Action): class DecisiveAction(Action):