@anuma/sdk 6.39.2 → 6.40.0

package/dist/server/index.mjs

@@ -19482,6 +19482,93 @@ function composeHooks(hooks) {
   };
 }
 
+// src/lib/chat/streamIdleTimeout.ts
+var StreamIdleTimeoutError = class extends Error {
+  constructor(idleTimeoutMs) {
+    super(`Stream timed out after ${idleTimeoutMs} ms without activity.`);
+    this.name = "StreamIdleTimeoutError";
+    this.idleTimeoutMs = idleTimeoutMs;
+  }
+};
+function withStreamIdleTimeout(transport, idleTimeoutMs) {
+  if (!(idleTimeoutMs > 0) || idleTimeoutMs === Infinity) return transport;
+  return (options) => ({
+    stream: (async function* () {
+      const controller = new AbortController();
+      let timer;
+      let settled = false;
+      let completed = false;
+      let iterator;
+      let interruption;
+      let rejectPending;
+      const disarm = () => {
+        if (timer !== void 0) clearTimeout(timer);
+        timer = void 0;
+      };
+      const interrupt = (error) => {
+        if (interruption) return;
+        interruption = error;
+        rejectPending?.(error);
+        controller.abort();
+      };
+      const arm = () => {
+        if (settled || controller.signal.aborted) return;
+        disarm();
+        timer = setTimeout(() => {
+          interrupt(new StreamIdleTimeoutError(idleTimeoutMs));
+        }, idleTimeoutMs);
+      };
+      const onAbort = () => {
+        controller.abort();
+      };
+      options.signal?.addEventListener("abort", onAbort, { once: true });
+      try {
+        if (options.signal?.aborted) {
+          const error = new Error("Request aborted");
+          error.name = "AbortError";
+          throw error;
+        }
+        arm();
+        iterator = transport({
+          ...options,
+          signal: controller.signal,
+          onActivity: () => {
+            arm();
+            options.onActivity?.();
+          }
+        }).stream[Symbol.asyncIterator]();
+        while (true) {
+          if (interruption) throw interruption;
+          const next = await new Promise((resolve, reject) => {
+            rejectPending = reject;
+            Promise.resolve(iterator.next()).then(resolve, reject);
+          });
+          rejectPending = void 0;
+          if (next.done) {
+            completed = true;
+            break;
+          }
+          arm();
+          yield next.value;
+        }
+      } finally {
+        settled = true;
+        rejectPending = void 0;
+        disarm();
+        options.signal?.removeEventListener("abort", onAbort);
+        if (!completed) {
+          controller.abort();
+          try {
+            void iterator?.return?.().catch(() => {
+            });
+          } catch {
+          }
+        }
+      }
+    })()
+  });
+}
+
 // src/lib/chat/useChat/utils.ts
 import { jsonrepair } from "jsonrepair";
 function parseToolArguments(raw) {
@@ -21645,7 +21732,20 @@ var defaultTransport = (options) => {
       } catch {
       }
     }
