diff --git a/nvim/lua/plugins/nvim-cmp.lua b/nvim/lua/plugins/nvim-cmp.lua
index 053c529..6919760 100644
--- a/nvim/lua/plugins/nvim-cmp.lua
+++ b/nvim/lua/plugins/nvim-cmp.lua
@@ -50,6 +50,15 @@ return {
         provider_options = {
           base_url = os.getenv 'LLAMA_API_BASE' or 'http://localhost:8080/ollama',
           api_key = os.getenv 'LLAMA_API_KEY',
+          prompt = function(lines_before, lines_after)
+            return lines_before
+            -- return '<PRE> ' .. lines_before .. ' <SUF>' .. lines_after .. ' <MID>'
+            -- return '<fim_prefix>' .. lines_before .. '<fim_suffix>' .. lines_after .. '<fim_middle>'
+            -- return '<|fim_prefix|>' .. lines_before .. '<|fim_suffix|>' .. lines_after .. '<|fim_middle|>'
+          end,
+          suffix = function(lines_after)
+            return lines_after
+          end,
         },
         max_lines = 500,
         run_on_every_keystroke = false,