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

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

@@ -1,12 +1,11 @@
-import { BotContext } from '@botonic/core';
-import { AiAgentArgs, AiAgentResponse } from './types';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import { StructuredTool } from '@langchain/core/tools';
+import { CompiledStateGraph } from '@langchain/langgraph';
+import { AgenticInputMessage, AgenticOutputMessage, AiAgentArgs } from './types';
 export declare class AiAgentClient {
-    private url;
-    constructor();
-    getInference(request: BotContext, aiAgentArgs: AiAgentArgs): Promise<AiAgentResponse>;
-    private agentTestInference;
-    private agentInference;
-    private getConfig;
-    private getData;
-    private getDataTest;
+    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

@@ -2,77 +2,49 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AiAgentClient = void 0;
 const tslib_1 = require("tslib");
-/* eslint-disable @typescript-eslint/naming-convention */
-const core_1 = require("@botonic/core");
-const axios_1 = tslib_1.__importDefault(require("axios"));
-const constants_1 = require("./constants");
-const errors_1 = require("./errors");
+const messages_1 = require("@langchain/core/messages");
+const prebuilt_1 = require("@langchain/langgraph/prebuilt");
+const tools_1 = require("./tools");
 class AiAgentClient {
-    constructor() {
-        this.url = `${constants_1.HUBTYPE_API_URL}/external/v1/ai/agent`;
-    }
-    getInference(request, aiAgentArgs) {
-        return tslib_1.__awaiter(this, void 0, void 0, function* () {
-            if ((0, core_1.isDev)(request.session)) {
-                return this.agentTestInference(request, aiAgentArgs);
-            }
-            return this.agentInference(request, aiAgentArgs);
-        });
-    }
-    agentTestInference(request, aiAgentArgs) {
-        return tslib_1.__awaiter(this, void 0, void 0, function* () {
-            const config = this.getConfig(request);
-            const data = this.getDataTest(request, aiAgentArgs);
-            const response = yield axios_1.default.post(`${this.url}/test/`, data, config);
-            return response.data;
+    constructor(aiAgentArgs, chatModel, tools = []) {
+        this.name = aiAgentArgs.name;
+        this.instructions = aiAgentArgs.instructions;
+        this.agent = (0, prebuilt_1.createReactAgent)({
+            llm: chatModel,
+            tools: tools,
+            prompt: this.instructions,
         });
     }
-    agentInference(request, aiAgentArgs) {
+    runAgent(_messages) {
         return tslib_1.__awaiter(this, void 0, void 0, function* () {
-            const config = this.getConfig(request);
-            const data = this.getData(request, aiAgentArgs);
-            const response = yield axios_1.default.post(`${this.url}/run/`, data, config);
-            return response.data;
+            const response = yield this.agent.invoke({
+                messages: _messages.map((message) => ({
+                    role: message.role,
+                    content: message.content,
+                })),
+            });
+            const lastMessage = response.messages.at(-1);
+            if (lastMessage instanceof messages_1.ToolMessage) {
+                if (lastMessage.name && tools_1.EXIT_TOOLS.includes(lastMessage.name)) {
+                    return {
+                        role: 'exit',
+                    };
+                }
+                return {
+                    role: 'tool',
+                    toolName: lastMessage.name,
+                    toolOutput: lastMessage.content,
+                };
+            }
+            if (lastMessage instanceof messages_1.AIMessage) {
+                return {
+                    role: 'assistant',
+                    content: lastMessage.content,
+                };
+            }
+            throw new Error('Last message is not a valid message');
         });
     }
-    getConfig(request) {
-        return {
-            headers: {
-                Authorization: `Bearer ${request.session._access_token}`,
-                'Content-Type': 'application/json',
-            },
-        };
-    }
-    getData(request, aiAgentArgs) {
-        if (!request.input.data) {
-            throw new errors_1.AiAgentError('No input data provided');
-        }
-        return {
-            message: request.input.message_id,
-            memory_length: 10,
-            name: aiAgentArgs.name,
-            instructions: aiAgentArgs.instructions,
-        };
-    }
-    getDataTest(request, aiAgentArgs) {
-        if (!request.input.data) {
-            throw new errors_1.AiAgentError('No input data provided');
-        }
-        // TODO: We can get messages from localStorage in Dev mode and transform them to the correct format {role: MessageRole, content: string}
-        // if (isDev(request.session)) {
-        //   const messages = localStorage.getItem('botonicState').
-        // }
-        return {
-            messages: [
-                {
-                    role: 'user',
-                    content: request.input.data,
-                },
-            ],
-            name: aiAgentArgs.name,
-            instructions: aiAgentArgs.instructions,
-        };
-    }
 }
 exports.AiAgentClient = AiAgentClient;
 //# sourceMappingURL=ai-agent-client.js.map

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":";;;;AAAA,yDAAyD;AACzD,wCAAiD;AACjD,0DAAyB;AAEzB,2CAA6C;AAC7C,qCAAuC;AASvC,MAAa,aAAa;IAGxB;QACE,IAAI,CAAC,GAAG,GAAG,GAAG,2BAAe,uBAAuB,CAAA;IACtD,CAAC;IAEK,YAAY,CAChB,OAAmB,EACnB,WAAwB;;YAExB,IAAI,IAAA,YAAK,EAAC,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC1B,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;aACrD;YAED,OAAO,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;QAClD,CAAC;KAAA;IAEa,kBAAkB,CAC9B,OAAmB,EACnB,WAAwB;;YAExB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;YACnD,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,IAAI,CAC/B,GAAG,IAAI,CAAC,GAAG,QAAQ,EACnB,IAAI,EACJ,MAAM,CACP,CAAA;YAED,OAAO,QAAQ,CAAC,IAAI,CAAA;QACtB,CAAC;KAAA;IAEa,cAAc,CAC1B,OAAmB,EACnB,WAAwB;;YAExB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;YAC/C,MAAM,QAAQ,GAAG,MAAM,eAAK,CAAC,IAAI,CAC/B,GAAG,IAAI,CAAC,GAAG,OAAO,EAClB,IAAI,EACJ,MAAM,CACP,CAAA;YAED,OAAO,QAAQ,CAAC,IAAI,CAAA;QACtB,CAAC;KAAA;IAEO,SAAS,CAAC,OAAmB;QACnC,OAAO;YACL,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,OAAO,CAAC,OAAO,CAAC,aAAa,EAAE;gBACxD,cAAc,EAAE,kBAAkB;aACnC;SACF,CAAA;IACH,CAAC;IAEO,OAAO,CACb,OAAmB,EACnB,WAAwB;QAExB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE;YACvB,MAAM,IAAI,qBAAY,CAAC,wBAAwB,CAAC,CAAA;SACjD;QAED,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;YACjC,aAAa,EAAE,EAAE;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,YAAY,EAAE,WAAW,CAAC,YAAY;SACvC,CAAA;IACH,CAAC;IAEO,WAAW,CACjB,OAAmB,EACnB,WAAwB;QAExB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE;YACvB,MAAM,IAAI,qBAAY,CAAC,wBAAwB,CAAC,CAAA;SACjD;QAED,wIAAwI;QACxI,gCAAgC;QAChC,2DAA2D;QAC3D,IAAI;QAEJ,OAAO;YACL,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI;iBAC5B;aACF;YACD,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,YAAY,EAAE,WAAW,CAAC,YAAY;SACvC,CAAA;IACH,CAAC;CACF;AAjGD,sCAiGC"}
+{"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"}

package/lib/cjs/constants.d.ts

@@ -1 +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;

package/lib/cjs/constants.js

@@ -1,5 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.HUBTYPE_API_URL = void 0;
+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.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_BASE = process.env.AZURE_OPENAI_API_BASE;
 //# 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"}
+{"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"}

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

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

package/lib/cjs/hubtype-client.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HubtypeClient = void 0;
+const tslib_1 = require("tslib");
+const axios_1 = tslib_1.__importDefault(require("axios"));
+const constants_1 = require("./constants");
+class HubtypeClient {
+    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.HubtypeClient = HubtypeClient;
+//# sourceMappingURL=hubtype-client.js.map

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

@@ -0,0 +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"}

package/lib/cjs/index.d.ts

@@ -1,9 +1,9 @@
 import { BotContext, Plugin } from '@botonic/core';
-import { AiAgentClient } from './ai-agent-client';
-import { AiAgentArgs, AiAgentResponse } from './types';
+import { AgenticOutputMessage, AiAgentArgs, PluginAiAgentOptions } from './types';
 export default class BotonicPluginAiAgents implements Plugin {
-    aiAgentClient: AiAgentClient;
-    constructor();
+    private readonly authToken?;
+    constructor(options?: PluginAiAgentOptions);
     pre(): void;
-    getInference(request: BotContext, aiAgentArgs: AiAgentArgs): Promise<AiAgentResponse>;
+    getInference(request: BotContext, aiAgentArgs: AiAgentArgs): Promise<AgenticOutputMessage | undefined>;
 }
+export * from './types';

package/lib/cjs/index.js

@@ -2,18 +2,43 @@
 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 utils_1 = require("./utils");
+const isProd = process.env.NODE_ENV === 'production';
 class BotonicPluginAiAgents {
-    constructor() {
-        this.aiAgentClient = new ai_agent_client_1.AiAgentClient();
+    constructor(options) {
+        this.authToken = options === null || options === void 0 ? void 0 : options.authToken;
     }
     pre() {
         return;
     }
     getInference(request, aiAgentArgs) {
         return tslib_1.__awaiter(this, void 0, void 0, function* () {
-            return this.aiAgentClient.getInference(request, aiAgentArgs);
+            try {
+                const authToken = isProd ? request.session._access_token : this.authToken;
+                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,
+                ]);
+                return yield aiAgentClient.runAgent(messages);
+            }
+            catch (error) {
+                console.error('error plugin returns undefined', error);
+                return undefined;
+            }
         });
     }
 }
 exports.default = BotonicPluginAiAgents;
+tslib_1.__exportStar(require("./types"), exports);
 //# sourceMappingURL=index.js.map

package/lib/cjs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AAEA,uDAAiD;AAGjD,MAAqB,qBAAqB;IAGxC;QACE,IAAI,CAAC,aAAa,GAAG,IAAI,+BAAa,EAAE,CAAA;IAC1C,CAAC;IAED,GAAG;QACD,OAAM;IACR,CAAC;IAEK,YAAY,CAChB,OAAmB,EACnB,WAAwB;;YAExB,OAAO,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;QAC9D,CAAC;KAAA;CACF;AAjBD,wCAiBC"}
+{"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"}

package/lib/cjs/tools.d.ts

@@ -0,0 +1,6 @@
+import { z } from 'zod';
+export declare const handoffToHumanAgent: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
+export declare const outOfContext: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
+export declare const finishConversation: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>;
+export declare const MANDATORY_TOOLS: import("@langchain/core/tools").DynamicStructuredTool<z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>, {}, {}, null>[];
+export declare const EXIT_TOOLS: string[];

package/lib/cjs/tools.js

@@ -0,0 +1,41 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EXIT_TOOLS = exports.MANDATORY_TOOLS = exports.finishConversation = exports.outOfContext = exports.handoffToHumanAgent = void 0;
+const tslib_1 = require("tslib");
+const tools_1 = require("@langchain/core/tools");
+const zod_1 = require("zod");
+exports.handoffToHumanAgent = (0, tools_1.tool)(() => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    return null;
+}), {
+    name: 'handoffToHumanAgent',
+    schema: zod_1.z.object({}),
+    description: 'Use this tool strictly when the user wants to talk with a human agent.',
+    returnDirect: true,
+});
+exports.outOfContext = (0, tools_1.tool)(() => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    return null;
+}), {
+    name: 'outOfContext',
+    schema: zod_1.z.object({}),
+    description: 'Use this when the user talks about something that is clearly out of your context. Example: "I want to buy a car" when you are a travel agent.',
+    returnDirect: true,
+});
+exports.finishConversation = (0, tools_1.tool)(() => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    return null;
+}), {
+    name: 'finishConversation',
+    schema: zod_1.z.object({}),
+    description: 'Use this when the user wants to end the conversation.',
+    returnDirect: true,
+});
+exports.MANDATORY_TOOLS = [
+    exports.handoffToHumanAgent,
+    exports.outOfContext,
+    exports.finishConversation,
+];
+exports.EXIT_TOOLS = [
+    exports.handoffToHumanAgent.name,
+    exports.outOfContext.name,
+    exports.finishConversation.name,
+];
+//# sourceMappingURL=tools.js.map

