opencode-graphiti 0.0.0-development

package/script/src/services/sdk-normalize.js

@@ -0,0 +1,67 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.unwrapSdkResponse = unwrapSdkResponse;
+exports.extractSdkMessages = extractSdkMessages;
+exports.extractSdkProviders = extractSdkProviders;
+exports.normalizeEpisode = normalizeEpisode;
+/**
+ * Normalize an SDK response that may be wrapped in `{ data: … }` or returned
+ * directly.  Returns the inner value cast to `T`, or `undefined` when the
+ * response is absent.
+ *
+ * This replaces the repeated `"data" in response ? (response as
+ * { data?: … }).data : response` pattern found in session.ts and
+ * context-limit.ts.
+ */
+function unwrapSdkResponse(response) {
+    if (response == null)
+        return undefined;
+    if (typeof response === "object" && "data" in response) {
+        return response.data;
+    }
+    return response;
+}
+/**
+ * Extract the messages array from a raw `session.messages()` response.
+ * Returns an empty array when the response is missing or malformed.
+ */
+function extractSdkMessages(response) {
+    const payload = unwrapSdkResponse(response);
+    return Array.isArray(payload) ? payload : [];
+}
+function extractSdkProviders(response) {
+    // provider.list() may return `{ providers: [...] }` directly (no data wrap).
+    if (response != null && typeof response === "object") {
+        const obj = response;
+        if (Array.isArray(obj["providers"])) {
+            return obj["providers"];
+        }
+        if ("data" in obj) {
+            const data = obj["data"];
+            if (data != null && typeof data === "object") {
+                const dataObj = data;
+                if (Array.isArray(dataObj["providers"])) {
+                    return dataObj["providers"];
+                }
+            }
+            if (Array.isArray(data))
+                return data;
+        }
+    }
+    return [];
+}
+/**
+ * Normalize a raw Graphiti episode object so that `sourceDescription` is
+ * always the canonical field regardless of whether the payload used
+ * camelCase (`sourceDescription`) or snake_case (`source_description`).
+ *
+ * Call this at the API boundary (e.g. inside `GraphitiClient.getEpisodes`)
+ * so all downstream consumers only need to read `episode.sourceDescription`.
+ */
+function normalizeEpisode(raw) {
+    const { source_description, ...rest } = raw;
+    return {
+        ...rest,
+        sourceDescription: rest.sourceDescription ?? source_description,
+    };
+}

package/script/src/session.d.ts

