Pedro/migrate context call to main prompt (#3400)

skyvern/forge/sdk/core/skyvern_context.py

@@ -31,6 +31,7 @@ class SkyvernContext:
     script_revision_id: str | None = None
     action_order: int = 0
     prompt: str | None = None
+    enable_parse_select_in_extract: bool = False
 
     def __repr__(self) -> str:
         return f"SkyvernContext(request_id={self.request_id}, organization_id={self.organization_id}, task_id={self.task_id}, step_id={self.step_id}, workflow_id={self.workflow_id}, workflow_run_id={self.workflow_run_id}, task_v2_id={self.task_v2_id}, max_steps_override={self.max_steps_override}, run_id={self.run_id})"

skyvern/forge/sdk/db/models.py

@@ -580,6 +580,7 @@ class ActionModel(Base):
     skyvern_element_hash = Column(String, nullable=True)
     skyvern_element_data = Column(JSON, nullable=True)
     action_json = Column(JSON, nullable=True)
+    input_or_select_context = Column(JSON, nullable=True)
     confidence_float = Column(Numeric, nullable=True)
 
     created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)