@iqai/adk 0.1.20 → 0.1.21

package/dist/index.js

@@ -1,4 +1,4 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class; var _class2; var _class3; var _class4; var _class5; var _class6; var _class7; var _class8; var _class9; var _class10; var _class11; var _class12; var _class13; var _class14; var _class15; var _class16; var _class17; var _class18; var _class19; var _class20; var _class21; var _class22; var _class23; var _class24; var _class25; var _class26; var _class27; var _class28; var _class29; var _class30; var _class31; var _class32; var _class33;var __defProp = Object.defineProperty;
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; } var _class; var _class2; var _class3; var _class4; var _class5; var _class6; var _class7; var _class8; var _class9; var _class10; var _class11; var _class12; var _class13; var _class14; var _class15; var _class16; var _class17; var _class18; var _class19; var _class20; var _class21; var _class22; var _class23; var _class24; var _class25; var _class26; var _class27; var _class28; var _class29; var _class30; var _class31; var _class32; var _class33; var _class34;var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
@@ -1353,7 +1353,7 @@ var BaseLlm = (_class5 = class {
    * @param llmRequest LlmRequest, the request to send to the LLM.
    * @returns BaseLLMConnection, the connection to the LLM.
    */
-  connect(llmRequest) {
+  connect(_llmRequest) {
     throw new Error(`Live connection is not supported for ${this.model}.`);
   }
 }, _class5);
@@ -8088,6 +8088,9 @@ var IdentityLlmRequestProcessor = class extends BaseLlmRequestProcessor {
 };
 var requestProcessor5 = new IdentityLlmRequestProcessor();
 
+// src/flows/llm-flows/instructions.ts
+
+
 // src/utils/instructions-utils.ts
 async function injectSessionState(template, readonlyContext) {
   const invocationContext = readonlyContext._invocationContext;
@@ -8202,6 +8205,22 @@ var InstructionsLlmRequestProcessor = class extends BaseLlmRequestProcessor {
       }
       llmRequest.appendInstructions([instruction]);
     }
+    if (agent.outputSchema) {
+      try {
+        const raw = _zodtojsonschema.zodToJsonSchema.call(void 0, agent.outputSchema, {
+          target: "jsonSchema7",
+          $refStrategy: "none"
+        });
+        const { $schema, ...json } = raw || {};
+        llmRequest.appendInstructions([
+          "You must respond with application/json that validates against this JSON Schema:",
+          "```json",
+          JSON.stringify(json, null, 2),
+          "```"
+        ]);
+      } catch (e5) {
+      }
+    }
     for await (const _ of []) {
       yield _;
     }
@@ -8493,12 +8512,87 @@ function removeThoughtFromRequest(llmRequest) {
 var requestProcessor7 = new NlPlanningRequestProcessor();
 var responseProcessor2 = new NlPlanningResponseProcessor();
 
+// src/flows/llm-flows/output-schema.ts
+init_logger();
+var OutputSchemaResponseProcessor = (_class26 = class extends BaseLlmResponseProcessor {constructor(...args2) { super(...args2); _class26.prototype.__init46.call(this); }
+  __init46() {this.logger = new Logger({ name: "OutputSchemaResponseProcessor" })}
+  async *runAsync(invocationContext, llmResponse) {
+    if (!llmResponse || !llmResponse.content || !llmResponse.content.parts || llmResponse.content.parts.length === 0) {
+      return;
+    }
+    const agent = invocationContext.agent;
+    if (!("outputSchema" in agent) || !agent.outputSchema) {
+      return;
+    }
+    let textContent = llmResponse.content.parts.map((part) => {
+      if (part && typeof part === "object" && "text" in part) {
+        return part.text || "";
+      }
+      return "";
+    }).join("");
+    if (!textContent.trim()) {
+      return;
+    }
+    try {
+      const parsed = JSON.parse(textContent);
+      const validated = agent.outputSchema.parse(parsed);
+      textContent = JSON.stringify(validated, null, 2);
+      llmResponse.content.parts = llmResponse.content.parts.map((part) => {
+        if (part && typeof part === "object" && "text" in part) {
+          return {
+            ...part,
+            text: textContent
+          };
+        }
+        return part;
+      });
+      this.logger.debug("Output schema validation successful", {
+        agent: agent.name,
+        originalLength: textContent.length,
+        validatedKeys: Object.keys(validated)
+      });
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      const detailedError = `Output schema validation failed for agent '${agent.name}': ${errorMessage}`;
+      this.logger.error(detailedError, {
+        agent: agent.name,
+        responseContent: textContent.substring(0, 200) + (textContent.length > 200 ? "..." : ""),
+        error: errorMessage
+      });
+      llmResponse.errorCode = "OUTPUT_SCHEMA_VALIDATION_FAILED";
+      llmResponse.errorMessage = detailedError;
+      llmResponse.error = new Error(detailedError);
+      const errorEvent = new Event({
+        id: Event.newId(),
+        invocationId: invocationContext.invocationId,
+        author: agent.name,
+        branch: invocationContext.branch,
+        content: {
+          role: "assistant",
+          parts: [
+            {
+              text: `Error: ${detailedError}`
+            }
+          ]
+        }
+      });
+      errorEvent.errorCode = "OUTPUT_SCHEMA_VALIDATION_FAILED";
+      errorEvent.errorMessage = detailedError;
+      errorEvent.error = new Error(detailedError);
+      yield errorEvent;
+    }
+  }
+}, _class26);
+var responseProcessor3 = new OutputSchemaResponseProcessor();
+
 // src/flows/llm-flows/shared-memory.ts
 var SharedMemoryRequestProcessor = class extends BaseLlmRequestProcessor {
   async *runAsync(invocationContext, llmRequest) {
     const memoryService = invocationContext.memoryService;
     if (!memoryService) return;
-    const lastUserEvent = invocationContext.session.events.findLast((e) => e.author === "user" && _optionalChain([e, 'access', _230 => _230.content, 'optionalAccess', _231 => _231.parts, 'optionalAccess', _232 => _232.length]));
+    const lastUserEvent = invocationContext.session.events.findLast(
+      (e) => e.author === "user" && _optionalChain([e, 'access', _230 => _230.content, 'optionalAccess', _231 => _231.parts, 'optionalAccess', _232 => _232.length])
+    );
     if (!lastUserEvent) return;
     const query = (_nullishCoalesce(lastUserEvent.content.parts, () => ( []))).map((p) => p.text || "").join(" ");
     const results = await memoryService.searchMemory({
@@ -8557,8 +8651,10 @@ var SingleFlow = class extends BaseLlmFlow {
     this.responseProcessors.push(
       responseProcessor2,
       // Phase 5: NL Planning
+      responseProcessor3,
+      // Phase 6: Output Schema validation and parsing - validates response against agent's output schema
       responseProcessor
-      // Phase 5: Code Execution (placeholder)
+      // Phase 7: Code Execution (placeholder)
     );
     this.logger.debug("SingleFlow initialized with processors");
   }
@@ -8660,7 +8756,7 @@ var AutoFlow = class extends SingleFlow {
 
 // src/agents/llm-agent.ts
 init_function_tool();
-var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
+var LlmAgent = (_class27 = class _LlmAgent extends BaseAgent {
   /**
    * The model to use for the agent
    * When not set, the agent will inherit the model from its ancestor
@@ -8731,13 +8827,27 @@ var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
    * The input schema when agent is used as a tool
    */
   
-  // Schema type - depends on specific implementation
   /**
    * The output schema when agent replies
    */
   
-  // Schema type - depends on specific implementation
-  __init46() {this.logger = new Logger({ name: "LlmAgent" })}
+  /**
+   * Callback or list of callbacks to be called before calling the LLM
+   */
+  
+  /**
+   * Callback or list of callbacks to be called after calling the LLM
+   */
+  
+  /**
+   * Callback or list of callbacks to be called before calling a tool
+   */
+  
+  /**
+   * Callback or list of callbacks to be called after calling a tool
+   */
+  
+  __init47() {this.logger = new Logger({ name: "LlmAgent" })}
   /**
    * Constructor for LlmAgent
    */
@@ -8748,7 +8858,7 @@ var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
       subAgents: config.subAgents,
       beforeAgentCallback: config.beforeAgentCallback,
       afterAgentCallback: config.afterAgentCallback
-    });_class26.prototype.__init46.call(this);;
+    });_class27.prototype.__init47.call(this);;
     this.model = config.model || "";
     this.instruction = config.instruction || "";
     this.globalInstruction = config.globalInstruction || "";
@@ -8767,6 +8877,11 @@ var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
     this.generateContentConfig = config.generateContentConfig;
     this.inputSchema = config.inputSchema;
     this.outputSchema = config.outputSchema;
+    this.beforeModelCallback = config.beforeModelCallback;
+    this.afterModelCallback = config.afterModelCallback;
+    this.beforeToolCallback = config.beforeToolCallback;
+    this.afterToolCallback = config.afterToolCallback;
+    this.validateOutputSchemaConfig();
   }
   /**
    * The resolved model field as BaseLLM
@@ -8783,13 +8898,15 @@ var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
       return new AiSdkLlm(this.model);
     }
     let ancestorAgent = this.parentAgent;
-    while (ancestorAgent !== null) {
+    while (ancestorAgent !== null && ancestorAgent !== void 0) {
       if (ancestorAgent instanceof _LlmAgent) {
         return ancestorAgent.canonicalModel;
       }
       ancestorAgent = ancestorAgent.parentAgent;
     }
-    throw new Error(`No model found for ${this.name}.`);
+    throw new Error(
+      `No model found for agent "${this.name}". Please specify a model directly on this agent using the 'model' property`
+    );
   }
   /**
    * The resolved instruction field to construct instruction for this agent
@@ -8829,6 +8946,80 @@ var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
     }
     return resolvedTools;
   }
+  /**
+   * Gets the canonical before model callbacks as an array
+   */
+  get canonicalBeforeModelCallbacks() {
+    if (!this.beforeModelCallback) {
+      return [];
+    }
+    if (Array.isArray(this.beforeModelCallback)) {
+      return this.beforeModelCallback;
+    }
+    return [this.beforeModelCallback];
+  }
+  /**
+   * Gets the canonical after model callbacks as an array
+   */
+  get canonicalAfterModelCallbacks() {
+    if (!this.afterModelCallback) {
+      return [];
+    }
+    if (Array.isArray(this.afterModelCallback)) {
+      return this.afterModelCallback;
+    }
+    return [this.afterModelCallback];
+  }
+  /**
+   * Gets the canonical before tool callbacks as an array
+   */
+  get canonicalBeforeToolCallbacks() {
+    if (!this.beforeToolCallback) {
+      return [];
+    }
+    if (Array.isArray(this.beforeToolCallback)) {
+      return this.beforeToolCallback;
+    }
+    return [this.beforeToolCallback];
+  }
+  /**
+   * Gets the canonical after tool callbacks as an array
+   */
+  get canonicalAfterToolCallbacks() {
+    if (!this.afterToolCallback) {
+      return [];
+    }
+    if (Array.isArray(this.afterToolCallback)) {
+      return this.afterToolCallback;
+    }
+    return [this.afterToolCallback];
+  }
+  /**
+   * Validates output schema configuration
+   * This matches the Python implementation's __check_output_schema
+   */
+  validateOutputSchemaConfig() {
+    if (!this.outputSchema) {
+      return;
+    }
+    if (!this.disallowTransferToParent || !this.disallowTransferToPeers) {
+      this.logger.warn(
+        `Invalid config for agent ${this.name}: output_schema cannot co-exist with agent transfer configurations. Setting disallow_transfer_to_parent=true, disallow_transfer_to_peers=true`
+      );
+      this.disallowTransferToParent = true;
+      this.disallowTransferToPeers = true;
+    }
+    if (this.subAgents && this.subAgents.length > 0) {
+      throw new Error(
+        `Invalid config for agent ${this.name}: if output_schema is set, sub_agents must be empty to disable agent transfer.`
+      );
+    }
+    if (this.tools && this.tools.length > 0) {
+      throw new Error(
+        `Invalid config for agent ${this.name}: if output_schema is set, tools must be empty`
+      );
+    }
+  }
   /**
    * Gets the appropriate LLM flow for this agent
    * This matches the Python implementation's _llm_flow property
@@ -8844,8 +9035,28 @@ var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
    * This matches the Python implementation's __maybe_save_output_to_state
    */
   maybeSaveOutputToState(event) {
+    if (event.author !== this.name) {
+      this.logger.debug(
+        `Skipping output save for agent ${this.name}: event authored by ${event.author}`
+      );
+      return;
+    }
     if (this.outputKey && event.isFinalResponse() && _optionalChain([event, 'access', _238 => _238.content, 'optionalAccess', _239 => _239.parts])) {
-      const result = event.content.parts.map((part) => part.text || "").join("");
+      let result = event.content.parts.map((part) => part.text || "").join("");
+      if (this.outputSchema) {
+        if (!result.trim()) {
+          return;
+        }
+        try {
+          const parsed = JSON.parse(result);
+          result = this.outputSchema.parse(parsed);
+        } catch (error) {
+          this.logger.error("Failed to validate output with schema:", error);
+          throw new Error(
+            `Output validation failed: ${error instanceof Error ? error.message : String(error)}`
+          );
+        }
+      }
       if (result) {
         if (!event.actions.stateDelta) {
           event.actions.stateDelta = {};
@@ -8884,7 +9095,7 @@ var LlmAgent = (_class26 = class _LlmAgent extends BaseAgent {
       yield errorEvent;
     }
   }
-}, _class26);
+}, _class27);
 
 // src/agents/sequential-agent.ts
 var SequentialAgent = class extends BaseAgent {
@@ -9076,7 +9287,7 @@ var LoopAgent = class extends BaseAgent {
 
 // src/agents/lang-graph-agent.ts
 init_logger();
-var LangGraphAgent = (_class27 = class extends BaseAgent {
+var LangGraphAgent = (_class28 = class extends BaseAgent {
   /**
    * Graph nodes (agents and their connections)
    */
@@ -9092,8 +9303,8 @@ var LangGraphAgent = (_class27 = class extends BaseAgent {
   /**
    * Results from node executions
    */
-  __init47() {this.results = []}
-  __init48() {this.logger = new Logger({ name: "LangGraphAgent" })}
+  __init48() {this.results = []}
+  __init49() {this.logger = new Logger({ name: "LangGraphAgent" })}
   /**
    * Constructor for LangGraphAgent
    */
@@ -9101,7 +9312,7 @@ var LangGraphAgent = (_class27 = class extends BaseAgent {
     super({
       name: config.name,
       description: config.description
-    });_class27.prototype.__init47.call(this);_class27.prototype.__init48.call(this);;
+    });_class28.prototype.__init48.call(this);_class28.prototype.__init49.call(this);;
     this.nodes = /* @__PURE__ */ new Map();
     for (const node of config.nodes) {
       if (this.nodes.has(node.name)) {
@@ -9297,7 +9508,7 @@ var LangGraphAgent = (_class27 = class extends BaseAgent {
     }
     this.maxSteps = maxSteps;
   }
-}, _class27);
+}, _class28);
 
 // src/agents/agent-builder.ts
 
@@ -9400,8 +9611,8 @@ var RunConfig = class {
 };
 
 // src/artifacts/in-memory-artifact-service.ts
-var InMemoryArtifactService = (_class28 = class {constructor() { _class28.prototype.__init49.call(this); }
-  __init49() {this.artifacts = /* @__PURE__ */ new Map()}
+var InMemoryArtifactService = (_class29 = class {constructor() { _class29.prototype.__init50.call(this); }
+  __init50() {this.artifacts = /* @__PURE__ */ new Map()}
   fileHasUserNamespace(filename) {
     return filename.startsWith("user:");
   }
@@ -9474,7 +9685,7 @@ var InMemoryArtifactService = (_class28 = class {constructor() { _class28.protot
     }
     return Array.from({ length: versions.length }, (_, i) => i);
   }
-}, _class28);
+}, _class29);
 
 // src/runners.ts
 init_logger();
@@ -9501,15 +9712,15 @@ function _extractWordsLower(text) {
   const words = text.match(/[A-Za-z]+/g) || [];
   return new Set(words.map((word) => word.toLowerCase()));
 }
-var InMemoryMemoryService = (_class29 = class {
+var InMemoryMemoryService = (_class30 = class {
   /**
    * Keys are app_name/user_id, session_id. Values are session event lists.
    */
-  __init50() {this._sessionEvents = /* @__PURE__ */ new Map()}
+  __init51() {this._sessionEvents = /* @__PURE__ */ new Map()}
   /**
    * Constructor for InMemoryMemoryService
    */
-  constructor() {;_class29.prototype.__init50.call(this);
+  constructor() {;_class30.prototype.__init51.call(this);
     this._sessionEvents = /* @__PURE__ */ new Map();
   }
   /**
@@ -9593,7 +9804,7 @@ var InMemoryMemoryService = (_class29 = class {
   clear() {
     this._sessionEvents.clear();
   }
-}, _class29);
+}, _class30);
 
 // src/sessions/in-memory-session-service.ts
 var _crypto = require('crypto');
@@ -9635,19 +9846,19 @@ var BaseSessionService = class {
 };
 
 // src/sessions/in-memory-session-service.ts
-var InMemorySessionService = (_class30 = class extends BaseSessionService {constructor(...args2) { super(...args2); _class30.prototype.__init51.call(this);_class30.prototype.__init52.call(this);_class30.prototype.__init53.call(this); }
+var InMemorySessionService = (_class31 = class extends BaseSessionService {constructor(...args3) { super(...args3); _class31.prototype.__init52.call(this);_class31.prototype.__init53.call(this);_class31.prototype.__init54.call(this); }
   /**
    * A map from app name to a map from user ID to a map from session ID to session.
    */
-  __init51() {this.sessions = /* @__PURE__ */ new Map()}
+  __init52() {this.sessions = /* @__PURE__ */ new Map()}
   /**
    * A map from app name to a map from user ID to a map from key to the value.
    */
-  __init52() {this.userState = /* @__PURE__ */ new Map()}
+  __init53() {this.userState = /* @__PURE__ */ new Map()}
   /**
    * A map from app name to a map from key to the value.
    */
-  __init53() {this.appState = /* @__PURE__ */ new Map()}
+  __init54() {this.appState = /* @__PURE__ */ new Map()}
   /**
    * Creates a new session.
    */
@@ -9844,7 +10055,7 @@ var InMemorySessionService = (_class30 = class extends BaseSessionService {const
     storageSession.lastUpdateTime = event.timestamp;
     return event;
   }
-}, _class30);
+}, _class31);
 
 // src/runners.ts
 function _findFunctionCallEventIfLastEventIsFunctionResponse(session) {
@@ -9870,7 +10081,7 @@ function _findFunctionCallEventIfLastEventIsFunctionResponse(session) {
   }
   return null;
 }
-var Runner = (_class31 = class {
+var Runner = (_class32 = class {
   /**
    * The app name of the runner.
    */
@@ -9891,7 +10102,7 @@ var Runner = (_class31 = class {
    * The memory service for the runner.
    */
   
-  __init54() {this.logger = new Logger({ name: "Runner" })}
+  __init55() {this.logger = new Logger({ name: "Runner" })}
   /**
    * Initializes the Runner.
    */
@@ -9901,7 +10112,7 @@ var Runner = (_class31 = class {
     artifactService,
     sessionService,
     memoryService
-  }) {;_class31.prototype.__init54.call(this);
+  }) {;_class32.prototype.__init55.call(this);
     this.appName = appName;
     this.agent = agent;
     this.artifactService = artifactService;
@@ -10108,7 +10319,7 @@ var Runner = (_class31 = class {
       runConfig
     });
   }
-}, _class31);
+}, _class32);
 var InMemoryRunner = class extends Runner {
   /**
    * Deprecated. Please don't use. The in-memory session service for the runner.
@@ -10131,18 +10342,18 @@ var InMemoryRunner = class extends Runner {
 };
 
 // src/agents/agent-builder.ts
-var AgentBuilder = (_class32 = class _AgentBuilder {
+var AgentBuilder = (_class33 = class _AgentBuilder {
   
   
   
   
   
-  __init55() {this.agentType = "llm"}
+  __init56() {this.agentType = "llm"}
   
   /**
    * Private constructor - use static create() method
    */
-  constructor(name) {;_class32.prototype.__init55.call(this);
+  constructor(name) {;_class33.prototype.__init56.call(this);
     this.config = { name };
   }
   /**
@@ -10188,6 +10399,14 @@ var AgentBuilder = (_class32 = class _AgentBuilder {
     this.config.instruction = instruction;
     return this;
   }
+  withInputSchema(schema) {
+    this.config.inputSchema = schema;
+    return this;
+  }
+  withOutputSchema(schema) {
+    this.config.outputSchema = schema;
+    return this;
+  }
   /**
    * Add tools to the agent
    * @param tools Tools to add to the agent
@@ -10394,6 +10613,14 @@ var AgentBuilder = (_class32 = class _AgentBuilder {
     }
     return { agent, runner, session };
   }
+  /**
+   * Type-safe build method for agents with output schemas
+   * Provides better type inference for the ask method return type
+   */
+  async buildWithSchema() {
+    const result = await this.build();
+    return result;
+  }
   /**
    * Quick execution helper - build and run a message
    * @param message Message to send to the agent (string or full message object)
@@ -10428,7 +10655,9 @@ var AgentBuilder = (_class32 = class _AgentBuilder {
           memoryService: this.memoryService,
           artifactService: this.artifactService,
           outputKey: this.config.outputKey,
-          sessionService: this.sessionService
+          sessionService: this.sessionService,
+          inputSchema: this.config.inputSchema,
+          outputSchema: this.config.outputSchema
         });
       }
       case "sequential":
@@ -10487,14 +10716,16 @@ var AgentBuilder = (_class32 = class _AgentBuilder {
     return `app-${this.config.name}`;
   }
   /**
-   * Create enhanced runner with simplified API
+   * Create enhanced runner with simplified API and proper typing
    * @param baseRunner The base runner instance
    * @param session The session instance
    * @returns Enhanced runner with simplified API
    */
   createEnhancedRunner(baseRunner, session) {
     const sessionOptions = this.sessionOptions;
+    const outputSchema = this.config.outputSchema;
     return {
+      __outputSchema: outputSchema,
       async ask(message) {
         const newMessage = typeof message === "string" ? { parts: [{ text: message }] } : typeof message === "object" && "contents" in message ? { parts: message.contents[message.contents.length - 1].parts } : message;
         let response = "";
@@ -10515,6 +10746,18 @@ var AgentBuilder = (_class32 = class _AgentBuilder {
             }
           }
         }
+        if (outputSchema) {
+          try {
+            const parsed = JSON.parse(response);
+            return outputSchema.parse(parsed);
+          } catch (parseError) {
+            try {
+              return outputSchema.parse(response);
+            } catch (validationError) {
+              return response.trim();
+            }
+          }
+        }
         return response.trim();
       },
       runAsync(params) {
@@ -10522,7 +10765,7 @@ var AgentBuilder = (_class32 = class _AgentBuilder {
       }
     };
   }
-}, _class32);
+}, _class33);
 
 // src/memory/index.ts
 var memory_exports = {};
@@ -10856,11 +11099,11 @@ var VertexAiSessionService = class extends BaseSessionService {
 
 // src/sessions/database-session-service.ts
 var _kysely = require('kysely');
-var DatabaseSessionService = (_class33 = class extends BaseSessionService {
+var DatabaseSessionService = (_class34 = class extends BaseSessionService {
   
-  __init56() {this.initialized = false}
+  __init57() {this.initialized = false}
   constructor(config) {
-    super();_class33.prototype.__init56.call(this);;
+    super();_class34.prototype.__init57.call(this);;
     this.db = config.db;
     if (!config.skipTableCreation) {
       this.initializeDatabase().catch((error) => {
@@ -10935,7 +11178,7 @@ var DatabaseSessionService = (_class33 = class extends BaseSessionService {
     if (!jsonString) return defaultValue;
     try {
       return JSON.parse(jsonString);
-    } catch (e5) {
+    } catch (e6) {
       return defaultValue;
     }
   }
@@ -11254,7 +11497,7 @@ var DatabaseSessionService = (_class33 = class extends BaseSessionService {
       }
     }
   }
-}, _class33);
+}, _class34);
 
 // src/sessions/database-factories.ts