From b6734d99e1f3b82ab61f93c42442b5a60df104f5 Mon Sep 17 00:00:00 2001
From: lexluo09 <39718951+lexluo09@users.noreply.github.com>
Date: Tue, 28 Nov 2023 15:42:13 +0800
Subject: [PATCH] [improvement](chat) add DemoChatConfig in cspider (#444)

---
 .../com/tencent/supersonic/ConfigureDemo.java | 23 ++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/launchers/standalone/src/main/java/com/tencent/supersonic/ConfigureDemo.java b/launchers/standalone/src/main/java/com/tencent/supersonic/ConfigureDemo.java
index 56a3a7c9f..750a82317 100644
--- a/launchers/standalone/src/main/java/com/tencent/supersonic/ConfigureDemo.java
+++ b/launchers/standalone/src/main/java/com/tencent/supersonic/ConfigureDemo.java
@@ -263,6 +263,23 @@ public class ConfigureDemo implements ApplicationListener<ApplicationReadyEvent>
         configService.addConfig(chatConfigBaseReq, user);
     }
 
+    public void addDemoChatConfig_5(Long modelId) {
+        ChatConfigBaseReq chatConfigBaseReq = new ChatConfigBaseReq();
+        chatConfigBaseReq.setModelId(modelId);
+
+        ChatDetailConfigReq chatDetailConfig = new ChatDetailConfigReq();
+        ChatDefaultConfigReq chatDefaultConfigDetail = new ChatDefaultConfigReq();
+        chatDefaultConfigDetail.setUnit(-1);
+        chatDefaultConfigDetail.setPeriod("DAY");
+        chatDetailConfig.setChatDefaultConfig(chatDefaultConfigDetail);
+        ItemVisibility visibility0 = new ItemVisibility();
+        chatDetailConfig.setVisibility(visibility0);
+        chatConfigBaseReq.setChatDetailConfig(chatDetailConfig);
+        List<RecommendedQuestionReq> recommendedQuestions = new ArrayList<>();
+        chatConfigBaseReq.setRecommendedQuestions(recommendedQuestions);
+        configService.addConfig(chatConfigBaseReq, user);
+    }
+
 
     private void addPlugin_1() {
         Plugin plugin1 = new Plugin();
@@ -357,7 +374,7 @@ public class ConfigureDemo implements ApplicationListener<ApplicationReadyEvent>
         LLMParserTool llmParserTool = new LLMParserTool();
         llmParserTool.setId("1");
         llmParserTool.setType(AgentToolType.LLM_S2SQL);
-        llmParserTool.setModelIds(Lists.newArrayList(5L,6L,7L,8L));
+        llmParserTool.setModelIds(Lists.newArrayList(5L, 6L, 7L, 8L));
         agentConfig.getTools().add(llmParserTool);
 
         agent.setAgentConfig(JSONObject.toJSONString(agentConfig));
@@ -375,6 +392,10 @@ public class ConfigureDemo implements ApplicationListener<ApplicationReadyEvent>
             addDemoChatConfig_2();
             addDemoChatConfig_3();
             addDemoChatConfig_4();
+            addDemoChatConfig_5(5L);
+            addDemoChatConfig_5(6L);
+            addDemoChatConfig_5(7L);
+            addDemoChatConfig_5(8L);
             addPlugin_1();
             addAgent1();
             addAgent2();