Async Memory Recall — 設計仕様書

type MemoryPrefetchHandle = {
  promise: Promise<RelevantAutoMemoryPromptResult>;
  /** promise.finally() により設定される。promise が確定するまで null。 */
  settledAt: number | null;
  /** メモリが注入された後に true — 二重注入を防ぐ。 */
  consumed: boolean;
  controller: AbortController;
};

// 新しいハンドルをセットする前に、前の UserQuery からの実行中プレフェッチを中断する
// （呼び出しが確定する前にユーザーが再度入力した場合の、孤立したサイドクエリを防ぐ）。
this.pendingMemoryPrefetch?.controller.abort();
this.pendingMemoryPrefetch = undefined;
 
const controller = new AbortController();
// 呼び出し元のシグナルをプレフェッチコントローラにブリッジする
// 親ターンでのユーザー中断（Ctrl-C / Esc）も呼び出しサイドクエリを終了させる。
const onParentAbort = () => controller.abort();
if (signal.aborted) {
  controller.abort();
} else {
  signal.addEventListener('abort', onParentAbort, { once: true });
}
 
const promise = this.config
  .getMemoryManager()
  .recall(projectRoot, partToString(request), {
    config: this.config,
    excludedFilePaths: this.surfacedRelevantAutoMemoryPaths,
    abortSignal: controller.signal,
  })
  .catch((error: unknown) => {
    if (!(error instanceof DOMException && error.name === 'AbortError')) {
      debugLogger.warn('Managed auto-memory recall prefetch failed.', error);
    }
    return EMPTY_RELEVANT_AUTO_MEMORY_RESULT;
  });
 
const handle: MemoryPrefetchHandle = {
  promise,
  settledAt: null,
  consumed: false,
  controller,
};
void promise.finally(() => {
  handle.settledAt = Date.now();
  signal.removeEventListener('abort', onParentAbort);
});
this.pendingMemoryPrefetch = handle;
// await しない — すぐに続行

const prefetchHandle = this.pendingMemoryPrefetch;
if (
  prefetchHandle &&
  prefetchHandle.settledAt !== null &&
  !prefetchHandle.consumed
) {
  prefetchHandle.consumed = true;
  this.pendingMemoryPrefetch = undefined;
  const result = await prefetchHandle.promise; // 既に確定済み、即座に返る
  if (result.prompt) {
    // unshift（push ではない）：メモリを systemReminders の先頭に保持し、
    // UserQuery ターンで system-reminder ブロックの先頭になるようにする。
    // （ToolResult ターンでは requestToSend に追記して
    // functionCall / functionResponse のペアリングを維持する — 以下参照）。
    systemReminders.unshift(result.prompt);
    for (const doc of result.selectedDocs) {
      this.surfacedRelevantAutoMemoryPaths.add(doc.filePath);
    }
  }
}

if (messageType === SendMessageType.ToolResult) {
  const prefetchHandle = this.pendingMemoryPrefetch;
  if (
    prefetchHandle &&
    prefetchHandle.settledAt !== null &&
    !prefetchHandle.consumed
  ) {
    prefetchHandle.consumed = true;
    this.pendingMemoryPrefetch = undefined;
    const result = await prefetchHandle.promise;
    if (result.prompt) {
      // 先頭ではなく追記（先頭にすると functionResponse 部分が前に来なくなる）。
      // ネイティブの Gemini パスでの functionCall/functionResponse の
      // ペアリングを壊さないようにする。
      requestToSend = [...requestToSend, result.prompt];
      for (const doc of result.selectedDocs) {
        this.surfacedRelevantAutoMemoryPaths.add(doc.filePath);
      }
    }
  }
}

this.pendingMemoryPrefetch?.controller.abort();
this.pendingMemoryPrefetch = undefined;

prefetchHandle.consumed = true;
this.pendingMemoryPrefetch = undefined;