@truefoundry/trueforge-core 0.0.0 → 0.1.0-rc.0

package/dist/core/runtime/AgentThreadOrchestrator.mjs

@@ -0,0 +1,418 @@
+// src/core/runtime/AgentThreadOrchestrator.ts
+import { AgentHarnessError, InvalidAgentSendInputError } from "../errors.mjs";
+import {
+  EventType,
+  newEventId
+} from "../events/schema.mjs";
+import { onSignalAbort } from "../util/abort.mjs";
+import { mergeAsyncGenerators } from "../util/promiseUtils.mjs";
+import "./AgentThread.mjs";
+import {
+  InternalEventType
+} from "./AgentThread.types.mjs";
+import {
+  assistantMessageContentToStringForSubAgent,
+  getThreadId,
+  isApprovalDecisionMessage,
+  isClientSideToolResponseMessage,
+  isInternalThreadDoneError
+} from "./contextUtils.mjs";
+import { addAgentThreadMetrics, createEmptyAgentThreadMetrics } from "./metrics.mjs";
+var MAX_PARALLEL_SUB_AGENTS = 5;
+function agentThreadEventToTerminalFields(event) {
+  switch (event.type) {
+    case InternalEventType.AGENT_DONE: {
+      if ("parent" in event && event.parent) return {};
+      if (event.status === "error") return {};
+      return { output: event.output };
+    }
+    case EventType.TOOL_APPROVAL_REQUIRED:
+    case EventType.TOOL_RESPONSE_REQUIRED:
+      return { requiredAction: event };
+    default:
+      return {};
+  }
+}
+function isUserToolApprovalOrResponseBatch(messages) {
+  const first = messages[0];
+  return first !== void 0 && (isApprovalDecisionMessage(first) || isClientSideToolResponseMessage(first));
+}
+function getMainThreadId(agentThreads) {
+  for (const thread of agentThreads.values()) {
+    if (!thread.parent) return thread.threadId;
+  }
+  throw new Error("Unreachable: no root thread found");
+}
+function getActiveAgentThreads(agentThreads) {
+  const nodes = new Set(agentThreads.keys());
+  const parentNodes = new Set(
+    [...agentThreads.values()].map((a) => a.parent?.thread_id).filter((a) => Boolean(a))
+  );
+  const leaves = [...nodes].filter((id) => !parentNodes.has(id));
+  if (leaves.length === 0) {
+    throw new Error("We cannot have zero leave agents");
+  }
+  return leaves.map((id) => {
+    const agentThread = agentThreads.get(id);
+    if (!agentThread) throw new Error("unreachable");
+    return agentThread;
+  });
+}
+function mergeAuthEvents(events) {
+  const serverMap = /* @__PURE__ */ new Map();
+  for (const event of events) {
+    for (const entry of event.mcp_servers) {
+      const existing = serverMap.get(entry.id);
+      if (existing) {
+        existing.thread_ids.push(...entry.thread_ids);
+      } else {
+        serverMap.set(entry.id, { ...entry, thread_ids: [...entry.thread_ids] });
+      }
+    }
+  }
+  return {
+    type: InternalEventType.MCP_AUTH_REQUIRED,
+    id: newEventId(),
+    created_at: (/* @__PURE__ */ new Date()).toISOString(),
+    thread_id: null,
+    mcp_servers: [...serverMap.values()]
+  };
+}
+function wrapGeneratorWithTrace(trace, generator) {
+  const wrapped = {
+    [Symbol.asyncIterator]() {
+      return this;
+    },
+    next(...[value]) {
+      return trace.runInContext(() => value === void 0 ? generator.next() : generator.next(value));
+    },
+    return(value) {
+      return trace.runInContext(() => generator.return(value));
+    },
+    throw(e) {
+      return trace.runInContext(() => generator.throw(e));
+    },
+    [Symbol.asyncDispose]() {
+      return generator[Symbol.asyncDispose]();
+    }
+  };
+  return wrapped;
+}
+function createRootAgentSpan(mainThread, tracing) {
+  const trace = tracing.startRootSpan(
+    JSON.stringify({ instructions: mainThread.definition.instruction, messages: mainThread.definition.messages })
+  );
+  let rootAgentErrorMessage;
+  let rootAgentCompleted = false;
+  return {
+    trace,
+    wrapThreadWithContext(thread, signal) {
+      return wrapGeneratorWithTrace(trace, thread.execute({ signal }));
+    },
+    setOutputFromEvent(chunk) {
+      rootAgentCompleted = true;
+      if (chunk.type === InternalEventType.AGENT_DONE) {
+        if (isInternalThreadDoneError(chunk)) {
+          rootAgentErrorMessage = chunk.error;
+          trace.setOutput(JSON.stringify({ error: chunk.error }));
+        } else {
+          const content = assistantMessageContentToStringForSubAgent(chunk.output.content);
+          trace.setOutput(JSON.stringify({ result: content }));
+        }
+      }
+    },
+    finalize(thread, error) {
+      trace.setMetrics(thread.getAgentThreadMetrics());
+      if (error) {
+        trace.setError(error);
+      } else if (rootAgentErrorMessage || !rootAgentCompleted) {
+        trace.setError(rootAgentErrorMessage ?? "Agent execution interrupted");
+      } else {
+        trace.setSuccess();
+      }
+    },
+    end() {
+      trace.end();
+    }
+  };
+}
+async function* wrapWithSubAgentSpan(parentTrace, currentThread, signal) {
+  const agentInfo = currentThread.agentInfo;
+  if (!agentInfo) {
+    throw new Error("Sub-agent thread is missing agentInfo");
+  }
+  const subTrace = parentTrace.startSubAgent(agentInfo.name, JSON.stringify({ instructions: agentInfo.input }));
+  const generator = wrapGeneratorWithTrace(subTrace, currentThread.execute({ signal }));
+  try {
+    for await (const event of generator) {
+      if (event.type === InternalEventType.AGENT_DONE) {
+        if (isInternalThreadDoneError(event)) {
+          subTrace.setOutput(JSON.stringify({ error: event.error }));
+          subTrace.setMetrics(currentThread.getAgentThreadMetrics());
+          subTrace.setError(event.error);
+        } else {
+          const content = assistantMessageContentToStringForSubAgent(event.output.content);
+          subTrace.setOutput(JSON.stringify({ result: content }));
+          subTrace.setMetrics(currentThread.getAgentThreadMetrics());
+          subTrace.setSuccess();
+        }
+      }
+      yield event;
+    }
+  } catch (error) {
+    subTrace.setError(error);
+    throw error;
+  } finally {
+    subTrace.end();
+  }
+}
+var AgentThreadOrchestrator = class {
+  agentThreads;
+  createDynamicSubAgentThread;
+  tracing;
+  logger;
+  // Finished sub-agents removed from `agentThreads`; kept so totals still include them.
+  finishedSubAgentMetrics = createEmptyAgentThreadMetrics();
+  constructor(params) {
+    this.agentThreads = params.agentThreads;
+    this.createDynamicSubAgentThread = params.createDynamicSubAgentThread;
+    this.tracing = params.tracing;
+    this.logger = params.logger.child({ module: "AgentThreadOrchestrator" });
+  }
+  /**
+   * Turn-wide metrics so far: live threads plus finished sub-agents.
+   * Safe mid-flight and after cancel/error (`execute()` may not return).
+   * Each thread is counted once — a sub-agent moves from live → finished atomically.
+   */
+  getMetrics() {
+    const total = createEmptyAgentThreadMetrics();
+    addAgentThreadMetrics(total, this.finishedSubAgentMetrics);
+    for (const thread of this.agentThreads.values()) {
+      addAgentThreadMetrics(total, thread.getAgentThreadMetrics());
+    }
+    return total;
+  }
+  async *send(messages) {
+    const byThread = /* @__PURE__ */ new Map();
+    for (const thread of this.agentThreads.values()) {
+      byThread.set(thread.threadId, []);
+    }
+    if (isUserToolApprovalOrResponseBatch(messages)) {
+      const grouped = /* @__PURE__ */ new Map();
+      for (const msg of messages) {
+        const threadId = msg.thread_id;
+        if (!byThread.has(threadId)) {
+          throw new InvalidAgentSendInputError(`unknown thread_id: ${threadId}`);
+        }
+        const batch = grouped.get(threadId) ?? [];
+        batch.push(msg);
+        grouped.set(threadId, batch);
+      }
+      for (const [threadId, batch] of grouped) {
+        byThread.set(threadId, batch);
+      }
+    } else if (messages.length > 0) {
+      if (this.agentThreads.size > 1) {
+        throw new InvalidAgentSendInputError(
+          "Cannot process user messages while sub agents are running, please send empty input for previous conversation to complete"
+        );
+      }
+      byThread.set(getMainThreadId(this.agentThreads), messages);
+    }
+    const validationErrors = [];
+    for (const [threadId, batch] of byThread) {
+      const thread = this.agentThreads.get(threadId);
+      if (!thread) {
+        throw new Error(`AgentThreadOrchestrator.send: unknown threadId ${threadId}`);
+      }
+      try {
+        thread.validateSendInput(batch);
+      } catch (e) {
+        if (e instanceof AgentHarnessError && e.code === "invalid_send_input") {
+          validationErrors.push(`thread ${threadId}: ${e.message}`);
+        } else {
+          throw e;
+        }
+      }
+    }
+    if (validationErrors.length > 0) {
+      throw new InvalidAgentSendInputError(validationErrors.join("; "));
+    }
+    for (const [threadId, batch] of byThread) {
+      yield* this.sendToThread(threadId, batch);
+    }
+  }
+  async *sendToThread(threadId, messages) {
+    const thread = this.agentThreads.get(threadId);
+    if (!thread) {
+      throw new Error(`AgentThreadOrchestrator.sendToThread: unknown threadId ${threadId}`);
+    }
+    yield* thread.send(messages);
+  }
+  async *processAgentStreamChunk(chunk, signal) {
+    if (chunk.type === InternalEventType.PASSTHROUGH) {
+      yield chunk.event;
+      return { shouldStopExecution: false };
+    }
+    if (!("thread_id" in chunk) || chunk.thread_id == null) {
+      yield chunk;
+      return { shouldStopExecution: false };
+    }
+    const currentThread = this.agentThreads.get(chunk.thread_id);
+    if (!currentThread) {
+      throw new Error("unreachable");
+    }
+    switch (chunk.type) {
+      case EventType.TOOL_APPROVAL_REQUIRED:
+      case EventType.TOOL_RESPONSE_REQUIRED:
+        yield chunk;
+        return { shouldStopExecution: true };
+      case InternalEventType.AGENT_DONE: {
+        if (chunk.parent) {
+          if (!chunk.send_to_parent) throw new Error("unreachable");
+          const parentThread = this.agentThreads.get(chunk.parent.thread_id);
+          if (!parentThread) throw new Error("unreachable: parent thread missing");
+          const subAgentToolIsOpen = parentThread.hasOpenToolCallId(chunk.parent.tool_call_id);
+          if (subAgentToolIsOpen) {
+            const parentToolResponse = {
+              type: EventType.TOOL_RESPONSE,
+              id: newEventId(),
+              created_at: (/* @__PURE__ */ new Date()).toISOString(),
+              thread_id: chunk.parent.thread_id,
+              tool_call_id: chunk.send_to_parent.tool_call_id,
+              content: ""
+            };
+            yield parentToolResponse;
+            yield* this.sendToThread(chunk.parent.thread_id, [chunk.send_to_parent]);
+          }
+          yield chunk;
+          addAgentThreadMetrics(this.finishedSubAgentMetrics, currentThread.getAgentThreadMetrics());
+          this.agentThreads.delete(chunk.thread_id);
+          return { shouldStopExecution: false };
+        }
+        yield chunk;
+        return { shouldStopExecution: true };
+      }
+      case InternalEventType.AGENT_CREATE_SUBAGENT: {
+        const parent = {
+          tool_call_id: chunk.tool_call_id,
+          thread_id: chunk.thread_id
+        };
+        const subAgentThreadId = getThreadId();
+        const subAgentThread = await this.createDynamicSubAgentThread({
+          parentDefinition: currentThread.definition,
+          request: chunk.agent_info,
+          threadId: subAgentThreadId,
+          parent,
+          signal
+        });
+        yield {
+          type: EventType.THREAD_CREATED,
+          id: newEventId(),
+          agent_info: {
+            type: "dynamic",
+            name: chunk.agent_info.name,
+            input: chunk.agent_info.input,
+            model: chunk.agent_info.model
+          },
+          created_at: (/* @__PURE__ */ new Date()).toISOString(),
+          parent,
+          title: chunk.agent_info.name,
+          thread_id: subAgentThreadId
+        };
+        this.agentThreads.set(subAgentThreadId, subAgentThread);
+        return { shouldStopExecution: false };
+      }
+      default:
+        yield chunk;
+        return { shouldStopExecution: false };
+    }
+  }
+  async *execute({
+    signal
+  }) {
+    const { agentThreads } = this;
+    let shouldStopExecution = false;
+    let caughtError;
+    let output = null;
+    const requiredActions = [];
+    let rootAgentError;
+    const pendingAuthEvents = [];
+    const mainThread = [...agentThreads.values()].find((e) => !e.parent);
+    if (!mainThread) {
+      throw new Error("Unreachable: no root thread found");
+    }
+    const rootSpan = createRootAgentSpan(mainThread, this.tracing);
+    onSignalAbort(signal, () => {
+      shouldStopExecution = true;
+    });
+    try {
+      while (agentThreads.size > 0) {
+        if (shouldStopExecution) break;
+        const activeThreads = getActiveAgentThreads(agentThreads);
+        for (let i = 0; i < activeThreads.length && !shouldStopExecution; i += MAX_PARALLEL_SUB_AGENTS) {
+          const batch = activeThreads.slice(i, i + MAX_PARALLEL_SUB_AGENTS);
+          const generators = batch.map((thread) => {
+            if (!thread.parent) {
+              return rootSpan.wrapThreadWithContext(thread, signal);
+            }
+            return wrapWithSubAgentSpan(rootSpan.trace, thread, signal);
+          });
+          for await (const chunk of mergeAsyncGenerators(generators, this.logger)) {
+            if (chunk.type === InternalEventType.MCP_AUTH_REQUIRED) {
+              pendingAuthEvents.push(chunk);
+              shouldStopExecution = true;
+              continue;
+            }
+            if (chunk.type === InternalEventType.AGENT_DONE && !chunk.parent) {
+              rootSpan.setOutputFromEvent(chunk);
+            }
+            const result = yield* this.processAgentStreamChunk(chunk, signal);
+            if (result.shouldStopExecution) {
+              shouldStopExecution = true;
+              if (!rootAgentError && chunk.type === InternalEventType.AGENT_DONE && isInternalThreadDoneError(chunk) && !chunk.parent) {
+                rootAgentError = { error: chunk.error, output: chunk.output };
+              }
+              if (chunk.type !== InternalEventType.PASSTHROUGH) {
+                const { output: outputContribution, requiredAction } = agentThreadEventToTerminalFields(chunk);
+                if (outputContribution) output = outputContribution;
+                if (requiredAction) requiredActions.push(requiredAction);
+              }
+            }
+          }
+        }
+      }
+    } catch (error) {
+      caughtError = error;
+      throw error;
+    } finally {
+      rootSpan.finalize(mainThread, caughtError);
+      rootSpan.end();
+    }
+    if (pendingAuthEvents.length > 0) {
+      const authEvent = mergeAuthEvents(pendingAuthEvents);
+      yield authEvent;
+      const authRequiredAction = {
+        type: EventType.MCP_AUTH_REQUIRED,
+        id: authEvent.id,
+        created_at: authEvent.created_at,
+        thread_id: authEvent.thread_id,
+        mcp_servers: authEvent.mcp_servers.map(({ thread_ids, ...s }) => {
+          void thread_ids;
+          return s;
+        })
+      };
+      requiredActions.push(authRequiredAction);
+    }
+    return {
+      output,
+      required_actions: requiredActions,
+      root_agent_error: rootAgentError
+    };
+  }
+};
+export {
+  AgentThreadOrchestrator,
+  wrapWithSubAgentSpan
+};
+//# sourceMappingURL=AgentThreadOrchestrator.mjs.map

