Function createMockToolHandlerContext

src/test-utils/test-helpers.ts:41–83

()

}

export function createMockToolHandlerContext(): {
  ctx: ToolHandlerContext;
  result: MockToolHandlerResult;
  run: <T>(fn: () => Promise<T>) => Promise<T>;
} {
  const events: AnyFragment[] = [];
  const attachments: ImageAttachment[] = [];
  const session = createRenderSession('text');
  const ctx: ToolHandlerContext = {
    emit: (fragment: AnyFragment) => {
      events.push(fragment);
      session.emit(fragment);
    },
    attach: (image) => {
      attachments.push(image);
      session.attach(image);
    },
  };
  const resultObj: MockToolHandlerResult = {
    events,
    attachments,
    get nextStepParams() {
      return ctx.nextStepParams;
    },
    text() {
      return renderCliTextTranscript({
        items: [],
        structuredOutput: ctx.structuredOutput,
        nextSteps: ctx.nextSteps,
      });
    },
    isError() {
      return ctx.structuredOutput?.result.didError === true;
    },
  };
  return {
    ctx,
    result: resultObj,
    run: async <T>(fn: () => Promise<T>): Promise<T> => {
      return handlerContextStorage.run(ctx, fn);
    },
  };
}

export async function runToolLogic<T>(logic: () => Promise<T>): Promise<{
  response: T;