feat: adding chunking feature, adding preview and partial accepting

lua/chatgpt_nvim/config.lua

@@ -28,7 +28,7 @@ local prompt_blocks = {
       project_name: example_project
       files:
       - path: "relative/path/to/file"
-    3. If more information or context is needed, ask the user (outside of the YAML) to provide that. For example:
+    3. If more information or context is needed, ask the user (outside of the YAML) to provide that.
     4. When all necessary information is gathered, provide the final YAML with the
       project's name and a list of files to be created or modified.
       Also explain the changes you made below the yaml.
@@ -46,6 +46,14 @@ local prompt_blocks = {
     As you generate code or provide guidance, you must consider the security impact of every decision.
     You will write and review code with a focus on minimizing vulnerabilities and following best security practices,
     such as validating all user inputs, avoiding unsafe libraries or functions, and following secure coding standards.
+  ]],
+  ["workflow-prompt"] = [[
+    You are a coding assistant focusing on making the Neovim ChatGPT workflow straightforward and user-friendly.
+    Provide a concise set of steps or guidance, reminding the user:
+    - How to list needed files for further context
+    - How to request additional information outside of the YAML
+    - How to finalize changes with a YAML response containing project_name and files
+    Always ensure that prompts and explanations remain clear and minimal, reducing user errors.
   ]]
 }
 
@@ -88,7 +96,14 @@ function M.load()
     token_limit = 128000,
     project_name = "",
     debug = false,
-    initial_files = {}
+    initial_files = {},
+
+    -- Additional new config flags
+    preview_changes = false,
+    interactive_file_selection = false,
+    partial_acceptance = false,
+    improved_debug = false,
+    enable_chunking = false
   }
 
   if fd then
@@ -119,10 +134,25 @@ function M.load()
         if type(result.initial_files) == "table" then
           config.initial_files = result.initial_files
         end
+        if type(result.preview_changes) == "boolean" then
+          config.preview_changes = result.preview_changes
+        end
+        if type(result.interactive_file_selection) == "boolean" then
+          config.interactive_file_selection = result.interactive_file_selection
+        end
+        if type(result.partial_acceptance) == "boolean" then
+          config.partial_acceptance = result.partial_acceptance
+        end
+        if type(result.improved_debug) == "boolean" then
+          config.improved_debug = result.improved_debug
+        end
+        if type(result.enable_chunking) == "boolean" then
+          config.enable_chunking = result.enable_chunking
+        end
       end
     end
   else
-    config.initial_prompt = "You are a coding assistant who receives a project's context and user instructions. You will guide the user through a workflow:\n1. First, ask the user which files are needed from the project to understand and perform the coding task.\n2. If more information or context is needed, ask for it outside of the YAML.\n3. Once all information is obtained, return the final YAML with the project_name and the files to be created/modified or deleted.\n\nThe final YAML must have:\nproject_name: <project_name>\nfiles:\n  - path: \"relative/path/to/file\"\n    content: |\n      <file content>\n  - path: \"relative/path/to/other_file\"\n    delete: true\n\nIf more context is needed at any step before providing the final YAML, request it outside of the YAML."
+    config.initial_prompt = "You are a coding assistant who receives a project's context and user instructions..."
   end
 
   -- Merge the default prompt blocks with the config's initial prompt
@@ -135,7 +165,6 @@ function M.load()
     end
     if #merged_prompt > 0 then
       local combined_blocks = table.concat(merged_prompt, "\n\n")
-      -- We now *append* or combine the existing initial_prompt with the default blocks
       if config.initial_prompt ~= "" then
         config.initial_prompt = config.initial_prompt .. "\n\n" .. combined_blocks
       else