opencode-graphiti 0.1.0 → 0.1.2

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

@@ -1,66 +1,7 @@
-import type { GraphitiConfig } from "../types/index.js";
-export interface CompactionDependencies {
-    sdkClient: {
-        session: {
-            summarize: (options: {
-                path: {
-                    id: string;
-                };
-                body?: {
-                    providerID: string;
-                    modelID: string;
-                };
-                query?: {
-                    directory?: string;
-                };
-            }) => Promise<unknown>;
-            promptAsync: (options: {
-                path: {
-                    id: string;
-                };
-                body?: {
-                    parts: Array<{
-                        type: "text";
-                        text: string;
-                    }>;
-                };
-                query?: {
-                    directory?: string;
-                };
-            }) => Promise<unknown>;
-        };
-        tui: {
-            showToast: (options?: {
-                body?: {
-                    title?: string;
-                    message: string;
-                    variant: "info" | "success" | "warning" | "error";
-                    duration?: number;
-                };
-                query?: {
-                    directory?: string;
-                };
-            }) => Promise<unknown>;
-        };
-        provider: {
-            list: (options?: {
-                directory?: string;
-            }) => Promise<unknown>;
-        };
-    };
-    directory: string;
-}
-export declare function createPreemptiveCompactionHandler(config: Pick<GraphitiConfig, "compactionThreshold" | "minTokensForCompaction" | "compactionCooldownMs" | "autoResumeAfterCompaction">, deps: CompactionDependencies): {
-    checkAndTriggerCompaction(sessionId: string, tokens: {
-        input: number;
-        output: number;
-        reasoning: number;
-        cache: {
-            read: number;
-            write: number;
-        };
-    }, providerID: string, modelID: string): Promise<void>;
-};
+import type { GraphitiFact, GraphitiNode } from "../types/index.js";
+/**
+ * Persist a compaction summary episode when enabled.
+ */
 export declare function handleCompaction(params: {
     client: {
         addEpisode: (args: {
@@ -71,23 +12,31 @@ export declare function handleCompaction(params: {
             sourceDescription?: string;
         }) => Promise<void>;
     };
-    config: GraphitiConfig;
     groupId: string;
     summary: string;
     sessionId: string;
 }): Promise<void>;
+/**
+ * Retrieve persistent fact context to include during compaction.
+ */
 export declare function getCompactionContext(params: {
     client: {
         searchFacts: (args: {
             query: string;
             groupIds?: string[];
             maxFacts?: number;
-        }) => Promise<Array<{
-            fact: string;
-        }>>;
+        }) => Promise<GraphitiFact[]>;
+        searchNodes: (args: {
+            query: string;
+            groupIds?: string[];
+            maxNodes?: number;
+        }) => Promise<GraphitiNode[]>;
+    };
+    characterBudget: number;
+    groupIds: {
+        project: string;
+        user?: string;
     };
-    config: GraphitiConfig;
-    groupId: string;
     contextStrings: string[];
 }): 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,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAGxD,MAAM,WAAW,sBAAsB;IACrC,SAAS,EAAE;QACT,OAAO,EAAE;YACP,SAAS,EAAE,CAAC,OAAO,EAAE;gBACnB,IAAI,EAAE;oBAAE,EAAE,EAAE,MAAM,CAAA;iBAAE,CAAC;gBACrB,IAAI,CAAC,EAAE;oBAAE,UAAU,EAAE,MAAM,CAAC;oBAAC,OAAO,EAAE,MAAM,CAAA;iBAAE,CAAC;gBAC/C,KAAK,CAAC,EAAE;oBAAE,SAAS,CAAC,EAAE,MAAM,CAAA;iBAAE,CAAC;aAChC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;YACvB,WAAW,EAAE,CAAC,OAAO,EAAE;gBACrB,IAAI,EAAE;oBAAE,EAAE,EAAE,MAAM,CAAA;iBAAE,CAAC;gBACrB,IAAI,CAAC,EAAE;oBAAE,KAAK,EAAE,KAAK,CAAC;wBAAE,IAAI,EAAE,MAAM,CAAC;wBAAC,IAAI,EAAE,MAAM,CAAA;qBAAE,CAAC,CAAA;iBAAE,CAAC;gBACxD,KAAK,CAAC,EAAE;oBAAE,SAAS,CAAC,EAAE,MAAM,CAAA;iBAAE,CAAC;aAChC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;SACxB,CAAC;QACF,GAAG,EAAE;YACH,SAAS,EAAE,CAAC,OAAO,CAAC,EAAE;gBACpB,IAAI,CAAC,EAAE;oBACL,KAAK,CAAC,EAAE,MAAM,CAAC;oBACf,OAAO,EAAE,MAAM,CAAC;oBAChB,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,GAAG,OAAO,CAAC;oBAClD,QAAQ,CAAC,EAAE,MAAM,CAAC;iBACnB,CAAC;gBACF,KAAK,CAAC,EAAE;oBAAE,SAAS,CAAC,EAAE,MAAM,CAAA;iBAAE,CAAC;aAChC,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;SACxB,CAAC;QACF,QAAQ,EAAE;YACR,IAAI,EAAE,CAAC,OAAO,CAAC,EAAE;gBAAE,SAAS,CAAC,EAAE,MAAM,CAAA;aAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;SAC9D,CAAC;KACH,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;CACnB;AAwDD,wBAAgB,iCAAiC,CAC/C,MAAM,EAAE,IAAI,CACV,cAAc,EACZ,qBAAqB,GACrB,wBAAwB,GACxB,sBAAsB,GACtB,2BAA2B,CAC9B,EACD,IAAI,EAAE,sBAAsB,GAC3B;IACD,yBAAyB,CACvB,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE;QACN,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,KAAK,EAAE,MAAM,CAAA;SAAE,CAAC;KACxC,EACD,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,IAAI,CAAC,CAAC;CAClB,CAiFA;AAED,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,MAAM,EAAE,cAAc,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB,GAAG,OAAO,CAAC,IAAI,CAAC,CAiBhB;AAED,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,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC,CAAC;KACxC,CAAC;IACF,MAAM,EAAE,cAAc,CAAC;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,cAAc,EAAE,MAAM,EAAE,CAAC;CAC1B,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAyBpB"}
