From a5bad60d8eb9b64995dd908d5cad8a453a1bcd04 Mon Sep 17 00:00:00 2001 From: Dominik Polakovics Date: Thu, 13 Feb 2025 11:54:27 +0100 Subject: [PATCH] feat: dont include chatgpt_config.yaml in file list --- lua/chatgpt_nvim/context.lua | 4 +++- lua/chatgpt_nvim/init.lua | 20 +++++++++++--------- 2 files changed, 14 insertions(+), 10 deletions(-) diff --git a/lua/chatgpt_nvim/context.lua b/lua/chatgpt_nvim/context.lua index d0be140..edef33d 100644 --- a/lua/chatgpt_nvim/context.lua +++ b/lua/chatgpt_nvim/context.lua @@ -148,7 +148,9 @@ function M.get_project_files(directories, conf) local rel_files = {} for _, f in ipairs(all_files) do local rel = vim.fn.fnamemodify(f, ":.") - table.insert(rel_files, rel) + if not rel:match("^%.?chatgpt_config%.yaml$") then + table.insert(rel_files, rel) + end end if conf.debug then diff --git a/lua/chatgpt_nvim/init.lua b/lua/chatgpt_nvim/init.lua index fcc3dfd..81d7d63 100644 --- a/lua/chatgpt_nvim/init.lua +++ b/lua/chatgpt_nvim/init.lua @@ -122,15 +122,17 @@ local function build_prompt(user_input, dirs, conf) -- 4.1) Dynamic file inclusion via @ operator in user_input local dynamic_files = {} for file in user_input:gmatch("@([^%s]+)") do - local already_included = false - for _, existing in ipairs(initial_files) do - if existing == file then - already_included = true - break + if file ~= "chatgpt_config.yaml" and file ~= ".chatgpt_config.yaml" then + local already_included = false + for _, existing in ipairs(initial_files) do + if existing == file then + already_included = true + break + end + end + if not already_included then + table.insert(dynamic_files, file) end - end - if not already_included then - table.insert(dynamic_files, file) end end @@ -173,7 +175,7 @@ local function build_prompt(user_input, dirs, conf) table.insert(final_sections, table.concat(env_lines, "\n")) local final_prompt = table.concat(final_sections, "\n\n") - final_prompt = final_prompt:gsub("%%PROJECT_NAME%%", conf.project_name) + final_prompt = final_prompt:gsub("%chatgpt.vim%", conf.project_name) return final_prompt end