From 6c8ebdfe1a6d37d742d37765bd2131a057f05ff7 Mon Sep 17 00:00:00 2001
From: lexluo09 <39718951+lexluo09@users.noreply.github.com>
Date: Wed, 27 Dec 2023 20:39:08 +0800
Subject: [PATCH] (improvement)(chat) Modify the default configuration related
 to text2sql numbers. (#575)

---
 .../tencent/supersonic/chat/config/OptimizationConfig.java  | 6 +++---
 .../supersonic/chat/parser/sql/llm/SqlPromptGenerator.java  | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/chat/core/src/main/java/com/tencent/supersonic/chat/config/OptimizationConfig.java b/chat/core/src/main/java/com/tencent/supersonic/chat/config/OptimizationConfig.java
index 51b4ac39d..c6583a15d 100644
--- a/chat/core/src/main/java/com/tencent/supersonic/chat/config/OptimizationConfig.java
+++ b/chat/core/src/main/java/com/tencent/supersonic/chat/config/OptimizationConfig.java
@@ -64,13 +64,13 @@ public class OptimizationConfig {
     @Value("${s2SQL.use.switch:true}")
     private boolean useS2SqlSwitch;
 
-    @Value("${text2sql.example.num:10}")
+    @Value("${text2sql.example.num:15}")
     private int text2sqlExampleNum;
 
-    @Value("${text2sql.fewShots.num:5}")
+    @Value("${text2sql.fewShots.num:10}")
     private int text2sqlFewShotsNum;
 
-    @Value("${text2sql.self.consistency.num:2}")
+    @Value("${text2sql.self.consistency.num:5}")
     private int text2sqlSelfConsistencyNum;
 
     @Value("${text2sql.collection.name:text2dsl_agent_collection}")
diff --git a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/SqlPromptGenerator.java b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/SqlPromptGenerator.java
index e6d260d82..db45044ba 100644
--- a/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/SqlPromptGenerator.java
+++ b/chat/core/src/main/java/com/tencent/supersonic/chat/parser/sql/llm/SqlPromptGenerator.java
@@ -107,7 +107,7 @@ public class SqlPromptGenerator {
         for (ElementValue priorLinking : linking) {
             String fieldName = priorLinking.getFieldName();
             String fieldValue = priorLinking.getFieldValue();
-            priorLinkingList.add("'" + fieldValue + "'是一个'" + fieldName + "'");
+            priorLinkingList.add("‘" + fieldValue + "‘是一个‘" + fieldName + "‘");
         }
         String currentDataStr = "当前的日期是" + currentDate;
         String linkingListStr = String.join("，", priorLinkingList);