@aigne/cli 1.60.0-beta → 1.74.0-beta

package/dist/tracer/terminal.mjs

@@ -0,0 +1,332 @@
+import { AIGNE_HUB_CREDITS_NOT_ENOUGH_ERROR_TYPE } from "../constants.mjs";
+import { terminalInput } from "../ui/utils/terminal-input.mjs";
+import checkbox_default from "../utils/inquirer/checkbox.mjs";
+import { AIGNEListr, AIGNEListrRenderer } from "../utils/listr.mjs";
+import { highlightUrl } from "../utils/string-utils.mjs";
+import { parseDuration } from "../utils/time.mjs";
+import chalk from "chalk";
+import { figures } from "@aigne/listr2";
+import { withProtocol } from "ufo";
+import { AIAgent, ChatModel, DEFAULT_OUTPUT_FILE_KEY, DEFAULT_OUTPUT_KEY, UserAgent, mergeContextUsage, newEmptyContextUsage } from "@aigne/core";
+import { flat, omit } from "@aigne/core/utils/type-utils";
+import { EOL } from "node:os";
+import * as prompts from "@inquirer/prompts";
+import { inspect } from "node:util";
+import { promiseWithResolvers } from "@aigne/core/utils/promise";
+import { markedTerminal } from "@aigne/marked-terminal";
+import { Marked } from "marked";
+import terminalImage from "terminal-image";
+import terminalLink from "terminal-link";
+
+//#region src/tracer/terminal.ts
+const CREDITS_ERROR_PROCESSED_FLAG = "$credits_error_processed";
+var TerminalTracer = class {
+	constructor(context, options = {}) {
+		this.context = context;
+		this.options = options;
+	}
+	tasks = {};
+	async run(agent, input$1, options) {
+		await this.context.observer?.serve();
+		const context = this.context.newContext({ reset: true });
+		const listr = new AIGNEListr({
+			formatRequest: (options$1) => this.formatRequest(agent, context, input$1, options$1),
+			formatResult: (result, options$1) => this.formatResult(agent, context, result, options$1)
+		}, [], {
+			concurrent: true,
+			exitOnError: false,
+			registerSignalListeners: false
+		});
+		this.listr = listr;
+		const collapsedMap = /* @__PURE__ */ new Map();
+		const hideContextIds = /* @__PURE__ */ new Set();
+		const onStart = async ({ context: context$1, agent: agent$1, ...event }) => {
+			const result = { options: { prompts: this.proxiedPrompts } };
+			if (agent$1 instanceof UserAgent) return result;
+			if (agent$1.taskRenderMode === "hide") {
+				hideContextIds.add(context$1.id);
+				return result;
+			}
+			if (agent$1.taskRenderMode === "collapse") collapsedMap.set(context$1.id, {
+				ancestor: { contextId: context$1.id },
+				usage: newEmptyContextUsage(),
+				models: /* @__PURE__ */ new Set()
+			});
+			if (context$1.parentId) {
+				if (hideContextIds.has(context$1.parentId)) {
+					hideContextIds.add(context$1.id);
+					return result;
+				}
+				const collapsed = collapsedMap.get(context$1.parentId);
+				if (collapsed) {
+					collapsedMap.set(context$1.id, collapsed);
+					return result;
+				}
+			}
+			const contextId = context$1.id;
+			const parentContextId = context$1.parentId;
+			const task = {
+				...promiseWithResolvers(),
+				agent: agent$1,
+				input: event.input,
+				listr: promiseWithResolvers(),
+				startTime: Date.now()
+			};
+			this.tasks[contextId] = task;
+			const listrTask = {
+				title: await this.formatTaskTitle(agent$1, { ...event }),
+				task: (ctx, taskWrapper) => {
+					const subtask = taskWrapper.newListr([{ task: () => task.promise }]);
+					task.listr.resolve({
+						subtask,
+						taskWrapper,
+						ctx
+					});
+					return subtask;
+				},
+				rendererOptions: {
+					persistentOutput: true,
+					outputBar: Number.POSITIVE_INFINITY,
+					bottomBar: Number.POSITIVE_INFINITY
+				}
+			};
+			const parentTask = parentContextId ? this.tasks[parentContextId] : void 0;
+			if (parentTask) parentTask.listr.promise.then(({ subtask }) => {
+				subtask.add(listrTask);
+			});
+			else listr.add(listrTask);
+			return result;
+		};
+		const onSuccess = async ({ context: context$1, agent: agent$1, output, ...event }) => {
+			const contextId = context$1.id;
+			const parentContextId = context$1.parentId;
+			const collapsed = collapsedMap.get(contextId);
+			if (collapsed) {
+				if (agent$1 instanceof ChatModel) {
+					const { usage, model } = output;
+					if (usage) mergeContextUsage(collapsed.usage, usage);
+					if (model) collapsed.models.add(model);
+				}
+				const task$1 = this.tasks[collapsed.ancestor.contextId];
+				if (task$1) {
+					task$1.usage = collapsed.usage;
+					task$1.extraTitleMetadata ??= {};
+					if (collapsed.models.size) task$1.extraTitleMetadata.model = [...collapsed.models].join(",");
+					const { taskWrapper: taskWrapper$1 } = await task$1.listr.promise;
+					taskWrapper$1.title = await this.formatTaskTitle(task$1.agent, {
+						input: task$1.input,
+						task: task$1,
+						usage: Boolean(task$1.usage.inputTokens || task$1.usage.outputTokens || task$1.usage.aigneHubCredits || task$1.usage.cacheCreationInputTokens || task$1.usage.cacheReadInputTokens),
+						time: context$1.id === collapsed.ancestor.contextId
+					});
+					if (context$1.id === collapsed.ancestor.contextId) task$1?.resolve();
+					return;
+				}
+			}
+			const task = this.tasks[contextId];
+			if (!task) return;
+			task.endTime = Date.now();
+			const { taskWrapper, ctx } = await task.listr.promise;
+			if (agent$1 instanceof ChatModel) {
+				const { usage, model } = output;
+				task.usage = usage;
+				task.extraTitleMetadata ??= {};
+				if (model) task.extraTitleMetadata.model = model;
+			}
+			taskWrapper.title = await this.formatTaskTitle(agent$1, {
+				...event,
+				task,
+				usage: true,
+				time: true
+			});
+			if (!parentContextId || !this.tasks[parentContextId]) Object.assign(ctx, output);
+			task.resolve();
+		};
+		let retryPromptPromise;
+		const onError = async ({ context: context$1, agent: agent$1, error, ...event }) => {
+			if ("type" in error && error.type === AIGNE_HUB_CREDITS_NOT_ENOUGH_ERROR_TYPE) {
+				if (!Object.hasOwn(error, CREDITS_ERROR_PROCESSED_FLAG)) {
+					Object.defineProperty(error, CREDITS_ERROR_PROCESSED_FLAG, {
+						value: true,
+						enumerable: false
+					});
+					const retry = await this.promptBuyCredits(error);
+					console.log("");
+					if (retry === "retry") return { retry: true };
+				}
+			}
+			if (agent$1 instanceof ChatModel) {
+				retryPromptPromise ??= this.proxiedPrompts.select({
+					message: chalk.red(`Error: ${error.message}`),
+					choices: [{
+						value: "retry",
+						name: "Retry"
+					}, {
+						value: "exit",
+						name: "Exit"
+					}]
+				}).then((result) => ({ retry: result === "retry" })).finally(() => {
+					retryPromptPromise = void 0;
+				});
+				const { retry } = await retryPromptPromise;
+				if (retry) return { retry: true };
+			}
+			const contextId = context$1.id;
+			const task = this.tasks[contextId];
+			if (!task) return;
+			task.endTime = Date.now();
+			const { taskWrapper } = await task.listr.promise;
+			taskWrapper.title = await this.formatTaskTitle(agent$1, {
+				...event,
+				task,
+				usage: true,
+				time: true
+			});
+			task.reject(error);
+		};
+		return {
+			result: await listr.run(() => context.invoke(agent, input$1, {
+				...options,
+				hooks: flat({
+					priority: "high",
+					onStart,
+					onSuccess,
+					onError
+				}, options?.hooks),
+				streaming: true,
+				newContext: false
+			})),
+			context
+		};
+	}
+	listr;
+	proxiedPrompts = new Proxy({}, { get: (_target, prop) => {
+		const method = prop === "checkbox" ? checkbox_default : prop === "input" ? terminalInput : prompts[prop];
+		if (typeof method !== "function") throw new Error(`Unsupported prompt method ${String(prop)}`);
+		return async (config) => {
+			const renderer = this.listr?.["renderer"] instanceof AIGNEListrRenderer ? this.listr["renderer"] : void 0;
+			await renderer?.pause();
+			try {
+				return await method({ ...config });
+			} finally {
+				await renderer?.resume();
+			}
+		};
+	} });
+	buyCreditsPromptPromise;
+	async promptBuyCredits(error) {
+		this.buyCreditsPromptPromise ??= (async () => {
+			return await this.proxiedPrompts.select({
+				message: highlightUrl(error.message),
+				choices: [{
+					name: "I have bought some credits, try again",
+					value: "retry"
+				}, {
+					name: "Exit",
+					value: "exit"
+				}]
+			});
+		})();
+		return this.buyCreditsPromptPromise.catch(() => "exit").finally(() => {
+			this.buyCreditsPromptPromise = void 0;
+		});
+	}
+	formatAigneHubCredits(credits, creditPrefix) {
+		const formattedCredits = credits % 1 !== 0 ? parseFloat(credits.toFixed(6)).toString() : credits.toFixed();
+		if (creditPrefix) return [chalk.grey("cost:"), chalk.blue(`${creditPrefix}${formattedCredits}`)];
+		return [chalk.blue(formattedCredits), chalk.grey("AIGNE Hub credits")];
+	}
+	formatTokenUsage(usage, extra) {
+		const content = [
+			[chalk.yellow(usage.inputTokens), chalk.grey("input tokens")],
+			usage.cacheReadInputTokens ? [chalk.green(usage.cacheReadInputTokens), chalk.grey("cached")] : void 0,
+			usage.cacheCreationInputTokens ? [chalk.yellow(usage.cacheCreationInputTokens), chalk.grey("cache write")] : void 0,
+			[chalk.cyan(usage.outputTokens), chalk.grey("output tokens")],
+			usage.aigneHubCredits ? this.formatAigneHubCredits(usage.aigneHubCredits, usage.creditPrefix) : void 0,
+			usage.agentCalls ? [chalk.magenta(usage.agentCalls), chalk.grey("agent calls")] : void 0
+		].filter((i) => !!i).map((i) => i.join(" "));
+		if (extra) content.unshift(...Object.entries(extra).filter(([k, v]) => k && v).map(([k, v]) => `${chalk.grey(k)}: ${v}`));
+		return `${chalk.grey("(")}${content.join(chalk.green(", "))}${chalk.grey(")")}`;
+	}
+	formatTimeUsage(startTime, endTime) {
+		const duration = endTime - startTime;
+		return chalk.grey(`[${parseDuration(duration)}]`);
+	}
+	async formatTaskTitle(agent, { task, usage, time, input: input$1 }) {
+		let title = agent.name;
+		if (agent.taskTitle) title += ` ${chalk.cyan(await agent.renderTaskTitle(input$1))}`;
+		if (usage && task?.usage) title += ` ${this.formatTokenUsage(task.usage, task.extraTitleMetadata)}`;
+		if (time && task?.startTime && task.endTime) title += ` ${this.formatTimeUsage(task.startTime, task.endTime)}`;
+		return title;
+	}
+	marked = new Marked().use({ walkTokens: (token) => {
+		if (token.type === "code") {
+			if (typeof token.lang === "string") token.lang = token.lang.trim().split(/\s+/)[0];
+		}
+	} }, markedTerminal({ forceHyperLink: false }, { theme: { string: chalk.green } }));
+	get outputKey() {
+		return this.options.outputKey || DEFAULT_OUTPUT_KEY;
+	}
+	get outputFileKey() {
+		return this.options.outputFileKey || DEFAULT_OUTPUT_FILE_KEY;
+	}
+	formatRequest(agent, _context, m = {}, { running = false } = {}) {
+		const prefix = `${chalk.grey(figures.pointer)} 💬 `;
+		const inputKey = agent instanceof AIAgent ? agent.inputKey : void 0;
+		const msg = inputKey ? m[inputKey] : void 0;
+		const message = inputKey ? omit(m, inputKey) : m;
+		const r = [msg && typeof msg === "string" ? this.marked.parse(msg, { async: false }).trim() : void 0, Object.keys(message).length > 0 ? inspect(message, {
+			colors: true,
+			...running ? this.runningInspectOptions : void 0
+		}) : void 0].filter(Boolean).join(EOL).trim();
+		if (!r) return void 0;
+		return `${prefix}${r}`;
+	}
+	formatResult(agent, context, m = {}, { running = false, renderImage = false } = {}) {
+		const { isTTY } = process.stdout;
+		const outputKey = this.outputKey || (agent instanceof AIAgent ? agent.outputKey : void 0);
+		const prefix = `${chalk.grey(figures.tick)} 🤖 ${this.formatTokenUsage(context.usage)}`;
+		const msg = outputKey ? m[outputKey] : void 0;
+		const message = outputKey ? omit(m, outputKey, this.outputFileKey) : m;
+		const text = msg && typeof msg === "string" ? isTTY ? this.marked.parse(msg, { async: false }).trim() : msg : void 0;
+		const json = Object.keys(message).length > 0 ? inspect(message, {
+			colors: isTTY,
+			...running ? this.runningInspectOptions : void 0
+		}) : void 0;
+		if (renderImage) return this.formatResultData(m).then((images) => {
+			return [
+				prefix,
+				text,
+				images,
+				json
+			].filter(Boolean).join(EOL.repeat(2));
+		});
+		return [
+			prefix,
+			text,
+			json
+		].filter(Boolean).join(EOL.repeat(2));
+	}
+	async formatResultData(output) {
+		const files = output[this.outputFileKey];
+		if (!Array.isArray(files)) return;
+		const options = { height: 30 };
+		return (await Promise.all(files.map(async (item) => {
+			const image = item.type === "local" ? await terminalImage.file(item.path, options) : item.type === "file" ? await terminalImage.buffer(Buffer.from(item.data, "base64"), options) : void 0;
+			const link = item.type === "local" ? withProtocol(item.path, "file://") : item.type === "url" ? item.url : void 0;
+			return [image, [
+				link ? chalk.cyan(terminalLink(link, link)) : void 0,
+				item.filename,
+				item.mimeType ? chalk.gray(item.mimeType) : void 0
+			].filter(Boolean).join(" ")].filter(Boolean).join(EOL);
+		}))).filter(Boolean).join(EOL);
+	}
+	runningInspectOptions = {
+		maxArrayLength: 3,
+		maxStringLength: 200
+	};
+};
+
+//#endregion
+export { TerminalTracer };
+//# sourceMappingURL=terminal.mjs.map

package/dist/tracer/terminal.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"terminal.mjs","names":["input","options","agent","context","task","checkbox"],"sources":["../../src/tracer/terminal.ts"],"sourcesContent":["import { EOL } from \"node:os\";\nimport { type InspectOptions, inspect } from \"node:util\";\nimport {\n  type Agent,\n  type AgentHooks,\n  AIAgent,\n  ChatModel,\n  type ChatModelOutput,\n  type Context,\n  type ContextUsage,\n  DEFAULT_OUTPUT_FILE_KEY,\n  DEFAULT_OUTPUT_KEY,\n  type FileUnionContent,\n  type InvokeOptions,\n  type Message,\n  mergeContextUsage,\n  newEmptyContextUsage,\n  UserAgent,\n} from \"@aigne/core\";\nimport { promiseWithResolvers } from \"@aigne/core/utils/promise\";\nimport { flat, omit } from \"@aigne/core/utils/type-utils\";\nimport { figures, type Listr } from \"@aigne/listr2\";\nimport { markedTerminal } from \"@aigne/marked-terminal\";\nimport * as prompts from \"@inquirer/prompts\";\nimport chalk from \"chalk\";\nimport { Marked } from \"marked\";\nimport terminalImage from \"terminal-image\";\nimport terminalLink from \"terminal-link\";\nimport { withProtocol } from \"ufo\";\nimport { AIGNE_HUB_CREDITS_NOT_ENOUGH_ERROR_TYPE } from \"../constants.js\";\nimport { terminalInput } from \"../ui/utils/terminal-input.js\";\nimport checkbox from \"../utils/inquirer/checkbox.js\";\nimport { AIGNEListr, AIGNEListrRenderer, type AIGNEListrTaskWrapper } from \"../utils/listr.js\";\nimport { highlightUrl } from \"../utils/string-utils.js\";\nimport { parseDuration } from \"../utils/time.js\";\n\nconst CREDITS_ERROR_PROCESSED_FLAG = \"$credits_error_processed\";\n\nexport interface TerminalTracerOptions {\n  outputKey?: string;\n  outputFileKey?: string;\n}\n\nexport class TerminalTracer {\n  constructor(\n    public readonly context: Context,\n    public readonly options: TerminalTracerOptions = {},\n  ) {}\n\n  private tasks: { [callId: string]: Task } = {};\n\n  async run(agent: Agent, input: Message, options?: InvokeOptions) {\n    await this.context.observer?.serve();\n\n    const context = this.context.newContext({ reset: true });\n\n    const listr = new AIGNEListr(\n      {\n        formatRequest: (options?: { running?: boolean }) =>\n          this.formatRequest(agent, context, input, options),\n        formatResult: (result, options?: { running?: boolean; renderImage?: boolean }) =>\n          this.formatResult(agent, context, result, options),\n      },\n      [],\n      { concurrent: true, exitOnError: false, registerSignalListeners: false },\n    );\n    this.listr = listr;\n\n    const collapsedMap = new Map<\n      string,\n      { ancestor: { contextId: string }; usage: ContextUsage; models: Set<string> }\n    >();\n    const hideContextIds = new Set<string>();\n\n    const onStart: AgentHooks[\"onStart\"] = async ({ context, agent, ...event }) => {\n      const result = { options: { prompts: this.proxiedPrompts } };\n\n      if (agent instanceof UserAgent) return result;\n\n      if (agent.taskRenderMode === \"hide\") {\n        hideContextIds.add(context.id);\n        return result;\n      }\n\n      if (agent.taskRenderMode === \"collapse\") {\n        collapsedMap.set(context.id, {\n          ancestor: { contextId: context.id },\n          usage: newEmptyContextUsage(),\n          models: new Set(),\n        });\n      }\n\n      if (context.parentId) {\n        if (hideContextIds.has(context.parentId)) {\n          hideContextIds.add(context.id);\n          return result;\n        }\n\n        const collapsed = collapsedMap.get(context.parentId);\n        if (collapsed) {\n          collapsedMap.set(context.id, collapsed);\n          return result;\n        }\n      }\n\n      const contextId = context.id;\n      const parentContextId = context.parentId;\n\n      const task: Task = {\n        ...promiseWithResolvers(),\n        agent,\n        input: event.input,\n        listr: promiseWithResolvers(),\n        startTime: Date.now(),\n      };\n      this.tasks[contextId] = task;\n\n      const listrTask: Parameters<typeof listr.add>[0] = {\n        title: await this.formatTaskTitle(agent, { ...event }),\n        task: (ctx, taskWrapper) => {\n          const subtask = taskWrapper.newListr([{ task: () => task.promise }]);\n          task.listr.resolve({ subtask, taskWrapper, ctx });\n          return subtask;\n        },\n        rendererOptions: {\n          persistentOutput: true,\n          outputBar: Number.POSITIVE_INFINITY,\n          bottomBar: Number.POSITIVE_INFINITY,\n        },\n      };\n\n      const parentTask = parentContextId ? this.tasks[parentContextId] : undefined;\n      if (parentTask) {\n        parentTask.listr.promise.then(({ subtask }) => {\n          subtask.add(listrTask);\n        });\n      } else {\n        listr.add(listrTask);\n      }\n\n      return result;\n    };\n\n    const onSuccess: AgentHooks[\"onSuccess\"] = async ({ context, agent, output, ...event }) => {\n      const contextId = context.id;\n      const parentContextId = context.parentId;\n\n      const collapsed = collapsedMap.get(contextId);\n      if (collapsed) {\n        if (agent instanceof ChatModel) {\n          const { usage, model } = output as ChatModelOutput;\n          if (usage) mergeContextUsage(collapsed.usage, usage);\n          if (model) collapsed.models.add(model);\n        }\n\n        const task = this.tasks[collapsed.ancestor.contextId];\n        if (task) {\n          task.usage = collapsed.usage;\n          task.extraTitleMetadata ??= {};\n          if (collapsed.models.size)\n            task.extraTitleMetadata.model = [...collapsed.models].join(\",\");\n\n          const { taskWrapper } = await task.listr.promise;\n\n          taskWrapper.title = await this.formatTaskTitle(task.agent, {\n            input: task.input,\n            task,\n            usage: Boolean(\n              task.usage.inputTokens ||\n                task.usage.outputTokens ||\n                task.usage.aigneHubCredits ||\n                task.usage.cacheCreationInputTokens ||\n                task.usage.cacheReadInputTokens,\n            ),\n            time: context.id === collapsed.ancestor.contextId,\n          });\n\n          if (context.id === collapsed.ancestor.contextId) {\n            task?.resolve();\n          }\n          return;\n        }\n      }\n\n      const task = this.tasks[contextId];\n      if (!task) return;\n\n      task.endTime = Date.now();\n\n      const { taskWrapper, ctx } = await task.listr.promise;\n\n      if (agent instanceof ChatModel) {\n        const { usage, model } = output as ChatModelOutput;\n        task.usage = usage;\n        task.extraTitleMetadata ??= {};\n        if (model) task.extraTitleMetadata.model = model;\n      }\n\n      taskWrapper.title = await this.formatTaskTitle(agent, {\n        ...event,\n        task,\n        usage: true,\n        time: true,\n      });\n\n      if (!parentContextId || !this.tasks[parentContextId]) {\n        Object.assign(ctx, output);\n      }\n\n      task.resolve();\n    };\n\n    let retryPromptPromise: Promise<{ retry?: boolean }> | undefined;\n\n    const onError: AgentHooks[\"onError\"] = async ({ context, agent, error, ...event }) => {\n      if (\"type\" in error && error.type === AIGNE_HUB_CREDITS_NOT_ENOUGH_ERROR_TYPE) {\n        if (!Object.hasOwn(error, CREDITS_ERROR_PROCESSED_FLAG)) {\n          Object.defineProperty(error, CREDITS_ERROR_PROCESSED_FLAG, {\n            value: true,\n            enumerable: false,\n          });\n\n          const retry = await this.promptBuyCredits(error);\n\n          console.log(\"\");\n\n          if (retry === \"retry\") {\n            return { retry: true };\n          }\n        }\n      }\n\n      if (agent instanceof ChatModel) {\n        retryPromptPromise ??= this.proxiedPrompts\n          .select({\n            message: chalk.red(`Error: ${error.message}`),\n            choices: [\n              { value: \"retry\", name: \"Retry\" },\n              { value: \"exit\", name: \"Exit\" },\n            ],\n          })\n          .then((result) => ({ retry: result === \"retry\" }))\n          .finally(() => {\n            retryPromptPromise = undefined;\n          });\n\n        const { retry } = await retryPromptPromise;\n        if (retry) return { retry: true };\n      }\n\n      const contextId = context.id;\n\n      const task = this.tasks[contextId];\n      if (!task) return;\n\n      task.endTime = Date.now();\n\n      const { taskWrapper } = await task.listr.promise;\n      taskWrapper.title = await this.formatTaskTitle(agent, {\n        ...event,\n        task,\n        usage: true,\n        time: true,\n      });\n\n      task.reject(error);\n    };\n\n    const result = await listr.run(() =>\n      context.invoke(agent, input, {\n        ...options,\n        hooks: flat(\n          {\n            priority: \"high\",\n            onStart,\n            onSuccess,\n            onError,\n          },\n          options?.hooks,\n        ),\n        streaming: true,\n        newContext: false,\n      }),\n    );\n\n    return { result, context };\n  }\n\n  private listr?: AIGNEListr;\n\n  private proxiedPrompts = new Proxy(\n    {},\n    {\n      get: (_target, prop) => {\n        const method =\n          prop === \"checkbox\"\n            ? checkbox\n            : prop === \"input\"\n              ? terminalInput\n              : // biome-ignore lint/performance/noDynamicNamespaceImportAccess: we need to access prompts dynamically\n                (prompts[prop as keyof typeof prompts] as (...args: any[]) => any);\n        if (typeof method !== \"function\")\n          throw new Error(`Unsupported prompt method ${String(prop)}`);\n\n        return async (config: any) => {\n          const renderer =\n            this.listr?.[\"renderer\"] instanceof AIGNEListrRenderer\n              ? this.listr[\"renderer\"]\n              : undefined;\n          await renderer?.pause();\n\n          try {\n            return await method({ ...config });\n          } finally {\n            await renderer?.resume();\n          }\n        };\n      },\n    },\n  ) as typeof prompts;\n\n  private buyCreditsPromptPromise: Promise<\"retry\" | \"exit\"> | undefined;\n\n  private async promptBuyCredits(error: Error) {\n    // Avoid multiple agents asking for credits, we will only show the prompt once\n    this.buyCreditsPromptPromise ??= (async () => {\n      const retry = await this.proxiedPrompts.select({\n        message: highlightUrl(error.message),\n        choices: [\n          {\n            name: \"I have bought some credits, try again\",\n            value: \"retry\" as const,\n          },\n          {\n            name: \"Exit\",\n            value: \"exit\" as const,\n          },\n        ],\n      });\n\n      return retry;\n    })();\n\n    return this.buyCreditsPromptPromise\n      .catch(() => \"exit\")\n      .finally(() => {\n        // Clear the promise so that we can show the prompt again if needed\n        this.buyCreditsPromptPromise = undefined;\n      });\n  }\n\n  formatAigneHubCredits(credits: number, creditPrefix?: string): [string, string] {\n    const hasDecimal = credits % 1 !== 0;\n    const formattedCredits = hasDecimal\n      ? parseFloat(credits.toFixed(6)).toString()\n      : credits.toFixed();\n\n    if (creditPrefix) {\n      return [chalk.grey(\"cost:\"), chalk.blue(`${creditPrefix}${formattedCredits}`)];\n    }\n\n    return [chalk.blue(formattedCredits), chalk.grey(\"AIGNE Hub credits\")];\n  }\n\n  formatTokenUsage(usage: Partial<ContextUsage>, extra?: { [key: string]: string }) {\n    const items = [\n      [chalk.yellow(usage.inputTokens), chalk.grey(\"input tokens\")],\n      usage.cacheReadInputTokens\n        ? [chalk.green(usage.cacheReadInputTokens), chalk.grey(\"cached\")]\n        : undefined,\n      usage.cacheCreationInputTokens\n        ? [chalk.yellow(usage.cacheCreationInputTokens), chalk.grey(\"cache write\")]\n        : undefined,\n      [chalk.cyan(usage.outputTokens), chalk.grey(\"output tokens\")],\n      usage.aigneHubCredits\n        ? this.formatAigneHubCredits(usage.aigneHubCredits, usage.creditPrefix)\n        : undefined,\n      usage.agentCalls ? [chalk.magenta(usage.agentCalls), chalk.grey(\"agent calls\")] : undefined,\n    ];\n\n    const content = items.filter((i) => !!i).map((i) => i.join(\" \"));\n\n    if (extra) {\n      content.unshift(\n        ...Object.entries(extra)\n          .filter(([k, v]) => k && v)\n          .map(([k, v]) => `${chalk.grey(k)}: ${v}`),\n      );\n    }\n\n    return `${chalk.grey(\"(\")}${content.join(chalk.green(\", \"))}${chalk.grey(\")\")}`;\n  }\n\n  formatTimeUsage(startTime: number, endTime: number) {\n    const duration = endTime - startTime;\n    return chalk.grey(`[${parseDuration(duration)}]`);\n  }\n\n  async formatTaskTitle(\n    agent: Agent,\n    { task, usage, time, input }: { task?: Task; usage?: boolean; time?: boolean; input: Message },\n  ) {\n    let title = agent.name;\n\n    if (agent.taskTitle) {\n      title += ` ${chalk.cyan(await agent.renderTaskTitle(input))}`;\n    }\n\n    if (usage && task?.usage)\n      title += ` ${this.formatTokenUsage(task.usage, task.extraTitleMetadata)}`;\n    if (time && task?.startTime && task.endTime)\n      title += ` ${this.formatTimeUsage(task.startTime, task.endTime)}`;\n\n    return title;\n  }\n\n  private marked = new Marked().use(\n    {\n      // marked-terminal does not support code block meta, so we need to strip it\n      walkTokens: (token) => {\n        if (token.type === \"code\") {\n          if (typeof token.lang === \"string\") {\n            token.lang = token.lang.trim().split(/\\s+/)[0];\n          }\n        }\n      },\n    },\n    markedTerminal(\n      { forceHyperLink: false },\n      {\n        theme: {\n          string: chalk.green,\n        },\n      },\n    ),\n  );\n\n  get outputKey() {\n    return this.options.outputKey || DEFAULT_OUTPUT_KEY;\n  }\n\n  get outputFileKey() {\n    return this.options.outputFileKey || DEFAULT_OUTPUT_FILE_KEY;\n  }\n\n  formatRequest(agent: Agent, _context: Context, m: Message = {}, { running = false } = {}) {\n    const prefix = `${chalk.grey(figures.pointer)} 💬 `;\n\n    const inputKey = agent instanceof AIAgent ? agent.inputKey : undefined;\n\n    const msg = inputKey ? m[inputKey] : undefined;\n    const message = inputKey ? omit(m, inputKey) : m;\n\n    const text =\n      msg && typeof msg === \"string\" ? this.marked.parse(msg, { async: false }).trim() : undefined;\n\n    const json =\n      Object.keys(message).length > 0\n        ? inspect(message, { colors: true, ...(running ? this.runningInspectOptions : undefined) })\n        : undefined;\n\n    const r = [text, json].filter(Boolean).join(EOL).trim();\n    if (!r) return undefined;\n\n    return `${prefix}${r}`;\n  }\n\n  formatResult(\n    agent: Agent,\n    context: Context,\n    m: Message = {},\n    { running = false, renderImage = false } = {},\n  ): string | Promise<string> {\n    const { isTTY } = process.stdout;\n    const outputKey = this.outputKey || (agent instanceof AIAgent ? agent.outputKey : undefined);\n\n    const prefix = `${chalk.grey(figures.tick)} 🤖 ${this.formatTokenUsage(context.usage)}`;\n\n    const msg = outputKey ? m[outputKey] : undefined;\n    const message = outputKey ? omit(m, outputKey, this.outputFileKey) : m;\n\n    const text =\n      msg && typeof msg === \"string\"\n        ? isTTY\n          ? this.marked.parse(msg, { async: false }).trim()\n          : msg\n        : undefined;\n\n    const json =\n      Object.keys(message).length > 0\n        ? inspect(message, { colors: isTTY, ...(running ? this.runningInspectOptions : undefined) })\n        : undefined;\n\n    if (renderImage) {\n      return this.formatResultData(m).then((images) => {\n        return [prefix, text, images, json].filter(Boolean).join(EOL.repeat(2));\n      });\n    }\n\n    return [prefix, text, json].filter(Boolean).join(EOL.repeat(2));\n  }\n\n  async formatResultData(output: Message): Promise<string | undefined> {\n    const files = output[this.outputFileKey] as FileUnionContent[];\n    if (!Array.isArray(files)) return;\n\n    const options: Parameters<typeof terminalImage.file>[1] = {\n      height: 30,\n    };\n\n    return (\n      await Promise.all(\n        files.map(async (item) => {\n          const image =\n            item.type === \"local\"\n              ? await terminalImage.file(item.path, options)\n              : item.type === \"file\"\n                ? await terminalImage.buffer(Buffer.from(item.data, \"base64\"), options)\n                : undefined;\n\n          const link =\n            item.type === \"local\"\n              ? withProtocol(item.path, \"file://\")\n              : item.type === \"url\"\n                ? item.url\n                : undefined;\n          const text = [\n            link ? chalk.cyan(terminalLink(link, link)) : undefined,\n            item.filename,\n            item.mimeType ? chalk.gray(item.mimeType) : undefined,\n          ]\n            .filter(Boolean)\n            .join(\" \");\n\n          return [image, text].filter(Boolean).join(EOL);\n        }),\n      )\n    )\n      .filter(Boolean)\n      .join(EOL);\n  }\n\n  protected runningInspectOptions: InspectOptions = {\n    maxArrayLength: 3,\n    maxStringLength: 200,\n  };\n}\n\ntype Task = ReturnType<typeof promiseWithResolvers<void>> & {\n  listr: ReturnType<\n    typeof promiseWithResolvers<{\n      ctx: object;\n      subtask: Listr;\n      taskWrapper: AIGNEListrTaskWrapper;\n    }>\n  >;\n  agent: Agent;\n  input: Message;\n  startTime?: number;\n  endTime?: number;\n  usage?: Partial<ContextUsage>;\n  extraTitleMetadata?: { [key: string]: string };\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAoCA,MAAM,+BAA+B;AAOrC,IAAa,iBAAb,MAA4B;CAC1B,YACE,AAAgB,SAChB,AAAgB,UAAiC,EAAE,EACnD;EAFgB;EACA;;CAGlB,AAAQ,QAAoC,EAAE;CAE9C,MAAM,IAAI,OAAc,SAAgB,SAAyB;AAC/D,QAAM,KAAK,QAAQ,UAAU,OAAO;EAEpC,MAAM,UAAU,KAAK,QAAQ,WAAW,EAAE,OAAO,MAAM,CAAC;EAExD,MAAM,QAAQ,IAAI,WAChB;GACE,gBAAgB,cACd,KAAK,cAAc,OAAO,SAASA,SAAOC,UAAQ;GACpD,eAAe,QAAQ,cACrB,KAAK,aAAa,OAAO,SAAS,QAAQA,UAAQ;GACrD,EACD,EAAE,EACF;GAAE,YAAY;GAAM,aAAa;GAAO,yBAAyB;GAAO,CACzE;AACD,OAAK,QAAQ;EAEb,MAAM,+BAAe,IAAI,KAGtB;EACH,MAAM,iCAAiB,IAAI,KAAa;EAExC,MAAM,UAAiC,OAAO,EAAE,oBAAS,gBAAO,GAAG,YAAY;GAC7E,MAAM,SAAS,EAAE,SAAS,EAAE,SAAS,KAAK,gBAAgB,EAAE;AAE5D,OAAIC,mBAAiB,UAAW,QAAO;AAEvC,OAAIA,QAAM,mBAAmB,QAAQ;AACnC,mBAAe,IAAIC,UAAQ,GAAG;AAC9B,WAAO;;AAGT,OAAID,QAAM,mBAAmB,WAC3B,cAAa,IAAIC,UAAQ,IAAI;IAC3B,UAAU,EAAE,WAAWA,UAAQ,IAAI;IACnC,OAAO,sBAAsB;IAC7B,wBAAQ,IAAI,KAAK;IAClB,CAAC;AAGJ,OAAIA,UAAQ,UAAU;AACpB,QAAI,eAAe,IAAIA,UAAQ,SAAS,EAAE;AACxC,oBAAe,IAAIA,UAAQ,GAAG;AAC9B,YAAO;;IAGT,MAAM,YAAY,aAAa,IAAIA,UAAQ,SAAS;AACpD,QAAI,WAAW;AACb,kBAAa,IAAIA,UAAQ,IAAI,UAAU;AACvC,YAAO;;;GAIX,MAAM,YAAYA,UAAQ;GAC1B,MAAM,kBAAkBA,UAAQ;GAEhC,MAAM,OAAa;IACjB,GAAG,sBAAsB;IACzB;IACA,OAAO,MAAM;IACb,OAAO,sBAAsB;IAC7B,WAAW,KAAK,KAAK;IACtB;AACD,QAAK,MAAM,aAAa;GAExB,MAAM,YAA6C;IACjD,OAAO,MAAM,KAAK,gBAAgBD,SAAO,EAAE,GAAG,OAAO,CAAC;IACtD,OAAO,KAAK,gBAAgB;KAC1B,MAAM,UAAU,YAAY,SAAS,CAAC,EAAE,YAAY,KAAK,SAAS,CAAC,CAAC;AACpE,UAAK,MAAM,QAAQ;MAAE;MAAS;MAAa;MAAK,CAAC;AACjD,YAAO;;IAET,iBAAiB;KACf,kBAAkB;KAClB,WAAW,OAAO;KAClB,WAAW,OAAO;KACnB;IACF;GAED,MAAM,aAAa,kBAAkB,KAAK,MAAM,mBAAmB;AACnE,OAAI,WACF,YAAW,MAAM,QAAQ,MAAM,EAAE,cAAc;AAC7C,YAAQ,IAAI,UAAU;KACtB;OAEF,OAAM,IAAI,UAAU;AAGtB,UAAO;;EAGT,MAAM,YAAqC,OAAO,EAAE,oBAAS,gBAAO,QAAQ,GAAG,YAAY;GACzF,MAAM,YAAYC,UAAQ;GAC1B,MAAM,kBAAkBA,UAAQ;GAEhC,MAAM,YAAY,aAAa,IAAI,UAAU;AAC7C,OAAI,WAAW;AACb,QAAID,mBAAiB,WAAW;KAC9B,MAAM,EAAE,OAAO,UAAU;AACzB,SAAI,MAAO,mBAAkB,UAAU,OAAO,MAAM;AACpD,SAAI,MAAO,WAAU,OAAO,IAAI,MAAM;;IAGxC,MAAME,SAAO,KAAK,MAAM,UAAU,SAAS;AAC3C,QAAIA,QAAM;AACR,YAAK,QAAQ,UAAU;AACvB,YAAK,uBAAuB,EAAE;AAC9B,SAAI,UAAU,OAAO,KACnB,QAAK,mBAAmB,QAAQ,CAAC,GAAG,UAAU,OAAO,CAAC,KAAK,IAAI;KAEjE,MAAM,EAAE,+BAAgB,MAAMA,OAAK,MAAM;AAEzC,mBAAY,QAAQ,MAAM,KAAK,gBAAgBA,OAAK,OAAO;MACzD,OAAOA,OAAK;MACZ;MACA,OAAO,QACLA,OAAK,MAAM,eACTA,OAAK,MAAM,gBACXA,OAAK,MAAM,mBACXA,OAAK,MAAM,4BACXA,OAAK,MAAM,qBACd;MACD,MAAMD,UAAQ,OAAO,UAAU,SAAS;MACzC,CAAC;AAEF,SAAIA,UAAQ,OAAO,UAAU,SAAS,UACpC,SAAM,SAAS;AAEjB;;;GAIJ,MAAM,OAAO,KAAK,MAAM;AACxB,OAAI,CAAC,KAAM;AAEX,QAAK,UAAU,KAAK,KAAK;GAEzB,MAAM,EAAE,aAAa,QAAQ,MAAM,KAAK,MAAM;AAE9C,OAAID,mBAAiB,WAAW;IAC9B,MAAM,EAAE,OAAO,UAAU;AACzB,SAAK,QAAQ;AACb,SAAK,uBAAuB,EAAE;AAC9B,QAAI,MAAO,MAAK,mBAAmB,QAAQ;;AAG7C,eAAY,QAAQ,MAAM,KAAK,gBAAgBA,SAAO;IACpD,GAAG;IACH;IACA,OAAO;IACP,MAAM;IACP,CAAC;AAEF,OAAI,CAAC,mBAAmB,CAAC,KAAK,MAAM,iBAClC,QAAO,OAAO,KAAK,OAAO;AAG5B,QAAK,SAAS;;EAGhB,IAAI;EAEJ,MAAM,UAAiC,OAAO,EAAE,oBAAS,gBAAO,OAAO,GAAG,YAAY;AACpF,OAAI,UAAU,SAAS,MAAM,SAAS,yCACpC;QAAI,CAAC,OAAO,OAAO,OAAO,6BAA6B,EAAE;AACvD,YAAO,eAAe,OAAO,8BAA8B;MACzD,OAAO;MACP,YAAY;MACb,CAAC;KAEF,MAAM,QAAQ,MAAM,KAAK,iBAAiB,MAAM;AAEhD,aAAQ,IAAI,GAAG;AAEf,SAAI,UAAU,QACZ,QAAO,EAAE,OAAO,MAAM;;;AAK5B,OAAIA,mBAAiB,WAAW;AAC9B,2BAAuB,KAAK,eACzB,OAAO;KACN,SAAS,MAAM,IAAI,UAAU,MAAM,UAAU;KAC7C,SAAS,CACP;MAAE,OAAO;MAAS,MAAM;MAAS,EACjC;MAAE,OAAO;MAAQ,MAAM;MAAQ,CAChC;KACF,CAAC,CACD,MAAM,YAAY,EAAE,OAAO,WAAW,SAAS,EAAE,CACjD,cAAc;AACb,0BAAqB;MACrB;IAEJ,MAAM,EAAE,UAAU,MAAM;AACxB,QAAI,MAAO,QAAO,EAAE,OAAO,MAAM;;GAGnC,MAAM,YAAYC,UAAQ;GAE1B,MAAM,OAAO,KAAK,MAAM;AACxB,OAAI,CAAC,KAAM;AAEX,QAAK,UAAU,KAAK,KAAK;GAEzB,MAAM,EAAE,gBAAgB,MAAM,KAAK,MAAM;AACzC,eAAY,QAAQ,MAAM,KAAK,gBAAgBD,SAAO;IACpD,GAAG;IACH;IACA,OAAO;IACP,MAAM;IACP,CAAC;AAEF,QAAK,OAAO,MAAM;;AAoBpB,SAAO;GAAE,QAjBM,MAAM,MAAM,UACzB,QAAQ,OAAO,OAAOF,SAAO;IAC3B,GAAG;IACH,OAAO,KACL;KACE,UAAU;KACV;KACA;KACA;KACD,EACD,SAAS,MACV;IACD,WAAW;IACX,YAAY;IACb,CAAC,CACH;GAEgB;GAAS;;CAG5B,AAAQ;CAER,AAAQ,iBAAiB,IAAI,MAC3B,EAAE,EACF,EACE,MAAM,SAAS,SAAS;EACtB,MAAM,SACJ,SAAS,aACLK,mBACA,SAAS,UACP,gBAEC,QAAQ;AACjB,MAAI,OAAO,WAAW,WACpB,OAAM,IAAI,MAAM,6BAA6B,OAAO,KAAK,GAAG;AAE9D,SAAO,OAAO,WAAgB;GAC5B,MAAM,WACJ,KAAK,QAAQ,uBAAuB,qBAChC,KAAK,MAAM,cACX;AACN,SAAM,UAAU,OAAO;AAEvB,OAAI;AACF,WAAO,MAAM,OAAO,EAAE,GAAG,QAAQ,CAAC;aAC1B;AACR,UAAM,UAAU,QAAQ;;;IAI/B,CACF;CAED,AAAQ;CAER,MAAc,iBAAiB,OAAc;AAE3C,OAAK,6BAA6B,YAAY;AAe5C,UAdc,MAAM,KAAK,eAAe,OAAO;IAC7C,SAAS,aAAa,MAAM,QAAQ;IACpC,SAAS,CACP;KACE,MAAM;KACN,OAAO;KACR,EACD;KACE,MAAM;KACN,OAAO;KACR,CACF;IACF,CAAC;MAGA;AAEJ,SAAO,KAAK,wBACT,YAAY,OAAO,CACnB,cAAc;AAEb,QAAK,0BAA0B;IAC/B;;CAGN,sBAAsB,SAAiB,cAAyC;EAE9E,MAAM,mBADa,UAAU,MAAM,IAE/B,WAAW,QAAQ,QAAQ,EAAE,CAAC,CAAC,UAAU,GACzC,QAAQ,SAAS;AAErB,MAAI,aACF,QAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,MAAM,KAAK,GAAG,eAAe,mBAAmB,CAAC;AAGhF,SAAO,CAAC,MAAM,KAAK,iBAAiB,EAAE,MAAM,KAAK,oBAAoB,CAAC;;CAGxE,iBAAiB,OAA8B,OAAmC;EAgBhF,MAAM,UAfQ;GACZ,CAAC,MAAM,OAAO,MAAM,YAAY,EAAE,MAAM,KAAK,eAAe,CAAC;GAC7D,MAAM,uBACF,CAAC,MAAM,MAAM,MAAM,qBAAqB,EAAE,MAAM,KAAK,SAAS,CAAC,GAC/D;GACJ,MAAM,2BACF,CAAC,MAAM,OAAO,MAAM,yBAAyB,EAAE,MAAM,KAAK,cAAc,CAAC,GACzE;GACJ,CAAC,MAAM,KAAK,MAAM,aAAa,EAAE,MAAM,KAAK,gBAAgB,CAAC;GAC7D,MAAM,kBACF,KAAK,sBAAsB,MAAM,iBAAiB,MAAM,aAAa,GACrE;GACJ,MAAM,aAAa,CAAC,MAAM,QAAQ,MAAM,WAAW,EAAE,MAAM,KAAK,cAAc,CAAC,GAAG;GACnF,CAEqB,QAAQ,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,MAAM,EAAE,KAAK,IAAI,CAAC;AAEhE,MAAI,MACF,SAAQ,QACN,GAAG,OAAO,QAAQ,MAAM,CACrB,QAAQ,CAAC,GAAG,OAAO,KAAK,EAAE,CAC1B,KAAK,CAAC,GAAG,OAAO,GAAG,MAAM,KAAK,EAAE,CAAC,IAAI,IAAI,CAC7C;AAGH,SAAO,GAAG,MAAM,KAAK,IAAI,GAAG,QAAQ,KAAK,MAAM,MAAM,KAAK,CAAC,GAAG,MAAM,KAAK,IAAI;;CAG/E,gBAAgB,WAAmB,SAAiB;EAClD,MAAM,WAAW,UAAU;AAC3B,SAAO,MAAM,KAAK,IAAI,cAAc,SAAS,CAAC,GAAG;;CAGnD,MAAM,gBACJ,OACA,EAAE,MAAM,OAAO,MAAM,kBACrB;EACA,IAAI,QAAQ,MAAM;AAElB,MAAI,MAAM,UACR,UAAS,IAAI,MAAM,KAAK,MAAM,MAAM,gBAAgBL,QAAM,CAAC;AAG7D,MAAI,SAAS,MAAM,MACjB,UAAS,IAAI,KAAK,iBAAiB,KAAK,OAAO,KAAK,mBAAmB;AACzE,MAAI,QAAQ,MAAM,aAAa,KAAK,QAClC,UAAS,IAAI,KAAK,gBAAgB,KAAK,WAAW,KAAK,QAAQ;AAEjE,SAAO;;CAGT,AAAQ,SAAS,IAAI,QAAQ,CAAC,IAC5B,EAEE,aAAa,UAAU;AACrB,MAAI,MAAM,SAAS,QACjB;OAAI,OAAO,MAAM,SAAS,SACxB,OAAM,OAAO,MAAM,KAAK,MAAM,CAAC,MAAM,MAAM,CAAC;;IAInD,EACD,eACE,EAAE,gBAAgB,OAAO,EACzB,EACE,OAAO,EACL,QAAQ,MAAM,OACf,EACF,CACF,CACF;CAED,IAAI,YAAY;AACd,SAAO,KAAK,QAAQ,aAAa;;CAGnC,IAAI,gBAAgB;AAClB,SAAO,KAAK,QAAQ,iBAAiB;;CAGvC,cAAc,OAAc,UAAmB,IAAa,EAAE,EAAE,EAAE,UAAU,UAAU,EAAE,EAAE;EACxF,MAAM,SAAS,GAAG,MAAM,KAAK,QAAQ,QAAQ,CAAC;EAE9C,MAAM,WAAW,iBAAiB,UAAU,MAAM,WAAW;EAE7D,MAAM,MAAM,WAAW,EAAE,YAAY;EACrC,MAAM,UAAU,WAAW,KAAK,GAAG,SAAS,GAAG;EAU/C,MAAM,IAAI,CAPR,OAAO,OAAO,QAAQ,WAAW,KAAK,OAAO,MAAM,KAAK,EAAE,OAAO,OAAO,CAAC,CAAC,MAAM,GAAG,QAGnF,OAAO,KAAK,QAAQ,CAAC,SAAS,IAC1B,QAAQ,SAAS;GAAE,QAAQ;GAAM,GAAI,UAAU,KAAK,wBAAwB;GAAY,CAAC,GACzF,OAEgB,CAAC,OAAO,QAAQ,CAAC,KAAK,IAAI,CAAC,MAAM;AACvD,MAAI,CAAC,EAAG,QAAO;AAEf,SAAO,GAAG,SAAS;;CAGrB,aACE,OACA,SACA,IAAa,EAAE,EACf,EAAE,UAAU,OAAO,cAAc,UAAU,EAAE,EACnB;EAC1B,MAAM,EAAE,UAAU,QAAQ;EAC1B,MAAM,YAAY,KAAK,cAAc,iBAAiB,UAAU,MAAM,YAAY;EAElF,MAAM,SAAS,GAAG,MAAM,KAAK,QAAQ,KAAK,CAAC,MAAM,KAAK,iBAAiB,QAAQ,MAAM;EAErF,MAAM,MAAM,YAAY,EAAE,aAAa;EACvC,MAAM,UAAU,YAAY,KAAK,GAAG,WAAW,KAAK,cAAc,GAAG;EAErE,MAAM,OACJ,OAAO,OAAO,QAAQ,WAClB,QACE,KAAK,OAAO,MAAM,KAAK,EAAE,OAAO,OAAO,CAAC,CAAC,MAAM,GAC/C,MACF;EAEN,MAAM,OACJ,OAAO,KAAK,QAAQ,CAAC,SAAS,IAC1B,QAAQ,SAAS;GAAE,QAAQ;GAAO,GAAI,UAAU,KAAK,wBAAwB;GAAY,CAAC,GAC1F;AAEN,MAAI,YACF,QAAO,KAAK,iBAAiB,EAAE,CAAC,MAAM,WAAW;AAC/C,UAAO;IAAC;IAAQ;IAAM;IAAQ;IAAK,CAAC,OAAO,QAAQ,CAAC,KAAK,IAAI,OAAO,EAAE,CAAC;IACvE;AAGJ,SAAO;GAAC;GAAQ;GAAM;GAAK,CAAC,OAAO,QAAQ,CAAC,KAAK,IAAI,OAAO,EAAE,CAAC;;CAGjE,MAAM,iBAAiB,QAA8C;EACnE,MAAM,QAAQ,OAAO,KAAK;AAC1B,MAAI,CAAC,MAAM,QAAQ,MAAM,CAAE;EAE3B,MAAM,UAAoD,EACxD,QAAQ,IACT;AAED,UACE,MAAM,QAAQ,IACZ,MAAM,IAAI,OAAO,SAAS;GACxB,MAAM,QACJ,KAAK,SAAS,UACV,MAAM,cAAc,KAAK,KAAK,MAAM,QAAQ,GAC5C,KAAK,SAAS,SACZ,MAAM,cAAc,OAAO,OAAO,KAAK,KAAK,MAAM,SAAS,EAAE,QAAQ,GACrE;GAER,MAAM,OACJ,KAAK,SAAS,UACV,aAAa,KAAK,MAAM,UAAU,GAClC,KAAK,SAAS,QACZ,KAAK,MACL;AASR,UAAO,CAAC,OARK;IACX,OAAO,MAAM,KAAK,aAAa,MAAM,KAAK,CAAC,GAAG;IAC9C,KAAK;IACL,KAAK,WAAW,MAAM,KAAK,KAAK,SAAS,GAAG;IAC7C,CACE,OAAO,QAAQ,CACf,KAAK,IAAI,CAEQ,CAAC,OAAO,QAAQ,CAAC,KAAK,IAAI;IAC9C,CACH,EAEA,OAAO,QAAQ,CACf,KAAK,IAAI;;CAGd,AAAU,wBAAwC;EAChD,gBAAgB;EAChB,iBAAiB;EAClB"}

package/dist/type.d.cts

@@ -0,0 +1,10 @@
+import { AgentInvokeOptions as AgentInvokeOptions$1, UserContext } from "@aigne/core";
+import * as prompts from "@inquirer/prompts";
+
+//#region src/type.d.ts
+interface AgentInvokeOptions<U extends UserContext = UserContext> extends AgentInvokeOptions$1<U> {
+  prompts: typeof prompts;
+}
+//#endregion
+export { AgentInvokeOptions };
+//# sourceMappingURL=type.d.cts.map

package/dist/type.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"type.d.cts","names":[],"sources":["../src/type.ts"],"mappings":";;;;UAGiB,kBAAA,WAA6B,WAAA,GAAc,WAAA,UAClD,oBAAA,CAAoB,CAAA;EAAA,OAAA,SACZ,OAAA;AAAA"}

package/dist/type.d.mts

@@ -0,0 +1,10 @@
+import { AgentInvokeOptions as AgentInvokeOptions$1, UserContext } from "@aigne/core";
+import * as prompts from "@inquirer/prompts";
+
+//#region src/type.d.ts
+interface AgentInvokeOptions<U extends UserContext = UserContext> extends AgentInvokeOptions$1<U> {
+  prompts: typeof prompts;
+}
+//#endregion
+export { AgentInvokeOptions };
+//# sourceMappingURL=type.d.mts.map

package/dist/type.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"type.d.mts","names":[],"sources":["../src/type.ts"],"mappings":";;;;UAGiB,kBAAA,WAA6B,WAAA,GAAc,WAAA,UAClD,oBAAA,CAAoB,CAAA;EAAA,OAAA,SACZ,OAAA;AAAA"}

package/dist/type.mjs

@@ -0,0 +1 @@
+export {  };

package/dist/ui/utils/terminal-input.cjs

@@ -0,0 +1,145 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+const require_text_buffer = require('./text-buffer.cjs');
+let chalk = require("chalk");
+chalk = require_rolldown_runtime.__toESM(chalk);
+let _inquirer_core = require("@inquirer/core");
+let ink = require("ink");
+let react = require("react");
+let react_jsx_runtime = require("react/jsx-runtime");
+
+//#region src/ui/utils/terminal-input.tsx
+async function terminalInput({ render: r = ink.render, ...options } = {}) {
+	return new Promise((resolve, reject) => {
+		const handleSigInt = () => {
+			reject(/* @__PURE__ */ new Error("Input aborted"));
+		};
+		process.addListener("SIGINT", handleSigInt);
+		const clean = () => process.removeListener("SIGINT", handleSigInt);
+		const app = r(/* @__PURE__ */ (0, react_jsx_runtime.jsx)(TerminalInput, {
+			...options,
+			onSubmit: (value) => {
+				if (options.clear) app.clear();
+				app.unmount();
+				resolve(value);
+				clean();
+			},
+			onError: (error) => {
+				if (options.clear) app.clear();
+				app.unmount();
+				reject(error);
+				clean();
+			}
+		}), { exitOnCtrlC: false });
+	});
+}
+function TerminalInput(props) {
+	const buffer = require_text_buffer.useTextBuffer({
+		initialText: props.default || "",
+		initialCursorOffset: props.default?.length || 0,
+		isValidPath: () => false,
+		viewport: {
+			width: 80,
+			height: 1
+		}
+	});
+	const textRef = (0, react.useRef)(buffer.text);
+	textRef.current = buffer.text;
+	const [status, setStatus] = (0, react.useState)("input");
+	const [errorMessage, setErrorMessage] = (0, react.useState)();
+	(0, ink.useInput)((character, key) => {
+		if (character === "c" && key.ctrl) {
+			setStatus("error");
+			setTimeout(() => {
+				props.onError(new _inquirer_core.ExitPromptError("Input aborted by user"));
+			});
+			return;
+		}
+		if (key.return) {
+			const input = textRef.current || props.default || "";
+			setStatus("validating");
+			setErrorMessage(void 0);
+			const validateInput = async () => {
+				try {
+					if (props.required && !input.trim()) {
+						setErrorMessage("You must provide a value");
+						setStatus("input");
+						return;
+					}
+					if (props.validate) {
+						const result = await props.validate(input);
+						if (result !== true) {
+							setErrorMessage(typeof result === "string" ? result : "You must provide a valid value");
+							setStatus("input");
+							return;
+						}
+					}
+					setStatus("success");
+					setTimeout(() => {
+						props.onSubmit(input);
+					});
+				} catch (error) {
+					setErrorMessage(error instanceof Error ? error.message : "Validation error");
+					setStatus("input");
+				}
+			};
+			validateInput();
+			return;
+		} else if (key.backspace) buffer.backspace();
+		else if (key.delete) buffer.backspace();
+		else if (key.downArrow) buffer.move("down");
+		else if (key.upArrow) buffer.move("up");
+		else if (key.leftArrow) buffer.move("left");
+		else if (key.rightArrow) buffer.move("right");
+		else if (character === "a" && key.ctrl) buffer.move("home");
+		else if (character === "e" && key.ctrl) buffer.move("end");
+		else {
+			buffer.handleInput({
+				...key,
+				name: character,
+				sequence: character,
+				paste: false
+			});
+			setErrorMessage(void 0);
+		}
+	});
+	const lines = [...buffer.lines];
+	if (status === "input") {
+		const [row, col] = buffer.cursor;
+		const currentLine = lines[row] || "";
+		lines[row] = currentLine.slice(0, col) + chalk.default.inverse(currentLine[col] || " ") + currentLine.slice(col + 1);
+	}
+	const label = props.message && chalk.default.bold(props.message);
+	const inline = props.inline !== false;
+	return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(ink.Box, {
+		flexDirection: "column",
+		children: [/* @__PURE__ */ (0, react_jsx_runtime.jsxs)(ink.Box, {
+			flexDirection: inline ? "row" : "column",
+			children: [/* @__PURE__ */ (0, react_jsx_runtime.jsxs)(ink.Text, { children: [
+				PREFIX[status],
+				" ",
+				!inline && label
+			] }), /* @__PURE__ */ (0, react_jsx_runtime.jsx)(ink.Box, {
+				flexShrink: 1,
+				flexGrow: 1,
+				marginLeft: inline ? 0 : 2,
+				marginRight: 1,
+				children: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)(ink.Text, { children: [!!label && inline && `${label} `, lines.join("\n")] })
+			})]
+		}), errorMessage && /* @__PURE__ */ (0, react_jsx_runtime.jsx)(ink.Box, {
+			marginLeft: 2,
+			children: /* @__PURE__ */ (0, react_jsx_runtime.jsx)(ink.Text, {
+				color: "red",
+				children: errorMessage
+			})
+		})]
+	});
+}
+const PREFIX = {
+	input: chalk.default.blue("?"),
+	validating: chalk.default.yellow("⋯"),
+	success: chalk.default.green("✔"),
+	error: chalk.default.red("✘")
+};
+
+//#endregion
+exports.terminalInput = terminalInput;