opencode-graphiti 0.1.5 → 0.1.7

package/README.md

@@ -22,12 +22,16 @@ reminded of recent project context — regardless of what survived the summary.
 
 This plugin connects to a Graphiti MCP server and:
 
-- Injects relevant memories into the first user message of each session
-- Periodically re-injects project memories as the conversation progresses
+- Searches Graphiti for relevant facts and entities on each user message
+- Injects memories into the system prompt via
+  `experimental.chat.system.transform`, keeping session titles clean
+- Detects context drift using Jaccard similarity and re-injects when the
+  conversation topic shifts
 - Buffers user and assistant messages, flushing them to Graphiti on idle or
   before compaction
 - Preserves key facts during context compaction
 - Saves compaction summaries as episodes so knowledge survives across boundaries
+- Annotates stale facts and filters expired ones automatically
 - Scopes memories per project (and per user) using directory-based group IDs
 
 ## Prerequisites
@@ -104,8 +108,12 @@ Create a config file at `~/.config/opencode/graphiti.jsonc`:
   // Prefix for project group IDs (e.g. "opencode-my-project")
   "groupIdPrefix": "opencode",
 
-  // Number of user messages between memory re-injections (0 = disabled)
-  "injectionInterval": 10
+  // Jaccard similarity threshold (0–1) below which memory is re-injected
+  // Lower values mean the topic must drift further before re-injection
+  "driftThreshold": 0.5,
+
+  // Number of days after which facts are annotated as stale
+  "factStaleDays": 30
 }
 ```
 
@@ -114,27 +122,36 @@ values.
 
 ## How It Works
 
-### Memory Injection (`chat.message`)
+### Memory Search and Caching (`chat.message`)
 
-On the first user message in a session, the plugin searches Graphiti for facts
-and entities relevant to the message content. Results are split into project and
-user scopes (70% / 30% budget split), formatted in XML-style `<memory>` blocks
-with explicit de-emphasis instructions, and injected into the conversation.
+On each user message the plugin searches Graphiti for facts and entities
+relevant to the message content. Results are split into project and user scopes
+(70% / 30% budget), deduplicated, filtered for validity, annotated with
+staleness if older than `factStaleDays`, and formatted as Markdown. The
+formatted context is cached on the session state for the system prompt hook to
+pick up.
 
-Memory is injected via `output.message.system` (system-level instruction) when
-available, which prevents memory from influencing session titles. If the system
-field is unavailable, the plugin falls back to prepending synthetic parts.
+On the very first message of a session, the plugin also loads the most recent
+session snapshot episode to prime the conversation with prior context.
 
 The injection budget is calculated dynamically: 5% of the model's context limit
 (resolved from the provider list) multiplied by 4 characters per token.
 
-### Re-injection (`chat.message`)
+### System Prompt Injection (`experimental.chat.system.transform`)
+
+A separate hook reads the cached memory context and appends it to the system
+prompt array. This approach keeps memory out of user message parts, which
+prevents it from influencing session titles or being treated as user
+instructions. The cache is cleared after injection so stale context is not
+re-injected on subsequent LLM calls within the same turn.
+
+### Drift-Based Re-injection (`chat.message`)
 
-When `injectionInterval` is greater than 0, the plugin periodically re-injects
-project-scoped memories as the conversation progresses. After every N user
-messages (where N is `injectionInterval`), stale synthetic memory parts are
-replaced with fresh results. Re-injection uses the full character budget for
-project memories only (no user scope).
+After the first injection, the plugin monitors for context drift on every user
+message. It searches Graphiti for the current message and compares the returned
+fact UUIDs against the previously injected set using Jaccard similarity. When
+similarity drops below `driftThreshold` (default 0.5), the memory cache is
+refreshed with project-scoped results only (no user scope).
 
 ### Message Buffering (`event`)
 

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

@@ -1 +1 @@
-{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/src/config.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAcvD;;GAEG;AACH,wBAAgB,UAAU,IAAI,cAAc,CAc3C"}
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/src/config.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAgBvD;;GAEG;AACH,wBAAgB,UAAU,IAAI,cAAc,CAc3C"}

package/esm/src/config.js

@@ -3,12 +3,14 @@ import * as z from "zod/mini";
 const DEFAULT_CONFIG = {
     endpoint: "http://localhost:8000/mcp",
     groupIdPrefix: "opencode",
-    injectionInterval: 10,
+    driftThreshold: 0.5,
+    factStaleDays: 30,
 };
 const GraphitiConfigSchema = z.object({
     endpoint: z.string(),
     groupIdPrefix: z.string(),
-    injectionInterval: z.number(),
+    driftThreshold: z.number(),
+    factStaleDays: z.number(),
 });
 /**
  * Load Graphiti configuration from JSONC files with defaults applied.

package/esm/src/handlers/chat.d.ts

@@ -7,7 +7,8 @@ type ChatMessageOutput = Parameters<ChatMessageHook>[1];
 /** Dependencies for the chat message handler. */
 export interface ChatHandlerDeps {
     sessionManager: SessionManager;
-    injectionInterval: number;
+    driftThreshold: number;
+    factStaleDays: number;
     client: GraphitiClient;
 }
 /** Creates the `chat.message` hook handler. */

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

@@ -1 +1 @@
-{"version":3,"file":"chat.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/chat.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAI5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAGpD,KAAK,eAAe,GAAG,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;AAC1D,KAAK,gBAAgB,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;AACvD,KAAK,iBAAiB,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;AAExD,iDAAiD;AACjD,MAAM,WAAW,eAAe;IAC9B,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,MAAM,EAAE,cAAc,CAAC;CACxB;AAED,+CAA+C;AAC/C,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,eAAe,IAyGvC,eAAe,gBAAgB,EAAE,QAAQ,iBAAiB,mBAuDzE"}
+{"version":3,"file":"chat.d.ts","sourceRoot":"","sources":["../../../src/src/handlers/chat.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAO5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAGpD,KAAK,eAAe,GAAG,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;AAC1D,KAAK,gBAAgB,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;AACvD,KAAK,iBAAiB,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;AAExD,iDAAiD;AACjD,MAAM,WAAW,eAAe;IAC9B,cAAc,EAAE,cAAc,CAAC;IAC/B,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,cAAc,CAAC;CACxB;AAED,+CAA+C;AAC/C,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,eAAe,IAiJvC,eAAe,gBAAgB,EAAE,QAAQ,iBAAiB,mBA6EzE"}

package/esm/src/handlers/chat.js

@@ -1,20 +1,11 @@
 import { calculateInjectionBudget } from "../services/context-limit.js";
-import { formatMemoryContext } from "../services/context.js";
+import { deduplicateContext, formatMemoryContext, } from "../services/context.js";
 import { logger } from "../services/logger.js";
 import { extractTextFromParts } from "../utils.js";
 /** Creates the `chat.message` hook handler. */
 export 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,
@@ -47,45 +38,85 @@ export function createChatHandler(deps) {
             userFactsPromise,
             userNodesPromise,
         ]);
-        const projectContext = formatMemoryContext(projectFacts, projectNodes);
-        const userContext = formatMemoryContext(userFacts, userNodes);
-        if (!projectContext && !userContext)
+        const projectContext = deduplicateContext({
+            facts: projectFacts,
+            nodes: projectNodes,
+        });
+        const userContext = deduplicateContext({
+            facts: userFacts,
+            nodes: userNodes,
+        });
+        const projectContextString = formatMemoryContext(projectContext.facts, projectContext.nodes, { factStaleDays });
+        const userContextString = 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.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 allFactUuids = [
+            ...projectContext.facts.map((fact) => fact.uuid),
+            ...userContext.facts.map((fact) => fact.uuid),
+        ];
+        const factUuids = seedFactUuids ?? Array.from(new Set(allFactUuids));
+        state.cachedMemoryContext = memoryContext;
+        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.length === 0 && right.length === 0)
+            return 1;
+        const leftSet = new Set(left);
+        const rightSet = new Set(right);
+        let intersection = 0;
+        for (const value of leftSet) {
+            if (rightSet.has(value))
+                intersection += 1;
         }
-        logger.info(`Injected ${projectFacts.length + userFacts.length} facts and ${projectNodes.length + userNodes.length} nodes`);
+        const union = leftSet.size + rightSet.size - intersection;
+        return union === 0 ? 1 : intersection / union;
     };
     return async ({ sessionID }, output) => {
         if (await sessionManager.isSubagentSession(sessionID)) {
@@ -112,19 +143,39 @@ export 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) {
+            try {
+                const driftFacts = await client.searchFacts({
+                    query: messageText,
+                    groupIds: [state.groupId],
+                    maxFacts: 20,
+                });
+                currentFactUuids = driftFacts.map((fact) => fact.uuid);
+                const similarity = computeJaccardSimilarity(currentFactUuids, state.lastInjectionFactUuids);
+                shouldReinject = similarity < driftThreshold;
+                if (!shouldReinject) {
+                    logger.debug("Skipping reinjection; similarity above threshold", {
+                        sessionID,
+                        similarity,
+                    });
+                    return;
+                }
+            }
+            catch (err) {
+                logger.error("Failed to check topic drift, skipping reinjection", {
+                    sessionID,
+                    err,
+                });
+                return;
+            }
+        }
         try {
-            const prefix = shouldReinject ? "graphiti-refresh-" : "graphiti-memory-";
             const useUserScope = shouldInjectOnFirst;
             const characterBudget = 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.error("Failed to inject memories:", err);

package/esm/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/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;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/esm/src/handlers/compacting.js

@@ -3,7 +3,7 @@ import { calculateInjectionBudget } from "../services/context-limit.js";
 import { logger } from "../services/logger.js";
 /** Creates the `experimental.session.compacting` hook handler. */
 export 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) {
@@ -20,6 +20,7 @@ export function createCompactingHandler(deps) {
                 user: state.userGroupId,
             },
             contextStrings: output.context,
+            factStaleDays,
         });
         if (additionalContext.length > 0) {
             output.context.push(...additionalContext);

package/esm/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/esm/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/esm/src/handlers/event.js

@@ -6,6 +6,41 @@ import { isTextPart, makeUserGroupId } from "../utils.js";
 export function createEventHandler(deps) {
     const { sessionManager, client, defaultGroupId, sdkClient, directory, groupIdPrefix, } = deps;
     const defaultUserGroupId = 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") {
@@ -24,7 +59,7 @@ export function createEventHandler(deps) {
                         groupId: defaultGroupId,
                         userGroupId: defaultUserGroupId,
                         injectedMemories: false,
-                        lastInjectionMessageCount: 0,
+                        lastInjectionFactUuids: [],
                         messageCount: 0,
                         pendingMessages: [],
                         contextLimit: 200_000,
@@ -71,6 +106,22 @@ export function createEventHandler(deps) {
                     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.info("Saved session snapshot", { sessionId });
+                    }
+                }
+                catch (err) {
+                    logger.error("Failed to save session snapshot", { sessionId, err });
+                }
                 await sessionManager.flushPendingMessages(sessionId, "Buffered messages from OpenCode session", 50);
                 return;
             }

package/esm/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/esm/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/esm/src/handlers/system.js

@@ -0,0 +1,18 @@
+import { logger } from "../services/logger.js";
+export 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.info("Injected memory context into system prompt");
+    };
+}

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;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/esm/src/index.js

@@ -2,6 +2,7 @@ import { loadConfig } from "./config.js";
 import { createChatHandler } from "./handlers/chat.js";
 import { createCompactingHandler } from "./handlers/compacting.js";
 import { createEventHandler } from "./handlers/event.js";
+import { createSystemHandler } from "./handlers/system.js";
 import { GraphitiClient } from "./services/client.js";
 import { logger } from "./services/logger.js";
 import { SessionManager } from "./session.js";
@@ -27,19 +28,24 @@ export const graphiti = async (input) => {
             sessionManager,
             client,
             defaultGroupId,
-            sdkClient: sdkClient,
+            sdkClient,
             directory: input.directory,
             groupIdPrefix: config.groupIdPrefix,
         }),
         "chat.message": createChatHandler({
             sessionManager,
-            injectionInterval: config.injectionInterval,
+            driftThreshold: config.driftThreshold,
+            factStaleDays: config.factStaleDays,
             client,
         }),
         "experimental.session.compacting": createCompactingHandler({
             sessionManager,
             client,
             defaultGroupId,
+            factStaleDays: config.factStaleDays,
+        }),
+        "experimental.chat.system.transform": createSystemHandler({
+            sessionManager,
         }),
     };
 };

package/esm/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/esm/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/esm/src/services/client.js

@@ -215,6 +215,29 @@ export 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.error("getEpisodes error:", err);
+            return [];
+        }
+    }
     /**
      * Check whether the Graphiti MCP server is reachable.
      */

package/esm/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/esm/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"}