@amemhq/core 2.1.1 → 2.1.2

package/dist/cli-migrate.cjs

@@ -37,6 +37,15 @@ __export(cli_migrate_exports, {
 });
 module.exports = __toCommonJS(cli_migrate_exports);
 
+// src/config.ts
+var os = __toESM(require("os"), 1);
+var path = __toESM(require("path"), 1);
+var _dataDir = process.env.AMEM_DATA_DIR || path.join(os.homedir(), ".amem");
+var _warn = (msg) => console.warn(msg);
+function warn(msg) {
+  _warn(msg);
+}
+
 // src/embedding.ts
 var pipeline = null;
 var extractor = null;
@@ -81,6 +90,12 @@ function applyModelPaths(env) {
     env.allowLocalModels = true;
   }
 }
+function humanBytes(n) {
+  if (n >= 1e9) return `${(n / 1e9).toFixed(2)} GB`;
+  if (n >= 1e6) return `${(n / 1e6).toFixed(1)} MB`;
+  if (n >= 1e3) return `${Math.round(n / 1e3)} kB`;
+  return `${n} B`;
+}
 function makeProgressReporter() {
   const lastPct = /* @__PURE__ */ new Map();
   return (e) => {
@@ -89,7 +104,7 @@ function makeProgressReporter() {
     const pct = Math.floor(e.progress / 10) * 10;
     if (lastPct.get(e.file) === pct) return;
     lastPct.set(e.file, pct);
-    const size = e.total ? ` of ${(e.total / 1e9).toFixed(2)} GB` : "";
+    const size = e.total ? ` of ${humanBytes(e.total)}` : "";
     console.log(`[amem] downloading ${e.file}: ${pct}%${size}`);
   };
 }
@@ -615,7 +630,7 @@ var _warned = /* @__PURE__ */ new Set();
 function warnOnce(key, message) {
   if (_warned.has(key)) return;
   _warned.add(key);
-  console.error(message);
+  warn(message);
 }
 function resolveProvider(role = "fast") {
   const raw = role === "strong" ? process.env.AMEM_LLM_STRONG_PROVIDER || _override.strong?.provider || void 0 : void 0;
@@ -679,9 +694,11 @@ async function llmCall(prompt, maxTokens = 500, role = "fast") {
   const isThinking = model.includes("gemini") || model.includes("pro-agent");
   const effectiveMaxTokens = isThinking ? Math.max(maxTokens * 8, 4e3) : maxTokens;
   try {
-    return provider === "openai" ? await openaiCall(prompt, model, effectiveMaxTokens, baseURL) : await anthropicCall(prompt, model, effectiveMaxTokens, baseURL);
+    const text = provider === "openai" ? await openaiCall(prompt, model, effectiveMaxTokens, baseURL) : await anthropicCall(prompt, model, effectiveMaxTokens, baseURL);
+    if (text === null) warn(`[amem] ${provider} answered with no text (model ${model})`);
+    return text;
   } catch (e) {
-    console.error(`[amem] LLM call failed: ${e.message}`);
+    warn(`[amem] LLM call failed: ${e.message}`);
     return null;
   }
 }
@@ -780,7 +797,8 @@ confidence guide (Story 27):
 
 Text: ${content}`;
   const raw = await llmCall(prompt, 400);
-  if (!raw)
+  if (!raw) {
+    warn("[amem] note construction got nothing back; storing with no keywords, tags or context");
     return {
       keywords: [],
       tags: [],
@@ -790,6 +808,7 @@ Text: ${content}`;
       topics: [],
       confidence: "medium"
     };
+  }
   try {
     const data = parseJsonLoose(raw);
     const rawCategory = typeof data.category === "string" ? data.category : "General";
@@ -808,7 +827,7 @@ Text: ${content}`;
       confidence
     };
   } catch (e) {
-    console.error(`[amem] Note construction parse failed: ${e.message}`);
+    warn(`[amem] Note construction parse failed: ${e.message}`);
     return {
       keywords: [],
       tags: [],
@@ -823,13 +842,6 @@ Text: ${content}`;
 
 // src/memory.ts
 var import_uuid = require("uuid");
-
-// src/config.ts
-var os = __toESM(require("os"), 1);
-var path = __toESM(require("path"), 1);
-var _dataDir = process.env.AMEM_DATA_DIR || path.join(os.homedir(), ".amem");
-
-// src/memory.ts
 var import_jieba = require("@node-rs/jieba");
 function buildEmbedText(note) {
   let text = note.content;
@@ -848,7 +860,7 @@ async function migrateCollection(opts) {
   const refreshFields = opts.refreshFields !== false;
   const dryRun = opts.dryRun !== false;
   const log = opts.logger?.info ?? ((m) => console.log(m));
-  const warn = opts.logger?.warn ?? ((m) => console.warn(m));
+  const warn2 = opts.logger?.warn ?? warn;
   if (from === to) throw new Error(`migrate: source and target are the same collection ("${from}")`);
   const model = getEmbeddingModel();
   const targetDim = await getEmbeddingDim();
@@ -916,7 +928,7 @@ async function migrateCollection(opts) {
         if (!note.context) note.context = built.context;
         refreshed++;
       } catch (e) {
-        warn(`[migrate] re-extract failed for ${note.id.slice(0, 8)} \u2014 keeping as-is: ${e.message}`);
+        warn2(`[migrate] re-extract failed for ${note.id.slice(0, 8)} \u2014 keeping as-is: ${e.message}`);
       }
     }
     const point = noteToPoint({ ...note, embedding: await encode(buildEmbedText(note)) });
@@ -929,7 +941,7 @@ async function migrateCollection(opts) {
   await flush();
   const finalCount = await countPointsRaw(to);
   if (finalCount !== notes.length) {
-    warn(`[migrate] target holds ${finalCount} point(s) but the source had ${notes.length} \u2014 check before switching`);
+    warn2(`[migrate] target holds ${finalCount} point(s) but the source had ${notes.length} \u2014 check before switching`);
   }
   log(`[migrate] done: ${migrated} written, ${alreadyDone.size} already present, ${refreshed} re-extracted.`);
   return {