@botonic/plugin-ai-agents 0.37.1 → 0.38.0-alpha.0

package/lib/esm/runner.js

@@ -1,5 +1,5 @@
 import { __awaiter } from "tslib";
-import { Runner } from '@openai/agents';
+import { InputGuardrailTripwireTriggered, Runner, RunToolCallItem, } from '@openai/agents';
 export class AIAgentRunner {
     constructor(agent) {
         this.agent = agent;
@@ -17,9 +17,30 @@ export class AIAgentRunner {
                     modelSettings: { temperature: 0 },
                 });
                 const result = yield runner.run(this.agent, messages, { context });
-                return ((_a = result.finalOutput) === null || _a === void 0 ? void 0 : _a.messages) || [];
+                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.getExecutedNameTools(result);
+                return {
+                    messages: hasExit
+                        ? []
+                        : outputMessages.filter(message => message.type !== 'exit'),
+                    toolsExecuted,
+                    exit: hasExit,
+                    inputGuardrailTriggered: [],
+                    outputGuardrailTriggered: [],
+                };
             }
             catch (error) {
+                if (error instanceof InputGuardrailTripwireTriggered) {
+                    return {
+                        messages: [],
+                        toolsExecuted: [],
+                        exit: true,
+                        inputGuardrailTriggered: error.result.output.outputInfo,
+                        outputGuardrailTriggered: [],
+                    };
+                }
                 if (attempt > maxRetries) {
                     throw error;
                 }
@@ -27,5 +48,14 @@ export class AIAgentRunner {
             }
         });
     }
+    getExecutedNameTools(result) {
+        var _a;
+        return (((_a = result.newItems) === null || _a === void 0 ? void 0 : _a.filter(item => item instanceof RunToolCallItem).map((item) => {
+            if (item.rawItem.type === 'function_call') {
+                return item.rawItem.name;
+            }
+            return '';
+        }).filter((toolName) => toolName !== '')) || []);
+    }
 }
 //# sourceMappingURL=runner.js.map

package/lib/esm/runner.js.map

