opencode-graphiti 0.1.0 → 0.1.2

package/script/src/services/client.js

@@ -1,44 +1,17 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GraphitiClient = void 0;
-const dntShim = __importStar(require("../../_dnt.shims.js"));
 const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
 const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
 const logger_js_1 = require("./logger.js");
+/**
+ * Graphiti MCP client wrapper for connecting, querying,
+ * and persisting episodes with basic reconnection handling.
+ */
 class GraphitiClient {
+    /**
+     * Create a Graphiti client bound to the given MCP endpoint URL.
+     */
     constructor(endpoint) {
         Object.defineProperty(this, "client", {
             enumerable: true,
@@ -66,12 +39,23 @@ class GraphitiClient {
         });
         this.endpoint = endpoint;
         this.client = new index_js_1.Client({ name: "opencode-graphiti", version: "0.1.0" });
-        const url = new dntShim.dntGlobalThis.URL(endpoint);
-        this.transport = new streamableHttp_js_1.StreamableHTTPClientTransport(url);
+        this.transport = new streamableHttp_js_1.StreamableHTTPClientTransport(new URL(endpoint));
+    }
+    /** Create a fresh MCP Client and Transport pair. */
+    createClientAndTransport() {
+        this.client = new index_js_1.Client({ name: "opencode-graphiti", version: "0.1.0" });
+        this.transport = new streamableHttp_js_1.StreamableHTTPClientTransport(new URL(this.endpoint));
     }
+    /**
+     * Establish a connection to the Graphiti MCP server.
+     * Creates a fresh Client/Transport if a previous attempt failed.
+     */
     async connect() {
         if (this.connected)
             return true;
+        // If a previous connect() tainted the Client's internal state,
+        // create fresh instances so the retry starts cleanly.
+        this.createClientAndTransport();
         try {
             await this.client.connect(this.transport);
             this.connected = true;
@@ -83,6 +67,9 @@ class GraphitiClient {
             return false;
         }
     }
+    /**
+     * Close the underlying MCP client connection.
+     */
     async disconnect() {
         if (this.connected) {
             await this.client.close();
@@ -126,17 +113,20 @@ class GraphitiClient {
     async reconnect() {
         this.connected = false;
         try {
-            await this.transport.close();
+            await this.client.close();
         }
         catch {
-            // ignore transport close errors
+            // ignore close errors on stale client
         }
-        this.transport = new streamableHttp_js_1.StreamableHTTPClientTransport(new URL(this.endpoint));
+        this.createClientAndTransport();
         await this.client.connect(this.transport);
         this.connected = true;
         logger_js_1.logger.info("Reconnected to Graphiti MCP server");
     }
-    // Public for testing
+    /**
+     * Parse MCP tool results into JSON when possible.
+     * Public for testing.
+     */
     parseToolResult(result) {
         const typedResult = result;
         const content = typedResult.content;
@@ -160,6 +150,9 @@ class GraphitiClient {
             return text;
         }
     }
+    /**
+     * Add an episode to Graphiti memory.
+     */
     async addEpisode(params) {
         await this.callTool("add_memory", {
             name: params.name,
@@ -170,6 +163,9 @@ class GraphitiClient {
         });
         logger_js_1.logger.debug("Added episode:", params.name);
     }
+    /**
+     * Search Graphiti facts matching the provided query.
+     */
     async searchFacts(params) {
         try {
             const result = await this.callTool("search_memory_facts", {
@@ -191,6 +187,9 @@ class GraphitiClient {
             return [];
         }
     }
+    /**
+     * Search Graphiti nodes matching the provided query.
+     */
     async searchNodes(params) {
         try {
             const result = await this.callTool("search_nodes", {
@@ -212,19 +211,9 @@ class GraphitiClient {
             return [];
         }
     }
-    async getEpisodes(params) {
-        try {
-            const result = await this.callTool("get_episodes", {
-                group_ids: params.groupIds,
-                max_episodes: params.maxEpisodes || 10,
-            });
-            return result || [];
-        }
-        catch (err) {
-            logger_js_1.logger.error("getEpisodes error:", err);
-            return [];
-        }
-    }
+    /**
+     * Check whether the Graphiti MCP server is reachable.
+     */
     async getStatus() {
         try {
             await this.callTool("get_status", {});

package/script/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/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,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/script/src/services/compaction.js

@@ -1,113 +1,15 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createPreemptiveCompactionHandler = createPreemptiveCompactionHandler;
 exports.handleCompaction = handleCompaction;
 exports.getCompactionContext = getCompactionContext;
+const context_js_1 = require("./context.js");
 const logger_js_1 = require("./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_js_1.logger.warn("Failed to fetch provider context limit", err);
-    }
-    state.contextLimitCache.set(modelKey, DEFAULT_CONTEXT_LIMIT);
-    return DEFAULT_CONTEXT_LIMIT;
-};
-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_js_1.logger.info("Preemptive compaction triggered", {
-                sessionId,
-                providerID,
-                modelID,
-                usageRatio,
-                totalTokens,
-                contextLimit,
-            });
-        }
-        catch (err) {
-            logger_js_1.logger.error("Preemptive compaction failed", err);
-        }
-        finally {
-            state.compactionInProgress.delete(sessionId);
-        }
-    };
-    return { checkAndTriggerCompaction };
-}
+/**
+ * Persist a compaction summary episode when enabled.
+ */
 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({
@@ -123,24 +25,95 @@ async function handleCompaction(params) {
         logger_js_1.logger.error("Failed to save compaction summary:", err);
     }
 }
+/**
+ * Retrieve persistent fact context to include during compaction.
+ */
 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(...(0, context_js_1.formatFactLines)(facts));
+            }
+            if (nodes.length > 0) {
+                lines.push("### Nodes");
+                lines.push(...(0, context_js_1.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_js_1.logger.error("Failed to get compaction context:", err);

package/script/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/script/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/script/src/services/context-limit.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveContextLimit = resolveContextLimit;
+exports.calculateInjectionBudget = calculateInjectionBudget;
+const logger_js_1 = require("./logger.js");
+const DEFAULT_CONTEXT_LIMIT = 200_000;
+const contextLimitCache = new Map();
+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_js_1.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).
+ */
+function calculateInjectionBudget(contextLimit) {
+    return Math.floor(contextLimit * 0.05 * 4);
+}

package/script/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/script/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/script/src/services/context.js

@@ -1,6 +1,26 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatNodeLines = exports.formatFactLines = void 0;
 exports.formatMemoryContext = formatMemoryContext;
+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}`;
+});
+exports.formatFactLines = formatFactLines;
+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}`;
+});
+exports.formatNodeLines = formatNodeLines;
+/**
+ * Format Graphiti facts and nodes into a user-facing context block.
+ */
 function formatMemoryContext(facts, nodes) {
     const sections = [];
     sections.push("# Persistent Memory (from Graphiti Knowledge Graph)");
@@ -10,26 +30,12 @@ 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(...(0, exports.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(...(0, exports.formatNodeLines)(nodes));
         sections.push("");
     }
     if (facts.length === 0 && nodes.length === 0) {