@botonic/plugin-ai-agents 0.36.0-alpha.8 → 0.36.0-alpha.9

package/lib/cjs/agent-builder.d.ts

@@ -0,0 +1,10 @@
+import { AIAgent, Tool } from './types';
+export declare class AIAgentBuilder {
+    private name;
+    private instructions;
+    private tools;
+    constructor(name: string, instructions: string, tools: Tool[]);
+    build(): AIAgent;
+    private addExtraInstructions;
+    private addHubtypeTools;
+}

package/lib/cjs/agent-builder.js

@@ -0,0 +1,43 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIAgentBuilder = void 0;
+const agents_1 = require("@openai/agents");
+const structured_output_1 = require("./structured-output");
+const tools_1 = require("./tools");
+class AIAgentBuilder {
+    constructor(name, instructions, tools) {
+        this.name = name;
+        this.instructions = this.addExtraInstructions(instructions);
+        this.tools = this.addHubtypeTools(tools);
+    }
+    build() {
+        return new agents_1.Agent({
+            name: this.name,
+            instructions: this.instructions,
+            tools: this.tools,
+            outputType: structured_output_1.OutputSchema,
+        });
+    }
+    addExtraInstructions(instructions) {
+        const metadata = `Current Date: ${new Date().toISOString()}`;
+        const example = {
+            messages: [
+                {
+                    type: 'text',
+                    content: {
+                        text: 'Hello, how can I help you today?',
+                    },
+                },
+            ],
+            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>`;
+    }
+    addHubtypeTools(tools) {
+        const hubtypeTools = [...tools_1.mandatoryTools];
+        return [...hubtypeTools, ...tools];
+    }
+}
+exports.AIAgentBuilder = AIAgentBuilder;
+//# sourceMappingURL=agent-builder.js.map

package/lib/cjs/agent-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-builder.js","sourceRoot":"","sources":["../../src/agent-builder.ts"],"names":[],"mappings":";;;AAAA,2CAAsC;AAEtC,2DAAkD;AAClD,mCAAwC;AAGxC,MAAa,cAAc;IAKzB,YAAY,IAAY,EAAE,YAAoB,EAAE,KAAa;QAC3D,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAA;QAC3D,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAA;IAC1C,CAAC;IAED,KAAK;QACH,OAAO,IAAI,cAAK,CAAC;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU,EAAE,gCAAY;SACzB,CAAC,CAAA;IACJ,CAAC;IAEO,oBAAoB,CAAC,YAAoB;QAC/C,MAAM,QAAQ,GAAG,iBAAiB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAA;QAC5D,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP,IAAI,EAAE,kCAAkC;qBACzC;iBACF;aACF;YACD,WAAW,EAAE,CAAC;SACf,CAAA;QACD,MAAM,MAAM,GAAG,yIAAyI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,cAAc,CAAA;QAC7L,OAAO,mBAAmB,YAAY,oCAAoC,QAAQ,8BAA8B,MAAM,aAAa,CAAA;IACrI,CAAC;IAEO,eAAe,CAAC,KAAa;QACnC,MAAM,YAAY,GAAW,CAAC,GAAG,sBAAc,CAAC,CAAA;QAChD,OAAO,CAAC,GAAG,YAAY,EAAE,GAAG,KAAK,CAAC,CAAA;IACpC,CAAC;CACF;AAzCD,wCAyCC"}

package/lib/cjs/index.d.ts

@@ -2,11 +2,11 @@ import { BotContext, Plugin } from '@botonic/core';
 import { AgenticOutputMessage, AiAgentArgs, CustomTool, PluginAiAgentOptions } from './types';
 export default class BotonicPluginAiAgents implements Plugin {
     private readonly authToken?;
-    customTools: CustomTool[];
+    toolDefinitions: CustomTool[];
     constructor(options?: PluginAiAgentOptions);
     pre(): void;
     getInference(request: BotContext, aiAgentArgs: AiAgentArgs): Promise<AgenticOutputMessage[] | undefined>;
     private getMessages;
-    private createTools;
+    private buildTools;
 }
 export * from './types';

package/lib/cjs/index.js

@@ -2,32 +2,37 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const tslib_1 = require("tslib");
 const agents_1 = require("@openai/agents");
+const agent_builder_1 = require("./agent-builder");
 const constants_1 = require("./constants");
 const hubtype_api_client_1 = require("./hubtype-api-client");
+const openai_1 = require("./openai");
 const runner_1 = require("./runner");
 class BotonicPluginAiAgents {
     constructor(options) {
-        this.customTools = [];
+        this.toolDefinitions = [];
+        (0, openai_1.setUpOpenAI)();
         this.authToken = options === null || options === void 0 ? void 0 : options.authToken;
-        this.customTools = (options === null || options === void 0 ? void 0 : options.customTools) || [];
+        this.toolDefinitions = (options === null || options === void 0 ? void 0 : options.customTools) || [];
     }
     pre() {
         return;
     }
     getInference(request, aiAgentArgs) {
+        var _a;
         return tslib_1.__awaiter(this, void 0, void 0, function* () {
             try {
                 const authToken = constants_1.isProd ? request.session._access_token : this.authToken;
                 if (!authToken) {
                     throw new Error('Auth token is required');
                 }
+                const tools = this.buildTools(((_a = aiAgentArgs.activeTools) === null || _a === void 0 ? void 0 : _a.map(tool => tool.name)) || []);
+                const agent = new agent_builder_1.AIAgentBuilder(aiAgentArgs.name, aiAgentArgs.instructions, tools).build();
                 const messages = yield this.getMessages(request, authToken, 10);
-                const availableTools = this.customTools.filter(tool => { var _a; return (_a = aiAgentArgs.activeTools) === null || _a === void 0 ? void 0 : _a.map(tool => tool.name).includes(tool.name); });
-                const tools = this.createTools(availableTools);
-                const runner = new runner_1.AIAgentRunner(aiAgentArgs.name, aiAgentArgs.instructions, tools);
-                const output = yield runner.run(messages);
-                console.log('\n\nOutput:', output);
-                return output;
+                const context = {
+                    authToken,
+                };
+                const runner = new runner_1.AIAgentRunner(agent);
+                return yield runner.run(messages, context);
             }
             catch (error) {
                 console.error('error plugin returns undefined', error);
@@ -45,13 +50,14 @@ class BotonicPluginAiAgents {
             return yield hubtypeClient.getLocalMessages(memoryLength);
         });
     }
-    createTools(customTools) {
-        return customTools.map(customTool => {
+    buildTools(activeToolNames) {
+        const availableTools = this.toolDefinitions.filter(tool => activeToolNames.includes(tool.name));
+        return availableTools.map(toolDefinition => {
             return (0, agents_1.tool)({
-                name: customTool.name,
-                description: customTool.description,
-                parameters: customTool.schema,
-                execute: customTool.func,
+                name: toolDefinition.name,
+                description: toolDefinition.description,
+                parameters: toolDefinition.schema,
+                execute: toolDefinition.func,
             });
         });
     }

package/lib/cjs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AACA,2CAAqC;AAErC,2CAAoC;AACpC,6DAAuD;AACvD,qCAAwC;AASxC,MAAqB,qBAAqB;IAIxC,YAAY,OAA8B;QAFnC,gBAAW,GAAiB,EAAE,CAAA;QAGnC,IAAI,CAAC,SAAS,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,CAAA;QACnC,IAAI,CAAC,WAAW,GAAG,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,KAAI,EAAE,CAAA;IAC/C,CAAC;IAED,GAAG;QACD,OAAM;IACR,CAAC;IAEK,YAAY,CAChB,OAAmB,EACnB,WAAwB;;YAExB,IAAI;gBACF,MAAM,SAAS,GAAG,kBAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;gBACzE,IAAI,CAAC,SAAS,EAAE;oBACd,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAA;iBAC1C;gBAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE,CAAC,CAAA;gBAC/D,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,WACpD,OAAA,MAAA,WAAW,CAAC,WAAW,0CAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,EAAA,CACpE,CAAA;gBACD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAA;gBAE9C,MAAM,MAAM,GAAG,IAAI,sBAAa,CAC9B,WAAW,CAAC,IAAI,EAChB,WAAW,CAAC,YAAY,EACxB,KAAK,CACN,CAAA;gBACD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;gBACzC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,CAAA;gBAElC,OAAO,MAAM,CAAA;aACd;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAA;gBACtD,OAAO,SAAS,CAAA;aACjB;QACH,CAAC;KAAA;IAEa,WAAW,CACvB,OAAmB,EACnB,SAAiB,EACjB,YAAoB;;YAEpB,IAAI,kBAAM,EAAE;gBACV,MAAM,aAAa,GAAG,IAAI,qCAAgB,CAAC,SAAS,CAAC,CAAA;gBACrD,OAAO,MAAM,aAAa,CAAC,WAAW,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;aAC9D;YACD,MAAM,aAAa,GAAG,IAAI,qCAAgB,CAAC,SAAS,CAAC,CAAA;YACrD,OAAO,MAAM,aAAa,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAA;QAC3D,CAAC;KAAA;IAEO,WAAW,CAAC,WAAyB;QAC3C,OAAO,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;YAClC,OAAO,IAAA,aAAI,EAAC;gBACV,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,UAAU,EAAE,UAAU,CAAC,MAAM;gBAC7B,OAAO,EAAE,UAAU,CAAC,IAAI;aACzB,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;CACF;AAnED,wCAmEC;AAED,kDAAuB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AACA,2CAAqC;AAErC,mDAAgD;AAChD,2CAAoC;AACpC,6DAAuD;AACvD,qCAAsC;AACtC,qCAAwC;AAWxC,MAAqB,qBAAqB;IAIxC,YAAY,OAA8B;QAFnC,oBAAe,GAAiB,EAAE,CAAA;QAGvC,IAAA,oBAAW,GAAE,CAAA;QACb,IAAI,CAAC,SAAS,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,CAAA;QACnC,IAAI,CAAC,eAAe,GAAG,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,KAAI,EAAE,CAAA;IACnD,CAAC;IAED,GAAG;QACD,OAAM;IACR,CAAC;IAEK,YAAY,CAChB,OAAmB,EACnB,WAAwB;;;YAExB,IAAI;gBACF,MAAM,SAAS,GAAG,kBAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;gBACzE,IAAI,CAAC,SAAS,EAAE;oBACd,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAA;iBAC1C;gBAED,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAC3B,CAAA,MAAA,WAAW,CAAC,WAAW,0CAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAI,EAAE,CACtD,CAAA;gBACD,MAAM,KAAK,GAAG,IAAI,8BAAc,CAC9B,WAAW,CAAC,IAAI,EAChB,WAAW,CAAC,YAAY,EACxB,KAAK,CACN,CAAC,KAAK,EAAE,CAAA;gBAET,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE,CAAC,CAAA;gBAC/D,MAAM,OAAO,GAAY;oBACvB,SAAS;iBACV,CAAA;gBAED,MAAM,MAAM,GAAG,IAAI,sBAAa,CAAC,KAAK,CAAC,CAAA;gBACvC,OAAO,MAAM,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;aAC3C;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAA;gBACtD,OAAO,SAAS,CAAA;aACjB;;KACF;IAEa,WAAW,CACvB,OAAmB,EACnB,SAAiB,EACjB,YAAoB;;YAEpB,IAAI,kBAAM,EAAE;gBACV,MAAM,aAAa,GAAG,IAAI,qCAAgB,CAAC,SAAS,CAAC,CAAA;gBACrD,OAAO,MAAM,aAAa,CAAC,WAAW,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;aAC9D;YACD,MAAM,aAAa,GAAG,IAAI,qCAAgB,CAAC,SAAS,CAAC,CAAA;YACrD,OAAO,MAAM,aAAa,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAA;QAC3D,CAAC;KAAA;IAEO,UAAU,CAAC,eAAyB;QAC1C,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CACxD,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CACpC,CAAA;QACD,OAAO,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE;YACzC,OAAO,IAAA,aAAI,EAAoB;gBAC7B,IAAI,EAAE,cAAc,CAAC,IAAI;gBACzB,WAAW,EAAE,cAAc,CAAC,WAAW;gBACvC,UAAU,EAAE,cAAc,CAAC,MAAM;gBACjC,OAAO,EAAE,cAAc,CAAC,IAAI;aAC7B,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;CACF;AAxED,wCAwEC;AAED,kDAAuB"}

package/lib/cjs/openai.d.ts

@@ -0,0 +1 @@
+export declare function setUpOpenAI(): void;

package/lib/cjs/openai.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setUpOpenAI = void 0;
+const agents_1 = require("@openai/agents");
+const openai_1 = require("openai");
+const constants_1 = require("./constants");
+function setUpOpenAI() {
+    setAzureOpenAIClient();
+    (0, agents_1.setOpenAIAPI)('chat_completions');
+    (0, agents_1.setTracingDisabled)(true);
+}
+exports.setUpOpenAI = setUpOpenAI;
+function setAzureOpenAIClient() {
+    const client = new openai_1.AzureOpenAI({
+        apiKey: constants_1.AZURE_OPENAI_API_KEY,
+        apiVersion: constants_1.AZURE_OPENAI_API_VERSION,
+        deployment: constants_1.AZURE_OPENAI_API_DEPLOYMENT_NAME,
+        baseURL: constants_1.AZURE_OPENAI_API_BASE,
+        dangerouslyAllowBrowser: !constants_1.isProd,
+    });
+    (0, agents_1.setDefaultOpenAIClient)(client);
+}
+//# sourceMappingURL=openai.js.map

package/lib/cjs/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/openai.ts"],"names":[],"mappings":";;;AAAA,2CAIuB;AACvB,mCAAoC;AACpC,2CAMoB;AAEpB,SAAgB,WAAW;IACzB,oBAAoB,EAAE,CAAA;IACtB,IAAA,qBAAY,EAAC,kBAAkB,CAAC,CAAA;IAChC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAA;AAC1B,CAAC;AAJD,kCAIC;AAED,SAAS,oBAAoB;IAC3B,MAAM,MAAM,GAAG,IAAI,oBAAW,CAAC;QAC7B,MAAM,EAAE,gCAAoB;QAC5B,UAAU,EAAE,oCAAwB;QACpC,UAAU,EAAE,4CAAgC;QAC5C,OAAO,EAAE,iCAAqB;QAC9B,uBAAuB,EAAE,CAAC,kBAAM;KACjC,CAAC,CAAA;IACF,IAAA,+BAAsB,EAAC,MAAM,CAAC,CAAA;AAChC,CAAC"}

package/lib/cjs/runner.d.ts

@@ -1,10 +1,7 @@
-import { Tool } from '@openai/agents';
-import { OutputMessage } from './structured-output';
-import { AgenticInputMessage } from './types';
+import { AIAgent, AgenticInputMessage, AgenticOutputMessage, Context } from './types';
 export declare class AIAgentRunner {
     private agent;
-    constructor(name: string, instructions: string, tools: Tool[]);
-    run(messages: AgenticInputMessage[], maxRetries?: number): Promise<OutputMessage[]>;
+    constructor(agent: AIAgent);
+    run(messages: AgenticInputMessage[], context: Context, maxRetries?: number): Promise<AgenticOutputMessage[]>;
     private runWithRetry;
-    private addExtraInstructions;
 }

package/lib/cjs/runner.js

@@ -3,73 +3,33 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.AIAgentRunner = void 0;
 const tslib_1 = require("tslib");
 const agents_1 = require("@openai/agents");
-const openai_1 = require("openai");
-const constants_1 = require("./constants");
-const structured_output_1 = require("./structured-output");
 class AIAgentRunner {
-    constructor(name, instructions, tools) {
-        if (constants_1.AZURE_OPENAI_API_KEY) {
-            const client = new openai_1.AzureOpenAI({
-                apiKey: constants_1.AZURE_OPENAI_API_KEY,
-                apiVersion: constants_1.AZURE_OPENAI_API_VERSION,
-                deployment: constants_1.AZURE_OPENAI_API_DEPLOYMENT_NAME,
-                baseURL: constants_1.AZURE_OPENAI_API_BASE,
-                dangerouslyAllowBrowser: !constants_1.isProd,
-            });
-            (0, agents_1.setDefaultOpenAIClient)(client);
-            (0, agents_1.setOpenAIAPI)('chat_completions');
-            (0, agents_1.setTracingDisabled)(true);
-        }
-        this.agent = new agents_1.Agent({
-            name: name,
-            instructions: this.addExtraInstructions(instructions),
-            tools: tools,
-            outputType: structured_output_1.OutputSchema,
-        });
+    constructor(agent) {
+        this.agent = agent;
     }
-    run(messages, maxRetries = 1) {
+    run(messages, context, maxRetries = 1) {
         return tslib_1.__awaiter(this, void 0, void 0, function* () {
-            return this.runWithRetry(messages, maxRetries, 1);
+            return this.runWithRetry(messages, context, maxRetries, 1);
         });
     }
-    runWithRetry(messages, maxRetries, attempt) {
+    runWithRetry(messages, context, maxRetries, attempt) {
         var _a;
         return tslib_1.__awaiter(this, void 0, void 0, function* () {
             try {
                 const runner = new agents_1.Runner({
                     modelSettings: { temperature: 0 },
                 });
-                const result = yield runner.run(this.agent, messages);
+                const result = yield runner.run(this.agent, messages, { context });
                 return ((_a = result.finalOutput) === null || _a === void 0 ? void 0 : _a.messages) || [];
             }
             catch (error) {
-                if (!(error instanceof agents_1.ModelBehaviorError)) {
-                    throw error;
-                }
                 if (attempt > maxRetries) {
                     throw error;
                 }
-                console.warn(`AI Agent execution failed due to ModelBehaviorError. Retrying... (Attempt ${attempt}/${maxRetries})`);
-                return this.runWithRetry(messages, maxRetries, attempt + 1);
+                return this.runWithRetry(messages, context, maxRetries, attempt + 1);
             }
         });
     }
-    addExtraInstructions(instructions) {
-        const metadata = `Current Date: ${new Date().toISOString()}`;
-        const outputExample = JSON.stringify({
-            messages: [
-                {
-                    type: 'text',
-                    content: {
-                        text: 'Hello, how can I help you today?',
-                    },
-                },
-            ],
-            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${outputExample}\n</example>`;
-        return `<instructions>\n${instructions}\n</instructions>\n\n<metadata>\n${metadata}\n</metadata>\n\n<output>\n${output}\n</output>`;
-    }
 }
 exports.AIAgentRunner = AIAgentRunner;
 //# sourceMappingURL=runner.js.map

package/lib/cjs/runner.js.map

@@ -1 +1 @@
-{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/runner.ts"],"names":[],"mappings":";;;;AAAA,2CAQuB;AACvB,mCAAoC;AAEpC,2CAMoB;AACpB,2DAAiE;AAGjE,MAAa,aAAa;IAGxB,YAAY,IAAY,EAAE,YAAoB,EAAE,KAAa;QAC3D,IAAI,gCAAoB,EAAE;YACxB,MAAM,MAAM,GAAG,IAAI,oBAAW,CAAC;gBAC7B,MAAM,EAAE,gCAAoB;gBAC5B,UAAU,EAAE,oCAAwB;gBACpC,UAAU,EAAE,4CAAgC;gBAC5C,OAAO,EAAE,iCAAqB;gBAC9B,uBAAuB,EAAE,CAAC,kBAAM;aACjC,CAAC,CAAA;YACF,IAAA,+BAAsB,EAAC,MAAa,CAAC,CAAA;YACrC,IAAA,qBAAY,EAAC,kBAAkB,CAAC,CAAA;YAChC,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAA;SACzB;QAED,IAAI,CAAC,KAAK,GAAG,IAAI,cAAK,CAAC;YACrB,IAAI,EAAE,IAAI;YACV,YAAY,EAAE,IAAI,CAAC,oBAAoB,CAAC,YAAY,CAAC;YACrD,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,gCAAY;SACzB,CAAC,CAAA;IACJ,CAAC;IAEK,GAAG,CACP,QAA+B,EAC/B,aAAqB,CAAC;;YAEtB,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC,CAAA;QACnD,CAAC;KAAA;IAEa,YAAY,CACxB,QAA+B,EAC/B,UAAkB,EAClB,OAAe;;;YAEf,IAAI;gBACF,MAAM,MAAM,GAAG,IAAI,eAAM,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,CAAC,CAAA;gBACrD,OAAO,CAAA,MAAA,MAAM,CAAC,WAAW,0CAAE,QAAQ,KAAI,EAAE,CAAA;aAC1C;YAAC,OAAO,KAAK,EAAE;gBACd,IAAI,CAAC,CAAC,KAAK,YAAY,2BAAkB,CAAC,EAAE;oBAC1C,MAAM,KAAK,CAAA;iBACZ;gBACD,IAAI,OAAO,GAAG,UAAU,EAAE;oBACxB,MAAM,KAAK,CAAA;iBACZ;gBACD,OAAO,CAAC,IAAI,CACV,6EAA6E,OAAO,IAAI,UAAU,GAAG,CACtG,CAAA;gBACD,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,GAAG,CAAC,CAAC,CAAA;aAC5D;;KACF;IAEO,oBAAoB,CAAC,YAAoB;QAC/C,MAAM,QAAQ,GAAG,iBAAiB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAA;QAE5D,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC;YACnC,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP,IAAI,EAAE,kCAAkC;qBACzC;iBACF;aACF;YACD,WAAW,EAAE,CAAC;SACf,CAAC,CAAA;QACF,MAAM,MAAM,GAAG,yIAAyI,aAAa,cAAc,CAAA;QAEnL,OAAO,mBAAmB,YAAY,oCAAoC,QAAQ,8BAA8B,MAAM,aAAa,CAAA;IACrI,CAAC;CACF;AA3ED,sCA2EC"}
+{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/runner.ts"],"names":[],"mappings":";;;;AAAA,2CAAuC;AAQvC,MAAa,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,eAAM,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;AAlCD,sCAkCC"}

package/lib/cjs/tools/index.d.ts

@@ -0,0 +1,2 @@
+import { Tool } from '../types';
+export declare const mandatoryTools: Tool[];

package/lib/cjs/tools/index.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mandatoryTools = void 0;
+exports.mandatoryTools = [];
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/index.ts"],"names":[],"mappings":";;;AAEa,QAAA,cAAc,GAAW,EAAE,CAAA"}

package/lib/cjs/types.d.ts

@@ -1,15 +1,20 @@
-import { AgentInputItem } from '@openai/agents';
+import { Agent, AgentInputItem, RunContext, Tool as OpenAITool } from '@openai/agents';
 import { ZodObject } from 'zod';
-import { OutputMessage } from './structured-output';
-export interface PluginAiAgentOptions {
-    authToken?: string;
-    customTools?: CustomTool[];
+import { OutputMessage, OutputSchema } from './structured-output';
+export interface Context {
+    authToken: string;
 }
 export interface CustomTool {
     name: string;
     description: string;
     schema: ZodObject<any>;
-    func: (input?: any) => Promise<any>;
+    func: (input?: any, runContext?: RunContext<Context>) => Promise<any>;
+}
+export type Tool = OpenAITool<Context>;
+export type AIAgent = Agent<Context, typeof OutputSchema>;
+export interface PluginAiAgentOptions {
+    authToken?: string;
+    customTools?: CustomTool[];
 }
 export interface AiAgentArgs {
     name: string;

package/lib/esm/agent-builder.d.ts

@@ -0,0 +1,10 @@
+import { AIAgent, Tool } from './types';
+export declare class AIAgentBuilder {
+    private name;
+    private instructions;
+    private tools;
+    constructor(name: string, instructions: string, tools: Tool[]);
+    build(): AIAgent;
+    private addExtraInstructions;
+    private addHubtypeTools;
+}

package/lib/esm/agent-builder.js

@@ -0,0 +1,39 @@
+import { Agent } from '@openai/agents';
+import { OutputSchema } from './structured-output';
+import { mandatoryTools } from './tools';
+export class AIAgentBuilder {
+    constructor(name, instructions, tools) {
+        this.name = name;
+        this.instructions = this.addExtraInstructions(instructions);
+        this.tools = this.addHubtypeTools(tools);
+    }
+    build() {
+        return new Agent({
+            name: this.name,
+            instructions: this.instructions,
+            tools: this.tools,
+            outputType: OutputSchema,
+        });
+    }
+    addExtraInstructions(instructions) {
+        const metadata = `Current Date: ${new Date().toISOString()}`;
+        const example = {
+            messages: [
+                {
+                    type: 'text',
+                    content: {
+                        text: 'Hello, how can I help you today?',
+                    },
+                },
+            ],
+            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>`;
+    }
+    addHubtypeTools(tools) {
+        const hubtypeTools = [...mandatoryTools];
+        return [...hubtypeTools, ...tools];
+    }
+}
+//# sourceMappingURL=agent-builder.js.map

package/lib/esm/agent-builder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-builder.js","sourceRoot":"","sources":["../../src/agent-builder.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAA;AAEtC,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAA;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAA;AAGxC,MAAM,OAAO,cAAc;IAKzB,YAAY,IAAY,EAAE,YAAoB,EAAE,KAAa;QAC3D,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAA;QAC3D,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAA;IAC1C,CAAC;IAED,KAAK;QACH,OAAO,IAAI,KAAK,CAAC;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU,EAAE,YAAY;SACzB,CAAC,CAAA;IACJ,CAAC;IAEO,oBAAoB,CAAC,YAAoB;QAC/C,MAAM,QAAQ,GAAG,iBAAiB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAA;QAC5D,MAAM,OAAO,GAAG;YACd,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP,IAAI,EAAE,kCAAkC;qBACzC;iBACF;aACF;YACD,WAAW,EAAE,CAAC;SACf,CAAA;QACD,MAAM,MAAM,GAAG,yIAAyI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,cAAc,CAAA;QAC7L,OAAO,mBAAmB,YAAY,oCAAoC,QAAQ,8BAA8B,MAAM,aAAa,CAAA;IACrI,CAAC;IAEO,eAAe,CAAC,KAAa;QACnC,MAAM,YAAY,GAAW,CAAC,GAAG,cAAc,CAAC,CAAA;QAChD,OAAO,CAAC,GAAG,YAAY,EAAE,GAAG,KAAK,CAAC,CAAA;IACpC,CAAC;CACF"}

package/lib/esm/index.d.ts

@@ -2,11 +2,11 @@ import { BotContext, Plugin } from '@botonic/core';
 import { AgenticOutputMessage, AiAgentArgs, CustomTool, PluginAiAgentOptions } from './types';
 export default class BotonicPluginAiAgents implements Plugin {
     private readonly authToken?;
-    customTools: CustomTool[];
+    toolDefinitions: CustomTool[];
     constructor(options?: PluginAiAgentOptions);
     pre(): void;
     getInference(request: BotContext, aiAgentArgs: AiAgentArgs): Promise<AgenticOutputMessage[] | undefined>;
     private getMessages;
-    private createTools;
+    private buildTools;
 }
 export * from './types';

package/lib/esm/index.js

@@ -1,31 +1,36 @@
 import { __awaiter } from "tslib";
 import { tool } from '@openai/agents';
+import { AIAgentBuilder } from './agent-builder';
 import { isProd } from './constants';
 import { HubtypeApiClient } from './hubtype-api-client';
+import { setUpOpenAI } from './openai';
 import { AIAgentRunner } from './runner';
 export default class BotonicPluginAiAgents {
     constructor(options) {
-        this.customTools = [];
+        this.toolDefinitions = [];
+        setUpOpenAI();
         this.authToken = options === null || options === void 0 ? void 0 : options.authToken;
-        this.customTools = (options === null || options === void 0 ? void 0 : options.customTools) || [];
+        this.toolDefinitions = (options === null || options === void 0 ? void 0 : options.customTools) || [];
     }
     pre() {
         return;
     }
     getInference(request, aiAgentArgs) {
+        var _a;
         return __awaiter(this, void 0, void 0, function* () {
             try {
                 const authToken = isProd ? request.session._access_token : this.authToken;
                 if (!authToken) {
                     throw new Error('Auth token is required');
                 }
+                const tools = this.buildTools(((_a = aiAgentArgs.activeTools) === null || _a === void 0 ? void 0 : _a.map(tool => tool.name)) || []);
+                const agent = new AIAgentBuilder(aiAgentArgs.name, aiAgentArgs.instructions, tools).build();
                 const messages = yield this.getMessages(request, authToken, 10);
-                const availableTools = this.customTools.filter(tool => { var _a; return (_a = aiAgentArgs.activeTools) === null || _a === void 0 ? void 0 : _a.map(tool => tool.name).includes(tool.name); });
-                const tools = this.createTools(availableTools);
-                const runner = new AIAgentRunner(aiAgentArgs.name, aiAgentArgs.instructions, tools);
-                const output = yield runner.run(messages);
-                console.log('\n\nOutput:', output);
-                return output;
+                const context = {
+                    authToken,
+                };
+                const runner = new AIAgentRunner(agent);
+                return yield runner.run(messages, context);
             }
             catch (error) {
                 console.error('error plugin returns undefined', error);
@@ -43,13 +48,14 @@ export default class BotonicPluginAiAgents {
             return yield hubtypeClient.getLocalMessages(memoryLength);
         });
     }
-    createTools(customTools) {
-        return customTools.map(customTool => {
+    buildTools(activeToolNames) {
+        const availableTools = this.toolDefinitions.filter(tool => activeToolNames.includes(tool.name));
+        return availableTools.map(toolDefinition => {
             return tool({
-                name: customTool.name,
-                description: customTool.description,
-                parameters: customTool.schema,
-                execute: customTool.func,
+                name: toolDefinition.name,
+                description: toolDefinition.description,
+                parameters: toolDefinition.schema,
+                execute: toolDefinition.func,
             });
         });
     }

package/lib/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAA;AAErC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AACpC,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAA;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAA;AASxC,MAAM,CAAC,OAAO,OAAO,qBAAqB;IAIxC,YAAY,OAA8B;QAFnC,gBAAW,GAAiB,EAAE,CAAA;QAGnC,IAAI,CAAC,SAAS,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,CAAA;QACnC,IAAI,CAAC,WAAW,GAAG,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,KAAI,EAAE,CAAA;IAC/C,CAAC;IAED,GAAG;QACD,OAAM;IACR,CAAC;IAEK,YAAY,CAChB,OAAmB,EACnB,WAAwB;;YAExB,IAAI;gBACF,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;gBACzE,IAAI,CAAC,SAAS,EAAE;oBACd,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAA;iBAC1C;gBAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE,CAAC,CAAA;gBAC/D,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,WACpD,OAAA,MAAA,WAAW,CAAC,WAAW,0CAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA,EAAA,CACpE,CAAA;gBACD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAA;gBAE9C,MAAM,MAAM,GAAG,IAAI,aAAa,CAC9B,WAAW,CAAC,IAAI,EAChB,WAAW,CAAC,YAAY,EACxB,KAAK,CACN,CAAA;gBACD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;gBACzC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,CAAA;gBAElC,OAAO,MAAM,CAAA;aACd;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAA;gBACtD,OAAO,SAAS,CAAA;aACjB;QACH,CAAC;KAAA;IAEa,WAAW,CACvB,OAAmB,EACnB,SAAiB,EACjB,YAAoB;;YAEpB,IAAI,MAAM,EAAE;gBACV,MAAM,aAAa,GAAG,IAAI,gBAAgB,CAAC,SAAS,CAAC,CAAA;gBACrD,OAAO,MAAM,aAAa,CAAC,WAAW,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;aAC9D;YACD,MAAM,aAAa,GAAG,IAAI,gBAAgB,CAAC,SAAS,CAAC,CAAA;YACrD,OAAO,MAAM,aAAa,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAA;QAC3D,CAAC;KAAA;IAEO,WAAW,CAAC,WAAyB;QAC3C,OAAO,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;YAClC,OAAO,IAAI,CAAC;gBACV,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,UAAU,EAAE,UAAU,CAAC,MAAM;gBAC7B,OAAO,EAAE,UAAU,CAAC,IAAI;aACzB,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;CACF;AAED,cAAc,SAAS,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAA;AAErC,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAA;AAChD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AACpC,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAA;AACvD,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAA;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAA;AAWxC,MAAM,CAAC,OAAO,OAAO,qBAAqB;IAIxC,YAAY,OAA8B;QAFnC,oBAAe,GAAiB,EAAE,CAAA;QAGvC,WAAW,EAAE,CAAA;QACb,IAAI,CAAC,SAAS,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,CAAA;QACnC,IAAI,CAAC,eAAe,GAAG,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW,KAAI,EAAE,CAAA;IACnD,CAAC;IAED,GAAG;QACD,OAAM;IACR,CAAC;IAEK,YAAY,CAChB,OAAmB,EACnB,WAAwB;;;YAExB,IAAI;gBACF,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;gBACzE,IAAI,CAAC,SAAS,EAAE;oBACd,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAA;iBAC1C;gBAED,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAC3B,CAAA,MAAA,WAAW,CAAC,WAAW,0CAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,KAAI,EAAE,CACtD,CAAA;gBACD,MAAM,KAAK,GAAG,IAAI,cAAc,CAC9B,WAAW,CAAC,IAAI,EAChB,WAAW,CAAC,YAAY,EACxB,KAAK,CACN,CAAC,KAAK,EAAE,CAAA;gBAET,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,EAAE,CAAC,CAAA;gBAC/D,MAAM,OAAO,GAAY;oBACvB,SAAS;iBACV,CAAA;gBAED,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,KAAK,CAAC,CAAA;gBACvC,OAAO,MAAM,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;aAC3C;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAA;gBACtD,OAAO,SAAS,CAAA;aACjB;;KACF;IAEa,WAAW,CACvB,OAAmB,EACnB,SAAiB,EACjB,YAAoB;;YAEpB,IAAI,MAAM,EAAE;gBACV,MAAM,aAAa,GAAG,IAAI,gBAAgB,CAAC,SAAS,CAAC,CAAA;gBACrD,OAAO,MAAM,aAAa,CAAC,WAAW,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;aAC9D;YACD,MAAM,aAAa,GAAG,IAAI,gBAAgB,CAAC,SAAS,CAAC,CAAA;YACrD,OAAO,MAAM,aAAa,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAA;QAC3D,CAAC;KAAA;IAEO,UAAU,CAAC,eAAyB;QAC1C,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CACxD,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CACpC,CAAA;QACD,OAAO,cAAc,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE;YACzC,OAAO,IAAI,CAAoB;gBAC7B,IAAI,EAAE,cAAc,CAAC,IAAI;gBACzB,WAAW,EAAE,cAAc,CAAC,WAAW;gBACvC,UAAU,EAAE,cAAc,CAAC,MAAM;gBACjC,OAAO,EAAE,cAAc,CAAC,IAAI;aAC7B,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;CACF;AAED,cAAc,SAAS,CAAA"}

package/lib/esm/openai.d.ts

@@ -0,0 +1 @@
+export declare function setUpOpenAI(): void;

package/lib/esm/openai.js

@@ -0,0 +1,19 @@
+import { setDefaultOpenAIClient, setOpenAIAPI, setTracingDisabled, } from '@openai/agents';
+import { AzureOpenAI } from 'openai';
+import { AZURE_OPENAI_API_BASE, AZURE_OPENAI_API_DEPLOYMENT_NAME, AZURE_OPENAI_API_KEY, AZURE_OPENAI_API_VERSION, isProd, } from './constants';
+export function setUpOpenAI() {
+    setAzureOpenAIClient();
+    setOpenAIAPI('chat_completions');
+    setTracingDisabled(true);
+}
+function setAzureOpenAIClient() {
+    const client = new AzureOpenAI({
+        apiKey: AZURE_OPENAI_API_KEY,
+        apiVersion: AZURE_OPENAI_API_VERSION,
+        deployment: AZURE_OPENAI_API_DEPLOYMENT_NAME,
+        baseURL: AZURE_OPENAI_API_BASE,
+        dangerouslyAllowBrowser: !isProd,
+    });
+    setDefaultOpenAIClient(client);
+}
+//# sourceMappingURL=openai.js.map

package/lib/esm/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/openai.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,EACtB,YAAY,EACZ,kBAAkB,GACnB,MAAM,gBAAgB,CAAA;AACvB,OAAO,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAA;AACpC,OAAO,EACL,qBAAqB,EACrB,gCAAgC,EAChC,oBAAoB,EACpB,wBAAwB,EACxB,MAAM,GACP,MAAM,aAAa,CAAA;AAEpB,MAAM,UAAU,WAAW;IACzB,oBAAoB,EAAE,CAAA;IACtB,YAAY,CAAC,kBAAkB,CAAC,CAAA;IAChC,kBAAkB,CAAC,IAAI,CAAC,CAAA;AAC1B,CAAC;AAED,SAAS,oBAAoB;IAC3B,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC;QAC7B,MAAM,EAAE,oBAAoB;QAC5B,UAAU,EAAE,wBAAwB;QACpC,UAAU,EAAE,gCAAgC;QAC5C,OAAO,EAAE,qBAAqB;QAC9B,uBAAuB,EAAE,CAAC,MAAM;KACjC,CAAC,CAAA;IACF,sBAAsB,CAAC,MAAM,CAAC,CAAA;AAChC,CAAC"}

package/lib/esm/runner.d.ts

@@ -1,10 +1,7 @@
-import { Tool } from '@openai/agents';
-import { OutputMessage } from './structured-output';
-import { AgenticInputMessage } from './types';
+import { AIAgent, AgenticInputMessage, AgenticOutputMessage, Context } from './types';
 export declare class AIAgentRunner {
     private agent;
-    constructor(name: string, instructions: string, tools: Tool[]);
-    run(messages: AgenticInputMessage[], maxRetries?: number): Promise<OutputMessage[]>;
+    constructor(agent: AIAgent);
+    run(messages: AgenticInputMessage[], context: Context, maxRetries?: number): Promise<AgenticOutputMessage[]>;
     private runWithRetry;
-    private addExtraInstructions;
 }

package/lib/esm/runner.js

@@ -1,71 +1,31 @@
 import { __awaiter } from "tslib";
-import { Agent, ModelBehaviorError, Runner, setDefaultOpenAIClient, setOpenAIAPI, setTracingDisabled, } from '@openai/agents';
-import { AzureOpenAI } from 'openai';
-import { AZURE_OPENAI_API_BASE, AZURE_OPENAI_API_DEPLOYMENT_NAME, AZURE_OPENAI_API_KEY, AZURE_OPENAI_API_VERSION, isProd, } from './constants';
-import { OutputSchema } from './structured-output';
+import { Runner } from '@openai/agents';
 export class AIAgentRunner {
-    constructor(name, instructions, tools) {
-        if (AZURE_OPENAI_API_KEY) {
-            const client = new AzureOpenAI({
-                apiKey: AZURE_OPENAI_API_KEY,
-                apiVersion: AZURE_OPENAI_API_VERSION,
-                deployment: AZURE_OPENAI_API_DEPLOYMENT_NAME,
-                baseURL: AZURE_OPENAI_API_BASE,
-                dangerouslyAllowBrowser: !isProd,
-            });
-            setDefaultOpenAIClient(client);
-            setOpenAIAPI('chat_completions');
-            setTracingDisabled(true);
-        }
-        this.agent = new Agent({
-            name: name,
-            instructions: this.addExtraInstructions(instructions),
-            tools: tools,
-            outputType: OutputSchema,
-        });
+    constructor(agent) {
+        this.agent = agent;
     }
-    run(messages, maxRetries = 1) {
+    run(messages, context, maxRetries = 1) {
         return __awaiter(this, void 0, void 0, function* () {
-            return this.runWithRetry(messages, maxRetries, 1);
+            return this.runWithRetry(messages, context, maxRetries, 1);
         });
     }
-    runWithRetry(messages, maxRetries, attempt) {
+    runWithRetry(messages, context, maxRetries, attempt) {
         var _a;
         return __awaiter(this, void 0, void 0, function* () {
             try {
                 const runner = new Runner({
                     modelSettings: { temperature: 0 },
                 });
-                const result = yield runner.run(this.agent, messages);
+                const result = yield runner.run(this.agent, messages, { context });
                 return ((_a = result.finalOutput) === null || _a === void 0 ? void 0 : _a.messages) || [];
             }
             catch (error) {
-                if (!(error instanceof ModelBehaviorError)) {
-                    throw error;
-                }
                 if (attempt > maxRetries) {
                     throw error;
                 }
-                console.warn(`AI Agent execution failed due to ModelBehaviorError. Retrying... (Attempt ${attempt}/${maxRetries})`);
-                return this.runWithRetry(messages, maxRetries, attempt + 1);
+                return this.runWithRetry(messages, context, maxRetries, attempt + 1);
             }
         });
     }
-    addExtraInstructions(instructions) {
-        const metadata = `Current Date: ${new Date().toISOString()}`;
-        const outputExample = JSON.stringify({
-            messages: [
-                {
-                    type: 'text',
-                    content: {
-                        text: 'Hello, how can I help you today?',
-                    },
-                },
-            ],
-            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${outputExample}\n</example>`;
-        return `<instructions>\n${instructions}\n</instructions>\n\n<metadata>\n${metadata}\n</metadata>\n\n<output>\n${output}\n</output>`;
-    }
 }
 //# 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,EACL,KAAK,EACL,kBAAkB,EAClB,MAAM,EACN,sBAAsB,EACtB,YAAY,EACZ,kBAAkB,GAEnB,MAAM,gBAAgB,CAAA;AACvB,OAAO,EAAE,WAAW,EAAE,MAAM,QAAQ,CAAA;AAEpC,OAAO,EACL,qBAAqB,EACrB,gCAAgC,EAChC,oBAAoB,EACpB,wBAAwB,EACxB,MAAM,GACP,MAAM,aAAa,CAAA;AACpB,OAAO,EAAiB,YAAY,EAAE,MAAM,qBAAqB,CAAA;AAGjE,MAAM,OAAO,aAAa;IAGxB,YAAY,IAAY,EAAE,YAAoB,EAAE,KAAa;QAC3D,IAAI,oBAAoB,EAAE;YACxB,MAAM,MAAM,GAAG,IAAI,WAAW,CAAC;gBAC7B,MAAM,EAAE,oBAAoB;gBAC5B,UAAU,EAAE,wBAAwB;gBACpC,UAAU,EAAE,gCAAgC;gBAC5C,OAAO,EAAE,qBAAqB;gBAC9B,uBAAuB,EAAE,CAAC,MAAM;aACjC,CAAC,CAAA;YACF,sBAAsB,CAAC,MAAa,CAAC,CAAA;YACrC,YAAY,CAAC,kBAAkB,CAAC,CAAA;YAChC,kBAAkB,CAAC,IAAI,CAAC,CAAA;SACzB;QAED,IAAI,CAAC,KAAK,GAAG,IAAI,KAAK,CAAC;YACrB,IAAI,EAAE,IAAI;YACV,YAAY,EAAE,IAAI,CAAC,oBAAoB,CAAC,YAAY,CAAC;YACrD,KAAK,EAAE,KAAK;YACZ,UAAU,EAAE,YAAY;SACzB,CAAC,CAAA;IACJ,CAAC;IAEK,GAAG,CACP,QAA+B,EAC/B,aAAqB,CAAC;;YAEtB,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC,CAAA;QACnD,CAAC;KAAA;IAEa,YAAY,CACxB,QAA+B,EAC/B,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,CAAC,CAAA;gBACrD,OAAO,CAAA,MAAA,MAAM,CAAC,WAAW,0CAAE,QAAQ,KAAI,EAAE,CAAA;aAC1C;YAAC,OAAO,KAAK,EAAE;gBACd,IAAI,CAAC,CAAC,KAAK,YAAY,kBAAkB,CAAC,EAAE;oBAC1C,MAAM,KAAK,CAAA;iBACZ;gBACD,IAAI,OAAO,GAAG,UAAU,EAAE;oBACxB,MAAM,KAAK,CAAA;iBACZ;gBACD,OAAO,CAAC,IAAI,CACV,6EAA6E,OAAO,IAAI,UAAU,GAAG,CACtG,CAAA;gBACD,OAAO,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,GAAG,CAAC,CAAC,CAAA;aAC5D;;KACF;IAEO,oBAAoB,CAAC,YAAoB;QAC/C,MAAM,QAAQ,GAAG,iBAAiB,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAA;QAE5D,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC;YACnC,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP,IAAI,EAAE,kCAAkC;qBACzC;iBACF;aACF;YACD,WAAW,EAAE,CAAC;SACf,CAAC,CAAA;QACF,MAAM,MAAM,GAAG,yIAAyI,aAAa,cAAc,CAAA;QAEnL,OAAO,mBAAmB,YAAY,oCAAoC,QAAQ,8BAA8B,MAAM,aAAa,CAAA;IACrI,CAAC;CACF"}
+{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/runner.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AAQvC,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"}

package/lib/esm/tools/index.d.ts

@@ -0,0 +1,2 @@
+import { Tool } from '../types';
+export declare const mandatoryTools: Tool[];

package/lib/esm/tools/index.js

@@ -0,0 +1,2 @@
+export const mandatoryTools = [];
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/index.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,cAAc,GAAW,EAAE,CAAA"}

package/lib/esm/types.d.ts

@@ -1,15 +1,20 @@
-import { AgentInputItem } from '@openai/agents';
+import { Agent, AgentInputItem, RunContext, Tool as OpenAITool } from '@openai/agents';
 import { ZodObject } from 'zod';
-import { OutputMessage } from './structured-output';
-export interface PluginAiAgentOptions {
-    authToken?: string;
-    customTools?: CustomTool[];
+import { OutputMessage, OutputSchema } from './structured-output';
+export interface Context {
+    authToken: string;
 }
 export interface CustomTool {
     name: string;
     description: string;
     schema: ZodObject<any>;
-    func: (input?: any) => Promise<any>;
+    func: (input?: any, runContext?: RunContext<Context>) => Promise<any>;
+}
+export type Tool = OpenAITool<Context>;
+export type AIAgent = Agent<Context, typeof OutputSchema>;
+export interface PluginAiAgentOptions {
+    authToken?: string;
+    customTools?: CustomTool[];
 }
 export interface AiAgentArgs {
     name: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@botonic/plugin-ai-agents",
-  "version": "0.36.0-alpha.8",
+  "version": "0.36.0-alpha.9",
   "main": "./lib/cjs/index.js",
   "module": "./lib/esm/index.js",
   "description": "Use AI Agents to generate your contents",

package/src/agent-builder.ts

@@ -0,0 +1,48 @@
+import { Agent } from '@openai/agents'
+
+import { OutputSchema } from './structured-output'
+import { mandatoryTools } from './tools'
+import { AIAgent, Tool } from './types'
+
+export class AIAgentBuilder {
+  private name: string
+  private instructions: string
+  private tools: Tool[]
+
+  constructor(name: string, instructions: string, tools: Tool[]) {
+    this.name = name
+    this.instructions = this.addExtraInstructions(instructions)
+    this.tools = this.addHubtypeTools(tools)
+  }
+
+  build(): AIAgent {
+    return new Agent({
+      name: this.name,
+      instructions: this.instructions,
+      tools: this.tools,
+      outputType: OutputSchema,
+    })
+  }
+
+  private addExtraInstructions(instructions: string): string {
+    const metadata = `Current Date: ${new Date().toISOString()}`
+    const example = {
+      messages: [
+        {
+          type: 'text',
+          content: {
+            text: 'Hello, how can I help you today?',
+          },
+        },
+      ],
+      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>`
+  }
+
+  private addHubtypeTools(tools: Tool[]): Tool[] {
+    const hubtypeTools: Tool[] = [...mandatoryTools]
+    return [...hubtypeTools, ...tools]
+  }
+}

package/src/index.ts

@@ -1,24 +1,29 @@
 import { BotContext, Plugin } from '@botonic/core'
 import { tool } from '@openai/agents'
 
+import { AIAgentBuilder } from './agent-builder'
 import { isProd } from './constants'
 import { HubtypeApiClient } from './hubtype-api-client'
+import { setUpOpenAI } from './openai'
 import { AIAgentRunner } from './runner'
 import {
   AgenticInputMessage,
   AgenticOutputMessage,
   AiAgentArgs,
+  Context,
   CustomTool,
   PluginAiAgentOptions,
+  Tool,
 } from './types'
 
 export default class BotonicPluginAiAgents implements Plugin {
   private readonly authToken?: string
-  public customTools: CustomTool[] = []
+  public toolDefinitions: CustomTool[] = []
 
   constructor(options?: PluginAiAgentOptions) {
+    setUpOpenAI()
     this.authToken = options?.authToken
-    this.customTools = options?.customTools || []
+    this.toolDefinitions = options?.customTools || []
   }
 
   pre(): void {
@@ -35,21 +40,22 @@ export default class BotonicPluginAiAgents implements Plugin {
         throw new Error('Auth token is required')
       }
 
-      const messages = await this.getMessages(request, authToken, 10)
-      const availableTools = this.customTools.filter(tool =>
-        aiAgentArgs.activeTools?.map(tool => tool.name).includes(tool.name)
+      const tools = this.buildTools(
+        aiAgentArgs.activeTools?.map(tool => tool.name) || []
       )
-      const tools = this.createTools(availableTools)
-
-      const runner = new AIAgentRunner(
+      const agent = new AIAgentBuilder(
         aiAgentArgs.name,
         aiAgentArgs.instructions,
         tools
-      )
-      const output = await runner.run(messages)
-      console.log('\n\nOutput:', output)
+      ).build()
+
+      const messages = await this.getMessages(request, authToken, 10)
+      const context: Context = {
+        authToken,
+      }
 
-      return output
+      const runner = new AIAgentRunner(agent)
+      return await runner.run(messages, context)
     } catch (error) {
       console.error('error plugin returns undefined', error)
       return undefined
@@ -69,13 +75,16 @@ export default class BotonicPluginAiAgents implements Plugin {
     return await hubtypeClient.getLocalMessages(memoryLength)
   }
 
-  private createTools(customTools: CustomTool[]) {
-    return customTools.map(customTool => {
-      return tool({
-        name: customTool.name,
-        description: customTool.description,
-        parameters: customTool.schema,
-        execute: customTool.func,
+  private buildTools(activeToolNames: string[]): Tool[] {
+    const availableTools = this.toolDefinitions.filter(tool =>
+      activeToolNames.includes(tool.name)
+    )
+    return availableTools.map(toolDefinition => {
+      return tool<any, Context, any>({
+        name: toolDefinition.name,
+        description: toolDefinition.description,
+        parameters: toolDefinition.schema,
+        execute: toolDefinition.func,
       })
     })
   }

package/src/openai.ts

@@ -0,0 +1,30 @@
+import {
+  setDefaultOpenAIClient,
+  setOpenAIAPI,
+  setTracingDisabled,
+} from '@openai/agents'
+import { AzureOpenAI } from 'openai'
+import {
+  AZURE_OPENAI_API_BASE,
+  AZURE_OPENAI_API_DEPLOYMENT_NAME,
+  AZURE_OPENAI_API_KEY,
+  AZURE_OPENAI_API_VERSION,
+  isProd,
+} from './constants'
+
+export function setUpOpenAI() {
+  setAzureOpenAIClient()
+  setOpenAIAPI('chat_completions')
+  setTracingDisabled(true)
+}
+
+function setAzureOpenAIClient() {
+  const client = new AzureOpenAI({
+    apiKey: AZURE_OPENAI_API_KEY,
+    apiVersion: AZURE_OPENAI_API_VERSION,
+    deployment: AZURE_OPENAI_API_DEPLOYMENT_NAME,
+    baseURL: AZURE_OPENAI_API_BASE,
+    dangerouslyAllowBrowser: !isProd,
+  })
+  setDefaultOpenAIClient(client)
+}

package/src/runner.ts

@@ -1,97 +1,43 @@
+import { Runner } from '@openai/agents'
 import {
-  Agent,
-  ModelBehaviorError,
-  Runner,
-  setDefaultOpenAIClient,
-  setOpenAIAPI,
-  setTracingDisabled,
-  Tool,
-} from '@openai/agents'
-import { AzureOpenAI } from 'openai'
-
-import {
-  AZURE_OPENAI_API_BASE,
-  AZURE_OPENAI_API_DEPLOYMENT_NAME,
-  AZURE_OPENAI_API_KEY,
-  AZURE_OPENAI_API_VERSION,
-  isProd,
-} from './constants'
-import { OutputMessage, OutputSchema } from './structured-output'
-import { AgenticInputMessage } from './types'
+  AIAgent,
+  AgenticInputMessage,
+  AgenticOutputMessage,
+  Context,
+} from './types'
 
 export class AIAgentRunner {
-  private agent: Agent<any, typeof OutputSchema>
-
-  constructor(name: string, instructions: string, tools: Tool[]) {
-    if (AZURE_OPENAI_API_KEY) {
-      const client = new AzureOpenAI({
-        apiKey: AZURE_OPENAI_API_KEY,
-        apiVersion: AZURE_OPENAI_API_VERSION,
-        deployment: AZURE_OPENAI_API_DEPLOYMENT_NAME,
-        baseURL: AZURE_OPENAI_API_BASE,
-        dangerouslyAllowBrowser: !isProd,
-      })
-      setDefaultOpenAIClient(client as any)
-      setOpenAIAPI('chat_completions')
-      setTracingDisabled(true)
-    }
+  private agent: AIAgent
 
-    this.agent = new Agent({
-      name: name,
-      instructions: this.addExtraInstructions(instructions),
-      tools: tools,
-      outputType: OutputSchema,
-    })
+  constructor(agent: AIAgent) {
+    this.agent = agent
   }
 
   async run(
     messages: AgenticInputMessage[],
+    context: Context,
     maxRetries: number = 1
-  ): Promise<OutputMessage[]> {
-    return this.runWithRetry(messages, maxRetries, 1)
+  ): Promise<AgenticOutputMessage[]> {
+    return this.runWithRetry(messages, context, maxRetries, 1)
   }
 
   private async runWithRetry(
     messages: AgenticInputMessage[],
+    context: Context,
     maxRetries: number,
     attempt: number
-  ): Promise<OutputMessage[]> {
+  ): Promise<AgenticOutputMessage[]> {
     try {
       const runner = new Runner({
         modelSettings: { temperature: 0 },
       })
-      const result = await runner.run(this.agent, messages)
+      const result = await runner.run(this.agent, messages, { context })
       return result.finalOutput?.messages || []
     } catch (error) {
-      if (!(error instanceof ModelBehaviorError)) {
-        throw error
-      }
       if (attempt > maxRetries) {
         throw error
       }
-      console.warn(
-        `AI Agent execution failed due to ModelBehaviorError. Retrying... (Attempt ${attempt}/${maxRetries})`
-      )
-      return this.runWithRetry(messages, maxRetries, attempt + 1)
+      return this.runWithRetry(messages, context, maxRetries, attempt + 1)
     }
   }
-
-  private addExtraInstructions(instructions: string): string {
-    const metadata = `Current Date: ${new Date().toISOString()}`
-
-    const outputExample = JSON.stringify({
-      messages: [
-        {
-          type: 'text',
-          content: {
-            text: 'Hello, how can I help you today?',
-          },
-        },
-      ],
-      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${outputExample}\n</example>`
-
-    return `<instructions>\n${instructions}\n</instructions>\n\n<metadata>\n${metadata}\n</metadata>\n\n<output>\n${output}\n</output>`
-  }
 }

package/src/tools/index.ts

@@ -0,0 +1,3 @@
+import { Tool } from '../types'
+
+export const mandatoryTools: Tool[] = []

package/src/types.ts

@@ -1,17 +1,30 @@
-import { AgentInputItem } from '@openai/agents'
+import {
+  Agent,
+  AgentInputItem,
+  RunContext,
+  Tool as OpenAITool,
+} from '@openai/agents'
 import { ZodObject } from 'zod'
 
-import { OutputMessage } from './structured-output'
-export interface PluginAiAgentOptions {
-  authToken?: string
-  customTools?: CustomTool[]
+import { OutputMessage, OutputSchema } from './structured-output'
+
+export interface Context {
+  authToken: string
 }
 
 export interface CustomTool {
   name: string
   description: string
   schema: ZodObject<any>
-  func: (input?: any) => Promise<any>
+  func: (input?: any, runContext?: RunContext<Context>) => Promise<any>
+}
+
+export type Tool = OpenAITool<Context>
+export type AIAgent = Agent<Context, typeof OutputSchema>
+
+export interface PluginAiAgentOptions {
+  authToken?: string
+  customTools?: CustomTool[]
 }
 
 export interface AiAgentArgs {