extract title from prompt (#7875)

* extract title from prompt

* tests

Author: joshspicer

src/github/copilotRemoteAgent.ts

@@ -712,7 +712,7 @@ export class CopilotRemoteAgentManager extends Disposable {
 			return { error: vscode.l10n.t('Failed to configure base branch \'{0}\' does not exist on the remote repository \'{1}/{2}\'. Please create the remote branch first.', base_ref, owner, repo), state: 'error' };
 		}
 
-		const title = extractTitle(problemContext);
+		const title = extractTitle(prompt, problemContext);
 		const { problemStatement, isTruncated } = truncatePrompt(prompt, problemContext);
 
 		if (isTruncated) {

src/github/copilotRemoteAgentUtils.ts

@@ -41,14 +41,22 @@ export function truncatePrompt(prompt: string, context?: string): { problemState
 	};
 }
 
-export function extractTitle(context: string | undefined): string | undefined {
-	if (!context) {
-		return;
+export function extractTitle(prompt: string, context: string | undefined): string | undefined {
+	const fromTitle = () => {
+		if (!prompt) {
+			return;
+		}
+		if (prompt.length <= 20) {
+			return prompt;
+		}
+		return prompt.substring(0, 20) + '...';
 	}
-	const titleMatch = context.match(/TITLE: \s*(.*)/i);
+	const titleMatch = context?.match(/TITLE: \s*(.*)/i);
 	if (titleMatch && titleMatch[1]) {
 		return titleMatch[1].trim();
 	}
+	return fromTitle();
+
 }
 
 export function formatBodyPlaceholder(title: string | undefined): string {

src/test/github/copilotRemoteAgentUtils.test.ts

@@ -61,46 +61,38 @@ describe('copilotRemoteAgentUtils', () => {
 	describe('extractTitle', () => {
 		it('should extract title from context with TITLE prefix', () => {
 			const context = 'Some initial text\nTITLE: Fix authentication bug\nSome other content';
-
-			const result = extractTitle(context);
-
+			const result = extractTitle('', context);
 			assert.strictEqual(result, 'Fix authentication bug');
 		});
 
 		it('should extract title with case insensitive matching', () => {
 			const context = 'Some text\ntitle: Add new feature\nMore text';
-
-			const result = extractTitle(context);
-
+			const result = extractTitle('', context);
 			assert.strictEqual(result, 'Add new feature');
 		});
 
 		it('should extract title with extra whitespace', () => {
 			const context = 'TITLE:   Refactor code structure   \n';
-
-			const result = extractTitle(context);
-
+			const result = extractTitle('', context);
 			assert.strictEqual(result, 'Refactor code structure');
 		});
 
-		it('should return undefined when no title is found', () => {
+		it('should use prompt when no title is found', () => {
 			const context = 'Some text without any title marker\nJust regular content';
-
-			const result = extractTitle(context);
-
-			assert.strictEqual(result, undefined);
+			const result = extractTitle('Default Title', context);
+			assert.strictEqual(result, 'Default Title');
 		});
 
-		it('should return undefined when context is undefined', () => {
-			const result = extractTitle(undefined);
-
-			assert.strictEqual(result, undefined);
+		it('should use prompt when context is undefined', () => {
+			const result = extractTitle('Default Title', undefined);
+			assert.strictEqual(result, 'Default Title');
 		});
 
-		it('should return undefined when context is empty string', () => {
-			const result = extractTitle('');
+		it('should return truncated title when context is empty string', () => {
+			const title = 'TEST TEST TEST TEST TEST TEST'; // will truncate to 20 characters
+			const result = extractTitle(title, '');
 
-			assert.strictEqual(result, undefined);
+			assert.strictEqual(result, 'TEST TEST TEST TEST ...');
 		});
 	});
 });