feat(dbChain): 新增上下文内容

src/main/java/cn/com/poc/agent_application/aggregate/impl/AgentApplicationServiceImpl.java

@@ -1133,6 +1133,7 @@ public class AgentApplicationServiceImpl implements AgentApplicationService {
             return null;
         }
         String query = getMessageQuestion(messages);
+        String context = getContext(messages);
         List<DBChainResult> results = new ArrayList<>();
         for (Integer databaseId : databaseIds) {
             BizKnowledgeDatabaseEntity bizKnowledgeDatabaseEntity = bizKnowledgeDatabaseService.get(databaseId.longValue());
@@ -1141,6 +1142,7 @@ public class AgentApplicationServiceImpl implements AgentApplicationService {
             }
             DBChainResponse dbChainResponse = new DBChainResponse();
             dbChainResponse.setQuestion(query);
+            dbChainResponse.setContext(context);
             dbChainResponse.setPrompt(null);
             dbChainResponse.setTableFilters(null);
             dbChainResponse.setMysqlUser(bizKnowledgeDatabaseEntity.getDbUsername());
@@ -1168,6 +1170,20 @@ public class AgentApplicationServiceImpl implements AgentApplicationService {
         return query;
     }
 
+    private String getContext(List<Message> messages) {
+        StringBuilder context = new StringBuilder();
+        if (messages.size() >= 6) {
+            for (int i = 6; i > 0; i--) {
+                Message message = messages.get(messages.size() - i);
+                context.append(message.getRole()).append(":").append(message.getContent()).append(StringUtils.LF);
+            }
+        } else {
+            for (Message message : messages) {
+                context.append(message.getRole()).append(":").append(message.getContent()).append(StringUtils.LF);
+            }
+        }
+        return context.toString();
+    }
 
     /**
      * 更新【记忆变量】结构

src/main/java/cn/com/poc/thirdparty/resource/demand/ai/entity/dbchain/DBChainResponse.java

@@ -15,6 +15,8 @@ public class DBChainResponse extends AbstractRequest<DBChainResult> implements S
 
     private String prompt;
 
+    private String context;
+
     private List<String> tableFilters;
 
     private String mysqlUser;
@@ -29,6 +31,13 @@ public class DBChainResponse extends AbstractRequest<DBChainResult> implements S
 
     private String apiKey;
 
+    public String getContext() {
+        return context;
+    }
+
+    public void setContext(String context) {
+        this.context = context;
+    }
 
     public String getQuestion() {
         return question;