diff --git a/README.md b/README.md
index 5f40e28..c505376 100644
--- a/README.md
+++ b/README.md
@@ -722,6 +722,8 @@ return {
 | AvanteConflictIncomingLabel | Incoming conflict label highlight             | Default to shade of `AvanteConflictIncoming` |
 | AvantePopupHint             | Usage hints in popup menus                    |                                              |
 | AvanteInlineHint            | The end-of-line hint displayed in visual mode |                                              |
+| AvantePromptInput           | The body highlight of the prompt input        |                                              |
+| AvantePromptInputBorder     | The border highlight of the prompt input      | Default to `NormalFloat`                     |
 
 See [highlights.lua](./lua/avante/highlights.lua) for more information
 
diff --git a/lua/avante/highlights.lua b/lua/avante/highlights.lua
index 90274dd..f0ccdbd 100644
--- a/lua/avante/highlights.lua
+++ b/lua/avante/highlights.lua
@@ -25,6 +25,8 @@ local Highlights = {
   BUTTON_PRIMARY_HOVER = { name = "AvanteButtonPrimaryHover", fg = "#1e222a", bg = "#56b6c2" },
   BUTTON_DANGER = { name = "AvanteButtonDanger", fg = "#1e222a", bg = "#ABB2BF" },
   BUTTON_DANGER_HOVER = { name = "AvanteButtonDangerHover", fg = "#1e222a", bg = "#e06c75" },
+  AVANTE_PROMPT_INPUT = { name = "AvantePromptInput" },
+  AVANTE_PROMPT_INPUT_BORDER = { name = "AvantePromptInputBorder", link = "NormalFloat" },
   AVANTE_SIDEBAR_WIN_SEPARATOR = {
     name = "AvanteSidebarWinSeparator",
     fg_link_bg = "NormalFloat",
diff --git a/lua/avante/ui/prompt_input.lua b/lua/avante/ui/prompt_input.lua
index 3b38cac..2ef5892 100644
--- a/lua/avante/ui/prompt_input.lua
+++ b/lua/avante/ui/prompt_input.lua
@@ -117,7 +117,11 @@ function PromptInput:open()
 
   api.nvim_set_option_value("wrap", false, { win = winid })
   api.nvim_set_option_value("winblend", 5, { win = winid })
-  api.nvim_set_option_value("winhighlight", "FloatBorder:NormalFloat", { win = winid })
+  api.nvim_set_option_value(
+    "winhighlight",
+    "FloatBorder:AvantePromptInputBorder,Normal:AvantePromptInput",
+    { win = winid }
+  )
   api.nvim_set_option_value("cursorline", true, { win = winid })
   api.nvim_set_option_value("modifiable", true, { buf = bufnr })