feat(test): add support for custom RAG context #195

Enable custom RAG context in test code generation task.
unit-mesh · May 29, 2024 · efb06f3 · efb06f3
1 parent 8b7db54
commit efb06f3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/kotlin/cc/unitmesh/devti/intentions/action/task/TestCodeGenTask.kt b/src/main/kotlin/cc/unitmesh/devti/intentions/action/task/TestCodeGenTask.kt
@@ -193,6 +193,8 @@ class TestCodeGenTask(val request: TestCodeGenRequest) :
     }
 
     private fun getRAGContext(testPromptContext: TestCodeGenContext): String {
+        if (!project.customAgentSetting.enableCustomRag) return ""
+
         val agent = loadRagApp() ?: return ""
 
         val query = testPromptContext.sourceCode