refactor(plugin-knowledge-bases): change api endpoint and response

hubtype · Aug 1, 2024 · 3e0fc7e · 3e0fc7e
1 parent 14d758a
commit 3e0fc7e
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 21 deletions.
diff --git a/packages/botonic-plugin-knowledge-bases/package.json b/packages/botonic-plugin-knowledge-bases/package.json
@@ -15,7 +15,7 @@
   },
   "dependencies": {
     "@babel/runtime": "^7.23.9",
-    "@botonic/core": "^0.28.1",
+    "@botonic/core": "^0.28.2",
     "axios": "^1.7.2"
   },
   "devDependencies": {

diff --git a/packages/botonic-plugin-knowledge-bases/src/hubtype-knowledge-api-service.ts b/packages/botonic-plugin-knowledge-bases/src/hubtype-knowledge-api-service.ts
@@ -23,16 +23,12 @@ export class HubtypeApiService {
       answer: string
       has_knowledge: boolean
       is_faithful: boolean
-      sources: {
-        knowledge_base_id: string
-        knowledge_source_id: string
-        knowledge_chunk_id: string
-      }[]
+      chunk_ids: string[]
     }>
   > {
     return await axios({
       method: 'POST',
-      url: `${this.host}/external/v1/ai/knowledge_bases/inference/`,
+      url: `${this.host}/external/v1/ai/knowledge_base/inference/`,
       headers: {
         Authorization: `Bearer ${authToken}`,
         'Content-Type': 'application/json',

diff --git a/packages/botonic-plugin-knowledge-bases/src/index.ts b/packages/botonic-plugin-knowledge-bases/src/index.ts
@@ -31,21 +31,13 @@ export default class BotonicPluginKnowledgeBases implements Plugin {
       sources
     )
 
-    const responseSources = response.data.sources.map(source => {
-      return {
-        knowledgeBaseId: source.knowledge_base_id,
-        knowledgeSourceId: source.knowledge_source_id,
-        knowledgeChunkId: source.knowledge_chunk_id,
-      }
-    })
-
     return {
       inferenceId: response.data.inference_id,
       question: response.data.question,
       answer: response.data.answer,
       hasKnowledge: response.data.has_knowledge,
       isFaithuful: response.data.is_faithful,
-      sources: responseSources,
+      chunkIds: response.data.chunk_ids,
     }
   }
 }

diff --git a/packages/botonic-plugin-knowledge-bases/src/types.ts b/packages/botonic-plugin-knowledge-bases/src/types.ts
@@ -10,9 +10,5 @@ export interface KnowledgeBaseResponse {
   answer: string
   hasKnowledge: boolean
   isFaithuful: boolean
-  sources: {
-    knowledgeBaseId: string
-    knowledgeSourceId: string
-    knowledgeChunkId: string
-  }[]
+  chunkIds: string[]
 }