@anuma/sdk 6.30.1 → 6.31.0

package/dist/server/index.mjs

@@ -13,7 +13,7 @@ import {
   taskTypeHeader,
   validateEndpointOverride,
   withInternalFlowMarker
-} from "./chunk-Z2QBL4HB.mjs";
+} from "./chunk-KSYV7UO2.mjs";
 import {
   BASE_URL,
   __commonJS,
@@ -19108,6 +19108,9 @@ async function updateMessageEmbeddingOp(ctx, uniqueId, vector, embeddingModel) {
   await ctx.database.write(async () => {
     await message.update((msg) => {
       msg._setRaw("vector", JSON.stringify(vector));
+      if (msg.origin === "chunks_discarded" && vector.length > 0) {
+        msg._setRaw("origin", "message");
+      }
       msg._setRaw("embedding_model", embeddingModel);
     });
   });
@@ -19124,6 +19127,9 @@ async function updateMessageChunksOp(ctx, uniqueId, chunks, embeddingModel) {
   await ctx.database.write(async () => {
     await message.update((msg) => {
       msg._setRaw("chunks", JSON.stringify(storedChunks));
+      if (msg.origin === "chunks_discarded" && storedChunks.length > 0 && storedChunks.every((chunk) => chunk.vector.length > 0)) {
+        msg._setRaw("origin", "message");
+      }
       msg._setRaw("embedding_model", embeddingModel);
     });
   });
@@ -21153,7 +21159,7 @@ async function getMemoriesNeedingTopicExtractionOp(ctx, options) {
     const id = raw.id;
     const topics = parseTopics(raw.topics);
     const linked = linkedNames.get(id);
-    if (topics !== null && topics.length > 0 && linksDivergeFromTopics(topics, linked ?? /* @__PURE__ */ new Set())) {
+    if (topics !== null && linksDivergeFromTopics(topics, linked ?? /* @__PURE__ */ new Set())) {
       topicsToRelinkAll.push(id);
       continue;
     }
@@ -21247,18 +21253,20 @@ async function relinkMemoryTopicsOp(ctx, memoryIds) {
   }
   const relinked = [];
   for (const id of Array.from(new Set(memoryIds))) {
-    let record;
-    try {
-      record = await ctx.vaultMemoryCollection.find(id);
-    } catch {
-      continue;
-    }
-    if (record.isDeleted || !isOwnedByCtxUser(ctx, record)) continue;
-    const topics = parseTopics(record.topics);
-    if (topics === null || topics.length === 0) continue;
     try {
-      await relinkMemoryEntitiesFromTopicsOp(entityCtx, id, topics);
-      relinked.push(id);
+      await ctx.database.write(async (writer) => {
+        let record;
+        try {
+          record = await ctx.vaultMemoryCollection.find(id);
+        } catch {
+          return;
+        }
+        if (record.isDeleted || !isOwnedByCtxUser(ctx, record)) return;
+        const topics = parseTopics(record.topics);
+        if (topics === null) return;
+        await writer.callWriter(() => relinkMemoryEntitiesFromTopicsOp(entityCtx, id, topics));
+        relinked.push(id);
+      });
     } catch (err) {
       getLogger().warn("[memory/topics] relink failed", err);
     }
@@ -28020,7 +28028,7 @@ async function tryConsolidate(trimmed, ctx, options, prepared) {
     content: m.content,
     similarity: m.similarity
   }));
-  const { consolidateMemory: doConsolidate } = await import("./consolidate-QWVI25WQ.mjs");
+  const { consolidateMemory: doConsolidate } = await import("./consolidate-MCNBSUXD.mjs");
   const decision = await doConsolidate(trimmed, candidates, consolidateOptions);
   if (decision.action === "create") return decision.fallbackReason ? null : { create: true };
   if (decision.action === "supersede" && decision.content) {
@@ -30682,6 +30690,28 @@ var BatchFailureError = class extends Error {
     this.terminal = terminal;
   }
 };
+var FlaggedBatchError = class extends Error {
+};
+function extracted(result) {
+  return result.failedCount === 0 && result.outcome !== "empty-after-retry";
+}
+function batchError(result) {
+  const failure = result.failure;
+  if (failure?.reason === "http-terminal") {
+    const status = failure.httpStatus;
+    if (status !== void 0 && REQUEST_REJECTED_STATUSES.has(status))
+      return new BatchFailureError(
+        `Extraction rejected the batch (HTTP ${status}); not retried this session`,
+        true
+      );
+    return new AccountFailureError(
+      `Extraction refused for the account (HTTP ${status ?? "unknown"}); retried next session`
+    );
+  }
+  if (result.failedCount > 0 || failure !== void 0 && CONTENT_FAILURE_REASONS.has(failure.reason))
+    return new BatchFailureError("Extraction batch incomplete; retained for retry", false);
+  return new Error("Extraction batch incomplete; retained for retry");
+}
 var AccountFailureError = class extends Error {
 };
 var LockedSourcesError = class extends Error {
@@ -30690,6 +30720,16 @@ var LockedSourcesError = class extends Error {
     this.persistentIds = persistentIds;
   }
 };
+function splitTurns(messages) {
+  const turns = [];
+  for (const message of messages) {
+    const last = turns[turns.length - 1];
+    if (!last || message.role === "user" && last.some((m) => m.role === "user"))
+      turns.push([message]);
+    else last.push(message);
+  }
+  return turns;
+}
 function idsOf(job) {
   return JSON.parse(String(job._getRaw("message_ids")));
 }
@@ -30840,6 +30880,10 @@ function createDurableAutoExtractor(options) {
       skippedThisSession.add(job.id);
       return false;
     }
+    if (error instanceof FlaggedBatchError) {
+      await acknowledge(job, ids, loaded, true);
+      return true;
+    }
     const counted = error instanceof BatchFailureError || error instanceof LockedSourcesError && error.persistentIds.length > 0;
     if (!counted) return false;
     const prior = sessionFailures.get(job.id);
@@ -30966,37 +31010,50 @@ function createDurableAutoExtractor(options) {
           const messages = loaded.flatMap(
             (m) => m && m.conversationId === conversationId && (m.role === "user" || m.role === "assistant") ? [{ id: m.uniqueId, role: m.role, content: m.content }] : []
           );
-          let success = messages.length === 0;
-          let result;
-          if (messages.length) {
-            result = await extractBatch(
-              messages,
+          const jobScope = currentScope() === "private" ? "private" : String(job._getRaw("scope"));
+          const jobFolderId = job._getRaw("folder_id") ?? null;
+          const runExtraction = async (batch, batchIds) => {
+            const result = await extractBatch(
+              batch,
               conversationId,
-              ids,
-              currentScope() === "private" ? "private" : String(job._getRaw("scope")),
-              job._getRaw("folder_id") ?? null
+              batchIds,
+              jobScope,
+              jobFolderId
             );
-            success = result.failedCount === 0 && result.outcome !== "empty-after-retry";
             options.onTurnComplete?.(result);
-          }
-          if (!success) {
-            const failure = result?.failure;
-            if (failure?.reason === "http-terminal") {
-              const status = failure.httpStatus;
-              if (status !== void 0 && REQUEST_REJECTED_STATUSES.has(status))
-                throw new BatchFailureError(
-                  `Extraction rejected the batch (HTTP ${status}); not retried this session`,
-                  true
+            return result;
+          };
+          const flaggedIds = /* @__PURE__ */ new Set();
+          if (messages.length) {
+            const result = await runExtraction(messages, ids);
+            if (result.failure?.reason === "content-flagged") {
+              const turns = splitTurns(messages);
+              if (turns.length < 2)
+                throw new FlaggedBatchError("Extraction batch refused by moderation; abandoned");
+              for (const turn of turns) {
+                const turnResult = await runExtraction(
+                  turn,
+                  turn.map((m) => m.id)
                 );
-              throw new AccountFailureError(
-                `Extraction refused for the account (HTTP ${status ?? "unknown"}); retried next session`
-              );
-            }
-            if ((result?.failedCount ?? 0) > 0 || failure !== void 0 && CONTENT_FAILURE_REASONS.has(failure.reason))
-              throw new BatchFailureError("Extraction batch incomplete; retained for retry", false);
-            throw new Error("Extraction batch incomplete; retained for retry");
+                if (turnResult.failure?.reason === "content-flagged")
+                  for (const m of turn) flaggedIds.add(m.id);
+                else if (!extracted(turnResult)) throw batchError(turnResult);
+              }
+              if (flaggedIds.size)
+                reportError(
+                  new Error(
+                    `Dropped ${flaggedIds.size} source id(s) of an extraction batch refused by moderation`
+                  ),
+                  conversationId
+                );
+            } else if (!extracted(result)) throw batchError(result);
           }
-          await acknowledge(job, ids, loaded, false);
+          await acknowledge(
+            job,
+            ids,
+            loaded,
+            ids.slice(-2).some((id) => flaggedIds.has(id))
+          );
           scheduleAfterSuccess = true;
         } catch (error) {
           attempts.set(job.id, attempted + 1);
@@ -33871,6 +33928,7 @@ function toolsToApiFormat(tools, apiType) {
       executor: _executor,
       skipContinuation: _skipContinuation,
       removeAfterExecution: _removeAfterExecution,
+      removeAfterResult: _removeAfterResult,
       executorTimeout: _executorTimeout,
       dependsOn: _dependsOn,
       deAnonymizeArgs: _deAnonymizeArgs,
@@ -36315,9 +36373,11 @@ Executing tool: ${toolInfo}
       }
       if (tools && apiTools) {
         const successfullyExecutedNames = /* @__PURE__ */ new Set();
+        const successfulResults = [];
         for (const r of executionResults) {
           if (!r.error && !isToolErrorResult(r.result) && "name" in r && r.name) {
             successfullyExecutedNames.add(r.name);
+            successfulResults.push(r.result);
           }
         }
         if (successfullyExecutedNames.size > 0) {
@@ -36326,7 +36386,15 @@ Executing tool: ${toolInfo}
             const tc = t;
             const func = tc.function;
             const toolName2 = typeof func?.name === "string" ? func.name : typeof tc.name === "string" ? tc.name : void 0;
-            if (tc.removeAfterExecution === true && toolName2 && successfullyExecutedNames.has(toolName2)) {
+            if (!toolName2) continue;
+            const removeAfterResult = tc.removeAfterResult;
+            if (tc.removeAfterExecution === true && successfullyExecutedNames.has(toolName2) || typeof removeAfterResult === "function" && successfulResults.some((result) => {
+              try {
+                return removeAfterResult(result);
+              } catch {
+                return false;
+              }
+            })) {
               toolsToRemove.add(toolName2);
             }
           }

package/dist/tools/index.cjs

@@ -1170,9 +1170,25 @@ var consoleLogger = {
   warn: console.warn.bind(console),
   error: console.error.bind(console)
 };
-var currentLogger = consoleLogger;
+var LOGGER_STATE_KEY = /* @__PURE__ */ Symbol.for("@anuma/sdk/logger/v1");
+var loggerGlobal = globalThis;
+var fallbackLoggerState = { logger: consoleLogger };
+var fallbackWarned = false;
+function getLoggerState() {
+  try {
+    return loggerGlobal[LOGGER_STATE_KEY] ?? (loggerGlobal[LOGGER_STATE_KEY] = fallbackLoggerState);
+  } catch {
+    if (!fallbackWarned) {
+      fallbackWarned = true;
+      fallbackLoggerState.logger.warn(
+        "@anuma/sdk: the global object rejects new properties. This entrypoint keeps its own logger, so setLogger does not reach other entrypoints."
+      );
+    }
+    return fallbackLoggerState;
+  }
+}
 function getLogger() {
-  return currentLogger;
+  return getLoggerState().logger;
 }
 
 // src/tools/googleDrive.ts

package/dist/tools/index.d.mts

@@ -40,6 +40,21 @@ type ToolConfig = LlmapiChatCompletionTool & {
      * execution fails, allowing the model to retry.
      */
     removeAfterExecution?: boolean;
+    /**
+     * Remove this tool for the rest of the turn once ANY tool returns a
+     * successful result this predicate accepts — its own or another's. Errors
+     * never count, as for `removeAfterExecution`.
+     *
+     * Only results of tools executed on this client are offered; tools the
+     * portal runs server-side never reach the predicate. A predicate that throws
+     * is treated as not matching.
+     *
+     * For a family of tools that has to leave together: when one of them
+     * declines an out-of-scope request, the model otherwise works through its
+     * siblings round after round until the round cap. Give every member the same
+     * predicate and the first refusal removes the whole family.
+     */
+    removeAfterResult?: (result: unknown) => boolean;
     /**
      * Timeout in milliseconds for this tool's executor.
      * Overrides the default 30-second timeout.

package/dist/tools/index.d.ts

@@ -40,6 +40,21 @@ type ToolConfig = LlmapiChatCompletionTool & {
      * execution fails, allowing the model to retry.
      */
     removeAfterExecution?: boolean;
+    /**
+     * Remove this tool for the rest of the turn once ANY tool returns a
+     * successful result this predicate accepts — its own or another's. Errors
+     * never count, as for `removeAfterExecution`.
+     *
+     * Only results of tools executed on this client are offered; tools the
+     * portal runs server-side never reach the predicate. A predicate that throws
+     * is treated as not matching.
+     *
+     * For a family of tools that has to leave together: when one of them
+     * declines an out-of-scope request, the model otherwise works through its
+     * siblings round after round until the round cap. Give every member the same
+     * predicate and the first refusal removes the whole family.
+     */
+    removeAfterResult?: (result: unknown) => boolean;
     /**
      * Timeout in milliseconds for this tool's executor.
      * Overrides the default 30-second timeout.

package/dist/tools/index.mjs

@@ -1014,9 +1014,25 @@ var consoleLogger = {
   warn: console.warn.bind(console),
   error: console.error.bind(console)
 };
-var currentLogger = consoleLogger;
+var LOGGER_STATE_KEY = /* @__PURE__ */ Symbol.for("@anuma/sdk/logger/v1");
+var loggerGlobal = globalThis;
+var fallbackLoggerState = { logger: consoleLogger };
+var fallbackWarned = false;
+function getLoggerState() {
+  try {
+    return loggerGlobal[LOGGER_STATE_KEY] ?? (loggerGlobal[LOGGER_STATE_KEY] = fallbackLoggerState);
+  } catch {
+    if (!fallbackWarned) {
+      fallbackWarned = true;
+      fallbackLoggerState.logger.warn(
+        "@anuma/sdk: the global object rejects new properties. This entrypoint keeps its own logger, so setLogger does not reach other entrypoints."
+      );
+    }
+    return fallbackLoggerState;
+  }
+}
 function getLogger() {
-  return currentLogger;
+  return getLoggerState().logger;
 }
 
 // src/tools/googleDrive.ts

package/dist/tools/selection/index.cjs

@@ -1270,9 +1270,25 @@ var consoleLogger = {
   warn: console.warn.bind(console),
   error: console.error.bind(console)
 };
-var currentLogger = consoleLogger;
+var LOGGER_STATE_KEY = /* @__PURE__ */ Symbol.for("@anuma/sdk/logger/v1");
+var loggerGlobal = globalThis;
+var fallbackLoggerState = { logger: consoleLogger };
+var fallbackWarned = false;
+function getLoggerState() {
+  try {
+    return loggerGlobal[LOGGER_STATE_KEY] ?? (loggerGlobal[LOGGER_STATE_KEY] = fallbackLoggerState);
+  } catch {
+    if (!fallbackWarned) {
+      fallbackWarned = true;
+      fallbackLoggerState.logger.warn(
+        "@anuma/sdk: the global object rejects new properties. This entrypoint keeps its own logger, so setLogger does not reach other entrypoints."
+      );
+    }
+    return fallbackLoggerState;
+  }
+}
 function getLogger() {
-  return currentLogger;
+  return getLoggerState().logger;
 }
 
 // src/lib/memoryEngine/chunking.ts
@@ -1662,6 +1678,9 @@ function clientToolToResponsesFormat(tool) {
     ...toolConfig.removeAfterExecution !== void 0 && {
       removeAfterExecution: toolConfig.removeAfterExecution
     },
+    ...toolConfig.removeAfterResult !== void 0 && {
+      removeAfterResult: toolConfig.removeAfterResult
+    },
     ...toolConfig.executorTimeout !== void 0 && {
       executorTimeout: toolConfig.executorTimeout
     },
@@ -1695,6 +1714,9 @@ function clientToolToCompletionsFormat(tool) {
     ...toolConfig.removeAfterExecution !== void 0 && {
       removeAfterExecution: toolConfig.removeAfterExecution
     },
+    ...toolConfig.removeAfterResult !== void 0 && {
+      removeAfterResult: toolConfig.removeAfterResult
+    },
     ...toolConfig.executorTimeout !== void 0 && {
       executorTimeout: toolConfig.executorTimeout
     },

package/dist/tools/selection/index.d.mts

@@ -32,6 +32,21 @@ type ToolConfig = LlmapiChatCompletionTool & {
      * execution fails, allowing the model to retry.
      */
     removeAfterExecution?: boolean;
+    /**
+     * Remove this tool for the rest of the turn once ANY tool returns a
+     * successful result this predicate accepts — its own or another's. Errors
+     * never count, as for `removeAfterExecution`.
+     *
+     * Only results of tools executed on this client are offered; tools the
+     * portal runs server-side never reach the predicate. A predicate that throws
+     * is treated as not matching.
+     *
+     * For a family of tools that has to leave together: when one of them
+     * declines an out-of-scope request, the model otherwise works through its
+     * siblings round after round until the round cap. Give every member the same
+     * predicate and the first refusal removes the whole family.
+     */
+    removeAfterResult?: (result: unknown) => boolean;
     /**
      * Timeout in milliseconds for this tool's executor.
      * Overrides the default 30-second timeout.

package/dist/tools/selection/index.d.ts

@@ -32,6 +32,21 @@ type ToolConfig = LlmapiChatCompletionTool & {
      * execution fails, allowing the model to retry.
      */
     removeAfterExecution?: boolean;
+    /**
+     * Remove this tool for the rest of the turn once ANY tool returns a
+     * successful result this predicate accepts — its own or another's. Errors
+     * never count, as for `removeAfterExecution`.
+     *
+     * Only results of tools executed on this client are offered; tools the
+     * portal runs server-side never reach the predicate. A predicate that throws
+     * is treated as not matching.
+     *
+     * For a family of tools that has to leave together: when one of them
+     * declines an out-of-scope request, the model otherwise works through its
+     * siblings round after round until the round cap. Give every member the same
+     * predicate and the first refusal removes the whole family.
+     */
+    removeAfterResult?: (result: unknown) => boolean;
     /**
      * Timeout in milliseconds for this tool's executor.
      * Overrides the default 30-second timeout.

package/dist/tools/selection/index.mjs

@@ -1148,9 +1148,25 @@ var consoleLogger = {
   warn: console.warn.bind(console),
   error: console.error.bind(console)
 };
-var currentLogger = consoleLogger;
+var LOGGER_STATE_KEY = /* @__PURE__ */ Symbol.for("@anuma/sdk/logger/v1");
+var loggerGlobal = globalThis;
+var fallbackLoggerState = { logger: consoleLogger };
+var fallbackWarned = false;
+function getLoggerState() {
+  try {
+    return loggerGlobal[LOGGER_STATE_KEY] ?? (loggerGlobal[LOGGER_STATE_KEY] = fallbackLoggerState);
+  } catch {
+    if (!fallbackWarned) {
+      fallbackWarned = true;
+      fallbackLoggerState.logger.warn(
+        "@anuma/sdk: the global object rejects new properties. This entrypoint keeps its own logger, so setLogger does not reach other entrypoints."
+      );
+    }
+    return fallbackLoggerState;
+  }
+}
 function getLogger() {
-  return currentLogger;
+  return getLoggerState().logger;
 }
 
 // src/lib/memoryEngine/chunking.ts
@@ -1540,6 +1556,9 @@ function clientToolToResponsesFormat(tool) {
     ...toolConfig.removeAfterExecution !== void 0 && {
       removeAfterExecution: toolConfig.removeAfterExecution
     },
+    ...toolConfig.removeAfterResult !== void 0 && {
+      removeAfterResult: toolConfig.removeAfterResult
+    },
     ...toolConfig.executorTimeout !== void 0 && {
       executorTimeout: toolConfig.executorTimeout
     },
@@ -1573,6 +1592,9 @@ function clientToolToCompletionsFormat(tool) {
     ...toolConfig.removeAfterExecution !== void 0 && {
       removeAfterExecution: toolConfig.removeAfterExecution
     },
+    ...toolConfig.removeAfterResult !== void 0 && {
+      removeAfterResult: toolConfig.removeAfterResult
+    },
     ...toolConfig.executorTimeout !== void 0 && {
       executorTimeout: toolConfig.executorTimeout
     },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@anuma/sdk",
-  "version": "6.30.1",
+  "version": "6.31.0",
   "description": "",
   "main": "./dist/index.cjs",
   "module": "./dist/index.mjs",
@@ -154,7 +154,7 @@
     "check:excluded-suites": "node scripts/assert-excluded-suites-still-fail.mjs",
     "knip": "knip",
     "check:export-consumers": "node scripts/assert-export-consumers.mjs",
-    "check-exports": "publint && attw --pack --ignore-rules no-resolution false-cjs && node scripts/assert-no-client-self-import.mjs",
+    "check-exports": "publint && attw --pack --ignore-rules no-resolution false-cjs && node scripts/assert-no-client-self-import.mjs && node scripts/assert-shared-logger.mjs",
     "check": "pnpm run lint && pnpm run knip && pnpm run check:export-consumers && pnpm run format && pnpm run generate",
     "eval:extraction": "tsx test/memory/src/extraction/benchmark.test.ts",
     "eval:topic": "tsx test/memory/src/topic/benchmark.ts",