refactor: rename generator

Author: phodal

src/main/kotlin/cc/unitmesh/devti/prompt/openai/OpenAIExecutor.kt

@@ -19,7 +19,7 @@ import kotlinx.coroutines.runBlocking
 
 class OpenAIExecutor(val openAIKey: String, val version: String) : AiExecutor, DevtiFlowAction {
     private val openAI: OpenAI = OpenAI(openAIKey)
-    private val gptPromptText = GptPromptText()
+    private val promptGenerator = PromptGenerator()
 
     @OptIn(BetaOpenAI::class)
     override suspend fun prompt(prompt: String): String {
@@ -39,31 +39,31 @@ class OpenAIExecutor(val openAIKey: String, val version: String) : AiExecutor, D
     }
 
     override fun fillStoryDetail(project: SimpleProjectInfo, story: String): String {
-        val promptText = gptPromptText.fillStoryDetail(project, story)
+        val promptText = promptGenerator.fillStoryDetail(project, story)
         return runBlocking {
             val prompt = prompt(promptText)
             return@runBlocking prompt
         }
     }
 
     override fun analysisEndpoint(storyDetail: String, classes: List<DtClass>): String {
-        val promptText = gptPromptText.fillEndpoint(storyDetail, classes)
+        val promptText = promptGenerator.fillEndpoint(storyDetail, classes)
         return runBlocking {
             val prompt = prompt(promptText)
             return@runBlocking prompt
         }
     }
 
     override fun needUpdateMethodForController(targetEndpoint: String, clazz: DtClass, storyDetail: String): String {
-        val promptText = gptPromptText.fillUpdateMethod(clazz, storyDetail)
+        val promptText = promptGenerator.fillUpdateMethod(clazz, storyDetail)
         logger.warn("needUpdateMethodForController prompt text: $promptText")
         return runBlocking {
             return@runBlocking prompt(promptText)
         }
     }
 
     fun codeCompleteFor(text: @NlsSafe String, className: @NlsSafe String?): String {
-        val promptText = gptPromptText.fillCodeComplete(text, className)
+        val promptText = promptGenerator.fillCodeComplete(text, className)
         logger.warn("codeCompleteFor prompt text: $promptText")
         return runBlocking {
             val prompt = prompt(promptText)
@@ -72,7 +72,7 @@ class OpenAIExecutor(val openAIKey: String, val version: String) : AiExecutor, D
     }
 
     fun autoComment(text: @NlsSafe String): String {
-        val promptText = gptPromptText.autoComment(text)
+        val promptText = promptGenerator.autoComment(text)
         logger.warn("autoComponent prompt text: $promptText")
         return runBlocking {
             val prompt = prompt(promptText)

src/main/kotlin/cc/unitmesh/devti/prompt/openai/GptPromptText.kt renamed to src/main/kotlin/cc/unitmesh/devti/prompt/openai/PromptGenerator.kt

@@ -5,7 +5,7 @@ import cc.unitmesh.devti.kanban.SimpleProjectInfo
 import com.intellij.openapi.util.NlsSafe
 import java.io.InputStream
 
-class GptPromptText() {
+class PromptGenerator() {
     // 1. read resources/prompts/create_story_detail.txt
     // 2. replace {project} with project name
     fun fillStoryDetail(project: SimpleProjectInfo, story: String): String {
@@ -17,22 +17,6 @@ class GptPromptText() {
             .replace("{story}", story)
     }
 
-    fun generateControllerCode(story: String): String {
-        return "This is a controller code about a ${story}"
-    }
-
-    fun generateServiceCode(story: String): String {
-        return "This is a service code about a ${story}"
-    }
-
-    fun generateModelCode(story: String): String {
-        return "This is a model code about a ${story}"
-    }
-
-    fun generateRepositoryCode(story: String): String {
-        return "This is a repository code about a ${story}"
-    }
-
     fun fillEndpoint(storyDetail: String, files: List<DtClass>): String {
         val promptText: InputStream = this::class.java.classLoader.getResourceAsStream("prompts/create_endpoint.txt")!!
         val promptTextString = promptText.bufferedReader().use { it.readText() }

src/test/kotlin/cc/unitmesh/devti/prompt/openai/GptPromptTextTest.kt renamed to src/test/kotlin/cc/unitmesh/devti/prompt/openai/PromptGeneratorTest.kt

@@ -5,15 +5,15 @@ import cc.unitmesh.devti.kanban.SimpleProjectInfo
 import org.junit.Assert.assertEquals
 import org.junit.Test
 
-class GptPromptTextTest {
+class PromptGeneratorTest {
 
     @Test
     fun should_fill_story_detail() {
-        val gptPromptText = GptPromptText()
+        val promptGenerator = PromptGenerator()
         val projectInfo = SimpleProjectInfo("", "DevTi", "description")
         val story = "story information"
 
-        val result = gptPromptText.fillStoryDetail(projectInfo, story)
+        val result = promptGenerator.fillStoryDetail(projectInfo, story)
         assertEquals(
             """你是一个敏捷项目的 BA，请根据如下的信息，编写用户故事。
 
@@ -39,10 +39,10 @@ AC 1:  xxx
 
     @Test
     fun should_fill_end_point_prompt() {
-        val gptPromptText = GptPromptText()
+        val promptGenerator = PromptGenerator()
         val storyDetail = "用户故事：可以选择宝贝出行服务"
         val files: List<DtClass> = listOf(DtClass("TaxiController", emptyList()), DtClass("GpsController", emptyList()))
-        val result = gptPromptText.fillEndpoint(storyDetail, files)
+        val result = promptGenerator.fillEndpoint(storyDetail, files)
         assertEquals("""请根据下面的用户故事 和 Controller 列表。要求：
 
 1. 返回最合适的 Controller 名字