@botonic/plugin-ai-agents 0.40.0-alpha.0 → 0.40.0

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

@@ -5,9 +5,27 @@ export class HubtypeApiClient {
     constructor(authToken) {
         this.authToken = authToken;
     }
+    retrieveSimilarChunks(query, sources) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const url = `${HUBTYPE_API_URL}/external/v1/ai/knowledge_base/similar_chunks/`;
+            const headers = {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${this.authToken}`,
+            };
+            const params = {
+                query,
+                source_ids: sources,
+                num_chunks: 5,
+            };
+            const response = yield axios.post(url, params, { headers });
+            return response.data.chunks.map((chunk) => chunk.text);
+        });
+    }
     getLocalMessages(maxMemoryLength) {
         return __awaiter(this, void 0, void 0, function* () {
-            const localBotonicState = localStorage.getItem('botonicState');
+            const localBotonicState = typeof globalThis !== 'undefined' && globalThis.localStorage
+                ? globalThis.localStorage.getItem('botonicState')
+                : null;
             const botonicState = JSON.parse(localBotonicState || '{}');
             const messages = botonicState.messages;
             const filteredMessages = messages

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

@@ -1 +1 @@
-{"version":3,"file":"hubtype-api-client.js","sourceRoot":"","sources":["../../src/hubtype-api-client.ts"],"names":[],"mappings":";AAEA,OAAO,KAAK,MAAM,OAAO,CAAA;AAEzB,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAA;AAe7C,MAAM,OAAO,gBAAgB;IAG3B,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAEK,gBAAgB,CACpB,eAAuB;;YAEvB,MAAM,iBAAiB,GAAG,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;YAC9D,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,IAAI,CAAC,CAAA;YAC1D,MAAM,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAA;YACtC,MAAM,gBAAgB,GAAG,QAAQ;iBAC9B,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;iBACpC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACf,IAAI,EAAE,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW;gBACtD,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI;aAC3B,CAAC,CAAC;iBACF,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAA;YAC9C,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAA;QACjD,CAAC;KAAA;IAEK,WAAW,CACf,OAAmB,EACnB,eAAuB;;YAEvB,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,eAAe;aAC9B,CAAA;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAiC,GAAG,EAAE;oBACpE,OAAO;oBACP,MAAM;iBACP,CAAC,CAAA;gBACF,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;gBACvC,OAAO,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAA;aAC5D;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;aACvD;QACH,CAAC;KAAA;IAEO,aAAa,CAAC,OAAuB;QAC3C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE;YAC3B,OAAO;gBACL,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,MAAM;aACb,CAAA;SACF;aAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE;YACvC,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,aAAa;wBACnB,IAAI,EAAE,OAAO,CAAC,OAAO;qBACtB;iBACF;gBACD,MAAM,EAAE,WAAW;aACpB,CAAA;SACF;aAAM;YACL,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAA;SACxC;IACH,CAAC;CACF"}
+{"version":3,"file":"hubtype-api-client.js","sourceRoot":"","sources":["../../src/hubtype-api-client.ts"],"names":[],"mappings":";AAEA,OAAO,KAAK,MAAM,OAAO,CAAA;AAEzB,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAA;AAe7C,MAAM,OAAO,gBAAgB;IAG3B,YAAY,SAAiB;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAA;IAC5B,CAAC;IAEK,qBAAqB,CACzB,KAAa,EACb,OAAiB;;YAEjB,MAAM,GAAG,GAAG,GAAG,eAAe,gDAAgD,CAAA;YAC9E,MAAM,OAAO,GAAG;gBACd,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE;aAC1C,CAAA;YACD,MAAM,MAAM,GAAG;gBACb,KAAK;gBACL,UAAU,EAAE,OAAO;gBACnB,UAAU,EAAE,CAAC;aACd,CAAA;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,OAAO,EAAE,CAAC,CAAA;YAC3D,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAU,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAC7D,CAAC;KAAA;IAEK,gBAAgB,CACpB,eAAuB;;YAEvB,MAAM,iBAAiB,GACrB,OAAO,UAAU,KAAK,WAAW,IAAI,UAAU,CAAC,YAAY;gBAC1D,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,cAAc,CAAC;gBACjD,CAAC,CAAC,IAAI,CAAA;YACV,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAiB,IAAI,IAAI,CAAC,CAAA;YAC1D,MAAM,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAA;YACtC,MAAM,gBAAgB,GAAG,QAAQ;iBAC9B,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;iBACpC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACf,IAAI,EAAE,OAAO,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW;gBACtD,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI;aAC3B,CAAC,CAAC;iBACF,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAA;YAC9C,OAAO,gBAAgB,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAA;QACjD,CAAC;KAAA;IAEK,WAAW,CACf,OAAmB,EACnB,eAAuB;;YAEvB,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,eAAe;aAC9B,CAAA;YAED,IAAI;gBACF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAiC,GAAG,EAAE;oBACpE,OAAO;oBACP,MAAM;iBACP,CAAC,CAAA;gBACF,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAA;gBACvC,OAAO,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAA;aAC5D;YAAC,OAAO,KAAK,EAAE;gBACd,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpB,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;aACvD;QACH,CAAC;KAAA;IAEO,aAAa,CAAC,OAAuB;QAC3C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE;YAC3B,OAAO;gBACL,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,MAAM;aACb,CAAA;SACF;aAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE;YACvC,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,aAAa;wBACnB,IAAI,EAAE,OAAO,CAAC,OAAO;qBACtB;iBACF;gBACD,MAAM,EAAE,WAAW;aACpB,CAAA;SACF;aAAM;YACL,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAA;SACxC;IACH,CAAC;CACF"}

package/lib/esm/index.d.ts

@@ -1,5 +1,5 @@
-import { BotContext, Plugin } from '@botonic/core';
-import { AiAgentArgs, CustomTool, InferenceResponse, PluginAiAgentOptions } from './types';
+import { AiAgentArgs, BotContext, Plugin } from '@botonic/core';
+import { CustomTool, InferenceResponse, PluginAiAgentOptions } from './types';
 export default class BotonicPluginAiAgents implements Plugin {
     private readonly authToken?;
     toolDefinitions: CustomTool[];

package/lib/esm/index.js

@@ -24,10 +24,11 @@ export default class BotonicPluginAiAgents {
                     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, request.session.user.contact_info || {}, aiAgentArgs.inputGuardrailRules || []).build();
+                const agent = new AIAgentBuilder(aiAgentArgs.name, aiAgentArgs.instructions, tools, request.session.user.contact_info || {}, aiAgentArgs.inputGuardrailRules || [], aiAgentArgs.sourceIds || []).build();
                 const messages = yield this.getMessages(request, authToken, MAX_MEMORY_LENGTH);
                 const context = {
                     authToken,
+                    sources: aiAgentArgs.sourceIds || [],
                 };
                 const runner = new AIAgentRunner(agent);
                 return yield runner.run(messages, context);

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,cAAc,EAAE,MAAM,iBAAiB,CAAA;AAChD,OAAO,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAA;AACvD,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,EACL,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,EACvC,WAAW,CAAC,mBAAmB,IAAI,EAAE,CACtC,CAAC,KAAK,EAAE,CAAA;gBAET,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CACrC,OAAO,EACP,SAAS,EACT,iBAAiB,CAClB,CAAA;gBACD,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;oBACL,QAAQ,EAAE,EAAE;oBACZ,aAAa,EAAE,EAAE;oBACjB,YAAY,EAAE,CAAC;oBACf,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,IAAI;oBACX,wBAAwB,EAAE,EAAE;oBAC5B,yBAAyB,EAAE,EAAE;iBAC9B,CAAA;aACF;;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"}
+{"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,iBAAiB,EAAE,MAAM,aAAa,CAAA;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAA;AACvD,OAAO,EAAE,WAAW,EAAE,MAAM,UAAU,CAAA;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAA;AAUxC,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,EACL,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,IAAI,EAAE,EACvC,WAAW,CAAC,mBAAmB,IAAI,EAAE,EACrC,WAAW,CAAC,SAAS,IAAI,EAAE,CAC5B,CAAC,KAAK,EAAE,CAAA;gBAET,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CACrC,OAAO,EACP,SAAS,EACT,iBAAiB,CAClB,CAAA;gBACD,MAAM,OAAO,GAAY;oBACvB,SAAS;oBACT,OAAO,EAAE,WAAW,CAAC,SAAS,IAAI,EAAE;iBACrC,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;oBACL,QAAQ,EAAE,EAAE;oBACZ,aAAa,EAAE,EAAE;oBACjB,YAAY,EAAE,CAAC;oBACf,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,IAAI;oBACX,wBAAwB,EAAE,EAAE;oBAC5B,yBAAyB,EAAE,EAAE;iBAC9B,CAAA;aACF;;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/structured-output/carousel.d.ts

@@ -10,13 +10,13 @@ export declare const CarouselSchema: z.ZodObject<{
             image: z.ZodString;
             button: z.ZodObject<{
                 text: z.ZodString;
-                url: z.ZodString;
+                url: z.ZodOptional<z.ZodNullable<z.ZodString>>;
             }, "strip", z.ZodTypeAny, {
                 text: string;
-                url: string;
+                url?: string | null | undefined;
             }, {
                 text: string;
-                url: string;
+                url?: string | null | undefined;
             }>;
         }, "strip", z.ZodTypeAny, {
             title: string;
@@ -24,7 +24,7 @@ export declare const CarouselSchema: z.ZodObject<{
             image: string;
             button: {
                 text: string;
-                url: string;
+                url?: string | null | undefined;
             };
         }, {
             title: string;
@@ -32,7 +32,7 @@ export declare const CarouselSchema: z.ZodObject<{
             image: string;
             button: {
                 text: string;
-                url: string;
+                url?: string | null | undefined;
             };
         }>, "many">;
     }, "strip", z.ZodTypeAny, {
@@ -42,7 +42,7 @@ export declare const CarouselSchema: z.ZodObject<{
             image: string;
             button: {
                 text: string;
-                url: string;
+                url?: string | null | undefined;
             };
         }[];
     }, {
@@ -52,7 +52,7 @@ export declare const CarouselSchema: z.ZodObject<{
             image: string;
             button: {
                 text: string;
-                url: string;
+                url?: string | null | undefined;
             };
         }[];
     }>;
@@ -65,7 +65,7 @@ export declare const CarouselSchema: z.ZodObject<{
             image: string;
             button: {
                 text: string;
-                url: string;
+                url?: string | null | undefined;
             };
         }[];
     };
@@ -78,7 +78,7 @@ export declare const CarouselSchema: z.ZodObject<{
             image: string;
             button: {
                 text: string;
-                url: string;
+                url?: string | null | undefined;
             };
         }[];
     };

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

@@ -9,7 +9,7 @@ export const CarouselSchema = z
             image: z.string(),
             button: z.object({
                 text: z.string(),
-                url: z.string(),
+                url: z.string().nullable().optional(),
             }),
         })),
     }),

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

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

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

@@ -27,25 +27,39 @@ export declare const OutputSchema: z.ZodObject<{
         type: z.ZodEnum<["textWithButtons"]>;
         content: z.ZodObject<{
             text: z.ZodString;
-            buttons: z.ZodArray<z.ZodString, "many">;
+            buttons: z.ZodArray<z.ZodObject<{
+                text: z.ZodString;
+            }, "strip", z.ZodTypeAny, {
+                text: string;
+            }, {
+                text: string;
+            }>, "many">;
         }, "strip", z.ZodTypeAny, {
             text: string;
-            buttons: string[];
+            buttons: {
+                text: string;
+            }[];
         }, {
             text: string;
-            buttons: string[];
+            buttons: {
+                text: string;
+            }[];
         }>;
     }, "strip", z.ZodTypeAny, {
         type: "textWithButtons";
         content: {
             text: string;
-            buttons: string[];
+            buttons: {
+                text: string;
+            }[];
         };
     }, {
         type: "textWithButtons";
         content: {
             text: string;
-            buttons: string[];
+            buttons: {
+                text: string;
+            }[];
         };
     }>, z.ZodObject<{
         type: z.ZodEnum<["carousel"]>;
@@ -56,13 +70,13 @@ export declare const OutputSchema: z.ZodObject<{
                 image: z.ZodString;
                 button: z.ZodObject<{
                     text: z.ZodString;
-                    url: z.ZodString;
+                    url: z.ZodOptional<z.ZodNullable<z.ZodString>>;
                 }, "strip", z.ZodTypeAny, {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 }, {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 }>;
             }, "strip", z.ZodTypeAny, {
                 title: string;
@@ -70,7 +84,7 @@ export declare const OutputSchema: z.ZodObject<{
                 image: string;
                 button: {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 };
             }, {
                 title: string;
@@ -78,7 +92,7 @@ export declare const OutputSchema: z.ZodObject<{
                 image: string;
                 button: {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 };
             }>, "many">;
         }, "strip", z.ZodTypeAny, {
@@ -88,7 +102,7 @@ export declare const OutputSchema: z.ZodObject<{
                 image: string;
                 button: {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 };
             }[];
         }, {
@@ -98,7 +112,7 @@ export declare const OutputSchema: z.ZodObject<{
                 image: string;
                 button: {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 };
             }[];
         }>;
@@ -111,7 +125,7 @@ export declare const OutputSchema: z.ZodObject<{
                 image: string;
                 button: {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 };
             }[];
         };
@@ -124,7 +138,7 @@ export declare const OutputSchema: z.ZodObject<{
                 image: string;
                 button: {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 };
             }[];
         };
@@ -145,7 +159,7 @@ export declare const OutputSchema: z.ZodObject<{
                 image: string;
                 button: {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 };
             }[];
         };
@@ -160,7 +174,9 @@ export declare const OutputSchema: z.ZodObject<{
         type: "textWithButtons";
         content: {
             text: string;
-            buttons: string[];
+            buttons: {
+                text: string;
+            }[];
         };
     })[];
 }, {
@@ -173,7 +189,7 @@ export declare const OutputSchema: z.ZodObject<{
                 image: string;
                 button: {
                     text: string;
-                    url: string;
+                    url?: string | null | undefined;
                 };
             }[];
         };
@@ -188,7 +204,9 @@ export declare const OutputSchema: z.ZodObject<{
         type: "textWithButtons";
         content: {
             text: string;
-            buttons: string[];
+            buttons: {
+                text: string;
+            }[];
         };
     })[];
 }>;

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

@@ -5,24 +5,38 @@ export declare const TextWithButtonsSchema: z.ZodObject<{
     type: z.ZodEnum<["textWithButtons"]>;
     content: z.ZodObject<{
         text: z.ZodString;
-        buttons: z.ZodArray<z.ZodString, "many">;
+        buttons: z.ZodArray<z.ZodObject<{
+            text: z.ZodString;
+        }, "strip", z.ZodTypeAny, {
+            text: string;
+        }, {
+            text: string;
+        }>, "many">;
     }, "strip", z.ZodTypeAny, {
         text: string;
-        buttons: string[];
+        buttons: {
+            text: string;
+        }[];
     }, {
         text: string;
-        buttons: string[];
+        buttons: {
+            text: string;
+        }[];
     }>;
 }, "strip", z.ZodTypeAny, {
     type: "textWithButtons";
     content: {
         text: string;
-        buttons: string[];
+        buttons: {
+            text: string;
+        }[];
     };
 }, {
     type: "textWithButtons";
     content: {
         text: string;
-        buttons: string[];
+        buttons: {
+            text: string;
+        }[];
     };
 }>;

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

@@ -4,7 +4,9 @@ export const TextWithButtonsSchema = z
     type: z.enum(['textWithButtons']),
     content: z.object({
         text: z.string(),
-        buttons: z.array(z.string()),
+        buttons: z.array(z.object({
+            text: z.string(),
+        })),
     }),
 })
     .describe('A text message with buttons to allow the user to use quick replies');

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

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

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

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

package/lib/esm/tools/index.js

@@ -1,2 +1,3 @@
+export { retrieveKnowledge } from './retrieve-knowledge';
 export const mandatoryTools = [];
 //# sourceMappingURL=index.js.map

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

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

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

@@ -0,0 +1,2 @@
+import { Context } from '../types';
+export declare const retrieveKnowledge: import("@openai/agents").FunctionTool<Context, any, any>;

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

@@ -0,0 +1,22 @@
+import { __awaiter } from "tslib";
+import { tool } from '@openai/agents';
+import { z } from 'zod';
+import { HubtypeApiClient } from '../hubtype-api-client';
+export const retrieveKnowledge = tool({
+    name: 'retrieve_knowledge',
+    description: 'Consult the knowledge base for information before answering. Use this tool to make sure the information you provide is faithful.',
+    parameters: z.object({
+        query: z.string().describe('The query to search the knowledge base for'),
+    }),
+    execute: ({ query }, runContext) => __awaiter(void 0, void 0, void 0, function* () {
+        const context = runContext === null || runContext === void 0 ? void 0 : runContext.context;
+        if (!context) {
+            throw new Error('Context is required');
+        }
+        const sources = context.sources;
+        const client = new HubtypeApiClient(context.authToken);
+        const chunks = yield client.retrieveSimilarChunks(query, sources);
+        return chunks;
+    }),
+});
+//# sourceMappingURL=retrieve-knowledge.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"retrieve-knowledge.js","sourceRoot":"","sources":["../../../src/tools/retrieve-knowledge.ts"],"names":[],"mappings":";AAAA,OAAO,EAAc,IAAI,EAAE,MAAM,gBAAgB,CAAA;AACjD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAA;AAGxD,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAoB;IACvD,IAAI,EAAE,oBAAoB;IAC1B,WAAW,EACT,kIAAkI;IACpI,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;QACnB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,4CAA4C,CAAC;KACzE,CAAC;IACF,OAAO,EAAE,CACP,EAAE,KAAK,EAAqB,EAC5B,UAAgC,EACb,EAAE;QACrB,MAAM,OAAO,GAAG,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,OAAO,CAAA;QACnC,IAAI,CAAC,OAAO,EAAE;YACZ,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAA;SACvC;QACD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAA;QAC/B,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QACtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAA;QACjE,OAAO,MAAM,CAAA;IACf,CAAC,CAAA;CACF,CAAC,CAAA"}

package/lib/esm/types.d.ts

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@botonic/plugin-ai-agents",
-  "version": "0.40.0-alpha.0",
+  "version": "0.40.0",
   "main": "./lib/cjs/index.js",
   "module": "./lib/esm/index.js",
   "description": "Use AI Agents to generate your contents",
@@ -13,7 +13,7 @@
     "lint_core": "../../node_modules/.bin/eslint_d --cache --quiet 'src/**/*.ts*'"
   },
   "dependencies": {
-    "@botonic/core": "0.40.0-alpha.0",
+    "@botonic/core": "^0.40.0",
     "@openai/agents": "^0.0.10",
     "axios": "^1.10.0",
     "openai": "^5.8.3",
@@ -45,4 +45,4 @@
     "javascript"
   ],
   "author": ""
-}
+}

package/src/agent-builder.ts

@@ -1,8 +1,8 @@
-import { Agent, InputGuardrail } from '@openai/agents'
+import { Agent, InputGuardrail, ModelSettings } from '@openai/agents'
 
 import { createInputGuardrail } from './guardrails'
 import { OutputSchema } from './structured-output'
-import { mandatoryTools } from './tools'
+import { mandatoryTools, retrieveKnowledge } from './tools'
 import { AIAgent, ContactInfo, GuardrailRule, Tool } from './types'
 
 export class AIAgentBuilder {
@@ -16,11 +16,12 @@ export class AIAgentBuilder {
     instructions: string,
     tools: Tool[],
     contactInfo: ContactInfo,
-    inputGuardrailRules: GuardrailRule[]
+    inputGuardrailRules: GuardrailRule[],
+    sourceIds: string[]
   ) {
     this.name = name
     this.instructions = this.addExtraInstructions(instructions, contactInfo)
-    this.tools = this.addHubtypeTools(tools)
+    this.tools = this.addHubtypeTools(tools, sourceIds)
     this.inputGuardrails = []
     if (inputGuardrailRules.length > 0) {
       const inputGuardrail = createInputGuardrail(inputGuardrailRules)
@@ -29,6 +30,12 @@ export class AIAgentBuilder {
   }
 
   build(): AIAgent {
+    const modelSettings: ModelSettings = {}
+
+    if (this.tools.includes(retrieveKnowledge)) {
+      modelSettings.toolChoice = retrieveKnowledge.name
+    }
+
     return new Agent({
       name: this.name,
       instructions: this.instructions,
@@ -36,6 +43,7 @@ export class AIAgentBuilder {
       outputType: OutputSchema,
       inputGuardrails: this.inputGuardrails,
       outputGuardrails: [],
+      modelSettings,
     })
   }
 
@@ -77,8 +85,11 @@ export class AIAgentBuilder {
     return `<output>\n${output}\n</output>`
   }
 
-  private addHubtypeTools(tools: Tool[]): Tool[] {
+  private addHubtypeTools(tools: Tool[], sourceIds: string[]): Tool[] {
     const hubtypeTools: Tool[] = [...mandatoryTools]
+    if (sourceIds.length > 0) {
+      hubtypeTools.push(retrieveKnowledge)
+    }
     return [...hubtypeTools, ...tools]
   }
 }

package/src/hubtype-api-client.ts

@@ -24,10 +24,31 @@ export class HubtypeApiClient {
     this.authToken = authToken
   }
 
+  async retrieveSimilarChunks(
+    query: string,
+    sources: string[]
+  ): Promise<string[]> {
+    const url = `${HUBTYPE_API_URL}/external/v1/ai/knowledge_base/similar_chunks/`
+    const headers = {
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${this.authToken}`,
+    }
+    const params = {
+      query,
+      source_ids: sources,
+      num_chunks: 5,
+    }
+    const response = await axios.post(url, params, { headers })
+    return response.data.chunks.map((chunk: any) => chunk.text)
+  }
+
   async getLocalMessages(
     maxMemoryLength: number
   ): Promise<AgenticInputMessage[]> {
-    const localBotonicState = localStorage.getItem('botonicState')
+    const localBotonicState =
+      typeof globalThis !== 'undefined' && globalThis.localStorage
+        ? globalThis.localStorage.getItem('botonicState')
+        : null
     const botonicState = JSON.parse(localBotonicState || '{}')
     const messages = botonicState.messages
     const filteredMessages = messages

package/src/index.ts

@@ -1,4 +1,4 @@
-import { BotContext, Plugin } from '@botonic/core'
+import { AiAgentArgs, BotContext, Plugin } from '@botonic/core'
 import { tool } from '@openai/agents'
 
 import { AIAgentBuilder } from './agent-builder'
@@ -8,7 +8,6 @@ import { setUpOpenAI } from './openai'
 import { AIAgentRunner } from './runner'
 import {
   AgenticInputMessage,
-  AiAgentArgs,
   Context,
   CustomTool,
   InferenceResponse,
@@ -48,7 +47,8 @@ export default class BotonicPluginAiAgents implements Plugin {
         aiAgentArgs.instructions,
         tools,
         request.session.user.contact_info || {},
-        aiAgentArgs.inputGuardrailRules || []
+        aiAgentArgs.inputGuardrailRules || [],
+        aiAgentArgs.sourceIds || []
       ).build()
 
       const messages = await this.getMessages(
@@ -58,6 +58,7 @@ export default class BotonicPluginAiAgents implements Plugin {
       )
       const context: Context = {
         authToken,
+        sources: aiAgentArgs.sourceIds || [],
       }
 
       const runner = new AIAgentRunner(agent)

package/src/structured-output/carousel.ts

@@ -14,7 +14,7 @@ export const CarouselSchema = z
           image: z.string(),
           button: z.object({
             text: z.string(),
-            url: z.string(),
+            url: z.string().nullable().optional(),
           }),
         })
       ),

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

@@ -8,7 +8,11 @@ export const TextWithButtonsSchema = z
     type: z.enum(['textWithButtons']),
     content: z.object({
       text: z.string(),
-      buttons: z.array(z.string()),
+      buttons: z.array(
+        z.object({
+          text: z.string(),
+        })
+      ),
     }),
   })
   .describe(

package/src/tools/index.ts

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