@puckeditor/plugin-ai 0.8.3-canary.c9f74a23 → 0.8.3-canary.ccfc525d

package/dist/index.js

@@ -4365,7 +4365,7 @@ var import_jsx_runtime35 = require("react/jsx-runtime");
 var q = (0, import_qler.default)();
 var DEFAULT_API_VERSION = "v2";
 var BUILD_OP_DEBOUNCE_MS = 1e3 / 60;
-var PLUGIN_AI_VERSION = true ? "0.8.3-canary.c9f74a23" : "unknown";
+var PLUGIN_AI_VERSION = true ? "0.8.3-canary.ccfc525d" : "unknown";
 var BENCHMARK = false;
 var prefixedUlid = (prefix = "") => `${prefix ? `${prefix}_` : ""}${(0, import_ulid.ulid)()}`;
 var getClassName18 = getClassNameFactory("Chat", styles_module_default);
@@ -4407,6 +4407,8 @@ function Chat2(props) {
   const removedAttachmentClientIdsRef = (0, import_react27.useRef)(/* @__PURE__ */ new Set());
   const uploadControllersRef = (0, import_react27.useRef)(/* @__PURE__ */ new Map());
   const pendingBuildOpsRef = (0, import_react27.useRef)([]);
+  const lastStreamActivityRef = (0, import_react27.useRef)(0);
+  const hiddenSinceRef = (0, import_react27.useRef)(null);
   const [toolStatus, setToolStatus] = (0, import_react27.useState)({});
   const [subagentState, setSubagentState] = (0, import_react27.useState)({});
   const attachmentConfig = (0, import_react27.useMemo)(
@@ -4519,6 +4521,7 @@ function Chat2(props) {
   );
   const processData = (0, import_react27.useCallback)(
     (dataPart) => {
+      lastStreamActivityRef.current = Date.now();
       switch (dataPart.type) {
         case "data-new-chat-created": {
           localChatId.current = dataPart.data.chatId;
@@ -4745,6 +4748,56 @@ function Chat2(props) {
     },
     [stop]
   );
+  const statusRef = (0, import_react27.useRef)(status);
+  (0, import_react27.useEffect)(() => {
+    statusRef.current = status;
+  }, [status]);
+  (0, import_react27.useEffect)(() => {
+    lastStreamActivityRef.current = Date.now();
+  }, [messages]);
+  (0, import_react27.useEffect)(() => {
+    const FROZEN_HIDDEN_MS = 3 * 6e4;
+    const RESUME_GRACE_MS = 12e3;
+    const isStreaming = () => statusRef.current === "streaming" || statusRef.current === "submitted";
+    let graceTimer;
+    const recoverIfSevered = () => {
+      if (!isStreaming()) return;
+      if (Date.now() - lastStreamActivityRef.current < RESUME_GRACE_MS) return;
+      stop();
+      setToolStatus((current) => {
+        let changed = false;
+        const next = {};
+        for (const [id, toolState] of Object.entries(current)) {
+          if (toolState.loading) {
+            changed = true;
+            next[id] = { ...toolState, loading: false, label: "Interrupted" };
+          } else {
+            next[id] = toolState;
+          }
+        }
+        return changed ? next : current;
+      });
+    };
+    const onVisibilityChange = () => {
+      if (document.hidden) {
+        hiddenSinceRef.current = isStreaming() ? Date.now() : null;
+        return;
+      }
+      const hiddenSince = hiddenSinceRef.current;
+      hiddenSinceRef.current = null;
+      if (graceTimer) clearTimeout(graceTimer);
+      if (hiddenSince === null || Date.now() - hiddenSince < FROZEN_HIDDEN_MS) {
+        return;
+      }
+      lastStreamActivityRef.current = 0;
+      graceTimer = setTimeout(recoverIfSevered, RESUME_GRACE_MS);
+    };
+    document.addEventListener("visibilitychange", onVisibilityChange);
+    return () => {
+      document.removeEventListener("visibilitychange", onVisibilityChange);
+      if (graceTimer) clearTimeout(graceTimer);
+    };
+  }, [stop]);
   const [forcedStatus, setForcedStatus] = (0, import_react27.useState)();
   const resolvedStatus = (0, import_react27.useMemo)(
     () => forcedStatus ?? status,

package/dist/index.mjs

@@ -4397,7 +4397,7 @@ import { Fragment as Fragment9, jsx as jsx35, jsxs as jsxs20 } from "react/jsx-r
 var q = qler();
 var DEFAULT_API_VERSION = "v2";
 var BUILD_OP_DEBOUNCE_MS = 1e3 / 60;
-var PLUGIN_AI_VERSION = true ? "0.8.3-canary.c9f74a23" : "unknown";
+var PLUGIN_AI_VERSION = true ? "0.8.3-canary.ccfc525d" : "unknown";
 var BENCHMARK = false;
 var prefixedUlid = (prefix = "") => `${prefix ? `${prefix}_` : ""}${ulid()}`;
 var getClassName18 = getClassNameFactory("Chat", styles_module_default);
@@ -4439,6 +4439,8 @@ function Chat2(props) {
   const removedAttachmentClientIdsRef = useRef13(/* @__PURE__ */ new Set());
   const uploadControllersRef = useRef13(/* @__PURE__ */ new Map());
   const pendingBuildOpsRef = useRef13([]);
+  const lastStreamActivityRef = useRef13(0);
+  const hiddenSinceRef = useRef13(null);
   const [toolStatus, setToolStatus] = useState16({});
   const [subagentState, setSubagentState] = useState16({});
   const attachmentConfig = useMemo6(
@@ -4551,6 +4553,7 @@ function Chat2(props) {
   );
   const processData = useCallback6(
     (dataPart) => {
+      lastStreamActivityRef.current = Date.now();
       switch (dataPart.type) {
         case "data-new-chat-created": {
           localChatId.current = dataPart.data.chatId;
@@ -4777,6 +4780,56 @@ function Chat2(props) {
     },
     [stop]
   );
+  const statusRef = useRef13(status);
+  useEffect13(() => {
+    statusRef.current = status;
+  }, [status]);
+  useEffect13(() => {
+    lastStreamActivityRef.current = Date.now();
+  }, [messages]);
+  useEffect13(() => {
+    const FROZEN_HIDDEN_MS = 3 * 6e4;
+    const RESUME_GRACE_MS = 12e3;
+    const isStreaming = () => statusRef.current === "streaming" || statusRef.current === "submitted";
+    let graceTimer;
+    const recoverIfSevered = () => {
+      if (!isStreaming()) return;
+      if (Date.now() - lastStreamActivityRef.current < RESUME_GRACE_MS) return;
+      stop();
+      setToolStatus((current) => {
+        let changed = false;
+        const next = {};
+        for (const [id, toolState] of Object.entries(current)) {
+          if (toolState.loading) {
+            changed = true;
+            next[id] = { ...toolState, loading: false, label: "Interrupted" };
+          } else {
+            next[id] = toolState;
+          }
+        }
+        return changed ? next : current;
+      });
+    };
+    const onVisibilityChange = () => {
+      if (document.hidden) {
+        hiddenSinceRef.current = isStreaming() ? Date.now() : null;
+        return;
+      }
+      const hiddenSince = hiddenSinceRef.current;
+      hiddenSinceRef.current = null;
+      if (graceTimer) clearTimeout(graceTimer);
+      if (hiddenSince === null || Date.now() - hiddenSince < FROZEN_HIDDEN_MS) {
+        return;
+      }
+      lastStreamActivityRef.current = 0;
+      graceTimer = setTimeout(recoverIfSevered, RESUME_GRACE_MS);
+    };
+    document.addEventListener("visibilitychange", onVisibilityChange);
+    return () => {
+      document.removeEventListener("visibilitychange", onVisibilityChange);
+      if (graceTimer) clearTimeout(graceTimer);
+    };
+  }, [stop]);
   const [forcedStatus, setForcedStatus] = useState16();
   const resolvedStatus = useMemo6(
     () => forcedStatus ?? status,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@puckeditor/plugin-ai",
-  "version": "0.8.3-canary.c9f74a23",
+  "version": "0.8.3-canary.ccfc525d",
   "author": "Chris Villa <chris@puckeditor.com>",
   "repository": {
     "type": "git",
@@ -52,12 +52,12 @@
     "tsup": "^8.2.4",
     "typescript": "^5.5.4",
     "vitest": "^4.1.0",
-    "@puckeditor/ai-types": "0.8.3-canary.c9f74a23",
-    "tsup-config": "0.8.3-canary.c9f74a23",
-    "eslint-config-custom": "0.8.3-canary.c9f74a23",
-    "@puckeditor/platform-types": "0.8.3-canary.c9f74a23",
-    "tsconfig": "0.8.3-canary.c9f74a23",
-    "reducer": "0.8.3-canary.c9f74a23"
+    "@puckeditor/ai-types": "0.8.3-canary.ccfc525d",
+    "@puckeditor/platform-types": "0.8.3-canary.ccfc525d",
+    "reducer": "0.8.3-canary.ccfc525d",
+    "tsconfig": "0.8.3-canary.ccfc525d",
+    "tsup-config": "0.8.3-canary.ccfc525d",
+    "eslint-config-custom": "0.8.3-canary.ccfc525d"
   },
   "peerDependencies": {
     "@puckeditor/core": "^0",