@@ -1 +1 @@
-{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/runner.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AASvC,MAAM,OAAO,aAAa;IAGxB,YAAY,KAAc;QACxB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;IACpB,CAAC;IAEK,GAAG,CACP,QAA+B,EAC/B,OAAgB,EAChB,aAAqB,CAAC;;YAEtB,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CAAA;QAC5D,CAAC;KAAA;IAEa,YAAY,CACxB,QAA+B,EAC/B,OAAgB,EAChB,UAAkB,EAClB,OAAe;;;YAEf,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;gBAClE,OAAO,CAAA,MAAA,MAAM,CAAC,WAAW,0CAAE,QAAQ,KAAI,EAAE,CAAA;aAC1C;YAAC,OAAO,KAAK,EAAE;gBACd,IAAI,OAAO,GAAG,UAAU,EAAE;oBACxB,MAAM,KAAK,CAAA;iBACZ;gBACD,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,GAAG,CAAC,CAAC,CAAA;aACrE;;KACF;CACF"}
+{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/runner.ts"],"names":[],"mappings":";AAAA,OAAO,EACL,+BAA+B,EAC/B,MAAM,EACN,eAAe,GAChB,MAAM,gBAAgB,CAAA;AAUvB,MAAM,OAAO,aAAa;IAGxB,YAAY,KAAc;QACxB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAA;IACpB,CAAC;IAEK,GAAG,CACP,QAA+B,EAC/B,OAAgB,EAChB,aAAqB,CAAC;;YAEtB,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC,CAAA;QAC5D,CAAC;KAAA;IAEa,YAAY,CACxB,QAA+B,EAC/B,OAAgB,EAChB,UAAkB,EAClB,OAAe;;;YAEf,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,oBAAoB,CAAC,MAAM,CAAC,CAAA;gBAEvD,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,uBAAuB,EAAE,EAAE;oBAC3B,wBAAwB,EAAE,EAAE;iBAC7B,CAAA;aACF;YAAC,OAAO,KAAK,EAAE;gBACd,IAAI,KAAK,YAAY,+BAA+B,EAAE;oBACpD,OAAO;wBACL,QAAQ,EAAE,EAAE;wBACZ,aAAa,EAAE,EAAE;wBACjB,IAAI,EAAE,IAAI;wBACV,uBAAuB,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU;wBACvD,wBAAwB,EAAE,EAAE;qBAC7B,CAAA;iBACF;gBACD,IAAI,OAAO,GAAG,UAAU,EAAE;oBACxB,MAAM,KAAK,CAAA;iBACZ;gBACD,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,GAAG,CAAC,CAAC,CAAA;aACrE;;KACF;IAEO,oBAAoB,CAAC,MAAM;;QACjC,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;YAC7B,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,KAAK,eAAe,EAAE;gBACzC,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAA;aACzB;YACD,OAAO,EAAE,CAAA;QACX,CAAC,EACA,MAAM,CAAC,CAAC,QAAgB,EAAE,EAAE,CAAC,QAAQ,KAAK,EAAE,CAAC,KAAI,EAAE,CACvD,CAAA;IACH,CAAC;CACF"}

package/lib/esm/structured-output/carousel.d.ts

@@ -1,20 +1,6 @@
+import { CarouselMessage } from '@botonic/core';
 import { z } from 'zod';
-import { BaseMessage } from './shared';
-interface CarouselElement {
-    title: string;
-    subtitle: string;
-    image: string;
-    button: {
-        text: string;
-        url: string;
-    };
-}
-export interface CarouselMessage extends BaseMessage {
-    type: 'carousel';
-    content: {
-        elements: CarouselElement[];
-    };
-}
+export type { CarouselMessage };
 export declare const CarouselSchema: z.ZodObject<{
     type: z.ZodEnum<["carousel"]>;
     content: z.ZodObject<{
@@ -97,4 +83,3 @@ export declare const CarouselSchema: z.ZodObject<{
         }[];
     };
 }>;
-export {};

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

@@ -1 +1 @@
-{"version":3,"file":"carousel.js","sourceRoot":"","sources":["../../../src/structured-output/carousel.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAkBvB,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,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;aAChB,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,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,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;aAChB,CAAC;SACH,CAAC,CACH;KACF,CAAC;CACH,CAAC;KACD,QAAQ,CAAC,kDAAkD,CAAC,CAAA"}

package/lib/esm/structured-output/exit.d.ts

@@ -1,8 +1,6 @@
+import { ExitMessage } from '@botonic/core';
 import z from 'zod';
-import { BaseMessage } from './shared';
-export interface ExitMessage extends BaseMessage {
-    type: 'exit';
-}
+export type { ExitMessage };
 export declare const ExitSchema: z.ZodObject<{
     type: z.ZodEnum<["exit"]>;
 }, "strip", z.ZodTypeAny, {

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

@@ -1 +1 @@
-{"version":3,"file":"exit.js","sourceRoot":"","sources":["../../../src/structured-output/exit.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAA;AAQnB,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,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"}

package/lib/esm/structured-output/index.d.ts

@@ -1,9 +1,5 @@
+import { OutputMessage } from '@botonic/core';
 import { z } from 'zod';
-import { CarouselMessage } from './carousel';
-import { ExitMessage } from './exit';
-import { TextMessage } from './text';
-import { TextWithButtonsMessage } from './text-with-buttons';
-export type OutputMessage = TextMessage | TextWithButtonsMessage | CarouselMessage | ExitMessage;
 export interface Output {
     messages: OutputMessage[];
 }

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

@@ -2,7 +2,7 @@ import { z } from 'zod';
 import { CarouselSchema } from './carousel';
 import { ExitSchema } from './exit';
 import { TextSchema } from './text';
-import { TextWithButtonsSchema, } from './text-with-buttons';
+import { TextWithButtonsSchema } from './text-with-buttons';
 export const OutputSchema = z
     .object({
     messages: z.array(z.union([TextSchema, TextWithButtonsSchema, CarouselSchema, ExitSchema])),

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/structured-output/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAmB,cAAc,EAAE,MAAM,YAAY,CAAA;AAC5D,OAAO,EAAe,UAAU,EAAE,MAAM,QAAQ,CAAA;AAChD,OAAO,EAAe,UAAU,EAAE,MAAM,QAAQ,CAAA;AAChD,OAAO,EAEL,qBAAqB,GACtB,MAAM,qBAAqB,CAAA;AAY5B,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,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"}

package/lib/esm/structured-output/text-with-buttons.d.ts

@@ -1,12 +1,6 @@
+import { TextWithButtonsMessage } from '@botonic/core';
 import z from 'zod';
-import { BaseMessage } from './shared';
-export interface TextWithButtonsMessage extends BaseMessage {
-    type: 'textWithButtons';
-    content: {
-        text: string;
-        buttons: string[];
-    };
-}
+export type { TextWithButtonsMessage };
 export declare const TextWithButtonsSchema: z.ZodObject<{
     type: z.ZodEnum<["textWithButtons"]>;
     content: z.ZodObject<{

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":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAA;AAYnB,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,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC7B,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,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,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;KAC7B,CAAC;CACH,CAAC;KACD,QAAQ,CACP,oEAAoE,CACrE,CAAA"}

package/lib/esm/structured-output/text.d.ts

@@ -1,11 +1,6 @@
+import { TextMessage } from '@botonic/core';
 import { z } from 'zod';
-import { BaseMessage } from './shared';
-export interface TextMessage extends BaseMessage {
-    type: 'text';
-    content: {
-        text: string;
-    };
-}
+export type { TextMessage };
 export declare const TextSchema: z.ZodObject<{
     type: z.ZodEnum<["text"]>;
     content: z.ZodObject<{

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

@@ -1 +1 @@
-{"version":3,"file":"text.js","sourceRoot":"","sources":["../../../src/structured-output/text.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAWvB,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,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"}

package/lib/esm/types.d.ts

@@ -1,6 +1,7 @@
-import { Agent, AgentInputItem, Tool as OpenAITool, RunContext as OpenAIRunContext } from '@openai/agents';
+import { AgenticOutputMessage, InferenceResponse, RunResult } from '@botonic/core';
+import { Agent, AgentInputItem, RunContext as OpenAIRunContext, Tool as OpenAITool } from '@openai/agents';
 import { ZodSchema } from 'zod';
-import { OutputMessage, OutputSchema } from './structured-output';
+import { OutputSchema } from './structured-output';
 export interface Context {
     authToken: string;
 }
@@ -11,6 +12,11 @@ export interface CustomTool {
     schema: ZodSchema;
     func: (input?: any, runContext?: RunContext) => Promise<any>;
 }
+export interface GuardrailRule {
+    name: string;
+    description: string;
+}
+export type ContactInfo = Record<string, string>;
 export type Tool = OpenAITool<Context>;
 export type AIAgent = Agent<Context, typeof OutputSchema>;
 export interface PluginAiAgentOptions {
@@ -23,6 +29,7 @@ export interface AiAgentArgs {
     activeTools?: {
         name: string;
     }[];
+    inputGuardrailRules?: GuardrailRule[];
 }
 export type AgenticInputMessage = AgentInputItem;
-export type AgenticOutputMessage = OutputMessage;
+export type { AgenticOutputMessage, InferenceResponse, RunResult };

package/package.json

@@ -1,19 +1,19 @@
 {
   "name": "@botonic/plugin-ai-agents",
-  "version": "0.37.1",
+  "version": "0.38.0-alpha.0",
   "main": "./lib/cjs/index.js",
   "module": "./lib/esm/index.js",
   "description": "Use AI Agents to generate your contents",
   "scripts": {
     "build": "rm -rf lib && ../../node_modules/.bin/tsc -p tsconfig.json && ../../node_modules/.bin/tsc -p tsconfig.esm.json",
     "build:watch": "npm run build -- --watch",
-    "test": "echo \"Error: no test specified\" && exit 1",
+    "test": "../../node_modules/.bin/jest --coverage",
     "prepublishOnly": "rm -rf lib && npm i && npm run build",
     "lint": "npm run lint_core -- --fix",
     "lint_core": "../../node_modules/.bin/eslint_d --cache --quiet 'src/**/*.ts*'"
   },
   "dependencies": {
-    "@botonic/core": "^0.37.0",
+    "@botonic/core": "0.38.0-alpha.0",
     "@openai/agents": "^0.0.10",
     "axios": "^1.10.0",
     "openai": "^5.8.3",

package/src/agent-builder.ts

@@ -1,18 +1,31 @@
-import { Agent } from '@openai/agents'
+import { Agent, InputGuardrail } from '@openai/agents'
 
+import { createInputGuardrail } from './guardrails'
 import { OutputSchema } from './structured-output'
 import { mandatoryTools } from './tools'
-import { AIAgent, Tool } from './types'
+import { AIAgent, ContactInfo, GuardrailRule, Tool } from './types'
 
 export class AIAgentBuilder {
   private name: string
   private instructions: string
   private tools: Tool[]
+  private inputGuardrails: InputGuardrail[]
 
-  constructor(name: string, instructions: string, tools: Tool[]) {
+  constructor(
+    name: string,
+    instructions: string,
+    tools: Tool[],
+    contactInfo: ContactInfo,
+    inputGuardrailRules: GuardrailRule[]
+  ) {
     this.name = name
-    this.instructions = this.addExtraInstructions(instructions)
+    this.instructions = this.addExtraInstructions(instructions, contactInfo)
     this.tools = this.addHubtypeTools(tools)
+    this.inputGuardrails = []
+    if (inputGuardrailRules.length > 0) {
+      const inputGuardrail = createInputGuardrail(inputGuardrailRules)
+      this.inputGuardrails.push(inputGuardrail)
+    }
   }
 
   build(): AIAgent {
@@ -21,11 +34,35 @@ export class AIAgentBuilder {
       instructions: this.instructions,
       tools: this.tools,
       outputType: OutputSchema,
+      inputGuardrails: this.inputGuardrails,
+      outputGuardrails: [],
     })
   }
 
-  private addExtraInstructions(instructions: string): string {
+  private addExtraInstructions(
+    initialInstructions: string,
+    contactInfo: ContactInfo
+  ): string {
+    const instructions = `<instructions>\n${initialInstructions}\n</instructions>`
+    const metadataInstructions = this.getMetadataInstructions()
+    const contactInfoInstructions = this.getContactInfoInstructions(contactInfo)
+    const outputInstructions = this.getOutputInstructions()
+    return `${instructions}\n\n${metadataInstructions}\n\n${contactInfoInstructions}\n\n${outputInstructions}`
+  }
+
+  private getContactInfoInstructions(contactInfo: ContactInfo): string {
+    const structuredContactInfo = Object.entries(contactInfo)
+      .map(([key, value]) => `${key}: ${value}`)
+      .join('\n')
+    return `<contact_info>\n${structuredContactInfo}\n</contact_info>`
+  }
+
+  private getMetadataInstructions(): string {
     const metadata = `Current Date: ${new Date().toISOString()}`
+    return `<metadata>\n${metadata}\n</metadata>`
+  }
+
+  private getOutputInstructions(): string {
     const example = {
       messages: [
         {
@@ -35,10 +72,9 @@ export class AIAgentBuilder {
           },
         },
       ],
-      numMessages: 1,
     }
     const output = `Return a JSON that follows the output schema provided. Never return multiple output schemas concatenated by a line break.\n<example>\n${JSON.stringify(example)}\n</example>`
-    return `<instructions>\n${instructions}\n</instructions>\n\n<metadata>\n${metadata}\n</metadata>\n\n<output>\n${output}\n</output>`
+    return `<output>\n${output}\n</output>`
   }
 
   private addHubtypeTools(tools: Tool[]): Tool[] {

package/src/guardrails/index.ts

@@ -0,0 +1 @@
+export * from './input'

package/src/guardrails/input.ts

@@ -0,0 +1,39 @@
+import { Agent, InputGuardrail, run, UserMessageItem } from '@openai/agents'
+import { z } from 'zod'
+
+import { GuardrailRule } from '../types'
+
+export function createInputGuardrail(rules: GuardrailRule[]): InputGuardrail {
+  const outputType = z.object(
+    Object.fromEntries(
+      rules.map(rule => [rule.name, z.boolean().describe(rule.description)])
+    )
+  )
+
+  const agent = new Agent({
+    name: 'InputGuardrail',
+    instructions:
+      'Check if the user triggers some of the following guardrails.',
+    outputType,
+  })
+
+  return {
+    name: 'InputGuardrail',
+    execute: async ({ input, context }) => {
+      const lastMessage = input[input.length - 1] as UserMessageItem
+      const result = await run(agent, [lastMessage], { context })
+      const finalOutput = result.finalOutput
+      if (finalOutput === undefined) {
+        throw new Error('Guardrail agent failed to produce output')
+      }
+      const triggered = Object.values(finalOutput).some(value => value === true)
+      const triggeredGuardrails = Object.keys(finalOutput).filter(
+        key => finalOutput[key] === true
+      )
+      return {
+        outputInfo: triggeredGuardrails,
+        tripwireTriggered: triggered,
+      }
+    },
+  }
+}

package/src/index.ts

@@ -8,10 +8,10 @@ import { setUpOpenAI } from './openai'
 import { AIAgentRunner } from './runner'
 import {
   AgenticInputMessage,
-  AgenticOutputMessage,
   AiAgentArgs,
   Context,
   CustomTool,
+  InferenceResponse,
   PluginAiAgentOptions,
   Tool,
 } from './types'
@@ -33,7 +33,7 @@ export default class BotonicPluginAiAgents implements Plugin {
   async getInference(
     request: BotContext,
     aiAgentArgs: AiAgentArgs
-  ): Promise<AgenticOutputMessage[] | undefined> {
+  ): Promise<InferenceResponse> {
     try {
       const authToken = isProd ? request.session._access_token : this.authToken
       if (!authToken) {
@@ -46,7 +46,9 @@ export default class BotonicPluginAiAgents implements Plugin {
       const agent = new AIAgentBuilder(
         aiAgentArgs.name,
         aiAgentArgs.instructions,
-        tools
+        tools,
+        request.session.user.contact_info || {},
+        aiAgentArgs.inputGuardrailRules || []
       ).build()
 
       const messages = await this.getMessages(request, authToken, 25)
@@ -58,6 +60,8 @@ export default class BotonicPluginAiAgents implements Plugin {
       return await runner.run(messages, context)
     } catch (error) {
       console.error('error plugin returns undefined', error)
+      // Here we can return a InferenceResponse as a exit
+      // but indicate that the inference failed
       return undefined
     }
   }

package/src/runner.ts

@@ -1,10 +1,15 @@
-import { Runner } from '@openai/agents'
+import {
+  InputGuardrailTripwireTriggered,
+  Runner,
+  RunToolCallItem,
+} from '@openai/agents'
 
 import {
   AgenticInputMessage,
   AgenticOutputMessage,
   AIAgent,
   Context,
+  RunResult,
 } from './types'
 
 export class AIAgentRunner {
@@ -18,7 +23,7 @@ export class AIAgentRunner {
     messages: AgenticInputMessage[],
     context: Context,
     maxRetries: number = 1
-  ): Promise<AgenticOutputMessage[]> {
+  ): Promise<RunResult> {
     return this.runWithRetry(messages, context, maxRetries, 1)
   }
 
@@ -27,18 +32,58 @@ export class AIAgentRunner {
     context: Context,
     maxRetries: number,
     attempt: number
-  ): Promise<AgenticOutputMessage[]> {
+  ): Promise<RunResult> {
     try {
       const runner = new Runner({
         modelSettings: { temperature: 0 },
       })
       const result = await runner.run(this.agent, messages, { context })
-      return result.finalOutput?.messages || []
+
+      const outputMessages = result.finalOutput?.messages || []
+      const hasExit =
+        outputMessages.length === 0 ||
+        outputMessages.some(message => message.type === 'exit')
+      const toolsExecuted = this.getExecutedNameTools(result)
+
+      return {
+        messages: hasExit
+          ? []
+          : (outputMessages.filter(
+              message => message.type !== 'exit'
+            ) as AgenticOutputMessage[]),
+        toolsExecuted,
+        exit: hasExit,
+        inputGuardrailTriggered: [],
+        outputGuardrailTriggered: [],
+      }
     } catch (error) {
+      if (error instanceof InputGuardrailTripwireTriggered) {
+        return {
+          messages: [],
+          toolsExecuted: [],
+          exit: true,
+          inputGuardrailTriggered: error.result.output.outputInfo,
+          outputGuardrailTriggered: [],
+        }
+      }
       if (attempt > maxRetries) {
         throw error
       }
       return this.runWithRetry(messages, context, maxRetries, attempt + 1)
     }
   }
+
+  private getExecutedNameTools(result) {
+    return (
+      result.newItems
+        ?.filter(item => item instanceof RunToolCallItem)
+        .map((item: RunToolCallItem) => {
+          if (item.rawItem.type === 'function_call') {
+            return item.rawItem.name
+          }
+          return ''
+        })
+        .filter((toolName: string) => toolName !== '') || []
+    )
+  }
 }

package/src/structured-output/carousel.ts

@@ -1,20 +1,7 @@
+import { CarouselMessage } from '@botonic/core'
 import { z } from 'zod'
 
-import { BaseMessage } from './shared'
-
-interface CarouselElement {
-  title: string
-  subtitle: string
-  image: string
-  button: { text: string; url: string }
-}
-
-export interface CarouselMessage extends BaseMessage {
-  type: 'carousel'
-  content: {
-    elements: CarouselElement[]
-  }
-}
+export type { CarouselMessage }
 
 export const CarouselSchema = z
   .object({

package/src/structured-output/exit.ts

@@ -1,10 +1,7 @@
+import { ExitMessage } from '@botonic/core'
 import z from 'zod'
 
-import { BaseMessage } from './shared'
-
-export interface ExitMessage extends BaseMessage {
-  type: 'exit'
-}
+export type { ExitMessage }
 
 export const ExitSchema = z
   .object({

package/src/structured-output/index.ts

@@ -1,18 +1,10 @@
+import { OutputMessage } from '@botonic/core'
 import { z } from 'zod'
 
-import { CarouselMessage, CarouselSchema } from './carousel'
-import { ExitMessage, ExitSchema } from './exit'
-import { TextMessage, TextSchema } from './text'
-import {
-  TextWithButtonsMessage,
-  TextWithButtonsSchema,
-} from './text-with-buttons'
-
-export type OutputMessage =
-  | TextMessage
-  | TextWithButtonsMessage
-  | CarouselMessage
-  | ExitMessage
+import { CarouselSchema } from './carousel'
+import { ExitSchema } from './exit'
+import { TextSchema } from './text'
+import { TextWithButtonsSchema } from './text-with-buttons'
 
 export interface Output {
   messages: OutputMessage[]

package/src/structured-output/text-with-buttons.ts

@@ -1,14 +1,7 @@
+import { TextWithButtonsMessage } from '@botonic/core'
 import z from 'zod'
 
-import { BaseMessage } from './shared'
-
-export interface TextWithButtonsMessage extends BaseMessage {
-  type: 'textWithButtons'
-  content: {
-    text: string
-    buttons: string[]
-  }
-}
+export type { TextWithButtonsMessage }
 
 export const TextWithButtonsSchema = z
   .object({

package/src/structured-output/text.ts

@@ -1,13 +1,7 @@
+import { TextMessage } from '@botonic/core'
 import { z } from 'zod'
 
-import { BaseMessage } from './shared'
-
-export interface TextMessage extends BaseMessage {
-  type: 'text'
-  content: {
-    text: string
-  }
-}
+export type { TextMessage }
 
 export const TextSchema = z
   .object({

package/src/types.ts

@@ -1,12 +1,17 @@
+import {
+  AgenticOutputMessage,
+  InferenceResponse,
+  RunResult,
+} from '@botonic/core'
 import {
   Agent,
   AgentInputItem,
-  Tool as OpenAITool,
   RunContext as OpenAIRunContext,
+  Tool as OpenAITool,
 } from '@openai/agents'
 import { ZodSchema } from 'zod'
 
-import { OutputMessage, OutputSchema } from './structured-output'
+import { OutputSchema } from './structured-output'
 
 export interface Context {
   authToken: string
@@ -21,9 +26,15 @@ export interface CustomTool {
   func: (input?: any, runContext?: RunContext) => Promise<any>
 }
 
+export interface GuardrailRule {
+  name: string
+  description: string
+}
+
+export type ContactInfo = Record<string, string>
+
 export type Tool = OpenAITool<Context>
 export type AIAgent = Agent<Context, typeof OutputSchema>
-
 export interface PluginAiAgentOptions {
   authToken?: string
   customTools?: CustomTool[]
@@ -33,7 +44,9 @@ export interface AiAgentArgs {
   name: string
   instructions: string
   activeTools?: { name: string }[]
+  inputGuardrailRules?: GuardrailRule[]
 }
 
 export type AgenticInputMessage = AgentInputItem
-export type AgenticOutputMessage = OutputMessage
+
+export type { AgenticOutputMessage, InferenceResponse, RunResult }

package/lib/cjs/structured-output/shared.d.ts

@@ -1,3 +0,0 @@
-export interface BaseMessage {
-    type: 'text' | 'textWithButtons' | 'carousel' | 'exit';
-}

package/lib/cjs/structured-output/shared.js

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

package/lib/cjs/structured-output/shared.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../src/structured-output/shared.ts"],"names":[],"mappings":""}

package/lib/esm/structured-output/shared.d.ts

@@ -1,3 +0,0 @@
-export interface BaseMessage {
-    type: 'text' | 'textWithButtons' | 'carousel' | 'exit';
-}