From d9916c9dc06ed6c7bc16ba995a369beb9b1dbcfe Mon Sep 17 00:00:00 2001
From: jerryjzhang <jerryjzhang@tencent.com>
Date: Wed, 26 Jun 2024 10:33:18 +0800
Subject: [PATCH] (improvement)(chat)Turn on multi-turn conversation for
 SmallTalkDemo.

---
 .../chat/server/executor/PlainTextExecutor.java          | 9 +++++----
 .../java/com/tencent/supersonic/demo/SmallTalkDemo.java  | 4 ++++
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/chat/server/src/main/java/com/tencent/supersonic/chat/server/executor/PlainTextExecutor.java b/chat/server/src/main/java/com/tencent/supersonic/chat/server/executor/PlainTextExecutor.java
index c3f2f889e..8a86069d1 100644
--- a/chat/server/src/main/java/com/tencent/supersonic/chat/server/executor/PlainTextExecutor.java
+++ b/chat/server/src/main/java/com/tencent/supersonic/chat/server/executor/PlainTextExecutor.java
@@ -26,10 +26,11 @@ import static com.tencent.supersonic.chat.server.parser.ParserConfig.PARSER_MULT
 public class PlainTextExecutor implements ChatExecutor {
 
     private static final String INSTRUCTION = ""
-            + "#Role: You are a nice person to talked to.\n"
-            + "#Task: You will have a small talk with the user, please respond quickly and nicely."
-            + "#History Conversations: %s\n"
-            + "#Current User Input: %s\n"
+            + "#Role: You are a nice person to talk to.\n"
+            + "#Task: Respond quickly and nicely to the user."
+            + "#Rules: 1.ALWAYS use the same language as the input.\n"
+            + "#History Inputs: %s\n"
+            + "#Current Input: %s\n"
             + "#Your response: ";
 
     @Override
diff --git a/launchers/standalone/src/main/java/com/tencent/supersonic/demo/SmallTalkDemo.java b/launchers/standalone/src/main/java/com/tencent/supersonic/demo/SmallTalkDemo.java
index 336731d86..63a428ba5 100644
--- a/launchers/standalone/src/main/java/com/tencent/supersonic/demo/SmallTalkDemo.java
+++ b/launchers/standalone/src/main/java/com/tencent/supersonic/demo/SmallTalkDemo.java
@@ -6,6 +6,7 @@ import com.tencent.supersonic.auth.api.authentication.pojo.User;
 import com.tencent.supersonic.chat.server.agent.Agent;
 import com.tencent.supersonic.chat.server.agent.AgentConfig;
 
+import com.tencent.supersonic.chat.server.agent.MultiTurnConfig;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.core.annotation.Order;
 import org.springframework.stereotype.Component;
@@ -25,6 +26,9 @@ public class SmallTalkDemo extends S2BaseDemo {
         agent.setAgentConfig(JSONObject.toJSONString(agentConfig));
         agent.setExamples(Lists.newArrayList("如何才能变帅",
                 "如何才能赚更多钱", "如何才能世界和平"));
+        MultiTurnConfig multiTurnConfig = new MultiTurnConfig();
+        multiTurnConfig.setEnableMultiTurn(true);
+        agent.setMultiTurnConfig(multiTurnConfig);
 
         agentService.createAgent(agent, User.getFakeUser());
     }