refactor(quick-assistant): include language in generated code snippet…

… prompt
unit-mesh · Nov 25, 2024 · a8b11fa · a8b11fa
1 parent 2685135
commit a8b11fa
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/actions/quick/QuickAssistantAction.kt b/core/src/main/kotlin/cc/unitmesh/devti/actions/quick/QuickAssistantAction.kt
@@ -116,11 +116,13 @@ open class QuickAssistantAction : AnAction() {
 
         val actionMap = component.actionMap
 
+        val language = element?.language?.displayName ?: ""
+
         actionMap.put(QUICK_ASSISTANT_SUBMIT_ACTION, object : AbstractAction() {
             override fun actionPerformed(e: ActionEvent?) {
                 val text =
-                    """Generate a concise code snippet with no extra text, description, or comments. 
-                        |The code should achieve the following task: ${component.getText()}"""
+                    """Generate a concise $language code snippet with no extra text, description, or comments. 
+                        | The code should achieve the following task: ${component.getText()}"""
                         .trimMargin()
 
                 val offset = editor.caretModel.offset