fix(amazonq): improve system prompts for rules

Author: avi-alpert

packages/amazonq/.changes/next-release/Bug Fix-62081c53-09b4-46b0-80b0-6eb853ffabd6.json

@@ -0,0 +1,4 @@
+{
+	"type": "Bug Fix",
+	"description": "Amazon Q chat: Improve responses for saved prompts and workspace rules"
+}

packages/core/src/codewhispererChat/controllers/chat/controller.ts

@@ -980,9 +980,15 @@ export class ChatController {
         if (Array.isArray(prompts) && prompts.length > 0) {
             triggerPayload.additionalContextLengths = this.telemetryHelper.getContextLengths(prompts)
             for (const prompt of prompts.slice(0, 20)) {
+                let description = prompt.description
+                // Add system prompt for user prompts and workspace rules
+                const contextType = this.telemetryHelper.getContextType(prompt)
+                if (contextType === 'rule' || contextType === 'prompt') {
+                    description = `You must follow the instructions in ${prompt.relativePath}. Below are lines ${prompt.startLine}-${prompt.endLine} of this file:\n`
+                }
                 const entry = {
                     name: prompt.name.substring(0, aditionalContentNameLimit),
-                    description: prompt.description.substring(0, aditionalContentNameLimit),
+                    description: description.substring(0, aditionalContentNameLimit),
                     innerContext: prompt.content.substring(0, additionalContentInnerContextLimit),
                 }
                 // make sure the relevantDocument + additionalContext
@@ -994,7 +1000,6 @@ export class ChatController {
                     break
                 }
 
-                const contextType = this.telemetryHelper.getContextType(prompt)
                 if (contextType === 'rule') {
                     triggerPayload.truncatedAdditionalContextLengths.ruleContextLength += entry.innerContext.length
                 } else if (contextType === 'prompt') {

packages/core/src/codewhispererChat/controllers/chat/telemetryHelper.ts

@@ -42,7 +42,7 @@ import { AuthUtil } from '../../../codewhisperer/util/authUtil'
 import { getSelectedCustomization } from '../../../codewhisperer/util/customizationUtil'
 import { undefinedIfEmpty } from '../../../shared/utilities/textUtilities'
 import { AdditionalContextPrompt } from '../../../amazonq/lsp/types'
-import { getUserPromptsDirectory } from '../../constants'
+import { getUserPromptsDirectory, promptFileExtension } from '../../constants'
 
 export function logSendTelemetryEventFailure(error: any) {
     let requestId: string | undefined
@@ -148,13 +148,14 @@ export class CWCTelemetryHelper {
     }
 
     public getContextType(prompt: AdditionalContextPrompt): string {
-        if (prompt.relativePath.startsWith(path.join('.amazonq', 'rules'))) {
-            return 'rule'
-        } else if (prompt.filePath.startsWith(getUserPromptsDirectory())) {
-            return 'prompt'
-        } else {
-            return 'file'
+        if (prompt.filePath.endsWith(promptFileExtension)) {
+            if (prompt.relativePath.startsWith(path.join('.amazonq', 'rules'))) {
+                return 'rule'
+            } else if (prompt.filePath.startsWith(getUserPromptsDirectory())) {
+                return 'prompt'
+            }
         }
+        return 'file'
     }
 
     public getContextLengths(prompts: AdditionalContextPrompt[]): AdditionalContextLengths {