非同期メモリリコール — 設計仕様

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) {
    // push ではなく unshift: UserQuery ターンでは system-reminder ブロックの
    // 先頭にメモリを配置するため。（ToolResult ターンでは functionCall /
    // functionResponse のペアリングを維持するために requestToSend に追加する
    // — 下記参照。）
    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 パーツを先頭に保ちつつ、モデルの
      // functionCall/functionResponse ペアリングをネイティブ Gemini パスで
      // 壊さないように追加（前置きではなく末尾に追加）する。
      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;