dns for browser session (#2972)

skyvern/forge/sdk/db/client.py

@@ -3011,6 +3011,8 @@ class AgentDB:
         self,
         browser_session_id: str,
         browser_address: str,
+        ip_address: str,
+        ecs_task_arn: str | None,
         organization_id: str | None = None,
     ) -> None:
         """Set the browser address for a persistent browser session."""
@@ -3026,6 +3028,8 @@ class AgentDB:
                 ).first()
                 if persistent_browser_session:
                     persistent_browser_session.browser_address = browser_address
+                    persistent_browser_session.ip_address = ip_address
+                    persistent_browser_session.ecs_task_arn = ecs_task_arn
                     # once the address is set, the session is started
                     persistent_browser_session.started_at = datetime.utcnow()
                     await session.commit()

skyvern/forge/sdk/db/models.py

@@ -696,6 +696,8 @@ class PersistentBrowserSessionModel(Base):
     browser_address = Column(String, nullable=True)
     status = Column(String, nullable=True, default="created")
     timeout_minutes = Column(Integer, nullable=True)
+    ip_address = Column(String, nullable=True)
+    ecs_task_arn = Column(String, nullable=True)
     started_at = Column(DateTime, nullable=True)
     completed_at = Column(DateTime, nullable=True)
     created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False, index=True)