opencode-graphiti 0.2.4 → 0.2.5-canary.274f482.20260604214739

package/esm/src/handlers/chat.js

@@ -34,7 +34,7 @@ export function createChatHandler(deps) {
             if (prepared) {
                 state.injectedMemories = true;
             }
-            logger.info("Prepared local session memory for chat transform", {
+            logger.info("Prepared local memory for chat transform", {
                 sessionID: canonicalSessionId,
                 sourceSessionID: sessionID,
                 hotTierReady: state.hotTierReady,
@@ -48,7 +48,7 @@ export function createChatHandler(deps) {
             }
         }
         catch (error) {
-            logger.warn("Unable to prepare local session memory for chat transform", {
+            logger.warn("Unable to prepare local memory for chat transform", {
                 sessionID,
                 error,
             });

package/esm/src/handlers/compacting.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"compacting.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/compacting.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpD,KAAK,cAAc,GAAG,WAAW,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;AAI5E,MAAM,WAAW,qBAAqB;IACpC,cAAc,EAAE,cAAc,CAAC;CAChC;AAED,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,qBAAqB,GAC1B,cAAc,CAmChB"}
+{"version":3,"file":"compacting.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/compacting.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpD,KAAK,cAAc,GAAG,WAAW,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;AAI5E,MAAM,WAAW,qBAAqB;IACpC,cAAc,EAAE,cAAc,CAAC;CAChC;AAED,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,qBAAqB,GAC1B,cAAc,CAqChB"}

package/esm/src/handlers/compacting.js

@@ -9,19 +9,19 @@ export function createCompactingHandler(deps) {
             if (!state?.isMain)
                 return;
             sessionManager.markResolvedSessionActive(sessionID, canonicalSessionId);
-            const prepared = await sessionManager.prepareInjection(canonicalSessionId);
+            const prepared = await sessionManager.prepareInjection(canonicalSessionId, undefined, { forCompaction: true });
             if (!prepared?.envelope)
                 return;
             output.context.push(prepared.envelope);
             sessionManager.clearPendingInjection(state, prepared);
-            logger.info("Injected local session_memory into compaction context", {
+            logger.info("Injected local memory into compaction context", {
                 sessionID: canonicalSessionId,
                 sourceSessionID: sessionID,
                 hotTierReady: state.hotTierReady,
             });
         }
         catch (error) {
-            logger.warn("Unable to prepare local session memory for compaction", {
+            logger.warn("Unable to prepare local memory for compaction", {
                 sessionID,
                 error,
             });

package/esm/src/handlers/messages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/messages.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAQjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAGpD,KAAK,qBAAqB,GAAG,WAAW,CACtC,KAAK,CAAC,sCAAsC,CAAC,CAC9C,CAAC;AAIF,MAAM,WAAW,mBAAmB;IAClC,cAAc,EAAE,cAAc,CAAC;CAChC;AAyFD,wBAAgB,qBAAqB,CACnC,IAAI,EAAE,mBAAmB,GACxB,qBAAqB,CAmEvB"}
+{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/messages.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAQjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAGpD,KAAK,qBAAqB,GAAG,WAAW,CACtC,KAAK,CAAC,sCAAsC,CAAC,CAC9C,CAAC;AAIF,MAAM,WAAW,mBAAmB;IAClC,cAAc,EAAE,cAAc,CAAC;CAChC;AAuGD,wBAAgB,qBAAqB,CACnC,IAAI,EAAE,mBAAmB,GACxB,qBAAqB,CAmEvB"}

package/esm/src/handlers/messages.js

@@ -9,16 +9,24 @@ const getTransformMessage = (input) => {
     return typeof message === "string" ? message : undefined;
 };
 const LEADING_SESSION_MEMORY_BLOCK = /^<session_memory\b[^>]*>[\s\S]*?<\/session_memory>(?:\r?\n){0,2}/;
-const LEADING_INJECTED_LEGACY_MEMORY_BLOCK_WITH_UUIDS = /^<memory\b(?=[^>]*\bdata-uuids=(["'])(?:[^"']*)\1)[^>]*>[\s\S]*?<\/memory>(?:\r?\n){0,2}/;
-const LEADING_INJECTED_EMPTY_LEGACY_MEMORY_BLOCK = /^<memory\b(?![^>]*\bdata-uuids=)[^>]*>\s*<\/memory>(?:\r?\n){0,2}/;
+const LEADING_MEMORY_BLOCK = /^<memory\b[^>]*>[\s\S]*?<\/memory>(?:\r?\n){0,2}/;
 const LEADING_PERSISTENT_MEMORY_BLOCK = /^<persistent_memory\b[^>]*>[\s\S]*?<\/persistent_memory>(?:\r?\n){0,2}/;
 const SESSION_MEMORY_SOURCE_ATTR_PATTERN = /<session_memory\b[^>]*\bsource=(['"])[^'"]+\1/i;
 const SESSION_MEMORY_GENERATED_SECTION_PATTERN = /<(?:session_snapshot|persistent_memory)\b/i;
+const MEMORY_VERSION_ATTR_PATTERN = /<memory\b[^>]*\bversion=(['"])2\1/i;
+const LEGACY_MEMORY_UUID_ATTR_PATTERN = /<memory\b[^>]*\bdata-uuids=(['"])(?:[^'"]*)\1/i;
+const EMPTY_MEMORY_BLOCK_PATTERN = /^<memory\b[^>]*>\s*<\/memory>$/i;
+const MEMORY_GENERATED_SECTION_PATTERN = /<(?:session_snapshot|persistent_memory)\b/i;
 const PERSISTENT_MEMORY_GENERATED_CONTENT_PATTERN = /<(?:node|fact|episode)\b/i;
 const USER_MEMORY_ENVELOPE_TAG_PATTERN = /<\/?(?:session_memory|memory|persistent_memory)\b[^>]*>/gi;
 const looksLikeInjectedSessionMemoryBlock = (block, allowAttrlessFollowup) => SESSION_MEMORY_SOURCE_ATTR_PATTERN.test(block) ||
     SESSION_MEMORY_GENERATED_SECTION_PATTERN.test(block) ||
     allowAttrlessFollowup;
+const looksLikeInjectedMemoryBlock = (block, allowAttrlessFollowup) => MEMORY_VERSION_ATTR_PATTERN.test(block) ||
+    LEGACY_MEMORY_UUID_ATTR_PATTERN.test(block) ||
+    EMPTY_MEMORY_BLOCK_PATTERN.test(block) ||
+    MEMORY_GENERATED_SECTION_PATTERN.test(block) ||
+    allowAttrlessFollowup;
 const looksLikeInjectedPersistentMemoryBlock = (block) => PERSISTENT_MEMORY_GENERATED_CONTENT_PATTERN.test(block);
 const scrubPromptMemoryText = (text) => {
     let scrubbed = text;
@@ -33,11 +41,10 @@ const scrubPromptMemoryText = (text) => {
             scrubbedInjectedPrefix = true;
             continue;
         }
-        const next = scrubbed
-            .replace(LEADING_INJECTED_LEGACY_MEMORY_BLOCK_WITH_UUIDS, "")
-            .replace(LEADING_INJECTED_EMPTY_LEGACY_MEMORY_BLOCK, "");
-        if (next !== scrubbed) {
-            scrubbed = next;
+        const leadingMemory = scrubbed.match(LEADING_MEMORY_BLOCK)?.[0];
+        if (leadingMemory &&
+            looksLikeInjectedMemoryBlock(leadingMemory, scrubbedInjectedPrefix)) {
+            scrubbed = scrubbed.slice(leadingMemory.length);
             scrubbedInjectedPrefix = true;
             continue;
         }
@@ -83,7 +90,7 @@ export function createMessagesHandler(deps) {
                 return;
             }
             textPart.text = `${prepared.envelope}\n\n${effectiveUserText}`;
-            logger.info("Injected canonical session_memory block", {
+            logger.info("Injected canonical memory block", {
                 sessionID: canonicalSessionId,
                 sourceSessionID,
                 rewroteExistingMemory: scrubbedUserText !== latestUserText,

package/esm/src/handlers/tool-before.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool-before.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/tool-before.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAC5E,OAAO,EAEL,KAAK,kBAAkB,EACvB,KAAK,eAAe,EACrB,MAAM,6BAA6B,CAAC;AAErC,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,2CAA2C,CAAC;AACzF,OAAO,KAAK,EAAE,+BAA+B,EAAE,MAAM,eAAe,CAAC;AAErE,KAAK,cAAc,GAAG,WAAW,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;AAIhE,MAAM,WAAW,qBAAqB;IACpC,oBAAoB,EAAE,+BAA+B,CAAC;IACtD,gBAAgB,EAAE,iBAAiB,CAAC;IACpC,eAAe,EAAE,uBAAuB,CAAC;IACzC,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,kBAAkB,KAAK,eAAe,CAAC;CAChE;AAqCD,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,qBAAqB,GAC1B,cAAc,CAuDhB"}
+{"version":3,"file":"tool-before.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/tool-before.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oCAAoC,CAAC;AAC5E,OAAO,EAEL,KAAK,kBAAkB,EACvB,KAAK,eAAe,EACrB,MAAM,6BAA6B,CAAC;AAErC,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,2CAA2C,CAAC;AACzF,OAAO,KAAK,EAAE,+BAA+B,EAAE,MAAM,eAAe,CAAC;AAErE,KAAK,cAAc,GAAG,WAAW,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;AAIhE,MAAM,WAAW,qBAAqB;IACpC,oBAAoB,EAAE,+BAA+B,CAAC;IACtD,gBAAgB,EAAE,iBAAiB,CAAC;IACpC,eAAe,EAAE,uBAAuB,CAAC;IACzC,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,kBAAkB,KAAK,eAAe,CAAC;CAChE;AAoCD,wBAAgB,uBAAuB,CACrC,IAAI,EAAE,qBAAqB,GAC1B,cAAc,CAuDhB"}

package/esm/src/handlers/tool-before.js

@@ -5,10 +5,10 @@ const toRecord = (value) => value && typeof value === "object" && !Array.isArray
     : {};
 const SESSION_MCP_TOOL_NAME_SET = new Set(SESSION_MCP_TOOL_NAMES);
 const isSessionMcpTool = (toolName) => SESSION_MCP_TOOL_NAME_SET.has(toolName);
-const injectRootSessionId = (args, canonicalSessionId) => ({
-    ...args,
-    root_session_id: canonicalSessionId,
-});
+const stripRootSessionId = (args) => {
+    const { root_session_id: _ignored, ...rest } = args;
+    return rest;
+};
 const resolveCanonicalSessionId = async (sessionCanonicalizer, sessionId) => {
     const cached = sessionCanonicalizer.getCachedCanonicalSessionId(sessionId);
     if (cached)
@@ -23,10 +23,10 @@ const resolveCanonicalSessionId = async (sessionCanonicalizer, sessionId) => {
 export function createToolBeforeHandler(deps) {
     const route = deps.routeToolCall ?? defaultRouteToolCall;
     return async ({ tool, sessionID, callID }, output) => {
-        const canonicalSessionId = await resolveCanonicalSessionId(deps.sessionCanonicalizer, sessionID);
         const sessionTool = isSessionMcpTool(tool);
+        const canonicalSessionId = await resolveCanonicalSessionId(deps.sessionCanonicalizer, sessionID);
         const args = sessionTool
-            ? injectRootSessionId(toRecord(output.args), canonicalSessionId)
+            ? stripRootSessionId(toRecord(output.args))
             : toRecord(output.args);
         if (sessionTool) {
             output.args = args;
@@ -42,7 +42,7 @@ export function createToolBeforeHandler(deps) {
                 return;
             case "modify":
                 output.args = sessionTool
-                    ? injectRootSessionId(toRecord(decision.args), canonicalSessionId)
+                    ? stripRootSessionId(toRecord(decision.args))
                     : decision.args;
                 deps.routingOutcomes.set(callID, {
                     source: "tool-routing",
@@ -64,7 +64,7 @@ export function createToolBeforeHandler(deps) {
                     action: "deny",
                     reason: decision.reason,
                 });
-                throw new Error(`Tool denied (${tool})`);
+                throw new Error(decision.guidance || `Tool denied (${tool})`);
         }
     };
 }

package/esm/src/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAe,MAAM,qBAAqB,CAAC;AAsE/D,eAAO,MAAM,gCAAgC,GAC3C,WAAW,OAAO,EAClB,UAAU,MAAM,KACf,IAOF,CAAC;AAEF,eAAO,MAAM,6BAA6B,GACxC,WAAW,OAAO,EAClB,UAAU,MAAM,KACf,IAOF,CAAC;AA+BF,eAAO,MAAM,QAAQ,EAAE,MA0OtB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAS,MAAM,EAAe,MAAM,qBAAqB,CAAC;AA6GtE,eAAO,MAAM,gCAAgC,GAC3C,WAAW,OAAO,EAClB,UAAU,MAAM,KACf,IAOF,CAAC;AAEF,eAAO,MAAM,6BAA6B,GACxC,WAAW,OAAO,EAClB,UAAU,MAAM,KACf,IAOF,CAAC;AAmCF,eAAO,MAAM,QAAQ,EAAE,MAkTtB,CAAC"}

package/esm/src/index.js

@@ -11,18 +11,32 @@ import { GraphitiConnectionManager } from "./services/connection-manager.js";
 import { GraphitiAsyncService } from "./services/graphiti-async.js";
 import { GraphitiMcpClient } from "./services/graphiti-mcp.js";
 import { redactEndpointUserInfo } from "./services/endpoint-redaction.js";
-import { notifyGraphitiAvailabilityIssue, setOpenCodeClient, } from "./services/opencode-warning.js";
+import { notifyDreamShutdownDelay, notifyGraphitiAvailabilityIssue, setOpenCodeClient, } from "./services/opencode-warning.js";
 import { RedisCacheService } from "./services/redis-cache.js";
 import { RedisClient } from "./services/redis-client.js";
 import { RedisEventsService } from "./services/redis-events.js";
+import { DreamStore } from "./services/dream-store.js";
 import { logger } from "./services/logger.js";
+import { SessionNotesService } from "./services/session-notes.js";
 import { RedisSnapshotService } from "./services/redis-snapshot.js";
 import { registerRuntimeTeardown } from "./services/runtime-teardown.js";
 import { createSessionExecutor } from "./services/session-executor.js";
-import { createSessionMcpRuntime } from "./services/session-mcp-runtime.js";
+import { createSessionMcpRuntime, SESSION_SEARCH_STRENGTHENED_DESCRIPTION, } from "./services/session-mcp-runtime.js";
 import { ToolGuidanceCache } from "./services/tool-guidance-cache.js";
 import { ToolRoutingOutcomeCache } from "./services/tool-routing-outcome-cache.js";
 import { makeGroupId, makeUserGroupId } from "./utils.js";
+const getTrackedRootSessionIds = (sessionManager) => {
+    const manager = sessionManager;
+    const tracked = manager.getTrackedRootSessionIds;
+    if (typeof tracked === "function") {
+        return tracked.call(sessionManager);
+    }
+    if (!(manager.sessions instanceof Map))
+        return [];
+    return [...manager.sessions.entries()]
+        .filter(([, state]) => state?.isMain)
+        .map(([sessionId]) => sessionId);
+};
 let activeRuntimeTeardown = null;
 let runtimeInitialization = Promise.resolve();
 export const warnOnGraphitiStartupUnavailable = (connected, endpoint) => {
@@ -40,6 +54,7 @@ export const warnOnRedisStartupUnavailable = (connected, endpoint) => {
 const defaultGraphitiDependencies = {
     loadConfig,
     setOpenCodeClient,
+    notifyDreamShutdownDelay,
     warnOnGraphitiStartupUnavailable,
     warnOnRedisStartupUnavailable,
     GraphitiConnectionManager,
@@ -49,6 +64,8 @@ const defaultGraphitiDependencies = {
     RedisEventsService,
     RedisSnapshotService,
     RedisCacheService,
+    DreamStore,
+    SessionNotesService,
     BatchDrainService,
     GraphitiAsyncService,
     createSessionExecutor,
@@ -64,6 +81,7 @@ const defaultGraphitiDependencies = {
     ToolRoutingOutcomeCache,
     makeGroupId,
     makeUserGroupId,
+    now: () => new Date().toISOString(),
 };
 export const graphiti = (input, dependencies = defaultGraphitiDependencies) => {
     const setup = runtimeInitialization.then(async () => {
@@ -137,13 +155,17 @@ export const graphiti = (input, dependencies = defaultGraphitiDependencies) => {
                 ttlSeconds: config.redis.cacheTtlSeconds,
                 driftThreshold: config.graphiti.driftThreshold,
             });
+            const dreamStore = new dependencies.DreamStore(redisClient);
+            const defaultGroupId = dependencies.makeGroupId(config.graphiti.groupIdPrefix, input.directory);
+            const defaultUserGroupId = dependencies.makeUserGroupId(config.graphiti.groupIdPrefix, input.directory);
+            const notesService = new dependencies.SessionNotesService(redisClient, {
+                groupId: defaultGroupId,
+            });
             const batchDrain = new dependencies.BatchDrainService(redisClient, redisEvents, {
                 batchSize: config.redis.batchSize,
                 batchMaxBytes: config.redis.batchMaxBytes,
                 drainRetryMax: config.redis.drainRetryMax,
             });
-            const defaultGroupId = dependencies.makeGroupId(config.graphiti.groupIdPrefix, input.directory);
-            const defaultUserGroupId = dependencies.makeUserGroupId(config.graphiti.groupIdPrefix, input.directory);
             const graphitiAsync = new dependencies.GraphitiAsyncService(graphitiClient, redisCache, batchDrain);
             startupCleanupTasks.unshift({
                 name: "graphiti-async",
@@ -153,6 +175,7 @@ export const graphiti = (input, dependencies = defaultGraphitiDependencies) => {
             const sessionMcpRuntime = dependencies.createSessionMcpRuntime({
                 redisClient,
                 graphitiCache: redisCache,
+                notesService,
                 sessionTtlSeconds: config.redis.sessionTtlSeconds,
                 groupId: defaultGroupId,
                 sessionExecutor,
@@ -164,12 +187,37 @@ export const graphiti = (input, dependencies = defaultGraphitiDependencies) => {
             });
             const sessionManager = new dependencies.SessionManager(defaultGroupId, defaultUserGroupId, input.client, redisEvents, redisSnapshot, redisCache, {
                 idleRetentionMs: config.redis.sessionTtlSeconds * 1000,
+                notesService,
                 runtimeStateMigrator: sessionMcpRuntime,
             });
             sessionMcpRuntime.setSessionCanonicalizer(sessionManager);
             const toolGuidanceCache = new dependencies.ToolGuidanceCache();
             const toolRoutingOutcomes = new dependencies.ToolRoutingOutcomeCache();
+            const sessionBiasState = new Map();
+            const chatHandler = dependencies.createChatHandler({
+                sessionManager,
+                redisEvents,
+                graphitiAsync,
+                drainTriggerSize: config.redis.batchSize,
+            });
+            const compactingHandler = dependencies
+                .createCompactingHandler({
+                sessionManager,
+            });
             startupTeardown = dependencies.registerRuntimeTeardown([
+                {
+                    name: "dream-shutdown-warning",
+                    run: async () => {
+                        const targetWatermark = dependencies.now();
+                        for (const rootSessionId of getTrackedRootSessionIds(sessionManager)) {
+                            const watermark = await dreamStore.getWatermark(rootSessionId);
+                            if (watermark === null || watermark < targetWatermark) {
+                                dependencies.notifyDreamShutdownDelay();
+                                return;
+                            }
+                        }
+                    },
+                },
                 {
                     name: "graphiti-drain-flush",
                     run: () => graphitiAsync.flushPendingGroups(sessionManager.getTrackedGroupIds()),
@@ -204,21 +252,45 @@ export const graphiti = (input, dependencies = defaultGraphitiDependencies) => {
                     sdkClient: input.client,
                     directory: input.directory,
                 }),
-                "chat.message": dependencies.createChatHandler({
-                    sessionManager,
-                    redisEvents,
-                    graphitiAsync,
-                    drainTriggerSize: config.redis.batchSize,
-                }),
-                "experimental.session.compacting": dependencies
-                    .createCompactingHandler({
-                    sessionManager,
-                }),
+                "chat.message": async (hookInput, output) => {
+                    const canonicalSessionId = sessionManager.getCachedCanonicalSessionId(hookInput.sessionID) ??
+                        await sessionManager.resolveCanonicalSessionId(hookInput.sessionID);
+                    if (canonicalSessionId && !sessionBiasState.has(canonicalSessionId)) {
+                        const priorEvents = await redisEvents.getRecentSessionEvents(canonicalSessionId, 1, false);
+                        if (priorEvents.length === 0) {
+                            sessionBiasState.set(canonicalSessionId, "new-session");
+                        }
+                    }
+                    await chatHandler(hookInput, output);
+                },
+                "experimental.session.compacting": async (hookInput, output) => {
+                    const canonicalSessionId = sessionManager.getCachedCanonicalSessionId(hookInput.sessionID) ??
+                        await sessionManager.resolveCanonicalSessionId(hookInput.sessionID);
+                    if (canonicalSessionId) {
+                        sessionBiasState.set(canonicalSessionId, "post-compaction");
+                    }
+                    await compactingHandler(hookInput, output);
+                },
                 "experimental.chat.messages.transform": dependencies
                     .createMessagesHandler({
                     sessionManager,
                 }),
                 tool: sessionMcpRuntime.tools,
+                "tool.definition": (hookInput, output) => {
+                    if (hookInput.toolID !== "session_search")
+                        return Promise.resolve();
+                    let anyBiased = false;
+                    for (const [sessionId, state] of sessionBiasState) {
+                        if (state === "normal")
+                            continue;
+                        anyBiased = true;
+                        sessionBiasState.delete(sessionId);
+                    }
+                    if (anyBiased) {
+                        output.description = SESSION_SEARCH_STRENGTHENED_DESCRIPTION;
+                    }
+                    return Promise.resolve();
+                },
                 "tool.execute.before": dependencies.createToolBeforeHandler({
                     sessionCanonicalizer: sessionManager,
                     guidanceThrottle: toolGuidanceCache,

package/esm/src/services/connection-manager.d.ts

@@ -54,8 +54,8 @@ type GraphitiConnectionManagerOptions = {
     connectionFactory?: ConnectionFactory;
     random?: () => number;
     now?: () => number;
-    setTimer?: (callback: () => void, delayMs: number) => TimerHandle;
-    clearTimer?: (timer: TimerHandle) => void;
+    setTimer?(callback: () => void, delayMs: number): TimerHandle;
+    clearTimer?(timer: TimerHandle): void;
 };
 export declare class GraphitiConnectionManager implements GraphitiToolCaller {
     private readonly endpoint;

package/esm/src/services/connection-manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"connection-manager.d.ts","sourceRoot":"","sources":["../../../src/src/services/connection-manager.ts"],"names":[],"mappings":"AAyDA,MAAM,MAAM,uBAAuB,GAC/B,YAAY,GACZ,WAAW,GACX,SAAS,GACT,SAAS,GACT,SAAS,CAAC;AAEd,KAAK,WAAW,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,MAAM,CAAC;AAE1D,qBAAa,oBAAqB,SAAQ,KAAK;IAI3C,QAAQ,CAAC,KAAK,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS;IAHnD,QAAQ,CAAC,IAAI,aAAa;gBAGf,KAAK,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,EACjD,OAAO,CAAC,EAAE,MAAM;CAYnB;AAED,qBAAa,yBAA0B,SAAQ,KAAK;IAClD,QAAQ,CAAC,IAAI,mBAAmB;gBAG9B,OAAO,SAA4D;CAKtE;AAED,qBAAa,2BAA4B,SAAQ,KAAK;IACpD,QAAQ,CAAC,IAAI,qBAAqB;gBAEtB,OAAO,SAA+B;CAInD;AAED,qBAAa,sBAAuB,SAAQ,KAAK;IAC/C,QAAQ,CAAC,IAAI,uBAAuB;gBAExB,OAAO,SAA+B;CAInD;AAED,qBAAa,2BAA4B,SAAQ,KAAK;IACpD,QAAQ,CAAC,IAAI,qBAAqB;gBAEtB,OAAO,SAA6B;CAIjD;AAED,wBAAgB,sBAAsB,CACpC,GAAG,EAAE,OAAO,GACX,GAAG,IAAI,oBAAoB,CAE7B;AAED,wBAAgB,sBAAsB,CACpC,GAAG,EAAE,OAAO,GACX,GAAG,IAAI,yBAAyB,GAAG,2BAA2B,CAGhE;AAED,MAAM,MAAM,mBAAmB,GAAG;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB,CAAC;AAEF,MAAM,WAAW,kBAAkB;IACjC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACzB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,QAAQ,CACN,OAAO,EAAE,mBAAmB,EAC5B,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,OAAO,CAAC,CAAC;CACrB;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,IAAI,IAAI,CAAC;IACd,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,KAAK,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5C,QAAQ,CACN,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,UAAU,CAAC,EAAE,MAAM,GAClB,OAAO,CAAC,OAAO,CAAC,CAAC;CACrB;AAWD,KAAK,iBAAiB,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,kBAAkB,CAAC;AA4BlE,KAAK,gCAAgC,GAAG;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,MAAM,CAAC,EAAE,MAAM,MAAM,CAAC;IACtB,GAAG,CAAC,EAAE,MAAM,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,OAAO,EAAE,MAAM,KAAK,WAAW,CAAC;IAClE,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;CAC3C,CAAC;AAkGF,qBAAa,yBAA0B,YAAW,kBAAkB;IAClE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAS;IACjD,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAC7C,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAC7C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAoB;IACtD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;IACtC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAe;IACnC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAGZ;IACjB,OAAO,CAAC,QAAQ,CAAC,cAAc,CAA+B;IAE9D,OAAO,CAAC,KAAK,CAAsC;IACnD,OAAO,CAAC,UAAU,CAAmC;IACrD,OAAO,CAAC,cAAc,CAAiC;IACvD,OAAO,CAAC,cAAc,CAA4B;IAClD,OAAO,CAAC,eAAe,CAAwB;IAC/C,OAAO,CAAC,YAAY,CAAuC;IAC3D,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,WAAW,CAA8B;IACjD,OAAO,CAAC,wBAAwB,CAA8B;gBAElD,OAAO,EAAE,gCAAgC;IAoBrD,QAAQ,IAAI,uBAAuB;IAInC,KAAK,IAAI,IAAI;IAcP,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAgDrB,KAAK,CAAC,SAAS,SAAwB,GAAG,OAAO,CAAC,OAAO,CAAC;IAwB1D,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,UAAU,SAAyB,GAClC,OAAO,CAAC,OAAO,CAAC;IA0Bb,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;YAerB,gBAAgB;YAoDhB,kCAAkC;YA0ElC,uCAAuC;IAyBrD,OAAO,CAAC,sBAAsB;YAIhB,uBAAuB;IASrC,OAAO,CAAC,sBAAsB;IAwC9B,OAAO,CAAC,mBAAmB;IAQ3B,OAAO,CAAC,cAAc;YAyCR,iBAAiB;IAgC/B,OAAO,CAAC,oBAAoB;IAQ5B,OAAO,CAAC,iBAAiB;IAOzB,OAAO,CAAC,gBAAgB;IAUxB,OAAO,CAAC,iBAAiB;IA6BzB,OAAO,CAAC,oBAAoB;IAO5B,OAAO,CAAC,mBAAmB;CAO5B"}
+{"version":3,"file":"connection-manager.d.ts","sourceRoot":"","sources":["../../../src/src/services/connection-manager.ts"],"names":[],"mappings":"AAyDA,MAAM,MAAM,uBAAuB,GAC/B,YAAY,GACZ,WAAW,GACX,SAAS,GACT,SAAS,GACT,SAAS,CAAC;AAEd,KAAK,WAAW,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,MAAM,CAAC;AAE1D,qBAAa,oBAAqB,SAAQ,KAAK;IAI3C,QAAQ,CAAC,KAAK,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS;IAHnD,QAAQ,CAAC,IAAI,aAAa;gBAGf,KAAK,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,EACjD,OAAO,CAAC,EAAE,MAAM;CAYnB;AAED,qBAAa,yBAA0B,SAAQ,KAAK;IAClD,QAAQ,CAAC,IAAI,mBAAmB;gBAG9B,OAAO,SAA4D;CAKtE;AAED,qBAAa,2BAA4B,SAAQ,KAAK;IACpD,QAAQ,CAAC,IAAI,qBAAqB;gBAEtB,OAAO,SAA+B;CAInD;AAED,qBAAa,sBAAuB,SAAQ,KAAK;IAC/C,QAAQ,CAAC,IAAI,uBAAuB;gBAExB,OAAO,SAA+B;CAInD;AAED,qBAAa,2BAA4B,SAAQ,KAAK;IACpD,QAAQ,CAAC,IAAI,qBAAqB;gBAEtB,OAAO,SAA6B;CAIjD;AAED,wBAAgB,sBAAsB,CACpC,GAAG,EAAE,OAAO,GACX,GAAG,IAAI,oBAAoB,CAE7B;AAED,wBAAgB,sBAAsB,CACpC,GAAG,EAAE,OAAO,GACX,GAAG,IAAI,yBAAyB,GAAG,2BAA2B,CAGhE;AAED,MAAM,MAAM,mBAAmB,GAAG;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB,CAAC;AAEF,MAAM,WAAW,kBAAkB;IACjC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACzB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,QAAQ,CACN,OAAO,EAAE,mBAAmB,EAC5B,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,OAAO,CAAC,CAAC;CACrB;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,IAAI,IAAI,CAAC;IACd,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,KAAK,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5C,QAAQ,CACN,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,UAAU,CAAC,EAAE,MAAM,GAClB,OAAO,CAAC,OAAO,CAAC,CAAC;CACrB;AAWD,KAAK,iBAAiB,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,kBAAkB,CAAC;AA4BlE,KAAK,gCAAgC,GAAG;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,MAAM,CAAC,EAAE,MAAM,MAAM,CAAC;IACtB,GAAG,CAAC,EAAE,MAAM,MAAM,CAAC;IACnB,QAAQ,CAAC,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,OAAO,EAAE,MAAM,GAAG,WAAW,CAAC;IAC9D,UAAU,CAAC,CAAC,KAAK,EAAE,WAAW,GAAG,IAAI,CAAC;CACvC,CAAC;AAkGF,qBAAa,yBAA0B,YAAW,kBAAkB;IAClE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAS;IACjD,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAC7C,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAC7C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAoB;IACtD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;IACtC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAe;IACnC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAGZ;IACjB,OAAO,CAAC,QAAQ,CAAC,cAAc,CAA+B;IAE9D,OAAO,CAAC,KAAK,CAAsC;IACnD,OAAO,CAAC,UAAU,CAAmC;IACrD,OAAO,CAAC,cAAc,CAAiC;IACvD,OAAO,CAAC,cAAc,CAA4B;IAClD,OAAO,CAAC,eAAe,CAAwB;IAC/C,OAAO,CAAC,YAAY,CAAuC;IAC3D,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,WAAW,CAA8B;IACjD,OAAO,CAAC,wBAAwB,CAA8B;gBAElD,OAAO,EAAE,gCAAgC;IAoBrD,QAAQ,IAAI,uBAAuB;IAInC,KAAK,IAAI,IAAI;IAcP,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAgDrB,KAAK,CAAC,SAAS,SAAwB,GAAG,OAAO,CAAC,OAAO,CAAC;IAwB1D,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,UAAU,SAAyB,GAClC,OAAO,CAAC,OAAO,CAAC;IA0Bb,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;YAerB,gBAAgB;YAoDhB,kCAAkC;YA0ElC,uCAAuC;IAyBrD,OAAO,CAAC,sBAAsB;YAIhB,uBAAuB;IASrC,OAAO,CAAC,sBAAsB;IAwC9B,OAAO,CAAC,mBAAmB;IAQ3B,OAAO,CAAC,cAAc;YAyCR,iBAAiB;IAgC/B,OAAO,CAAC,oBAAoB;IAQ5B,OAAO,CAAC,iBAAiB;IAOzB,OAAO,CAAC,gBAAgB;IAUxB,OAAO,CAAC,iBAAiB;IA6BzB,OAAO,CAAC,oBAAoB;IAO5B,OAAO,CAAC,mBAAmB;CAO5B"}

package/esm/src/services/dream-store.d.ts

@@ -0,0 +1,21 @@
+import type { NormalizedMemoryResult } from "../types/index.js";
+import type { RedisClient } from "./redis-client.js";
+export type DreamSummaryRecord = {
+    rootSessionId: string;
+    granularity: string;
+    created_at: string;
+    body: string;
+};
+export declare class DreamStore {
+    private readonly redis;
+    constructor(redis: RedisClient);
+    putSummary(record: DreamSummaryRecord): Promise<void>;
+    getSummariesAround(input: {
+        rootSessionId: string;
+        when: string;
+        query?: string;
+    }): Promise<NormalizedMemoryResult[]>;
+    getWatermark(rootSessionId: string): Promise<string | null>;
+    setWatermark(rootSessionId: string, value: string): Promise<void>;
+}
+//# sourceMappingURL=dream-store.d.ts.map

package/esm/src/services/dream-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dream-store.d.ts","sourceRoot":"","sources":["../../../src/src/services/dream-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAChE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAErD,MAAM,MAAM,kBAAkB,GAAG;IAC/B,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAgEF,qBAAa,UAAU;IACT,OAAO,CAAC,QAAQ,CAAC,KAAK;gBAAL,KAAK,EAAE,WAAW;IAEzC,UAAU,CAAC,MAAM,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAYrD,kBAAkB,CAAC,KAAK,EAAE;QAC9B,aAAa,EAAE,MAAM,CAAC;QACtB,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;IA8C/B,YAAY,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAI3D,YAAY,CAAC,aAAa,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAGxE"}

package/esm/src/services/dream-store.js

@@ -0,0 +1,96 @@
+const dreamSummariesKey = (rootSessionId) => `session:${rootSessionId}:dream:summaries`;
+const dreamSummaryField = (record) => `${record.granularity}:${record.created_at}`;
+const dreamWatermarkKey = (rootSessionId) => `session:${rootSessionId}:dream:watermark`;
+const normalizeText = (value) => value.trim().replace(/\s+/g, " ");
+const tokenize = (value) => normalizeText(value).toLowerCase().match(/[a-z0-9]{2,}/g) ?? [];
+const parseStoredRecord = (value) => {
+    try {
+        const parsed = JSON.parse(value);
+        if (typeof parsed.granularity !== "string" ||
+            typeof parsed.created_at !== "string" ||
+            typeof parsed.body !== "string") {
+            return null;
+        }
+        return {
+            granularity: parsed.granularity,
+            created_at: parsed.created_at,
+            body: parsed.body,
+        };
+    }
+    catch {
+        return null;
+    }
+};
+const scoreSummary = (record, query) => {
+    const normalizedQuery = normalizeText(query ?? "").toLowerCase();
+    if (!normalizedQuery)
+        return 1;
+    const body = normalizeText(record.body).toLowerCase();
+    if (body === normalizedQuery)
+        return 1;
+    if (body.includes(normalizedQuery))
+        return 0.95;
+    const queryTokens = [...new Set(tokenize(normalizedQuery))];
+    if (queryTokens.length === 0)
+        return 0;
+    const matched = queryTokens.filter((token) => body.includes(token));
+    if (matched.length === 0)
+        return 0;
+    return Number((matched.length / queryTokens.length).toFixed(6));
+};
+export class DreamStore {
+    constructor(redis) {
+        Object.defineProperty(this, "redis", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: redis
+        });
+    }
+    async putSummary(record) {
+        await this.redis.setHashFields(dreamSummariesKey(record.rootSessionId), {
+            [dreamSummaryField(record)]: JSON.stringify({
+                granularity: record.granularity,
+                created_at: record.created_at,
+                body: record.body,
+            }),
+        });
+    }
+    async getSummariesAround(input) {
+        const summaries = Object.values(await this.redis.getHashAll(dreamSummariesKey(input.rootSessionId)))
+            .map(parseStoredRecord)
+            .filter((record) => record !== null)
+            .map((record) => ({
+            record,
+            score: scoreSummary(record, input.query),
+        }))
+            .filter(({ score }) => score > 0)
+            .sort((left, right) => left.record.created_at.localeCompare(right.record.created_at));
+        const before = summaries.filter(({ record }) => record.created_at < input.when);
+        const exact = summaries.filter(({ record }) => record.created_at === input.when);
+        const after = summaries.filter(({ record }) => record.created_at > input.when);
+        const selected = [
+            ...before.slice(-1),
+            ...exact,
+            ...after.slice(0, 1),
+        ];
+        return selected.map(({ record, score }) => ({
+            type: "summary",
+            ref: `session:${input.rootSessionId}:summary:dream:${record.granularity}:${record.created_at}`,
+            snippet: record.body,
+            score,
+            id: `${record.granularity}:${record.created_at}`,
+            root_session_id: input.rootSessionId,
+            scope: "session",
+            granularity: record.granularity,
+            created_at: record.created_at,
+            source: "dream",
+        }));
+    }
+    async getWatermark(rootSessionId) {
+        return await this.redis.getString(dreamWatermarkKey(rootSessionId));
+    }
+    async setWatermark(rootSessionId, value) {
+        await this.redis.setString(dreamWatermarkKey(rootSessionId), value);
+    }
+}

package/esm/src/services/exact-history.d.ts

@@ -0,0 +1,10 @@
+import type { NormalizedMemoryResult } from "../types/index.js";
+export type ExactHistoryAdapter = {
+    search(input: {
+        rootSessionId: string;
+        query: string;
+        when: string;
+    }): Promise<NormalizedMemoryResult[]>;
+};
+export declare const createExactHistoryAdapter: () => ExactHistoryAdapter;
+//# sourceMappingURL=exact-history.d.ts.map

package/esm/src/services/exact-history.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"exact-history.d.ts","sourceRoot":"","sources":["../../../src/src/services/exact-history.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAEhE,MAAM,MAAM,mBAAmB,GAAG;IAChC,MAAM,CAAC,KAAK,EAAE;QACZ,aAAa,EAAE,MAAM,CAAC;QACtB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;KACd,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC,CAAC;CACvC,CAAC;AAEF,eAAO,MAAM,yBAAyB,QAAO,mBAE3C,CAAC"}

package/esm/src/services/exact-history.js

@@ -0,0 +1,3 @@
+export const createExactHistoryAdapter = () => ({
+    search: () => Promise.resolve([]),
+});

package/esm/src/services/graphiti-async.d.ts

@@ -3,8 +3,8 @@ import type { GraphitiMcpClient } from "./graphiti-mcp.js";
 import type { RedisCacheService } from "./redis-cache.js";
 type TimerHandle = ReturnType<typeof setTimeout> | number;
 type GraphitiAsyncServiceOptions = {
-    setTimer?: (callback: () => void, delayMs: number) => TimerHandle;
-    clearTimer?: (timer: TimerHandle) => void;
+    setTimer?(callback: () => void, delayMs: number): TimerHandle;
+    clearTimer?(timer: TimerHandle): void;
 };
 export declare class GraphitiAsyncService {
     private readonly graphiti;

package/esm/src/services/graphiti-async.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"graphiti-async.d.ts","sourceRoot":"","sources":["../../../src/src/services/graphiti-async.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAE3D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAE1D,KAAK,WAAW,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,MAAM,CAAC;AAE1D,KAAK,2BAA2B,GAAG;IACjC,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,OAAO,EAAE,MAAM,KAAK,WAAW,CAAC;IAClE,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;CAC3C,CAAC;AAEF,qBAAa,oBAAoB;IAqB7B,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,oBAAoB;IAxBvC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,+BAA+B,CAAU;IACjE,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAoC;IAClE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAGZ;IACjB,OAAO,CAAC,QAAQ,CAAC,cAAc,CAA+B;IAC9D,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAkC;IACnE,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAMhC;IACJ,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAoC;IACpE,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAoC;IACnE,OAAO,CAAC,OAAO,CAAS;gBAGL,QAAQ,EAAE,iBAAiB,EAC3B,KAAK,EAAE,iBAAiB,EACxB,KAAK,EAAE,iBAAiB,EACxB,iBAAiB,SAAQ,EACzB,oBAAoB,SACiB,EACtD,OAAO,GAAE,2BAAgC;IAQrC,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB7D,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAsB9B,OAAO,CAAC,aAAa;IAcrB,OAAO,CAAC,gBAAgB;IAqBxB,OAAO,CAAC,kBAAkB;IAO1B,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAyBrC,oBAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IA2E1D,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;CAsDrC"}
+{"version":3,"file":"graphiti-async.d.ts","sourceRoot":"","sources":["../../../src/src/services/graphiti-async.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAE3D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAE1D,KAAK,WAAW,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,MAAM,CAAC;AAE1D,KAAK,2BAA2B,GAAG;IACjC,QAAQ,CAAC,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,OAAO,EAAE,MAAM,GAAG,WAAW,CAAC;IAC9D,UAAU,CAAC,CAAC,KAAK,EAAE,WAAW,GAAG,IAAI,CAAC;CACvC,CAAC;AAEF,qBAAa,oBAAoB;IAqB7B,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,KAAK;IACtB,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAClC,OAAO,CAAC,QAAQ,CAAC,oBAAoB;IAxBvC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,+BAA+B,CAAU;IACjE,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAoC;IAClE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAGZ;IACjB,OAAO,CAAC,QAAQ,CAAC,cAAc,CAA+B;IAC9D,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAkC;IACnE,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAMhC;IACJ,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAoC;IACpE,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAoC;IACnE,OAAO,CAAC,OAAO,CAAS;gBAGL,QAAQ,EAAE,iBAAiB,EAC3B,KAAK,EAAE,iBAAiB,EACxB,KAAK,EAAE,iBAAiB,EACxB,iBAAiB,SAAQ,EACzB,oBAAoB,SACiB,EACtD,OAAO,GAAE,2BAAgC;IAQrC,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAwB7D,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAsB9B,OAAO,CAAC,aAAa;IAcrB,OAAO,CAAC,gBAAgB;IAqBxB,OAAO,CAAC,kBAAkB;IAO1B,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAyBrC,oBAAoB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IA2E1D,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;CAsDrC"}

package/esm/src/services/memory-results.d.ts

@@ -0,0 +1,6 @@
+import type { NormalizedMemoryResult } from "../types/index.js";
+export declare const compareWeightedResults: (left: NormalizedMemoryResult, right: NormalizedMemoryResult) => number;
+export declare const orderMemoryResults: (results: NormalizedMemoryResult[], options: {
+    mode: "query" | "reflection";
+}) => NormalizedMemoryResult[];
+//# sourceMappingURL=memory-results.d.ts.map

package/esm/src/services/memory-results.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-results.d.ts","sourceRoot":"","sources":["../../../src/src/services/memory-results.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAEhE,eAAO,MAAM,sBAAsB,GACjC,MAAM,sBAAsB,EAC5B,OAAO,sBAAsB,KAC5B,MAQF,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAC7B,SAAS,sBAAsB,EAAE,EACjC,SAAS;IAAE,IAAI,EAAE,OAAO,GAAG,YAAY,CAAA;CAAE,KACxC,sBAAsB,EAexB,CAAC"}

package/esm/src/services/memory-results.js

@@ -0,0 +1,23 @@
+export const compareWeightedResults = (left, right) => {
+    if (right.score !== left.score) {
+        return right.score - left.score;
+    }
+    if (right.created_at !== left.created_at) {
+        return right.created_at.localeCompare(left.created_at);
+    }
+    return left.ref.localeCompare(right.ref);
+};
+export const orderMemoryResults = (results, options) => {
+    if (options.mode === "reflection") {
+        return results
+            .filter((result) => result.type === "summary")
+            .sort((left, right) => left.created_at.localeCompare(right.created_at));
+    }
+    const primary = results
+        .filter((result) => result.type === "entry" || result.type === "note")
+        .sort(compareWeightedResults);
+    const summaries = results
+        .filter((result) => result.type === "summary")
+        .sort(compareWeightedResults);
+    return [...primary, ...summaries];
+};

package/esm/src/services/memory-search.d.ts

@@ -0,0 +1,29 @@
+import type { NormalizedMemoryResult } from "../types/index.js";
+import type { ExactHistoryAdapter } from "./exact-history.js";
+import type { SessionMcpResponseMap } from "./session-mcp-types.js";
+import type { SessionNotesService } from "./session-notes.js";
+export type SummarySearchAdapter = {
+    search(input: {
+        rootSessionId: string;
+        query: string;
+        when: string;
+    }): Promise<NormalizedMemoryResult[]>;
+};
+export type MemorySearchService = {
+    search(input: {
+        rootSessionId: string;
+        query: string;
+        when: string;
+    }): Promise<SessionMcpResponseMap["session_search"]>;
+};
+type MemorySearchServiceOptions = {
+    exactHistoryAdapter: ExactHistoryAdapter;
+    notesService: Pick<SessionNotesService, "searchNotes">;
+    summarySearchAdapter: SummarySearchAdapter;
+    groupId: string;
+    resultLimit?: number;
+};
+export declare const createSummarySearchAdapter: () => SummarySearchAdapter;
+export declare const createMemorySearchService: (options: MemorySearchServiceOptions) => MemorySearchService;
+export {};
+//# sourceMappingURL=memory-search.d.ts.map

package/esm/src/services/memory-search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-search.d.ts","sourceRoot":"","sources":["../../../src/src/services/memory-search.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAEhE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AACpE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAE9D,MAAM,MAAM,oBAAoB,GAAG;IACjC,MAAM,CAAC,KAAK,EAAE;QACZ,aAAa,EAAE,MAAM,CAAC;QACtB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;KACd,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC,CAAC;CACvC,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,MAAM,CAAC,KAAK,EAAE;QACZ,aAAa,EAAE,MAAM,CAAC;QACtB,KAAK,EAAE,MAAM,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;KACd,GAAG,OAAO,CAAC,qBAAqB,CAAC,gBAAgB,CAAC,CAAC,CAAC;CACtD,CAAC;AAEF,KAAK,0BAA0B,GAAG;IAChC,mBAAmB,EAAE,mBAAmB,CAAC;IACzC,YAAY,EAAE,IAAI,CAAC,mBAAmB,EAAE,aAAa,CAAC,CAAC;IACvD,oBAAoB,EAAE,oBAAoB,CAAC;IAC3C,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,eAAO,MAAM,0BAA0B,QAAO,oBAE5C,CAAC;AAMH,eAAO,MAAM,yBAAyB,GACpC,SAAS,0BAA0B,KAClC,mBAqDF,CAAC"}