From 82bbd42ceae6504d13d693fe608c082d4cfe15d6 Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Mon, 16 Dec 2024 14:31:52 -0800
Subject: [PATCH] create observer models with workflow_permanent_id

---
 skyvern/forge/sdk/db/client.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py
index dc474fc254..db10d70ac9 100644
--- 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: