@smyslenny/agent-memory 5.0.1 → 5.0.2

package/dist/index.d.ts

@@ -330,6 +330,16 @@ declare function fusionScore(input: {
     recency_boost?: number;
 }): number;
 declare function fuseHybridResults(lexical: SearchResult[], vector: VectorSearchResult[], limit: number, recency_boost?: number): HybridRecallResult[];
+interface RelatedLink {
+    memory: Memory;
+    sourceId: string;
+    weight: number;
+}
+/**
+ * Fetch related memories from the links table for a set of source memory IDs.
+ * Shared by both recall and surface expansion paths.
+ */
+declare function fetchRelatedLinks(db: Database.Database, sourceIds: string[], agentId: string, excludeIds: Set<string>, maxPerSource?: number): RelatedLink[];
 declare function recallMemories(db: Database.Database, query: string, opts?: RecallOptions): Promise<HybridRecallResponse>;
 declare function reindexEmbeddings(db: Database.Database, opts?: ReindexOptions): Promise<ReindexEmbeddingsResult>;
 declare function reindexMemorySearch(db: Database.Database, opts?: ReindexOptions): Promise<ReindexSearchResult>;
@@ -410,6 +420,8 @@ interface SurfaceResult {
     lexical_rank?: number;
     semantic_rank?: number;
     semantic_similarity?: number;
+    related_source_id?: string;
+    match_type?: "direct" | "related";
 }
 interface SurfaceResponse {
     count: number;
@@ -831,4 +843,4 @@ declare function formatNarrativeBoot(layers: {
  */
 declare function boot(db: Database.Database, opts?: WarmBootOptions): WarmBootResult;
 
-export { type AgentMemoryHttpServer, type ReflectInput as AppReflectInput, type ReflectProgressEvent as AppReflectProgressEvent, type AutoIngestWatcher, type AutoIngestWatcherOptions, type BootResult, type ConflictInfo, type ConflictType, type CreateMemoryInput, type DedupScoreBreakdown, type EmbeddingProvider, type EmbeddingProviderConfig, type EmbeddingProviderKind, type EmbeddingProviderOptions, type EmbeddingStatus, type EvictionCandidate, type ExportResult, type FeedbackEventInput, type FeedbackEventRecord, type FeedbackSource, type FeedbackSummary, type GovernResult, type GuardAction, type GuardInput, type GuardResult, type HttpJobStatus, type HttpServerOptions, type HybridRecallResponse, type HybridRecallResult, type IngestExtractedItem, type IngestResult, type IngestRunOptions, type MaintenanceJob, type MaintenancePhase, type MaintenanceStatus, type Memory, type MemoryType, type MergeContext, type MergePlan, type Path, type PendingEmbeddingRecord, type Priority, type RecallInput, type ReflectCheckpoint, type ReflectOptions, type ReflectProgressEvent$1 as ReflectProgressEvent, type ReflectRunResult, type ReflectRunners, type ReflectStats, type ReflectStep, type ReindexEmbeddingsResult, type ReindexInput, type ReindexProgressEvent, type ReindexSearchResult, type RememberInput, type SearchResult, type StatusResult, type StoredEmbedding, type SurfaceInput, type SurfaceIntent, type SurfaceResponse, type SurfaceResult, type SyncInput, type SyncResult, type TidyResult, type UpdateMemoryInput, type VectorSearchResult, type WarmBootOptions, type WarmBootResult, boot, buildFtsQuery, buildMergePlan, calculateVitality, classifyIngestType, completeMaintenanceJob, computeEvictionScore, contentHash, cosineSimilarity, countMemories, createEmbeddingProvider, createHttpServer, createInitialCheckpoint, createLocalHttpEmbeddingProvider, createMaintenanceJob, createMemory, createOpenAICompatibleEmbeddingProvider, createPath, decodeVector, deleteMemory, deletePath, encodeVector, exportMemories, extractIngestItems, failMaintenanceJob, findResumableMaintenanceJob, formatNarrativeBoot, formatRelativeDate, fuseHybridResults, fusionScore, getConfiguredEmbeddingProviderId, getDecayedMemories, getEmbedding, getEmbeddingProvider, getEmbeddingProviderConfigFromEnv, getEmbeddingProviderFromEnv, getFeedbackScore, getFeedbackSummary, getMaintenanceJob, getMemory, getMemoryStatus, getPath, getPathByUri, getPathsByDomain, getPathsByMemory, getPathsByPrefix, guard, healthcheckEmbeddingProvider, ingestText, isStaleContent, listMemories, listPendingEmbeddings, loadWarmBootLayers, markAllEmbeddingsPending, markEmbeddingFailed, markMemoryEmbeddingPending, parseUri, priorityPrior, rankEvictionCandidates, rebuildBm25Index, recallMemories, recallMemory, recordAccess, recordFeedbackEvent, recordPassiveFeedback, reflectMemories, reindexEmbeddings, reindexMemories, reindexMemorySearch, rememberMemory, runAutoIngestWatcher, runDecay, runGovern, runReflectOrchestrator, runTidy, searchBM25, searchByVector, slugify, splitIngestBlocks, startHttpServer, surfaceMemories, syncBatch, syncOne, tokenize, updateMaintenanceCheckpoint, updateMemory, upsertReadyEmbedding };
+export { type AgentMemoryHttpServer, type ReflectInput as AppReflectInput, type ReflectProgressEvent as AppReflectProgressEvent, type AutoIngestWatcher, type AutoIngestWatcherOptions, type BootResult, type ConflictInfo, type ConflictType, type CreateMemoryInput, type DedupScoreBreakdown, type EmbeddingProvider, type EmbeddingProviderConfig, type EmbeddingProviderKind, type EmbeddingProviderOptions, type EmbeddingStatus, type EvictionCandidate, type ExportResult, type FeedbackEventInput, type FeedbackEventRecord, type FeedbackSource, type FeedbackSummary, type GovernResult, type GuardAction, type GuardInput, type GuardResult, type HttpJobStatus, type HttpServerOptions, type HybridRecallResponse, type HybridRecallResult, type IngestExtractedItem, type IngestResult, type IngestRunOptions, type MaintenanceJob, type MaintenancePhase, type MaintenanceStatus, type Memory, type MemoryType, type MergeContext, type MergePlan, type Path, type PendingEmbeddingRecord, type Priority, type RecallInput, type ReflectCheckpoint, type ReflectOptions, type ReflectProgressEvent$1 as ReflectProgressEvent, type ReflectRunResult, type ReflectRunners, type ReflectStats, type ReflectStep, type ReindexEmbeddingsResult, type ReindexInput, type ReindexProgressEvent, type ReindexSearchResult, type RelatedLink, type RememberInput, type SearchResult, type StatusResult, type StoredEmbedding, type SurfaceInput, type SurfaceIntent, type SurfaceResponse, type SurfaceResult, type SyncInput, type SyncResult, type TidyResult, type UpdateMemoryInput, type VectorSearchResult, type WarmBootOptions, type WarmBootResult, boot, buildFtsQuery, buildMergePlan, calculateVitality, classifyIngestType, completeMaintenanceJob, computeEvictionScore, contentHash, cosineSimilarity, countMemories, createEmbeddingProvider, createHttpServer, createInitialCheckpoint, createLocalHttpEmbeddingProvider, createMaintenanceJob, createMemory, createOpenAICompatibleEmbeddingProvider, createPath, decodeVector, deleteMemory, deletePath, encodeVector, exportMemories, extractIngestItems, failMaintenanceJob, fetchRelatedLinks, findResumableMaintenanceJob, formatNarrativeBoot, formatRelativeDate, fuseHybridResults, fusionScore, getConfiguredEmbeddingProviderId, getDecayedMemories, getEmbedding, getEmbeddingProvider, getEmbeddingProviderConfigFromEnv, getEmbeddingProviderFromEnv, getFeedbackScore, getFeedbackSummary, getMaintenanceJob, getMemory, getMemoryStatus, getPath, getPathByUri, getPathsByDomain, getPathsByMemory, getPathsByPrefix, guard, healthcheckEmbeddingProvider, ingestText, isStaleContent, listMemories, listPendingEmbeddings, loadWarmBootLayers, markAllEmbeddingsPending, markEmbeddingFailed, markMemoryEmbeddingPending, parseUri, priorityPrior, rankEvictionCandidates, rebuildBm25Index, recallMemories, recallMemory, recordAccess, recordFeedbackEvent, recordPassiveFeedback, reflectMemories, reindexEmbeddings, reindexMemories, reindexMemorySearch, rememberMemory, runAutoIngestWatcher, runDecay, runGovern, runReflectOrchestrator, runTidy, searchBM25, searchByVector, slugify, splitIngestBlocks, startHttpServer, surfaceMemories, syncBatch, syncOne, tokenize, updateMaintenanceCheckpoint, updateMemory, upsertReadyEmbedding };

package/dist/index.js

@@ -1,4 +1,4 @@
-// AgentMemory v2 — Sleep-cycle memory for AI agents
+// AgentMemory — Sleep-cycle memory for AI agents
 
 // src/core/memory.ts
 import { createHash as createHash2 } from "crypto";
@@ -1433,21 +1433,20 @@ async function searchVectorBranch(db, query, opts) {
     before: opts.before
   });
 }
-function expandRelated(db, results, agentId, maxTotal) {
-  const existingIds = new Set(results.map((r) => r.memory.id));
+function fetchRelatedLinks(db, sourceIds, agentId, excludeIds, maxPerSource = 5) {
   const related = [];
-  for (const result of results) {
+  for (const sourceId of sourceIds) {
     const links = db.prepare(
       `SELECT l.target_id, l.weight, m.*
        FROM links l
        JOIN memories m ON m.id = l.target_id
        WHERE l.agent_id = ? AND l.source_id = ?
        ORDER BY l.weight DESC
-       LIMIT 5`
-    ).all(agentId, result.memory.id);
+       LIMIT ?`
+    ).all(agentId, sourceId, maxPerSource);
     for (const link of links) {
-      if (existingIds.has(link.target_id)) continue;
-      existingIds.add(link.target_id);
+      if (excludeIds.has(link.target_id)) continue;
+      excludeIds.add(link.target_id);
       const relatedMemory = {
         id: link.id,
         content: link.content,
@@ -1470,12 +1469,30 @@ function expandRelated(db, results, agentId, maxTotal) {
       };
       related.push({
         memory: relatedMemory,
-        score: result.score * link.weight * 0.6,
-        related_source_id: result.memory.id,
-        match_type: "related"
+        sourceId,
+        weight: link.weight
       });
     }
   }
+  return related;
+}
+function expandRelated(db, results, agentId, maxTotal) {
+  const existingIds = new Set(results.map((r) => r.memory.id));
+  const links = fetchRelatedLinks(
+    db,
+    results.map((r) => r.memory.id),
+    agentId,
+    existingIds
+  );
+  const related = links.map((link) => {
+    const sourceResult = results.find((r) => r.memory.id === link.sourceId);
+    return {
+      memory: link.memory,
+      score: (sourceResult?.score ?? 0) * link.weight * 0.6,
+      related_source_id: link.sourceId,
+      match_type: "related"
+    };
+  });
   const directResults = results.map((r) => ({
     ...r,
     match_type: "direct"
@@ -2501,7 +2518,8 @@ async function surfaceMemories(db, input) {
       }),
       lexical_rank: signal.queryRank ?? signal.recentRank ?? signal.taskRank,
       semantic_rank: signal.semanticRank,
-      semantic_similarity: signal.semanticSimilarity
+      semantic_similarity: signal.semanticSimilarity,
+      match_type: "direct"
     };
   }).sort((left, right) => {
     if (right.score !== left.score) return right.score - left.score;
@@ -2510,6 +2528,42 @@ async function surfaceMemories(db, input) {
     if (left.memory.priority !== right.memory.priority) return left.memory.priority - right.memory.priority;
     return right.memory.updated_at.localeCompare(left.memory.updated_at);
   }).slice(0, limit);
+  if (input.related) {
+    const existingIds = new Set(results.map((r) => r.memory.id));
+    const links = fetchRelatedLinks(
+      db,
+      results.map((r) => r.memory.id),
+      agentId,
+      existingIds
+    );
+    const relatedResults = links.map((link) => {
+      const sourceResult = results.find((r) => r.memory.id === link.sourceId);
+      const feedbackSummary = getFeedbackSummary(db, link.memory.id, agentId);
+      return {
+        memory: link.memory,
+        score: (sourceResult?.score ?? 0) * link.weight * 0.6,
+        semantic_score: 0,
+        lexical_score: 0,
+        task_match: 0,
+        vitality: link.memory.vitality,
+        priority_prior: priorityPrior(link.memory.priority),
+        feedback_score: feedbackSummary.score,
+        feedback_summary: feedbackSummary,
+        reason_codes: [`type:${link.memory.type}`, "related"],
+        related_source_id: link.sourceId,
+        match_type: "related"
+      };
+    });
+    const maxTotal = Math.floor(limit * 1.5);
+    const combined = [...results, ...relatedResults].sort((a, b) => b.score - a.score).slice(0, maxTotal);
+    return {
+      count: combined.length,
+      query: trimmedQuery,
+      task: trimmedTask,
+      intent: input.intent,
+      results: combined
+    };
+  }
   return {
     count: results.length,
     query: trimmedQuery,
@@ -3129,6 +3183,8 @@ function formatRecallResponse(result) {
       vector_rank: row.vector_rank,
       bm25_score: row.bm25_score,
       vector_score: row.vector_score,
+      related_source_id: row.related_source_id,
+      match_type: row.match_type,
       updated_at: row.memory.updated_at
     }))
   };
@@ -3153,6 +3209,8 @@ function formatSurfaceResponse(result) {
       feedback_score: row.feedback_score,
       feedback_summary: row.feedback_summary,
       reason_codes: row.reason_codes,
+      related_source_id: row.related_source_id,
+      match_type: row.match_type,
       updated_at: row.memory.updated_at
     }))
   };
@@ -3322,7 +3380,11 @@ function createHttpServer(options) {
           emotion_val: asNumber(body.emotion_val),
           agent_id: asString(body.agent_id) ?? defaultAgentId,
           conservative: asBoolean(body.conservative),
-          provider: options?.provider
+          provider: options?.provider,
+          emotion_tag: asString(body.emotion_tag),
+          source_session: asString(body.source_session),
+          source_context: asString(body.source_context),
+          observed_at: asString(body.observed_at)
         });
         sendJson(res, 200, result);
         return;
@@ -3337,7 +3399,12 @@ function createHttpServer(options) {
           query,
           limit: asNumber(body.limit),
           agent_id: asString(body.agent_id) ?? defaultAgentId,
-          provider: options?.provider
+          provider: options?.provider,
+          related: asBoolean(body.related),
+          after: asString(body.after),
+          before: asString(body.before),
+          recency_boost: asNumber(body.recency_boost),
+          emotion_tag: asString(body.emotion_tag)
         });
         sendJson(res, 200, formatRecallResponse(result));
         return;
@@ -3357,7 +3424,12 @@ function createHttpServer(options) {
           types,
           limit: asNumber(body.limit),
           agent_id: asString(body.agent_id) ?? defaultAgentId,
-          provider: options?.provider
+          provider: options?.provider,
+          related: asBoolean(body.related),
+          after: asString(body.after),
+          before: asString(body.before),
+          recency_boost: asNumber(body.recency_boost),
+          emotion_tag: asString(body.emotion_tag)
         });
         sendJson(res, 200, formatSurfaceResponse(result));
         return;
@@ -3954,6 +4026,7 @@ export {
   exportMemories,
   extractIngestItems,
   failMaintenanceJob,
+  fetchRelatedLinks,
   findResumableMaintenanceJob,
   formatNarrativeBoot,
   formatRelativeDate,