@botonic/plugin-knowledge-bases 0.23.0 → 0.23.1-alpha.1

package/lib/cjs/hubtype-knowledge-api-service.d.ts

@@ -1,12 +1,15 @@
 import { AxiosResponse } from 'axios';
 export declare class HubtypeApiService {
-    private knowledgeBaseId;
     private host;
+    private knowledgeBaseId;
     private timeout;
-    constructor(knowledgeBaseId: string, host: string, timeout?: number);
+    constructor(host: string, knowledgeBaseId: string, timeout?: number);
     inference(authToken: string, chatId: string): Promise<AxiosResponse<{
-        ai: string;
+        answer: string;
         has_knowledge: boolean;
-        sources: string[];
+        sources: {
+            knowledge_source_id: string;
+            page?: number;
+        }[];
     }>>;
 }

package/lib/cjs/hubtype-knowledge-api-service.js

@@ -2,10 +2,11 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.HubtypeApiService = void 0;
 const tslib_1 = require("tslib");
+/* eslint-disable @typescript-eslint/naming-convention */
 const axios_1 = tslib_1.__importDefault(require("axios"));
 const DEFAULT_TIMEOUT = 10000;
 class HubtypeApiService {
-    constructor(knowledgeBaseId, host, timeout) {
+    constructor(host, knowledgeBaseId, timeout) {
         this.knowledgeBaseId = knowledgeBaseId;
         this.host = host;
         this.timeout = timeout || DEFAULT_TIMEOUT;
@@ -13,12 +14,14 @@ class HubtypeApiService {
     async inference(authToken, chatId) {
         return await (0, axios_1.default)({
             method: 'POST',
-            url: `${this.host}v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
+            url: `${this.host}/external/v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
             headers: {
                 Authorization: `Bearer ${authToken}`,
                 'Content-Type': 'application/json',
             },
-            data: { chat_id: chatId },
+            data: {
+                chat_id: chatId,
+            },
             timeout: this.timeout,
         });
     }

package/lib/cjs/hubtype-knowledge-api-service.js.map

@@ -1 +1 @@
-{"version":3,"file":"hubtype-knowledge-api-service.js","sourceRoot":"","sources":["../../src/hubtype-knowledge-api-service.ts"],"names":[],"mappings":";;;;AAAA,0DAA4C;AAE5C,MAAM,eAAe,GAAG,KAAK,CAAA;AAE7B,MAAa,iBAAiB;IAK5B,YAAY,eAAuB,EAAE,IAAY,EAAE,OAAgB;QACjE,IAAI,CAAC,eAAe,GAAG,eAAe,CAAA;QACtC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,eAAe,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,SAAS,CACb,SAAiB,EACjB,MAAc;QAQd,OAAO,MAAM,IAAA,eAAK,EAAC;YACjB,MAAM,EAAE,MAAM;YACd,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,yBAAyB,IAAI,CAAC,eAAe,aAAa;YAC3E,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,SAAS,EAAE;gBACpC,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAA;IACJ,CAAC;CACF;AAhCD,8CAgCC"}
+{"version":3,"file":"hubtype-knowledge-api-service.js","sourceRoot":"","sources":["../../src/hubtype-knowledge-api-service.ts"],"names":[],"mappings":";;;;AAAA,yDAAyD;AACzD,0DAA4C;AAE5C,MAAM,eAAe,GAAG,KAAK,CAAA;AAE7B,MAAa,iBAAiB;IAK5B,YAAY,IAAY,EAAE,eAAuB,EAAE,OAAgB;QACjE,IAAI,CAAC,eAAe,GAAG,eAAe,CAAA;QACtC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,eAAe,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,SAAS,CACb,SAAiB,EACjB,MAAc;QAWd,OAAO,MAAM,IAAA,eAAK,EAAC;YACjB,MAAM,EAAE,MAAM;YACd,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,mCAAmC,IAAI,CAAC,eAAe,aAAa;YACrF,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,SAAS,EAAE;gBACpC,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE;gBACJ,OAAO,EAAE,MAAM;aAChB;YACD,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAA;IACJ,CAAC;CACF;AArCD,8CAqCC"}

package/lib/cjs/index.d.ts

@@ -1,10 +1,10 @@
 import { HubtypeSession, Plugin, PluginPreRequest } from '@botonic/core';
-import { KnowledgebaseResponse, PluginKnowledgeBaseOptions } from './types';
+import { KnowledgeBaseResponse, PluginKnowledgeBaseOptions } from './types';
 export default class BotonicPluginKnowledgeBases implements Plugin {
     private readonly apiService;
     private readonly authToken;
     constructor(options: PluginKnowledgeBaseOptions);
     pre(_request: PluginPreRequest): Promise<void>;
-    getIaResponse(session: HubtypeSession): Promise<KnowledgebaseResponse>;
+    getInference(session: HubtypeSession): Promise<KnowledgeBaseResponse>;
 }
 export { PluginKnowledgeBaseOptions } from './types';

package/lib/cjs/index.js

@@ -4,30 +4,26 @@ const hubtype_knowledge_api_service_1 = require("./hubtype-knowledge-api-service
 const isProd = process.env.NODE_ENV === 'production';
 class BotonicPluginKnowledgeBases {
     constructor(options) {
-        this.apiService = new hubtype_knowledge_api_service_1.HubtypeApiService(options.knowledgeBaseId, options.host, options.timeout);
+        this.apiService = new hubtype_knowledge_api_service_1.HubtypeApiService(options.host, options.knowledgeBaseId, options.timeout);
         this.authToken = options.authToken || '';
     }
     async pre(_request) {
         return;
     }
-    async getIaResponse(session) {
-        try {
-            const authToken = isProd ? session._access_token : this.authToken;
-            const response = await this.apiService.inference(authToken, session.user.id);
+    async getInference(session) {
+        const authToken = isProd ? session._access_token : this.authToken;
+        const response = await this.apiService.inference(authToken, session.user.id);
+        const sources = response.data.sources.map(source => {
             return {
-                ai: response.data.ai,
-                hasKnowledge: response.data.has_knowledge,
-                sources: response.data.sources,
+                knowledgeSourceId: source.knowledge_source_id,
+                page: source.page,
             };
-        }
-        catch (e) {
-            console.error(e);
-            return {
-                ai: '',
-                hasKnowledge: false,
-                sources: [],
-            };
-        }
+        });
+        return {
+            answer: response.data.answer,
+            hasKnowledge: response.data.has_knowledge,
+            sources,
+        };
     }
 }
 exports.default = BotonicPluginKnowledgeBases;

package/lib/cjs/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;AAEA,mFAAmE;AAGnE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAA;AAEpD,MAAqB,2BAA2B;IAI9C,YAAY,OAAmC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,iDAAiB,CACrC,OAAO,CAAC,eAAe,EACvB,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,OAAO,CAChB,CAAA;QACD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAA;IAC1C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,QAA0B;QAClC,OAAM;IACR,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAuB;QACzC,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;YACjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAC9C,SAAS,EACT,OAAO,CAAC,IAAI,CAAC,EAAE,CAChB,CAAA;YACD,OAAO;gBACL,EAAE,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACpB,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa;gBACzC,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,OAAO;aAC/B,CAAA;SACF;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;YAChB,OAAO;gBACL,EAAE,EAAE,EAAE;gBACN,YAAY,EAAE,KAAK;gBACnB,OAAO,EAAE,EAAE;aACZ,CAAA;SACF;IACH,CAAC;CACF;AAtCD,8CAsCC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;AAEA,mFAAmE;AAGnE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAA;AAEpD,MAAqB,2BAA2B;IAI9C,YAAY,OAAmC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,iDAAiB,CACrC,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,eAAe,EACvB,OAAO,CAAC,OAAO,CAChB,CAAA;QACD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAA;IAC1C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,QAA0B;QAClC,OAAM;IACR,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,OAAuB;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;QAEjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAE5E,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;YACjD,OAAO;gBACL,iBAAiB,EAAE,MAAM,CAAC,mBAAmB;gBAC7C,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM;YAC5B,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa;YACzC,OAAO;SACR,CAAA;IACH,CAAC;CACF;AAnCD,8CAmCC"}

package/lib/cjs/types.d.ts

@@ -4,8 +4,11 @@ export interface PluginKnowledgeBaseOptions {
     authToken?: string;
     timeout?: number;
 }
-export interface KnowledgebaseResponse {
-    ai: string;
+export interface KnowledgeBaseResponse {
+    answer: string;
     hasKnowledge: boolean;
-    sources: string[];
+    sources: {
+        knowledgeSourceId: string;
+        page?: number;
+    }[];
 }

package/lib/esm/hubtype-knowledge-api-service.d.ts

@@ -1,12 +1,15 @@
 import { AxiosResponse } from 'axios';
 export declare class HubtypeApiService {
-    private knowledgeBaseId;
     private host;
+    private knowledgeBaseId;
     private timeout;
-    constructor(knowledgeBaseId: string, host: string, timeout?: number);
+    constructor(host: string, knowledgeBaseId: string, timeout?: number);
     inference(authToken: string, chatId: string): Promise<AxiosResponse<{
-        ai: string;
+        answer: string;
         has_knowledge: boolean;
-        sources: string[];
+        sources: {
+            knowledge_source_id: string;
+            page?: number;
+        }[];
     }>>;
 }

package/lib/esm/hubtype-knowledge-api-service.js

@@ -1,7 +1,8 @@
+/* eslint-disable @typescript-eslint/naming-convention */
 import axios from 'axios';
 const DEFAULT_TIMEOUT = 10000;
 export class HubtypeApiService {
-    constructor(knowledgeBaseId, host, timeout) {
+    constructor(host, knowledgeBaseId, timeout) {
         this.knowledgeBaseId = knowledgeBaseId;
         this.host = host;
         this.timeout = timeout || DEFAULT_TIMEOUT;
@@ -9,12 +10,14 @@ export class HubtypeApiService {
     async inference(authToken, chatId) {
         return await axios({
             method: 'POST',
-            url: `${this.host}v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
+            url: `${this.host}/external/v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
             headers: {
                 Authorization: `Bearer ${authToken}`,
                 'Content-Type': 'application/json',
             },
-            data: { chat_id: chatId },
+            data: {
+                chat_id: chatId,
+            },
             timeout: this.timeout,
         });
     }

package/lib/esm/hubtype-knowledge-api-service.js.map

@@ -1 +1 @@
-{"version":3,"file":"hubtype-knowledge-api-service.js","sourceRoot":"","sources":["../../src/hubtype-knowledge-api-service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAwB,MAAM,OAAO,CAAA;AAE5C,MAAM,eAAe,GAAG,KAAK,CAAA;AAE7B,MAAM,OAAO,iBAAiB;IAK5B,YAAY,eAAuB,EAAE,IAAY,EAAE,OAAgB;QACjE,IAAI,CAAC,eAAe,GAAG,eAAe,CAAA;QACtC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,eAAe,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,SAAS,CACb,SAAiB,EACjB,MAAc;QAQd,OAAO,MAAM,KAAK,CAAC;YACjB,MAAM,EAAE,MAAM;YACd,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,yBAAyB,IAAI,CAAC,eAAe,aAAa;YAC3E,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,SAAS,EAAE;gBACpC,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAA;IACJ,CAAC;CACF"}
+{"version":3,"file":"hubtype-knowledge-api-service.js","sourceRoot":"","sources":["../../src/hubtype-knowledge-api-service.ts"],"names":[],"mappings":"AAAA,yDAAyD;AACzD,OAAO,KAAwB,MAAM,OAAO,CAAA;AAE5C,MAAM,eAAe,GAAG,KAAK,CAAA;AAE7B,MAAM,OAAO,iBAAiB;IAK5B,YAAY,IAAY,EAAE,eAAuB,EAAE,OAAgB;QACjE,IAAI,CAAC,eAAe,GAAG,eAAe,CAAA;QACtC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,eAAe,CAAA;IAC3C,CAAC;IAED,KAAK,CAAC,SAAS,CACb,SAAiB,EACjB,MAAc;QAWd,OAAO,MAAM,KAAK,CAAC;YACjB,MAAM,EAAE,MAAM;YACd,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,mCAAmC,IAAI,CAAC,eAAe,aAAa;YACrF,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,SAAS,EAAE;gBACpC,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE;gBACJ,OAAO,EAAE,MAAM;aAChB;YACD,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC,CAAA;IACJ,CAAC;CACF"}

package/lib/esm/index.d.ts

@@ -1,10 +1,10 @@
 import { HubtypeSession, Plugin, PluginPreRequest } from '@botonic/core';
-import { KnowledgebaseResponse, PluginKnowledgeBaseOptions } from './types';
+import { KnowledgeBaseResponse, PluginKnowledgeBaseOptions } from './types';
 export default class BotonicPluginKnowledgeBases implements Plugin {
     private readonly apiService;
     private readonly authToken;
     constructor(options: PluginKnowledgeBaseOptions);
     pre(_request: PluginPreRequest): Promise<void>;
-    getIaResponse(session: HubtypeSession): Promise<KnowledgebaseResponse>;
+    getInference(session: HubtypeSession): Promise<KnowledgeBaseResponse>;
 }
 export { PluginKnowledgeBaseOptions } from './types';

package/lib/esm/index.js

@@ -2,30 +2,26 @@ import { HubtypeApiService } from './hubtype-knowledge-api-service';
 const isProd = process.env.NODE_ENV === 'production';
 export default class BotonicPluginKnowledgeBases {
     constructor(options) {
-        this.apiService = new HubtypeApiService(options.knowledgeBaseId, options.host, options.timeout);
+        this.apiService = new HubtypeApiService(options.host, options.knowledgeBaseId, options.timeout);
         this.authToken = options.authToken || '';
     }
     async pre(_request) {
         return;
     }
-    async getIaResponse(session) {
-        try {
-            const authToken = isProd ? session._access_token : this.authToken;
-            const response = await this.apiService.inference(authToken, session.user.id);
+    async getInference(session) {
+        const authToken = isProd ? session._access_token : this.authToken;
+        const response = await this.apiService.inference(authToken, session.user.id);
+        const sources = response.data.sources.map(source => {
             return {
-                ai: response.data.ai,
-                hasKnowledge: response.data.has_knowledge,
-                sources: response.data.sources,
+                knowledgeSourceId: source.knowledge_source_id,
+                page: source.page,
             };
-        }
-        catch (e) {
-            console.error(e);
-            return {
-                ai: '',
-                hasKnowledge: false,
-                sources: [],
-            };
-        }
+        });
+        return {
+            answer: response.data.answer,
+            hasKnowledge: response.data.has_knowledge,
+            sources,
+        };
     }
 }
 //# 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,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAGnE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAA;AAEpD,MAAM,CAAC,OAAO,OAAO,2BAA2B;IAI9C,YAAY,OAAmC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,iBAAiB,CACrC,OAAO,CAAC,eAAe,EACvB,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,OAAO,CAChB,CAAA;QACD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAA;IAC1C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,QAA0B;QAClC,OAAM;IACR,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,OAAuB;QACzC,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;YACjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAC9C,SAAS,EACT,OAAO,CAAC,IAAI,CAAC,EAAE,CAChB,CAAA;YACD,OAAO;gBACL,EAAE,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACpB,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa;gBACzC,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,OAAO;aAC/B,CAAA;SACF;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;YAChB,OAAO;gBACL,EAAE,EAAE,EAAE;gBACN,YAAY,EAAE,KAAK;gBACnB,OAAO,EAAE,EAAE;aACZ,CAAA;SACF;IACH,CAAC;CACF"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAGnE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAA;AAEpD,MAAM,CAAC,OAAO,OAAO,2BAA2B;IAI9C,YAAY,OAAmC;QAC7C,IAAI,CAAC,UAAU,GAAG,IAAI,iBAAiB,CACrC,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,eAAe,EACvB,OAAO,CAAC,OAAO,CAChB,CAAA;QACD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAA;IAC1C,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,QAA0B;QAClC,OAAM;IACR,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,OAAuB;QACxC,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;QAEjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAE5E,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;YACjD,OAAO;gBACL,iBAAiB,EAAE,MAAM,CAAC,mBAAmB;gBAC7C,IAAI,EAAE,MAAM,CAAC,IAAI;aAClB,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,OAAO;YACL,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM;YAC5B,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa;YACzC,OAAO;SACR,CAAA;IACH,CAAC;CACF"}

package/lib/esm/types.d.ts

@@ -4,8 +4,11 @@ export interface PluginKnowledgeBaseOptions {
     authToken?: string;
     timeout?: number;
 }
-export interface KnowledgebaseResponse {
-    ai: string;
+export interface KnowledgeBaseResponse {
+    answer: string;
     hasKnowledge: boolean;
-    sources: string[];
+    sources: {
+        knowledgeSourceId: string;
+        page?: number;
+    }[];
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@botonic/plugin-knowledge-bases",
-  "version": "0.23.0",
+  "version": "0.23.1-alpha.1",
   "description": "Use a Hubtype to make the bot respond through a knowledge base.",
   "main": "./lib/cjs/index.js",
   "module": "./lib/esm/index.js",
@@ -29,7 +29,7 @@
   ],
   "dependencies": {
     "@babel/runtime": "^7.22.5",
-    "@botonic/core": "^0.23.0",
+    "@botonic/core": "0.23.1-alpha.0",
     "axios": "^1.4.0"
   },
   "devDependencies": {

package/src/hubtype-knowledge-api-service.ts

@@ -1,13 +1,14 @@
+/* eslint-disable @typescript-eslint/naming-convention */
 import axios, { AxiosResponse } from 'axios'
 
 const DEFAULT_TIMEOUT = 10000
 
 export class HubtypeApiService {
-  private knowledgeBaseId: string
   private host: string
+  private knowledgeBaseId: string
   private timeout: number
 
-  constructor(knowledgeBaseId: string, host: string, timeout?: number) {
+  constructor(host: string, knowledgeBaseId: string, timeout?: number) {
     this.knowledgeBaseId = knowledgeBaseId
     this.host = host
     this.timeout = timeout || DEFAULT_TIMEOUT
@@ -18,19 +19,24 @@ export class HubtypeApiService {
     chatId: string
   ): Promise<
     AxiosResponse<{
-      ai: string
+      answer: string
       has_knowledge: boolean
-      sources: string[]
+      sources: {
+        knowledge_source_id: string
+        page?: number
+      }[]
     }>
   > {
     return await axios({
       method: 'POST',
-      url: `${this.host}v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
+      url: `${this.host}/external/v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
       headers: {
         Authorization: `Bearer ${authToken}`,
         'Content-Type': 'application/json',
       },
-      data: { chat_id: chatId },
+      data: {
+        chat_id: chatId,
+      },
       timeout: this.timeout,
     })
   }

package/src/index.ts

@@ -1,7 +1,7 @@
 import { HubtypeSession, Plugin, PluginPreRequest } from '@botonic/core'
 
 import { HubtypeApiService } from './hubtype-knowledge-api-service'
-import { KnowledgebaseResponse, PluginKnowledgeBaseOptions } from './types'
+import { KnowledgeBaseResponse, PluginKnowledgeBaseOptions } from './types'
 
 const isProd = process.env.NODE_ENV === 'production'
 
@@ -11,8 +11,8 @@ export default class BotonicPluginKnowledgeBases implements Plugin {
 
   constructor(options: PluginKnowledgeBaseOptions) {
     this.apiService = new HubtypeApiService(
-      options.knowledgeBaseId,
       options.host,
+      options.knowledgeBaseId,
       options.timeout
     )
     this.authToken = options.authToken || ''
@@ -22,25 +22,22 @@ export default class BotonicPluginKnowledgeBases implements Plugin {
     return
   }
 
-  async getIaResponse(session: HubtypeSession): Promise<KnowledgebaseResponse> {
-    try {
-      const authToken = isProd ? session._access_token : this.authToken
-      const response = await this.apiService.inference(
-        authToken,
-        session.user.id
-      )
-      return {
-        ai: response.data.ai,
-        hasKnowledge: response.data.has_knowledge,
-        sources: response.data.sources,
-      }
-    } catch (e) {
-      console.error(e)
+  async getInference(session: HubtypeSession): Promise<KnowledgeBaseResponse> {
+    const authToken = isProd ? session._access_token : this.authToken
+
+    const response = await this.apiService.inference(authToken, session.user.id)
+
+    const sources = response.data.sources.map(source => {
       return {
-        ai: '',
-        hasKnowledge: false,
-        sources: [],
+        knowledgeSourceId: source.knowledge_source_id,
+        page: source.page,
       }
+    })
+
+    return {
+      answer: response.data.answer,
+      hasKnowledge: response.data.has_knowledge,
+      sources,
     }
   }
 }

package/src/types.ts

@@ -5,8 +5,11 @@ export interface PluginKnowledgeBaseOptions {
   timeout?: number
 }
 
-export interface KnowledgebaseResponse {
-  ai: string
+export interface KnowledgeBaseResponse {
+  answer: string
   hasKnowledge: boolean
-  sources: string[]
+  sources: {
+    knowledgeSourceId: string
+    page?: number
+  }[]
 }