@botonic/plugin-ai-agents 0.43.0-beta.0 → 0.44.0-alpha.0

package/lib/esm/runner.js

@@ -1,54 +1,58 @@
-import { __awaiter } from "tslib";
-import { InputGuardrailTripwireTriggered, Runner, RunToolCallItem, } from '@openai/agents';
-import { retrieveKnowledge } from './tools';
-export class AIAgentRunner {
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIAgentRunner = void 0;
+const agents_1 = require("@openai/agents");
+const tools_1 = require("./tools");
+class AIAgentRunner {
     constructor(agent) {
         this.agent = agent;
     }
-    run(messages, context) {
-        var _a;
-        return __awaiter(this, void 0, void 0, function* () {
-            console.log('Running agent with messages', JSON.stringify(messages, null, 2));
-            try {
-                const runner = new Runner({
-                    modelSettings: { temperature: 0 },
-                });
-                const result = yield runner.run(this.agent, messages, { context });
-                const outputMessages = ((_a = result.finalOutput) === null || _a === void 0 ? void 0 : _a.messages) || [];
-                const hasExit = outputMessages.length === 0 ||
-                    outputMessages.some(message => message.type === 'exit');
-                const toolsExecuted = this.getToolsExecuted(result, context);
+    async run(messages, context) {
+        try {
+            const runner = new agents_1.Runner({
+                modelSettings: { temperature: 0 },
+            });
+            // Type assertion to bypass strict type checking - the actual return type from runner.run()
+            // doesn't perfectly match our interface, but the properties we access are compatible
+            const result = (await runner.run(this.agent, messages, {
+                context,
+            }));
+            const outputMessages = result.finalOutput?.messages || [];
+            const hasExit = outputMessages.length === 0 ||
+                outputMessages.some(message => message.type === 'exit');
+            const toolsExecuted = this.getToolsExecuted(result, context);
+            return {
+                messages: hasExit
+                    ? []
+                    : outputMessages.filter(message => message.type !== 'exit'),
+                toolsExecuted,
+                exit: hasExit,
+                memoryLength: messages.length,
+                error: false,
+                inputGuardrailsTriggered: [],
+                outputGuardrailsTriggered: [],
+            };
+        }
+        catch (error) {
+            if (error instanceof agents_1.InputGuardrailTripwireTriggered) {
                 return {
-                    messages: hasExit
-                        ? []
-                        : outputMessages.filter(message => message.type !== 'exit'),
-                    toolsExecuted,
-                    exit: hasExit,
-                    memoryLength: messages.length,
+                    messages: [],
+                    memoryLength: 0,
+                    toolsExecuted: [],
+                    exit: true,
                     error: false,
-                    inputGuardrailsTriggered: [],
+                    inputGuardrailsTriggered: error.result.output.outputInfo,
                     outputGuardrailsTriggered: [],
                 };
             }
-            catch (error) {
-                if (error instanceof InputGuardrailTripwireTriggered) {
-                    return {
-                        messages: [],
-                        memoryLength: 0,
-                        toolsExecuted: [],
-                        exit: true,
-                        error: false,
-                        inputGuardrailsTriggered: error.result.output.outputInfo,
-                        outputGuardrailsTriggered: [],
-                    };
-                }
-                throw error;
-            }
-        });
+            throw error;
+        }
     }
     getToolsExecuted(result, context) {
-        var _a;
-        return (((_a = result.newItems) === null || _a === void 0 ? void 0 : _a.filter(item => item instanceof RunToolCallItem).map((item) => this.getToolExecutionInfo(item, context)).filter((toolExecution) => toolExecution.toolName !== '')) || []);
+        return (result.newItems
+            ?.filter(item => item instanceof agents_1.RunToolCallItem)
+            .map((item) => this.getToolExecutionInfo(item, context))
+            .filter((toolExecution) => toolExecution.toolName !== '') || []);
     }
     getToolExecutionInfo(item, context) {
         if (item.rawItem.type !== 'function_call') {
@@ -59,7 +63,7 @@ export class AIAgentRunner {
         }
         const toolName = item.rawItem.name;
         const toolArguments = this.getSafeToolArguments(item.rawItem.arguments);
-        if (toolName === retrieveKnowledge.name) {
+        if (toolName === tools_1.retrieveKnowledge.name) {
             return {
                 toolName,
                 toolArguments,
@@ -81,4 +85,5 @@ export class AIAgentRunner {
         }
     }
 }
+exports.AIAgentRunner = AIAgentRunner;
 //# sourceMappingURL=runner.js.map

package/lib/esm/runner.js.map

@@ -1 +1 @@
-{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/runner.ts"],"names":[],"mappings":";AACA,OAAO,EACL,+BAA+B,EAC/B,MAAM,EACN,eAAe,GAChB,MAAM,gBAAgB,CAAA;AAEvB,OAAO,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAA;AAS3C,MAAM,OAAO,aAAa;IAMxB,YAAY,KAAoC;QAC9C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;IACpB,CAAC;IAEK,GAAG,CACP,QAA+B,EAC/B,OAAsC;;;YAEtC,OAAO,CAAC,GAAG,CACT,6BAA6B,EAC7B,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAClC,CAAA;YACD,IAAI;gBACF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;oBACxB,aAAa,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE;iBAClC,CAAC,CAAA;gBACF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,OAAO,EAAE,CAAC,CAAA;gBAElE,MAAM,cAAc,GAAG,CAAA,MAAA,MAAM,CAAC,WAAW,0CAAE,QAAQ,KAAI,EAAE,CAAA;gBACzD,MAAM,OAAO,GACX,cAAc,CAAC,MAAM,KAAK,CAAC;oBAC3B,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAA;gBACzD,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;gBAE5D,OAAO;oBACL,QAAQ,EAAE,OAAO;wBACf,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAE,cAAc,CAAC,MAAM,CACpB,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CACR;oBAChC,aAAa;oBACb,IAAI,EAAE,OAAO;oBACb,YAAY,EAAE,QAAQ,CAAC,MAAM;oBAC7B,KAAK,EAAE,KAAK;oBACZ,wBAAwB,EAAE,EAAE;oBAC5B,yBAAyB,EAAE,EAAE;iBAC9B,CAAA;aACF;YAAC,OAAO,KAAK,EAAE;gBACd,IAAI,KAAK,YAAY,+BAA+B,EAAE;oBACpD,OAAO;wBACL,QAAQ,EAAE,EAAE;wBACZ,YAAY,EAAE,CAAC;wBACf,aAAa,EAAE,EAAE;wBACjB,IAAI,EAAE,IAAI;wBACV,KAAK,EAAE,KAAK;wBACZ,wBAAwB,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU;wBACxD,yBAAyB,EAAE,EAAE;qBAC9B,CAAA;iBACF;gBACD,MAAM,KAAK,CAAA;aACZ;;KACF;IAEO,gBAAgB,CACtB,MAAM,EACN,OAAsC;;QAEtC,OAAO,CACL,CAAA,MAAA,MAAM,CAAC,QAAQ,0CACX,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,YAAY,eAAe,EAC/C,GAAG,CAAC,CAAC,IAAqB,EAAE,EAAE,CAC7B,IAAI,CAAC,oBAAoB,CAAC,IAAuB,EAAE,OAAO,CAAC,EAE5D,MAAM,CACL,CAAC,aAA4B,EAAE,EAAE,CAAC,aAAa,CAAC,QAAQ,KAAK,EAAE,CAChE,KAAI,EAAE,CACV,CAAA;IACH,CAAC;IAEO,oBAAoB,CAC1B,IAAqB,EACrB,OAAsC;QAEtC,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,eAAe,EAAE;YACzC,OAAO;gBACL,QAAQ,EAAE,EAAE;gBACZ,aAAa,EAAE,EAAE;aAClB,CAAA;SACF;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAA;QAClC,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAEvE,IAAI,QAAQ,KAAK,iBAAiB,CAAC,IAAI,EAAE;YACvC,OAAO;gBACL,QAAQ;gBACR,aAAa;gBACb,uBAAuB,EAAE,OAAO,CAAC,aAAa,CAAC,SAAS;gBACxD,sBAAsB,EAAE,OAAO,CAAC,aAAa,CAAC,SAAS;aACxD,CAAA;SACF;QAED,OAAO;YACL,QAAQ;YACR,aAAa;SACd,CAAA;IACH,CAAC;IAEO,oBAAoB,CAAC,gBAAwB;QACnD,IAAI;YACF,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAA;SACpC;QAAC,OAAO,KAAK,EAAE;YACd,OAAO,EAAE,CAAA;SACV;IACH,CAAC;CACF"}
+{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/runner.ts"],"names":[],"mappings":";;;AACA,2CAIuB;AAEvB,mCAA2C;AAkB3C,MAAa,aAAa;IAMxB,YAAY,KAAoC;QAC9C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;IACpB,CAAC;IAED,KAAK,CAAC,GAAG,CACP,QAA+B,EAC/B,OAAsC;QAEtC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC;gBACxB,aAAa,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE;aAClC,CAAC,CAAA;YACF,2FAA2F;YAC3F,qFAAqF;YACrF,MAAM,MAAM,GAAG,CAAC,MAAM,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,EAAE;gBACrD,OAAO;aACR,CAAC,CAAwB,CAAA;YAE1B,MAAM,cAAc,GAAG,MAAM,CAAC,WAAW,EAAE,QAAQ,IAAI,EAAE,CAAA;YACzD,MAAM,OAAO,GACX,cAAc,CAAC,MAAM,KAAK,CAAC;gBAC3B,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAA;YACzD,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;YAE5D,OAAO;gBACL,QAAQ,EAAE,OAAO;oBACf,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAE,cAAc,CAAC,MAAM,CACpB,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CACR;gBAChC,aAAa;gBACb,IAAI,EAAE,OAAO;gBACb,YAAY,EAAE,QAAQ,CAAC,MAAM;gBAC7B,KAAK,EAAE,KAAK;gBACZ,wBAAwB,EAAE,EAAE;gBAC5B,yBAAyB,EAAE,EAAE;aAC9B,CAAA;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,wCAA+B,EAAE,CAAC;gBACrD,OAAO;oBACL,QAAQ,EAAE,EAAE;oBACZ,YAAY,EAAE,CAAC;oBACf,aAAa,EAAE,EAAE;oBACjB,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,KAAK;oBACZ,wBAAwB,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU;oBACxD,yBAAyB,EAAE,EAAE;iBAC9B,CAAA;YACH,CAAC;YACD,MAAM,KAAK,CAAA;QACb,CAAC;IACH,CAAC;IAEO,gBAAgB,CACtB,MAAM,EACN,OAAsC;QAEtC,OAAO,CACL,MAAM,CAAC,QAAQ;YACb,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,YAAY,wBAAe,CAAC;aAChD,GAAG,CAAC,CAAC,IAAqB,EAAE,EAAE,CAC7B,IAAI,CAAC,oBAAoB,CAAC,IAAuB,EAAE,OAAO,CAAC,CAC5D;aACA,MAAM,CACL,CAAC,aAA4B,EAAE,EAAE,CAAC,aAAa,CAAC,QAAQ,KAAK,EAAE,CAChE,IAAI,EAAE,CACV,CAAA;IACH,CAAC;IAEO,oBAAoB,CAC1B,IAAqB,EACrB,OAAsC;QAEtC,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YAC1C,OAAO;gBACL,QAAQ,EAAE,EAAE;gBACZ,aAAa,EAAE,EAAE;aAClB,CAAA;QACH,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAA;QAClC,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAEvE,IAAI,QAAQ,KAAK,yBAAiB,CAAC,IAAI,EAAE,CAAC;YACxC,OAAO;gBACL,QAAQ;gBACR,aAAa;gBACb,uBAAuB,EAAE,OAAO,CAAC,aAAa,CAAC,SAAS;gBACxD,sBAAsB,EAAE,OAAO,CAAC,aAAa,CAAC,SAAS;aACxD,CAAA;QACH,CAAC;QAED,OAAO;YACL,QAAQ;YACR,aAAa;SACd,CAAA;IACH,CAAC;IAEO,oBAAoB,CAAC,gBAAwB;QACnD,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAA;QACrC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,EAAE,CAAA;QACX,CAAC;IACH,CAAC;CACF;AA9GD,sCA8GC"}

package/lib/esm/structured-output/carousel.js

@@ -1,16 +1,19 @@
-import { z } from 'zod';
-export const CarouselSchema = z
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CarouselSchema = void 0;
+const zod_1 = require("zod");
+exports.CarouselSchema = zod_1.z
     .object({
-    type: z.enum(['carousel']),
-    content: z.object({
-        text: z.string().nullable().optional(),
-        elements: z.array(z.object({
-            title: z.string(),
-            subtitle: z.string(),
-            image: z.string(),
-            button: z.object({
-                text: z.string(),
-                url: z.string().nullable().optional(),
+    type: zod_1.z.enum(['carousel']),
+    content: zod_1.z.object({
+        text: zod_1.z.string().nullable().optional(),
+        elements: zod_1.z.array(zod_1.z.object({
+            title: zod_1.z.string(),
+            subtitle: zod_1.z.string(),
+            image: zod_1.z.string(),
+            button: zod_1.z.object({
+                text: zod_1.z.string(),
+                url: zod_1.z.string().nullable().optional(),
             }),
         })),
     }),

package/lib/esm/structured-output/carousel.js.map

@@ -1 +1 @@
-{"version":3,"file":"carousel.js","sourceRoot":"","sources":["../../../src/structured-output/carousel.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAIvB,MAAM,CAAC,MAAM,cAAc,GAAG,CAAC;KAC5B,MAAM,CAAC;IACN,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,CAAC;IAC1B,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;QACtC,QAAQ,EAAE,CAAC,CAAC,KAAK,CACf,CAAC,CAAC,MAAM,CAAC;YACP,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;YACjB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;YACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;YACjB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;gBACf,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;gBAChB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;aACtC,CAAC;SACH,CAAC,CACH;KACF,CAAC;CACH,CAAC;KACD,QAAQ,CAAC,kDAAkD,CAAC,CAAA"}
+{"version":3,"file":"carousel.js","sourceRoot":"","sources":["../../../src/structured-output/carousel.ts"],"names":[],"mappings":";;;AACA,6BAAuB;AAIV,QAAA,cAAc,GAAG,OAAC;KAC5B,MAAM,CAAC;IACN,IAAI,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,CAAC;IAC1B,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;QACtC,QAAQ,EAAE,OAAC,CAAC,KAAK,CACf,OAAC,CAAC,MAAM,CAAC;YACP,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;YACjB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;YACpB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;YACjB,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC;gBACf,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;gBAChB,GAAG,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;aACtC,CAAC;SACH,CAAC,CACH;KACF,CAAC;CACH,CAAC;KACD,QAAQ,CAAC,kDAAkD,CAAC,CAAA"}

package/lib/esm/structured-output/exit.js

@@ -1,7 +1,11 @@
-import z from 'zod';
-export const ExitSchema = z
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExitSchema = void 0;
+const tslib_1 = require("tslib");
+const zod_1 = tslib_1.__importDefault(require("zod"));
+exports.ExitSchema = zod_1.default
     .object({
-    type: z.enum(['exit']),
+    type: zod_1.default.enum(['exit']),
 })
     .describe('An exit message. This message should only be used to exit the agent due to out of context or the user asking to exit the conversation');
 //# sourceMappingURL=exit.js.map

package/lib/esm/structured-output/exit.js.map

@@ -1 +1 @@
-{"version":3,"file":"exit.js","sourceRoot":"","sources":["../../../src/structured-output/exit.ts"],"names":[],"mappings":"AACA,OAAO,CAAC,MAAM,KAAK,CAAA;AAInB,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC;KACxB,MAAM,CAAC;IACN,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC;CACvB,CAAC;KACD,QAAQ,CACP,uIAAuI,CACxI,CAAA"}
+{"version":3,"file":"exit.js","sourceRoot":"","sources":["../../../src/structured-output/exit.ts"],"names":[],"mappings":";;;;AACA,sDAAmB;AAIN,QAAA,UAAU,GAAG,aAAC;KACxB,MAAM,CAAC;IACN,IAAI,EAAE,aAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC;CACvB,CAAC;KACD,QAAQ,CACP,uIAAuI,CACxI,CAAA"}

package/lib/esm/structured-output/index.js

@@ -1,11 +1,14 @@
-import { z } from 'zod';
-import { CarouselSchema } from './carousel';
-import { ExitSchema } from './exit';
-import { TextSchema } from './text';
-import { TextWithButtonsSchema } from './text-with-buttons';
-export const OutputSchema = z
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OutputSchema = void 0;
+const zod_1 = require("zod");
+const carousel_1 = require("./carousel");
+const exit_1 = require("./exit");
+const text_1 = require("./text");
+const text_with_buttons_1 = require("./text-with-buttons");
+exports.OutputSchema = zod_1.z
     .object({
-    messages: z.array(z.union([TextSchema, TextWithButtonsSchema, CarouselSchema, ExitSchema])),
+    messages: zod_1.z.array(zod_1.z.union([text_1.TextSchema, text_with_buttons_1.TextWithButtonsSchema, carousel_1.CarouselSchema, exit_1.ExitSchema])),
 })
     .describe('The messages to be sent to the user');
 //# sourceMappingURL=index.js.map

package/lib/esm/structured-output/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/structured-output/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AACnC,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AACnC,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAA;AAM3D,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC;KAC1B,MAAM,CAAC;IACN,QAAQ,EAAE,CAAC,CAAC,KAAK,CACf,CAAC,CAAC,KAAK,CAAC,CAAC,UAAU,EAAE,qBAAqB,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC,CACzE;CACF,CAAC;KACD,QAAQ,CAAC,qCAAqC,CAAC,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/structured-output/index.ts"],"names":[],"mappings":";;;AACA,6BAAuB;AAEvB,yCAA2C;AAC3C,iCAAmC;AACnC,iCAAmC;AACnC,2DAA2D;AAM9C,QAAA,YAAY,GAAG,OAAC;KAC1B,MAAM,CAAC;IACN,QAAQ,EAAE,OAAC,CAAC,KAAK,CACf,OAAC,CAAC,KAAK,CAAC,CAAC,iBAAU,EAAE,yCAAqB,EAAE,yBAAc,EAAE,iBAAU,CAAC,CAAC,CACzE;CACF,CAAC;KACD,QAAQ,CAAC,qCAAqC,CAAC,CAAA"}

package/lib/esm/structured-output/text-with-buttons.js

@@ -1,12 +1,16 @@
-import z from 'zod';
-export const TextWithButtonsSchema = z
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TextWithButtonsSchema = void 0;
+const tslib_1 = require("tslib");
+const zod_1 = tslib_1.__importDefault(require("zod"));
+exports.TextWithButtonsSchema = zod_1.default
     .object({
-    type: z.enum(['textWithButtons']),
-    content: z.object({
-        text: z.string(),
-        buttons: z.array(z.object({
-            text: z.string(),
-            url: z.string().nullable().optional(),
+    type: zod_1.default.enum(['textWithButtons']),
+    content: zod_1.default.object({
+        text: zod_1.default.string(),
+        buttons: zod_1.default.array(zod_1.default.object({
+            text: zod_1.default.string(),
+            url: zod_1.default.string().nullable().optional(),
         })),
     }),
 })

package/lib/esm/structured-output/text-with-buttons.js.map

@@ -1 +1 @@
-{"version":3,"file":"text-with-buttons.js","sourceRoot":"","sources":["../../../src/structured-output/text-with-buttons.ts"],"names":[],"mappings":"AACA,OAAO,CAAC,MAAM,KAAK,CAAA;AAInB,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC;KACnC,MAAM,CAAC;IACN,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,iBAAiB,CAAC,CAAC;IACjC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,OAAO,EAAE,CAAC,CAAC,KAAK,CACd,CAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;SACtC,CAAC,CACH;KACF,CAAC;CACH,CAAC;KACD,QAAQ,CACP,oEAAoE,CACrE,CAAA"}
+{"version":3,"file":"text-with-buttons.js","sourceRoot":"","sources":["../../../src/structured-output/text-with-buttons.ts"],"names":[],"mappings":";;;;AACA,sDAAmB;AAIN,QAAA,qBAAqB,GAAG,aAAC;KACnC,MAAM,CAAC;IACN,IAAI,EAAE,aAAC,CAAC,IAAI,CAAC,CAAC,iBAAiB,CAAC,CAAC;IACjC,OAAO,EAAE,aAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,aAAC,CAAC,MAAM,EAAE;QAChB,OAAO,EAAE,aAAC,CAAC,KAAK,CACd,aAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,aAAC,CAAC,MAAM,EAAE;YAChB,GAAG,EAAE,aAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;SACtC,CAAC,CACH;KACF,CAAC;CACH,CAAC;KACD,QAAQ,CACP,oEAAoE,CACrE,CAAA"}

package/lib/esm/structured-output/text.js

@@ -1,9 +1,12 @@
-import { z } from 'zod';
-export const TextSchema = z
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TextSchema = void 0;
+const zod_1 = require("zod");
+exports.TextSchema = zod_1.z
     .object({
-    type: z.enum(['text']),
-    content: z.object({
-        text: z.string(),
+    type: zod_1.z.enum(['text']),
+    content: zod_1.z.object({
+        text: zod_1.z.string(),
     }),
 })
     .describe('A text message');

package/lib/esm/structured-output/text.js.map

@@ -1 +1 @@
-{"version":3,"file":"text.js","sourceRoot":"","sources":["../../../src/structured-output/text.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAIvB,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC;KACxB,MAAM,CAAC;IACN,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC;IACtB,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;KACjB,CAAC;CACH,CAAC;KACD,QAAQ,CAAC,gBAAgB,CAAC,CAAA"}
+{"version":3,"file":"text.js","sourceRoot":"","sources":["../../../src/structured-output/text.ts"],"names":[],"mappings":";;;AACA,6BAAuB;AAIV,QAAA,UAAU,GAAG,OAAC;KACxB,MAAM,CAAC;IACN,IAAI,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC;IACtB,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;KACjB,CAAC;CACH,CAAC;KACD,QAAQ,CAAC,gBAAgB,CAAC,CAAA"}

package/lib/esm/tools/index.js

@@ -1,3 +1,7 @@
-export { retrieveKnowledge } from './retrieve-knowledge';
-export const mandatoryTools = [];
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mandatoryTools = exports.retrieveKnowledge = void 0;
+var retrieve_knowledge_1 = require("./retrieve-knowledge");
+Object.defineProperty(exports, "retrieveKnowledge", { enumerable: true, get: function () { return retrieve_knowledge_1.retrieveKnowledge; } });
+exports.mandatoryTools = [];
 //# sourceMappingURL=index.js.map

package/lib/esm/tools/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAA;AAExD,MAAM,CAAC,MAAM,cAAc,GAAW,EAAE,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/index.ts"],"names":[],"mappings":";;;AACA,2DAAwD;AAA/C,uHAAA,iBAAiB,OAAA;AAEb,QAAA,cAAc,GAAW,EAAE,CAAA"}

package/lib/esm/tools/retrieve-knowledge.d.ts

@@ -1,2 +1,9 @@
+import { z } from 'zod';
 import { Context } from '../types';
-export declare const retrieveKnowledge: import("@openai/agents").FunctionTool<Context<import("@botonic/core").ResolvedPlugins, any>, any, any>;
+export declare const retrieveKnowledge: import("@openai/agents").FunctionTool<Context<import("@botonic/core").ResolvedPlugins, any>, z.ZodObject<{
+    query: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    query: string;
+}, {
+    query: string;
+}>, string>;

package/lib/esm/tools/retrieve-knowledge.js

@@ -1,21 +1,24 @@
-import { __awaiter } from "tslib";
-import { tool } from '@openai/agents';
-import { z } from 'zod';
-import { HubtypeApiClient } from '../hubtype-api-client';
-export const retrieveKnowledge = tool({
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.retrieveKnowledge = void 0;
+const agents_1 = require("@openai/agents");
+const zod_1 = require("zod");
+const hubtype_api_client_1 = require("../hubtype-api-client");
+exports.retrieveKnowledge = (0, agents_1.tool)({
     name: 'retrieve_knowledge',
     description: 'Consult the knowledge base for information before answering. Use this tool to make sure the information you provide is faithful.',
-    parameters: z.object({
-        query: z.string().describe('The query to search the knowledge base for'),
+    parameters: zod_1.z.object({
+        query: zod_1.z.string().describe('The query to search the knowledge base for'),
     }),
-    execute: ({ query }, runContext) => __awaiter(void 0, void 0, void 0, function* () {
-        const context = runContext === null || runContext === void 0 ? void 0 : runContext.context;
+    execute: async (input, runContext) => {
+        const context = runContext?.context;
+        const query = input.query;
         if (!context) {
             throw new Error('Context is required');
         }
         const sourceIds = context.sourceIds;
-        const client = new HubtypeApiClient(context.authToken);
-        const chunks = yield client.retrieveSimilarChunks(query, sourceIds);
+        const client = new hubtype_api_client_1.HubtypeApiClient(context.authToken);
+        const chunks = await client.retrieveSimilarChunks(query, sourceIds);
         const chunkTexts = chunks.map(chunk => chunk.text);
         context.knowledgeUsed = {
             query,
@@ -24,6 +27,6 @@ export const retrieveKnowledge = tool({
             chunkTexts,
         };
         return chunkTexts;
-    }),
+    },
 });
 //# sourceMappingURL=retrieve-knowledge.js.map

package/lib/esm/tools/retrieve-knowledge.js.map

@@ -1 +1 @@
-{"version":3,"file":"retrieve-knowledge.js","sourceRoot":"","sources":["../../../src/tools/retrieve-knowledge.ts"],"names":[],"mappings":";AAAA,OAAO,EAAc,IAAI,EAAE,MAAM,gBAAgB,CAAA;AACjD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAA;AAGxD,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAoB;IACvD,IAAI,EAAE,oBAAoB;IAC1B,WAAW,EACT,kIAAkI;IACpI,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;QACnB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;KACzE,CAAC;IACF,OAAO,EAAE,CACP,EAAE,KAAK,EAAqB,EAC5B,UAAgC,EACb,EAAE;QACrB,MAAM,OAAO,GAAG,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,OAAO,CAAA;QACnC,IAAI,CAAC,OAAO,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAA;SACvC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAA;QACnC,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QACtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAA;QACnE,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAElD,OAAO,CAAC,aAAa,GAAG;YACtB,KAAK;YACL,SAAS;YACT,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;YACxC,UAAU;SACX,CAAA;QAED,OAAO,UAAU,CAAA;IACnB,CAAC,CAAA;CACF,CAAC,CAAA"}
+{"version":3,"file":"retrieve-knowledge.js","sourceRoot":"","sources":["../../../src/tools/retrieve-knowledge.ts"],"names":[],"mappings":";;;AAAA,2CAAiD;AACjD,6BAAuB;AAEvB,8DAAwD;AAG3C,QAAA,iBAAiB,GAAG,IAAA,aAAI,EAAC;IACpC,IAAI,EAAE,oBAAoB;IAC1B,WAAW,EACT,kIAAkI;IACpI,UAAU,EAAE,OAAC,CAAC,MAAM,CAAC;QACnB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;KACzE,CAAC;IACF,OAAO,EAAE,KAAK,EACZ,KAAwB,EACxB,UAAgC,EACb,EAAE;QACrB,MAAM,OAAO,GAAG,UAAU,EAAE,OAAO,CAAA;QACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAA;QACzB,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAA;QACxC,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,CAAA;QACnC,MAAM,MAAM,GAAG,IAAI,qCAAgB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QACtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAA;QACnE,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAElD,OAAO,CAAC,aAAa,GAAG;YACtB,KAAK;YACL,SAAS;YACT,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;YACxC,UAAU;SACX,CAAA;QAED,OAAO,UAAU,CAAA;IACnB,CAAC;CACF,CAAC,CAAA"}

package/lib/esm/types.d.ts

@@ -1,5 +1,5 @@
 import { AgenticOutputMessage, AiAgentArgs, BotContext, GuardrailRule, InferenceResponse, ResolvedPlugins, RunResult } from '@botonic/core';
-import { Agent, AgentInputItem, RunContext as OpenAIRunContext, Tool as OpenAITool } from '@openai/agents';
+import { Agent, AgentInputItem, AgentOutputType, RunContext as OpenAIRunContext, Tool as OpenAITool } from '@openai/agents';
 import { ZodSchema } from 'zod';
 import { OutputSchema } from './structured-output';
 export interface Context<TPlugins extends ResolvedPlugins = ResolvedPlugins, TExtraData = any> {
@@ -24,9 +24,8 @@ export interface Chunk {
     id: string;
     text: string;
 }
-export type ContactInfo = Record<string, string>;
 export type Tool<TPlugins extends ResolvedPlugins = ResolvedPlugins, TExtraData = any> = OpenAITool<Context<TPlugins, TExtraData>>;
-export type AIAgent<TPlugins extends ResolvedPlugins = ResolvedPlugins, TExtraData = any> = Agent<Context<TPlugins, TExtraData>, typeof OutputSchema>;
+export type AIAgent<TPlugins extends ResolvedPlugins = ResolvedPlugins, TExtraData = any> = Agent<Context<TPlugins, TExtraData>, AgentOutputType<typeof OutputSchema>>;
 export type MessageHistoryApiVersion = 'v1' | 'v2';
 export interface MemoryOptions {
     maxMessages?: number;

package/lib/esm/types.js

@@ -1,2 +1,3 @@
-export {};
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
 //# sourceMappingURL=types.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@botonic/plugin-ai-agents",
-  "version": "0.43.0-beta.0",
+  "version": "0.44.0-alpha.0",
   "main": "./lib/cjs/index.js",
   "module": "./lib/esm/index.js",
   "description": "Use AI Agents to generate your contents",
@@ -13,10 +13,10 @@
     "lint_core": "../../node_modules/.bin/eslint_d --cache --quiet 'src/**/*.ts*'"
   },
   "dependencies": {
-    "@botonic/core": "^0.42.0",
-    "@openai/agents": "^0.1.11",
-    "axios": "^1.12.2",
-    "openai": "^5.23.2",
+    "@botonic/core": "^0.44.0-alpha.0",
+    "@openai/agents": "^0.3.9",
+    "axios": "^1.13.2",
+    "openai": "^6.0.0",
     "uuid": "^10.0.0",
     "zod": "3.25.76"
   },

package/src/agent-builder.ts

@@ -1,10 +1,29 @@
-import { ResolvedPlugins } from '@botonic/core'
-import { Agent, InputGuardrail, ModelSettings } from '@openai/agents'
+import { CampaignV2, ContactInfo, ResolvedPlugins } from '@botonic/core'
+import {
+  Agent,
+  AgentOutputType,
+  InputGuardrail,
+  ModelSettings,
+} from '@openai/agents'
 
+import { OPENAI_MODEL, OPENAI_PROVIDER } from './constants'
 import { createInputGuardrail } from './guardrails'
 import { OutputSchema } from './structured-output'
 import { mandatoryTools, retrieveKnowledge } from './tools'
-import { AIAgent, ContactInfo, GuardrailRule, Tool } from './types'
+import { AIAgent, Context, GuardrailRule, Tool } from './types'
+
+interface AIAgentBuilderOptions<
+  TPlugins extends ResolvedPlugins = ResolvedPlugins,
+  TExtraData = any,
+> {
+  name: string
+  instructions: string
+  tools: Tool<TPlugins, TExtraData>[]
+  campaignContext?: CampaignV2
+  contactInfo: ContactInfo[]
+  inputGuardrailRules: GuardrailRule[]
+  sourceIds: string[]
+}
 
 export class AIAgentBuilder<
   TPlugins extends ResolvedPlugins = ResolvedPlugins,
@@ -15,33 +34,46 @@ export class AIAgentBuilder<
   private tools: Tool<TPlugins, TExtraData>[]
   private inputGuardrails: InputGuardrail[]
 
-  constructor(
-    name: string,
-    instructions: string,
-    tools: Tool<TPlugins, TExtraData>[],
-    contactInfo: ContactInfo,
-    inputGuardrailRules: GuardrailRule[],
-    sourceIds: string[]
-  ) {
-    this.name = name
-    this.instructions = this.addExtraInstructions(instructions, contactInfo)
-    this.tools = this.addHubtypeTools(tools, sourceIds)
+  constructor(options: AIAgentBuilderOptions<TPlugins, TExtraData>) {
+    this.name = options.name
+    this.instructions = this.addExtraInstructions(
+      options.instructions,
+      options.contactInfo,
+      options.campaignContext
+    )
+    this.tools = this.addHubtypeTools(options.tools, options.sourceIds)
     this.inputGuardrails = []
-    if (inputGuardrailRules.length > 0) {
-      const inputGuardrail = createInputGuardrail(inputGuardrailRules)
+    if (options.inputGuardrailRules.length > 0) {
+      const inputGuardrail = createInputGuardrail(options.inputGuardrailRules)
       this.inputGuardrails.push(inputGuardrail)
     }
   }
 
   build(): AIAgent<TPlugins, TExtraData> {
-    const modelSettings: ModelSettings = {}
+    const modelSettings: ModelSettings = {} as ModelSettings
+    if (OPENAI_PROVIDER === 'openai') {
+      modelSettings.reasoning = { effort: 'none' }
+      modelSettings.text = { verbosity: 'medium' }
+    }
 
-    if (this.tools.includes(retrieveKnowledge)) {
+    if (this.tools.includes(retrieveKnowledge) && OPENAI_PROVIDER === 'azure') {
       modelSettings.toolChoice = retrieveKnowledge.name
     }
 
-    return new Agent({
+    // When using standard OpenAI API, we need to specify the model
+    // Azure OpenAI uses deployment name instead
+    const model = OPENAI_PROVIDER === 'openai' ? OPENAI_MODEL : undefined
+
+    // TODO: Improve type safety - replace AgentOutputType<any> with AgentOutputType<typeof OutputSchema>
+    // Currently using explicit type parameters to avoid type inference issues where Agent constructor
+    // infers ZodObject instead of AgentOutputType<typeof OutputSchema>. The explicit type parameters
+    // ensure compatibility with AIAgent type definition. Future improvements:
+    // 1. Update @openai/agents package to properly infer AgentOutputType from outputType parameter
+    // 2. Replace AgentOutputType<any> with AgentOutputType<typeof OutputSchema> once type system allows
+    // 3. Consider updating AIAgent type definition if @openai/agents types change significantly
+    return new Agent<Context<TPlugins, TExtraData>, AgentOutputType<any>>({
       name: this.name,
+      model,
       instructions: this.instructions,
       tools: this.tools,
       outputType: OutputSchema,
@@ -53,20 +85,34 @@ export class AIAgentBuilder<
 
   private addExtraInstructions(
     initialInstructions: string,
-    contactInfo: ContactInfo
+    contactInfo: ContactInfo[],
+    campaignContext?: CampaignV2
   ): string {
-    const instructions = `<instructions>\n${initialInstructions}\n</instructions>`
+    const instructions = `<instructions>\n${initialInstructions.trim()}\n</instructions>`
     const metadataInstructions = this.getMetadataInstructions()
     const contactInfoInstructions = this.getContactInfoInstructions(contactInfo)
+    const campaignInstructions = this.getCampaignInstructions(campaignContext)
     const outputInstructions = this.getOutputInstructions()
-    return `${instructions}\n\n${metadataInstructions}\n\n${contactInfoInstructions}\n\n${outputInstructions}`
+    return `${instructions}\n\n${metadataInstructions}\n\n${contactInfoInstructions}\n\n${campaignInstructions}\n\n${outputInstructions}`
   }
 
-  private getContactInfoInstructions(contactInfo: ContactInfo): string {
-    const structuredContactInfo = Object.entries(contactInfo)
-      .map(([key, value]) => `${key}: ${value}`)
+  private getContactInfoInstructions(contactInfo: ContactInfo[]): string {
+    const structuredContactInfo = contactInfo
+      .map(
+        info =>
+          ` <contact_info>
+              <name>${info.name}</name>
+              <value>${info.value}</value>
+              <type>${info.type}</type>
+              ${
+                info.description
+                  ? `<description>${info.description}</description>`
+                  : ''
+              }
+            </contact_info>`
+      )
       .join('\n')
-    return `<contact_info>\n${structuredContactInfo}\n</contact_info>`
+    return `<contact_info_fields>\n${structuredContactInfo}</contact_info_fields>`
   }
 
   private getMetadataInstructions(): string {
@@ -74,6 +120,13 @@ export class AIAgentBuilder<
     return `<metadata>\n${metadata}\n</metadata>`
   }
 
+  private getCampaignInstructions(campaignContext?: CampaignV2): string {
+    if (!campaignContext || !campaignContext.agent_context) {
+      return ''
+    }
+    return `<campaign_context>\n${campaignContext.agent_context}\n</campaign_context>`
+  }
+
   private getOutputInstructions(): string {
     const example = {
       messages: [

package/src/constants.ts

@@ -1,6 +1,13 @@
 export const HUBTYPE_API_URL =
   process.env.HUBTYPE_API_URL || 'https://api.hubtype.com'
 
+// OpenAI Provider Configuration
+export const OPENAI_API_KEY = process.env.OPENAI_API_KEY // pragma: allowlist secret
+export const OPENAI_MODEL = process.env.OPENAI_MODEL || 'gpt-4.1-mini'
+export const OPENAI_PROVIDER: 'openai' | 'azure' =
+  (process.env.OPENAI_PROVIDER as 'openai' | 'azure') || 'azure'
+
+// Azure OpenAI Configuration
 export const AZURE_OPENAI_API_KEY = process.env.AZURE_OPENAI_API_KEY // pragma: allowlist secret
 export const AZURE_OPENAI_API_BASE = process.env.AZURE_OPENAI_API_BASE
 export const AZURE_OPENAI_API_DEPLOYMENT_NAME =

package/src/guardrails/input.ts

@@ -22,7 +22,7 @@ export function createInputGuardrail(rules: GuardrailRule[]): InputGuardrail {
     execute: async ({ input, context }) => {
       const lastMessage = input[input.length - 1] as UserMessageItem
       const result = await run(agent, [lastMessage], { context })
-      const finalOutput = result.finalOutput
+      const finalOutput = result.finalOutput as Record<string, boolean>
       if (finalOutput === undefined) {
         throw new Error('Guardrail agent failed to produce output')
       }