feat: add contents of the message box to the prompt to customize gene…

…rated content

src/generate-commit-msg.ts

@@ -13,8 +13,8 @@ import { ProgressHandler } from './utils';
  * @param {string} diff - The diff string representing changes to be committed.
  * @returns {Promise<Array<{ role: string, content: string }>>} - A promise that resolves to an array of messages for the chat completion.
  */
-const generateCommitMessageChatCompletionPrompt = async (diff: string) => {
-  const INIT_MESSAGES_PROMPT = await getMainCommitPrompt();
+const generateCommitMessageChatCompletionPrompt = async (diff: string, extras: string) => {
+  const INIT_MESSAGES_PROMPT = await getMainCommitPrompt(extras);
   const chatContextAsCompletionRequest = [...INIT_MESSAGES_PROMPT];
 
   chatContextAsCompletionRequest.push({
@@ -85,7 +85,8 @@ export async function generateCommitMsg(arg) {
         }
 
         progress.report({ message: 'Analyzing changes...' });
-        const messages = await generateCommitMessageChatCompletionPrompt(diff);
+        const extras = scmInputBox.value;
+        const messages = await generateCommitMessageChatCompletionPrompt(diff, extras);
 
         progress.report({ message: 'Generating commit message...' });
         try {

src/prompts.ts

@@ -6,7 +6,7 @@ import { ConfigKeys, ConfigurationManager } from './config';
  * @param {string} language - The language to be used in the prompt.
  * @returns {Object} - The main prompt object containing role and content.
  */
-const INIT_MAIN_PROMPT = (language: string) => ({
+const INIT_MAIN_PROMPT = (language: string, extras: string) => ({
   role: 'system',
   content:
     ConfigurationManager.getInstance().getConfig<string>(ConfigKeys.SYSTEM_PROMPT) ||
@@ -78,6 +78,7 @@ You will act as a git commit message generator. When receiving a git diff, you w
 3. NO additional text or explanations
 4. NO questions or comments
 5. NO formatting instructions or metadata
+${extras ? `6. VERY IMPORTANT! ${extras}` : ''}
 
 ## Examples
 
@@ -100,17 +101,18 @@ OUTPUT:
 - rename port variable to uppercase (PORT) to follow constant naming convention
 - add environment variable port support for flexible deployment
 
-Remember: All output MUST be in ${language} language. You are to act as a pure commit message generator. Your response should contain NOTHING but the commit message itself.`
+Remember: All output MUST be in ${language} language. You are to act as a pure commit message generator. Your response should contain NOTHING but the commit message itself.
+`
 });
 
 /**
  * Retrieves the main commit prompt.
  *
  * @returns {Promise<Array<Object>>} - A promise that resolves to an array of prompts.
  */
-export const getMainCommitPrompt = async () => {
+export const getMainCommitPrompt = async (extras: string) => {
   const language = ConfigurationManager.getInstance().getConfig<string>(
     ConfigKeys.AI_COMMIT_LANGUAGE
   );
-  return [INIT_MAIN_PROMPT(language)];
+  return [INIT_MAIN_PROMPT(language, extras)];
 };