@@ -0,0 +1,74 @@
+import type { OpencodeClient } from "@opencode-ai/sdk";
+import { type RedisCacheService } from "./services/redis-cache.js";
+import type { RedisEventsService } from "./services/redis-events.js";
+import type { RedisSnapshotService } from "./services/redis-snapshot.js";
+import { type PreparedSessionMemory } from "./types/index.js";
+export type SessionState = {
+    groupId: string;
+    userGroupId: string;
+    injectedMemories: boolean;
+    messageCount: number;
+    contextLimit: number;
+    isMain: boolean;
+    hotTierReady: boolean;
+    latestUserRequest?: string;
+    latestRefreshQuery?: string;
+    pendingInjection?: PreparedSessionMemory;
+    pendingInjectionGeneration: number;
+};
+type TimerHandle = ReturnType<typeof setTimeout> | number;
+export interface SessionManagerOptions {
+    idleRetentionMs?: number;
+    setTimer?: (callback: () => void, delayMs: number) => TimerHandle;
+    clearTimer?: (timer: TimerHandle) => void;
+}
+export declare class SessionManager {
+    private readonly defaultGroupId;
+    private readonly defaultUserGroupId;
+    private readonly sdkClient;
+    private readonly redisEvents;
+    private readonly redisSnapshot;
+    private readonly redisCache;
+    private sessions;
+    private parentIdCache;
+    private canonicalSessionIdCache;
+    private temporaryRootSessionIds;
+    private pendingAssistantMessages;
+    private pendingAssistantCompletions;
+    private bufferedAssistantMessageIds;
+    private sessionLifecycles;
+    private readonly idleRetentionMs;
+    private readonly setTimerImpl;
+    private readonly clearTimerImpl;
+    constructor(defaultGroupId: string, defaultUserGroupId: string, sdkClient: OpencodeClient, redisEvents: RedisEventsService, redisSnapshot: RedisSnapshotService, redisCache: RedisCacheService, options?: SessionManagerOptions);
+    createDefaultState(groupId: string, userGroupId: string): SessionState;
+    getState(sessionId: string): SessionState | undefined;
+    setState(sessionId: string, state: SessionState): void;
+    markSessionActive(sessionId: string): void;
+    markResolvedSessionActive(sessionId: string, canonicalSessionId?: string): void;
+    private markLifecycleActive;
+    captureIdleCleanupGeneration(sessionId: string): number | null;
+    scheduleIdleSessionCleanup(sessionId: string, expectedActivityGeneration?: number): void;
+    setParentId(sessionId: string, parentId: string | null): void;
+    private mergeSessionState;
+    private migrateTemporaryRootRuntimeState;
+    resolveParentId(sessionId: string): Promise<string | null | undefined>;
+    resolveCanonicalSessionId(sessionId: string, visited?: Set<string>): Promise<string | undefined>;
+    resolveSessionState(sessionId: string): Promise<{
+        state: SessionState | null;
+        resolved: boolean;
+        canonicalSessionId?: string;
+    }>;
+    bufferAssistantPart(sessionId: string, messageId: string, text: string, sourceSessionId?: string): void;
+    isAssistantBuffered(sessionId: string, messageId: string): boolean;
+    hasPendingAssistantCompletion(sessionId: string, messageId: string): boolean;
+    finalizeAssistantMessage(_state: SessionState, sessionId: string, messageId: string, source: string): string | null;
+    deletePendingAssistant(sessionId: string, messageId: string): void;
+    clearPendingInjection(state: SessionState, prepared?: PreparedSessionMemory | null): void;
+    purgeAssistantBufferSource(sourceSessionId: string): void;
+    prepareInjection(sessionId: string, lastRequest?: string): Promise<PreparedSessionMemory | null>;
+    deleteSession(sessionId: string): void;
+    private getLifecycle;
+}
+export {};
+//# sourceMappingURL=session.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../src/src/session.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAGvD,OAAO,EAEL,KAAK,iBAAiB,EACvB,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAQrE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AACzE,OAAO,EAEL,KAAK,qBAAqB,EAE3B,MAAM,kBAAkB,CAAC;AAoK1B,MAAM,MAAM,YAAY,GAAG;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,OAAO,CAAC;IAChB,YAAY,EAAE,OAAO,CAAC;IACtB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,gBAAgB,CAAC,EAAE,qBAAqB,CAAC;IACzC,0BAA0B,EAAE,MAAM,CAAC;CACpC,CAAC;AAEF,KAAK,WAAW,GAAG,UAAU,CAAC,OAAO,UAAU,CAAC,GAAG,MAAM,CAAC;AAE1D,MAAM,WAAW,qBAAqB;IACpC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,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;AAOD,qBAAa,cAAc;IAoBvB,OAAO,CAAC,QAAQ,CAAC,cAAc;IAC/B,OAAO,CAAC,QAAQ,CAAC,kBAAkB;IACnC,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,UAAU;IAxB7B,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,aAAa,CAAoC;IACzD,OAAO,CAAC,uBAAuB,CAA6B;IAC5D,OAAO,CAAC,uBAAuB,CAAqB;IACpD,OAAO,CAAC,wBAAwB,CAG5B;IACJ,OAAO,CAAC,2BAA2B,CAAqB;IACxD,OAAO,CAAC,2BAA2B,CAA6B;IAChE,OAAO,CAAC,iBAAiB,CAAuC;IAChE,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAGZ;IACjB,OAAO,CAAC,QAAQ,CAAC,cAAc,CAA+B;gBAG3C,cAAc,EAAE,MAAM,EACtB,kBAAkB,EAAE,MAAM,EAC1B,SAAS,EAAE,cAAc,EACzB,WAAW,EAAE,kBAAkB,EAC/B,aAAa,EAAE,oBAAoB,EACnC,UAAU,EAAE,iBAAiB,EAC9C,OAAO,GAAE,qBAA0B;IASrC,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,YAAY;IAgBtE,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAIrD,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,GAAG,IAAI;IAItD,iBAAiB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAQ1C,yBAAyB,CACvB,SAAS,EAAE,MAAM,EACjB,kBAAkB,CAAC,EAAE,MAAM,GAC1B,IAAI;IAOP,OAAO,CAAC,mBAAmB;IAS3B,4BAA4B,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAM9D,0BAA0B,CACxB,SAAS,EAAE,MAAM,EACjB,0BAA0B,CAAC,EAAE,MAAM,GAClC,IAAI;IAsCP,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI;IAwB7D,OAAO,CAAC,iBAAiB;IAwBzB,OAAO,CAAC,gCAAgC;IAsFlC,eAAe,CACnB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IAgC/B,yBAAyB,CAC7B,SAAS,EAAE,MAAM,EACjB,OAAO,GAAE,GAAG,CAAC,MAAM,CAAqB,GACvC,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAgCxB,mBAAmB,CACvB,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC;QACT,KAAK,EAAE,YAAY,GAAG,IAAI,CAAC;QAC3B,QAAQ,EAAE,OAAO,CAAC;QAClB,kBAAkB,CAAC,EAAE,MAAM,CAAC;KAC7B,CAAC;IAiBF,mBAAmB,CACjB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EACjB,IAAI,EAAE,MAAM,EACZ,eAAe,SAAY,GAC1B,IAAI;IASP,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO;IAIlE,6BAA6B,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO;IAI5E,wBAAwB,CACtB,MAAM,EAAE,YAAY,EACpB,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,GACb,MAAM,GAAG,IAAI;IA0BhB,sBAAsB,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAMlE,qBAAqB,CACnB,KAAK,EAAE,YAAY,EACnB,QAAQ,CAAC,EAAE,qBAAqB,GAAG,IAAI,GACtC,IAAI;IAOP,0BAA0B,CAAC,eAAe,EAAE,MAAM,GAAG,IAAI;IAoBnD,gBAAgB,CACpB,SAAS,EAAE,MAAM,EACjB,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC;IAkLxC,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAuCtC,OAAO,CAAC,YAAY;CAWrB"}