Modified specs

lavague-ai · Sep 24, 2024 · c883193 · c883193
1 parent bc9a1d0
commit c883193
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lavague-core/lavague/trajectory/base.py b/lavague-core/lavague/trajectory/base.py
@@ -1,4 +1,4 @@
-from typing import Optional, List
+from typing import Optional, List, Tuple
 from pydantic import BaseModel, SerializeAsAny
 from lavague.action import Action, WebExtractionAction, WebNavigationAction
 from enum import Enum
@@ -14,9 +14,9 @@ class Trajectory(BaseModel):
 
     run_id: str
     start_url: str
+    viewport_size: Tuple[int, int]
     objective: str
     status: TrajectoryStatus
-    output: Optional[str]
     actions: List[SerializeAsAny[Action]]
 
     def write_to_file(self, file_path: str):