-    return response;
+    if (!response.body || !options.onActivity) return response;
+    const body = response.body.pipeThrough(
+      new TransformStream({
+        transform(chunk, controller) {
+          if (chunk.byteLength > 0) options.onActivity?.();
+          controller.enqueue(chunk);
+        }
+      })
+    );
+    return new Response(body, {
+      status: response.status,
+      statusText: response.statusText,
+      headers: response.headers
+    });
   };
   return createSseClient({
     method: options.method ?? "POST",
@@ -21702,6 +21802,7 @@ async function runToolLoop(options) {
     conversationId,
     smoothing,
     signal,
+    idleTimeoutMs = 12e4,
     resumable,
     detachSignal,
     onStreamMeta,
@@ -21715,13 +21816,14 @@ async function runToolLoop(options) {
     onStepFinish,
     onRequest,
     onStreamRetry,
-    transport: makeStreamingRequest = defaultTransport,
+    transport = defaultTransport,
     preProcessors,
     maxConnectorCalls = 2,
     hooks: hooksOption,
     piiRedaction,
     onPiiRedacted
   } = options;
+  const makeStreamingRequest = withStreamIdleTimeout(transport, idleTimeoutMs);
   const hooks = Array.isArray(hooksOption) ? composeHooks(hooksOption) : hooksOption;
   const runId = generateRunId();
   let messages = options.messages;
@@ -33302,6 +33404,14 @@ function createLocalMemoryStore(options) {
       getLogger().warn("[memory/store] Failed to embed memory:", err);
     });
   };
+  const maintenance = {
+    createDecaySweeper: (sweeperOptions) => createDecaySweeper({ ...sweeperOptions, vaultCtx }),
+    getTopicBacklog: (backlogOptions) => getMemoriesNeedingTopicExtractionOp(vaultCtx, backlogOptions),
+    extractTopics: (memoryIds, extractOptions) => extractAndLinkEntitiesForMemoriesOp(vaultCtx, memoryIds, extractOptions),
+    stampTopicsExtracted: (memoryIds, extractedAt, version) => stampTopicsExtractedAtOp(vaultCtx, memoryIds, extractedAt, version),
+    relinkTopics: (memoryIds) => relinkMemoryTopicsOp(vaultCtx, memoryIds),
+    backfillTopics: (memoryIds) => backfillMemoryTopicsOp(vaultCtx, memoryIds)
+  };
   return {
     masksEmbeddingInput: options.embeddingOptions.maskInput !== void 0,
     list: (listOptions) => getAllVaultMemoriesOp(vaultCtx, listOptions),
@@ -33315,21 +33425,22 @@ function createLocalMemoryStore(options) {
     memoriesByTopics: (names) => getMemoriesByEntityNamesOp(entityCtx, names),
     topicsByMemories: (memoryIds) => getEntitiesByMemoryIdsOp(entityCtx, memoryIds),
     listTopics: async () => {
-      const links = await entityCtx.memoryEntityCollection.query(...linksOwnedBy).fetch();
+      const links = await entityCtx.memoryEntityCollection.query(...linksOwnedBy).unsafeFetchRaw();
       const members = /* @__PURE__ */ new Map();
       for (const link of links) {
-        const entityId = String(link.entityId);
+        const entityId = String(link.entity_id);
         const set = members.get(entityId) ?? /* @__PURE__ */ new Set();
-        set.add(String(link.memoryId));
+        set.add(String(link.memory_id));
         members.set(entityId, set);
       }
-      const entities = userId === void 0 ? await entityCtx.entityCollection.query().fetch() : members.size ? await entityCtx.entityCollection.query(import_watermelondb24.Q.where("id", import_watermelondb24.Q.oneOf([...members.keys()]))).fetch() : [];
+      const entities = userId === void 0 ? await entityCtx.entityCollection.query().unsafeFetchRaw() : members.size ? await entityCtx.entityCollection.query(import_watermelondb24.Q.where("id", import_watermelondb24.Q.oneOf([...members.keys()]))).unsafeFetchRaw() : [];
       const byName = /* @__PURE__ */ new Map();
       for (const entity of entities) {
-        const count = members.get(entity.id)?.size ?? 0;
-        const existing = byName.get(entity.canonicalName);
-        byName.set(entity.canonicalName, {
-          name: entity.canonicalName,
+        const name = String(entity.canonical_name);
+        const count = members.get(String(entity.id))?.size ?? 0;
+        const existing = byName.get(name);
+        byName.set(name, {
+          name,
           kind: existing?.kind ?? entity.kind ?? null,
           memoryCount: (existing?.memoryCount ?? 0) + count
         });
@@ -33420,14 +33531,8 @@ function createLocalMemoryStore(options) {
       }
       return () => subscriptions.forEach((s) => s.unsubscribe());
     },
-    maintenance: {
-      createDecaySweeper: (sweeperOptions) => createDecaySweeper({ ...sweeperOptions, vaultCtx }),
-      getTopicBacklog: (backlogOptions) => getMemoriesNeedingTopicExtractionOp(vaultCtx, backlogOptions),
-      extractTopics: (memoryIds, extractOptions) => extractAndLinkEntitiesForMemoriesOp(vaultCtx, memoryIds, extractOptions),
-      stampTopicsExtracted: (memoryIds, extractedAt, version) => stampTopicsExtractedAtOp(vaultCtx, memoryIds, extractedAt, version),
-      relinkTopics: (memoryIds) => relinkMemoryTopicsOp(vaultCtx, memoryIds),
-      backfillTopics: (memoryIds) => backfillMemoryTopicsOp(vaultCtx, memoryIds)
-    }
+    maintenance,
+    topicMaintenance: maintenance
   };
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@anuma/sdk",
-  "version": "6.39.2",
+  "version": "6.40.0",
   "description": "",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",