diff --git a/components/agents/AgentGenerationWorkspaceDarkV2.tsx b/components/agents/AgentGenerationWorkspaceDarkV2.tsx
index 152997579..ec6f64890 100644
--- a/components/agents/AgentGenerationWorkspaceDarkV2.tsx
+++ b/components/agents/AgentGenerationWorkspaceDarkV2.tsx
@@ -4191,9 +4191,9 @@ export function AgentGenerationWorkspaceDarkV2({
         void refreshHistory();
       }
     },
-    onFinish({ messages: finishedMessages }) {
+    onFinish({ messages: finishedMessages, isAbort, isDisconnect, isError }) {
       const lastUser = finishedMessages.filter((message) => message.role === "user").at(-1);
-      setFinishedSceneRequestId(lastUser?.id || null);
+      if (!isAbort && !isDisconnect && !isError) setFinishedSceneRequestId(lastUser?.id || null);
       void refreshHistory();
     },
     experimental_throttle: 100,