Added info about Tabs in the ActionOutput

lavague-ai · Oct 3, 2024 · eef5b99 · eef5b99
1 parent 5b194b9
commit eef5b99
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git 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
@@ -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