package/dist/core/runtime/AgentThreadOrchestrator.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/core/runtime/AgentThreadOrchestrator.ts"],"sourcesContent":["import type { Logger } from 'winston';\nimport { AgentHarnessError, InvalidAgentSendInputError } from '../errors';\nimport {\n  EventType,\n  newEventId,\n  type ActionRequiredEvent,\n  type ModelMessageEvent,\n  type ToolResponseEvent,\n  type UserToolApprovalMessage,\n  type UserToolResponseMessage,\n} from '../events/schema';\nimport type { AgentExecutionTrace, AgentTracing } from '../tracing/AgentTracing';\nimport { onSignalAbort } from '../util/abort';\nimport { mergeAsyncGenerators } from '../util/promiseUtils';\nimport { AgentThread } from './AgentThread';\nimport type { AgentThreadRuntimeSendBatch } from './AgentThread.types';\nimport {\n  InternalEventType,\n  type AgentThreadAppendContext,\n  type AgentThreadEvent,\n  type AgentThreadExecutionEvent,\n  type AgentThreadExecutionResult,\n  type AgentThreadSendBatch,\n  type InternalMCPAuthRequiredEvent,\n  type InternalMCPServerAuthInfo,\n} from './AgentThread.types';\nimport {\n  assistantMessageContentToStringForSubAgent,\n  getThreadId,\n  isApprovalDecisionMessage,\n  isClientSideToolResponseMessage,\n  isInternalThreadDoneError,\n} from './contextUtils';\nimport type { CreateDynamicSubAgentThread } from './CreateDynamicSubAgentThread';\nimport { addAgentThreadMetrics, createEmptyAgentThreadMetrics, type AgentThreadMetrics } from './metrics';\n\nconst MAX_PARALLEL_SUB_AGENTS = 5;\n\nfunction agentThreadEventToTerminalFields(event: AgentThreadExecutionEvent): {\n  output?: ModelMessageEvent | undefined;\n  requiredAction?: ActionRequiredEvent | undefined;\n} {\n  switch (event.type) {\n    case InternalEventType.AGENT_DONE: {\n      if ('parent' in event && event.parent) return {};\n      if (event.status === 'error') return {};\n      return { output: event.output };\n    }\n    case EventType.TOOL_APPROVAL_REQUIRED:\n    case EventType.TOOL_RESPONSE_REQUIRED:\n      return { requiredAction: event };\n    default:\n      return {};\n  }\n}\n\ntype UserToolApprovalOrResponseBatch = (UserToolApprovalMessage | UserToolResponseMessage)[];\n\nfunction isUserToolApprovalOrResponseBatch(\n  messages: AgentThreadSendBatch,\n): messages is UserToolApprovalOrResponseBatch {\n  const first = messages[0];\n  return first !== undefined && (isApprovalDecisionMessage(first) || isClientSideToolResponseMessage(first));\n}\n\nfunction getMainThreadId(agentThreads: Map<string, AgentThread>): string {\n  for (const thread of agentThreads.values()) {\n    if (!thread.parent) return thread.threadId;\n  }\n  throw new Error('Unreachable: no root thread found');\n}\n\nfunction getActiveAgentThreads(agentThreads: Map<string, AgentThread>): AgentThread[] {\n  const nodes = new Set(agentThreads.keys());\n  const parentNodes = new Set(\n    [...agentThreads.values()].map(a => a.parent?.thread_id).filter((a): a is string => Boolean(a)),\n  );\n  const leaves = [...nodes].filter(id => !parentNodes.has(id));\n  if (leaves.length === 0) {\n    throw new Error('We cannot have zero leave agents');\n  }\n  return leaves.map(id => {\n    const agentThread = agentThreads.get(id);\n    if (!agentThread) throw new Error('unreachable');\n    return agentThread;\n  });\n}\n\nfunction mergeAuthEvents(events: InternalMCPAuthRequiredEvent[]): InternalMCPAuthRequiredEvent {\n  const serverMap = new Map<string, InternalMCPServerAuthInfo>();\n  for (const event of events) {\n    for (const entry of event.mcp_servers) {\n      const existing = serverMap.get(entry.id);\n      if (existing) {\n        existing.thread_ids.push(...entry.thread_ids);\n      } else {\n        serverMap.set(entry.id, { ...entry, thread_ids: [...entry.thread_ids] });\n      }\n    }\n  }\n  return {\n    type: InternalEventType.MCP_AUTH_REQUIRED,\n    id: newEventId(),\n    created_at: new Date().toISOString(),\n    thread_id: null,\n    mcp_servers: [...serverMap.values()],\n  };\n}\n\nfunction wrapGeneratorWithTrace<T, TReturn = void, TNext = unknown>(\n  trace: AgentExecutionTrace,\n  generator: AsyncGenerator<T, TReturn, TNext>,\n): AsyncGenerator<T, TReturn, TNext> {\n  const wrapped: AsyncGenerator<T, TReturn, TNext> = {\n    [Symbol.asyncIterator]() {\n      return this;\n    },\n    next(...[value]: [] | [TNext]) {\n      return trace.runInContext(() => (value === undefined ? generator.next() : generator.next(value)));\n    },\n    return(value: TReturn | PromiseLike<TReturn>) {\n      return trace.runInContext(() => generator.return(value));\n    },\n    throw(e: unknown) {\n      return trace.runInContext(() => generator.throw(e));\n    },\n    [Symbol.asyncDispose]() {\n      return generator[Symbol.asyncDispose]();\n    },\n  };\n  return wrapped;\n}\n\ninterface RootAgentSpanHandle {\n  trace: AgentExecutionTrace;\n  wrapThreadWithContext(thread: AgentThread, signal?: AbortSignal): AsyncGenerator<AgentThreadEvent>;\n  setOutputFromEvent(chunk: AgentThreadEvent): void;\n  finalize(thread: AgentThread, error?: unknown): void;\n  end(): void;\n}\n\nfunction createRootAgentSpan(mainThread: AgentThread, tracing: AgentTracing): RootAgentSpanHandle {\n  const trace = tracing.startRootSpan(\n    JSON.stringify({ instructions: mainThread.definition.instruction, messages: mainThread.definition.messages }),\n  );\n  let rootAgentErrorMessage: string | undefined;\n  let rootAgentCompleted = false;\n\n  return {\n    trace,\n    wrapThreadWithContext(thread: AgentThread, signal?: AbortSignal) {\n      return wrapGeneratorWithTrace(trace, thread.execute({ signal }));\n    },\n    setOutputFromEvent(chunk: AgentThreadEvent) {\n      rootAgentCompleted = true;\n      if (chunk.type === InternalEventType.AGENT_DONE) {\n        if (isInternalThreadDoneError(chunk)) {\n          rootAgentErrorMessage = chunk.error;\n          trace.setOutput(JSON.stringify({ error: chunk.error }));\n        } else {\n          const content = assistantMessageContentToStringForSubAgent(chunk.output.content);\n          trace.setOutput(JSON.stringify({ result: content }));\n        }\n      }\n    },\n    finalize(thread: AgentThread, error?: unknown) {\n      trace.setMetrics(thread.getAgentThreadMetrics());\n      if (error) {\n        trace.setError(error);\n      } else if (rootAgentErrorMessage || !rootAgentCompleted) {\n        trace.setError(rootAgentErrorMessage ?? 'Agent execution interrupted');\n      } else {\n        trace.setSuccess();\n      }\n    },\n    end() {\n      trace.end();\n    },\n  };\n}\n\n/** @internal Exported for sub-agent span finalization tests. */\nexport async function* wrapWithSubAgentSpan(\n  parentTrace: AgentExecutionTrace,\n  currentThread: AgentThread,\n  signal?: AbortSignal,\n): AsyncGenerator<AgentThreadEvent> {\n  const agentInfo = currentThread.agentInfo;\n  if (!agentInfo) {\n    throw new Error('Sub-agent thread is missing agentInfo');\n  }\n  const subTrace = parentTrace.startSubAgent(agentInfo.name, JSON.stringify({ instructions: agentInfo.input }));\n  const generator = wrapGeneratorWithTrace(subTrace, currentThread.execute({ signal }));\n\n  try {\n    for await (const event of generator) {\n      if (event.type === InternalEventType.AGENT_DONE) {\n        if (isInternalThreadDoneError(event)) {\n          subTrace.setOutput(JSON.stringify({ error: event.error }));\n          subTrace.setMetrics(currentThread.getAgentThreadMetrics());\n          subTrace.setError(event.error);\n        } else {\n          const content = assistantMessageContentToStringForSubAgent(event.output.content);\n          subTrace.setOutput(JSON.stringify({ result: content }));\n          subTrace.setMetrics(currentThread.getAgentThreadMetrics());\n          subTrace.setSuccess();\n        }\n        // end() only in finally so every path finalizes exactly once\n      }\n      yield event;\n    }\n  } catch (error) {\n    subTrace.setError(error);\n    throw error;\n  } finally {\n    subTrace.end();\n  }\n}\n\nexport interface AgentThreadOrchestratorInput {\n  agentThreads: Map<string, AgentThread>;\n  createDynamicSubAgentThread: CreateDynamicSubAgentThread;\n  tracing: AgentTracing;\n  logger: Logger;\n}\n\nexport class AgentThreadOrchestrator {\n  private readonly agentThreads: Map<string, AgentThread>;\n  private readonly createDynamicSubAgentThread: CreateDynamicSubAgentThread;\n  private readonly tracing: AgentTracing;\n  private readonly logger: Logger;\n  // Finished sub-agents removed from `agentThreads`; kept so totals still include them.\n  private finishedSubAgentMetrics: AgentThreadMetrics = createEmptyAgentThreadMetrics();\n\n  constructor(params: AgentThreadOrchestratorInput) {\n    this.agentThreads = params.agentThreads;\n    this.createDynamicSubAgentThread = params.createDynamicSubAgentThread;\n    this.tracing = params.tracing;\n    this.logger = params.logger.child({ module: 'AgentThreadOrchestrator' });\n  }\n\n  /**\n   * Turn-wide metrics so far: live threads plus finished sub-agents.\n   * Safe mid-flight and after cancel/error (`execute()` may not return).\n   * Each thread is counted once — a sub-agent moves from live → finished atomically.\n   */\n  public getMetrics(): AgentThreadMetrics {\n    const total = createEmptyAgentThreadMetrics();\n    addAgentThreadMetrics(total, this.finishedSubAgentMetrics);\n    for (const thread of this.agentThreads.values()) {\n      addAgentThreadMetrics(total, thread.getAgentThreadMetrics());\n    }\n    return total;\n  }\n\n  public async *send(messages: AgentThreadSendBatch): AsyncGenerator<AgentThreadAppendContext, void, unknown> {\n    const byThread = new Map<string, AgentThreadRuntimeSendBatch>();\n    for (const thread of this.agentThreads.values()) {\n      byThread.set(thread.threadId, []);\n    }\n\n    if (isUserToolApprovalOrResponseBatch(messages)) {\n      const grouped = new Map<string, UserToolApprovalOrResponseBatch[number][]>();\n      for (const msg of messages) {\n        const threadId = msg.thread_id;\n        if (!byThread.has(threadId)) {\n          throw new InvalidAgentSendInputError(`unknown thread_id: ${threadId}`);\n        }\n        const batch = grouped.get(threadId) ?? [];\n        batch.push(msg);\n        grouped.set(threadId, batch);\n      }\n      for (const [threadId, batch] of grouped) {\n        byThread.set(threadId, batch);\n      }\n    } else if (messages.length > 0) {\n      if (this.agentThreads.size > 1) {\n        throw new InvalidAgentSendInputError(\n          'Cannot process user messages while sub agents are running, please send empty input for previous conversation to complete',\n        );\n      }\n      byThread.set(getMainThreadId(this.agentThreads), messages);\n    }\n\n    const validationErrors: string[] = [];\n    for (const [threadId, batch] of byThread) {\n      const thread = this.agentThreads.get(threadId);\n      if (!thread) {\n        throw new Error(`AgentThreadOrchestrator.send: unknown threadId ${threadId}`);\n      }\n      try {\n        thread.validateSendInput(batch);\n      } catch (e) {\n        if (e instanceof AgentHarnessError && e.code === 'invalid_send_input') {\n          validationErrors.push(`thread ${threadId}: ${e.message}`);\n        } else {\n          throw e;\n        }\n      }\n    }\n    if (validationErrors.length > 0) {\n      throw new InvalidAgentSendInputError(validationErrors.join('; '));\n    }\n\n    for (const [threadId, batch] of byThread) {\n      yield* this.sendToThread(threadId, batch);\n    }\n  }\n\n  private async *sendToThread(\n    threadId: string,\n    messages: AgentThreadRuntimeSendBatch,\n  ): AsyncGenerator<AgentThreadAppendContext, void, unknown> {\n    const thread = this.agentThreads.get(threadId);\n    if (!thread) {\n      throw new Error(`AgentThreadOrchestrator.sendToThread: unknown threadId ${threadId}`);\n    }\n    yield* thread.send(messages);\n  }\n\n  private async *processAgentStreamChunk(\n    chunk: Exclude<AgentThreadEvent, InternalMCPAuthRequiredEvent>,\n    signal: AbortSignal,\n  ): AsyncGenerator<AgentThreadExecutionEvent, { shouldStopExecution: boolean }, unknown> {\n    if (chunk.type === InternalEventType.PASSTHROUGH) {\n      yield chunk.event;\n      return { shouldStopExecution: false };\n    }\n\n    if (!('thread_id' in chunk) || chunk.thread_id == null) {\n      yield chunk;\n      return { shouldStopExecution: false };\n    }\n\n    const currentThread = this.agentThreads.get(chunk.thread_id);\n    if (!currentThread) {\n      throw new Error('unreachable');\n    }\n\n    switch (chunk.type) {\n      case EventType.TOOL_APPROVAL_REQUIRED:\n      case EventType.TOOL_RESPONSE_REQUIRED:\n        yield chunk;\n        return { shouldStopExecution: true };\n      case InternalEventType.AGENT_DONE: {\n        if (chunk.parent) {\n          if (!chunk.send_to_parent) throw new Error('unreachable');\n          const parentThread = this.agentThreads.get(chunk.parent.thread_id);\n          if (!parentThread) throw new Error('unreachable: parent thread missing');\n          const subAgentToolIsOpen = parentThread.hasOpenToolCallId(chunk.parent.tool_call_id);\n          if (subAgentToolIsOpen) {\n            const parentToolResponse: ToolResponseEvent = {\n              type: EventType.TOOL_RESPONSE,\n              id: newEventId(),\n              created_at: new Date().toISOString(),\n              thread_id: chunk.parent.thread_id,\n              tool_call_id: chunk.send_to_parent.tool_call_id,\n              content: '',\n            };\n            yield parentToolResponse;\n            yield* this.sendToThread(chunk.parent.thread_id, [chunk.send_to_parent]);\n          }\n          yield chunk;\n          // Move metrics to the finished bucket and drop the live entry with no `yield` between,\n          // so getMetrics() counts this sub-agent once. After `yield chunk` per durable-state.\n          addAgentThreadMetrics(this.finishedSubAgentMetrics, currentThread.getAgentThreadMetrics());\n          this.agentThreads.delete(chunk.thread_id);\n          return { shouldStopExecution: false };\n        }\n        yield chunk;\n        return { shouldStopExecution: true };\n      }\n      case InternalEventType.AGENT_CREATE_SUBAGENT: {\n        const parent = {\n          tool_call_id: chunk.tool_call_id,\n          thread_id: chunk.thread_id,\n        };\n        const subAgentThreadId = getThreadId();\n        const subAgentThread = await this.createDynamicSubAgentThread({\n          parentDefinition: currentThread.definition,\n          request: chunk.agent_info,\n          threadId: subAgentThreadId,\n          parent,\n          signal,\n        });\n\n        yield {\n          type: EventType.THREAD_CREATED,\n          id: newEventId(),\n          agent_info: {\n            type: 'dynamic',\n            name: chunk.agent_info.name,\n            input: chunk.agent_info.input,\n            model: chunk.agent_info.model,\n          },\n          created_at: new Date().toISOString(),\n          parent,\n          title: chunk.agent_info.name,\n          thread_id: subAgentThreadId,\n        };\n        this.agentThreads.set(subAgentThreadId, subAgentThread);\n        return { shouldStopExecution: false };\n      }\n      default:\n        yield chunk;\n        return { shouldStopExecution: false };\n    }\n  }\n\n  public async *execute({\n    signal,\n  }: {\n    signal: AbortSignal;\n  }): AsyncGenerator<AgentThreadExecutionEvent, AgentThreadExecutionResult, unknown> {\n    const { agentThreads } = this;\n\n    let shouldStopExecution = false;\n    let caughtError: unknown;\n    let output: ModelMessageEvent | null = null;\n    const requiredActions: ActionRequiredEvent[] = [];\n    let rootAgentError: AgentThreadExecutionResult['root_agent_error'];\n    const pendingAuthEvents: InternalMCPAuthRequiredEvent[] = [];\n    const mainThread = [...agentThreads.values()].find(e => !e.parent);\n    if (!mainThread) {\n      throw new Error('Unreachable: no root thread found');\n    }\n    const rootSpan = createRootAgentSpan(mainThread, this.tracing);\n\n    onSignalAbort(signal, () => {\n      shouldStopExecution = true;\n    });\n\n    try {\n      while (agentThreads.size > 0) {\n        if (shouldStopExecution) break;\n\n        const activeThreads = getActiveAgentThreads(agentThreads);\n\n        for (let i = 0; i < activeThreads.length && !shouldStopExecution; i += MAX_PARALLEL_SUB_AGENTS) {\n          const batch = activeThreads.slice(i, i + MAX_PARALLEL_SUB_AGENTS);\n          const generators = batch.map(thread => {\n            if (!thread.parent) {\n              return rootSpan.wrapThreadWithContext(thread, signal);\n            }\n            return wrapWithSubAgentSpan(rootSpan.trace, thread, signal);\n          });\n\n          for await (const chunk of mergeAsyncGenerators(generators, this.logger)) {\n            if (chunk.type === InternalEventType.MCP_AUTH_REQUIRED) {\n              pendingAuthEvents.push(chunk);\n              shouldStopExecution = true;\n              continue;\n            }\n            if (chunk.type === InternalEventType.AGENT_DONE && !chunk.parent) {\n              // Root-thread only — sub-agent AGENT_DONE would overwrite root trace output /\n              // leave a stale error; sub-agent spans finalize in wrapWithSubAgentSpan.\n              rootSpan.setOutputFromEvent(chunk);\n            }\n\n            const result = yield* this.processAgentStreamChunk(chunk, signal);\n            if (result.shouldStopExecution) {\n              shouldStopExecution = true;\n              if (\n                !rootAgentError &&\n                chunk.type === InternalEventType.AGENT_DONE &&\n                isInternalThreadDoneError(chunk) &&\n                !chunk.parent\n              ) {\n                rootAgentError = { error: chunk.error, output: chunk.output };\n              }\n              if (chunk.type !== InternalEventType.PASSTHROUGH) {\n                const { output: outputContribution, requiredAction } = agentThreadEventToTerminalFields(chunk);\n                if (outputContribution) output = outputContribution;\n                if (requiredAction) requiredActions.push(requiredAction);\n              }\n            }\n          }\n        }\n      }\n    } catch (error) {\n      caughtError = error;\n      throw error;\n    } finally {\n      rootSpan.finalize(mainThread, caughtError);\n      rootSpan.end();\n    }\n\n    if (pendingAuthEvents.length > 0) {\n      const authEvent = mergeAuthEvents(pendingAuthEvents);\n      yield authEvent;\n      const authRequiredAction: ActionRequiredEvent = {\n        type: EventType.MCP_AUTH_REQUIRED,\n        id: authEvent.id,\n        created_at: authEvent.created_at,\n        thread_id: authEvent.thread_id,\n        mcp_servers: authEvent.mcp_servers.map(({ thread_ids, ...s }) => {\n          void thread_ids;\n          return s;\n        }),\n      };\n      requiredActions.push(authRequiredAction);\n    }\n\n    return {\n      output,\n      required_actions: requiredActions,\n      root_agent_error: rootAgentError,\n    };\n  }\n}\n"],"mappings":";AACA,SAAS,mBAAmB,kCAAkC;AAC9D;AAAA,EACE;AAAA,EACA;AAAA,OAMK;AAEP,SAAS,qBAAqB;AAC9B,SAAS,4BAA4B;AACrC,OAA4B;AAE5B;AAAA,EACE;AAAA,OAQK;AACP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAEP,SAAS,uBAAuB,qCAA8D;AAE9F,IAAM,0BAA0B;AAEhC,SAAS,iCAAiC,OAGxC;AACA,UAAQ,MAAM,MAAM;AAAA,IAClB,KAAK,kBAAkB,YAAY;AACjC,UAAI,YAAY,SAAS,MAAM,OAAQ,QAAO,CAAC;AAC/C,UAAI,MAAM,WAAW,QAAS,QAAO,CAAC;AACtC,aAAO,EAAE,QAAQ,MAAM,OAAO;AAAA,IAChC;AAAA,IACA,KAAK,UAAU;AAAA,IACf,KAAK,UAAU;AACb,aAAO,EAAE,gBAAgB,MAAM;AAAA,IACjC;AACE,aAAO,CAAC;AAAA,EACZ;AACF;AAIA,SAAS,kCACP,UAC6C;AAC7C,QAAM,QAAQ,SAAS,CAAC;AACxB,SAAO,UAAU,WAAc,0BAA0B,KAAK,KAAK,gCAAgC,KAAK;AAC1G;AAEA,SAAS,gBAAgB,cAAgD;AACvE,aAAW,UAAU,aAAa,OAAO,GAAG;AAC1C,QAAI,CAAC,OAAO,OAAQ,QAAO,OAAO;AAAA,EACpC;AACA,QAAM,IAAI,MAAM,mCAAmC;AACrD;AAEA,SAAS,sBAAsB,cAAuD;AACpF,QAAM,QAAQ,IAAI,IAAI,aAAa,KAAK,CAAC;AACzC,QAAM,cAAc,IAAI;AAAA,IACtB,CAAC,GAAG,aAAa,OAAO,CAAC,EAAE,IAAI,OAAK,EAAE,QAAQ,SAAS,EAAE,OAAO,CAAC,MAAmB,QAAQ,CAAC,CAAC;AAAA,EAChG;AACA,QAAM,SAAS,CAAC,GAAG,KAAK,EAAE,OAAO,QAAM,CAAC,YAAY,IAAI,EAAE,CAAC;AAC3D,MAAI,OAAO,WAAW,GAAG;AACvB,UAAM,IAAI,MAAM,kCAAkC;AAAA,EACpD;AACA,SAAO,OAAO,IAAI,QAAM;AACtB,UAAM,cAAc,aAAa,IAAI,EAAE;AACvC,QAAI,CAAC,YAAa,OAAM,IAAI,MAAM,aAAa;AAC/C,WAAO;AAAA,EACT,CAAC;AACH;AAEA,SAAS,gBAAgB,QAAsE;AAC7F,QAAM,YAAY,oBAAI,IAAuC;AAC7D,aAAW,SAAS,QAAQ;AAC1B,eAAW,SAAS,MAAM,aAAa;AACrC,YAAM,WAAW,UAAU,IAAI,MAAM,EAAE;AACvC,UAAI,UAAU;AACZ,iBAAS,WAAW,KAAK,GAAG,MAAM,UAAU;AAAA,MAC9C,OAAO;AACL,kBAAU,IAAI,MAAM,IAAI,EAAE,GAAG,OAAO,YAAY,CAAC,GAAG,MAAM,UAAU,EAAE,CAAC;AAAA,MACzE;AAAA,IACF;AAAA,EACF;AACA,SAAO;AAAA,IACL,MAAM,kBAAkB;AAAA,IACxB,IAAI,WAAW;AAAA,IACf,aAAY,oBAAI,KAAK,GAAE,YAAY;AAAA,IACnC,WAAW;AAAA,IACX,aAAa,CAAC,GAAG,UAAU,OAAO,CAAC;AAAA,EACrC;AACF;AAEA,SAAS,uBACP,OACA,WACmC;AACnC,QAAM,UAA6C;AAAA,IACjD,CAAC,OAAO,aAAa,IAAI;AACvB,aAAO;AAAA,IACT;AAAA,IACA,QAAQ,CAAC,KAAK,GAAiB;AAC7B,aAAO,MAAM,aAAa,MAAO,UAAU,SAAY,UAAU,KAAK,IAAI,UAAU,KAAK,KAAK,CAAE;AAAA,IAClG;AAAA,IACA,OAAO,OAAuC;AAC5C,aAAO,MAAM,aAAa,MAAM,UAAU,OAAO,KAAK,CAAC;AAAA,IACzD;AAAA,IACA,MAAM,GAAY;AAChB,aAAO,MAAM,aAAa,MAAM,UAAU,MAAM,CAAC,CAAC;AAAA,IACpD;AAAA,IACA,CAAC,OAAO,YAAY,IAAI;AACtB,aAAO,UAAU,OAAO,YAAY,EAAE;AAAA,IACxC;AAAA,EACF;AACA,SAAO;AACT;AAUA,SAAS,oBAAoB,YAAyB,SAA4C;AAChG,QAAM,QAAQ,QAAQ;AAAA,IACpB,KAAK,UAAU,EAAE,cAAc,WAAW,WAAW,aAAa,UAAU,WAAW,WAAW,SAAS,CAAC;AAAA,EAC9G;AACA,MAAI;AACJ,MAAI,qBAAqB;AAEzB,SAAO;AAAA,IACL;AAAA,IACA,sBAAsB,QAAqB,QAAsB;AAC/D,aAAO,uBAAuB,OAAO,OAAO,QAAQ,EAAE,OAAO,CAAC,CAAC;AAAA,IACjE;AAAA,IACA,mBAAmB,OAAyB;AAC1C,2BAAqB;AACrB,UAAI,MAAM,SAAS,kBAAkB,YAAY;AAC/C,YAAI,0BAA0B,KAAK,GAAG;AACpC,kCAAwB,MAAM;AAC9B,gBAAM,UAAU,KAAK,UAAU,EAAE,OAAO,MAAM,MAAM,CAAC,CAAC;AAAA,QACxD,OAAO;AACL,gBAAM,UAAU,2CAA2C,MAAM,OAAO,OAAO;AAC/E,gBAAM,UAAU,KAAK,UAAU,EAAE,QAAQ,QAAQ,CAAC,CAAC;AAAA,QACrD;AAAA,MACF;AAAA,IACF;AAAA,IACA,SAAS,QAAqB,OAAiB;AAC7C,YAAM,WAAW,OAAO,sBAAsB,CAAC;AAC/C,UAAI,OAAO;AACT,cAAM,SAAS,KAAK;AAAA,MACtB,WAAW,yBAAyB,CAAC,oBAAoB;AACvD,cAAM,SAAS,yBAAyB,6BAA6B;AAAA,MACvE,OAAO;AACL,cAAM,WAAW;AAAA,MACnB;AAAA,IACF;AAAA,IACA,MAAM;AACJ,YAAM,IAAI;AAAA,IACZ;AAAA,EACF;AACF;AAGA,gBAAuB,qBACrB,aACA,eACA,QACkC;AAClC,QAAM,YAAY,cAAc;AAChC,MAAI,CAAC,WAAW;AACd,UAAM,IAAI,MAAM,uCAAuC;AAAA,EACzD;AACA,QAAM,WAAW,YAAY,cAAc,UAAU,MAAM,KAAK,UAAU,EAAE,cAAc,UAAU,MAAM,CAAC,CAAC;AAC5G,QAAM,YAAY,uBAAuB,UAAU,cAAc,QAAQ,EAAE,OAAO,CAAC,CAAC;AAEpF,MAAI;AACF,qBAAiB,SAAS,WAAW;AACnC,UAAI,MAAM,SAAS,kBAAkB,YAAY;AAC/C,YAAI,0BAA0B,KAAK,GAAG;AACpC,mBAAS,UAAU,KAAK,UAAU,EAAE,OAAO,MAAM,MAAM,CAAC,CAAC;AACzD,mBAAS,WAAW,cAAc,sBAAsB,CAAC;AACzD,mBAAS,SAAS,MAAM,KAAK;AAAA,QAC/B,OAAO;AACL,gBAAM,UAAU,2CAA2C,MAAM,OAAO,OAAO;AAC/E,mBAAS,UAAU,KAAK,UAAU,EAAE,QAAQ,QAAQ,CAAC,CAAC;AACtD,mBAAS,WAAW,cAAc,sBAAsB,CAAC;AACzD,mBAAS,WAAW;AAAA,QACtB;AAAA,MAEF;AACA,YAAM;AAAA,IACR;AAAA,EACF,SAAS,OAAO;AACd,aAAS,SAAS,KAAK;AACvB,UAAM;AAAA,EACR,UAAE;AACA,aAAS,IAAI;AAAA,EACf;AACF;AASO,IAAM,0BAAN,MAA8B;AAAA,EAClB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAET,0BAA8C,8BAA8B;AAAA,EAEpF,YAAY,QAAsC;AAChD,SAAK,eAAe,OAAO;AAC3B,SAAK,8BAA8B,OAAO;AAC1C,SAAK,UAAU,OAAO;AACtB,SAAK,SAAS,OAAO,OAAO,MAAM,EAAE,QAAQ,0BAA0B,CAAC;AAAA,EACzE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOO,aAAiC;AACtC,UAAM,QAAQ,8BAA8B;AAC5C,0BAAsB,OAAO,KAAK,uBAAuB;AACzD,eAAW,UAAU,KAAK,aAAa,OAAO,GAAG;AAC/C,4BAAsB,OAAO,OAAO,sBAAsB,CAAC;AAAA,IAC7D;AACA,WAAO;AAAA,EACT;AAAA,EAEA,OAAc,KAAK,UAAyF;AAC1G,UAAM,WAAW,oBAAI,IAAyC;AAC9D,eAAW,UAAU,KAAK,aAAa,OAAO,GAAG;AAC/C,eAAS,IAAI,OAAO,UAAU,CAAC,CAAC;AAAA,IAClC;AAEA,QAAI,kCAAkC,QAAQ,GAAG;AAC/C,YAAM,UAAU,oBAAI,IAAuD;AAC3E,iBAAW,OAAO,UAAU;AAC1B,cAAM,WAAW,IAAI;AACrB,YAAI,CAAC,SAAS,IAAI,QAAQ,GAAG;AAC3B,gBAAM,IAAI,2BAA2B,sBAAsB,QAAQ,EAAE;AAAA,QACvE;AACA,cAAM,QAAQ,QAAQ,IAAI,QAAQ,KAAK,CAAC;AACxC,cAAM,KAAK,GAAG;AACd,gBAAQ,IAAI,UAAU,KAAK;AAAA,MAC7B;AACA,iBAAW,CAAC,UAAU,KAAK,KAAK,SAAS;AACvC,iBAAS,IAAI,UAAU,KAAK;AAAA,MAC9B;AAAA,IACF,WAAW,SAAS,SAAS,GAAG;AAC9B,UAAI,KAAK,aAAa,OAAO,GAAG;AAC9B,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AACA,eAAS,IAAI,gBAAgB,KAAK,YAAY,GAAG,QAAQ;AAAA,IAC3D;AAEA,UAAM,mBAA6B,CAAC;AACpC,eAAW,CAAC,UAAU,KAAK,KAAK,UAAU;AACxC,YAAM,SAAS,KAAK,aAAa,IAAI,QAAQ;AAC7C,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,kDAAkD,QAAQ,EAAE;AAAA,MAC9E;AACA,UAAI;AACF,eAAO,kBAAkB,KAAK;AAAA,MAChC,SAAS,GAAG;AACV,YAAI,aAAa,qBAAqB,EAAE,SAAS,sBAAsB;AACrE,2BAAiB,KAAK,UAAU,QAAQ,KAAK,EAAE,OAAO,EAAE;AAAA,QAC1D,OAAO;AACL,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IACF;AACA,QAAI,iBAAiB,SAAS,GAAG;AAC/B,YAAM,IAAI,2BAA2B,iBAAiB,KAAK,IAAI,CAAC;AAAA,IAClE;AAEA,eAAW,CAAC,UAAU,KAAK,KAAK,UAAU;AACxC,aAAO,KAAK,aAAa,UAAU,KAAK;AAAA,IAC1C;AAAA,EACF;AAAA,EAEA,OAAe,aACb,UACA,UACyD;AACzD,UAAM,SAAS,KAAK,aAAa,IAAI,QAAQ;AAC7C,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI,MAAM,0DAA0D,QAAQ,EAAE;AAAA,IACtF;AACA,WAAO,OAAO,KAAK,QAAQ;AAAA,EAC7B;AAAA,EAEA,OAAe,wBACb,OACA,QACsF;AACtF,QAAI,MAAM,SAAS,kBAAkB,aAAa;AAChD,YAAM,MAAM;AACZ,aAAO,EAAE,qBAAqB,MAAM;AAAA,IACtC;AAEA,QAAI,EAAE,eAAe,UAAU,MAAM,aAAa,MAAM;AACtD,YAAM;AACN,aAAO,EAAE,qBAAqB,MAAM;AAAA,IACtC;AAEA,UAAM,gBAAgB,KAAK,aAAa,IAAI,MAAM,SAAS;AAC3D,QAAI,CAAC,eAAe;AAClB,YAAM,IAAI,MAAM,aAAa;AAAA,IAC/B;AAEA,YAAQ,MAAM,MAAM;AAAA,MAClB,KAAK,UAAU;AAAA,MACf,KAAK,UAAU;AACb,cAAM;AACN,eAAO,EAAE,qBAAqB,KAAK;AAAA,MACrC,KAAK,kBAAkB,YAAY;AACjC,YAAI,MAAM,QAAQ;AAChB,cAAI,CAAC,MAAM,eAAgB,OAAM,IAAI,MAAM,aAAa;AACxD,gBAAM,eAAe,KAAK,aAAa,IAAI,MAAM,OAAO,SAAS;AACjE,cAAI,CAAC,aAAc,OAAM,IAAI,MAAM,oCAAoC;AACvE,gBAAM,qBAAqB,aAAa,kBAAkB,MAAM,OAAO,YAAY;AACnF,cAAI,oBAAoB;AACtB,kBAAM,qBAAwC;AAAA,cAC5C,MAAM,UAAU;AAAA,cAChB,IAAI,WAAW;AAAA,cACf,aAAY,oBAAI,KAAK,GAAE,YAAY;AAAA,cACnC,WAAW,MAAM,OAAO;AAAA,cACxB,cAAc,MAAM,eAAe;AAAA,cACnC,SAAS;AAAA,YACX;AACA,kBAAM;AACN,mBAAO,KAAK,aAAa,MAAM,OAAO,WAAW,CAAC,MAAM,cAAc,CAAC;AAAA,UACzE;AACA,gBAAM;AAGN,gCAAsB,KAAK,yBAAyB,cAAc,sBAAsB,CAAC;AACzF,eAAK,aAAa,OAAO,MAAM,SAAS;AACxC,iBAAO,EAAE,qBAAqB,MAAM;AAAA,QACtC;AACA,cAAM;AACN,eAAO,EAAE,qBAAqB,KAAK;AAAA,MACrC;AAAA,MACA,KAAK,kBAAkB,uBAAuB;AAC5C,cAAM,SAAS;AAAA,UACb,cAAc,MAAM;AAAA,UACpB,WAAW,MAAM;AAAA,QACnB;AACA,cAAM,mBAAmB,YAAY;AACrC,cAAM,iBAAiB,MAAM,KAAK,4BAA4B;AAAA,UAC5D,kBAAkB,cAAc;AAAA,UAChC,SAAS,MAAM;AAAA,UACf,UAAU;AAAA,UACV;AAAA,UACA;AAAA,QACF,CAAC;AAED,cAAM;AAAA,UACJ,MAAM,UAAU;AAAA,UAChB,IAAI,WAAW;AAAA,UACf,YAAY;AAAA,YACV,MAAM;AAAA,YACN,MAAM,MAAM,WAAW;AAAA,YACvB,OAAO,MAAM,WAAW;AAAA,YACxB,OAAO,MAAM,WAAW;AAAA,UAC1B;AAAA,UACA,aAAY,oBAAI,KAAK,GAAE,YAAY;AAAA,UACnC;AAAA,UACA,OAAO,MAAM,WAAW;AAAA,UACxB,WAAW;AAAA,QACb;AACA,aAAK,aAAa,IAAI,kBAAkB,cAAc;AACtD,eAAO,EAAE,qBAAqB,MAAM;AAAA,MACtC;AAAA,MACA;AACE,cAAM;AACN,eAAO,EAAE,qBAAqB,MAAM;AAAA,IACxC;AAAA,EACF;AAAA,EAEA,OAAc,QAAQ;AAAA,IACpB;AAAA,EACF,GAEmF;AACjF,UAAM,EAAE,aAAa,IAAI;AAEzB,QAAI,sBAAsB;AAC1B,QAAI;AACJ,QAAI,SAAmC;AACvC,UAAM,kBAAyC,CAAC;AAChD,QAAI;AACJ,UAAM,oBAAoD,CAAC;AAC3D,UAAM,aAAa,CAAC,GAAG,aAAa,OAAO,CAAC,EAAE,KAAK,OAAK,CAAC,EAAE,MAAM;AACjE,QAAI,CAAC,YAAY;AACf,YAAM,IAAI,MAAM,mCAAmC;AAAA,IACrD;AACA,UAAM,WAAW,oBAAoB,YAAY,KAAK,OAAO;AAE7D,kBAAc,QAAQ,MAAM;AAC1B,4BAAsB;AAAA,IACxB,CAAC;AAED,QAAI;AACF,aAAO,aAAa,OAAO,GAAG;AAC5B,YAAI,oBAAqB;AAEzB,cAAM,gBAAgB,sBAAsB,YAAY;AAExD,iBAAS,IAAI,GAAG,IAAI,cAAc,UAAU,CAAC,qBAAqB,KAAK,yBAAyB;AAC9F,gBAAM,QAAQ,cAAc,MAAM,GAAG,IAAI,uBAAuB;AAChE,gBAAM,aAAa,MAAM,IAAI,YAAU;AACrC,gBAAI,CAAC,OAAO,QAAQ;AAClB,qBAAO,SAAS,sBAAsB,QAAQ,MAAM;AAAA,YACtD;AACA,mBAAO,qBAAqB,SAAS,OAAO,QAAQ,MAAM;AAAA,UAC5D,CAAC;AAED,2BAAiB,SAAS,qBAAqB,YAAY,KAAK,MAAM,GAAG;AACvE,gBAAI,MAAM,SAAS,kBAAkB,mBAAmB;AACtD,gCAAkB,KAAK,KAAK;AAC5B,oCAAsB;AACtB;AAAA,YACF;AACA,gBAAI,MAAM,SAAS,kBAAkB,cAAc,CAAC,MAAM,QAAQ;AAGhE,uBAAS,mBAAmB,KAAK;AAAA,YACnC;AAEA,kBAAM,SAAS,OAAO,KAAK,wBAAwB,OAAO,MAAM;AAChE,gBAAI,OAAO,qBAAqB;AAC9B,oCAAsB;AACtB,kBACE,CAAC,kBACD,MAAM,SAAS,kBAAkB,cACjC,0BAA0B,KAAK,KAC/B,CAAC,MAAM,QACP;AACA,iCAAiB,EAAE,OAAO,MAAM,OAAO,QAAQ,MAAM,OAAO;AAAA,cAC9D;AACA,kBAAI,MAAM,SAAS,kBAAkB,aAAa;AAChD,sBAAM,EAAE,QAAQ,oBAAoB,eAAe,IAAI,iCAAiC,KAAK;AAC7F,oBAAI,mBAAoB,UAAS;AACjC,oBAAI,eAAgB,iBAAgB,KAAK,cAAc;AAAA,cACzD;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,SAAS,OAAO;AACd,oBAAc;AACd,YAAM;AAAA,IACR,UAAE;AACA,eAAS,SAAS,YAAY,WAAW;AACzC,eAAS,IAAI;AAAA,IACf;AAEA,QAAI,kBAAkB,SAAS,GAAG;AAChC,YAAM,YAAY,gBAAgB,iBAAiB;AACnD,YAAM;AACN,YAAM,qBAA0C;AAAA,QAC9C,MAAM,UAAU;AAAA,QAChB,IAAI,UAAU;AAAA,QACd,YAAY,UAAU;AAAA,QACtB,WAAW,UAAU;AAAA,QACrB,aAAa,UAAU,YAAY,IAAI,CAAC,EAAE,YAAY,GAAG,EAAE,MAAM;AAC/D,eAAK;AACL,iBAAO;AAAA,QACT,CAAC;AAAA,MACH;AACA,sBAAgB,KAAK,kBAAkB;AAAA,IACzC;AAEA,WAAO;AAAA,MACL;AAAA,MACA,kBAAkB;AAAA,MAClB,kBAAkB;AAAA,IACpB;AAAA,EACF;AACF;","names":[]}

package/dist/core/runtime/CreateDynamicSubAgentThread.d.ts

@@ -0,0 +1,11 @@
+import type { AgentInfo, AgentParent } from '../events/schema';
+import type { AgentDefinition } from './AgentDefinition';
+import type { AgentThread } from './AgentThread';
+export type CreateDynamicSubAgentThread = (input: {
+    parentDefinition: AgentDefinition;
+    request: AgentInfo;
+    threadId: string;
+    parent: AgentParent;
+    signal: AbortSignal;
+}) => Promise<AgentThread>;
+//# sourceMappingURL=CreateDynamicSubAgentThread.d.ts.map

package/dist/core/runtime/CreateDynamicSubAgentThread.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CreateDynamicSubAgentThread.d.ts","sourceRoot":"","sources":["../../../src/core/runtime/CreateDynamicSubAgentThread.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAEjD,MAAM,MAAM,2BAA2B,GAAG,CAAC,KAAK,EAAE;IAChD,gBAAgB,EAAE,eAAe,CAAC;IAClC,OAAO,EAAE,SAAS,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,WAAW,CAAC;IACpB,MAAM,EAAE,WAAW,CAAC;CACrB,KAAK,OAAO,CAAC,WAAW,CAAC,CAAC"}

package/dist/core/runtime/CreateDynamicSubAgentThread.js

@@ -0,0 +1,19 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/core/runtime/CreateDynamicSubAgentThread.ts
+var CreateDynamicSubAgentThread_exports = {};
+module.exports = __toCommonJS(CreateDynamicSubAgentThread_exports);
+//# sourceMappingURL=CreateDynamicSubAgentThread.js.map

package/dist/core/runtime/CreateDynamicSubAgentThread.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/core/runtime/CreateDynamicSubAgentThread.ts"],"sourcesContent":["import type { AgentInfo, AgentParent } from '../events/schema';\nimport type { AgentDefinition } from './AgentDefinition';\nimport type { AgentThread } from './AgentThread';\n\nexport type CreateDynamicSubAgentThread = (input: {\n  parentDefinition: AgentDefinition;\n  request: AgentInfo;\n  threadId: string;\n  parent: AgentParent;\n  signal: AbortSignal;\n}) => Promise<AgentThread>;\n"],"mappings":";;;;;;;;;;;;;;;;AAAA;AAAA;","names":[]}

package/dist/core/runtime/CreateDynamicSubAgentThread.mjs

@@ -0,0 +1 @@
+//# sourceMappingURL=CreateDynamicSubAgentThread.mjs.map

package/dist/core/runtime/CreateDynamicSubAgentThread.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/core/runtime/DeferredTool.d.ts

@@ -0,0 +1,34 @@
+import type { CallToolRequest } from '@modelcontextprotocol/sdk/types.js';
+import type { Logger } from 'winston';
+import { InstructionBuilder } from '../InstructionBuilder';
+import type { InternalToolCallInfo } from '../llm/LLMTypes';
+import { type IToolSet } from '../mcp/IMCPServer';
+import { LocalToolMCP, type ToolDefinition } from '../mcp/LocalToolMCP';
+import type { AgentTracing } from '../tracing/AgentTracing';
+export declare const DEFERRED_TOOLS_SERVER_ID = "deferred-tools";
+export declare const LIST_TOOLS_NAME = "list_tools";
+export declare const GET_TOOL_INFO_NAME = "get_tool_info";
+export declare const GET_TOOL_OUTPUT_SCHEMA_NAME = "get_tool_output_schema";
+export declare class DeferredTool extends LocalToolMCP {
+    readonly name = "deferred-tools";
+    readonly displayName = "DeferredTools";
+    readonly description = "Deferred tool loading for user MCP servers";
+    private readonly serverMap;
+    private readonly tools;
+    private readonly logger;
+    constructor(servers: readonly IToolSet[], options: {
+        tracing: AgentTracing;
+        logger: Logger;
+    });
+    buildInstruction(builder: InstructionBuilder): void;
+    protected getTools(): ToolDefinition[];
+    toolCallInfo(params: CallToolRequest['params'], resolveUnderlyingTool?: boolean): Promise<InternalToolCallInfo>;
+    private resolveServer;
+    private resolveServerTools;
+    private buildTools;
+    private buildListTools;
+    private buildGetToolInfo;
+    private buildGetToolOutputSchema;
+    private buildCallTool;
+}
+//# sourceMappingURL=DeferredTool.d.ts.map

package/dist/core/runtime/DeferredTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"DeferredTool.d.ts","sourceRoot":"","sources":["../../../src/core/runtime/DeferredTool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oCAAoC,CAAC;AAE1E,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAGtC,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAC5D,OAAO,EAML,KAAK,QAAQ,EAGd,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAc,YAAY,EAAE,KAAK,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACpF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAG5D,eAAO,MAAM,wBAAwB,mBAAmB,CAAC;AAEzD,eAAO,MAAM,eAAe,eAAe,CAAC;AAC5C,eAAO,MAAM,kBAAkB,kBAAkB,CAAC;AAClD,eAAO,MAAM,2BAA2B,2BAA2B,CAAC;AAkCpE,qBAAa,YAAa,SAAQ,YAAY;IAC5C,QAAQ,CAAC,IAAI,oBAA4B;IACzC,QAAQ,CAAC,WAAW,mBAAmB;IACvC,SAAkB,WAAW,gDAAgD;IAE7E,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAwB;IAClD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAmB;IACzC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAEhC,YACE,OAAO,EAAE,SAAS,QAAQ,EAAE,EAC5B,OAAO,EAAE;QACP,OAAO,EAAE,YAAY,CAAC;QACtB,MAAM,EAAE,MAAM,CAAC;KAChB,EAMF;IAED,gBAAgB,CAAC,OAAO,EAAE,kBAAkB,GAAG,IAAI,CAqBlD;IAED,SAAS,CAAC,QAAQ,IAAI,cAAc,EAAE,CAErC;IAEc,YAAY,CACzB,MAAM,EAAE,eAAe,CAAC,QAAQ,CAAC,EACjC,qBAAqB,CAAC,EAAE,OAAO,GAC9B,OAAO,CAAC,oBAAoB,CAAC,CA8B/B;IAED,OAAO,CAAC,aAAa;YAIP,kBAAkB;IAiBhC,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,cAAc;IA8BtB,OAAO,CAAC,gBAAgB;IA+CxB,OAAO,CAAC,wBAAwB;IA8ChC,OAAO,CAAC,aAAa;CA+CtB"}