package/lib/cjs/tools.js.map

@@ -0,0 +1 @@
+{"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/cjs/types.d.ts

@@ -1,31 +1,29 @@
+export interface PluginAiAgentOptions {
+    authToken?: string;
+}
 export interface AiAgentArgs {
     name: string;
     instructions: string;
 }
-export interface Config {
-    headers: {
-        Authorization: string;
-        'Content-Type': string;
-    };
+export interface AgenticBaseMessage {
+    role: 'assistant' | 'user' | 'tool' | 'exit';
 }
-export interface AiAgentRequestDataTest {
-    messages: {
-        role: MessageRole;
-        content: string;
-    }[];
-    name: string;
-    instructions: string;
+export interface AssistantMessage extends AgenticBaseMessage {
+    role: 'assistant';
+    content: string;
 }
-export type MessageRole = 'user' | 'assistant';
-export interface AiAgentRequestData {
-    message: string;
-    memory_length: number;
-    name: string;
-    instructions: string;
+export interface UserMessage extends AgenticBaseMessage {
+    role: 'user';
+    content: string;
+}
+export interface ToolMessage extends AgenticBaseMessage {
+    role: 'tool';
+    toolName: string;
+    toolOutput: string | null;
 }
-export interface AiAgentResponse {
-    message: {
-        role: string;
-        content: string;
-    };
+export interface ExitMessage extends AgenticBaseMessage {
+    role: 'exit';
 }
+export type AgenticInputMessage = AssistantMessage | UserMessage;
+export type AgenticOutputMessage = AssistantMessage | ToolMessage | ExitMessage;
+export type AgenticMessage = AssistantMessage | UserMessage | ToolMessage | ExitMessage;

package/lib/cjs/utils.d.ts

@@ -0,0 +1,2 @@
+import { AzureChatOpenAI } from '@langchain/openai';
+export declare function loadChatModel(provider: 'azureOpenAI'): AzureChatOpenAI;

package/lib/cjs/utils.js

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadChatModel = void 0;
+const openai_1 = require("@langchain/openai");
+const constants_1 = require("./constants");
+function loadChatModel(provider) {
+    switch (provider) {
+        case 'azureOpenAI':
+            return new openai_1.AzureChatOpenAI({
+                azureOpenAIApiVersion: constants_1.AZURE_OPENAI_API_VERSION,
+                azureOpenAIApiKey: constants_1.AZURE_OPENAI_API_KEY,
+                azureOpenAIApiDeploymentName: constants_1.AZURE_OPENAI_API_DEPLOYMENT_NAME,
+                azureOpenAIEndpoint: constants_1.AZURE_OPENAI_API_BASE,
+                temperature: 0,
+                // other params...
+            });
+        default:
+            throw new Error(`Provider ${provider} not supported`);
+    }
+}
+exports.loadChatModel = loadChatModel;
+//# sourceMappingURL=utils.js.map

package/lib/cjs/utils.js.map

@@ -0,0 +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"}

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

@@ -1,12 +1,11 @@
-import { BotContext } from '@botonic/core';
-import { AiAgentArgs, AiAgentResponse } from './types';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import { StructuredTool } from '@langchain/core/tools';
+import { CompiledStateGraph } from '@langchain/langgraph';
+import { AgenticInputMessage, AgenticOutputMessage, AiAgentArgs } from './types';
 export declare class AiAgentClient {
-    private url;
-    constructor();
-    getInference(request: BotContext, aiAgentArgs: AiAgentArgs): Promise<AiAgentResponse>;
-    private agentTestInference;
-    private agentInference;
-    private getConfig;
-    private getData;
-    private getDataTest;
+    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,74 +1,46 @@
 import { __awaiter } from "tslib";
-/* eslint-disable @typescript-eslint/naming-convention */
-import { isDev } from '@botonic/core';
-import axios from 'axios';
-import { HUBTYPE_API_URL } from './constants';
-import { AiAgentError } from './errors';
+import { AIMessage as LangchainAIMessage, ToolMessage as LangchainToolMessage, } from '@langchain/core/messages';
+import { createReactAgent } from '@langchain/langgraph/prebuilt';
+import { EXIT_TOOLS } from './tools';
 export class AiAgentClient {
-    constructor() {
-        this.url = `${HUBTYPE_API_URL}/external/v1/ai/agent`;
-    }
-    getInference(request, aiAgentArgs) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (isDev(request.session)) {
-                return this.agentTestInference(request, aiAgentArgs);
-            }
-            return this.agentInference(request, aiAgentArgs);
-        });
-    }
-    agentTestInference(request, aiAgentArgs) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const config = this.getConfig(request);
-            const data = this.getDataTest(request, aiAgentArgs);
-            const response = yield axios.post(`${this.url}/test/`, data, config);
-            return response.data;
+    constructor(aiAgentArgs, chatModel, tools = []) {
+        this.name = aiAgentArgs.name;
+        this.instructions = aiAgentArgs.instructions;
+        this.agent = createReactAgent({
+            llm: chatModel,
+            tools: tools,
+            prompt: this.instructions,
         });
     }
-    agentInference(request, aiAgentArgs) {
+    runAgent(_messages) {
         return __awaiter(this, void 0, void 0, function* () {
-            const config = this.getConfig(request);
-            const data = this.getData(request, aiAgentArgs);
-            const response = yield axios.post(`${this.url}/run/`, data, config);
-            return response.data;
+            const response = yield this.agent.invoke({
+                messages: _messages.map((message) => ({
+                    role: message.role,
+                    content: message.content,
+                })),
+            });
+            const lastMessage = response.messages.at(-1);
+            if (lastMessage instanceof LangchainToolMessage) {
+                if (lastMessage.name && EXIT_TOOLS.includes(lastMessage.name)) {
+                    return {
+                        role: 'exit',
+                    };
+                }
+                return {
+                    role: 'tool',
+                    toolName: lastMessage.name,
+                    toolOutput: lastMessage.content,
+                };
+            }
+            if (lastMessage instanceof LangchainAIMessage) {
+                return {
+                    role: 'assistant',
+                    content: lastMessage.content,
+                };
+            }
+            throw new Error('Last message is not a valid message');
         });
     }
-    getConfig(request) {
-        return {
-            headers: {
-                Authorization: `Bearer ${request.session._access_token}`,
-                'Content-Type': 'application/json',
-            },
-        };
-    }
-    getData(request, aiAgentArgs) {
-        if (!request.input.data) {
-            throw new AiAgentError('No input data provided');
-        }
-        return {
-            message: request.input.message_id,
-            memory_length: 10,
-            name: aiAgentArgs.name,
-            instructions: aiAgentArgs.instructions,
-        };
-    }
-    getDataTest(request, aiAgentArgs) {
-        if (!request.input.data) {
-            throw new AiAgentError('No input data provided');
-        }
-        // TODO: We can get messages from localStorage in Dev mode and transform them to the correct format {role: MessageRole, content: string}
-        // if (isDev(request.session)) {
-        //   const messages = localStorage.getItem('botonicState').
-        // }
-        return {
-            messages: [
-                {
-                    role: 'user',
-                    content: request.input.data,
-                },
-            ],
-            name: aiAgentArgs.name,
-            instructions: aiAgentArgs.instructions,
-        };
-    }
 }
 //# sourceMappingURL=ai-agent-client.js.map

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":";AAAA,yDAAyD;AACzD,OAAO,EAAc,KAAK,EAAE,MAAM,eAAe,CAAA;AACjD,OAAO,KAAK,MAAM,OAAO,CAAA;AAEzB,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAA;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAA;AASvC,MAAM,OAAO,aAAa;IAGxB;QACE,IAAI,CAAC,GAAG,GAAG,GAAG,eAAe,uBAAuB,CAAA;IACtD,CAAC;IAEK,YAAY,CAChB,OAAmB,EACnB,WAAwB;;YAExB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC1B,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;aACrD;YAED,OAAO,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;QAClD,CAAC;KAAA;IAEa,kBAAkB,CAC9B,OAAmB,EACnB,WAAwB;;YAExB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;YACnD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,GAAG,IAAI,CAAC,GAAG,QAAQ,EACnB,IAAI,EACJ,MAAM,CACP,CAAA;YAED,OAAO,QAAQ,CAAC,IAAI,CAAA;QACtB,CAAC;KAAA;IAEa,cAAc,CAC1B,OAAmB,EACnB,WAAwB;;YAExB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;YACtC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,CAAC,CAAA;YAC/C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,GAAG,IAAI,CAAC,GAAG,OAAO,EAClB,IAAI,EACJ,MAAM,CACP,CAAA;YAED,OAAO,QAAQ,CAAC,IAAI,CAAA;QACtB,CAAC;KAAA;IAEO,SAAS,CAAC,OAAmB;QACnC,OAAO;YACL,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,OAAO,CAAC,OAAO,CAAC,aAAa,EAAE;gBACxD,cAAc,EAAE,kBAAkB;aACnC;SACF,CAAA;IACH,CAAC;IAEO,OAAO,CACb,OAAmB,EACnB,WAAwB;QAExB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE;YACvB,MAAM,IAAI,YAAY,CAAC,wBAAwB,CAAC,CAAA;SACjD;QAED,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,UAAU;YACjC,aAAa,EAAE,EAAE;YACjB,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,YAAY,EAAE,WAAW,CAAC,YAAY;SACvC,CAAA;IACH,CAAC;IAEO,WAAW,CACjB,OAAmB,EACnB,WAAwB;QAExB,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE;YACvB,MAAM,IAAI,YAAY,CAAC,wBAAwB,CAAC,CAAA;SACjD;QAED,wIAAwI;QACxI,gCAAgC;QAChC,2DAA2D;QAC3D,IAAI;QAEJ,OAAO;YACL,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI;iBAC5B;aACF;YACD,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,YAAY,EAAE,WAAW,CAAC,YAAY;SACvC,CAAA;IACH,CAAC;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,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"}

package/lib/esm/constants.d.ts

@@ -1 +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;

package/lib/esm/constants.js

@@ -1,2 +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_BASE = process.env.AZURE_OPENAI_API_BASE;
 //# 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"}
+{"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"}

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

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