@react-grab/ami 0.0.88 → 0.0.89

package/dist/client.js

@@ -17362,11 +17362,60 @@ var q = async (e) => {
   return { success: o2.success, patches: t2.patches, error: o2.error };
 };
 
+// ../utils/dist/client.js
+var CONNECTION_CHECK_TTL_MS = 5e3;
+var STORAGE_KEY = "react-grab:agent-sessions";
+var isRecord = (value) => typeof value === "object" && value !== null;
+var getStoredAgentContext = (storage, sessionId, storageKey = STORAGE_KEY) => {
+  const rawSessions = storage.getItem(storageKey);
+  if (!rawSessions) throw new Error("No sessions to resume");
+  let parsed;
+  try {
+    parsed = JSON.parse(rawSessions);
+  } catch {
+    throw new Error("Failed to parse stored sessions");
+  }
+  if (!isRecord(parsed)) throw new Error("Invalid stored sessions");
+  const storedSession = parsed[sessionId];
+  if (!isRecord(storedSession)) {
+    throw new Error(`Session ${sessionId} not found`);
+  }
+  const context = storedSession.context;
+  if (!isRecord(context)) throw new Error(`Session ${sessionId} is invalid`);
+  const content = context.content;
+  const prompt = context.prompt;
+  if (typeof content !== "string" || typeof prompt !== "string") {
+    throw new Error(`Session ${sessionId} is invalid`);
+  }
+  const options = context.options;
+  const storedSessionId = context.sessionId;
+  return {
+    content,
+    prompt,
+    options,
+    sessionId: typeof storedSessionId === "string" ? storedSessionId : void 0
+  };
+};
+var createCachedConnectionChecker = (checkConnection, ttlMs = CONNECTION_CHECK_TTL_MS) => {
+  let cache = null;
+  return async () => {
+    const now = Date.now();
+    if (cache && now - cache.timestamp < ttlMs) return cache.result;
+    try {
+      const result = await checkConnection();
+      cache = { result, timestamp: now };
+      return result;
+    } catch {
+      cache = { result: false, timestamp: now };
+      return false;
+    }
+  };
+};
+
 // src/constants.ts
 var COMPLETED_STATUS = "COMPLETED_STATUS";
 
 // src/client.ts
-var STORAGE_KEY = "react-grab:agent-sessions";
 var TOKEN_STORAGE_KEY = "react-grab:ami-token";
 var BRIDGE_TOKEN_STORAGE_KEY = "react-grab:ami-bridge-token";
 var loadCachedAuth = () => {
@@ -17491,9 +17540,8 @@ ${context.content}`;
       return { status: COMPLETED_STATUS, messages, chatId };
   }
 };
-var CONNECTION_CHECK_TTL_MS = 5e3;
+var isReactGrabApi = (value) => typeof value === "object" && value !== null && "setAgent" in value;
 var createAmiAgentProvider = (projectId) => {
-  let connectionCache = null;
   let lastAgentMessages = null;
   const sessionData = /* @__PURE__ */ new Map();
   const getLatestProjectId = async (token) => {
@@ -17615,16 +17663,10 @@ var createAmiAgentProvider = (projectId) => {
     },
     resume: async function* (sessionId, signal, storage) {
       const startTime = Date.now();
-      const savedSessions = storage.getItem(STORAGE_KEY);
-      if (!savedSessions) {
-        throw new Error("No sessions to resume");
-      }
-      const sessionsObject = JSON.parse(savedSessions);
-      const session = sessionsObject[sessionId];
-      if (!session) {
-        throw new Error(`Session ${sessionId} not found`);
-      }
-      const context = session.context;
+      const storedContext = getStoredAgentContext(storage, sessionId);
+      const context = {
+        content: storedContext.content,
+        prompt: storedContext.prompt};
       const auth = await getOrCreateAuth();
       yield "Resuming...";
       const statusQueue = [];
@@ -17722,21 +17764,10 @@ var createAmiAgentProvider = (projectId) => {
     },
     supportsResume: true,
     supportsFollowUp: true,
-    checkConnection: async () => {
-      const now = Date.now();
-      if (connectionCache && now - connectionCache.timestamp < CONNECTION_CHECK_TTL_MS) {
-        return connectionCache.result;
-      }
-      try {
-        const response = await fetch(l, { method: "HEAD" });
-        const result = response.ok;
-        connectionCache = { result, timestamp: now };
-        return result;
-      } catch {
-        connectionCache = { result: false, timestamp: now };
-        return false;
-      }
-    },
+    checkConnection: createCachedConnectionChecker(async () => {
+      const response = await fetch(l, { method: "HEAD" });
+      return response.ok;
+    }, CONNECTION_CHECK_TTL_MS),
     undo: async () => {
       if (!lastAgentMessages) return;
       try {
@@ -17753,24 +17784,25 @@ var createAmiAgentProvider = (projectId) => {
 var attachAgent = async () => {
   if (typeof window === "undefined") return;
   const provider = createAmiAgentProvider();
-  const attach = (api2) => {
-    api2.setAgent({ provider, storage: sessionStorage });
+  const attach = (api) => {
+    api.setAgent({ provider, storage: sessionStorage });
   };
-  const api = window.__REACT_GRAB__;
-  if (api) {
-    attach(api);
+  const existingApi = window.__REACT_GRAB__;
+  if (isReactGrabApi(existingApi)) {
+    attach(existingApi);
     return;
   }
   window.addEventListener(
     "react-grab:init",
     (event) => {
-      const customEvent = event;
-      attach(customEvent.detail);
+      if (!(event instanceof CustomEvent)) return;
+      if (!isReactGrabApi(event.detail)) return;
+      attach(event.detail);
     },
     { once: true }
   );
   const apiAfterListener = window.__REACT_GRAB__;
-  if (apiAfterListener) {
+  if (isReactGrabApi(apiAfterListener)) {
     attach(apiAfterListener);
   }
 };

package/dist/server.cjs

@@ -7,9 +7,11 @@ function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 var pc__default = /*#__PURE__*/_interopDefault(pc);
 
 // src/server.ts
-var VERSION = "0.0.88";
+var VERSION = "0.0.89";
 try {
-  fetch(`https://www.react-grab.com/api/version?source=ami&t=${Date.now()}`).catch(() => {
+  fetch(
+    `https://www.react-grab.com/api/version?source=ami&t=${Date.now()}`
+  ).catch(() => {
   });
 } catch {
 }

package/dist/server.js

@@ -1,9 +1,11 @@
 import pc from 'picocolors';
 
 // src/server.ts
-var VERSION = "0.0.88";
+var VERSION = "0.0.89";
 try {
-  fetch(`https://www.react-grab.com/api/version?source=ami&t=${Date.now()}`).catch(() => {
+  fetch(
+    `https://www.react-grab.com/api/version?source=ami&t=${Date.now()}`
+  ).catch(() => {
   });
 } catch {
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@react-grab/ami",
-  "version": "0.0.88",
+  "version": "0.0.89",
   "type": "module",
   "bin": {
     "react-grab-ami": "./dist/cli.cjs"
@@ -22,12 +22,13 @@
   ],
   "devDependencies": {
     "@types/node": "^22.10.7",
-    "tsup": "^8.4.0"
+    "tsup": "^8.4.0",
+    "@react-grab/utils": "0.0.89"
   },
   "dependencies": {
     "ami-sdk": "^0.0.6",
     "picocolors": "^1.1.1",
-    "react-grab": "0.0.88"
+    "react-grab": "0.0.89"
   },
   "scripts": {
     "dev": "tsup --watch",