@iqai/adk 0.1.14 → 0.1.16

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @iqai/adk
 
+## 0.1.16
+
+### Patch Changes
+
+- 335fa8e: Updates createTool to use zod v3 instead of v4
+
+## 0.1.15
+
+### Patch Changes
+
+- d5d9750: update transfer agent tool to work
+
 ## 0.1.14
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1,10 +1,10 @@
 import { Part, Content, Blob, SpeechConfig, AudioTranscriptionConfig, RealtimeInputConfig, ProactivityConfig, FunctionDeclaration, GroundingMetadata, GenerateContentResponseUsageMetadata, GenerateContentConfig, Schema, LiveConnectConfig, GoogleGenAI, FunctionCall } from '@google/genai';
 export { Blob, Content, FunctionDeclaration, Schema as JSONSchema } from '@google/genai';
 import { LanguageModel } from 'ai';
-import * as z from 'zod/v4';
+import * as z from 'zod';
+import { z as z$1 } from 'zod';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { CreateMessageRequestSchema, CreateMessageResultSchema, Tool } from '@modelcontextprotocol/sdk/types.js';
-import { z as z$1 } from 'zod';
 import { Kysely, Generated } from 'kysely';
 import { StorageOptions } from '@google-cloud/storage';
 import { Tracer } from '@opentelemetry/api';
