From c08b82b6ade9092eca1de93bc9c4ff0e80329dfb Mon Sep 17 00:00:00 2001
From: ramnique <30795890+ramnique@users.noreply.github.com>
Date: Tue, 28 Jan 2025 08:32:22 +0530
Subject: [PATCH] fix chat reset bug on workflow save

---
 .../projects/[projectId]/workflow/workflow_editor.tsx    | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/apps/rowboat/app/projects/[projectId]/workflow/workflow_editor.tsx b/apps/rowboat/app/projects/[projectId]/workflow/workflow_editor.tsx
index 97ff5bea..0e564638 100644
--- a/apps/rowboat/app/projects/[projectId]/workflow/workflow_editor.tsx
+++ b/apps/rowboat/app/projects/[projectId]/workflow/workflow_editor.tsx
@@ -39,6 +39,7 @@ interface StateItem {
     publishSuccess: boolean;
     pendingChanges: boolean;
     chatKey: number;
+    lastUpdatedAt: string;
 }
 
 interface State {
@@ -194,7 +195,7 @@ function reducer(state: State, action: Action): State {
             newState = produce(state, draft => {
                 draft.present.saving = action.saving;
                 draft.present.pendingChanges = action.saving;
-                draft.present.workflow.lastUpdatedAt = !action.saving ? new Date().toISOString() : state.present.workflow.lastUpdatedAt;
+                draft.present.lastUpdatedAt = !action.saving ? new Date().toISOString() : state.present.workflow.lastUpdatedAt;
             });
             break;
         }
@@ -257,6 +258,7 @@ function reducer(state: State, action: Action): State {
                                 name: action.agent.name || newAgentName
                             };
                             draft.pendingChanges = true;
+                            draft.chatKey++;
                             break;
                         }
                         case "add_tool": {
@@ -281,6 +283,7 @@ function reducer(state: State, action: Action): State {
                                 name: action.tool.name || newToolName
                             };
                             draft.pendingChanges = true;
+                            draft.chatKey++;
                             break;
                         }
                         case "add_prompt": {
@@ -303,6 +306,7 @@ function reducer(state: State, action: Action): State {
                                 name: action.prompt.name || newPromptName
                             };
                             draft.pendingChanges = true;
+                            draft.chatKey++;
                             break;
                         }
                         case "delete_agent":
@@ -493,6 +497,7 @@ export function WorkflowEditor({
             publishSuccess: false,
             pendingChanges: false,
             chatKey: 0,
+            lastUpdatedAt: workflow.lastUpdatedAt,
         }
     });
     const [chatMessages, setChatMessages] = useState<z.infer<typeof apiV1.ChatMessage>[]>([]);
@@ -726,7 +731,7 @@ export function WorkflowEditor({
                         <div>Saving...</div>
                     </div>}
                     {!state.present.saving && state.present.workflow && <div>
-                        Updated <RelativeTime date={new Date(state.present.workflow.lastUpdatedAt)} />
+                        Updated <RelativeTime date={new Date(state.present.lastUpdatedAt)} />
                     </div>}
                 </div>}
                 {!isLive && <>