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

phodal · phodal · commit efb06f3287eb · 2024-05-29T17:04:42.000+08:00
Enable custom RAG context in test code generation task.
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

Original file line number	Diff line number	Diff line change
`@@ -193,6 +193,8 @@ class TestCodeGenTask(val request: TestCodeGenRequest) :`
`193`	`193`	`}`
`194`	`194`
`195`	`195`	`private fun getRAGContext(testPromptContext: TestCodeGenContext): String {`
	`196`	`+ if (!project.customAgentSetting.enableCustomRag) return ""`
	`197`	`+`
`196`	`198`	`val agent = loadRagApp() ?: return ""`
`197`	`199`
`198`	`200`	`val query = testPromptContext.sourceCode`