1. 优化回答格式

2. 使用view提高准确度
3. 优化提示词

Author: wangle201210

eino/eino.go

@@ -2,6 +2,7 @@ package eino
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 )
 
@@ -40,3 +41,25 @@ func ChoiceSQL(sqls, ddl, question string) (sql string, err error) {
 	}
 	return result.Content, nil
 }
+
+func PrettyRes(sql, question string, runResult []map[string]interface{}) (res string, err error) {
+	ctx := context.Background()
+	marshal, err := json.Marshal(runResult)
+	if err != nil {
+		return "", err
+	}
+	messages, err := prettyMessages(question, sql, string(marshal))
+	if err != nil {
+		return "", err
+	}
+
+	cm, err := createOpenAIChatModel(ctx)
+	if err != nil {
+		return "", err
+	}
+	result, err := generate(ctx, cm, messages)
+	if err != nil {
+		return "", fmt.Errorf("优化回答失败: %w", err)
+	}
+	return result.Content, nil
+}

eino/message.go

@@ -31,6 +31,9 @@ func createTemplate() prompt.ChatTemplate {
 			"Pay attention to use CURDATE() function to get the current date, if the question involves \"today\"."+
 			"Can only perform queries and does not accept any modification or deletion functions."+
 			"Use the following table schema info to create your SQL query:\n{ddl}\n"+
+			"I have simplified the syntax of view to be similar to table syntax."+
+			"If the view can meet the needs, use the view first, otherwise use the table"+
+			"Think about it step by step:\n1. Determine the tables to be joined...\n2. Identify the filter conditions...\n3. Choose the aggregation method..."+
 			"The returned content can only contain SQL statements, without explanations or other information, and should not be labeled with SQL tags.",
 		),
 		schema.MessagesPlaceholder("chat_history", true),
@@ -84,3 +87,36 @@ func choiceSqlMessages(sqls, ddl, question string) ([]*schema.Message, error) {
 	}
 	return messages, nil
 }
+
+// createAnswerTemplate 创建并返回一个配置好的聊天模板
+func createAnswerTemplate() prompt.ChatTemplate {
+	// 创建模板，使用 FString 格式
+	return prompt.FromMessages(schema.FString,
+		// 系统消息模板
+		schema.SystemMessage("{role}"+
+			"Given an input question, first create a syntactically correct MySQL query to run, then look at the results of the query and return the answer to the input question."+
+			"The SQL statement obtained is: {sql}。 "+
+			"The result obtained after executing SQL is: {result}。"+
+			"Answer user questions based on the executed SQL and the obtained answers.",
+		),
+		schema.MessagesPlaceholder("chat_history", true),
+		// 用户消息模板
+		schema.UserMessage("Question: {question}"),
+	)
+}
+
+// prettyMessages 优化回答
+func prettyMessages(question, sql, answer string) ([]*schema.Message, error) {
+	template := createAnswerTemplate()
+	data := map[string]any{
+		"role":     role,
+		"question": question,
+		"sql":      sql,
+		"result":   answer,
+	}
+	messages, err := formatMessages(template, data)
+	if err != nil {
+		return nil, err
+	}
+	return messages, nil
+}

mysql/mysql.go

@@ -19,6 +19,9 @@ type Db struct {
 	DataSourceName string   // 数据库连接字信息
 	db             *gorm.DB // gorm数据库实例
 	ddl            []string // 数据库表结构定义列表
+
+	OnlyView  bool // 是否只使用 视图
+	OnlyTable bool // 是否只使用 表
 }
 
 // Init 初始化数据库连接并获取表结构
@@ -62,12 +65,37 @@ func (x *Db) getDdl() ([]string, error) {
 		if err := x.db.Raw(fmt.Sprintf("SHOW CREATE TABLE `%s`", tableName)).Scan(&result).Error; err != nil {
 			return nil, fmt.Errorf("获取表%s的结构失败: %w", tableName, err)
 		}
-
-		createTable, ok := result["Create Table"].(string)
-		if !ok {
-			return nil, fmt.Errorf("表%s的结构格式异常", tableName)
+		if d, e := result["Create Table"]; e {
+			if x.OnlyView {
+				continue
+			}
+			createTable := d.(string)
+			ddls = append(ddls, createTable)
+			continue
+		}
+		if v, e := result["View"]; e {
+			if x.OnlyTable {
+				continue
+			}
+			var columns []ColumnMeta
+			// 其实table也可以这样使用
+			if err := x.db.Raw(fmt.Sprintf("SHOW FULL COLUMNS FROM %s", v)).Scan(&columns).Error; err != nil {
+				return nil, fmt.Errorf("获取表%s的结构失败: %w", tableName, err)
+			}
+			createTable := fmt.Sprintf("CREATE VIEW `%s` (", v)
+			for _, column := range columns {
+				createTable += fmt.Sprintf("\n`%s` %s", column.Field, column.Type)
+				if column.Comment != "" {
+					createTable += fmt.Sprintf(" COMMENT '%s'", column.Comment)
+				}
+				createTable += ","
+			}
+			createTable = createTable[:len(createTable)-1]
+			createTable += ")"
+			ddls = append(ddls, createTable)
+			continue
 		}
-		ddls = append(ddls, createTable)
+		return nil, fmt.Errorf("获取表%s的结构失败: %w", tableName, err)
 	}
 	return ddls, nil
 }

mysql/view.go

@@ -0,0 +1,9 @@
+package mysql
+
+// ColumnMeta 定义列的元数据
+type ColumnMeta struct {
+	Field   string // 列名
+	Type    string // 数据类型
+	Key     string
+	Comment string // 备注，很重要
+}

text2sql.go

@@ -24,6 +24,25 @@ type Text2sql struct {
 	question   string    // 用户输入的问题
 	sqls       []string  // 生成的SQL列表
 	removeSqls []string  // 去除空白后的SQL列表
+
+	OnlyView  bool // 是否只使用 视图
+	OnlyTable bool // 是否只使用 表
+}
+
+func (x *Text2sql) Pretty(question string) (sql string, runResult string, err error) {
+	var res []map[string]interface{}
+	sql, res, err = x.Do(question)
+	if len(res) == 0 {
+		runResult = ""
+		return
+	}
+	// 优化回答
+	runResult, err = eino.PrettyRes(sql, question, res)
+	if err != nil {
+		err = fmt.Errorf("优化回答失败: %w", err)
+		return
+	}
+	return
 }
 
 // Do 执行文本到SQL的转换过程
@@ -48,7 +67,7 @@ func (x *Text2sql) Do(question string) (sql string, runResult []map[string]inter
 	}
 
 	// 初始化数据库连接
-	db := &mysql.Db{DataSourceName: x.DbLink}
+	db := &mysql.Db{DataSourceName: x.DbLink, OnlyView: x.OnlyView, OnlyTable: x.OnlyTable}
 	if err = db.Init(); err != nil {
 		err = fmt.Errorf("初始化数据库失败: %w", err)
 		return