Chore: Refactor error investigation prompts to improve context handling (#34)

* Refactor error investigation prompts to improve context handling

* bump version

Author: tninja

HISTORY.org

@@ -3,6 +3,7 @@
 
 ** Main branch change
 
+- Chore: Refactor error investigation prompts to improve context handling
 - Chore: Add clipboard context option to ai-code-send-command function
 - Chore: add compilation buffer content as error context
 - Chore: Refactor AI shell command prompt to include marked files in dired mode

ai-code-discussion.el

@@ -130,15 +130,24 @@ Argument ARG is the prefix argument."
                                  (buffer-substring-no-properties (point-min) (point-max)))))
          (region-text (when (region-active-p)
                         (buffer-substring-no-properties (region-beginning) (region-end))))
-         (context-text (or compilation-content region-text))
+         (buffer-file buffer-file-name)
+         (full-buffer-context (when (and (not buffer-file) (not region-text))
+                                (buffer-substring-no-properties (point-min) (point-max))))
          (function-name (which-function))
-         (files-context-string (ai-code--get-context-files-string)))
+         (files-context-string (ai-code--get-context-files-string))
+         (context-section
+          (if full-buffer-context
+              (concat "\n\nContext:\n" full-buffer-context)
+            (concat
+             (when compilation-content
+               (concat "\n\nCompilation output:\n" compilation-content))
+             (when (and region-text (not compilation-content))
+               (concat "\n\nSelected code:\n" region-text)))))
+         (default-question "How to fix the error in this code? Please analyze the error, explain the root cause, and provide the corrected code to resolve the issue: "))
     (if (or arg (not (derived-mode-p 'prog-mode)))
-        (let* ((initial-prompt (if (and context-text (not (derived-mode-p 'prog-mode)))
-                                   (concat "Investigate the exception and fix the code:\n\n" context-text)
-                                 (or context-text "")))
-               (prompt (ai-code-read-string "Investigate exception (no context): " initial-prompt))
+        (let* ((prompt (ai-code-read-string "Investigate exception (no context): " default-question))
                (final-prompt (concat prompt
+                                     context-section
                                      (when function-name (format "\nFunction: %s" function-name))
                                      files-context-string)))
           (ai-code--insert-prompt final-prompt))
@@ -151,15 +160,12 @@ Argument ARG is the prefix argument."
                     (format "Investigate exception in function %s: " function-name)
                   "Investigate selected exception: "))
                (function-name
-                (format "Investigate exceptions in function %s: " function-name))
+                    (format "Investigate exceptions in function %s: " function-name))
                (t "Investigate exceptions in code: ")))
-             (initial-prompt (ai-code-read-string prompt-label
-                                                  (or context-text "How to fix the error in this code? Please analyze the error, explain the root cause, and provide the corrected code to resolve the issue: ")))
+             (initial-prompt (ai-code-read-string prompt-label default-question))
              (final-prompt
               (concat initial-prompt
-                      (when compilation-content (concat "\n\nCompilation output:\n" compilation-content))
-                      (when (and region-text (not compilation-content))
-                        (concat "\n\nSelected code:\n" region-text))
+                      context-section
                       (when function-name (format "\nFunction: %s" function-name))
                       files-context-string
                       (concat "\n\nNote: Please focus on how to fix the error. Your response should include:\n"

ai-code-interface.el

@@ -1,7 +1,7 @@
 ;;; ai-code-interface.el --- AI code interface for editing AI prompt files -*- lexical-binding: t; -*-
 
 ;; Author: Kang Tu <tninja@gmail.com>
-;; Version: 0.48
+;; Version: 0.49
 ;; Package-Requires: ((emacs "26.1") (transient "0.8.0") (magit "2.1.0"))
 
 ;; SPDX-License-Identifier: Apache-2.0