@@ -1304,6 +1304,18 @@ interface LlmAgentConfig<T extends BaseLlm = BaseLlm> {
      * Description of the agent
      */
     description: string;
+    /**
+     * Sub-agents that this agent can delegate to
+     */
+    subAgents?: BaseAgent[];
+    /**
+     * Callback or list of callbacks to be invoked before the agent run
+     */
+    beforeAgentCallback?: BeforeAgentCallback;
+    /**
+     * Callback or list of callbacks to be invoked after the agent run
+     */
+    afterAgentCallback?: AfterAgentCallback;
     /**
      * The LLM model to use
      * When not set, the agent will inherit the model from its ancestor
@@ -1877,6 +1889,10 @@ declare class TransferToAgentTool extends BaseTool {
      * Constructor for TransferToAgentTool
      */
     constructor();
+    /**
+     * Get the function declaration for the tool
+     */
+    getDeclaration(): FunctionDeclaration;
     /**
      * Execute the transfer to agent action
      */
@@ -3921,6 +3937,8 @@ interface AgentBuilderConfig {
     planner?: BasePlanner;
     codeExecutor?: BaseCodeExecutor;
     subAgents?: BaseAgent[];
+    beforeAgentCallback?: BeforeAgentCallback;
+    afterAgentCallback?: AfterAgentCallback;
     maxIterations?: number;
     nodes?: LangGraphNode[];
     rootNode?: string;
@@ -4080,6 +4098,24 @@ declare class AgentBuilder {
      * @returns This builder instance for chaining
      */
     withOutputKey(outputKey: string): this;
+    /**
+     * Add sub-agents to the agent
+     * @param subAgents Sub-agents to add to the agent
+     * @returns This builder instance for chaining
+     */
+    withSubAgents(subAgents: BaseAgent[]): this;
+    /**
+     * Set the before agent callback
+     * @param callback Callback to invoke before agent execution
+     * @returns This builder instance for chaining
+     */
+    withBeforeAgentCallback(callback: BeforeAgentCallback): this;
+    /**
+     * Set the after agent callback
+     * @param callback Callback to invoke after agent execution
+     * @returns This builder instance for chaining
+     */
+    withAfterAgentCallback(callback: AfterAgentCallback): this;
     /**
      * Configure as a sequential agent
      * @param subAgents Sub-agents to execute in sequence

package/dist/index.d.ts

@@ -1,10 +1,10 @@
 import { Part, Content, Blob, SpeechConfig, AudioTranscriptionConfig, RealtimeInputConfig, ProactivityConfig, FunctionDeclaration, GroundingMetadata, GenerateContentResponseUsageMetadata, GenerateContentConfig, Schema, LiveConnectConfig, GoogleGenAI, FunctionCall } from '@google/genai';
 export { Blob, Content, FunctionDeclaration, Schema as JSONSchema } from '@google/genai';
 import { LanguageModel } from 'ai';
-import * as z from 'zod/v4';
+import * as z from 'zod';
+import { z as z$1 } from 'zod';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { CreateMessageRequestSchema, CreateMessageResultSchema, Tool } from '@modelcontextprotocol/sdk/types.js';
-import { z as z$1 } from 'zod';
 import { Kysely, Generated } from 'kysely';
 import { StorageOptions } from '@google-cloud/storage';
 import { Tracer } from '@opentelemetry/api';
@@ -1304,6 +1304,18 @@ interface LlmAgentConfig<T extends BaseLlm = BaseLlm> {
      * Description of the agent
      */
     description: string;
+    /**
+     * Sub-agents that this agent can delegate to
+     */
+    subAgents?: BaseAgent[];
+    /**
+     * Callback or list of callbacks to be invoked before the agent run
+     */
+    beforeAgentCallback?: BeforeAgentCallback;
+    /**
+     * Callback or list of callbacks to be invoked after the agent run
+     */
+    afterAgentCallback?: AfterAgentCallback;
     /**
      * The LLM model to use
      * When not set, the agent will inherit the model from its ancestor
@@ -1877,6 +1889,10 @@ declare class TransferToAgentTool extends BaseTool {
      * Constructor for TransferToAgentTool
      */
     constructor();
+    /**
+     * Get the function declaration for the tool
+     */
+    getDeclaration(): FunctionDeclaration;
     /**
      * Execute the transfer to agent action
      */
@@ -3921,6 +3937,8 @@ interface AgentBuilderConfig {
     planner?: BasePlanner;
     codeExecutor?: BaseCodeExecutor;
     subAgents?: BaseAgent[];
+    beforeAgentCallback?: BeforeAgentCallback;
+    afterAgentCallback?: AfterAgentCallback;
     maxIterations?: number;
     nodes?: LangGraphNode[];
     rootNode?: string;
@@ -4080,6 +4098,24 @@ declare class AgentBuilder {
      * @returns This builder instance for chaining
      */
     withOutputKey(outputKey: string): this;
+    /**
+     * Add sub-agents to the agent
+     * @param subAgents Sub-agents to add to the agent
+     * @returns This builder instance for chaining
+     */
+    withSubAgents(subAgents: BaseAgent[]): this;
+    /**
+     * Set the before agent callback
+     * @param callback Callback to invoke before agent execution
+     * @returns This builder instance for chaining
+     */
+    withBeforeAgentCallback(callback: BeforeAgentCallback): this;
+    /**
+     * Set the after agent callback
+     * @param callback Callback to invoke after agent execution
+     * @returns This builder instance for chaining
+     */
+    withAfterAgentCallback(callback: AfterAgentCallback): this;
     /**
      * Configure as a sequential agent
      * @param subAgents Sub-agents to execute in sequence

package/dist/index.js

@@ -3855,7 +3855,8 @@ init_base_tool();
 
 // src/tools/base/create-tool.ts
 init_base_tool();
-var _v4 = require('zod/v4'); var z = _interopRequireWildcard(_v4);
+var _zod = require('zod'); var z = _interopRequireWildcard(_zod);
+var _zodtojsonschema = require('zod-to-json-schema');
 var CreatedTool = class extends BaseTool {
   
   
@@ -3885,7 +3886,7 @@ var CreatedTool = class extends BaseTool {
     } catch (error) {
       if (error instanceof z.ZodError) {
         return {
-          error: `Invalid arguments for ${this.name}: ${z.prettifyError(error)}`
+          error: `Invalid arguments for ${this.name}: ${error.message}`
         };
       }
       return {
@@ -3903,7 +3904,11 @@ var CreatedTool = class extends BaseTool {
    * Builds the function declaration from the Zod schema
    */
   buildDeclaration() {
-    const parameters = z.toJSONSchema(this.schema);
+    const rawParameters = _zodtojsonschema.zodToJsonSchema.call(void 0, this.schema, {
+      target: "jsonSchema7",
+      $refStrategy: "none"
+    });
+    const { $schema, ...parameters } = rawParameters;
     return {
       name: this.name,
       description: this.description,
@@ -4846,6 +4851,7 @@ var GetUserChoiceTool = (_class18 = class extends BaseTool {
 // src/tools/common/transfer-to-agent-tool.ts
 init_logger();
 init_base_tool();
+
 var TransferToAgentTool = (_class19 = class extends BaseTool {
   __init29() {this.logger = new Logger({ name: "TransferToAgentTool" })}
   /**
@@ -4854,9 +4860,28 @@ var TransferToAgentTool = (_class19 = class extends BaseTool {
   constructor() {
     super({
       name: "transfer_to_agent",
-      description: "Transfer the question to another agent."
+      description: "Transfer the question to another agent when it's more suitable to answer the user's question according to the agent's description. Use this function when you determine that another agent in the system would be better equipped to handle the user's request based on their specialized capabilities and expertise areas."
     });_class19.prototype.__init29.call(this);;
   }
+  /**
+   * Get the function declaration for the tool
+   */
+  getDeclaration() {
+    return {
+      name: this.name,
+      description: this.description,
+      parameters: {
+        type: _genai.Type.OBJECT,
+        properties: {
+          agent_name: {
+            type: _genai.Type.STRING,
+            description: "The name of the agent to transfer control to"
+          }
+        },
+        required: ["agent_name"]
+      }
+    };
+  }
   /**
    * Execute the transfer to agent action
    */
@@ -8464,12 +8489,12 @@ var SingleFlow = class extends BaseLlmFlow {
 };
 
 // src/flows/llm-flows/agent-transfer.ts
+var _dedent = require('dedent'); var _dedent2 = _interopRequireDefault(_dedent);
 var AgentTransferLlmRequestProcessor = class extends BaseLlmRequestProcessor {
   /**
    * Processes agent transfer by adding transfer instructions and tools
    * if the agent has transfer targets available
    */
-  // eslint-disable-next-line @typescript-eslint/require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
     if (!("subAgents" in agent) || typeof agent.subAgents !== "object") {
@@ -8487,60 +8512,61 @@ var AgentTransferLlmRequestProcessor = class extends BaseLlmRequestProcessor {
     const transferToAgentTool = new TransferToAgentTool();
     const toolContext = new ToolContext(invocationContext);
     await transferToAgentTool.processLlmRequest(toolContext, llmRequest);
-    for await (const _ of []) {
-      yield _;
+    const shouldYield = false;
+    if (shouldYield) {
+      yield {};
     }
   }
 };
 function buildTargetAgentsInfo(targetAgent) {
-  return `
-Agent name: ${targetAgent.name}
-Agent description: ${targetAgent.description}
-`;
+  return _dedent2.default`
+		Agent name: ${targetAgent.name}
+		Agent description: ${targetAgent.description}
+	`;
 }
 function buildTargetAgentsInstructions(agent, targetAgents) {
   const lineBreak = "\n";
   const transferFunctionName = "transfer_to_agent";
-  let instructions = `
-You have a list of other agents to transfer to:
+  let instructions = _dedent2.default`
+		You have a list of other agents to transfer to:
 
-${targetAgents.map((targetAgent) => buildTargetAgentsInfo(targetAgent)).join(lineBreak)}
+		${targetAgents.map((targetAgent) => buildTargetAgentsInfo(targetAgent)).join(lineBreak)}
 
-If you are the best to answer the question according to your description, you
-can answer it.
+		If you are the best to answer the question according to your description, you
+		can answer it.
 
-If another agent is better for answering the question according to its
-description, call \`${transferFunctionName}\` function to transfer the
-question to that agent. When transferring, do not generate any text other than
-the function call.
+		If another agent is better for answering the question according to its
+		description, call \`${transferFunctionName}\` function to transfer the
+		question to that agent. When transferring, do not generate any text other than
+		the function call.
 `;
   if (agent.parentAgent && !agent.disallowTransferToParent) {
-    instructions += `
-Your parent agent is ${agent.parentAgent.name}. If neither the other agents nor
-you are best for answering the question according to the descriptions, transfer
-to your parent agent.
-`;
+    instructions += _dedent2.default`
+			Your parent agent is ${agent.parentAgent.name}. If neither the other agents nor
+			you are best for answering the question according to the descriptions, transfer
+			to your parent agent.
+		`;
   }
   return instructions;
 }
 function getTransferTargets(agent) {
-  const result = [];
+  const targets = [];
   if (agent.subAgents && Array.isArray(agent.subAgents)) {
-    result.push(...agent.subAgents);
+    targets.push(...agent.subAgents);
   }
   if (!agent.parentAgent || !("subAgents" in agent.parentAgent)) {
-    return result;
+    return targets;
   }
   if (!agent.disallowTransferToParent) {
-    result.push(agent.parentAgent);
+    targets.push(agent.parentAgent);
   }
   if (!agent.disallowTransferToPeers && agent.parentAgent.subAgents) {
     const peerAgents = agent.parentAgent.subAgents.filter(
       (peerAgent) => peerAgent.name !== agent.name
     );
-    result.push(...peerAgents);
+    targets.push(...peerAgents);
   }
-  return result;
+  return targets;
 }
 var requestProcessor8 = new AgentTransferLlmRequestProcessor();
 
@@ -8642,7 +8668,10 @@ var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
   constructor(config) {
     super({
       name: config.name,
-      description: config.description
+      description: config.description,
+      subAgents: config.subAgents,
+      beforeAgentCallback: config.beforeAgentCallback,
+      afterAgentCallback: config.afterAgentCallback
     });_class26.prototype.__init46.call(this);;
     this.model = config.model || "";
     this.instruction = config.instruction || "";
@@ -10119,6 +10148,33 @@ var AgentBuilder = (_class32 = class _AgentBuilder {
     this.config.outputKey = outputKey;
     return this;
   }
+  /**
+   * Add sub-agents to the agent
+   * @param subAgents Sub-agents to add to the agent
+   * @returns This builder instance for chaining
+   */
+  withSubAgents(subAgents) {
+    this.config.subAgents = subAgents;
+    return this;
+  }
+  /**
+   * Set the before agent callback
+   * @param callback Callback to invoke before agent execution
+   * @returns This builder instance for chaining
+   */
+  withBeforeAgentCallback(callback) {
+    this.config.beforeAgentCallback = callback;
+    return this;
+  }
+  /**
+   * Set the after agent callback
+   * @param callback Callback to invoke after agent execution
+   * @returns This builder instance for chaining
+   */
+  withAfterAgentCallback(callback) {
+    this.config.afterAgentCallback = callback;
+    return this;
+  }
   /**
    * Configure as a sequential agent
    * @param subAgents Sub-agents to execute in sequence
@@ -10290,6 +10346,9 @@ var AgentBuilder = (_class32 = class _AgentBuilder {
           tools: this.config.tools,
           planner: this.config.planner,
           codeExecutor: this.config.codeExecutor,
+          subAgents: this.config.subAgents,
+          beforeAgentCallback: this.config.beforeAgentCallback,
+          afterAgentCallback: this.config.afterAgentCallback,
           memoryService: this.memoryService,
           artifactService: this.artifactService,
           outputKey: this.config.outputKey,
@@ -11122,7 +11181,7 @@ var DatabaseSessionService = (_class33 = class extends BaseSessionService {
 }, _class33);
 
 // src/sessions/database-factories.ts
-var _dedent = require('dedent'); var _dedent2 = _interopRequireDefault(_dedent);
+
 
 function createDependencyError(packageName, dbType) {
   return new Error(

package/dist/index.mjs

@@ -3855,7 +3855,8 @@ init_base_tool();
 
 // src/tools/base/create-tool.ts
 init_base_tool();
-import * as z from "zod/v4";
+import * as z from "zod";
+import { zodToJsonSchema } from "zod-to-json-schema";
 var CreatedTool = class extends BaseTool {
   func;
   schema;
@@ -3885,7 +3886,7 @@ var CreatedTool = class extends BaseTool {
     } catch (error) {
       if (error instanceof z.ZodError) {
         return {
-          error: `Invalid arguments for ${this.name}: ${z.prettifyError(error)}`
+          error: `Invalid arguments for ${this.name}: ${error.message}`
         };
       }
       return {
@@ -3903,7 +3904,11 @@ var CreatedTool = class extends BaseTool {
    * Builds the function declaration from the Zod schema
    */
   buildDeclaration() {
-    const parameters = z.toJSONSchema(this.schema);
+    const rawParameters = zodToJsonSchema(this.schema, {
+      target: "jsonSchema7",
+      $refStrategy: "none"
+    });
+    const { $schema, ...parameters } = rawParameters;
     return {
       name: this.name,
       description: this.description,
@@ -4846,6 +4851,7 @@ var GetUserChoiceTool = class extends BaseTool {
 // src/tools/common/transfer-to-agent-tool.ts
 init_logger();
 init_base_tool();
+import { Type as Type8 } from "@google/genai";
 var TransferToAgentTool = class extends BaseTool {
   logger = new Logger({ name: "TransferToAgentTool" });
   /**
@@ -4854,9 +4860,28 @@ var TransferToAgentTool = class extends BaseTool {
   constructor() {
     super({
       name: "transfer_to_agent",
-      description: "Transfer the question to another agent."
+      description: "Transfer the question to another agent when it's more suitable to answer the user's question according to the agent's description. Use this function when you determine that another agent in the system would be better equipped to handle the user's request based on their specialized capabilities and expertise areas."
     });
   }
+  /**
+   * Get the function declaration for the tool
+   */
+  getDeclaration() {
+    return {
+      name: this.name,
+      description: this.description,
+      parameters: {
+        type: Type8.OBJECT,
+        properties: {
+          agent_name: {
+            type: Type8.STRING,
+            description: "The name of the agent to transfer control to"
+          }
+        },
+        required: ["agent_name"]
+      }
+    };
+  }
   /**
    * Execute the transfer to agent action
    */
@@ -4869,7 +4894,7 @@ var TransferToAgentTool = class extends BaseTool {
 // src/tools/common/load-memory-tool.ts
 init_logger();
 init_base_tool();
-import { Type as Type8 } from "@google/genai";
+import { Type as Type9 } from "@google/genai";
 var LoadMemoryTool = class extends BaseTool {
   logger = new Logger({ name: "LoadMemoryTool" });
   /**
@@ -4889,10 +4914,10 @@ var LoadMemoryTool = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: Type8.OBJECT,
+        type: Type9.OBJECT,
         properties: {
           query: {
-            type: Type8.STRING,
+            type: Type9.STRING,
             description: "The query to load memories for"
           }
         },
@@ -4923,7 +4948,7 @@ var LoadMemoryTool = class extends BaseTool {
 
 // src/tools/common/load-artifacts-tool.ts
 init_base_tool();
-import { Type as Type9 } from "@google/genai";
+import { Type as Type10 } from "@google/genai";
 var LoadArtifactsTool = class extends BaseTool {
   constructor() {
     super({
@@ -4939,12 +4964,12 @@ var LoadArtifactsTool = class extends BaseTool {
       name: this.name,
       description: this.description,
       parameters: {
-        type: Type9.OBJECT,
+        type: Type10.OBJECT,
         properties: {
           artifact_names: {
-            type: Type9.ARRAY,
+            type: Type10.ARRAY,
             items: {
-              type: Type9.STRING
+              type: Type10.STRING
             },
             description: "List of artifact names to load"
           }
@@ -5563,7 +5588,7 @@ init_logger();
 init_base_tool();
 
 // src/tools/mcp/schema-conversion.ts
-import { Type as Type10 } from "@google/genai";
+import { Type as Type11 } from "@google/genai";
 function adkToMcpToolType(tool) {
   const declaration = tool.getDeclaration();
   const params = declarationToJsonSchema(declaration);
@@ -5592,13 +5617,13 @@ function jsonSchemaToDeclaration(name, description, schema) {
       parameters = schema;
     } else {
       parameters = {
-        type: Type10.OBJECT,
+        type: Type11.OBJECT,
         properties: schema
       };
     }
   } else {
     parameters = {
-      type: Type10.OBJECT,
+      type: Type11.OBJECT,
       properties: {}
     };
   }
@@ -5610,7 +5635,7 @@ function jsonSchemaToDeclaration(name, description, schema) {
 }
 function normalizeJsonSchema(schema) {
   if (!schema) {
-    return { type: Type10.OBJECT, properties: {} };
+    return { type: Type11.OBJECT, properties: {} };
   }
   const normalizedSchema = { ...schema };
   if (!normalizedSchema.type) {
@@ -5627,39 +5652,39 @@ function normalizeJsonSchema(schema) {
     case "integer":
       return normalizeNumberSchema(normalizedSchema);
     case "boolean":
-      return { type: Type10.BOOLEAN };
+      return { type: Type11.BOOLEAN };
     case "null":
-      return { type: Type10.NULL };
+      return { type: Type11.NULL };
     default:
       return normalizedSchema;
   }
 }
 function determineSchemaType(schema) {
   if (schema.properties || schema.required || schema.additionalProperties !== void 0) {
-    return Type10.OBJECT;
+    return Type11.OBJECT;
   }
   if (schema.items) {
-    return Type10.ARRAY;
+    return Type11.ARRAY;
   }
   if (schema.enum !== void 0) {
-    if (schema.enum.length === 0) return Type10.STRING;
+    if (schema.enum.length === 0) return Type11.STRING;
     const firstItem = schema.enum[0];
-    if (typeof firstItem === "string") return Type10.STRING;
-    if (typeof firstItem === "number") return Type10.NUMBER;
-    if (typeof firstItem === "boolean") return Type10.BOOLEAN;
-    return Type10.STRING;
+    if (typeof firstItem === "string") return Type11.STRING;
+    if (typeof firstItem === "number") return Type11.NUMBER;
+    if (typeof firstItem === "boolean") return Type11.BOOLEAN;
+    return Type11.STRING;
   }
   if (schema.minLength !== void 0 || schema.maxLength !== void 0 || schema.pattern) {
-    return Type10.STRING;
+    return Type11.STRING;
   }
   if (schema.minimum !== void 0 || schema.maximum !== void 0 || schema.exclusiveMinimum !== void 0 || schema.exclusiveMaximum !== void 0) {
-    return schema.multipleOf === void 0 || schema.multipleOf % 1 === 0 ? Type10.INTEGER : Type10.NUMBER;
+    return schema.multipleOf === void 0 || schema.multipleOf % 1 === 0 ? Type11.INTEGER : Type11.NUMBER;
   }
-  return Type10.OBJECT;
+  return Type11.OBJECT;
 }
 function normalizeObjectSchema(schema) {
   const normalizedSchema = {
-    type: Type10.OBJECT,
+    type: Type11.OBJECT,
     properties: {}
   };
   if (schema.properties) {
@@ -5677,7 +5702,7 @@ function normalizeObjectSchema(schema) {
 }
 function normalizeArraySchema(schema) {
   const normalizedSchema = {
-    type: Type10.ARRAY
+    type: Type11.ARRAY
   };
   if (schema.items) {
     normalizedSchema.items = normalizeJsonSchema(
@@ -5694,7 +5719,7 @@ function normalizeArraySchema(schema) {
 }
 function normalizeStringSchema(schema) {
   const normalizedSchema = {
-    type: Type10.STRING
+    type: Type11.STRING
   };
   if (schema.minLength !== void 0)
     normalizedSchema.minLength = schema.minLength;
@@ -5727,7 +5752,7 @@ function mcpSchemaToParameters(mcpTool) {
   }
   if (!schema) {
     return {
-      type: Type10.OBJECT,
+      type: Type11.OBJECT,
       properties: {}
     };
   }
@@ -8464,12 +8489,12 @@ var SingleFlow = class extends BaseLlmFlow {
 };
 
 // src/flows/llm-flows/agent-transfer.ts
+import dedent from "dedent";
 var AgentTransferLlmRequestProcessor = class extends BaseLlmRequestProcessor {
   /**
    * Processes agent transfer by adding transfer instructions and tools
    * if the agent has transfer targets available
    */
-  // eslint-disable-next-line @typescript-eslint/require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
     if (!("subAgents" in agent) || typeof agent.subAgents !== "object") {
@@ -8487,60 +8512,61 @@ var AgentTransferLlmRequestProcessor = class extends BaseLlmRequestProcessor {
     const transferToAgentTool = new TransferToAgentTool();
     const toolContext = new ToolContext(invocationContext);
     await transferToAgentTool.processLlmRequest(toolContext, llmRequest);
-    for await (const _ of []) {
-      yield _;
+    const shouldYield = false;
+    if (shouldYield) {
+      yield {};
     }
   }
 };
 function buildTargetAgentsInfo(targetAgent) {
-  return `
-Agent name: ${targetAgent.name}
-Agent description: ${targetAgent.description}
-`;
+  return dedent`
+		Agent name: ${targetAgent.name}
+		Agent description: ${targetAgent.description}
+	`;
 }
 function buildTargetAgentsInstructions(agent, targetAgents) {
   const lineBreak = "\n";
   const transferFunctionName = "transfer_to_agent";
-  let instructions = `
-You have a list of other agents to transfer to:
+  let instructions = dedent`
+		You have a list of other agents to transfer to:
 
-${targetAgents.map((targetAgent) => buildTargetAgentsInfo(targetAgent)).join(lineBreak)}
+		${targetAgents.map((targetAgent) => buildTargetAgentsInfo(targetAgent)).join(lineBreak)}
 
-If you are the best to answer the question according to your description, you
-can answer it.
+		If you are the best to answer the question according to your description, you
+		can answer it.
 
-If another agent is better for answering the question according to its
-description, call \`${transferFunctionName}\` function to transfer the
-question to that agent. When transferring, do not generate any text other than
-the function call.
+		If another agent is better for answering the question according to its
+		description, call \`${transferFunctionName}\` function to transfer the
+		question to that agent. When transferring, do not generate any text other than
+		the function call.
 `;
   if (agent.parentAgent && !agent.disallowTransferToParent) {
-    instructions += `
-Your parent agent is ${agent.parentAgent.name}. If neither the other agents nor
-you are best for answering the question according to the descriptions, transfer
-to your parent agent.
-`;
+    instructions += dedent`
+			Your parent agent is ${agent.parentAgent.name}. If neither the other agents nor
+			you are best for answering the question according to the descriptions, transfer
+			to your parent agent.
+		`;
   }
   return instructions;
 }
 function getTransferTargets(agent) {
-  const result = [];
+  const targets = [];
   if (agent.subAgents && Array.isArray(agent.subAgents)) {
-    result.push(...agent.subAgents);
+    targets.push(...agent.subAgents);
   }
   if (!agent.parentAgent || !("subAgents" in agent.parentAgent)) {
-    return result;
+    return targets;
   }
   if (!agent.disallowTransferToParent) {
-    result.push(agent.parentAgent);
+    targets.push(agent.parentAgent);
   }
   if (!agent.disallowTransferToPeers && agent.parentAgent.subAgents) {
     const peerAgents = agent.parentAgent.subAgents.filter(
       (peerAgent) => peerAgent.name !== agent.name
     );
-    result.push(...peerAgents);
+    targets.push(...peerAgents);
   }
-  return result;
+  return targets;
 }
 var requestProcessor8 = new AgentTransferLlmRequestProcessor();
 
@@ -8642,7 +8668,10 @@ var LlmAgent = class _LlmAgent extends BaseAgent {
   constructor(config) {
     super({
       name: config.name,
-      description: config.description
+      description: config.description,
+      subAgents: config.subAgents,
+      beforeAgentCallback: config.beforeAgentCallback,
+      afterAgentCallback: config.afterAgentCallback
     });
     this.model = config.model || "";
     this.instruction = config.instruction || "";
@@ -10119,6 +10148,33 @@ var AgentBuilder = class _AgentBuilder {
     this.config.outputKey = outputKey;
     return this;
   }
+  /**
+   * Add sub-agents to the agent
+   * @param subAgents Sub-agents to add to the agent
+   * @returns This builder instance for chaining
+   */
+  withSubAgents(subAgents) {
+    this.config.subAgents = subAgents;
+    return this;
+  }
+  /**
+   * Set the before agent callback
+   * @param callback Callback to invoke before agent execution
+   * @returns This builder instance for chaining
+   */
+  withBeforeAgentCallback(callback) {
+    this.config.beforeAgentCallback = callback;
+    return this;
+  }
+  /**
+   * Set the after agent callback
+   * @param callback Callback to invoke after agent execution
+   * @returns This builder instance for chaining
+   */
+  withAfterAgentCallback(callback) {
+    this.config.afterAgentCallback = callback;
+    return this;
+  }
   /**
    * Configure as a sequential agent
    * @param subAgents Sub-agents to execute in sequence
@@ -10290,6 +10346,9 @@ var AgentBuilder = class _AgentBuilder {
           tools: this.config.tools,
           planner: this.config.planner,
           codeExecutor: this.config.codeExecutor,
+          subAgents: this.config.subAgents,
+          beforeAgentCallback: this.config.beforeAgentCallback,
+          afterAgentCallback: this.config.afterAgentCallback,
           memoryService: this.memoryService,
           artifactService: this.artifactService,
           outputKey: this.config.outputKey,
@@ -11122,11 +11181,11 @@ var DatabaseSessionService = class extends BaseSessionService {
 };
 
 // src/sessions/database-factories.ts
-import dedent from "dedent";
+import dedent2 from "dedent";
 import { Kysely, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
 function createDependencyError(packageName, dbType) {
   return new Error(
-    dedent`
+    dedent2`
 		Missing required peer dependency: ${packageName}
 		To use ${dbType} sessions, install the required package:
 			npm install ${packageName}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@iqai/adk",
-  "version": "0.1.14",
+  "version": "0.1.16",
   "description": "Agent Development Kit for TypeScript with multi-provider LLM support",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -44,7 +44,8 @@
     "kysely": "^0.28.2",
     "openai": "^4.93.0",
     "uuid": "^11.1.0",
-    "zod": "^3.25.67"
+    "zod": "^3.25.67",
+    "zod-to-json-schema": "^3.24.6"
   },
   "peerDependencies": {
     "better-sqlite3": "^11.10.0",