@stackmemoryai/stackmemory 0.2.8 → 0.3.0

package/dist/core/digest/frame-digest-integration.js

@@ -0,0 +1,172 @@
+import { EnhancedHybridDigestGenerator } from "./enhanced-hybrid-digest.js";
+import { logger } from "../monitoring/logger.js";
+class FrameDigestIntegration {
+  frameManager;
+  digestGenerator;
+  frameActivityMap = /* @__PURE__ */ new Map();
+  constructor(frameManager, db, llmProvider) {
+    this.frameManager = frameManager;
+    this.digestGenerator = new EnhancedHybridDigestGenerator(
+      db,
+      {
+        enableAIGeneration: true,
+        maxTokens: 200
+      },
+      llmProvider
+    );
+    this.setupHooks();
+  }
+  /**
+   * Setup frame lifecycle hooks
+   */
+  setupHooks() {
+    const originalAddEvent = this.frameManager.addEvent.bind(this.frameManager);
+    this.frameManager.addEvent = (eventType, payload, frameId) => {
+      const result = originalAddEvent(eventType, payload, frameId);
+      if (eventType === "tool_call") {
+        this.digestGenerator.recordToolCall();
+      }
+      if (eventType === "user_message") {
+        this.digestGenerator.recordUserInput();
+      }
+      const targetFrameId = frameId || this.frameManager.getCurrentFrameId();
+      if (targetFrameId) {
+        this.frameActivityMap.set(targetFrameId, Date.now());
+      }
+      return result;
+    };
+    const originalCreateFrame = this.frameManager.createFrame.bind(
+      this.frameManager
+    );
+    this.frameManager.createFrame = (options) => {
+      const frameId = originalCreateFrame(options);
+      this.digestGenerator.onFrameOpened(frameId);
+      this.frameActivityMap.set(frameId, Date.now());
+      return frameId;
+    };
+    const originalCloseFrame = this.frameManager.closeFrame.bind(
+      this.frameManager
+    );
+    this.frameManager.closeFrame = (frameId, outputs) => {
+      const targetFrameId = frameId || this.frameManager.getCurrentFrameId();
+      if (targetFrameId) {
+        const digest = this.generateEnhancedDigest(targetFrameId);
+        const enhancedOutputs = {
+          ...outputs,
+          digest: digest.json,
+          digestText: digest.text
+        };
+        this.digestGenerator.onFrameClosed(targetFrameId);
+        this.frameActivityMap.delete(targetFrameId);
+        originalCloseFrame(frameId, enhancedOutputs);
+      } else {
+        originalCloseFrame(frameId, outputs);
+      }
+    };
+  }
+  /**
+   * Generate enhanced digest for a frame
+   */
+  generateEnhancedDigest(frameId) {
+    const frame = this.frameManager.getFrame(frameId);
+    if (!frame) {
+      logger.warn("Frame not found for digest generation", { frameId });
+      return { text: "", json: {} };
+    }
+    const events = this.frameManager.getFrameEvents(frameId);
+    const anchors = this.getFrameAnchors(frameId);
+    const digestInput = {
+      frame: this.convertFrame(frame),
+      events: events.map(this.convertEvent),
+      anchors: anchors.map(this.convertAnchor)
+    };
+    const hybridDigest = this.digestGenerator.generateDigest(digestInput);
+    return {
+      text: hybridDigest.text,
+      json: {
+        deterministic: hybridDigest.deterministic,
+        aiGenerated: hybridDigest.aiGenerated,
+        status: hybridDigest.status,
+        generatedAt: Date.now()
+      }
+    };
+  }
+  /**
+   * Convert FrameManager frame to DigestInput frame
+   */
+  convertFrame(frame) {
+    return frame;
+  }
+  /**
+   * Convert FrameManager event to DigestInput event
+   */
+  convertEvent(event) {
+    return event;
+  }
+  /**
+   * Convert FrameManager anchor to DigestInput anchor
+   */
+  convertAnchor(anchor) {
+    return anchor;
+  }
+  /**
+   * Calculate importance score based on frame characteristics
+   */
+  calculateImportanceScore(frame) {
+    let score = 0.5;
+    const typeScores = {
+      task: 0.6,
+      debug: 0.8,
+      review: 0.7,
+      write: 0.5,
+      tool_scope: 0.3,
+      subtask: 0.4
+    };
+    score = typeScores[frame.type] || score;
+    score -= frame.depth * 0.05;
+    if (frame.closed_at) {
+      const durationMinutes = (frame.closed_at - frame.created_at) / 60;
+      if (durationMinutes > 10) score += 0.1;
+      if (durationMinutes > 30) score += 0.1;
+    }
+    return Math.max(0, Math.min(1, score));
+  }
+  /**
+   * Get frame anchors (wrapper for proper typing)
+   */
+  getFrameAnchors(frameId) {
+    return [];
+  }
+  /**
+   * Handle user interruption
+   */
+  handleUserInterruption() {
+    this.digestGenerator.handleInterruption();
+  }
+  /**
+   * Get idle status
+   */
+  getIdleStatus() {
+    return this.digestGenerator.getIdleStatus();
+  }
+  /**
+   * Force process digest queue
+   */
+  async forceProcessQueue() {
+    await this.digestGenerator.forceProcessQueue();
+  }
+  /**
+   * Cleanup
+   */
+  shutdown() {
+    this.digestGenerator.shutdown();
+  }
+}
+function enhanceFrameManagerWithDigest(frameManager, db, llmProvider) {
+  return new FrameDigestIntegration(frameManager, db, llmProvider);
+}
+export {
+  FrameDigestIntegration,
+  enhanceFrameManagerWithDigest
+};
+//# sourceMappingURL=frame-digest-integration.js.map

