Spring AI借助全局参数实现智能数据库操作与个性化待办管理

String conversation_id = "123";
OpenAiChatOptions openAiChatOptions = OpenAiChatOptions.builder()
        .withModel("hunyuan-pro").withTemperature(0.5)
        .build();
String systemPrompt = """
        - Role: 个人助理小助手
        - Background: 用户需要一个多功能的AI助手，可以提供实时的天气信息、详尽的旅游攻略以及帮助记录待办事项。
        - Profile: 你是一个专业的旅行天气小助手，具备强大的信息检索能力和数据处理能力，能够为用户提供精确的天气信息、详尽的旅游攻略，并帮助管理日常待办事项。
        - Skills: 你拥有强大的网络搜索能力、数据处理能力以及用户交互能力，能够快速准确地为用户提供所需信息。
        - Goals: 提供准确的天气信息，制定包含航班、酒店、火车信息的详尽旅游攻略，并帮助用户记录和管理待办事项。
        - Constrains: 提供的信息必须准确无误，旅游攻略应详尽实用，待办事项管理应简洁高效。
        - OutputFormat: 友好的对话式回复，包含必要的详细信息和格式化的数据。
        - Workflow:
          1. 接收用户的天气查询请求，并提供准确的天气信息。
          2. 根据用户的旅游目的地，搜索并提供包括航班、酒店、火车在内的旅游攻略。
          3. 接收用户的待办事项，并提供简洁的记录和提醒服务。
        """;
ChatMemory chatMemory1 = messageChatMemoryAdvisor.getChatMemory();
String content = this.myChatClientWithSystem
        .prompt()
        .system(systemPrompt)
        .user(userInput)
        .options(openAiChatOptions)
        .advisors(messageChatMemoryAdvisor,myLoggerAdvisor,promptChatKnowledageAdvisor,promptChatDateAdvisor)
        .advisors(advisor -> advisor.param("chat_memory_conversation_id", conversation_id)
                .param("chat_memory_response_size", 100))
        .functions("CurrentWeather","TravelPlanning","toDoListFunctionWithContext")
        .toolContext(Map.of("sessionId", conversation_id, "userMemory", chatMemory1,"client",chatClient))
        .call()
        .content();
log.info("content: {}", content);
ChatDataPO chatDataPO = ChatDataPO.builder().code("text").data(ChildData.builder().text(content).build()).build();
return chatDataPO;

create table todo_info(
  id int(11) auto_increment primary key,
  todo_info varchar(1000) not null,
  todo_Date date not null,
  done boolean not null default false
)

public class ToDoListInfoService implements BiFunction<ToDoListInfoService.ToDoRequest, ToolContext, ToDoListInfoService.ToDoResponse> {
    private JdbcTemplate jdbcTemplate;
    public ToDoListInfoService(JdbcTemplate jdbcTemplate) {
        this.jdbcTemplate = jdbcTemplate;
    }
    @JsonClassDescription("crud:c 代表增加；r：代表查询，u：代表更新，d：代表删除")
    public record ToDoRequest(String crud) {}
    public record ToDoResponse(String message) {}
    @Override
    public ToDoResponse apply(ToDoListInfoService.ToDoRequest request, ToolContext toolContext) {}
}

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-jdbc</artifactId>
</dependency>

@Bean
public FunctionCallback toDoListFunctionWithContext(JdbcTemplate jdbcTemplate) {
    return FunctionCallbackWrapper.builder(new ToDoListInfoService(jdbcTemplate))
            .withName("toDoListFunctionWithContext") // (1) function name
            .withDescription("添加待办,crud:c 代表增加；r：代表查询，u：代表更新，d：代表删除") // (2) function description
            .build();
}

@Slf4j
public class MyMessageChatMemoryAdvisor extends MessageChatMemoryAdvisor {
    private ChatMemory chatMemory;
    public MyMessageChatMemoryAdvisor(ChatMemory chatMemory) {
        super(chatMemory);
        this.chatMemory = chatMemory;
    }
    public MyMessageChatMemoryAdvisor(ChatMemory chatMemory, String defaultConversationId, int chatHistoryWindowSize) {
        super(chatMemory, defaultConversationId, chatHistoryWindowSize);
        this.chatMemory = chatMemory;
    }
    public MyMessageChatMemoryAdvisor(ChatMemory chatMemory, String defaultConversationId, int chatHistoryWindowSize, int order) {
        super(chatMemory, defaultConversationId, chatHistoryWindowSize, order);
        this.chatMemory = chatMemory;
    }
    public ChatMemory getChatMemory() {
        return chatMemory;
    }
}

public ToDoResponse apply(ToDoListInfoService.ToDoRequest request, ToolContext toolContext) {
        String tableinfo = """
                - Role: SQL语句生成专家
                - Background: 用户需要根据特定的表结构和参数信息生成精准的MySQL查询或修改语句，以实现数据库操作的自动化和效率化。
                - Profile: 你是一位经验丰富的数据库管理员和SQL专家，精通MySQL数据库的各种查询和修改语句，能够根据用户提供的表结构和参数信息快速生成正确的SQL语句。
                - Skills: 你具备深厚的数据库理论知识和丰富的实践经验，能够理解复杂的表结构，准确把握用户需求，并据此生成高效、准确的SQL语句。
                - Goals: 根据用户提供的表结构和参数信息，生成可以直接执行的MySQL查询或修改语句，确保语句的正确性和执行的成功率。
                - Constrains: 生成的SQL语句必须符合MySQL的语法规则，能够直接在MySQL数据库中执行，且不包含任何额外的信息或提示。
                - OutputFormat: 纯SQL文本语句，格式规范，无多余信息。禁止使用markdown格式。
                - Workflow:
                  1. 分析用户提供的表结构信息和参数信息。
                  2. 根据分析结果，确定需要执行的数据库操作类型（查询、插入、更新或删除）。
                  3. 结合操作类型和用户提供的信息，生成符合MySQL语法的SQL语句。
                 -example:
                  q.帮我创建一个待办：明天9点提醒我读书。今天日期是2024-11-07 20:20:11
                  a：INSERT INTO todo_info (todo_info, todo_date) VALUES ('明天8点读书', '2024-11-08 08:00:00');
                 - tableinfo：
                     create table todo_info(
                     id int(11) auto_increment primary key,
                     todo_info varchar(1000) not null,
                     todo_Date date not null,
                     done boolean not null default false
                     )
                """;
        String crud = request.crud;
        ChatMemory chatMemory = (ChatMemory)toolContext.getContext().get("userMemory");
        String conversation_id = (String)toolContext.getContext().get("sessionId");
        ChatClient client = (ChatClient)toolContext.getContext().get("client");
        List<Message> messages = chatMemory.get(conversation_id, 1);
        var userqa =  messages.get(0).getContent();
        String jsonString = "执行成功";
        String content = client.prompt()
                .system(tableinfo)
                .user("请根据当前问题生成相应SQL文本即可，禁止生成SQL以外的内容:"+userqa+"，今天日期是:"+ DateUtil.now())
                .call().content();
        try {
            if (crud.equals("r")) {
                jsonString = "查询到待办内容如下：" + JSONObject.toJSONString(jdbcTemplate.queryForList(content));
            } else {
                jdbcTemplate.execute(content);
            }
        }catch (Exception e){
            log.info("ToDoListInfoService:{}", e.getMessage());
            jsonString = "执行失败了";
        }
        log.info("ToDoListInfoService:{}", content);
        return new ToDoResponse(jsonString);