+{"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,CA8GpB"}

package/esm/src/services/compaction.js

@@ -1,108 +1,11 @@
+import { formatFactLines, formatNodeLines } from "./context.js";
 import { logger } from "./logger.js";
-const DEFAULT_CONTEXT_LIMIT = 200_000;
-const RESUME_DELAY_MS = 500;
-const delay = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
-const buildModelKey = (providerID, modelID) => `${providerID}/${modelID}`;
-const resolveContextLimit = async (providerID, modelID, deps, state) => {
-    const modelKey = buildModelKey(providerID, modelID);
-    const cached = state.contextLimitCache.get(modelKey);
-    if (cached)
-        return cached;
-    try {
-        const providers = await deps.sdkClient.provider.list({
-            directory: deps.directory,
-        });
-        const list = providers.providers ?? [];
-        for (const provider of list) {
-            const providerInfo = provider;
-            if (providerInfo.id !== providerID)
-                continue;
-            const models = providerInfo.models ?? [];
-            for (const model of models) {
-                const modelInfo = model;
-                if (modelInfo.id !== modelID)
-                    continue;
-                const contextLimit = modelInfo.limit?.context;
-                if (typeof contextLimit === "number" && contextLimit > 0) {
-                    state.contextLimitCache.set(modelKey, contextLimit);
-                    return contextLimit;
-                }
-            }
-        }
-    }
-    catch (err) {
-        logger.warn("Failed to fetch provider context limit", err);
-    }
-    state.contextLimitCache.set(modelKey, DEFAULT_CONTEXT_LIMIT);
-    return DEFAULT_CONTEXT_LIMIT;
-};
-export function createPreemptiveCompactionHandler(config, deps) {
-    const state = {
-        lastCompactionTime: new Map(),
-        compactionInProgress: new Set(),
-        contextLimitCache: new Map(),
-    };
-    const checkAndTriggerCompaction = async (sessionId, tokens, providerID, modelID) => {
-        const totalTokens = tokens.input + tokens.cache.read + tokens.output +
-            tokens.reasoning;
-        if (totalTokens < (config.minTokensForCompaction ?? 0))
-            return;
-        if (state.compactionInProgress.has(sessionId))
-            return;
-        const lastCompaction = state.lastCompactionTime.get(sessionId) ?? 0;
-        if (Date.now() - lastCompaction < (config.compactionCooldownMs ?? 0)) {
-            return;
-        }
-        const contextLimit = await resolveContextLimit(providerID, modelID, deps, state);
-        const usageRatio = totalTokens / contextLimit;
-        if (usageRatio < (config.compactionThreshold ?? 1))
-            return;
-        state.compactionInProgress.add(sessionId);
-        try {
-            await deps.sdkClient.tui.showToast({
-                body: {
-                    title: "Graphiti",
-                    message: "Compacting session to preserve context...",
-                    variant: "info",
-                    duration: 3000,
-                },
-                query: { directory: deps.directory },
-            });
-            await deps.sdkClient.session.summarize({
-                path: { id: sessionId },
-                body: { providerID, modelID },
-                query: { directory: deps.directory },
-            });
-            state.lastCompactionTime.set(sessionId, Date.now());
-            if (config.autoResumeAfterCompaction) {
-                await delay(RESUME_DELAY_MS);
-                await deps.sdkClient.session.promptAsync({
-                    path: { id: sessionId },
-                    body: { parts: [{ type: "text", text: "Continue" }] },
-                    query: { directory: deps.directory },
-                });
-            }
-            logger.info("Preemptive compaction triggered", {
-                sessionId,
-                providerID,
-                modelID,
-                usageRatio,
-                totalTokens,
-                contextLimit,
-            });
-        }
-        catch (err) {
-            logger.error("Preemptive compaction failed", err);
-        }
-        finally {
-            state.compactionInProgress.delete(sessionId);
-        }
-    };
-    return { checkAndTriggerCompaction };
-}
+/**
+ * Persist a compaction summary episode when enabled.
+ */
 export async function handleCompaction(params) {
-    const { client, config, groupId, summary, sessionId } = params;
-    if (!config.enableCompactionSave || !summary)
+    const { client, groupId, summary, sessionId } = params;
+    if (!summary)
         return;
     try {
         await client.addEpisode({
@@ -118,24 +21,95 @@ export async function handleCompaction(params) {
         logger.error("Failed to save compaction summary:", err);
     }
 }
+/**
+ * Retrieve persistent fact context to include during compaction.
+ */
 export async function getCompactionContext(params) {
-    const { client, config, groupId, contextStrings } = params;
+    const { client, characterBudget, groupIds, contextStrings } = params;
     try {
         const queryText = contextStrings.slice(0, 3).join(" ").slice(0, 500);
         if (!queryText.trim())
             return [];
-        const facts = await client.searchFacts({
+        const projectFactsPromise = client.searchFacts({
             query: queryText,
-            groupIds: [groupId],
-            maxFacts: config.maxFacts,
+            groupIds: [groupIds.project],
+            maxFacts: 50,
         });
-        if (facts.length === 0)
+        const projectNodesPromise = client.searchNodes({
+            query: queryText,
+            groupIds: [groupIds.project],
+            maxNodes: 30,
+        });
+        const userGroupId = groupIds.user;
+        const userFactsPromise = userGroupId
+            ? client.searchFacts({
+                query: queryText,
+                groupIds: [userGroupId],
+                maxFacts: 20,
+            })
+            : Promise.resolve([]);
+        const userNodesPromise = userGroupId
+            ? client.searchNodes({
+                query: queryText,
+                groupIds: [userGroupId],
+                maxNodes: 10,
+            })
+            : Promise.resolve([]);
+        const [projectFacts, projectNodes, userFacts, userNodes] = await Promise
+            .all([
+            projectFactsPromise,
+            projectNodesPromise,
+            userFactsPromise,
+            userNodesPromise,
+        ]);
+        if (projectFacts.length === 0 && projectNodes.length === 0 &&
+            userFacts.length === 0 && userNodes.length === 0) {
             return [];
-        const lines = [
-            "## Persistent Knowledge (preserve these facts during compaction):",
-            ...facts.map((fact) => `- ${fact.fact}`),
+        }
+        const buildSection = (header, facts, nodes) => {
+            const lines = [];
+            lines.push(header);
+            if (facts.length > 0) {
+                lines.push("### Facts");
+                lines.push(...formatFactLines(facts));
+            }
+            if (nodes.length > 0) {
+                lines.push("### Nodes");
+                lines.push(...formatNodeLines(nodes));
+            }
+            return lines.join("\n");
+        };
+        const projectSection = buildSection("## Persistent Knowledge (Project)", projectFacts, projectNodes);
+        const userSection = buildSection("## Persistent Knowledge (User)", userFacts, userNodes);
+        const headerLines = [
+            "## Current Goal",
+            "- ",
+            "",
+            "## Work Completed",
+            "- ",
+            "",
+            "## Remaining Tasks",
+            "- ",
+            "",
+            "## Constraints & Decisions",
+            "- ",
+            "",
+            "## Persistent Knowledge",
         ];
-        return [lines.join("\n")];
+        const header = headerLines.join("\n");
+        const base = `${header}\n`;
+        const remainingBudget = Math.max(characterBudget - base.length, 0);
+        const projectBudget = Math.floor(remainingBudget * 0.7);
+        const userBudget = remainingBudget - projectBudget;
+        const truncatedProject = projectSection.slice(0, projectBudget);
+        const truncatedUser = userSection.slice(0, userBudget);
+        const sections = [header];
+        if (truncatedProject.trim())
+            sections.push(truncatedProject);
+        if (truncatedUser.trim())
+            sections.push(truncatedUser);
+        const content = sections.join("\n").slice(0, characterBudget);
+        return [content];
     }
     catch (err) {
         logger.error("Failed to get compaction context:", err);

package/esm/src/services/context-limit.d.ts

@@ -0,0 +1,14 @@
+export interface ProviderListClient {
+    provider: {
+        list: (options?: {
+            directory?: string;
+        }) => Promise<unknown>;
+    };
+}
+export declare function resolveContextLimit(providerID: string, modelID: string, client: ProviderListClient, directory: string): Promise<number>;
+/**
+ * Calculate the character budget for memory injection
+ * (5% of context limit * 4 chars/token).
+ */
+export declare function calculateInjectionBudget(contextLimit: number): number;
+//# sourceMappingURL=context-limit.d.ts.map

package/esm/src/services/context-limit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"context-limit.d.ts","sourceRoot":"","sources":["../../../src/src/services/context-limit.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE;QACR,IAAI,EAAE,CAAC,OAAO,CAAC,EAAE;YAAE,SAAS,CAAC,EAAE,MAAM,CAAA;SAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;KAC9D,CAAC;CACH;AAID,wBAAsB,mBAAmB,CACvC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,kBAAkB,EAC1B,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,MAAM,CAAC,CA+BjB;AAED;;;GAGG;AACH,wBAAgB,wBAAwB,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM,CAErE"}

package/esm/src/services/context-limit.js

@@ -0,0 +1,41 @@
+import { logger } from "./logger.js";
+const DEFAULT_CONTEXT_LIMIT = 200_000;
+const contextLimitCache = new Map();
+export async function resolveContextLimit(providerID, modelID, client, directory) {
+    const modelKey = `${providerID}/${modelID}`;
+    const cached = contextLimitCache.get(modelKey);
+    if (cached)
+        return cached;
+    try {
+        const providers = await client.provider.list({ directory });
+        const list = providers.providers ?? [];
+        for (const provider of list) {
+            const providerInfo = provider;
+            if (providerInfo.id !== providerID)
+                continue;
+            const models = providerInfo.models ?? [];
+            for (const model of models) {
+                const modelInfo = model;
+                if (modelInfo.id !== modelID)
+                    continue;
+                const contextLimit = modelInfo.limit?.context;
+                if (typeof contextLimit === "number" && contextLimit > 0) {
+                    contextLimitCache.set(modelKey, contextLimit);
+                    return contextLimit;
+                }
+            }
+        }
+    }
+    catch (err) {
+        logger.warn("Failed to fetch provider context limit", err);
+    }
+    contextLimitCache.set(modelKey, DEFAULT_CONTEXT_LIMIT);
+    return DEFAULT_CONTEXT_LIMIT;
+}
+/**
+ * Calculate the character budget for memory injection
+ * (5% of context limit * 4 chars/token).
+ */
+export function calculateInjectionBudget(contextLimit) {
+    return Math.floor(contextLimit * 0.05 * 4);
+}

package/esm/src/services/context.d.ts

@@ -1,3 +1,8 @@
 import type { GraphitiFact, GraphitiNode } from "../types/index.js";
+export declare const formatFactLines: (facts: GraphitiFact[]) => string[];
+export declare const formatNodeLines: (nodes: GraphitiNode[]) => string[];
+/**
+ * Format Graphiti facts and nodes into a user-facing context block.
+ */
 export declare function formatMemoryContext(facts: GraphitiFact[], nodes: GraphitiNode[]): string;
 //# sourceMappingURL=context.d.ts.map

package/esm/src/services/context.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../../src/src/services/context.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEpE,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,YAAY,EAAE,EACrB,KAAK,EAAE,YAAY,EAAE,GACpB,MAAM,CA0CR"}
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../../src/src/services/context.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEpE,eAAO,MAAM,eAAe,GAAI,OAAO,YAAY,EAAE,KAAG,MAAM,EAO1D,CAAC;AAEL,eAAO,MAAM,eAAe,GAAI,OAAO,YAAY,EAAE,KAAG,MAAM,EAK1D,CAAC;AAEL;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,YAAY,EAAE,EACrB,KAAK,EAAE,YAAY,EAAE,GACpB,MAAM,CA8BR"}

package/esm/src/services/context.js

@@ -1,3 +1,20 @@
+export const formatFactLines = (facts) => facts.map((fact) => {
+    const entities = [];
+    if (fact.source_node?.name)
+        entities.push(fact.source_node.name);
+    if (fact.target_node?.name)
+        entities.push(fact.target_node.name);
+    const entityStr = entities.length > 0 ? ` [${entities.join(" -> ")}]` : "";
+    return `- ${fact.fact}${entityStr}`;
+});
+export const formatNodeLines = (nodes) => nodes.map((node) => {
+    const labels = node.labels?.length ? ` (${node.labels.join(", ")})` : "";
+    const summary = node.summary ? `: ${node.summary}` : "";
+    return `- **${node.name}**${labels}${summary}`;
+});
+/**
+ * Format Graphiti facts and nodes into a user-facing context block.
+ */
 export function formatMemoryContext(facts, nodes) {
     const sections = [];
     sections.push("# Persistent Memory (from Graphiti Knowledge Graph)");
@@ -7,26 +24,12 @@ export function formatMemoryContext(facts, nodes) {
     sections.push("");
     if (facts.length > 0) {
         sections.push("## Known Facts");
-        for (const fact of facts) {
-            const entities = [];
-            if (fact.source_node?.name)
-                entities.push(fact.source_node.name);
-            if (fact.target_node?.name)
-                entities.push(fact.target_node.name);
-            const entityStr = entities.length > 0
-                ? ` [${entities.join(" -> ")}]`
-                : "";
-            sections.push(`- ${fact.fact}${entityStr}`);
-        }
+        sections.push(...formatFactLines(facts));
         sections.push("");
     }
     if (nodes.length > 0) {
         sections.push("## Known Entities");
-        for (const node of nodes) {
-            const labels = node.labels?.length ? ` (${node.labels.join(", ")})` : "";
-            const summary = node.summary ? `: ${node.summary}` : "";
-            sections.push(`- **${node.name}**${labels}${summary}`);
-        }
+        sections.push(...formatNodeLines(nodes));
         sections.push("");
     }
     if (facts.length === 0 && nodes.length === 0) {

package/esm/src/session.d.ts

@@ -0,0 +1,90 @@
+import type { GraphitiClient } from "./services/client.js";
+/**
+ * Per-session state tracked by the plugin.
+ */
+export type SessionState = {
+    /** Graphiti group ID for this session. */
+    groupId: string;
+    /** Graphiti group ID for user-scoped memories. */
+    userGroupId: string;
+    /** Whether memories have been injected into this session yet. */
+    injectedMemories: boolean;
+    /** Message count at last memory injection. */
+    lastInjectionMessageCount: number;
+    /** Count of messages observed in this session. */
+    messageCount: number;
+    /** Buffered message strings awaiting flush. */
+    pendingMessages: string[];
+    /** Context window limit in tokens. */
+    contextLimit: number;
+    /** True when this session is the primary (non-subagent) session. */
+    isMain: boolean;
+};
+/**
+ * Minimal SDK client interface needed for session operations.
+ */
+export interface SdkSessionClient {
+    session: {
+        /** Retrieve session metadata by ID. */
+        get: (args: {
+            path: {
+                id: string;
+            };
+        }) => Promise<unknown>;
+        /** List recent messages for a session. */
+        messages: (args: {
+            sessionID: string;
+            limit?: number;
+        }) => Promise<unknown>;
+    };
+}
+/**
+ * Manages session lifecycle, parent ID resolution, message buffering,
+ * and flushing of pending messages to Graphiti.
+ */
+/**
+ * Tracks per-session state, parent resolution, message buffering,
+ * and flushing pending messages to Graphiti.
+ */
+export declare class SessionManager {
+    private readonly defaultGroupId;
+    private readonly defaultUserGroupId;
+    private readonly sdkClient;
+    private readonly graphitiClient;
+    private sessions;
+    private parentIdCache;
+    private pendingAssistantMessages;
+    private bufferedAssistantMessageIds;
+    constructor(defaultGroupId: string, defaultUserGroupId: string, sdkClient: SdkSessionClient, graphitiClient: GraphitiClient);
+    /** Get the current session state, if present. */
+    getState(sessionId: string): SessionState | undefined;
+    /** Persist session state for the given session ID. */
+    setState(sessionId: string, state: SessionState): void;
+    /** Cache a resolved parent ID for a session. */
+    setParentId(sessionId: string, parentId: string | null): void;
+    /** Resolve and cache the parent ID for a session. */
+    resolveParentId(sessionId: string): Promise<string | null | undefined>;
+    /** Resolve the session state, initializing if needed. */
+    resolveSessionState(sessionId: string): Promise<{
+        state: SessionState | null;
+        resolved: boolean;
+    }>;
+    /** Determine whether a session is a subagent session. */
+    isSubagentSession(sessionId: string): Promise<boolean>;
+    /** Buffer partial assistant text for a streaming message. */
+    bufferAssistantPart(sessionId: string, messageId: string, text: string): void;
+    /** Check if an assistant message has already been finalized. */
+    isAssistantBuffered(sessionId: string, messageId: string): boolean;
+    /**
+     * Finalize a buffered assistant message and append it to pending messages.
+     */
+    finalizeAssistantMessage(state: SessionState, sessionId: string, messageId: string, source: string): void;
+    /** Flush pending buffered messages to Graphiti when size thresholds permit. */
+    flushPendingMessages(sessionId: string, sourceDescription: string, minBytes: number): Promise<void>;
+    /** Remove a pending assistant message by key. */
+    deletePendingAssistant(sessionId: string, messageId: string): void;
+    /** Clear cached data for a session. */
+    deleteSession(sessionId: string): void;
+    private fetchLatestAssistantMessage;
+}
+//# sourceMappingURL=session.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../src/src/session.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAI3D;;GAEG;AACH,MAAM,MAAM,YAAY,GAAG;IACzB,0CAA0C;IAC1C,OAAO,EAAE,MAAM,CAAC;IAChB,kDAAkD;IAClD,WAAW,EAAE,MAAM,CAAC;IACpB,iEAAiE;IACjE,gBAAgB,EAAE,OAAO,CAAC;IAC1B,8CAA8C;IAC9C,yBAAyB,EAAE,MAAM,CAAC;IAClC,kDAAkD;IAClD,YAAY,EAAE,MAAM,CAAC;IACrB,+CAA+C;IAC/C,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,sCAAsC;IACtC,YAAY,EAAE,MAAM,CAAC;IACrB,oEAAoE;IACpE,MAAM,EAAE,OAAO,CAAC;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE;QACP,uCAAuC;QACvC,GAAG,EAAE,CAAC,IAAI,EAAE;YAAE,IAAI,EAAE;gBAAE,EAAE,EAAE,MAAM,CAAA;aAAE,CAAA;SAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1D,0CAA0C;QAC1C,QAAQ,EAAE,CAAC,IAAI,EAAE;YAAE,SAAS,EAAE,MAAM,CAAC;YAAC,KAAK,CAAC,EAAE,MAAM,CAAA;SAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;KAC7E,CAAC;CACH;AAED;;;GAGG;AACH;;;GAGG;AACH,qBAAa,cAAc;IAUvB,OAAO,CAAC,QAAQ,CAAC,cAAc;IAC/B,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IACnC,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,cAAc;IAZjC,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,aAAa,CAAoC;IACzD,OAAO,CAAC,wBAAwB,CAG5B;IACJ,OAAO,CAAC,2BAA2B,CAAqB;gBAGrC,cAAc,EAAE,MAAM,EACtB,kBAAkB,EAAE,MAAM,EAC1B,SAAS,EAAE,gBAAgB,EAC3B,cAAc,EAAE,cAAc;IAGjD,iDAAiD;IACjD,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAIrD,sDAAsD;IACtD,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,GAAG,IAAI;IAItD,gDAAgD;IAChD,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI;IAI7D,qDAAqD;IAC/C,eAAe,CACnB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IAsBrC,yDAAyD;IACnD,mBAAmB,CACvB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC;QAAE,KAAK,EAAE,YAAY,GAAG,IAAI,CAAC;QAAC,QAAQ,EAAE,OAAO,CAAA;KAAE,CAAC;IAyB7D,yDAAyD;IACnD,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAK5D,6DAA6D;IAC7D,mBAAmB,CACjB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,MAAM,GACX,IAAI;IAKP,gEAAgE;IAChE,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO;IAKlE;;OAEG;IACH,wBAAwB,CACtB,KAAK,EAAE,YAAY,EACnB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,IAAI;IAwCP,+EAA+E;IACzE,oBAAoB,CACxB,SAAS,EAAE,MAAM,EACjB,iBAAiB,EAAE,MAAM,EACzB,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAgFhB,iDAAiD;IACjD,sBAAsB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAKlE,uCAAuC;IACvC,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;YAexB,2BAA2B;CAiC1C"}