diff --git a/packages/botonic-plugin-flow-builder/src/action/knowledge-bases.ts b/packages/botonic-plugin-flow-builder/src/action/knowledge-bases.ts
index 73395c692..ab84519bf 100644
--- a/packages/botonic-plugin-flow-builder/src/action/knowledge-bases.ts
+++ b/packages/botonic-plugin-flow-builder/src/action/knowledge-bases.ts
@@ -71,16 +71,17 @@ async function getContentsWithKnowledgeResponse(
     return undefined
   }
 
-  return updateContentsWithAnswer(contents, knowledgeBaseResponse.answer)
+  return updateContentsWithResponse(contents, knowledgeBaseResponse)
 }
 
-function updateContentsWithAnswer(
+function updateContentsWithResponse(
   contents: FlowContent[],
-  answer: string
+  response: KnowledgeBaseResponse
 ): FlowContent[] {
   return contents.map(content => {
     if (content instanceof FlowKnowledgeBase) {
-      content.text = answer
+      content.text = response.answer
+      content.inferenceId = response.inferenceId
     }
 
     return content
diff --git a/packages/botonic-plugin-flow-builder/src/content-fields/flow-knowledge-base.tsx b/packages/botonic-plugin-flow-builder/src/content-fields/flow-knowledge-base.tsx
index 4ab2ff475..3ae13a16a 100644
--- a/packages/botonic-plugin-flow-builder/src/content-fields/flow-knowledge-base.tsx
+++ b/packages/botonic-plugin-flow-builder/src/content-fields/flow-knowledge-base.tsx
@@ -5,9 +5,10 @@ import { ContentFieldsBase } from './content-fields-base'
 import { HtKnowledgeBaseNode } from './hubtype-fields'
 
 export class FlowKnowledgeBase extends ContentFieldsBase {
-  public code = ''
+  public code: string = ''
   public sources: string[] = []
-  public text = ''
+  public text: string = ''
+  public inferenceId?: string
 
   static fromHubtypeCMS(component: HtKnowledgeBaseNode): FlowKnowledgeBase {
     const newKnowledgeBase = new FlowKnowledgeBase(component.id)
@@ -18,6 +19,10 @@ export class FlowKnowledgeBase extends ContentFieldsBase {
   }
 
   toBotonic(id: string): JSX.Element {
-    return <Text key={id}>{this.text}</Text>
+    return (
+      <Text key={id} withfeedback={true} inferenceid={this.inferenceId}>
+        {this.text}
+      </Text>
+    )
   }
 }