package/dist/core/digest/frame-digest-integration.js.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../src/core/digest/frame-digest-integration.ts"],
+  "sourcesContent": ["/**\n * Integration module for Enhanced Hybrid Digest with Frame Manager\n * Connects the 60/40 digest generator with frame lifecycle events\n */\n\nimport Database from 'better-sqlite3';\nimport {\n  FrameManager,\n  Frame,\n  Event,\n  Anchor,\n} from '../context/frame-manager.js';\nimport { EnhancedHybridDigestGenerator } from './enhanced-hybrid-digest.js';\nimport { DigestInput, DigestLLMProvider } from './types.js';\nimport { logger } from '../monitoring/logger.js';\n\n/**\n * Frame Digest Integration\n * Enhances FrameManager with hybrid digest capabilities\n */\nexport class FrameDigestIntegration {\n  private frameManager: FrameManager;\n  private digestGenerator: EnhancedHybridDigestGenerator;\n  private frameActivityMap = new Map<string, number>();\n\n  constructor(\n    frameManager: FrameManager,\n    db: Database.Database,\n    llmProvider?: DigestLLMProvider\n  ) {\n    this.frameManager = frameManager;\n    this.digestGenerator = new EnhancedHybridDigestGenerator(\n      db,\n      {\n        enableAIGeneration: true,\n        maxTokens: 200,\n      },\n      llmProvider\n    );\n\n    this.setupHooks();\n  }\n\n  /**\n   * Setup frame lifecycle hooks\n   */\n  private setupHooks(): void {\n    // Track frame activity\n    const originalAddEvent = this.frameManager.addEvent.bind(this.frameManager);\n    this.frameManager.addEvent = (\n      eventType: Event['event_type'],\n      payload: Record<string, any>,\n      frameId?: string\n    ): string => {\n      const result = originalAddEvent(eventType, payload, frameId);\n\n      // Track tool calls for idle detection\n      if (eventType === 'tool_call') {\n        this.digestGenerator.recordToolCall();\n      }\n\n      // Track user messages\n      if (eventType === 'user_message') {\n        this.digestGenerator.recordUserInput();\n      }\n\n      // Track frame activity\n      const targetFrameId = frameId || this.frameManager.getCurrentFrameId();\n      if (targetFrameId) {\n        this.frameActivityMap.set(targetFrameId, Date.now());\n      }\n\n      return result;\n    };\n\n    // Hook into frame creation\n    const originalCreateFrame = this.frameManager.createFrame.bind(\n      this.frameManager\n    );\n    this.frameManager.createFrame = (options: any): string => {\n      const frameId = originalCreateFrame(options);\n      this.digestGenerator.onFrameOpened(frameId);\n      this.frameActivityMap.set(frameId, Date.now());\n      return frameId;\n    };\n\n    // Hook into frame closure\n    const originalCloseFrame = this.frameManager.closeFrame.bind(\n      this.frameManager\n    );\n    this.frameManager.closeFrame = (\n      frameId?: string,\n      outputs?: Record<string, any>\n    ): void => {\n      const targetFrameId = frameId || this.frameManager.getCurrentFrameId();\n\n      if (targetFrameId) {\n        // Generate enhanced digest\n        const digest = this.generateEnhancedDigest(targetFrameId);\n\n        // Merge digest outputs with provided outputs\n        const enhancedOutputs = {\n          ...outputs,\n          digest: digest.json,\n          digestText: digest.text,\n        };\n\n        // Notify digest generator of frame closure\n        this.digestGenerator.onFrameClosed(targetFrameId);\n        this.frameActivityMap.delete(targetFrameId);\n\n        // Call original with enhanced outputs\n        originalCloseFrame(frameId, enhancedOutputs);\n      } else {\n        originalCloseFrame(frameId, outputs);\n      }\n    };\n  }\n\n  /**\n   * Generate enhanced digest for a frame\n   */\n  private generateEnhancedDigest(frameId: string): {\n    text: string;\n    json: Record<string, any>;\n  } {\n    // Get frame data\n    const frame = this.frameManager.getFrame(frameId);\n    if (!frame) {\n      logger.warn('Frame not found for digest generation', { frameId });\n      return { text: '', json: {} };\n    }\n\n    // Get events and anchors\n    const events = this.frameManager.getFrameEvents(frameId);\n    const anchors = this.getFrameAnchors(frameId);\n\n    // Convert to digest input format\n    const digestInput: DigestInput = {\n      frame: this.convertFrame(frame),\n      events: events.map(this.convertEvent),\n      anchors: anchors.map(this.convertAnchor),\n    };\n\n    // Generate hybrid digest\n    const hybridDigest = this.digestGenerator.generateDigest(digestInput);\n\n    // Format for frame manager\n    return {\n      text: hybridDigest.text,\n      json: {\n        deterministic: hybridDigest.deterministic,\n        aiGenerated: hybridDigest.aiGenerated,\n        status: hybridDigest.status,\n        generatedAt: Date.now(),\n      },\n    };\n  }\n\n  /**\n   * Convert FrameManager frame to DigestInput frame\n   */\n  private convertFrame(frame: Frame): Frame {\n    // Frame types are the same - just pass through with type assertion\n    return frame;\n  }\n\n  /**\n   * Convert FrameManager event to DigestInput event\n   */\n  private convertEvent(event: Event): Event {\n    // Events are the same type - just pass through\n    return event;\n  }\n\n  /**\n   * Convert FrameManager anchor to DigestInput anchor\n   */\n  private convertAnchor(anchor: Anchor): Anchor {\n    // Anchors are the same type - just pass through\n    return anchor;\n  }\n\n  /**\n   * Calculate importance score based on frame characteristics\n   */\n  private calculateImportanceScore(frame: Frame): number {\n    let score = 0.5; // Base score\n\n    // Adjust based on frame type\n    const typeScores: Record<string, number> = {\n      task: 0.6,\n      debug: 0.8,\n      review: 0.7,\n      write: 0.5,\n      tool_scope: 0.3,\n      subtask: 0.4,\n    };\n    score = typeScores[frame.type] || score;\n\n    // Adjust based on depth (deeper frames are usually less important)\n    score -= frame.depth * 0.05;\n\n    // Adjust based on duration (longer frames might be more important)\n    if (frame.closed_at) {\n      const durationMinutes = (frame.closed_at - frame.created_at) / 60;\n      if (durationMinutes > 10) score += 0.1;\n      if (durationMinutes > 30) score += 0.1;\n    }\n\n    // Clamp between 0 and 1\n    return Math.max(0, Math.min(1, score));\n  }\n\n  /**\n   * Get frame anchors (wrapper for proper typing)\n   */\n  private getFrameAnchors(frameId: string): Anchor[] {\n    // This would typically use frameManager's method, but it's private\n    // For now, return empty array - in production would need to expose this\n    return [];\n  }\n\n  /**\n   * Handle user interruption\n   */\n  public handleUserInterruption(): void {\n    this.digestGenerator.handleInterruption();\n  }\n\n  /**\n   * Get idle status\n   */\n  public getIdleStatus(): ReturnType<\n    EnhancedHybridDigestGenerator['getIdleStatus']\n  > {\n    return this.digestGenerator.getIdleStatus();\n  }\n\n  /**\n   * Force process digest queue\n   */\n  public async forceProcessQueue(): Promise<void> {\n    await this.digestGenerator.forceProcessQueue();\n  }\n\n  /**\n   * Cleanup\n   */\n  public shutdown(): void {\n    this.digestGenerator.shutdown();\n  }\n}\n\n/**\n * Factory function to enhance existing FrameManager\n */\nexport function enhanceFrameManagerWithDigest(\n  frameManager: FrameManager,\n  db: Database.Database,\n  llmProvider?: DigestLLMProvider\n): FrameDigestIntegration {\n  return new FrameDigestIntegration(frameManager, db, llmProvider);\n}\n"],
+  "mappings": "AAYA,SAAS,qCAAqC;AAE9C,SAAS,cAAc;AAMhB,MAAM,uBAAuB;AAAA,EAC1B;AAAA,EACA;AAAA,EACA,mBAAmB,oBAAI,IAAoB;AAAA,EAEnD,YACE,cACA,IACA,aACA;AACA,SAAK,eAAe;AACpB,SAAK,kBAAkB,IAAI;AAAA,MACzB;AAAA,MACA;AAAA,QACE,oBAAoB;AAAA,QACpB,WAAW;AAAA,MACb;AAAA,MACA;AAAA,IACF;AAEA,SAAK,WAAW;AAAA,EAClB;AAAA;AAAA;AAAA;AAAA,EAKQ,aAAmB;AAEzB,UAAM,mBAAmB,KAAK,aAAa,SAAS,KAAK,KAAK,YAAY;AAC1E,SAAK,aAAa,WAAW,CAC3B,WACA,SACA,YACW;AACX,YAAM,SAAS,iBAAiB,WAAW,SAAS,OAAO;AAG3D,UAAI,cAAc,aAAa;AAC7B,aAAK,gBAAgB,eAAe;AAAA,MACtC;AAGA,UAAI,cAAc,gBAAgB;AAChC,aAAK,gBAAgB,gBAAgB;AAAA,MACvC;AAGA,YAAM,gBAAgB,WAAW,KAAK,aAAa,kBAAkB;AACrE,UAAI,eAAe;AACjB,aAAK,iBAAiB,IAAI,eAAe,KAAK,IAAI,CAAC;AAAA,MACrD;AAEA,aAAO;AAAA,IACT;AAGA,UAAM,sBAAsB,KAAK,aAAa,YAAY;AAAA,MACxD,KAAK;AAAA,IACP;AACA,SAAK,aAAa,cAAc,CAAC,YAAyB;AACxD,YAAM,UAAU,oBAAoB,OAAO;AAC3C,WAAK,gBAAgB,cAAc,OAAO;AAC1C,WAAK,iBAAiB,IAAI,SAAS,KAAK,IAAI,CAAC;AAC7C,aAAO;AAAA,IACT;AAGA,UAAM,qBAAqB,KAAK,aAAa,WAAW;AAAA,MACtD,KAAK;AAAA,IACP;AACA,SAAK,aAAa,aAAa,CAC7B,SACA,YACS;AACT,YAAM,gBAAgB,WAAW,KAAK,aAAa,kBAAkB;AAErE,UAAI,eAAe;AAEjB,cAAM,SAAS,KAAK,uBAAuB,aAAa;AAGxD,cAAM,kBAAkB;AAAA,UACtB,GAAG;AAAA,UACH,QAAQ,OAAO;AAAA,UACf,YAAY,OAAO;AAAA,QACrB;AAGA,aAAK,gBAAgB,cAAc,aAAa;AAChD,aAAK,iBAAiB,OAAO,aAAa;AAG1C,2BAAmB,SAAS,eAAe;AAAA,MAC7C,OAAO;AACL,2BAAmB,SAAS,OAAO;AAAA,MACrC;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,uBAAuB,SAG7B;AAEA,UAAM,QAAQ,KAAK,aAAa,SAAS,OAAO;AAChD,QAAI,CAAC,OAAO;AACV,aAAO,KAAK,yCAAyC,EAAE,QAAQ,CAAC;AAChE,aAAO,EAAE,MAAM,IAAI,MAAM,CAAC,EAAE;AAAA,IAC9B;AAGA,UAAM,SAAS,KAAK,aAAa,eAAe,OAAO;AACvD,UAAM,UAAU,KAAK,gBAAgB,OAAO;AAG5C,UAAM,cAA2B;AAAA,MAC/B,OAAO,KAAK,aAAa,KAAK;AAAA,MAC9B,QAAQ,OAAO,IAAI,KAAK,YAAY;AAAA,MACpC,SAAS,QAAQ,IAAI,KAAK,aAAa;AAAA,IACzC;AAGA,UAAM,eAAe,KAAK,gBAAgB,eAAe,WAAW;AAGpE,WAAO;AAAA,MACL,MAAM,aAAa;AAAA,MACnB,MAAM;AAAA,QACJ,eAAe,aAAa;AAAA,QAC5B,aAAa,aAAa;AAAA,QAC1B,QAAQ,aAAa;AAAA,QACrB,aAAa,KAAK,IAAI;AAAA,MACxB;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,aAAa,OAAqB;AAExC,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,aAAa,OAAqB;AAExC,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,cAAc,QAAwB;AAE5C,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,yBAAyB,OAAsB;AACrD,QAAI,QAAQ;AAGZ,UAAM,aAAqC;AAAA,MACzC,MAAM;AAAA,MACN,OAAO;AAAA,MACP,QAAQ;AAAA,MACR,OAAO;AAAA,MACP,YAAY;AAAA,MACZ,SAAS;AAAA,IACX;AACA,YAAQ,WAAW,MAAM,IAAI,KAAK;AAGlC,aAAS,MAAM,QAAQ;AAGvB,QAAI,MAAM,WAAW;AACnB,YAAM,mBAAmB,MAAM,YAAY,MAAM,cAAc;AAC/D,UAAI,kBAAkB,GAAI,UAAS;AACnC,UAAI,kBAAkB,GAAI,UAAS;AAAA,IACrC;AAGA,WAAO,KAAK,IAAI,GAAG,KAAK,IAAI,GAAG,KAAK,CAAC;AAAA,EACvC;AAAA;AAAA;AAAA;AAAA,EAKQ,gBAAgB,SAA2B;AAGjD,WAAO,CAAC;AAAA,EACV;AAAA;AAAA;AAAA;AAAA,EAKO,yBAA+B;AACpC,SAAK,gBAAgB,mBAAmB;AAAA,EAC1C;AAAA;AAAA;AAAA;AAAA,EAKO,gBAEL;AACA,WAAO,KAAK,gBAAgB,cAAc;AAAA,EAC5C;AAAA;AAAA;AAAA;AAAA,EAKA,MAAa,oBAAmC;AAC9C,UAAM,KAAK,gBAAgB,kBAAkB;AAAA,EAC/C;AAAA;AAAA;AAAA;AAAA,EAKO,WAAiB;AACtB,SAAK,gBAAgB,SAAS;AAAA,EAChC;AACF;AAKO,SAAS,8BACd,cACA,IACA,aACwB;AACxB,SAAO,IAAI,uBAAuB,cAAc,IAAI,WAAW;AACjE;",
+  "names": []
+}