opencode-graphiti 0.1.5 → 0.1.6

package/script/src/handlers/chat.js

@@ -7,17 +7,8 @@ const logger_js_1 = require("../services/logger.js");
 const utils_js_1 = require("../utils.js");
 /** Creates the `chat.message` hook handler. */
 function createChatHandler(deps) {
-    const { sessionManager, injectionInterval, client } = deps;
-    const removeSyntheticMemoryParts = (parts) => parts.filter((part) => {
-        if (part.type !== "text")
-            return true;
-        if (part.id?.startsWith("graphiti-memory-"))
-            return false;
-        if (part.id?.startsWith("graphiti-refresh-"))
-            return false;
-        return true;
-    });
-    const injectMemoryContext = async (state, messageText, output, prefix, useUserScope, characterBudget, shouldReinject) => {
+    const { sessionManager, driftThreshold, factStaleDays, client } = deps;
+    const searchAndCacheMemoryContext = async (state, messageText, useUserScope, characterBudget, seedFactUuids) => {
         const userGroupId = state.userGroupId;
         const projectFactsPromise = client.searchFacts({
             query: messageText,
@@ -50,45 +41,83 @@ function createChatHandler(deps) {
             userFactsPromise,
             userNodesPromise,
         ]);
-        const projectContext = (0, context_js_1.formatMemoryContext)(projectFacts, projectNodes);
-        const userContext = (0, context_js_1.formatMemoryContext)(userFacts, userNodes);
-        if (!projectContext && !userContext)
+        const projectContext = (0, context_js_1.deduplicateContext)({
+            facts: projectFacts,
+            nodes: projectNodes,
+        });
+        const userContext = (0, context_js_1.deduplicateContext)({
+            facts: userFacts,
+            nodes: userNodes,
+        });
+        const projectContextString = (0, context_js_1.formatMemoryContext)(projectContext.facts, projectContext.nodes, { factStaleDays });
+        const userContextString = (0, context_js_1.formatMemoryContext)(userContext.facts, userContext.nodes, { factStaleDays });
+        if (!projectContextString && !userContextString)
             return;
+        let snapshotPrimer = "";
+        if (useUserScope && characterBudget > 0) {
+            try {
+                const episodes = await client.getEpisodes({
+                    groupId: state.groupId,
+                    lastN: 10,
+                });
+                const snapshot = episodes
+                    .filter((episode) => {
+                    const description = episode.sourceDescription ??
+                        episode.source_description ?? "";
+                    return description === "session-snapshot";
+                })
+                    .sort((a, b) => {
+                    const aTime = a.created_at ? Date.parse(a.created_at) : 0;
+                    const bTime = b.created_at ? Date.parse(b.created_at) : 0;
+                    return bTime - aTime;
+                })[0];
+                if (snapshot?.content) {
+                    const snapshotBudget = Math.min(characterBudget, 1200);
+                    snapshotPrimer = [
+                        "## Session Snapshot",
+                        "> Most recent session snapshot; use to restore active strategy and open questions.",
+                        "",
+                        snapshot.content.slice(0, snapshotBudget),
+                    ].join("\n");
+                }
+            }
+            catch (err) {
+                logger_js_1.logger.error("Failed to load session snapshot", { err });
+            }
+        }
         const projectBudget = useUserScope
             ? Math.floor(characterBudget * 0.7)
             : characterBudget;
         const userBudget = characterBudget - projectBudget;
-        const truncatedProject = projectContext.slice(0, projectBudget);
-        const truncatedUser = useUserScope ? userContext.slice(0, userBudget) : "";
-        const memoryContext = [truncatedProject, truncatedUser]
+        const truncatedProject = projectContextString.slice(0, projectBudget);
+        const truncatedUser = useUserScope
+            ? userContextString.slice(0, userBudget)
+            : "";
+        const memoryContext = [snapshotPrimer, truncatedProject, truncatedUser]
             .filter((section) => section.trim().length > 0)
             .join("\n\n")
             .slice(0, characterBudget);
         if (!memoryContext)
             return;
-        if ("system" in output.message) {
-            try {
-                output.message.system = memoryContext;
-                return;
-            }
-            catch (_err) {
-                // Fall through to synthetic injection.
-            }
-        }
-        if (shouldReinject) {
-            output.parts = removeSyntheticMemoryParts(output.parts);
-        }
-        {
-            output.parts.unshift({
-                type: "text",
-                text: memoryContext,
-                id: `${prefix}${Date.now()}`,
-                sessionID: output.message.sessionID,
-                messageID: output.message.id,
-                synthetic: true,
-            });
+        const factUuids = seedFactUuids ??
+            new Set([
+                ...projectContext.facts.map((fact) => fact.uuid),
+                ...userContext.facts.map((fact) => fact.uuid),
+            ]);
+        state.cachedMemoryContext = memoryContext;
+        logger_js_1.logger.info(`Cached ${projectFacts.length + userFacts.length} facts and ${projectNodes.length + userNodes.length} nodes for system prompt injection`);
+        state.lastInjectionFactUuids = factUuids;
+    };
+    const computeJaccardSimilarity = (left, right) => {
+        if (left.size === 0 && right.size === 0)
+            return 1;
+        let intersection = 0;
+        for (const value of left) {
+            if (right.has(value))
+                intersection += 1;
         }
-        logger_js_1.logger.info(`Injected ${projectFacts.length + userFacts.length} facts and ${projectNodes.length + userNodes.length} nodes`);
+        const union = left.size + right.size - intersection;
+        return union === 0 ? 1 : intersection / union;
     };
     return async ({ sessionID }, output) => {
         if (await sessionManager.isSubagentSession(sessionID)) {
@@ -115,19 +144,30 @@ function createChatHandler(deps) {
             messageLength: messageText.length,
         });
         const shouldInjectOnFirst = !state.injectedMemories;
-        const shouldReinject = !shouldInjectOnFirst &&
-            injectionInterval > 0 &&
-            (state.messageCount - state.lastInjectionMessageCount) >=
-                injectionInterval;
-        if (!shouldInjectOnFirst && !shouldReinject)
-            return;
+        let shouldReinject = false;
+        let currentFactUuids = null;
+        if (!shouldInjectOnFirst) {
+            const driftFacts = await client.searchFacts({
+                query: messageText,
+                groupIds: [state.groupId],
+                maxFacts: 20,
+            });
+            currentFactUuids = new Set(driftFacts.map((fact) => fact.uuid));
+            const similarity = computeJaccardSimilarity(currentFactUuids, state.lastInjectionFactUuids);
+            shouldReinject = similarity < driftThreshold;
+            if (!shouldReinject) {
+                logger_js_1.logger.debug("Skipping reinjection; drift above threshold", {
+                    sessionID,
+                    similarity,
+                });
+                return;
+            }
+        }
         try {
-            const prefix = shouldReinject ? "graphiti-refresh-" : "graphiti-memory-";
             const useUserScope = shouldInjectOnFirst;
             const characterBudget = (0, context_limit_js_1.calculateInjectionBudget)(state.contextLimit);
-            await injectMemoryContext(state, messageText, output, prefix, useUserScope, characterBudget, shouldReinject);
+            await searchAndCacheMemoryContext(state, messageText, useUserScope, characterBudget, currentFactUuids);
             state.injectedMemories = true;
-            state.lastInjectionMessageCount = state.messageCount;
         }
         catch (err) {
             logger_js_1.logger.error("Failed to inject memories:", err);

package/script/src/handlers/compacting.d.ts

@@ -9,6 +9,7 @@ export interface CompactingHandlerDeps {
     sessionManager: SessionManager;
     client: GraphitiClient;
     defaultGroupId: string;
+    factStaleDays: number;
 }
 /** Creates the `experimental.session.compacting` hook handler. */
 export declare function createCompactingHandler(deps: CompactingHandlerDeps): ({ sessionID }: CompactingInput, output: CompactingOutput) => Promise<void>;

package/script/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;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAI5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpD,KAAK,cAAc,GAAG,WAAW,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;AAC5E,KAAK,eAAe,GAAG,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;AACrD,KAAK,gBAAgB,GAAG,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;AAEtD,+CAA+C;AAC/C,MAAM,WAAW,qBAAqB;IACpC,cAAc,EAAE,cAAc,CAAC;IAC/B,MAAM,EAAE,cAAc,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;CACxB;AAED,kEAAkE;AAClE,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,qBAAqB,IAI/D,eAAe,eAAe,EAC9B,QAAQ,gBAAgB,mBAyB3B"}
+{"version":3,"file":"compacting.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/compacting.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAI5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpD,KAAK,cAAc,GAAG,WAAW,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;AAC5E,KAAK,eAAe,GAAG,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;AACrD,KAAK,gBAAgB,GAAG,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;AAEtD,+CAA+C;AAC/C,MAAM,WAAW,qBAAqB;IACpC,cAAc,EAAE,cAAc,CAAC;IAC/B,MAAM,EAAE,cAAc,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,kEAAkE;AAClE,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,qBAAqB,IAI/D,eAAe,eAAe,EAC9B,QAAQ,gBAAgB,mBA0B3B"}

package/script/src/handlers/compacting.js

@@ -6,7 +6,7 @@ const context_limit_js_1 = require("../services/context-limit.js");
 const logger_js_1 = require("../services/logger.js");
 /** Creates the `experimental.session.compacting` hook handler. */
 function createCompactingHandler(deps) {
-    const { sessionManager, client, defaultGroupId } = deps;
+    const { sessionManager, client, defaultGroupId, factStaleDays } = deps;
     return async ({ sessionID }, output) => {
         const state = sessionManager.getState(sessionID);
         if (!state?.isMain) {
@@ -23,6 +23,7 @@ function createCompactingHandler(deps) {
                 user: state.userGroupId,
             },
             contextStrings: output.context,
+            factStaleDays,
         });
         if (additionalContext.length > 0) {
             output.context.push(...additionalContext);

package/script/src/handlers/event.d.ts

@@ -1,6 +1,6 @@
 import type { Hooks } from "@opencode-ai/plugin";
+import type { OpencodeClient } from "@opencode-ai/sdk";
 import type { GraphitiClient } from "../services/client.js";
-import type { ProviderListClient } from "../services/context-limit.js";
 import type { SessionManager } from "../session.js";
 type EventHook = NonNullable<Hooks["event"]>;
 type EventInput = Parameters<EventHook>[0];
@@ -9,7 +9,7 @@ export interface EventHandlerDeps {
     sessionManager: SessionManager;
     client: GraphitiClient;
     defaultGroupId: string;
-    sdkClient: ProviderListClient;
+    sdkClient: OpencodeClient;
     directory: string;
     groupIdPrefix: string;
 }

package/script/src/handlers/event.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"event.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/event.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE5D,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAGvE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAGpD,KAAK,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;AAC7C,KAAK,UAAU,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;AAE3C,0CAA0C;AAC1C,MAAM,WAAW,gBAAgB;IAC/B,cAAc,EAAE,cAAc,CAAC;IAC/B,MAAM,EAAE,cAAc,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,kBAAkB,CAAC;IAC9B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,wCAAwC;AACxC,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,gBAAgB,IAWzC,WAAW,UAAU,mBA+JpC"}
+{"version":3,"file":"event.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/event.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAI5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAGpD,KAAK,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;AAC7C,KAAK,UAAU,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;AAE3C,0CAA0C;AAC1C,MAAM,WAAW,gBAAgB;IAC/B,cAAc,EAAE,cAAc,CAAC;IAC/B,MAAM,EAAE,cAAc,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,cAAc,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,wCAAwC;AACxC,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,gBAAgB,IAkDzC,WAAW,UAAU,mBAkLpC"}

package/script/src/handlers/event.js

@@ -9,6 +9,41 @@ const utils_js_1 = require("../utils.js");
 function createEventHandler(deps) {
     const { sessionManager, client, defaultGroupId, sdkClient, directory, groupIdPrefix, } = deps;
     const defaultUserGroupId = (0, utils_js_1.makeUserGroupId)(groupIdPrefix);
+    const buildSessionSnapshot = (sessionId, messages) => {
+        const recentMessages = messages.slice(-12);
+        const recentAssistant = [...recentMessages]
+            .reverse()
+            .find((message) => message.startsWith("Assistant:"))
+            ?.replace(/^Assistant:\s*/, "")
+            .trim();
+        const recentUser = [...recentMessages]
+            .reverse()
+            .find((message) => message.startsWith("User:"))
+            ?.replace(/^User:\s*/, "")
+            .trim();
+        const questionRegex = /[^\n\r?]{3,200}\?/g;
+        const questions = recentMessages
+            .flatMap((message) => {
+            const text = message.replace(/^(User|Assistant):\s*/, "");
+            return text.match(questionRegex) ?? [];
+        })
+            .map((question) => question.trim());
+        const uniqueQuestions = Array.from(new Set(questions)).slice(0, 6);
+        const lines = [];
+        lines.push(`Session ${sessionId} working snapshot`);
+        if (recentUser)
+            lines.push(`Recent user focus: ${recentUser}`);
+        if (recentAssistant) {
+            lines.push(`Recent assistant focus: ${recentAssistant}`);
+        }
+        if (uniqueQuestions.length > 0) {
+            lines.push("Open questions:");
+            for (const question of uniqueQuestions) {
+                lines.push(`- ${question}`);
+            }
+        }
+        return lines.join("\n");
+    };
     return async ({ event }) => {
         try {
             if (event.type === "session.created") {
@@ -27,7 +62,7 @@ function createEventHandler(deps) {
                         groupId: defaultGroupId,
                         userGroupId: defaultUserGroupId,
                         injectedMemories: false,
-                        lastInjectionMessageCount: 0,
+                        lastInjectionFactUuids: new Set(),
                         messageCount: 0,
                         pendingMessages: [],
                         contextLimit: 200_000,
@@ -74,6 +109,22 @@ function createEventHandler(deps) {
                     logger_js_1.logger.debug("Ignoring non-main idle session:", sessionId);
                     return;
                 }
+                try {
+                    const snapshotContent = buildSessionSnapshot(sessionId, state.pendingMessages);
+                    if (snapshotContent.trim()) {
+                        await client.addEpisode({
+                            name: `Snapshot: ${sessionId}`,
+                            episodeBody: snapshotContent,
+                            groupId: state.groupId,
+                            source: "text",
+                            sourceDescription: "session-snapshot",
+                        });
+                        logger_js_1.logger.info("Saved session snapshot", { sessionId });
+                    }
+                }
+                catch (err) {
+                    logger_js_1.logger.error("Failed to save session snapshot", { sessionId, err });
+                }
                 await sessionManager.flushPendingMessages(sessionId, "Buffered messages from OpenCode session", 50);
                 return;
             }

package/script/src/handlers/system.d.ts

@@ -0,0 +1,11 @@
+import type { Hooks } from "@opencode-ai/plugin";
+import type { SessionManager } from "../session.js";
+type SystemTransformHook = NonNullable<Hooks["experimental.chat.system.transform"]>;
+type SystemTransformInput = Parameters<SystemTransformHook>[0];
+type SystemTransformOutput = Parameters<SystemTransformHook>[1];
+export interface SystemHandlerDeps {
+    sessionManager: SessionManager;
+}
+export declare function createSystemHandler(deps: SystemHandlerDeps): ({ sessionID }: SystemTransformInput, output: SystemTransformOutput) => Promise<void>;
+export {};
+//# sourceMappingURL=system.d.ts.map

package/script/src/handlers/system.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"system.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/system.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpD,KAAK,mBAAmB,GAAG,WAAW,CACpC,KAAK,CAAC,oCAAoC,CAAC,CAC5C,CAAC;AACF,KAAK,oBAAoB,GAAG,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;AAC/D,KAAK,qBAAqB,GAAG,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;AAEhE,MAAM,WAAW,iBAAiB;IAChC,cAAc,EAAE,cAAc,CAAC;CAChC;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,iBAAiB,IAMvD,eAAe,oBAAoB,EACnC,QAAQ,qBAAqB,mBAYhC"}

package/script/src/handlers/system.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createSystemHandler = createSystemHandler;
+const logger_js_1 = require("../services/logger.js");
+function createSystemHandler(deps) {
+    const { sessionManager } = deps;
+    // Assumes chat.message hook completes before system.transform fires for the same turn.
+    // deno-lint-ignore require-await
+    return async ({ sessionID }, output) => {
+        if (!sessionID)
+            return;
+        const state = sessionManager.getState(sessionID);
+        if (!state?.isMain)
+            return;
+        if (!state.cachedMemoryContext)
+            return;
+        output.system.push(state.cachedMemoryContext);
+        state.cachedMemoryContext = undefined;
+        logger_js_1.logger.info("Injected memory context into system prompt");
+    };
+}

package/script/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;AAW/D;;GAEG;AACH,eAAO,MAAM,QAAQ,EAAE,MAkDtB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAe,MAAM,qBAAqB,CAAC;AAW/D;;GAEG;AACH,eAAO,MAAM,QAAQ,EAAE,MAuDtB,CAAC"}

package/script/src/index.js

@@ -5,6 +5,7 @@ const config_js_1 = require("./config.js");
 const chat_js_1 = require("./handlers/chat.js");
 const compacting_js_1 = require("./handlers/compacting.js");
 const event_js_1 = require("./handlers/event.js");
+const system_js_1 = require("./handlers/system.js");
 const client_js_1 = require("./services/client.js");
 const logger_js_1 = require("./services/logger.js");
 const session_js_1 = require("./session.js");
@@ -30,19 +31,24 @@ const graphiti = async (input) => {
             sessionManager,
             client,
             defaultGroupId,
-            sdkClient: sdkClient,
+            sdkClient,
             directory: input.directory,
             groupIdPrefix: config.groupIdPrefix,
         }),
         "chat.message": (0, chat_js_1.createChatHandler)({
             sessionManager,
-            injectionInterval: config.injectionInterval,
+            driftThreshold: config.driftThreshold,
+            factStaleDays: config.factStaleDays,
             client,
         }),
         "experimental.session.compacting": (0, compacting_js_1.createCompactingHandler)({
             sessionManager,
             client,
             defaultGroupId,
+            factStaleDays: config.factStaleDays,
+        }),
+        "experimental.chat.system.transform": (0, system_js_1.createSystemHandler)({
+            sessionManager,
         }),
     };
 };

package/script/src/services/client.d.ts

@@ -1,4 +1,4 @@
-import type { GraphitiFact, GraphitiNode } from "../types/index.js";
+import type { GraphitiEpisode, GraphitiFact, GraphitiNode } from "../types/index.js";
 /**
  * Graphiti MCP client wrapper for connecting, querying,
  * and persisting episodes with basic reconnection handling.
@@ -57,6 +57,13 @@ export declare class GraphitiClient {
         groupIds?: string[];
         maxNodes?: number;
     }): Promise<GraphitiNode[]>;
+    /**
+     * Retrieve recent episodes for a group.
+     */
+    getEpisodes(params: {
+        groupId?: string;
+        lastN?: number;
+    }): Promise<GraphitiEpisode[]>;
     /**
      * Check whether the Graphiti MCP server is reachable.
      */

package/script/src/services/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../src/src/services/client.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,YAAY,EAEZ,YAAY,EAEb,MAAM,mBAAmB,CAAC;AAG3B;;;GAGG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAgC;IACjD,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,QAAQ,CAAS;IAEzB;;OAEG;gBACS,QAAQ,EAAE,MAAM;IAS5B,oDAAoD;IACpD,OAAO,CAAC,wBAAwB;IAUhC;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;IAgBjC;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;YAOnB,QAAQ;IAkCtB,OAAO,CAAC,gBAAgB;YASV,SAAS;IAavB;;;OAGG;IACH,eAAe,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO;IAyBzC;;OAEG;IACG,UAAU,CAAC,MAAM,EAAE;QACvB,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;QACrC,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B,GAAG,OAAO,CAAC,IAAI,CAAC;IAWjB;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAsB3B;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAsB3B;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;CAQpC"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../src/src/services/client.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EACV,eAAe,EACf,YAAY,EAEZ,YAAY,EAEb,MAAM,mBAAmB,CAAC;AAG3B;;;GAGG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAgC;IACjD,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,QAAQ,CAAS;IAEzB;;OAEG;gBACS,QAAQ,EAAE,MAAM;IAS5B,oDAAoD;IACpD,OAAO,CAAC,wBAAwB;IAUhC;;;OAGG;IACG,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC;IAgBjC;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;YAOnB,QAAQ;IAkCtB,OAAO,CAAC,gBAAgB;YASV,SAAS;IAavB;;;OAGG;IACH,eAAe,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO;IAyBzC;;OAEG;IACG,UAAU,CAAC,MAAM,EAAE;QACvB,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;QACrC,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B,GAAG,OAAO,CAAC,IAAI,CAAC;IAWjB;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAsB3B;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE;QACxB,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAsB3B;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE;QACxB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAqB9B;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;CAQpC"}

package/script/src/services/client.js

@@ -221,6 +221,29 @@ class GraphitiClient {
             return [];
         }
     }
+    /**
+     * Retrieve recent episodes for a group.
+     */
+    async getEpisodes(params) {
+        try {
+            const result = await this.callTool("get_episodes", {
+                group_id: params.groupId,
+                last_n: params.lastN,
+            });
+            if (Array.isArray(result))
+                return result;
+            if (result &&
+                typeof result === "object" &&
+                Array.isArray(result.episodes)) {
+                return result.episodes;
+            }
+            return [];
+        }
+        catch (err) {
+            logger_js_1.logger.error("getEpisodes error:", err);
+            return [];
+        }
+    }
     /**
      * Check whether the Graphiti MCP server is reachable.
      */

package/script/src/services/compaction.d.ts

@@ -1,4 +1,13 @@
 import type { GraphitiFact, GraphitiNode } from "../types/index.js";
+export declare const classifyFacts: (facts: GraphitiFact[], now: Date) => {
+    decisions: GraphitiFact[];
+    active: GraphitiFact[];
+    background: GraphitiFact[];
+};
+export declare const takeFactsWithinBudget: (facts: GraphitiFact[], budget: number, formatOptions: {
+    factStaleDays: number;
+    now: Date;
+}) => GraphitiFact[];
 /**
  * Persist a compaction summary episode when enabled.
  */
@@ -38,5 +47,6 @@ export declare function getCompactionContext(params: {
         user?: string;
     };
     contextStrings: string[];
+    factStaleDays?: number;
 }): Promise<string[]>;
 //# sourceMappingURL=compaction.d.ts.map

package/script/src/services/compaction.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"compaction.d.ts","sourceRoot":"","sources":["../../../src/src/services/compaction.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAIpE;;GAEG;AACH,wBAAsB,gBAAgB,CAAC,MAAM,EAAE;IAC7C,MAAM,EAAE;QACN,UAAU,EAAE,CAAC,IAAI,EAAE;YACjB,IAAI,EAAE,MAAM,CAAC;YACb,WAAW,EAAE,MAAM,CAAC;YACpB,OAAO,CAAC,EAAE,MAAM,CAAC;YACjB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;YACrC,iBAAiB,CAAC,EAAE,MAAM,CAAC;SAC5B,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACrB,CAAC;IACF,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB,GAAG,OAAO,CAAC,IAAI,CAAC,CAiBhB;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CAAC,MAAM,EAAE;IACjD,MAAM,EAAE;QACN,WAAW,EAAE,CAAC,IAAI,EAAE;YAClB,KAAK,EAAE,MAAM,CAAC;YACd,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;YACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;SACnB,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;QAC9B,WAAW,EAAE,CAAC,IAAI,EAAE;YAClB,KAAK,EAAE,MAAM,CAAC;YACd,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;YACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;SACnB,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;KAC/B,CAAC;IACF,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE;QACR,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;IACF,cAAc,EAAE,MAAM,EAAE,CAAC;CAC1B,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAgIpB"}
+{"version":3,"file":"compaction.d.ts","sourceRoot":"","sources":["../../../src/src/services/compaction.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AA0BpE,eAAO,MAAM,aAAa,GACxB,OAAO,YAAY,EAAE,EACrB,KAAK,IAAI,KACR;IACD,SAAS,EAAE,YAAY,EAAE,CAAC;IAC1B,MAAM,EAAE,YAAY,EAAE,CAAC;IACvB,UAAU,EAAE,YAAY,EAAE,CAAC;CA4B5B,CAAC;AAEF,eAAO,MAAM,qBAAqB,GAChC,OAAO,YAAY,EAAE,EACrB,QAAQ,MAAM,EACd,eAAe;IAAE,aAAa,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,IAAI,CAAA;CAAE,KAClD,YAAY,EAoBd,CAAC;AAEF;;GAEG;AACH,wBAAsB,gBAAgB,CAAC,MAAM,EAAE;IAC7C,MAAM,EAAE;QACN,UAAU,EAAE,CAAC,IAAI,EAAE;YACjB,IAAI,EAAE,MAAM,CAAC;YACb,WAAW,EAAE,MAAM,CAAC;YACpB,OAAO,CAAC,EAAE,MAAM,CAAC;YACjB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;YACrC,iBAAiB,CAAC,EAAE,MAAM,CAAC;SAC5B,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KACrB,CAAC;IACF,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB,GAAG,OAAO,CAAC,IAAI,CAAC,CAiBhB;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CAAC,MAAM,EAAE;IACjD,MAAM,EAAE;QACN,WAAW,EAAE,CAAC,IAAI,EAAE;YAClB,KAAK,EAAE,MAAM,CAAC;YACd,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;YACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;SACnB,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;QAC9B,WAAW,EAAE,CAAC,IAAI,EAAE;YAClB,KAAK,EAAE,MAAM,CAAC;YACd,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;YACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;SACnB,KAAK,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;KAC/B,CAAC;IACF,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE;QACR,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;IACF,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CA0KpB"}

package/script/src/services/compaction.js

@@ -1,9 +1,76 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.takeFactsWithinBudget = exports.classifyFacts = void 0;
 exports.handleCompaction = handleCompaction;
 exports.getCompactionContext = getCompactionContext;
 const context_js_1 = require("./context.js");
 const logger_js_1 = require("./logger.js");
+const DAY_MS = 24 * 60 * 60 * 1000;
+const DECISION_KEYWORDS = [
+    "decided",
+    "must",
+    "should",
+    "prefer",
+    "constraint",
+    "require",
+    "chose",
+    "always",
+    "never",
+    "schema",
+    "architecture",
+    "agreed",
+    "design",
+    "selected",
+];
+const classifyFacts = (facts, now) => {
+    const decisions = [];
+    const active = [];
+    const background = [];
+    const cutoff = now.getTime() - 7 * DAY_MS;
+    for (const fact of facts) {
+        const text = fact.fact.toLowerCase();
+        // Use word boundary regex to match whole words only
+        const hasDecisionKeyword = DECISION_KEYWORDS.some((keyword) => {
+            const regex = new RegExp(`\\b${keyword}\\b`, "i");
+            return regex.test(text);
+        });
+        if (hasDecisionKeyword) {
+            decisions.push(fact);
+            continue;
+        }
+        const validAt = fact.valid_at ? Date.parse(fact.valid_at) : NaN;
+        if (!Number.isNaN(validAt) && validAt >= cutoff) {
+            active.push(fact);
+            continue;
+        }
+        background.push(fact);
+    }
+    return { decisions, active, background };
+};
+exports.classifyFacts = classifyFacts;
+const takeFactsWithinBudget = (facts, budget, formatOptions) => {
+    if (budget <= 0 || facts.length === 0)
+        return [];
+    const classified = (0, exports.classifyFacts)(facts, formatOptions.now);
+    const prioritized = [
+        ...classified.decisions,
+        ...classified.active,
+        ...classified.background,
+    ];
+    const lines = (0, context_js_1.formatFactLines)(prioritized, formatOptions);
+    const selected = [];
+    let remaining = budget;
+    for (let i = 0; i < lines.length; i++) {
+        const line = lines[i];
+        const length = line.length + 1;
+        if (remaining - length < 0)
+            continue;
+        selected.push(prioritized[i]);
+        remaining -= length;
+    }
+    return selected;
+};
+exports.takeFactsWithinBudget = takeFactsWithinBudget;
 /**
  * Persist a compaction summary episode when enabled.
  */
@@ -30,6 +97,8 @@ async function handleCompaction(params) {
  */
 async function getCompactionContext(params) {
     const { client, characterBudget, groupIds, contextStrings } = params;
+    const now = new Date();
+    const factStaleDays = params.factStaleDays ?? 30;
     try {
         const queryText = contextStrings.slice(0, 3).join(" ").slice(0, 500);
         if (!queryText.trim())
@@ -70,14 +139,40 @@ async function getCompactionContext(params) {
             userFacts.length === 0 && userNodes.length === 0) {
             return [];
         }
-        const buildSection = (header, facts, nodes) => {
+        const formatOptions = { factStaleDays, now };
+        const projectContext = (0, context_js_1.deduplicateContext)({
+            facts: projectFacts,
+            nodes: projectNodes,
+        });
+        const userContext = (0, context_js_1.deduplicateContext)({
+            facts: userFacts,
+            nodes: userNodes,
+        });
+        const buildSection = (header, facts, nodes, budget) => {
             const lines = [];
             lines.push(header);
             lines.push("<instruction>Background context only; do not reference in titles, summaries, or opening responses unless directly relevant.</instruction>");
-            if (facts.length > 0) {
-                lines.push("<facts>");
-                lines.push(...(0, context_js_1.formatFactLines)(facts));
-                lines.push("</facts>");
+            const classified = (0, exports.classifyFacts)(facts, now);
+            const decisionBudget = Math.floor(budget * 0.4);
+            const activeBudget = Math.floor(budget * 0.35);
+            const backgroundBudget = budget - decisionBudget - activeBudget;
+            const selectedDecisions = (0, exports.takeFactsWithinBudget)(classified.decisions, decisionBudget, formatOptions);
+            const selectedActive = (0, exports.takeFactsWithinBudget)(classified.active, activeBudget, formatOptions);
+            const selectedBackground = (0, exports.takeFactsWithinBudget)(classified.background, backgroundBudget, formatOptions);
+            if (selectedDecisions.length > 0) {
+                lines.push("<decisions>");
+                lines.push(...(0, context_js_1.formatFactLines)(selectedDecisions, formatOptions));
+                lines.push("</decisions>");
+            }
+            if (selectedActive.length > 0) {
+                lines.push("<active_context>");
+                lines.push(...(0, context_js_1.formatFactLines)(selectedActive, formatOptions));
+                lines.push("</active_context>");
+            }
+            if (selectedBackground.length > 0) {
+                lines.push("<background>");
+                lines.push(...(0, context_js_1.formatFactLines)(selectedBackground, formatOptions));
+                lines.push("</background>");
             }
             if (nodes.length > 0) {
                 lines.push("<nodes>");
@@ -86,25 +181,19 @@ async function getCompactionContext(params) {
             }
             return lines.join("\n");
         };
-        const projectSection = buildSection('<memory source="project">', projectFacts, projectNodes);
-        const userSection = buildSection('<memory source="user">', userFacts, userNodes);
         const headerLines = [
             "<summary>",
-            "<current_goal>",
-            "- ",
-            "</current_goal>",
-            "",
-            "<work_completed>",
+            "<decisions>",
             "- ",
-            "</work_completed>",
+            "</decisions>",
             "",
-            "<remaining_tasks>",
+            "<active_context>",
             "- ",
-            "</remaining_tasks>",
+            "</active_context>",
             "",
-            "<constraints_decisions>",
+            "<background>",
             "- ",
-            "</constraints_decisions>",
+            "</background>",
             "",
             "<persistent_memory>",
         ];
@@ -113,6 +202,8 @@ async function getCompactionContext(params) {
         const remainingBudget = Math.max(characterBudget - base.length, 0);
         const projectBudget = Math.floor(remainingBudget * 0.7);
         const userBudget = remainingBudget - projectBudget;
+        const projectSection = buildSection('<memory source="project">', projectContext.facts, projectContext.nodes, projectBudget);
+        const userSection = buildSection('<memory source="user">', userContext.facts, userContext.nodes, userBudget);
         const truncatedProject = projectSection.slice(0, projectBudget);
         const truncatedUser = userSection.slice(0, userBudget);
         const sections = [header];