diff --git a/exts/database/src/main/kotlin/cc/unitmesh/database/flow/GenSqlFlow.kt b/exts/database/src/main/kotlin/cc/unitmesh/database/flow/GenSqlFlow.kt
index 36216d3e3e..13b71bf039 100644
--- a/exts/database/src/main/kotlin/cc/unitmesh/database/flow/GenSqlFlow.kt
+++ b/exts/database/src/main/kotlin/cc/unitmesh/database/flow/GenSqlFlow.kt
@@ -2,6 +2,7 @@ package cc.unitmesh.database.flow
 
 import cc.unitmesh.database.DbContextActionProvider
 import cc.unitmesh.devti.AutoDevBundle
+import cc.unitmesh.devti.flow.TaskFlow
 import cc.unitmesh.devti.gui.chat.ChatCodingPanel
 import cc.unitmesh.devti.llms.LLMProvider
 import cc.unitmesh.devti.template.TemplateRender
@@ -16,10 +17,10 @@ class GenSqlFlow(
     val ui: ChatCodingPanel,
     val llm: LLMProvider,
     val project: Project
-) {
+) : TaskFlow {
     private val logger = logger<GenSqlFlow>()
 
-    fun clarify(): String {
+    override fun clarify(): String {
         val stepOnePrompt = generateStepOnePrompt(genSqlContext, actions)
 
         LLMCoroutineScope.scope(project).runCatching {
@@ -35,7 +36,8 @@ class GenSqlFlow(
         }
     }
 
-    fun generate(tableNames: List<String>): String {
+    override fun design(context: Any): String {
+        val tableNames = context as List<String>
         val stepTwoPrompt = generateStepTwoPrompt(genSqlContext, actions, tableNames)
 
         LLMCoroutineScope.scope(project).runCatching {
diff --git a/exts/database/src/main/kotlin/cc/unitmesh/database/flow/GenSqlTask.kt b/exts/database/src/main/kotlin/cc/unitmesh/database/flow/GenSqlTask.kt
index f3c12e0239..a53874a74a 100644
--- a/exts/database/src/main/kotlin/cc/unitmesh/database/flow/GenSqlTask.kt
+++ b/exts/database/src/main/kotlin/cc/unitmesh/database/flow/GenSqlTask.kt
@@ -36,7 +36,7 @@ class GenSqlTask(
 
         indicator.fraction = 0.6
         indicator.text = AutoDevBundle.message("migration.database.sql.generate.generate")
-        val sqlScript = flow.generate(tableNames)
+        val sqlScript = flow.design(tableNames)
 
         logger.info("SQL Script: $sqlScript")
         WriteCommandAction.runWriteCommandAction(project, "Gen SQL", "cc.unitmesh.livingDoc", {
diff --git a/src/main/kotlin/cc/unitmesh/devti/flow/TaskFlow.kt b/src/main/kotlin/cc/unitmesh/devti/flow/TaskFlow.kt
new file mode 100644
index 0000000000..90d9f9d796
--- /dev/null
+++ b/src/main/kotlin/cc/unitmesh/devti/flow/TaskFlow.kt
@@ -0,0 +1,11 @@
+package cc.unitmesh.devti.flow
+
+interface TaskFlow {
+    fun clarify(): String
+    fun design(context: Any): String {
+        return ""
+    }
+    fun execute(): String {
+        return ""
+    }
+}
\ No newline at end of file