opencode-graphiti 0.1.10 → 0.1.12

package/script/src/services/client.js

@@ -1,138 +1,41 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GraphitiClient = void 0;
-const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
-const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
-const deno_js_1 = __importDefault(require("../../deno.js"));
+const connection_manager_js_1 = require("./connection-manager.js");
 const logger_js_1 = require("./logger.js");
 const sdk_normalize_js_1 = require("./sdk-normalize.js");
 /**
- * Graphiti MCP client wrapper for connecting, querying,
- * and persisting episodes with basic reconnection handling.
+ * Graphiti domain adapter over the connection manager.
  */
 class GraphitiClient {
-    /**
-     * Create a Graphiti client bound to the given MCP endpoint URL.
-     */
-    constructor(endpoint) {
-        Object.defineProperty(this, "client", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "transport", {
+    constructor(endpointOrManager) {
+        Object.defineProperty(this, "toolCaller", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "connected", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "endpoint", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.endpoint = endpoint;
-        this.client = new index_js_1.Client({
-            name: deno_js_1.default.name,
-            version: deno_js_1.default.version,
-        });
-        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: deno_js_1.default.name,
-            version: deno_js_1.default.version,
-        });
-        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;
-            logger_js_1.logger.info("Connected to Graphiti MCP server at", this.endpoint);
-            return true;
+        if (typeof endpointOrManager === "string") {
+            this.toolCaller = new connection_manager_js_1.GraphitiConnectionManager({
+                endpoint: endpointOrManager,
+            });
         }
-        catch (err) {
-            logger_js_1.logger.error("Failed to connect to Graphiti:", err);
-            return false;
+        else {
+            this.toolCaller = endpointOrManager;
         }
     }
-    /**
-     * Close the underlying MCP client connection.
-     */
-    async disconnect() {
-        if (this.connected) {
-            await this.client.close();
-            this.connected = false;
-        }
+    start() {
+        this.toolCaller.start();
     }
-    async callTool(name, args) {
-        if (!this.connected) {
-            const ok = await this.connect();
-            if (!ok)
-                throw new Error("Not connected to Graphiti");
-        }
-        // Sanitize arguments: omit task_id (and others) if null or undefined
-        const sanitizedArgs = Object.fromEntries(Object.entries(args).filter(([_, v]) => v !== null && v !== undefined));
-        try {
-            const result = await this.client.callTool({
-                name,
-                arguments: sanitizedArgs,
-            });
-            return this.parseToolResult(result);
-        }
-        catch (err) {
-            if (this.isSessionExpired(err)) {
-                logger_js_1.logger.warn("Graphiti session expired, reconnecting...");
-                await this.reconnect();
-                const result = await this.client.callTool({
-                    name,
-                    arguments: sanitizedArgs,
-                });
-                return this.parseToolResult(result);
-            }
-            throw err;
-        }
+    async stop() {
+        await this.toolCaller.stop();
     }
-    isSessionExpired(err) {
-        return !!(err &&
-            typeof err === "object" &&
-            "code" in err &&
-            err.code === 404);
+    async connect() {
+        this.toolCaller.start();
+        return await this.toolCaller.ready();
     }
-    async reconnect() {
-        this.connected = false;
-        try {
-            await this.client.close();
-        }
-        catch {
-            // ignore close errors on stale client
-        }
-        this.createClientAndTransport();
-        await this.client.connect(this.transport);
-        this.connected = true;
-        logger_js_1.logger.info("Reconnected to Graphiti MCP server");
+    async ready(timeoutMs) {
+        return await this.toolCaller.ready(timeoutMs);
     }
     /**
      * Parse MCP tool results into JSON when possible.
@@ -161,19 +64,6 @@ class GraphitiClient {
             return text;
         }
     }
-    /**
-     * Add an episode to Graphiti memory.
-     */
-    async addEpisode(params) {
-        await this.callTool("add_memory", {
-            name: params.name,
-            episode_body: params.episodeBody,
-            group_id: params.groupId,
-            source: params.source || "text",
-            source_description: params.sourceDescription || "",
-        });
-        logger_js_1.logger.debug("Added episode:", params.name);
-    }
     /**
      * Extract an array from a tool result that may be a bare array or a
      * wrapped-array response object (`{ [key]: T[] }`).
@@ -190,6 +80,30 @@ class GraphitiClient {
         }
         return null;
     }
+    /**
+     * Add an episode to Graphiti memory.
+     */
+    async addEpisode(params) {
+        try {
+            await this.callTool("add_memory", {
+                name: params.name,
+                episode_body: params.episodeBody,
+                group_id: params.groupId,
+                source: params.source || "text",
+                source_description: params.sourceDescription || "",
+            });
+            logger_js_1.logger.debug("Added episode:", params.name);
+        }
+        catch (err) {
+            if ((0, connection_manager_js_1.isGraphitiOfflineError)(err) ||
+                (0, connection_manager_js_1.isGraphitiTimeoutError)(err) ||
+                err instanceof connection_manager_js_1.GraphitiTransportError ||
+                err instanceof connection_manager_js_1.GraphitiSessionExpiredError) {
+                logger_js_1.logger.warn("addEpisode failed due to Graphiti availability issue", err);
+            }
+            throw err;
+        }
+    }
     /**
      * Search Graphiti facts matching the provided query.
      */
@@ -203,6 +117,19 @@ class GraphitiClient {
             return this.parseWrappedArray(result, "facts") ?? [];
         }
         catch (err) {
+            if ((0, connection_manager_js_1.isGraphitiTimeoutError)(err)) {
+                logger_js_1.logger.warn("searchFacts request timed out; returning no facts");
+                return [];
+            }
+            if ((0, connection_manager_js_1.isGraphitiOfflineError)(err)) {
+                logger_js_1.logger.warn("searchFacts unavailable; returning no facts");
+                return [];
+            }
+            if (err instanceof connection_manager_js_1.GraphitiTransportError ||
+                err instanceof connection_manager_js_1.GraphitiSessionExpiredError) {
+                logger_js_1.logger.warn("searchFacts unavailable during reconnect; returning no facts");
+                return [];
+            }
             logger_js_1.logger.error("searchFacts error:", err);
             return [];
         }
@@ -220,6 +147,19 @@ class GraphitiClient {
             return this.parseWrappedArray(result, "nodes") ?? [];
         }
         catch (err) {
+            if ((0, connection_manager_js_1.isGraphitiTimeoutError)(err)) {
+                logger_js_1.logger.warn("searchNodes request timed out; returning no nodes");
+                return [];
+            }
+            if ((0, connection_manager_js_1.isGraphitiOfflineError)(err)) {
+                logger_js_1.logger.warn("searchNodes unavailable; returning no nodes");
+                return [];
+            }
+            if (err instanceof connection_manager_js_1.GraphitiTransportError ||
+                err instanceof connection_manager_js_1.GraphitiSessionExpiredError) {
+                logger_js_1.logger.warn("searchNodes unavailable during reconnect; returning no nodes");
+                return [];
+            }
             logger_js_1.logger.error("searchNodes error:", err);
             return [];
         }
@@ -238,6 +178,19 @@ class GraphitiClient {
             return raw.map(sdk_normalize_js_1.normalizeEpisode);
         }
         catch (err) {
+            if ((0, connection_manager_js_1.isGraphitiTimeoutError)(err)) {
+                logger_js_1.logger.warn("getEpisodes request timed out; returning no episodes");
+                return [];
+            }
+            if ((0, connection_manager_js_1.isGraphitiOfflineError)(err)) {
+                logger_js_1.logger.warn("getEpisodes unavailable; returning no episodes");
+                return [];
+            }
+            if (err instanceof connection_manager_js_1.GraphitiTransportError ||
+                err instanceof connection_manager_js_1.GraphitiSessionExpiredError) {
+                logger_js_1.logger.warn("getEpisodes unavailable during reconnect; returning no episodes");
+                return [];
+            }
             logger_js_1.logger.error("getEpisodes error:", err);
             return [];
         }
@@ -254,5 +207,9 @@ class GraphitiClient {
             return false;
         }
     }
+    async callTool(name, args) {
+        const result = await this.toolCaller.callTool(name, args);
+        return this.parseToolResult(result);
+    }
 }
 exports.GraphitiClient = GraphitiClient;

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

@@ -0,0 +1,97 @@
+export type GraphitiConnectionState = "connecting" | "connected" | "offline" | "closing";
+type TimerHandle = ReturnType<typeof setTimeout> | number;
+export declare class GraphitiOfflineError extends Error {
+    readonly state: "offline" | "closing";
+    readonly kind = "offline";
+    constructor(state: "offline" | "closing", message?: string);
+}
+export declare class GraphitiQueueTimeoutError extends Error {
+    readonly kind = "queue-timeout";
+    constructor(message?: string);
+}
+export declare class GraphitiRequestTimeoutError extends Error {
+    readonly kind = "request-timeout";
+    constructor(message?: string);
+}
+export declare class GraphitiTransportError extends Error {
+    readonly kind = "transport-failure";
+    constructor(message?: string);
+}
+export declare class GraphitiSessionExpiredError extends Error {
+    readonly kind = "session-expired";
+    constructor(message?: string);
+}
+export declare function isGraphitiOfflineError(err: unknown): err is GraphitiOfflineError;
+export declare function isGraphitiTimeoutError(err: unknown): err is GraphitiQueueTimeoutError | GraphitiRequestTimeoutError;
+export type GraphitiToolRequest = {
+    name: string;
+    arguments?: Record<string, unknown>;
+};
+export interface GraphitiConnection {
+    connect(): Promise<void>;
+    close(): Promise<void>;
+    callTool(request: GraphitiToolRequest): Promise<unknown>;
+}
+export interface GraphitiToolCaller {
+    start(): void;
+    stop(): Promise<void>;
+    ready(timeoutMs?: number): Promise<boolean>;
+    callTool(name: string, args: Record<string, unknown>, deadlineMs?: number): Promise<unknown>;
+}
+type ConnectionFactory = (endpoint: string) => GraphitiConnection;
+type GraphitiConnectionManagerOptions = {
+    endpoint: string;
+    requestDeadlineMs?: number;
+    queueCapacity?: number;
+    startupTimeoutMs?: number;
+    reconnectInitialDelayMs?: number;
+    reconnectMaxDelayMs?: number;
+    reconnectMultiplier?: number;
+    reconnectJitter?: number;
+    connectionFactory?: ConnectionFactory;
+    random?: () => number;
+    setTimer?: (callback: () => void, delayMs: number) => TimerHandle;
+    clearTimer?: (timer: TimerHandle) => void;
+};
+export declare class GraphitiConnectionManager implements GraphitiToolCaller {
+    private readonly endpoint;
+    private readonly requestDeadlineMs;
+    private readonly queueCapacity;
+    private readonly startupTimeoutMs;
+    private readonly reconnectInitialDelayMs;
+    private readonly reconnectMaxDelayMs;
+    private readonly reconnectMultiplier;
+    private readonly reconnectJitter;
+    private readonly connectionFactory;
+    private readonly random;
+    private readonly setTimerImpl;
+    private readonly clearTimerImpl;
+    private state;
+    private connection;
+    private connectPromise;
+    private reconnectTimer;
+    private pendingRequests;
+    private readyWaiters;
+    private reconnectDelayMs;
+    private started;
+    private flushingQueue;
+    constructor(options: GraphitiConnectionManagerOptions);
+    getState(): GraphitiConnectionState;
+    start(): void;
+    stop(): Promise<void>;
+    ready(timeoutMs?: number): Promise<boolean>;
+    callTool(name: string, args: Record<string, unknown>, deadlineMs?: number): Promise<unknown>;
+    reconnect(): Promise<boolean>;
+    private performReconnect;
+    private executeConnectedCall;
+    private enqueueRequest;
+    private flushPendingQueue;
+    private removePendingRequest;
+    private clearPendingTimer;
+    private rejectAllPending;
+    private scheduleReconnect;
+    private cancelReconnectTimer;
+    private resolveReadyWaiters;
+}
+export {};
+//# sourceMappingURL=connection-manager.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"connection-manager.d.ts","sourceRoot":"","sources":["../../../src/src/services/connection-manager.ts"],"names":[],"mappings":"AAKA,MAAM,MAAM,uBAAuB,GAC/B,YAAY,GACZ,WAAW,GACX,SAAS,GACT,SAAS,CAAC;AAEd,KAAK,WAAW,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,MAAM,CAAC;AAE1D,qBAAa,oBAAqB,SAAQ,KAAK;IAGjC,QAAQ,CAAC,KAAK,EAAE,SAAS,GAAG,SAAS;IAFjD,QAAQ,CAAC,IAAI,aAAa;gBAEL,KAAK,EAAE,SAAS,GAAG,SAAS,EAAE,OAAO,CAAC,EAAE,MAAM;CASpE;AAED,qBAAa,yBAA0B,SAAQ,KAAK;IAClD,QAAQ,CAAC,IAAI,mBAAmB;gBAG9B,OAAO,SAA4D;CAKtE;AAED,qBAAa,2BAA4B,SAAQ,KAAK;IACpD,QAAQ,CAAC,IAAI,qBAAqB;gBAEtB,OAAO,SAA+B;CAInD;AAED,qBAAa,sBAAuB,SAAQ,KAAK;IAC/C,QAAQ,CAAC,IAAI,uBAAuB;gBAExB,OAAO,SAA+B;CAInD;AAED,qBAAa,2BAA4B,SAAQ,KAAK;IACpD,QAAQ,CAAC,IAAI,qBAAqB;gBAEtB,OAAO,SAA6B;CAIjD;AAED,wBAAgB,sBAAsB,CACpC,GAAG,EAAE,OAAO,GACX,GAAG,IAAI,oBAAoB,CAE7B;AAED,wBAAgB,sBAAsB,CACpC,GAAG,EAAE,OAAO,GACX,GAAG,IAAI,yBAAyB,GAAG,2BAA2B,CAGhE;AAED,MAAM,MAAM,mBAAmB,GAAG;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC,CAAC;AAEF,MAAM,WAAW,kBAAkB;IACjC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACzB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,QAAQ,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CAC1D;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,IAAI,IAAI,CAAC;IACd,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACtB,KAAK,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5C,QAAQ,CACN,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,UAAU,CAAC,EAAE,MAAM,GAClB,OAAO,CAAC,OAAO,CAAC,CAAC;CACrB;AAUD,KAAK,iBAAiB,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,kBAAkB,CAAC;AAElE,KAAK,gCAAgC,GAAG;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;IACtC,MAAM,CAAC,EAAE,MAAM,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,OAAO,EAAE,MAAM,KAAK,WAAW,CAAC;IAClE,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;CAC3C,CAAC;AAsEF,qBAAa,yBAA0B,YAAW,kBAAkB;IAClE,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAS;IACjD,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAC7C,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAC7C,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAoB;IACtD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;IACtC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAGZ;IACjB,OAAO,CAAC,QAAQ,CAAC,cAAc,CAA+B;IAE9D,OAAO,CAAC,KAAK,CAAsC;IACnD,OAAO,CAAC,UAAU,CAAmC;IACrD,OAAO,CAAC,cAAc,CAAiC;IACvD,OAAO,CAAC,cAAc,CAA4B;IAClD,OAAO,CAAC,eAAe,CAAwB;IAC/C,OAAO,CAAC,YAAY,CAAuC;IAC3D,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,aAAa,CAAS;gBAElB,OAAO,EAAE,gCAAgC;IAkBrD,QAAQ,IAAI,uBAAuB;IAInC,KAAK,IAAI,IAAI;IAMP,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAyBrB,KAAK,CAAC,SAAS,SAAwB,GAAG,OAAO,CAAC,OAAO,CAAC;IAwB1D,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,UAAU,SAAyB,GAClC,OAAO,CAAC,OAAO,CAAC;IAsBb,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;YAerB,gBAAgB;YAmDhB,oBAAoB;IAoDlC,OAAO,CAAC,cAAc;YAuCR,iBAAiB;IAuB/B,OAAO,CAAC,oBAAoB;IAQ5B,OAAO,CAAC,iBAAiB;IAOzB,OAAO,CAAC,gBAAgB;IAUxB,OAAO,CAAC,iBAAiB;IA4BzB,OAAO,CAAC,oBAAoB;IAO5B,OAAO,CAAC,mBAAmB;CAO5B"}