@botonic/plugin-ai-agents 0.36.0-alpha.2 → 0.36.0-alpha.3

package/README.md

@@ -0,0 +1,14 @@
+# Botonic Plugin AI Agents
+
+## env variables
+
+You need this env variables when deploy a bot that use this plugin
+
+.env
+
+```
+AZURE_OPENAI_API_KEY='FAKE_API_KEY' // pragma: allowlist secret
+AZURE_OPENAI_API_BASE='FAKE_API_URL_BASE'
+AZURE_OPENAI_API_DEPLOYMENT_NAME='gpt-4o-mini_p1'
+AZURE_OPENAI_API_VERSION='2024-12-01-preview'
+```

package/lib/cjs/ai-agent-client.d.ts

@@ -3,8 +3,6 @@ import { StructuredTool } from '@langchain/core/tools';
 import { CompiledStateGraph } from '@langchain/langgraph';
 import { AgenticInputMessage, AgenticOutputMessage, AiAgentArgs } from './types';
 export declare class AiAgentClient {
-    name: string;
-    instructions: string;
     agent: CompiledStateGraph<any, any>;
     constructor(aiAgentArgs: AiAgentArgs, chatModel: BaseChatModel, tools?: StructuredTool[]);
     runAgent(_messages: AgenticInputMessage[]): Promise<AgenticOutputMessage>;

package/lib/cjs/ai-agent-client.js

@@ -4,15 +4,14 @@ exports.AiAgentClient = void 0;
 const tslib_1 = require("tslib");
 const messages_1 = require("@langchain/core/messages");
 const prebuilt_1 = require("@langchain/langgraph/prebuilt");
-const tools_1 = require("./tools");
+const default_1 = require("./tools/default");
 class AiAgentClient {
     constructor(aiAgentArgs, chatModel, tools = []) {
-        this.name = aiAgentArgs.name;
-        this.instructions = aiAgentArgs.instructions;
         this.agent = (0, prebuilt_1.createReactAgent)({
+            name: aiAgentArgs.name,
             llm: chatModel,
             tools: tools,
-            prompt: this.instructions,
+            prompt: aiAgentArgs.instructions,
         });
     }
     runAgent(_messages) {
@@ -25,7 +24,7 @@ class AiAgentClient {
             });
             const lastMessage = response.messages.at(-1);
             if (lastMessage instanceof messages_1.ToolMessage) {
-                if (lastMessage.name && tools_1.EXIT_TOOLS.includes(lastMessage.name)) {
+                if (lastMessage.name && default_1.EXIT_TOOLS.includes(lastMessage.name)) {
                     return {
                         role: 'exit',
                     };

package/lib/cjs/ai-agent-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai-agent-client.js","sourceRoot":"","sources":["../../src/ai-agent-client.ts"],"names":[],"mappings":";;;;AACA,uDAGiC;AAGjC,4DAAgE;AAEhE,mCAAoC;AAUpC,MAAa,aAAa;IAKxB,YACE,WAAwB,EACxB,SAAwB,EACxB,QAA0B,EAAE;QAE5B,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAA;QAC5B,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,YAAY,CAAA;QAE5C,IAAI,CAAC,KAAK,GAAG,IAAA,2BAAgB,EAAC;YAC5B,GAAG,EAAE,SAAS;YACd,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,IAAI,CAAC,YAAY;SAC1B,CAAC,CAAA;IACJ,CAAC;IAEK,QAAQ,CACZ,SAAgC;;YAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,OAA4B,EAAE,EAAE,CAAC,CAAC;oBACzD,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,OAAO,EAAE,OAAO,CAAC,OAAO;iBACzB,CAAC,CAAC;aACJ,CAAC,CAAA;YACF,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAE5C,IAAI,WAAW,YAAY,sBAAoB,EAAE;gBAC/C,IAAI,WAAW,CAAC,IAAI,IAAI,kBAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;oBAC7D,OAAO;wBACL,IAAI,EAAE,MAAM;qBACE,CAAA;iBACjB;gBACD,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,WAAW,CAAC,IAAI;oBAC1B,UAAU,EAAE,WAAW,CAAC,OAAO;iBACjB,CAAA;aACjB;YAED,IAAI,WAAW,YAAY,oBAAkB,EAAE;gBAC7C,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,WAAW,CAAC,OAAO;iBACT,CAAA;aACtB;YAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;QACxD,CAAC;KAAA;CACF;AArDD,sCAqDC"}
+{"version":3,"file":"ai-agent-client.js","sourceRoot":"","sources":["../../src/ai-agent-client.ts"],"names":[],"mappings":";;;;AACA,uDAGiC;AAGjC,4DAAgE;AAEhE,6CAA4C;AAU5C,MAAa,aAAa;IAGxB,YACE,WAAwB,EACxB,SAAwB,EACxB,QAA0B,EAAE;QAE5B,IAAI,CAAC,KAAK,GAAG,IAAA,2BAAgB,EAAC;YAC5B,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,GAAG,EAAE,SAAS;YACd,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,WAAW,CAAC,YAAY;SACjC,CAAC,CAAA;IACJ,CAAC;IAEK,QAAQ,CACZ,SAAgC;;YAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,OAA4B,EAAE,EAAE,CAAC,CAAC;oBACzD,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,OAAO,EAAE,OAAO,CAAC,OAAO;iBACzB,CAAC,CAAC;aACJ,CAAC,CAAA;YACF,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAE5C,IAAI,WAAW,YAAY,sBAAoB,EAAE;gBAC/C,IAAI,WAAW,CAAC,IAAI,IAAI,oBAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;oBAC7D,OAAO;wBACL,IAAI,EAAE,MAAM;qBACE,CAAA;iBACjB;gBACD,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,WAAW,CAAC,IAAI;oBAC1B,UAAU,EAAE,WAAW,CAAC,OAAO;iBACjB,CAAA;aACjB;YAED,IAAI,WAAW,YAAY,oBAAkB,EAAE;gBAC7C,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,WAAW,CAAC,OAAO;iBACT,CAAA;aACtB;YAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;QACxD,CAAC;KAAA;CACF;AAjDD,sCAiDC"}

package/lib/cjs/constants.d.ts

@@ -1,5 +1,5 @@
 export declare const HUBTYPE_API_URL: string;
 export declare const AZURE_OPENAI_API_KEY: string | undefined;
-export declare const AZURE_OPENAI_API_DEPLOYMENT_NAME: string | undefined;
-export declare const AZURE_OPENAI_API_VERSION: string | undefined;
 export declare const AZURE_OPENAI_API_BASE: string | undefined;
+export declare const AZURE_OPENAI_API_DEPLOYMENT_NAME: string;
+export declare const AZURE_OPENAI_API_VERSION: string;

package/lib/cjs/constants.js

@@ -1,9 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AZURE_OPENAI_API_BASE = exports.AZURE_OPENAI_API_VERSION = exports.AZURE_OPENAI_API_DEPLOYMENT_NAME = exports.AZURE_OPENAI_API_KEY = exports.HUBTYPE_API_URL = void 0;
+exports.AZURE_OPENAI_API_VERSION = exports.AZURE_OPENAI_API_DEPLOYMENT_NAME = exports.AZURE_OPENAI_API_BASE = exports.AZURE_OPENAI_API_KEY = exports.HUBTYPE_API_URL = void 0;
 exports.HUBTYPE_API_URL = process.env.HUBTYPE_API_URL || 'https://api.hubtype.com';
-exports.AZURE_OPENAI_API_KEY = process.env.AZURE_OPENAI_API_KEY;
-exports.AZURE_OPENAI_API_DEPLOYMENT_NAME = process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME;
-exports.AZURE_OPENAI_API_VERSION = process.env.AZURE_OPENAI_API_VERSION;
+exports.AZURE_OPENAI_API_KEY = process.env.AZURE_OPENAI_API_KEY; // pragma: allowlist secret
 exports.AZURE_OPENAI_API_BASE = process.env.AZURE_OPENAI_API_BASE;
+exports.AZURE_OPENAI_API_DEPLOYMENT_NAME = process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME || 'gpt-4o-mini_p1';
+exports.AZURE_OPENAI_API_VERSION = process.env.AZURE_OPENAI_API_VERSION || '2024-12-01-preview';
 //# sourceMappingURL=constants.js.map

package/lib/cjs/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/constants.ts"],"names":[],"mappings":";;;AAAa,QAAA,eAAe,GAC1B,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,yBAAyB,CAAA;AAE7C,QAAA,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAA;AACvD,QAAA,gCAAgC,GAC3C,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAA;AACjC,QAAA,wBAAwB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAA;AAC/D,QAAA,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAA"}
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/constants.ts"],"names":[],"mappings":";;;AAAa,QAAA,eAAe,GAC1B,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,yBAAyB,CAAA;AAE7C,QAAA,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAA,CAAC,2BAA2B;AACnF,QAAA,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAA;AACzD,QAAA,gCAAgC,GAC3C,OAAO,CAAC,GAAG,CAAC,gCAAgC,IAAI,gBAAgB,CAAA;AACrD,QAAA,wBAAwB,GACnC,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,oBAAoB,CAAA"}

package/lib/cjs/hubtype-api-client.d.ts

@@ -0,0 +1,7 @@
+import { BotContext } from '@botonic/core';
+import { AgenticInputMessage } from './types';
+export declare class HubtypeApiClient {
+    private readonly authToken;
+    constructor(authToken: string);
+    getMessages(request: BotContext, memoryLength: number): Promise<AgenticInputMessage[]>;
+}

package/lib/cjs/hubtype-api-client.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HubtypeApiClient = void 0;
+const tslib_1 = require("tslib");
+const axios_1 = tslib_1.__importDefault(require("axios"));
+const constants_1 = require("./constants");
+class HubtypeApiClient {
+    constructor(authToken) {
+        this.authToken = authToken;
+    }
+    getMessages(request, memoryLength) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            const url = `${constants_1.HUBTYPE_API_URL}/external/v1/ai/agent/message_history/`;
+            const headers = {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${this.authToken}`,
+            };
+            const params = {
+                last_message_id: request.input.message_id,
+                num_messages: memoryLength,
+            };
+            try {
+                const messages = yield axios_1.default.get(url, {
+                    headers,
+                    params,
+                });
+                return messages.data.messages;
+            }
+            catch (error) {
+                console.error(error);
+                throw new Error('Failed to get messages from Hubtype');
+            }
+        });
+    }
+}
+exports.HubtypeApiClient = HubtypeApiClient;
+//# sourceMappingURL=hubtype-api-client.js.map

package/lib/cjs/hubtype-api-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hubtype-api-client.js","sourceRoot":"","sources":["../../src/hubtype-api-client.ts"],"names":[],"mappings":";;;;AACA,0DAAyB;AAEzB,2CAA6C;AAG7C,MAAa,gBAAgB;IAG3B,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAEK,WAAW,CACf,OAAmB,EACnB,YAAoB;;YAEpB,MAAM,GAAG,GAAG,GAAG,2BAAe,wCAAwC,CAAA;YACtE,MAAM,OAAO,GAAG;gBACd,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;aAC1C,CAAA;YACD,MAAM,MAAM,GAAG;gBACb,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;gBACzC,YAAY,EAAE,YAAY;aAC3B,CAAA;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,GAAG,CAC9B,GAAG,EACH;oBACE,OAAO;oBACP,MAAM;iBACP,CACF,CAAA;gBAED,OAAO,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;aAC9B;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;aACvD;QACH,CAAC;KAAA;CACF;AApCD,4CAoCC"}

package/lib/cjs/hubtype-client.d.ts

@@ -1,6 +1,6 @@
 import { BotContext } from '@botonic/core';
 import { AgenticInputMessage } from './types';
-export declare class HubtypeClient {
+export declare class HubtypeApiClient {
     private readonly authToken;
     constructor(authToken: string);
     getMessages(request: BotContext, memoryLength: number): Promise<AgenticInputMessage[]>;

package/lib/cjs/hubtype-client.js

@@ -1,10 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.HubtypeClient = void 0;
+exports.HubtypeApiClient = void 0;
 const tslib_1 = require("tslib");
 const axios_1 = tslib_1.__importDefault(require("axios"));
 const constants_1 = require("./constants");
-class HubtypeClient {
+class HubtypeApiClient {
     constructor(authToken) {
         this.authToken = authToken;
     }
@@ -33,5 +33,5 @@ class HubtypeClient {
         });
     }
 }
-exports.HubtypeClient = HubtypeClient;
+exports.HubtypeApiClient = HubtypeApiClient;
 //# sourceMappingURL=hubtype-client.js.map

package/lib/cjs/hubtype-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"hubtype-client.js","sourceRoot":"","sources":["../../src/hubtype-client.ts"],"names":[],"mappings":";;;;AACA,0DAAyB;AAEzB,2CAA6C;AAG7C,MAAa,aAAa;IAGxB,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAEK,WAAW,CACf,OAAmB,EACnB,YAAoB;;YAEpB,MAAM,GAAG,GAAG,GAAG,2BAAe,wCAAwC,CAAA;YACtE,MAAM,OAAO,GAAG;gBACd,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;aAC1C,CAAA;YACD,MAAM,MAAM,GAAG;gBACb,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;gBACzC,YAAY,EAAE,YAAY;aAC3B,CAAA;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,GAAG,CAC9B,GAAG,EACH;oBACE,OAAO;oBACP,MAAM;iBACP,CACF,CAAA;gBAED,OAAO,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;aAC9B;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;aACvD;QACH,CAAC;KAAA;CACF;AApCD,sCAoCC"}
+{"version":3,"file":"hubtype-client.js","sourceRoot":"","sources":["../../src/hubtype-client.ts"],"names":[],"mappings":";;;;AACA,0DAAyB;AAEzB,2CAA6C;AAG7C,MAAa,gBAAgB;IAG3B,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAEK,WAAW,CACf,OAAmB,EACnB,YAAoB;;YAEpB,MAAM,GAAG,GAAG,GAAG,2BAAe,wCAAwC,CAAA;YACtE,MAAM,OAAO,GAAG;gBACd,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;aAC1C,CAAA;YACD,MAAM,MAAM,GAAG;gBACb,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;gBACzC,YAAY,EAAE,YAAY;aAC3B,CAAA;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,GAAG,CAC9B,GAAG,EACH;oBACE,OAAO;oBACP,MAAM;iBACP,CACF,CAAA;gBAED,OAAO,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;aAC9B;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;aACvD;QACH,CAAC;KAAA;CACF;AApCD,4CAoCC"}

package/lib/cjs/index.d.ts

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

package/lib/cjs/index.js

@@ -2,13 +2,16 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const tslib_1 = require("tslib");
 const ai_agent_client_1 = require("./ai-agent-client");
-const hubtype_client_1 = require("./hubtype-client");
-const tools_1 = require("./tools");
+const hubtype_api_client_1 = require("./hubtype-api-client");
+const custom_1 = require("./tools/custom");
+const default_1 = require("./tools/default");
 const utils_1 = require("./utils");
 const isProd = process.env.NODE_ENV === 'production';
 class BotonicPluginAiAgents {
     constructor(options) {
+        this.customTools = [];
         this.authToken = options === null || options === void 0 ? void 0 : options.authToken;
+        this.customTools = (options === null || options === void 0 ? void 0 : options.customTools) || [];
     }
     pre() {
         return;
@@ -20,16 +23,11 @@ class BotonicPluginAiAgents {
                 if (!authToken) {
                     throw new Error('Auth token is required');
                 }
-                const hubtypeClient = new hubtype_client_1.HubtypeClient(authToken);
-                const messages = isProd
-                    ? yield hubtypeClient.getMessages(request, 10)
-                    : [{ role: 'user', content: request.input.data }];
-                const customTools = [];
-                const chatModel = (0, utils_1.loadChatModel)('azureOpenAI');
-                const aiAgentClient = new ai_agent_client_1.AiAgentClient(aiAgentArgs, chatModel, [
-                    ...customTools,
-                    ...tools_1.MANDATORY_TOOLS,
-                ]);
+                const messages = yield this.getMessages(request, authToken, 10);
+                const customTools = this.customTools.map(customTool => (0, custom_1.createCustomTool)(customTool));
+                const tools = [...customTools, ...default_1.MANDATORY_TOOLS];
+                const chatModel = (0, utils_1.loadChatModel)(utils_1.AiProvider.AzureOpenAI);
+                const aiAgentClient = new ai_agent_client_1.AiAgentClient(aiAgentArgs, chatModel, tools);
                 return yield aiAgentClient.runAgent(messages);
             }
             catch (error) {
@@ -38,6 +36,15 @@ class BotonicPluginAiAgents {
             }
         });
     }
+    getMessages(request, authToken, memoryLength) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            if (isProd) {
+                const hubtypeClient = new hubtype_api_client_1.HubtypeApiClient(authToken);
+                return yield hubtypeClient.getMessages(request, memoryLength);
+            }
+            return [{ role: 'user', content: request.input.data }];
+        });
+    }
 }
 exports.default = BotonicPluginAiAgents;
 tslib_1.__exportStar(require("./types"), exports);

package/lib/cjs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AAEA,uDAAiD;AACjD,qDAAgD;AAChD,mCAAyC;AAOzC,mCAAuC;AAEvC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAA;AAEpD,MAAqB,qBAAqB;IAExC,YAAY,OAA8B;QACxC,IAAI,CAAC,SAAS,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,CAAA;IACrC,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,aAAa,GAAG,IAAI,8BAAa,CAAC,SAAS,CAAC,CAAA;gBAClD,MAAM,QAAQ,GAAG,MAAM;oBACrB,CAAC,CAAC,MAAM,aAAa,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;oBAC9C,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI,EAAiB,CAAC,CAAA;gBAElE,MAAM,WAAW,GAAG,EAAE,CAAA;gBACtB,MAAM,SAAS,GAAG,IAAA,qBAAa,EAAC,aAAa,CAAC,CAAA;gBAC9C,MAAM,aAAa,GAAG,IAAI,+BAAa,CAAC,WAAW,EAAE,SAAS,EAAE;oBAC9D,GAAG,WAAW;oBACd,GAAG,uBAAe;iBACnB,CAAC,CAAA;gBAEF,OAAO,MAAM,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;aAC9C;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAA;gBACtD,OAAO,SAAS,CAAA;aACjB;QACH,CAAC;KAAA;CACF;AAtCD,wCAsCC;AAED,kDAAuB"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AAEA,uDAAiD;AACjD,6DAAuD;AACvD,2CAAiD;AACjD,6CAAiD;AASjD,mCAAmD;AAEnD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAA;AAEpD,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,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;gBAE/D,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CACpD,IAAA,yBAAgB,EAAC,UAAU,CAAC,CAC7B,CAAA;gBAED,MAAM,KAAK,GAAG,CAAC,GAAG,WAAW,EAAE,GAAG,yBAAe,CAAC,CAAA;gBAClD,MAAM,SAAS,GAAG,IAAA,qBAAa,EAAC,kBAAU,CAAC,WAAW,CAAC,CAAA;gBACvD,MAAM,aAAa,GAAG,IAAI,+BAAa,CAAC,WAAW,EAAE,SAAS,EAAE,KAAK,CAAC,CAAA;gBAEtE,OAAO,MAAM,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;aAC9C;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,qCAAgB,CAAC,SAAS,CAAC,CAAA;gBACrD,OAAO,MAAM,aAAa,CAAC,WAAW,CAAC,OAAO,EAAE,YAAY,CAAC,CAAA;aAC9D;YAED,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI,EAAiB,CAAC,CAAA;QACvE,CAAC;KAAA;CACF;AApDD,wCAoDC;AAED,kDAAuB"}

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

@@ -0,0 +1,3 @@
+import { DynamicStructuredTool } from '@langchain/core/tools';
+import { CustomTool } from '../types';
+export declare function createCustomTool(customTool: CustomTool): DynamicStructuredTool;

package/lib/cjs/tools/custom.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createCustomTool = void 0;
+const tools_1 = require("@langchain/core/tools");
+function createCustomTool(customTool) {
+    return (0, tools_1.tool)(customTool.func, {
+        name: customTool.name,
+        description: customTool.description,
+        schema: customTool.schema,
+        returnDirect: customTool.returnDirect || false,
+    });
+}
+exports.createCustomTool = createCustomTool;
+//# sourceMappingURL=custom.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"custom.js","sourceRoot":"","sources":["../../../src/tools/custom.ts"],"names":[],"mappings":";;;AAAA,iDAAmE;AAInE,SAAgB,gBAAgB,CAC9B,UAAsB;IAEtB,OAAO,IAAA,YAAI,EAAC,UAAU,CAAC,IAAI,EAAE;QAC3B,IAAI,EAAE,UAAU,CAAC,IAAI;QACrB,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,KAAK;KAC/C,CAAC,CAAA;AACJ,CAAC;AATD,4CASC"}

package/lib/cjs/{tools.js → tools/default.js}

@@ -38,4 +38,4 @@ exports.EXIT_TOOLS = [
     exports.outOfContext.name,
     exports.finishConversation.name,
 ];
-//# sourceMappingURL=tools.js.map
+//# sourceMappingURL=default.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"default.js","sourceRoot":"","sources":["../../../src/tools/default.ts"],"names":[],"mappings":";;;;AAAA,iDAA4C;AAC5C,6BAAuB;AAEV,QAAA,mBAAmB,GAAG,IAAA,YAAI,EACrC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,qBAAqB;IAC3B,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,wEAAwE;IAC1E,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,YAAY,GAAG,IAAA,YAAI,EAC9B,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,cAAc;IACpB,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,+IAA+I;IACjJ,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,kBAAkB,GAAG,IAAA,YAAI,EACpC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,oBAAoB;IAC1B,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EAAE,uDAAuD;IACpE,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,eAAe,GAAG;IAC7B,2BAAmB;IACnB,oBAAY;IACZ,0BAAkB;CACnB,CAAA;AAEY,QAAA,UAAU,GAAG;IACxB,2BAAmB,CAAC,IAAI;IACxB,oBAAY,CAAC,IAAI;IACjB,0BAAkB,CAAC,IAAI;CACxB,CAAA"}

package/lib/cjs/types.d.ts

@@ -1,5 +1,14 @@
+import { ZodSchema } from 'zod';
 export interface PluginAiAgentOptions {
     authToken?: string;
+    customTools?: CustomTool[];
+}
+export interface CustomTool {
+    name: string;
+    description: string;
+    schema: ZodSchema;
+    returnDirect?: boolean;
+    func: (input?: any) => Promise<any>;
 }
 export interface AiAgentArgs {
     name: string;

package/lib/cjs/utils.d.ts

@@ -1,2 +1,5 @@
 import { AzureChatOpenAI } from '@langchain/openai';
-export declare function loadChatModel(provider: 'azureOpenAI'): AzureChatOpenAI;
+export declare enum AiProvider {
+    AzureOpenAI = "azureOpenAI"
+}
+export declare function loadChatModel(provider: AiProvider): AzureChatOpenAI;

package/lib/cjs/utils.js

@@ -1,11 +1,15 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.loadChatModel = void 0;
+exports.loadChatModel = exports.AiProvider = void 0;
 const openai_1 = require("@langchain/openai");
 const constants_1 = require("./constants");
+var AiProvider;
+(function (AiProvider) {
+    AiProvider["AzureOpenAI"] = "azureOpenAI";
+})(AiProvider = exports.AiProvider || (exports.AiProvider = {}));
 function loadChatModel(provider) {
     switch (provider) {
-        case 'azureOpenAI':
+        case AiProvider.AzureOpenAI:
             return new openai_1.AzureChatOpenAI({
                 azureOpenAIApiVersion: constants_1.AZURE_OPENAI_API_VERSION,
                 azureOpenAIApiKey: constants_1.AZURE_OPENAI_API_KEY,

package/lib/cjs/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/utils.ts"],"names":[],"mappings":";;;AAAA,8CAAmD;AAEnD,2CAKoB;AAEpB,SAAgB,aAAa,CAAC,QAAuB;IACnD,QAAQ,QAAQ,EAAE;QAChB,KAAK,aAAa;YAChB,OAAO,IAAI,wBAAe,CAAC;gBACzB,qBAAqB,EAAE,oCAAwB;gBAC/C,iBAAiB,EAAE,gCAAoB;gBACvC,4BAA4B,EAAE,4CAAgC;gBAC9D,mBAAmB,EAAE,iCAAqB;gBAC1C,WAAW,EAAE,CAAC;gBACd,kBAAkB;aACnB,CAAC,CAAA;QACJ;YACE,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,gBAAgB,CAAC,CAAA;KACxD;AACH,CAAC;AAdD,sCAcC"}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/utils.ts"],"names":[],"mappings":";;;AAAA,8CAAmD;AAEnD,2CAKoB;AAEpB,IAAY,UAEX;AAFD,WAAY,UAAU;IACpB,yCAA2B,CAAA;AAC7B,CAAC,EAFW,UAAU,GAAV,kBAAU,KAAV,kBAAU,QAErB;AAED,SAAgB,aAAa,CAAC,QAAoB;IAChD,QAAQ,QAAQ,EAAE;QAChB,KAAK,UAAU,CAAC,WAAW;YACzB,OAAO,IAAI,wBAAe,CAAC;gBACzB,qBAAqB,EAAE,oCAAwB;gBAC/C,iBAAiB,EAAE,gCAAoB;gBACvC,4BAA4B,EAAE,4CAAgC;gBAC9D,mBAAmB,EAAE,iCAAqB;gBAC1C,WAAW,EAAE,CAAC;gBACd,kBAAkB;aACnB,CAAC,CAAA;QACJ;YACE,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,gBAAgB,CAAC,CAAA;KACxD;AACH,CAAC;AAdD,sCAcC"}

package/lib/esm/ai-agent-client.d.ts

@@ -3,8 +3,6 @@ import { StructuredTool } from '@langchain/core/tools';
 import { CompiledStateGraph } from '@langchain/langgraph';
 import { AgenticInputMessage, AgenticOutputMessage, AiAgentArgs } from './types';
 export declare class AiAgentClient {
-    name: string;
-    instructions: string;
     agent: CompiledStateGraph<any, any>;
     constructor(aiAgentArgs: AiAgentArgs, chatModel: BaseChatModel, tools?: StructuredTool[]);
     runAgent(_messages: AgenticInputMessage[]): Promise<AgenticOutputMessage>;

package/lib/esm/ai-agent-client.js

@@ -1,15 +1,14 @@
 import { __awaiter } from "tslib";
 import { AIMessage as LangchainAIMessage, ToolMessage as LangchainToolMessage, } from '@langchain/core/messages';
 import { createReactAgent } from '@langchain/langgraph/prebuilt';
-import { EXIT_TOOLS } from './tools';
+import { EXIT_TOOLS } from './tools/default';
 export class AiAgentClient {
     constructor(aiAgentArgs, chatModel, tools = []) {
-        this.name = aiAgentArgs.name;
-        this.instructions = aiAgentArgs.instructions;
         this.agent = createReactAgent({
+            name: aiAgentArgs.name,
             llm: chatModel,
             tools: tools,
-            prompt: this.instructions,
+            prompt: aiAgentArgs.instructions,
         });
     }
     runAgent(_messages) {

package/lib/esm/ai-agent-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai-agent-client.js","sourceRoot":"","sources":["../../src/ai-agent-client.ts"],"names":[],"mappings":";AACA,OAAO,EACL,SAAS,IAAI,kBAAkB,EAC/B,WAAW,IAAI,oBAAoB,GACpC,MAAM,0BAA0B,CAAA;AAGjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAA;AAEhE,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAA;AAUpC,MAAM,OAAO,aAAa;IAKxB,YACE,WAAwB,EACxB,SAAwB,EACxB,QAA0B,EAAE;QAE5B,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC,IAAI,CAAA;QAC5B,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,YAAY,CAAA;QAE5C,IAAI,CAAC,KAAK,GAAG,gBAAgB,CAAC;YAC5B,GAAG,EAAE,SAAS;YACd,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,IAAI,CAAC,YAAY;SAC1B,CAAC,CAAA;IACJ,CAAC;IAEK,QAAQ,CACZ,SAAgC;;YAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,OAA4B,EAAE,EAAE,CAAC,CAAC;oBACzD,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,OAAO,EAAE,OAAO,CAAC,OAAO;iBACzB,CAAC,CAAC;aACJ,CAAC,CAAA;YACF,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAE5C,IAAI,WAAW,YAAY,oBAAoB,EAAE;gBAC/C,IAAI,WAAW,CAAC,IAAI,IAAI,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;oBAC7D,OAAO;wBACL,IAAI,EAAE,MAAM;qBACE,CAAA;iBACjB;gBACD,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,WAAW,CAAC,IAAI;oBAC1B,UAAU,EAAE,WAAW,CAAC,OAAO;iBACjB,CAAA;aACjB;YAED,IAAI,WAAW,YAAY,kBAAkB,EAAE;gBAC7C,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,WAAW,CAAC,OAAO;iBACT,CAAA;aACtB;YAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;QACxD,CAAC;KAAA;CACF"}
+{"version":3,"file":"ai-agent-client.js","sourceRoot":"","sources":["../../src/ai-agent-client.ts"],"names":[],"mappings":";AACA,OAAO,EACL,SAAS,IAAI,kBAAkB,EAC/B,WAAW,IAAI,oBAAoB,GACpC,MAAM,0BAA0B,CAAA;AAGjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAA;AAEhE,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAU5C,MAAM,OAAO,aAAa;IAGxB,YACE,WAAwB,EACxB,SAAwB,EACxB,QAA0B,EAAE;QAE5B,IAAI,CAAC,KAAK,GAAG,gBAAgB,CAAC;YAC5B,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,GAAG,EAAE,SAAS;YACd,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,WAAW,CAAC,YAAY;SACjC,CAAC,CAAA;IACJ,CAAC;IAEK,QAAQ,CACZ,SAAgC;;YAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;gBACvC,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,OAA4B,EAAE,EAAE,CAAC,CAAC;oBACzD,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,OAAO,EAAE,OAAO,CAAC,OAAO;iBACzB,CAAC,CAAC;aACJ,CAAC,CAAA;YACF,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;YAE5C,IAAI,WAAW,YAAY,oBAAoB,EAAE;gBAC/C,IAAI,WAAW,CAAC,IAAI,IAAI,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE;oBAC7D,OAAO;wBACL,IAAI,EAAE,MAAM;qBACE,CAAA;iBACjB;gBACD,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,QAAQ,EAAE,WAAW,CAAC,IAAI;oBAC1B,UAAU,EAAE,WAAW,CAAC,OAAO;iBACjB,CAAA;aACjB;YAED,IAAI,WAAW,YAAY,kBAAkB,EAAE;gBAC7C,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,WAAW,CAAC,OAAO;iBACT,CAAA;aACtB;YAED,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;QACxD,CAAC;KAAA;CACF"}

package/lib/esm/constants.d.ts

@@ -1,5 +1,5 @@
 export declare const HUBTYPE_API_URL: string;
 export declare const AZURE_OPENAI_API_KEY: string | undefined;
-export declare const AZURE_OPENAI_API_DEPLOYMENT_NAME: string | undefined;
-export declare const AZURE_OPENAI_API_VERSION: string | undefined;
 export declare const AZURE_OPENAI_API_BASE: string | undefined;
+export declare const AZURE_OPENAI_API_DEPLOYMENT_NAME: string;
+export declare const AZURE_OPENAI_API_VERSION: string;

package/lib/esm/constants.js

@@ -1,6 +1,6 @@
 export const HUBTYPE_API_URL = process.env.HUBTYPE_API_URL || 'https://api.hubtype.com';
-export const AZURE_OPENAI_API_KEY = process.env.AZURE_OPENAI_API_KEY;
-export const AZURE_OPENAI_API_DEPLOYMENT_NAME = process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME;
-export const AZURE_OPENAI_API_VERSION = process.env.AZURE_OPENAI_API_VERSION;
+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 = process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME || 'gpt-4o-mini_p1';
+export const AZURE_OPENAI_API_VERSION = process.env.AZURE_OPENAI_API_VERSION || '2024-12-01-preview';
 //# sourceMappingURL=constants.js.map

package/lib/esm/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/constants.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,eAAe,GAC1B,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,yBAAyB,CAAA;AAE1D,MAAM,CAAC,MAAM,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAA;AACpE,MAAM,CAAC,MAAM,gCAAgC,GAC3C,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAA;AAC9C,MAAM,CAAC,MAAM,wBAAwB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAA;AAC5E,MAAM,CAAC,MAAM,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAA"}
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/constants.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,eAAe,GAC1B,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,yBAAyB,CAAA;AAE1D,MAAM,CAAC,MAAM,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAA,CAAC,2BAA2B;AAChG,MAAM,CAAC,MAAM,qBAAqB,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAA;AACtE,MAAM,CAAC,MAAM,gCAAgC,GAC3C,OAAO,CAAC,GAAG,CAAC,gCAAgC,IAAI,gBAAgB,CAAA;AAClE,MAAM,CAAC,MAAM,wBAAwB,GACnC,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,oBAAoB,CAAA"}

package/lib/esm/hubtype-api-client.d.ts

@@ -0,0 +1,7 @@
+import { BotContext } from '@botonic/core';
+import { AgenticInputMessage } from './types';
+export declare class HubtypeApiClient {
+    private readonly authToken;
+    constructor(authToken: string);
+    getMessages(request: BotContext, memoryLength: number): Promise<AgenticInputMessage[]>;
+}

package/lib/esm/hubtype-api-client.js

@@ -0,0 +1,33 @@
+import { __awaiter } from "tslib";
+import axios from 'axios';
+import { HUBTYPE_API_URL } from './constants';
+export class HubtypeApiClient {
+    constructor(authToken) {
+        this.authToken = authToken;
+    }
+    getMessages(request, memoryLength) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const url = `${HUBTYPE_API_URL}/external/v1/ai/agent/message_history/`;
+            const headers = {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${this.authToken}`,
+            };
+            const params = {
+                last_message_id: request.input.message_id,
+                num_messages: memoryLength,
+            };
+            try {
+                const messages = yield axios.get(url, {
+                    headers,
+                    params,
+                });
+                return messages.data.messages;
+            }
+            catch (error) {
+                console.error(error);
+                throw new Error('Failed to get messages from Hubtype');
+            }
+        });
+    }
+}
+//# sourceMappingURL=hubtype-api-client.js.map

package/lib/esm/hubtype-api-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hubtype-api-client.js","sourceRoot":"","sources":["../../src/hubtype-api-client.ts"],"names":[],"mappings":";AACA,OAAO,KAAK,MAAM,OAAO,CAAA;AAEzB,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAA;AAG7C,MAAM,OAAO,gBAAgB;IAG3B,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAEK,WAAW,CACf,OAAmB,EACnB,YAAoB;;YAEpB,MAAM,GAAG,GAAG,GAAG,eAAe,wCAAwC,CAAA;YACtE,MAAM,OAAO,GAAG;gBACd,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;aAC1C,CAAA;YACD,MAAM,MAAM,GAAG;gBACb,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;gBACzC,YAAY,EAAE,YAAY;aAC3B,CAAA;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAC9B,GAAG,EACH;oBACE,OAAO;oBACP,MAAM;iBACP,CACF,CAAA;gBAED,OAAO,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;aAC9B;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;aACvD;QACH,CAAC;KAAA;CACF"}

package/lib/esm/hubtype-client.d.ts

@@ -1,6 +1,6 @@
 import { BotContext } from '@botonic/core';
 import { AgenticInputMessage } from './types';
-export declare class HubtypeClient {
+export declare class HubtypeApiClient {
     private readonly authToken;
     constructor(authToken: string);
     getMessages(request: BotContext, memoryLength: number): Promise<AgenticInputMessage[]>;

package/lib/esm/hubtype-client.js

@@ -1,7 +1,7 @@
 import { __awaiter } from "tslib";
 import axios from 'axios';
 import { HUBTYPE_API_URL } from './constants';
-export class HubtypeClient {
+export class HubtypeApiClient {
     constructor(authToken) {
         this.authToken = authToken;
     }

package/lib/esm/hubtype-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"hubtype-client.js","sourceRoot":"","sources":["../../src/hubtype-client.ts"],"names":[],"mappings":";AACA,OAAO,KAAK,MAAM,OAAO,CAAA;AAEzB,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAA;AAG7C,MAAM,OAAO,aAAa;IAGxB,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAEK,WAAW,CACf,OAAmB,EACnB,YAAoB;;YAEpB,MAAM,GAAG,GAAG,GAAG,eAAe,wCAAwC,CAAA;YACtE,MAAM,OAAO,GAAG;gBACd,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;aAC1C,CAAA;YACD,MAAM,MAAM,GAAG;gBACb,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;gBACzC,YAAY,EAAE,YAAY;aAC3B,CAAA;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAC9B,GAAG,EACH;oBACE,OAAO;oBACP,MAAM;iBACP,CACF,CAAA;gBAED,OAAO,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;aAC9B;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;aACvD;QACH,CAAC;KAAA;CACF"}
+{"version":3,"file":"hubtype-client.js","sourceRoot":"","sources":["../../src/hubtype-client.ts"],"names":[],"mappings":";AACA,OAAO,KAAK,MAAM,OAAO,CAAA;AAEzB,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAA;AAG7C,MAAM,OAAO,gBAAgB;IAG3B,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAEK,WAAW,CACf,OAAmB,EACnB,YAAoB;;YAEpB,MAAM,GAAG,GAAG,GAAG,eAAe,wCAAwC,CAAA;YACtE,MAAM,OAAO,GAAG;gBACd,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;aAC1C,CAAA;YACD,MAAM,MAAM,GAAG;gBACb,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;gBACzC,YAAY,EAAE,YAAY;aAC3B,CAAA;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAC9B,GAAG,EACH;oBACE,OAAO;oBACP,MAAM;iBACP,CACF,CAAA;gBAED,OAAO,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;aAC9B;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;aACvD;QACH,CAAC;KAAA;CACF"}

package/lib/esm/index.d.ts

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

package/lib/esm/index.js

@@ -1,12 +1,15 @@
 import { __awaiter } from "tslib";
 import { AiAgentClient } from './ai-agent-client';
-import { HubtypeClient } from './hubtype-client';
-import { MANDATORY_TOOLS } from './tools';
-import { loadChatModel } from './utils';
+import { HubtypeApiClient } from './hubtype-api-client';
+import { createCustomTool } from './tools/custom';
+import { MANDATORY_TOOLS } from './tools/default';
+import { AiProvider, loadChatModel } from './utils';
 const isProd = process.env.NODE_ENV === 'production';
 export default class BotonicPluginAiAgents {
     constructor(options) {
+        this.customTools = [];
         this.authToken = options === null || options === void 0 ? void 0 : options.authToken;
+        this.customTools = (options === null || options === void 0 ? void 0 : options.customTools) || [];
     }
     pre() {
         return;
@@ -18,16 +21,11 @@ export default class BotonicPluginAiAgents {
                 if (!authToken) {
                     throw new Error('Auth token is required');
                 }
-                const hubtypeClient = new HubtypeClient(authToken);
-                const messages = isProd
-                    ? yield hubtypeClient.getMessages(request, 10)
-                    : [{ role: 'user', content: request.input.data }];
-                const customTools = [];
-                const chatModel = loadChatModel('azureOpenAI');
-                const aiAgentClient = new AiAgentClient(aiAgentArgs, chatModel, [
-                    ...customTools,
-                    ...MANDATORY_TOOLS,
-                ]);
+                const messages = yield this.getMessages(request, authToken, 10);
+                const customTools = this.customTools.map(customTool => createCustomTool(customTool));
+                const tools = [...customTools, ...MANDATORY_TOOLS];
+                const chatModel = loadChatModel(AiProvider.AzureOpenAI);
+                const aiAgentClient = new AiAgentClient(aiAgentArgs, chatModel, tools);
                 return yield aiAgentClient.runAgent(messages);
             }
             catch (error) {
@@ -36,6 +34,15 @@ export default class BotonicPluginAiAgents {
             }
         });
     }
+    getMessages(request, authToken, memoryLength) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (isProd) {
+                const hubtypeClient = new HubtypeApiClient(authToken);
+                return yield hubtypeClient.getMessages(request, memoryLength);
+            }
+            return [{ role: 'user', content: request.input.data }];
+        });
+    }
 }
 export * from './types';
 //# sourceMappingURL=index.js.map

package/lib/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAA;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAA;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAA;AAOzC,OAAO,EAAE,aAAa,EAAE,MAAM,SAAS,CAAA;AAEvC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAA;AAEpD,MAAM,CAAC,OAAO,OAAO,qBAAqB;IAExC,YAAY,OAA8B;QACxC,IAAI,CAAC,SAAS,GAAG,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS,CAAA;IACrC,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,aAAa,GAAG,IAAI,aAAa,CAAC,SAAS,CAAC,CAAA;gBAClD,MAAM,QAAQ,GAAG,MAAM;oBACrB,CAAC,CAAC,MAAM,aAAa,CAAC,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;oBAC9C,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI,EAAiB,CAAC,CAAA;gBAElE,MAAM,WAAW,GAAG,EAAE,CAAA;gBACtB,MAAM,SAAS,GAAG,aAAa,CAAC,aAAa,CAAC,CAAA;gBAC9C,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,WAAW,EAAE,SAAS,EAAE;oBAC9D,GAAG,WAAW;oBACd,GAAG,eAAe;iBACnB,CAAC,CAAA;gBAEF,OAAO,MAAM,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;aAC9C;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAA;gBACtD,OAAO,SAAS,CAAA;aACjB;QACH,CAAC;KAAA;CACF;AAED,cAAc,SAAS,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAA;AACjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAA;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAA;AACjD,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAA;AASjD,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,SAAS,CAAA;AAEnD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAA;AAEpD,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;gBAE/D,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CACpD,gBAAgB,CAAC,UAAU,CAAC,CAC7B,CAAA;gBAED,MAAM,KAAK,GAAG,CAAC,GAAG,WAAW,EAAE,GAAG,eAAe,CAAC,CAAA;gBAClD,MAAM,SAAS,GAAG,aAAa,CAAC,UAAU,CAAC,WAAW,CAAC,CAAA;gBACvD,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,WAAW,EAAE,SAAS,EAAE,KAAK,CAAC,CAAA;gBAEtE,OAAO,MAAM,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;aAC9C;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;YAED,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI,EAAiB,CAAC,CAAA;QACvE,CAAC;KAAA;CACF;AAED,cAAc,SAAS,CAAA"}

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

@@ -0,0 +1,3 @@
+import { DynamicStructuredTool } from '@langchain/core/tools';
+import { CustomTool } from '../types';
+export declare function createCustomTool(customTool: CustomTool): DynamicStructuredTool;

package/lib/esm/tools/custom.js

@@ -0,0 +1,10 @@
+import { tool } from '@langchain/core/tools';
+export function createCustomTool(customTool) {
+    return tool(customTool.func, {
+        name: customTool.name,
+        description: customTool.description,
+        schema: customTool.schema,
+        returnDirect: customTool.returnDirect || false,
+    });
+}
+//# sourceMappingURL=custom.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"custom.js","sourceRoot":"","sources":["../../../src/tools/custom.ts"],"names":[],"mappings":"AAAA,OAAO,EAAyB,IAAI,EAAE,MAAM,uBAAuB,CAAA;AAInE,MAAM,UAAU,gBAAgB,CAC9B,UAAsB;IAEtB,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE;QAC3B,IAAI,EAAE,UAAU,CAAC,IAAI;QACrB,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,KAAK;KAC/C,CAAC,CAAA;AACJ,CAAC"}

package/lib/esm/{tools.js → tools/default.js}

@@ -35,4 +35,4 @@ export const EXIT_TOOLS = [
     outOfContext.name,
     finishConversation.name,
 ];
-//# sourceMappingURL=tools.js.map
+//# sourceMappingURL=default.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"default.js","sourceRoot":"","sources":["../../../src/tools/default.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAA;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,MAAM,CAAC,MAAM,mBAAmB,GAAG,IAAI,CACrC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,qBAAqB;IAC3B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,wEAAwE;IAC1E,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,CAC9B,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,cAAc;IACpB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,+IAA+I;IACjJ,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,CACpC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,oBAAoB;IAC1B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EAAE,uDAAuD;IACpE,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,mBAAmB;IACnB,YAAY;IACZ,kBAAkB;CACnB,CAAA;AAED,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,mBAAmB,CAAC,IAAI;IACxB,YAAY,CAAC,IAAI;IACjB,kBAAkB,CAAC,IAAI;CACxB,CAAA"}

package/lib/esm/types.d.ts

@@ -1,5 +1,14 @@
+import { ZodSchema } from 'zod';
 export interface PluginAiAgentOptions {
     authToken?: string;
+    customTools?: CustomTool[];
+}
+export interface CustomTool {
+    name: string;
+    description: string;
+    schema: ZodSchema;
+    returnDirect?: boolean;
+    func: (input?: any) => Promise<any>;
 }
 export interface AiAgentArgs {
     name: string;

package/lib/esm/utils.d.ts

@@ -1,2 +1,5 @@
 import { AzureChatOpenAI } from '@langchain/openai';
-export declare function loadChatModel(provider: 'azureOpenAI'): AzureChatOpenAI;
+export declare enum AiProvider {
+    AzureOpenAI = "azureOpenAI"
+}
+export declare function loadChatModel(provider: AiProvider): AzureChatOpenAI;

package/lib/esm/utils.js

@@ -1,8 +1,12 @@
 import { AzureChatOpenAI } from '@langchain/openai';
 import { AZURE_OPENAI_API_BASE, AZURE_OPENAI_API_DEPLOYMENT_NAME, AZURE_OPENAI_API_KEY, AZURE_OPENAI_API_VERSION, } from './constants';
+export var AiProvider;
+(function (AiProvider) {
+    AiProvider["AzureOpenAI"] = "azureOpenAI";
+})(AiProvider || (AiProvider = {}));
 export function loadChatModel(provider) {
     switch (provider) {
-        case 'azureOpenAI':
+        case AiProvider.AzureOpenAI:
             return new AzureChatOpenAI({
                 azureOpenAIApiVersion: AZURE_OPENAI_API_VERSION,
                 azureOpenAIApiKey: AZURE_OPENAI_API_KEY,

package/lib/esm/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAA;AAEnD,OAAO,EACL,qBAAqB,EACrB,gCAAgC,EAChC,oBAAoB,EACpB,wBAAwB,GACzB,MAAM,aAAa,CAAA;AAEpB,MAAM,UAAU,aAAa,CAAC,QAAuB;IACnD,QAAQ,QAAQ,EAAE;QAChB,KAAK,aAAa;YAChB,OAAO,IAAI,eAAe,CAAC;gBACzB,qBAAqB,EAAE,wBAAwB;gBAC/C,iBAAiB,EAAE,oBAAoB;gBACvC,4BAA4B,EAAE,gCAAgC;gBAC9D,mBAAmB,EAAE,qBAAqB;gBAC1C,WAAW,EAAE,CAAC;gBACd,kBAAkB;aACnB,CAAC,CAAA;QACJ;YACE,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,gBAAgB,CAAC,CAAA;KACxD;AACH,CAAC"}
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAA;AAEnD,OAAO,EACL,qBAAqB,EACrB,gCAAgC,EAChC,oBAAoB,EACpB,wBAAwB,GACzB,MAAM,aAAa,CAAA;AAEpB,MAAM,CAAN,IAAY,UAEX;AAFD,WAAY,UAAU;IACpB,yCAA2B,CAAA;AAC7B,CAAC,EAFW,UAAU,KAAV,UAAU,QAErB;AAED,MAAM,UAAU,aAAa,CAAC,QAAoB;IAChD,QAAQ,QAAQ,EAAE;QAChB,KAAK,UAAU,CAAC,WAAW;YACzB,OAAO,IAAI,eAAe,CAAC;gBACzB,qBAAqB,EAAE,wBAAwB;gBAC/C,iBAAiB,EAAE,oBAAoB;gBACvC,4BAA4B,EAAE,gCAAgC;gBAC9D,mBAAmB,EAAE,qBAAqB;gBAC1C,WAAW,EAAE,CAAC;gBACd,kBAAkB;aACnB,CAAC,CAAA;QACJ;YACE,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,gBAAgB,CAAC,CAAA;KACxD;AACH,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@botonic/plugin-ai-agents",
-  "version": "0.36.0-alpha.2",
+  "version": "0.36.0-alpha.3",
   "main": "./lib/cjs/index.js",
   "module": "./lib/esm/index.js",
   "description": "Use AI Agents to generate your contents",
@@ -18,7 +18,8 @@
     "@langchain/langgraph": "^0.3.0",
     "@langchain/openai": "^0.5.12",
     "axios": "^1.9.0",
-    "uuid": "^10.0.0"
+    "uuid": "^10.0.0",
+    "zod": "^3.25.57"
   },
   "devDependencies": {
     "@types/uuid": "^10.0.0"

package/src/ai-agent-client.ts

@@ -7,7 +7,7 @@ import { StructuredTool } from '@langchain/core/tools'
 import { CompiledStateGraph } from '@langchain/langgraph'
 import { createReactAgent } from '@langchain/langgraph/prebuilt'
 
-import { EXIT_TOOLS } from './tools'
+import { EXIT_TOOLS } from './tools/default'
 import {
   AgenticInputMessage,
   AgenticOutputMessage,
@@ -18,8 +18,6 @@ import {
 } from './types'
 
 export class AiAgentClient {
-  public name: string
-  public instructions: string
   public agent: CompiledStateGraph<any, any> // TODO: apply RunInput, RunOutput, etc.
 
   constructor(
@@ -27,13 +25,11 @@ export class AiAgentClient {
     chatModel: BaseChatModel,
     tools: StructuredTool[] = []
   ) {
-    this.name = aiAgentArgs.name
-    this.instructions = aiAgentArgs.instructions
-
     this.agent = createReactAgent({
+      name: aiAgentArgs.name,
       llm: chatModel,
       tools: tools,
-      prompt: this.instructions,
+      prompt: aiAgentArgs.instructions,
     })
   }
 

package/src/constants.ts

@@ -1,8 +1,9 @@
 export const HUBTYPE_API_URL =
   process.env.HUBTYPE_API_URL || 'https://api.hubtype.com'
 
-export const AZURE_OPENAI_API_KEY = process.env.AZURE_OPENAI_API_KEY
-export const AZURE_OPENAI_API_DEPLOYMENT_NAME =
-  process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME
-export const AZURE_OPENAI_API_VERSION = process.env.AZURE_OPENAI_API_VERSION
+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 =
+  process.env.AZURE_OPENAI_API_DEPLOYMENT_NAME || 'gpt-4o-mini_p1'
+export const AZURE_OPENAI_API_VERSION =
+  process.env.AZURE_OPENAI_API_VERSION || '2024-12-01-preview'

package/src/hubtype-api-client.ts

@@ -0,0 +1,43 @@
+import { BotContext } from '@botonic/core'
+import axios from 'axios'
+
+import { HUBTYPE_API_URL } from './constants'
+import { AgenticInputMessage } from './types'
+
+export class HubtypeApiClient {
+  private readonly authToken: string
+
+  constructor(authToken: string) {
+    this.authToken = authToken
+  }
+
+  async getMessages(
+    request: BotContext,
+    memoryLength: number
+  ): Promise<AgenticInputMessage[]> {
+    const url = `${HUBTYPE_API_URL}/external/v1/ai/agent/message_history/`
+    const headers = {
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${this.authToken}`,
+    }
+    const params = {
+      last_message_id: request.input.message_id,
+      num_messages: memoryLength,
+    }
+
+    try {
+      const messages = await axios.get<{ messages: AgenticInputMessage[] }>(
+        url,
+        {
+          headers,
+          params,
+        }
+      )
+
+      return messages.data.messages
+    } catch (error) {
+      console.error(error)
+      throw new Error('Failed to get messages from Hubtype')
+    }
+  }
+}

package/src/hubtype-client.ts

@@ -4,7 +4,7 @@ import axios from 'axios'
 import { HUBTYPE_API_URL } from './constants'
 import { AgenticInputMessage } from './types'
 
-export class HubtypeClient {
+export class HubtypeApiClient {
   private readonly authToken: string
 
   constructor(authToken: string) {

package/src/index.ts

@@ -1,22 +1,28 @@
 import { BotContext, Plugin } from '@botonic/core'
 
 import { AiAgentClient } from './ai-agent-client'
-import { HubtypeClient } from './hubtype-client'
-import { MANDATORY_TOOLS } from './tools'
+import { HubtypeApiClient } from './hubtype-api-client'
+import { createCustomTool } from './tools/custom'
+import { MANDATORY_TOOLS } from './tools/default'
 import {
+  AgenticInputMessage,
   AgenticOutputMessage,
   AiAgentArgs,
+  CustomTool,
   PluginAiAgentOptions,
   UserMessage,
 } from './types'
-import { loadChatModel } from './utils'
+import { AiProvider, loadChatModel } from './utils'
 
 const isProd = process.env.NODE_ENV === 'production'
 
 export default class BotonicPluginAiAgents implements Plugin {
   private readonly authToken?: string
+  public customTools: CustomTool[] = []
+
   constructor(options?: PluginAiAgentOptions) {
     this.authToken = options?.authToken
+    this.customTools = options?.customTools || []
   }
 
   pre(): void {
@@ -33,17 +39,15 @@ export default class BotonicPluginAiAgents implements Plugin {
         throw new Error('Auth token is required')
       }
 
-      const hubtypeClient = new HubtypeClient(authToken)
-      const messages = isProd
-        ? await hubtypeClient.getMessages(request, 10)
-        : [{ role: 'user', content: request.input.data } as UserMessage]
+      const messages = await this.getMessages(request, authToken, 10)
+
+      const customTools = this.customTools.map(customTool =>
+        createCustomTool(customTool)
+      )
 
-      const customTools = []
-      const chatModel = loadChatModel('azureOpenAI')
-      const aiAgentClient = new AiAgentClient(aiAgentArgs, chatModel, [
-        ...customTools,
-        ...MANDATORY_TOOLS,
-      ])
+      const tools = [...customTools, ...MANDATORY_TOOLS]
+      const chatModel = loadChatModel(AiProvider.AzureOpenAI)
+      const aiAgentClient = new AiAgentClient(aiAgentArgs, chatModel, tools)
 
       return await aiAgentClient.runAgent(messages)
     } catch (error) {
@@ -51,6 +55,19 @@ export default class BotonicPluginAiAgents implements Plugin {
       return undefined
     }
   }
+
+  private async getMessages(
+    request: BotContext,
+    authToken: string,
+    memoryLength: number
+  ): Promise<AgenticInputMessage[]> {
+    if (isProd) {
+      const hubtypeClient = new HubtypeApiClient(authToken)
+      return await hubtypeClient.getMessages(request, memoryLength)
+    }
+
+    return [{ role: 'user', content: request.input.data } as UserMessage]
+  }
 }
 
 export * from './types'

package/src/tools/custom.ts

@@ -0,0 +1,14 @@
+import { DynamicStructuredTool, tool } from '@langchain/core/tools'
+
+import { CustomTool } from '../types'
+
+export function createCustomTool(
+  customTool: CustomTool
+): DynamicStructuredTool {
+  return tool(customTool.func, {
+    name: customTool.name,
+    description: customTool.description,
+    schema: customTool.schema,
+    returnDirect: customTool.returnDirect || false,
+  })
+}

package/src/{tools.ts → tools/default.ts}

@@ -44,6 +44,7 @@ export const MANDATORY_TOOLS = [
   outOfContext,
   finishConversation,
 ]
+
 export const EXIT_TOOLS = [
   handoffToHumanAgent.name,
   outOfContext.name,

package/src/types.ts

@@ -1,6 +1,17 @@
+import { ZodSchema } from 'zod'
 export interface PluginAiAgentOptions {
   authToken?: string
+  customTools?: CustomTool[]
 }
+
+export interface CustomTool {
+  name: string
+  description: string
+  schema: ZodSchema
+  returnDirect?: boolean
+  func: (input?: any) => Promise<any>
+}
+
 export interface AiAgentArgs {
   name: string
   instructions: string

package/src/utils.ts

@@ -7,9 +7,13 @@ import {
   AZURE_OPENAI_API_VERSION,
 } from './constants'
 
-export function loadChatModel(provider: 'azureOpenAI') {
+export enum AiProvider {
+  AzureOpenAI = 'azureOpenAI',
+}
+
+export function loadChatModel(provider: AiProvider) {
   switch (provider) {
-    case 'azureOpenAI':
+    case AiProvider.AzureOpenAI:
       return new AzureChatOpenAI({
         azureOpenAIApiVersion: AZURE_OPENAI_API_VERSION,
         azureOpenAIApiKey: AZURE_OPENAI_API_KEY,

package/lib/cjs/tools.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/tools.ts"],"names":[],"mappings":";;;;AAAA,iDAA4C;AAC5C,6BAAuB;AAEV,QAAA,mBAAmB,GAAG,IAAA,YAAI,EACrC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,qBAAqB;IAC3B,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,wEAAwE;IAC1E,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,YAAY,GAAG,IAAA,YAAI,EAC9B,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,cAAc;IACpB,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,+IAA+I;IACjJ,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,kBAAkB,GAAG,IAAA,YAAI,EACpC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,oBAAoB;IAC1B,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EAAE,uDAAuD;IACpE,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAEY,QAAA,eAAe,GAAG;IAC7B,2BAAmB;IACnB,oBAAY;IACZ,0BAAkB;CACnB,CAAA;AACY,QAAA,UAAU,GAAG;IACxB,2BAAmB,CAAC,IAAI;IACxB,oBAAY,CAAC,IAAI;IACjB,0BAAkB,CAAC,IAAI;CACxB,CAAA"}

package/lib/esm/tools.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/tools.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAA;AAC5C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,MAAM,CAAC,MAAM,mBAAmB,GAAG,IAAI,CACrC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,qBAAqB;IAC3B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,wEAAwE;IAC1E,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,CAC9B,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,cAAc;IACpB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EACT,+IAA+I;IACjJ,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,CACpC,GAAS,EAAE;IACT,OAAO,IAAI,CAAA;AACb,CAAC,CAAA,EACD;IACE,IAAI,EAAE,oBAAoB;IAC1B,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;IACpB,WAAW,EAAE,uDAAuD;IACpE,YAAY,EAAE,IAAI;CACnB,CACF,CAAA;AAED,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,mBAAmB;IACnB,YAAY;IACZ,kBAAkB;CACnB,CAAA;AACD,MAAM,CAAC,MAAM,UAAU,GAAG;IACxB,mBAAmB,CAAC,IAAI;IACxB,YAAY,CAAC,IAAI;IACjB,kBAAkB,CAAC,IAAI;CACxB,CAAA"}