From eef5b990420080f52df83f540ac21ba704de57a3 Mon Sep 17 00:00:00 2001
From: dhuynh95 <daniel.huynh@mithrilsecurity.io>
Date: Thu, 3 Oct 2024 08:49:29 +0200
Subject: [PATCH] Added info about Tabs in the ActionOutput

---
 lavague-sdk/lavague/sdk/action/__init__.py | 1 +
 lavague-sdk/lavague/sdk/action/base.py     | 6 +++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/lavague-sdk/lavague/sdk/action/__init__.py b/lavague-sdk/lavague/sdk/action/__init__.py
index 4ac94e06..35463369 100644
--- a/lavague-sdk/lavague/sdk/action/__init__.py
+++ b/lavague-sdk/lavague/sdk/action/__init__.py
@@ -5,6 +5,7 @@
     ActionParser,
     DEFAULT_PARSER,
     UnhandledTypeException,
+    Tab,
 )
 
 from lavague.sdk.action.navigation import WebNavigationAction
diff --git a/lavague-sdk/lavague/sdk/action/base.py b/lavague-sdk/lavague/sdk/action/base.py
index e348fe24..b0d57de8 100644
--- a/lavague-sdk/lavague/sdk/action/base.py
+++ b/lavague-sdk/lavague/sdk/action/base.py
@@ -13,6 +13,9 @@ class ActionType(str, Enum):
     NAVIGATION = "web_navigation"
     EXTRACTION = "web_extraction"
 
+class Tab(BaseModel):
+    url: str
+    title: str
 
 T = TypeVar("T")
 
@@ -23,7 +26,8 @@ class Action(BaseModel, Generic[T]):
     step_id: str = Field(default_factory=lambda: str(uuid4()))
     action_type: ActionType
     action_output: List[T]
-    url: str
+    tabs: List[Tab]
+    focused_tab: int
     status: ActionStatus
     instruction: str