create observer models with workflow_permanent_id

Skyvern-AI · Dec 16, 2024 · 82bbd42 · 82bbd42
1 parent 2b85605
commit 82bbd42
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py
@@ -1793,6 +1793,7 @@ async def create_observer_cruise(
         self,
         workflow_run_id: str | None = None,
         workflow_id: str | None = None,
+        workflow_permanent_id: str | None = None,
         prompt: str | None = None,
         url: str | None = None,
         organization_id: str | None = None,
@@ -1801,6 +1802,7 @@ async def create_observer_cruise(
             new_observer_cruise = ObserverCruiseModel(
                 workflow_run_id=workflow_run_id,
                 workflow_id=workflow_id,
+                workflow_permanent_id=workflow_permanent_id,
                 prompt=prompt,
                 url=url,
                 organization_id=organization_id,
@@ -1815,6 +1817,7 @@ async def create_observer_thought(
         observer_cruise_id: str,
         workflow_run_id: str | None = None,
         workflow_id: str | None = None,
+        workflow_permanent_id: str | None = None,
         workflow_run_block_id: str | None = None,
         user_input: str | None = None,
         observation: str | None = None,
@@ -1827,6 +1830,7 @@ async def create_observer_thought(
                 observer_cruise_id=observer_cruise_id,
                 workflow_run_id=workflow_run_id,
                 workflow_id=workflow_id,
+                workflow_permanent_id=workflow_permanent_id,
                 workflow_run_block_id=workflow_run_block_id,
                 user_input=user_input,
                 observation=observation,
@@ -1845,6 +1849,7 @@ async def update_observer_cruise(
         status: ObserverCruiseStatus | None = None,
         workflow_run_id: str | None = None,
         workflow_id: str | None = None,
+        workflow_permanent_id: str | None = None,
         url: str | None = None,
         prompt: str | None = None,
         organization_id: str | None = None,
@@ -1864,6 +1869,8 @@ async def update_observer_cruise(
                     observer_cruise.workflow_run_id = workflow_run_id
                 if workflow_id:
                     observer_cruise.workflow_id = workflow_id
+                if workflow_permanent_id:
+                    observer_cruise.workflow_permanent_id = workflow_permanent_id
                 if url:
                     observer_cruise.url = url
                 if prompt: