@tonyclaw/agent-inspector 4.0.7 → 4.0.9

package/.output/server/_libs/mariozechner__pi-agent-core.mjs

@@ -1,824 +1 @@
-import "./typebox.mjs";
-import { s as streamSimple, v as validateToolArguments } from "./mariozechner__pi-ai.mjs";
-import "./partial-json.mjs";
-import "./anthropic-ai__sdk.mjs";
-import "./openai.mjs";
-import "./google__genai.mjs";
-import "./p-retry.mjs";
-import "./react.mjs";
-import "./retry.mjs";
-import "./google-auth-library.mjs";
-import "child_process";
-import "path";
-import "querystring";
-import "fs";
-import "./gaxios.mjs";
-import "https";
-import "./extend.mjs";
-import "stream";
-import "./gcp-metadata.mjs";
-import "os";
-import "./json-bigint.mjs";
-import "./bignumber.js.mjs";
-import "./google-logging-utils.mjs";
-import "process";
-import "events";
-import "util";
-import "./base64-js.mjs";
-import "crypto";
-import "./ecdsa-sig-formatter.mjs";
-import "./safe-buffer.mjs";
-import "buffer";
-import "./jws.mjs";
-import "./jwa.mjs";
-import "./buffer-equal-constant-time.mjs";
-import "fs/promises";
-import "node:stream";
-import "node:stream/promises";
-import "./ws.mjs";
-import "http";
-import "net";
-import "tls";
-import "url";
-import "zlib";
-import "./mistralai__mistralai.mjs";
-import "./zod.mjs";
-import "./zod-to-json-schema.mjs";
-import "./opentelemetry__api.mjs";
-import "./@opentelemetry/semantic-conventions+[...].mjs";
-async function runAgentLoop(prompts, context, config, emit, signal, streamFn) {
-  const newMessages = [...prompts];
-  const currentContext = {
-    ...context,
-    messages: [...context.messages, ...prompts]
-  };
-  await emit({ type: "agent_start" });
-  await emit({ type: "turn_start" });
-  for (const prompt of prompts) {
-    await emit({ type: "message_start", message: prompt });
-    await emit({ type: "message_end", message: prompt });
-  }
-  await runLoop(currentContext, newMessages, config, signal, emit, streamFn);
-  return newMessages;
-}
-async function runAgentLoopContinue(context, config, emit, signal, streamFn) {
-  if (context.messages.length === 0) {
-    throw new Error("Cannot continue: no messages in context");
-  }
-  if (context.messages[context.messages.length - 1].role === "assistant") {
-    throw new Error("Cannot continue from message role: assistant");
-  }
-  const newMessages = [];
-  const currentContext = { ...context };
-  await emit({ type: "agent_start" });
-  await emit({ type: "turn_start" });
-  await runLoop(currentContext, newMessages, config, signal, emit, streamFn);
-  return newMessages;
-}
-async function runLoop(currentContext, newMessages, config, signal, emit, streamFn) {
-  let firstTurn = true;
-  let pendingMessages = await config.getSteeringMessages?.() || [];
-  while (true) {
-    let hasMoreToolCalls = true;
-    while (hasMoreToolCalls || pendingMessages.length > 0) {
-      if (!firstTurn) {
-        await emit({ type: "turn_start" });
-      } else {
-        firstTurn = false;
-      }
-      if (pendingMessages.length > 0) {
-        for (const message2 of pendingMessages) {
-          await emit({ type: "message_start", message: message2 });
-          await emit({ type: "message_end", message: message2 });
-          currentContext.messages.push(message2);
-          newMessages.push(message2);
-        }
-        pendingMessages = [];
-      }
-      const message = await streamAssistantResponse(currentContext, config, signal, emit, streamFn);
-      newMessages.push(message);
-      if (message.stopReason === "error" || message.stopReason === "aborted") {
-        await emit({ type: "turn_end", message, toolResults: [] });
-        await emit({ type: "agent_end", messages: newMessages });
-        return;
-      }
-      const toolCalls = message.content.filter((c) => c.type === "toolCall");
-      const toolResults = [];
-      hasMoreToolCalls = false;
-      if (toolCalls.length > 0) {
-        const executedToolBatch = await executeToolCalls(currentContext, message, config, signal, emit);
-        toolResults.push(...executedToolBatch.messages);
-        hasMoreToolCalls = !executedToolBatch.terminate;
-        for (const result of toolResults) {
-          currentContext.messages.push(result);
-          newMessages.push(result);
-        }
-      }
-      await emit({ type: "turn_end", message, toolResults });
-      if (await config.shouldStopAfterTurn?.({
-        message,
-        toolResults,
-        context: currentContext,
-        newMessages
-      })) {
-        await emit({ type: "agent_end", messages: newMessages });
-        return;
-      }
-      pendingMessages = await config.getSteeringMessages?.() || [];
-    }
-    const followUpMessages = await config.getFollowUpMessages?.() || [];
-    if (followUpMessages.length > 0) {
-      pendingMessages = followUpMessages;
-      continue;
-    }
-    break;
-  }
-  await emit({ type: "agent_end", messages: newMessages });
-}
-async function streamAssistantResponse(context, config, signal, emit, streamFn) {
-  let messages = context.messages;
-  if (config.transformContext) {
-    messages = await config.transformContext(messages, signal);
-  }
-  const llmMessages = await config.convertToLlm(messages);
-  const llmContext = {
-    systemPrompt: context.systemPrompt,
-    messages: llmMessages,
-    tools: context.tools
-  };
-  const streamFunction = streamFn || streamSimple;
-  const resolvedApiKey = (config.getApiKey ? await config.getApiKey(config.model.provider) : void 0) || config.apiKey;
-  const response = await streamFunction(config.model, llmContext, {
-    ...config,
-    apiKey: resolvedApiKey,
-    signal
-  });
-  let partialMessage = null;
-  let addedPartial = false;
-  for await (const event of response) {
-    switch (event.type) {
-      case "start":
-        partialMessage = event.partial;
-        context.messages.push(partialMessage);
-        addedPartial = true;
-        await emit({ type: "message_start", message: { ...partialMessage } });
-        break;
-      case "text_start":
-      case "text_delta":
-      case "text_end":
-      case "thinking_start":
-      case "thinking_delta":
-      case "thinking_end":
-      case "toolcall_start":
-      case "toolcall_delta":
-      case "toolcall_end":
-        if (partialMessage) {
-          partialMessage = event.partial;
-          context.messages[context.messages.length - 1] = partialMessage;
-          await emit({
-            type: "message_update",
-            assistantMessageEvent: event,
-            message: { ...partialMessage }
-          });
-        }
-        break;
-      case "done":
-      case "error": {
-        const finalMessage2 = await response.result();
-        if (addedPartial) {
-          context.messages[context.messages.length - 1] = finalMessage2;
-        } else {
-          context.messages.push(finalMessage2);
-        }
-        if (!addedPartial) {
-          await emit({ type: "message_start", message: { ...finalMessage2 } });
-        }
-        await emit({ type: "message_end", message: finalMessage2 });
-        return finalMessage2;
-      }
-    }
-  }
-  const finalMessage = await response.result();
-  if (addedPartial) {
-    context.messages[context.messages.length - 1] = finalMessage;
-  } else {
-    context.messages.push(finalMessage);
-    await emit({ type: "message_start", message: { ...finalMessage } });
-  }
-  await emit({ type: "message_end", message: finalMessage });
-  return finalMessage;
-}
-async function executeToolCalls(currentContext, assistantMessage, config, signal, emit) {
-  const toolCalls = assistantMessage.content.filter((c) => c.type === "toolCall");
-  const hasSequentialToolCall = toolCalls.some((tc) => currentContext.tools?.find((t) => t.name === tc.name)?.executionMode === "sequential");
-  if (config.toolExecution === "sequential" || hasSequentialToolCall) {
-    return executeToolCallsSequential(currentContext, assistantMessage, toolCalls, config, signal, emit);
-  }
-  return executeToolCallsParallel(currentContext, assistantMessage, toolCalls, config, signal, emit);
-}
-async function executeToolCallsSequential(currentContext, assistantMessage, toolCalls, config, signal, emit) {
-  const finalizedCalls = [];
-  const messages = [];
-  for (const toolCall of toolCalls) {
-    await emit({
-      type: "tool_execution_start",
-      toolCallId: toolCall.id,
-      toolName: toolCall.name,
-      args: toolCall.arguments
-    });
-    const preparation = await prepareToolCall(currentContext, assistantMessage, toolCall, config, signal);
-    let finalized;
-    if (preparation.kind === "immediate") {
-      finalized = {
-        toolCall,
-        result: preparation.result,
-        isError: preparation.isError
-      };
-    } else {
-      const executed = await executePreparedToolCall(preparation, signal, emit);
-      finalized = await finalizeExecutedToolCall(currentContext, assistantMessage, preparation, executed, config, signal);
-    }
-    await emitToolExecutionEnd(finalized, emit);
-    const toolResultMessage = createToolResultMessage(finalized);
-    await emitToolResultMessage(toolResultMessage, emit);
-    finalizedCalls.push(finalized);
-    messages.push(toolResultMessage);
-  }
-  return {
-    messages,
-    terminate: shouldTerminateToolBatch(finalizedCalls)
-  };
-}
-async function executeToolCallsParallel(currentContext, assistantMessage, toolCalls, config, signal, emit) {
-  const finalizedCalls = [];
-  for (const toolCall of toolCalls) {
-    await emit({
-      type: "tool_execution_start",
-      toolCallId: toolCall.id,
-      toolName: toolCall.name,
-      args: toolCall.arguments
-    });
-    const preparation = await prepareToolCall(currentContext, assistantMessage, toolCall, config, signal);
-    if (preparation.kind === "immediate") {
-      const finalized = {
-        toolCall,
-        result: preparation.result,
-        isError: preparation.isError
-      };
-      await emitToolExecutionEnd(finalized, emit);
-      finalizedCalls.push(finalized);
-      continue;
-    }
-    finalizedCalls.push(async () => {
-      const executed = await executePreparedToolCall(preparation, signal, emit);
-      const finalized = await finalizeExecutedToolCall(currentContext, assistantMessage, preparation, executed, config, signal);
-      await emitToolExecutionEnd(finalized, emit);
-      return finalized;
-    });
-  }
-  const orderedFinalizedCalls = await Promise.all(finalizedCalls.map((entry) => typeof entry === "function" ? entry() : Promise.resolve(entry)));
-  const messages = [];
-  for (const finalized of orderedFinalizedCalls) {
-    const toolResultMessage = createToolResultMessage(finalized);
-    await emitToolResultMessage(toolResultMessage, emit);
-    messages.push(toolResultMessage);
-  }
-  return {
-    messages,
-    terminate: shouldTerminateToolBatch(orderedFinalizedCalls)
-  };
-}
-function shouldTerminateToolBatch(finalizedCalls) {
-  return finalizedCalls.length > 0 && finalizedCalls.every((finalized) => finalized.result.terminate === true);
-}
-function prepareToolCallArguments(tool, toolCall) {
-  if (!tool.prepareArguments) {
-    return toolCall;
-  }
-  const preparedArguments = tool.prepareArguments(toolCall.arguments);
-  if (preparedArguments === toolCall.arguments) {
-    return toolCall;
-  }
-  return {
-    ...toolCall,
-    arguments: preparedArguments
-  };
-}
-async function prepareToolCall(currentContext, assistantMessage, toolCall, config, signal) {
-  const tool = currentContext.tools?.find((t) => t.name === toolCall.name);
-  if (!tool) {
-    return {
-      kind: "immediate",
-      result: createErrorToolResult(`Tool ${toolCall.name} not found`),
-      isError: true
-    };
-  }
-  try {
-    const preparedToolCall = prepareToolCallArguments(tool, toolCall);
-    const validatedArgs = validateToolArguments(tool, preparedToolCall);
-    if (config.beforeToolCall) {
-      const beforeResult = await config.beforeToolCall({
-        assistantMessage,
-        toolCall,
-        args: validatedArgs,
-        context: currentContext
-      }, signal);
-      if (beforeResult?.block) {
-        return {
-          kind: "immediate",
-          result: createErrorToolResult(beforeResult.reason || "Tool execution was blocked"),
-          isError: true
-        };
-      }
-    }
-    return {
-      kind: "prepared",
-      toolCall,
-      tool,
-      args: validatedArgs
-    };
-  } catch (error) {
-    return {
-      kind: "immediate",
-      result: createErrorToolResult(error instanceof Error ? error.message : String(error)),
-      isError: true
-    };
-  }
-}
-async function executePreparedToolCall(prepared, signal, emit) {
-  const updateEvents = [];
-  try {
-    const result = await prepared.tool.execute(prepared.toolCall.id, prepared.args, signal, (partialResult) => {
-      updateEvents.push(Promise.resolve(emit({
-        type: "tool_execution_update",
-        toolCallId: prepared.toolCall.id,
-        toolName: prepared.toolCall.name,
-        args: prepared.toolCall.arguments,
-        partialResult
-      })));
-    });
-    await Promise.all(updateEvents);
-    return { result, isError: false };
-  } catch (error) {
-    await Promise.all(updateEvents);
-    return {
-      result: createErrorToolResult(error instanceof Error ? error.message : String(error)),
-      isError: true
-    };
-  }
-}
-async function finalizeExecutedToolCall(currentContext, assistantMessage, prepared, executed, config, signal) {
-  let result = executed.result;
-  let isError = executed.isError;
-  if (config.afterToolCall) {
-    try {
-      const afterResult = await config.afterToolCall({
-        assistantMessage,
-        toolCall: prepared.toolCall,
-        args: prepared.args,
-        result,
-        isError,
-        context: currentContext
-      }, signal);
-      if (afterResult) {
-        result = {
-          content: afterResult.content ?? result.content,
-          details: afterResult.details ?? result.details,
-          terminate: afterResult.terminate ?? result.terminate
-        };
-        isError = afterResult.isError ?? isError;
-      }
-    } catch (error) {
-      result = createErrorToolResult(error instanceof Error ? error.message : String(error));
-      isError = true;
-    }
-  }
-  return {
-    toolCall: prepared.toolCall,
-    result,
-    isError
-  };
-}
-function createErrorToolResult(message) {
-  return {
-    content: [{ type: "text", text: message }],
-    details: {}
-  };
-}
-async function emitToolExecutionEnd(finalized, emit) {
-  await emit({
-    type: "tool_execution_end",
-    toolCallId: finalized.toolCall.id,
-    toolName: finalized.toolCall.name,
-    result: finalized.result,
-    isError: finalized.isError
-  });
-}
-function createToolResultMessage(finalized) {
-  return {
-    role: "toolResult",
-    toolCallId: finalized.toolCall.id,
-    toolName: finalized.toolCall.name,
-    content: finalized.result.content,
-    details: finalized.result.details,
-    isError: finalized.isError,
-    timestamp: Date.now()
-  };
-}
-async function emitToolResultMessage(toolResultMessage, emit) {
-  await emit({ type: "message_start", message: toolResultMessage });
-  await emit({ type: "message_end", message: toolResultMessage });
-}
-function defaultConvertToLlm(messages) {
-  return messages.filter((message) => message.role === "user" || message.role === "assistant" || message.role === "toolResult");
-}
-const EMPTY_USAGE = {
-  input: 0,
-  output: 0,
-  cacheRead: 0,
-  cacheWrite: 0,
-  totalTokens: 0,
-  cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
-};
-const DEFAULT_MODEL = {
-  id: "unknown",
-  name: "unknown",
-  api: "unknown",
-  provider: "unknown",
-  baseUrl: "",
-  reasoning: false,
-  input: [],
-  cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-  contextWindow: 0,
-  maxTokens: 0
-};
-function createMutableAgentState(initialState) {
-  let tools = initialState?.tools?.slice() ?? [];
-  let messages = initialState?.messages?.slice() ?? [];
-  return {
-    systemPrompt: initialState?.systemPrompt ?? "",
-    model: initialState?.model ?? DEFAULT_MODEL,
-    thinkingLevel: initialState?.thinkingLevel ?? "off",
-    get tools() {
-      return tools;
-    },
-    set tools(nextTools) {
-      tools = nextTools.slice();
-    },
-    get messages() {
-      return messages;
-    },
-    set messages(nextMessages) {
-      messages = nextMessages.slice();
-    },
-    isStreaming: false,
-    streamingMessage: void 0,
-    pendingToolCalls: /* @__PURE__ */ new Set(),
-    errorMessage: void 0
-  };
-}
-class PendingMessageQueue {
-  mode;
-  messages = [];
-  constructor(mode) {
-    this.mode = mode;
-  }
-  enqueue(message) {
-    this.messages.push(message);
-  }
-  hasItems() {
-    return this.messages.length > 0;
-  }
-  drain() {
-    if (this.mode === "all") {
-      const drained = this.messages.slice();
-      this.messages = [];
-      return drained;
-    }
-    const first = this.messages[0];
-    if (!first) {
-      return [];
-    }
-    this.messages = this.messages.slice(1);
-    return [first];
-  }
-  clear() {
-    this.messages = [];
-  }
-}
-class Agent {
-  _state;
-  listeners = /* @__PURE__ */ new Set();
-  steeringQueue;
-  followUpQueue;
-  convertToLlm;
-  transformContext;
-  streamFn;
-  getApiKey;
-  onPayload;
-  onResponse;
-  beforeToolCall;
-  afterToolCall;
-  activeRun;
-  /** Session identifier forwarded to providers for cache-aware backends. */
-  sessionId;
-  /** Optional per-level thinking token budgets forwarded to the stream function. */
-  thinkingBudgets;
-  /** Preferred transport forwarded to the stream function. */
-  transport;
-  /** Optional cap for provider-requested retry delays. */
-  maxRetryDelayMs;
-  /** Tool execution strategy for assistant messages that contain multiple tool calls. */
-  toolExecution;
-  constructor(options = {}) {
-    this._state = createMutableAgentState(options.initialState);
-    this.convertToLlm = options.convertToLlm ?? defaultConvertToLlm;
-    this.transformContext = options.transformContext;
-    this.streamFn = options.streamFn ?? streamSimple;
-    this.getApiKey = options.getApiKey;
-    this.onPayload = options.onPayload;
-    this.onResponse = options.onResponse;
-    this.beforeToolCall = options.beforeToolCall;
-    this.afterToolCall = options.afterToolCall;
-    this.steeringQueue = new PendingMessageQueue(options.steeringMode ?? "one-at-a-time");
-    this.followUpQueue = new PendingMessageQueue(options.followUpMode ?? "one-at-a-time");
-    this.sessionId = options.sessionId;
-    this.thinkingBudgets = options.thinkingBudgets;
-    this.transport = options.transport ?? "auto";
-    this.maxRetryDelayMs = options.maxRetryDelayMs;
-    this.toolExecution = options.toolExecution ?? "parallel";
-  }
-  /**
-   * Subscribe to agent lifecycle events.
-   *
-   * Listener promises are awaited in subscription order and are included in
-   * the current run's settlement. Listeners also receive the active abort
-   * signal for the current run.
-   *
-   * `agent_end` is the final emitted event for a run, but the agent does not
-   * become idle until all awaited listeners for that event have settled.
-   */
-  subscribe(listener) {
-    this.listeners.add(listener);
-    return () => this.listeners.delete(listener);
-  }
-  /**
-   * Current agent state.
-   *
-   * Assigning `state.tools` or `state.messages` copies the provided top-level array.
-   */
-  get state() {
-    return this._state;
-  }
-  /** Controls how queued steering messages are drained. */
-  set steeringMode(mode) {
-    this.steeringQueue.mode = mode;
-  }
-  get steeringMode() {
-    return this.steeringQueue.mode;
-  }
-  /** Controls how queued follow-up messages are drained. */
-  set followUpMode(mode) {
-    this.followUpQueue.mode = mode;
-  }
-  get followUpMode() {
-    return this.followUpQueue.mode;
-  }
-  /** Queue a message to be injected after the current assistant turn finishes. */
-  steer(message) {
-    this.steeringQueue.enqueue(message);
-  }
-  /** Queue a message to run only after the agent would otherwise stop. */
-  followUp(message) {
-    this.followUpQueue.enqueue(message);
-  }
-  /** Remove all queued steering messages. */
-  clearSteeringQueue() {
-    this.steeringQueue.clear();
-  }
-  /** Remove all queued follow-up messages. */
-  clearFollowUpQueue() {
-    this.followUpQueue.clear();
-  }
-  /** Remove all queued steering and follow-up messages. */
-  clearAllQueues() {
-    this.clearSteeringQueue();
-    this.clearFollowUpQueue();
-  }
-  /** Returns true when either queue still contains pending messages. */
-  hasQueuedMessages() {
-    return this.steeringQueue.hasItems() || this.followUpQueue.hasItems();
-  }
-  /** Active abort signal for the current run, if any. */
-  get signal() {
-    return this.activeRun?.abortController.signal;
-  }
-  /** Abort the current run, if one is active. */
-  abort() {
-    this.activeRun?.abortController.abort();
-  }
-  /**
-   * Resolve when the current run and all awaited event listeners have finished.
-   *
-   * This resolves after `agent_end` listeners settle.
-   */
-  waitForIdle() {
-    return this.activeRun?.promise ?? Promise.resolve();
-  }
-  /** Clear transcript state, runtime state, and queued messages. */
-  reset() {
-    this._state.messages = [];
-    this._state.isStreaming = false;
-    this._state.streamingMessage = void 0;
-    this._state.pendingToolCalls = /* @__PURE__ */ new Set();
-    this._state.errorMessage = void 0;
-    this.clearFollowUpQueue();
-    this.clearSteeringQueue();
-  }
-  async prompt(input, images) {
-    if (this.activeRun) {
-      throw new Error("Agent is already processing a prompt. Use steer() or followUp() to queue messages, or wait for completion.");
-    }
-    const messages = this.normalizePromptInput(input, images);
-    await this.runPromptMessages(messages);
-  }
-  /** Continue from the current transcript. The last message must be a user or tool-result message. */
-  async continue() {
-    if (this.activeRun) {
-      throw new Error("Agent is already processing. Wait for completion before continuing.");
-    }
-    const lastMessage = this._state.messages[this._state.messages.length - 1];
-    if (!lastMessage) {
-      throw new Error("No messages to continue from");
-    }
-    if (lastMessage.role === "assistant") {
-      const queuedSteering = this.steeringQueue.drain();
-      if (queuedSteering.length > 0) {
-        await this.runPromptMessages(queuedSteering, { skipInitialSteeringPoll: true });
-        return;
-      }
-      const queuedFollowUps = this.followUpQueue.drain();
-      if (queuedFollowUps.length > 0) {
-        await this.runPromptMessages(queuedFollowUps);
-        return;
-      }
-      throw new Error("Cannot continue from message role: assistant");
-    }
-    await this.runContinuation();
-  }
-  normalizePromptInput(input, images) {
-    if (Array.isArray(input)) {
-      return input;
-    }
-    if (typeof input !== "string") {
-      return [input];
-    }
-    const content = [{ type: "text", text: input }];
-    if (images && images.length > 0) {
-      content.push(...images);
-    }
-    return [{ role: "user", content, timestamp: Date.now() }];
-  }
-  async runPromptMessages(messages, options = {}) {
-    await this.runWithLifecycle(async (signal) => {
-      await runAgentLoop(messages, this.createContextSnapshot(), this.createLoopConfig(options), (event) => this.processEvents(event), signal, this.streamFn);
-    });
-  }
-  async runContinuation() {
-    await this.runWithLifecycle(async (signal) => {
-      await runAgentLoopContinue(this.createContextSnapshot(), this.createLoopConfig(), (event) => this.processEvents(event), signal, this.streamFn);
-    });
-  }
-  createContextSnapshot() {
-    return {
-      systemPrompt: this._state.systemPrompt,
-      messages: this._state.messages.slice(),
-      tools: this._state.tools.slice()
-    };
-  }
-  createLoopConfig(options = {}) {
-    let skipInitialSteeringPoll = options.skipInitialSteeringPoll === true;
-    return {
-      model: this._state.model,
-      reasoning: this._state.thinkingLevel === "off" ? void 0 : this._state.thinkingLevel,
-      sessionId: this.sessionId,
-      onPayload: this.onPayload,
-      onResponse: this.onResponse,
-      transport: this.transport,
-      thinkingBudgets: this.thinkingBudgets,
-      maxRetryDelayMs: this.maxRetryDelayMs,
-      toolExecution: this.toolExecution,
-      beforeToolCall: this.beforeToolCall,
-      afterToolCall: this.afterToolCall,
-      convertToLlm: this.convertToLlm,
-      transformContext: this.transformContext,
-      getApiKey: this.getApiKey,
-      getSteeringMessages: async () => {
-        if (skipInitialSteeringPoll) {
-          skipInitialSteeringPoll = false;
-          return [];
-        }
-        return this.steeringQueue.drain();
-      },
-      getFollowUpMessages: async () => this.followUpQueue.drain()
-    };
-  }
-  async runWithLifecycle(executor) {
-    if (this.activeRun) {
-      throw new Error("Agent is already processing.");
-    }
-    const abortController = new AbortController();
-    let resolvePromise = () => {
-    };
-    const promise = new Promise((resolve) => {
-      resolvePromise = resolve;
-    });
-    this.activeRun = { promise, resolve: resolvePromise, abortController };
-    this._state.isStreaming = true;
-    this._state.streamingMessage = void 0;
-    this._state.errorMessage = void 0;
-    try {
-      await executor(abortController.signal);
-    } catch (error) {
-      await this.handleRunFailure(error, abortController.signal.aborted);
-    } finally {
-      this.finishRun();
-    }
-  }
-  async handleRunFailure(error, aborted) {
-    const failureMessage = {
-      role: "assistant",
-      content: [{ type: "text", text: "" }],
-      api: this._state.model.api,
-      provider: this._state.model.provider,
-      model: this._state.model.id,
-      usage: EMPTY_USAGE,
-      stopReason: aborted ? "aborted" : "error",
-      errorMessage: error instanceof Error ? error.message : String(error),
-      timestamp: Date.now()
-    };
-    this._state.messages.push(failureMessage);
-    this._state.errorMessage = failureMessage.errorMessage;
-    await this.processEvents({ type: "agent_end", messages: [failureMessage] });
-  }
-  finishRun() {
-    this._state.isStreaming = false;
-    this._state.streamingMessage = void 0;
-    this._state.pendingToolCalls = /* @__PURE__ */ new Set();
-    this.activeRun?.resolve();
-    this.activeRun = void 0;
-  }
-  /**
-   * Reduce internal state for a loop event, then await listeners.
-   *
-   * `agent_end` only means no further loop events will be emitted. The run is
-   * considered idle later, after all awaited listeners for `agent_end` finish
-   * and `finishRun()` clears runtime-owned state.
-   */
-  async processEvents(event) {
-    switch (event.type) {
-      case "message_start":
-        this._state.streamingMessage = event.message;
-        break;
-      case "message_update":
-        this._state.streamingMessage = event.message;
-        break;
-      case "message_end":
-        this._state.streamingMessage = void 0;
-        this._state.messages.push(event.message);
-        break;
-      case "tool_execution_start": {
-        const pendingToolCalls = new Set(this._state.pendingToolCalls);
-        pendingToolCalls.add(event.toolCallId);
-        this._state.pendingToolCalls = pendingToolCalls;
-        break;
-      }
-      case "tool_execution_end": {
-        const pendingToolCalls = new Set(this._state.pendingToolCalls);
-        pendingToolCalls.delete(event.toolCallId);
-        this._state.pendingToolCalls = pendingToolCalls;
-        break;
-      }
-      case "turn_end":
-        if (event.message.role === "assistant" && event.message.errorMessage) {
-          this._state.errorMessage = event.message.errorMessage;
-        }
-        break;
-      case "agent_end":
-        this._state.streamingMessage = void 0;
-        break;
-    }
-    const signal = this.activeRun?.abortController.signal;
-    if (!signal) {
-      throw new Error("Agent listener invoked outside active run");
-    }
-    for (const listener of this.listeners) {
-      await listener(event, signal);
-    }
-  }
-}
-export {
-  Agent,
-  runAgentLoop,
-  runAgentLoopContinue
-};
+import"./typebox.mjs";import{s as _,v as b}from"./mariozechner__pi-ai.mjs";import"./partial-json.mjs";import"./anthropic-ai__sdk.mjs";import"./openai.mjs";import"./google__genai.mjs";import"./p-retry.mjs";import"./react.mjs";import"./retry.mjs";import"./google-auth-library.mjs";import"child_process";import"path";import"querystring";import"fs";import"./gaxios.mjs";import"https";import"./extend.mjs";import"stream";import"./gcp-metadata.mjs";import"os";import"./json-bigint.mjs";import"./bignumber.js.mjs";import"./google-logging-utils.mjs";import"process";import"events";import"util";import"./base64-js.mjs";import"crypto";import"./ecdsa-sig-formatter.mjs";import"./safe-buffer.mjs";import"buffer";import"./jws.mjs";import"./jwa.mjs";import"./buffer-equal-constant-time.mjs";import"fs/promises";import"node:stream";import"node:stream/promises";import"./ws.mjs";import"http";import"net";import"tls";import"url";import"zlib";import"./mistralai__mistralai.mjs";import"./zod.mjs";import"./zod-to-json-schema.mjs";import"./opentelemetry__api.mjs";import"./@opentelemetry/semantic-conventions+[...].mjs";async function x(t,e,s,a,o,r){const n=[...t],l={...e,messages:[...e.messages,...t]};await a({type:"agent_start"}),await a({type:"turn_start"});for(const i of t)await a({type:"message_start",message:i}),await a({type:"message_end",message:i});return await C(l,n,s,o,a,r),n}async function P(t,e,s,a,o){if(t.messages.length===0)throw new Error("Cannot continue: no messages in context");if(t.messages[t.messages.length-1].role==="assistant")throw new Error("Cannot continue from message role: assistant");const r=[],n={...t};return await s({type:"agent_start"}),await s({type:"turn_start"}),await C(n,r,e,a,s,o),r}async function C(t,e,s,a,o,r){let n=!0,l=await s.getSteeringMessages?.()||[];for(;;){let i=!0;for(;i||l.length>0;){if(n?n=!1:await o({type:"turn_start"}),l.length>0){for(const c of l)await o({type:"message_start",message:c}),await o({type:"message_end",message:c}),t.messages.push(c),e.push(c);l=[]}const u=await A(t,s,a,o,r);if(e.push(u),u.stopReason==="error"||u.stopReason==="aborted"){await o({type:"turn_end",message:u,toolResults:[]}),await o({type:"agent_end",messages:e});return}const g=u.content.filter(c=>c.type==="toolCall"),p=[];if(i=!1,g.length>0){const c=await S(t,u,s,a,o);p.push(...c.messages),i=!c.terminate;for(const h of p)t.messages.push(h),e.push(h)}if(await o({type:"turn_end",message:u,toolResults:p}),await s.shouldStopAfterTurn?.({message:u,toolResults:p,context:t,newMessages:e})){await o({type:"agent_end",messages:e});return}l=await s.getSteeringMessages?.()||[]}const m=await s.getFollowUpMessages?.()||[];if(m.length>0){l=m;continue}break}await o({type:"agent_end",messages:e})}async function A(t,e,s,a,o){let r=t.messages;e.transformContext&&(r=await e.transformContext(r,s));const n=await e.convertToLlm(r),l={systemPrompt:t.systemPrompt,messages:n,tools:t.tools},i=o||_,m=(e.getApiKey?await e.getApiKey(e.model.provider):void 0)||e.apiKey,u=await i(e.model,l,{...e,apiKey:m,signal:s});let g=null,p=!1;for await(const h of u)switch(h.type){case"start":g=h.partial,t.messages.push(g),p=!0,await a({type:"message_start",message:{...g}});break;case"text_start":case"text_delta":case"text_end":case"thinking_start":case"thinking_delta":case"thinking_end":case"toolcall_start":case"toolcall_delta":case"toolcall_end":g&&(g=h.partial,t.messages[t.messages.length-1]=g,await a({type:"message_update",assistantMessageEvent:h,message:{...g}}));break;case"done":case"error":{const d=await u.result();return p?t.messages[t.messages.length-1]=d:t.messages.push(d),p||await a({type:"message_start",message:{...d}}),await a({type:"message_end",message:d}),d}}const c=await u.result();return p?t.messages[t.messages.length-1]=c:(t.messages.push(c),await a({type:"message_start",message:{...c}})),await a({type:"message_end",message:c}),c}async function S(t,e,s,a,o){const r=e.content.filter(l=>l.type==="toolCall"),n=r.some(l=>t.tools?.find(i=>i.name===l.name)?.executionMode==="sequential");return s.toolExecution==="sequential"||n?Q(t,e,r,s,a,o):U(t,e,r,s,a,o)}async function Q(t,e,s,a,o,r){const n=[],l=[];for(const i of s){await r({type:"tool_execution_start",toolCallId:i.id,toolName:i.name,args:i.arguments});const m=await E(t,e,i,a,o);let u;if(m.kind==="immediate")u={toolCall:i,result:m.result,isError:m.isError};else{const p=await M(m,o,r);u=await R(t,e,m,p,a,o)}await w(u,r);const g=k(u);await v(g,r),n.push(u),l.push(g)}return{messages:l,terminate:T(n)}}async function U(t,e,s,a,o,r){const n=[];for(const m of s){await r({type:"tool_execution_start",toolCallId:m.id,toolName:m.name,args:m.arguments});const u=await E(t,e,m,a,o);if(u.kind==="immediate"){const g={toolCall:m,result:u.result,isError:u.isError};await w(g,r),n.push(g);continue}n.push(async()=>{const g=await M(u,o,r),p=await R(t,e,u,g,a,o);return await w(p,r),p})}const l=await Promise.all(n.map(m=>typeof m=="function"?m():Promise.resolve(m))),i=[];for(const m of l){const u=k(m);await v(u,r),i.push(u)}return{messages:i,terminate:T(l)}}function T(t){return t.length>0&&t.every(e=>e.result.terminate===!0)}function L(t,e){if(!t.prepareArguments)return e;const s=t.prepareArguments(e.arguments);return s===e.arguments?e:{...e,arguments:s}}async function E(t,e,s,a,o){const r=t.tools?.find(n=>n.name===s.name);if(!r)return{kind:"immediate",result:f(`Tool ${s.name} not found`),isError:!0};try{const n=L(r,s),l=b(r,n);if(a.beforeToolCall){const i=await a.beforeToolCall({assistantMessage:e,toolCall:s,args:l,context:t},o);if(i?.block)return{kind:"immediate",result:f(i.reason||"Tool execution was blocked"),isError:!0}}return{kind:"prepared",toolCall:s,tool:r,args:l}}catch(n){return{kind:"immediate",result:f(n instanceof Error?n.message:String(n)),isError:!0}}}async function M(t,e,s){const a=[];try{const o=await t.tool.execute(t.toolCall.id,t.args,e,r=>{a.push(Promise.resolve(s({type:"tool_execution_update",toolCallId:t.toolCall.id,toolName:t.toolCall.name,args:t.toolCall.arguments,partialResult:r})))});return await Promise.all(a),{result:o,isError:!1}}catch(o){return await Promise.all(a),{result:f(o instanceof Error?o.message:String(o)),isError:!0}}}async function R(t,e,s,a,o,r){let n=a.result,l=a.isError;if(o.afterToolCall)try{const i=await o.afterToolCall({assistantMessage:e,toolCall:s.toolCall,args:s.args,result:n,isError:l,context:t},r);i&&(n={content:i.content??n.content,details:i.details??n.details,terminate:i.terminate??n.terminate},l=i.isError??l)}catch(i){n=f(i instanceof Error?i.message:String(i)),l=!0}return{toolCall:s.toolCall,result:n,isError:l}}function f(t){return{content:[{type:"text",text:t}],details:{}}}async function w(t,e){await e({type:"tool_execution_end",toolCallId:t.toolCall.id,toolName:t.toolCall.name,result:t.result,isError:t.isError})}function k(t){return{role:"toolResult",toolCallId:t.toolCall.id,toolName:t.toolCall.name,content:t.result.content,details:t.result.details,isError:t.isError,timestamp:Date.now()}}async function v(t,e){await e({type:"message_start",message:t}),await e({type:"message_end",message:t})}function I(t){return t.filter(e=>e.role==="user"||e.role==="assistant"||e.role==="toolResult")}const F={input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},q={id:"unknown",name:"unknown",api:"unknown",provider:"unknown",baseUrl:"",reasoning:!1,input:[],cost:{input:0,output:0,cacheRead:0,cacheWrite:0},contextWindow:0,maxTokens:0};function D(t){let e=t?.tools?.slice()??[],s=t?.messages?.slice()??[];return{systemPrompt:t?.systemPrompt??"",model:t?.model??q,thinkingLevel:t?.thinkingLevel??"off",get tools(){return e},set tools(a){e=a.slice()},get messages(){return s},set messages(a){s=a.slice()},isStreaming:!1,streamingMessage:void 0,pendingToolCalls:new Set,errorMessage:void 0}}class y{mode;messages=[];constructor(e){this.mode=e}enqueue(e){this.messages.push(e)}hasItems(){return this.messages.length>0}drain(){if(this.mode==="all"){const s=this.messages.slice();return this.messages=[],s}const e=this.messages[0];return e?(this.messages=this.messages.slice(1),[e]):[]}clear(){this.messages=[]}}class Ue{_state;listeners=new Set;steeringQueue;followUpQueue;convertToLlm;transformContext;streamFn;getApiKey;onPayload;onResponse;beforeToolCall;afterToolCall;activeRun;sessionId;thinkingBudgets;transport;maxRetryDelayMs;toolExecution;constructor(e={}){this._state=D(e.initialState),this.convertToLlm=e.convertToLlm??I,this.transformContext=e.transformContext,this.streamFn=e.streamFn??_,this.getApiKey=e.getApiKey,this.onPayload=e.onPayload,this.onResponse=e.onResponse,this.beforeToolCall=e.beforeToolCall,this.afterToolCall=e.afterToolCall,this.steeringQueue=new y(e.steeringMode??"one-at-a-time"),this.followUpQueue=new y(e.followUpMode??"one-at-a-time"),this.sessionId=e.sessionId,this.thinkingBudgets=e.thinkingBudgets,this.transport=e.transport??"auto",this.maxRetryDelayMs=e.maxRetryDelayMs,this.toolExecution=e.toolExecution??"parallel"}subscribe(e){return this.listeners.add(e),()=>this.listeners.delete(e)}get state(){return this._state}set steeringMode(e){this.steeringQueue.mode=e}get steeringMode(){return this.steeringQueue.mode}set followUpMode(e){this.followUpQueue.mode=e}get followUpMode(){return this.followUpQueue.mode}steer(e){this.steeringQueue.enqueue(e)}followUp(e){this.followUpQueue.enqueue(e)}clearSteeringQueue(){this.steeringQueue.clear()}clearFollowUpQueue(){this.followUpQueue.clear()}clearAllQueues(){this.clearSteeringQueue(),this.clearFollowUpQueue()}hasQueuedMessages(){return this.steeringQueue.hasItems()||this.followUpQueue.hasItems()}get signal(){return this.activeRun?.abortController.signal}abort(){this.activeRun?.abortController.abort()}waitForIdle(){return this.activeRun?.promise??Promise.resolve()}reset(){this._state.messages=[],this._state.isStreaming=!1,this._state.streamingMessage=void 0,this._state.pendingToolCalls=new Set,this._state.errorMessage=void 0,this.clearFollowUpQueue(),this.clearSteeringQueue()}async prompt(e,s){if(this.activeRun)throw new Error("Agent is already processing a prompt. Use steer() or followUp() to queue messages, or wait for completion.");const a=this.normalizePromptInput(e,s);await this.runPromptMessages(a)}async continue(){if(this.activeRun)throw new Error("Agent is already processing. Wait for completion before continuing.");const e=this._state.messages[this._state.messages.length-1];if(!e)throw new Error("No messages to continue from");if(e.role==="assistant"){const s=this.steeringQueue.drain();if(s.length>0){await this.runPromptMessages(s,{skipInitialSteeringPoll:!0});return}const a=this.followUpQueue.drain();if(a.length>0){await this.runPromptMessages(a);return}throw new Error("Cannot continue from message role: assistant")}await this.runContinuation()}normalizePromptInput(e,s){if(Array.isArray(e))return e;if(typeof e!="string")return[e];const a=[{type:"text",text:e}];return s&&s.length>0&&a.push(...s),[{role:"user",content:a,timestamp:Date.now()}]}async runPromptMessages(e,s={}){await this.runWithLifecycle(async a=>{await x(e,this.createContextSnapshot(),this.createLoopConfig(s),o=>this.processEvents(o),a,this.streamFn)})}async runContinuation(){await this.runWithLifecycle(async e=>{await P(this.createContextSnapshot(),this.createLoopConfig(),s=>this.processEvents(s),e,this.streamFn)})}createContextSnapshot(){return{systemPrompt:this._state.systemPrompt,messages:this._state.messages.slice(),tools:this._state.tools.slice()}}createLoopConfig(e={}){let s=e.skipInitialSteeringPoll===!0;return{model:this._state.model,reasoning:this._state.thinkingLevel==="off"?void 0:this._state.thinkingLevel,sessionId:this.sessionId,onPayload:this.onPayload,onResponse:this.onResponse,transport:this.transport,thinkingBudgets:this.thinkingBudgets,maxRetryDelayMs:this.maxRetryDelayMs,toolExecution:this.toolExecution,beforeToolCall:this.beforeToolCall,afterToolCall:this.afterToolCall,convertToLlm:this.convertToLlm,transformContext:this.transformContext,getApiKey:this.getApiKey,getSteeringMessages:async()=>s?(s=!1,[]):this.steeringQueue.drain(),getFollowUpMessages:async()=>this.followUpQueue.drain()}}async runWithLifecycle(e){if(this.activeRun)throw new Error("Agent is already processing.");const s=new AbortController;let a=()=>{};const o=new Promise(r=>{a=r});this.activeRun={promise:o,resolve:a,abortController:s},this._state.isStreaming=!0,this._state.streamingMessage=void 0,this._state.errorMessage=void 0;try{await e(s.signal)}catch(r){await this.handleRunFailure(r,s.signal.aborted)}finally{this.finishRun()}}async handleRunFailure(e,s){const a={role:"assistant",content:[{type:"text",text:""}],api:this._state.model.api,provider:this._state.model.provider,model:this._state.model.id,usage:F,stopReason:s?"aborted":"error",errorMessage:e instanceof Error?e.message:String(e),timestamp:Date.now()};this._state.messages.push(a),this._state.errorMessage=a.errorMessage,await this.processEvents({type:"agent_end",messages:[a]})}finishRun(){this._state.isStreaming=!1,this._state.streamingMessage=void 0,this._state.pendingToolCalls=new Set,this.activeRun?.resolve(),this.activeRun=void 0}async processEvents(e){switch(e.type){case"message_start":this._state.streamingMessage=e.message;break;case"message_update":this._state.streamingMessage=e.message;break;case"message_end":this._state.streamingMessage=void 0,this._state.messages.push(e.message);break;case"tool_execution_start":{const a=new Set(this._state.pendingToolCalls);a.add(e.toolCallId),this._state.pendingToolCalls=a;break}case"tool_execution_end":{const a=new Set(this._state.pendingToolCalls);a.delete(e.toolCallId),this._state.pendingToolCalls=a;break}case"turn_end":e.message.role==="assistant"&&e.message.errorMessage&&(this._state.errorMessage=e.message.errorMessage);break;case"agent_end":this._state.streamingMessage=void 0;break}const s=this.activeRun?.abortController.signal;if(!s)throw new Error("Agent listener invoked outside active run");for(const a of this.listeners)await a(e,s)}}export{Ue as Agent,x as runAgentLoop,P as runAgentLoopContinue};