feat: add maximum length flag (#91)

src/cli.ts

@@ -25,6 +25,12 @@ cli(
 				alias: 'g',
 				default: 1,
 			},
+			length: {
+				type: Number,
+				description: 'Maximum length of the generated message',
+				alias: 'l',
+				default: 50,
+			},
 		},
 
 		commands: [
@@ -45,6 +51,7 @@ cli(
 			aicommits(
 				argv.flags.generate,
 				rawArgv,
+				argv.flags.length,
 			);
 		}
 	},

src/commands/aicommits.ts

@@ -16,6 +16,7 @@ import { generateCommitMessage } from '../utils/openai.js';
 export default async (
 	generate: number,
 	rawArgv: string[],
+	maximumLength: number,
 ) => (async () => {
 	intro(bgCyan(black(' aicommits ')));
 
@@ -45,6 +46,7 @@ export default async (
 		OPENAI_KEY,
 		staged.diff,
 		generate,
+		maximumLength,
 	);
 	s.stop('Changes analyzed');
 

src/utils/openai.ts

@@ -4,14 +4,15 @@ const sanitizeMessage = (message: string) => message.trim().replace(/[\n\r]/g, '
 
 const deduplicateMessages = (array: string[]) => Array.from(new Set(array));
 
-const promptTemplate = 'Write an insightful but concise Git commit message in a complete sentence in present tense for the following diff without prefacing it with anything:';
+const promptTemplate = 'Write an insightful but concise Git commit message in a complete sentence in present tense for the following diff without prefacing it with anything, making sure it is exactly or less than [count] characters long:';
 
 export const generateCommitMessage = async (
 	apiKey: string,
 	diff: string,
 	completions: number,
+	maximumLength: number,
 ) => {
-	const prompt = `${promptTemplate}\n${diff}`;
+	const prompt = `${promptTemplate.replace('[count]', maximumLength.toString())}\n${diff}`;
 
 	// Accounting for GPT-3's input req of 4k tokens (approx 8k chars)
 	if (prompt.length > 8000) {