@theokit/sdk 2.22.0 → 2.23.0

package/dist/a2a/index.js

@@ -18423,15 +18423,33 @@ var MaxDelegationDepthError = class extends Error {
   maxDepth;
   code = "max_delegation_depth";
 };
-async function applyDelegationStart(spec, input) {
+async function applyDelegationStart(spec, input, iteration) {
   if (spec.onDelegationStart === void 0) return { input };
-  const decision = await spec.onDelegationStart({ input, name: spec.name });
+  const decision = await spec.onDelegationStart({ input, name: spec.name, iteration });
   if (decision === void 0) return { input };
   if (decision.proceed === false)
     return { reject: decision.rejectionReason ?? "(delegation rejected)" };
-  return { input: decision.modifiedInput ?? input };
+  return {
+    input: decision.modifiedInput ?? input,
+    ...decision.modifiedMaxSteps !== void 0 ? { maxSteps: decision.modifiedMaxSteps } : {}
+  };
+}
+async function collectChildToolResults(run) {
+  const lines = [];
+  for await (const event of run.stream()) {
+    if (event.type === "tool_call" && event.status === "completed") {
+      const rendered = typeof event.result === "string" ? event.result : JSON.stringify(event.result ?? null);
+      lines.push(`${event.name}: ${rendered}`);
+    }
+  }
+  if (lines.length === 0) return "";
+  return `
+
+<subagent-tool-results>
+${lines.join("\n")}
+</subagent-tool-results>`;
 }
-async function runChildAgent(spec, input, signal) {
+async function runChildAgent(spec, input, signal, maxSteps) {
   const { Agent: Agent2 } = await Promise.resolve().then(() => (init_agent(), agent_exports));
   const agent = await Agent2.create({
     ...spec.model ? { model: { id: spec.model } } : {},
@@ -18439,17 +18457,22 @@ async function runChildAgent(spec, input, signal) {
     tools: spec.tools ?? []
   });
   try {
-    const run = signal !== void 0 ? await agent.send(input, { signal }) : await agent.send(input);
+    const sendOptions = {
+      ...signal !== void 0 ? { signal } : {},
+      ...maxSteps !== void 0 ? { maxIterations: maxSteps } : {}
+    };
+    const run = Object.keys(sendOptions).length > 0 ? await agent.send(input, sendOptions) : await agent.send(input);
     const result = await run.wait();
-    return result.result ?? "(no response)";
+    const text = result.result ?? "(no response)";
+    return spec.includeToolResults === true ? text + await collectChildToolResults(run) : text;
   } finally {
     agent.dispose();
   }
 }
-async function notifyDelegationError(spec, input, error) {
+async function notifyDelegationError(spec, input, error, iteration) {
   if (spec.onDelegationComplete === void 0) return;
   try {
-    await spec.onDelegationComplete({ input, name: spec.name, error });
+    await spec.onDelegationComplete({ input, name: spec.name, error, iteration });
   } catch {
   }
 }
@@ -18464,9 +18487,9 @@ ${preamble}
 Task:
 ${input}`;
 }
-async function applyDelegationComplete(spec, input, result) {
+async function applyDelegationComplete(spec, input, result, iteration) {
   if (spec.onDelegationComplete === void 0) return result;
-  const completion = await spec.onDelegationComplete({ input, name: spec.name, result });
+  const completion = await spec.onDelegationComplete({ input, name: spec.name, result, iteration });
   return completion?.feedback !== void 0 ? result + completion.feedback : result;
 }
 function defineSubAgent(spec, _parentDepth = 0) {
@@ -18478,23 +18501,26 @@ function defineSubAgent(spec, _parentDepth = 0) {
   const inputSchema = z.object({
     input: z.string().describe("Task for the subagent")
   });
+  let iteration = 0;
   return {
     name: spec.name,
     description: spec.description,
     inputSchema,
     handler: async (rawInput, ctx) => {
       const { input: parsed } = inputSchema.parse(rawInput);
-      const start = await applyDelegationStart(spec, parsed);
+      iteration += 1;
+      const capturedIteration = iteration;
+      const start = await applyDelegationStart(spec, parsed, capturedIteration);
       if ("reject" in start) return start.reject;
       const input = applyMessageFilter(spec, start.input, ctx?.messages);
       let result;
       try {
-        result = await runChildAgent(spec, input, ctx?.signal);
+        result = await runChildAgent(spec, input, ctx?.signal, start.maxSteps);
       } catch (error) {
-        await notifyDelegationError(spec, input, error);
+        await notifyDelegationError(spec, input, error, capturedIteration);
         throw error;
       }
-      return applyDelegationComplete(spec, input, result);
+      return applyDelegationComplete(spec, input, result, capturedIteration);
     }
   };
 }