feat(database): add SQL generation templates #80

This commit adds two new SQL generation templates for the Database Administrator. The first template, `sql-gen-clarify.vm`, prompts the user to choose the best tables based on their requirements. The second template, `sql-gen-design.vm`, prompts the user to write SQL queries based on their requirements and table information. Both templates provide clear instructions and examples to guide the user.

Author: phodal

exts/database/src/main/kotlin/cc/unitmesh/database/actions/GenSqlScriptBySelection.kt

@@ -43,21 +43,24 @@ class GenSqlScriptBySelection : AbstractChatIntention() {
 
         val tableColumns = DbContextProvider(dasTables).getTableColumns(dasTables.map { it.name })
 
-        val prompt = """
-            |Database: $databaseVersion
-            |Requirement: $selectedText
-            |Tables: ${dasTables.joinToString { it.name }}
-            |Columns: ${tableColumns.joinToString { it }}
-        """.trimMargin()
-
-        println(prompt)
+        DbContext(
+            requirement = selectedText ?: "",
+            databaseVersion = databaseVersion.let {
+                "name: ${it.name}, version: ${it.version}"
+            },
+            schemaName = schemaName,
+            tableNames = dasTables.map { it.name },
+        )
     }
 }
 
 data class DbContext(
+    val requirement: String,
     val databaseVersion: String,
     val schemaName: String,
     val tableNames: List<String>,
+    // for step 2
+    val tableInfos: List<String> = emptyList(),
 ) {
 }
 

src/main/resources/genius/sql/sql-gen-clarify.vm

@@ -0,0 +1,21 @@
+You are a professional Database Administrator.
+According to the user's requirements, you should choose the best Tables for the user in List.
+
+— User use database: ${context.databaseVersion}
+- User schema name: ${context.schemaName}
+- User tables: ${context.tableNames}
+
+For example:
+
+- Question(requirements): calculate the average trip length by subscriber type.// User tables: trips, users, subscriber_type
+- Answer: [trips, subscriber_type]
+
+----
+
+Here are the requirements:
+
+```
+${context.requirement}
+```
+
+Please choose the best Tables for the user, just return the table names in a list, no explain.

src/main/resources/genius/sql/sql-gen-design.vm

@@ -0,0 +1,27 @@
+You are a professional Database Administrator.
+According to the user's requirements, and Tables info, write SQL for the user.
+
+— User use database: ${context.databaseVersion}
+- User schema name: ${context.schemaName}
+- User tableInfos: ${context.tableInfos}
+
+For example:
+
+- Question(requirements): calculate the average trip length by subscriber type.
+// table `subscriber_type`: average_trip_length: int, subscriber_type: string
+- Answer:
+```sql
+select average_trip_length from subscriber_type where subscriber_type = 'subscriber'
+```
+
+----
+
+Here are the requirements:
+
+```
+${context.requirement}
+```
+
+Please write your SQL here:
+
+```sql