@botonic/plugin-knowledge-bases 0.26.1-alpha.0 → 0.27.0-alpha.1

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

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

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

@@ -6,22 +6,22 @@ const tslib_1 = require("tslib");
 const axios_1 = tslib_1.__importDefault(require("axios"));
 const DEFAULT_TIMEOUT = 10000;
 class HubtypeApiService {
-    constructor(host, knowledgeBaseId, timeout) {
-        this.knowledgeBaseId = knowledgeBaseId;
+    constructor(host, timeout) {
         this.host = host;
         this.timeout = timeout || DEFAULT_TIMEOUT;
     }
-    inference(authToken, chatId) {
+    inference(authToken, question, sources) {
         return tslib_1.__awaiter(this, void 0, void 0, function* () {
             return yield (0, axios_1.default)({
                 method: 'POST',
-                url: `${this.host}/external/v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
+                url: `${this.host}/external/v1/ai/knowledge_bases/inference/`,
                 headers: {
                     Authorization: `Bearer ${authToken}`,
                     'Content-Type': 'application/json',
                 },
                 data: {
-                    chat_id: chatId,
+                    question,
+                    sources,
                 },
                 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,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;IAEK,SAAS,CACb,SAAiB,EACjB,MAAc;;YAad,OAAO,MAAM,IAAA,eAAK,EAAC;gBACjB,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,mCAAmC,IAAI,CAAC,eAAe,aAAa;gBACrF,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,SAAS,EAAE;oBACpC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE;oBACJ,OAAO,EAAE,MAAM;iBAChB;gBACD,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAA;QACJ,CAAC;KAAA;CACF;AAvCD,8CAuCC"}
+{"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;IAI5B,YAAY,IAAY,EAAE,OAAgB;QACxC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,eAAe,CAAA;IAC3C,CAAC;IAEK,SAAS,CACb,SAAiB,EACjB,QAAgB,EAChB,OAAiB;;YAejB,OAAO,MAAM,IAAA,eAAK,EAAC;gBACjB,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,4CAA4C;gBAC7D,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,SAAS,EAAE;oBACpC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE;oBACJ,QAAQ;oBACR,OAAO;iBACR;gBACD,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAA;QACJ,CAAC;KAAA;CACF;AAzCD,8CAyCC"}

package/lib/cjs/index.d.ts

@@ -5,6 +5,6 @@ export default class BotonicPluginKnowledgeBases implements Plugin {
     private readonly authToken;
     constructor(options: PluginKnowledgeBaseOptions);
     pre(_request: PluginPreRequest): Promise<void>;
-    getInference(session: Session): Promise<KnowledgeBaseResponse>;
+    getInference(session: Session, userInput: string, sources: string[]): Promise<KnowledgeBaseResponse>;
 }
 export { PluginKnowledgeBaseOptions } from './types';

package/lib/cjs/index.js

@@ -5,7 +5,7 @@ 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.host, options.knowledgeBaseId, options.timeout);
+        this.apiService = new hubtype_knowledge_api_service_1.HubtypeApiService(options.host, options.timeout);
         this.authToken = options.authToken || '';
     }
     pre(_request) {
@@ -13,21 +13,24 @@ class BotonicPluginKnowledgeBases {
             return;
         });
     }
-    getInference(session) {
+    getInference(session, userInput, sources) {
         return tslib_1.__awaiter(this, void 0, void 0, function* () {
             const authToken = isProd ? session._access_token : this.authToken;
-            const response = yield this.apiService.inference(authToken, session.user.id);
-            const sources = response.data.sources.map(source => {
+            const response = yield this.apiService.inference(authToken, userInput, sources);
+            const responseSources = response.data.sources.map(source => {
                 return {
+                    knowledgeBaseId: source.knowledge_base_id,
                     knowledgeSourceId: source.knowledge_source_id,
-                    page: source.page,
+                    knowledgeChunkId: source.knowledge_chunk_id,
                 };
             });
             return {
+                inferenceId: response.data.inference_id,
                 question: response.data.question,
                 answer: response.data.answer,
-                hasKnowledge: response.data.has_knowledge && response.data.is_faithful,
-                sources,
+                hasKnowledge: response.data.has_knowledge,
+                isFaithuful: response.data.is_faithful,
+                sources: responseSources,
             };
         });
     }

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,IAAI,EACZ,OAAO,CAAC,eAAe,EACvB,OAAO,CAAC,OAAO,CAChB,CAAA;QACD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAA;IAC1C,CAAC;IAEK,GAAG,CAAC,QAA0B;;YAClC,OAAM;QACR,CAAC;KAAA;IAEK,YAAY,CAAC,OAAgB;;YACjC,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;YAEjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;YAE5E,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;gBACjD,OAAO;oBACL,iBAAiB,EAAE,MAAM,CAAC,mBAAmB;oBAC7C,IAAI,EAAE,MAAM,CAAC,IAAI;iBAClB,CAAA;YACH,CAAC,CAAC,CAAA;YAEF,OAAO;gBACL,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,QAAQ;gBAChC,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM;gBAC5B,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa,IAAI,QAAQ,CAAC,IAAI,CAAC,WAAW;gBACtE,OAAO;aACR,CAAA;QACH,CAAC;KAAA;CACF;AApCD,8CAoCC"}
+{"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,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAA;QACtE,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAA;IAC1C,CAAC;IAEK,GAAG,CAAC,QAA0B;;YAClC,OAAM;QACR,CAAC;KAAA;IAEK,YAAY,CAChB,OAAgB,EAChB,SAAiB,EACjB,OAAiB;;YAEjB,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;YAEjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAC9C,SAAS,EACT,SAAS,EACT,OAAO,CACR,CAAA;YAED,MAAM,eAAe,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;gBACzD,OAAO;oBACL,eAAe,EAAE,MAAM,CAAC,iBAAiB;oBACzC,iBAAiB,EAAE,MAAM,CAAC,mBAAmB;oBAC7C,gBAAgB,EAAE,MAAM,CAAC,kBAAkB;iBAC5C,CAAA;YACH,CAAC,CAAC,CAAA;YAEF,OAAO;gBACL,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY;gBACvC,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,QAAQ;gBAChC,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM;gBAC5B,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa;gBACzC,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,WAAW;gBACtC,OAAO,EAAE,eAAe;aACzB,CAAA;QACH,CAAC;KAAA;CACF;AA3CD,8CA2CC"}

package/lib/cjs/types.d.ts

@@ -1,15 +1,17 @@
 export interface PluginKnowledgeBaseOptions {
-    knowledgeBaseId: string;
     host: string;
     authToken?: string;
     timeout?: number;
 }
 export interface KnowledgeBaseResponse {
+    inferenceId: string;
     question: string;
     answer: string;
     hasKnowledge: boolean;
+    isFaithuful: boolean;
     sources: {
+        knowledgeBaseId: string;
         knowledgeSourceId: string;
-        page?: number;
+        knowledgeChunkId: string;
     }[];
 }

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

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

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

@@ -3,22 +3,22 @@ import { __awaiter } from "tslib";
 import axios from 'axios';
 const DEFAULT_TIMEOUT = 10000;
 export class HubtypeApiService {
-    constructor(host, knowledgeBaseId, timeout) {
-        this.knowledgeBaseId = knowledgeBaseId;
+    constructor(host, timeout) {
         this.host = host;
         this.timeout = timeout || DEFAULT_TIMEOUT;
     }
-    inference(authToken, chatId) {
+    inference(authToken, question, sources) {
         return __awaiter(this, void 0, void 0, function* () {
             return yield axios({
                 method: 'POST',
-                url: `${this.host}/external/v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
+                url: `${this.host}/external/v1/ai/knowledge_bases/inference/`,
                 headers: {
                     Authorization: `Bearer ${authToken}`,
                     'Content-Type': 'application/json',
                 },
                 data: {
-                    chat_id: chatId,
+                    question,
+                    sources,
                 },
                 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,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;IAEK,SAAS,CACb,SAAiB,EACjB,MAAc;;YAad,OAAO,MAAM,KAAK,CAAC;gBACjB,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,mCAAmC,IAAI,CAAC,eAAe,aAAa;gBACrF,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,SAAS,EAAE;oBACpC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE;oBACJ,OAAO,EAAE,MAAM;iBAChB;gBACD,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAA;QACJ,CAAC;KAAA;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;IAI5B,YAAY,IAAY,EAAE,OAAgB;QACxC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;QAChB,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,eAAe,CAAA;IAC3C,CAAC;IAEK,SAAS,CACb,SAAiB,EACjB,QAAgB,EAChB,OAAiB;;YAejB,OAAO,MAAM,KAAK,CAAC;gBACjB,MAAM,EAAE,MAAM;gBACd,GAAG,EAAE,GAAG,IAAI,CAAC,IAAI,4CAA4C;gBAC7D,OAAO,EAAE;oBACP,aAAa,EAAE,UAAU,SAAS,EAAE;oBACpC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE;oBACJ,QAAQ;oBACR,OAAO;iBACR;gBACD,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAA;QACJ,CAAC;KAAA;CACF"}

package/lib/esm/index.d.ts

@@ -5,6 +5,6 @@ export default class BotonicPluginKnowledgeBases implements Plugin {
     private readonly authToken;
     constructor(options: PluginKnowledgeBaseOptions);
     pre(_request: PluginPreRequest): Promise<void>;
-    getInference(session: Session): Promise<KnowledgeBaseResponse>;
+    getInference(session: Session, userInput: string, sources: string[]): Promise<KnowledgeBaseResponse>;
 }
 export { PluginKnowledgeBaseOptions } from './types';

package/lib/esm/index.js

@@ -3,7 +3,7 @@ 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.host, options.knowledgeBaseId, options.timeout);
+        this.apiService = new HubtypeApiService(options.host, options.timeout);
         this.authToken = options.authToken || '';
     }
     pre(_request) {
@@ -11,21 +11,24 @@ export default class BotonicPluginKnowledgeBases {
             return;
         });
     }
-    getInference(session) {
+    getInference(session, userInput, sources) {
         return __awaiter(this, void 0, void 0, function* () {
             const authToken = isProd ? session._access_token : this.authToken;
-            const response = yield this.apiService.inference(authToken, session.user.id);
-            const sources = response.data.sources.map(source => {
+            const response = yield this.apiService.inference(authToken, userInput, sources);
+            const responseSources = response.data.sources.map(source => {
                 return {
+                    knowledgeBaseId: source.knowledge_base_id,
                     knowledgeSourceId: source.knowledge_source_id,
-                    page: source.page,
+                    knowledgeChunkId: source.knowledge_chunk_id,
                 };
             });
             return {
+                inferenceId: response.data.inference_id,
                 question: response.data.question,
                 answer: response.data.answer,
-                hasKnowledge: response.data.has_knowledge && response.data.is_faithful,
-                sources,
+                hasKnowledge: response.data.has_knowledge,
+                isFaithuful: response.data.is_faithful,
+                sources: responseSources,
             };
         });
     }

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,IAAI,EACZ,OAAO,CAAC,eAAe,EACvB,OAAO,CAAC,OAAO,CAChB,CAAA;QACD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAA;IAC1C,CAAC;IAEK,GAAG,CAAC,QAA0B;;YAClC,OAAM;QACR,CAAC;KAAA;IAEK,YAAY,CAAC,OAAgB;;YACjC,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;YAEjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;YAE5E,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;gBACjD,OAAO;oBACL,iBAAiB,EAAE,MAAM,CAAC,mBAAmB;oBAC7C,IAAI,EAAE,MAAM,CAAC,IAAI;iBAClB,CAAA;YACH,CAAC,CAAC,CAAA;YAEF,OAAO;gBACL,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,QAAQ;gBAChC,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM;gBAC5B,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa,IAAI,QAAQ,CAAC,IAAI,CAAC,WAAW;gBACtE,OAAO;aACR,CAAA;QACH,CAAC;KAAA;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,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAA;QACtE,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,EAAE,CAAA;IAC1C,CAAC;IAEK,GAAG,CAAC,QAA0B;;YAClC,OAAM;QACR,CAAC;KAAA;IAEK,YAAY,CAChB,OAAgB,EAChB,SAAiB,EACjB,OAAiB;;YAEjB,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAA;YAEjE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAC9C,SAAS,EACT,SAAS,EACT,OAAO,CACR,CAAA;YAED,MAAM,eAAe,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;gBACzD,OAAO;oBACL,eAAe,EAAE,MAAM,CAAC,iBAAiB;oBACzC,iBAAiB,EAAE,MAAM,CAAC,mBAAmB;oBAC7C,gBAAgB,EAAE,MAAM,CAAC,kBAAkB;iBAC5C,CAAA;YACH,CAAC,CAAC,CAAA;YAEF,OAAO;gBACL,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY;gBACvC,QAAQ,EAAE,QAAQ,CAAC,IAAI,CAAC,QAAQ;gBAChC,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM;gBAC5B,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,aAAa;gBACzC,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,WAAW;gBACtC,OAAO,EAAE,eAAe;aACzB,CAAA;QACH,CAAC;KAAA;CACF"}

package/lib/esm/types.d.ts

@@ -1,15 +1,17 @@
 export interface PluginKnowledgeBaseOptions {
-    knowledgeBaseId: string;
     host: string;
     authToken?: string;
     timeout?: number;
 }
 export interface KnowledgeBaseResponse {
+    inferenceId: string;
     question: string;
     answer: string;
     hasKnowledge: boolean;
+    isFaithuful: boolean;
     sources: {
+        knowledgeBaseId: string;
         knowledgeSourceId: string;
-        page?: number;
+        knowledgeChunkId: string;
     }[];
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@botonic/plugin-knowledge-bases",
-  "version": "0.26.1-alpha.0",
+  "version": "0.27.0-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",
@@ -15,7 +15,7 @@
   },
   "dependencies": {
     "@babel/runtime": "^7.23.9",
-    "@botonic/core": "0.26.1-alpha.0",
+    "@botonic/core": "0.27.0-alpha.1",
     "axios": "^1.6.8"
   },
   "devDependencies": {

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

@@ -5,39 +5,41 @@ const DEFAULT_TIMEOUT = 10000
 
 export class HubtypeApiService {
   private host: string
-  private knowledgeBaseId: string
   private timeout: number
 
-  constructor(host: string, knowledgeBaseId: string, timeout?: number) {
-    this.knowledgeBaseId = knowledgeBaseId
+  constructor(host: string, timeout?: number) {
     this.host = host
     this.timeout = timeout || DEFAULT_TIMEOUT
   }
 
   async inference(
     authToken: string,
-    chatId: string
+    question: string,
+    sources: string[]
   ): Promise<
     AxiosResponse<{
+      inference_id: string
       question: string
       answer: string
       has_knowledge: boolean
       is_faithful: boolean
       sources: {
+        knowledge_base_id: string
         knowledge_source_id: string
-        page?: number
+        knowledge_chunk_id: string
       }[]
     }>
   > {
     return await axios({
       method: 'POST',
-      url: `${this.host}/external/v1/ai/knowledge_bases/${this.knowledgeBaseId}/inference/`,
+      url: `${this.host}/external/v1/ai/knowledge_bases/inference/`,
       headers: {
         Authorization: `Bearer ${authToken}`,
         'Content-Type': 'application/json',
       },
       data: {
-        chat_id: chatId,
+        question,
+        sources,
       },
       timeout: this.timeout,
     })

package/src/index.ts

@@ -10,11 +10,7 @@ export default class BotonicPluginKnowledgeBases implements Plugin {
   private readonly authToken: string
 
   constructor(options: PluginKnowledgeBaseOptions) {
-    this.apiService = new HubtypeApiService(
-      options.host,
-      options.knowledgeBaseId,
-      options.timeout
-    )
+    this.apiService = new HubtypeApiService(options.host, options.timeout)
     this.authToken = options.authToken || ''
   }
 
@@ -22,23 +18,34 @@ export default class BotonicPluginKnowledgeBases implements Plugin {
     return
   }
 
-  async getInference(session: Session): Promise<KnowledgeBaseResponse> {
+  async getInference(
+    session: Session,
+    userInput: string,
+    sources: string[]
+  ): Promise<KnowledgeBaseResponse> {
     const authToken = isProd ? session._access_token : this.authToken
 
-    const response = await this.apiService.inference(authToken, session.user.id)
+    const response = await this.apiService.inference(
+      authToken,
+      userInput,
+      sources
+    )
 
-    const sources = response.data.sources.map(source => {
+    const responseSources = response.data.sources.map(source => {
       return {
+        knowledgeBaseId: source.knowledge_base_id,
         knowledgeSourceId: source.knowledge_source_id,
-        page: source.page,
+        knowledgeChunkId: source.knowledge_chunk_id,
       }
     })
 
     return {
+      inferenceId: response.data.inference_id,
       question: response.data.question,
       answer: response.data.answer,
-      hasKnowledge: response.data.has_knowledge && response.data.is_faithful,
-      sources,
+      hasKnowledge: response.data.has_knowledge,
+      isFaithuful: response.data.is_faithful,
+      sources: responseSources,
     }
   }
 }

package/src/types.ts

@@ -1,16 +1,18 @@
 export interface PluginKnowledgeBaseOptions {
-  knowledgeBaseId: string
   host: string
   authToken?: string
   timeout?: number
 }
 
 export interface KnowledgeBaseResponse {
+  inferenceId: string
   question: string
   answer: string
   hasKnowledge: boolean
+  isFaithuful: boolean
   sources: {
+    knowledgeBaseId: string
     knowledgeSourceId: string
-    page?: number
+    knowledgeChunkId: string
   }[]
 }