Function readViaLegacyAssembly

packages/server/src/routes/snapshot.ts:132–176

(ix: IInstantiationService, sid: string)

}

async function readViaLegacyAssembly(ix: IInstantiationService, sid: string) {
  return ix.invokeFunction(async (a) => {
    const broadcast = a.get(IWSBroadcastService);
    const sessionService = a.get(ISessionService);
    const messageService = a.get(IMessageService);
    const promptService = a.get(IPromptService);
    const approvals = a.get(IApprovalService) as ApprovalService;
    const questions = a.get(IQuestionService) as QuestionService;

    let snapState = await broadcast.getSnapshotState(sid);
    let session: Session | undefined;
    let items: Message[] = [];
    let hasMore = false;

    for (let attempt = 0; attempt < MAX_ASSEMBLY_ATTEMPTS; attempt++) {
      session = await sessionService.get(sid);
      const page = await messageService.list(sid, {
        page_size: SNAPSHOT_MESSAGE_PAGE_SIZE,
      });
      items = [...page.items].reverse();
      hasMore = page.has_more;

      const post = await broadcast.getSnapshotState(sid);
      const stable = post.seq === snapState.seq && post.epoch === snapState.epoch;
      snapState = post;
      if (stable) break;
    }

    const currentPromptId = promptService.getCurrentPromptId(sid);
    const inFlightTurn = snapState.inFlightTurn;
    if (inFlightTurn !== null && currentPromptId !== undefined) {
      inFlightTurn.current_prompt_id = currentPromptId;
    }

    return {
      as_of_seq: snapState.seq,
      epoch: snapState.epoch,
      session: session!,
      messages: { items, has_more: hasMore },
      in_flight_turn: inFlightTurn,
      pending_approvals: approvals.listPending(sid),
      pending_questions: questions.listPending(sid),
    };
  });
}