diff --git a/termui/termui.go b/termui/termui.go
index 7e4def9..1575473 100644
--- a/termui/termui.go
+++ b/termui/termui.go
@@ -43,6 +43,8 @@
 🏷️  {{.input.title}}
 {{else if eq .msg.ToolName "precommit" -}}
 🌱 git branch: sketch/{{.input.branch_name}}
+{{else if eq .msg.ToolName "knowledge_base" -}}
+📚 Knowledge: {{.input.topic}}
 {{else if eq .msg.ToolName "str_replace_editor" -}}
  ✏️  {{.input.file_path -}}
 {{else if eq .msg.ToolName "codereview" -}}