You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.9 KiB
97 lines
3.9 KiB
package io.mybatis.learn.s03;
|
|
|
|
import io.mybatis.learn.core.AgentRunner;
|
|
import io.mybatis.learn.core.tools.BashTool;
|
|
import io.mybatis.learn.core.tools.EditFileTool;
|
|
import io.mybatis.learn.core.tools.ReadFileTool;
|
|
import io.mybatis.learn.core.tools.WriteFileTool;
|
|
import org.springframework.ai.chat.client.ChatClient;
|
|
import org.springframework.ai.chat.model.ChatModel;
|
|
import org.springframework.boot.CommandLineRunner;
|
|
import org.springframework.boot.SpringApplication;
|
|
import org.springframework.boot.WebApplicationType;
|
|
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
|
|
|
/**
|
|
* S03 - TodoWrite：让 Agent 追踪自己的进度。
|
|
* <p>
|
|
* 格言: "没有计划的 agent 走哪算哪"
|
|
* <p>
|
|
* 核心模式:
|
|
* <pre>
|
|
* +----------+ +-------+ +---------+
|
|
* | User | ---> | LLM | ---> | Tools |
|
|
* | prompt | | | | + todo |
|
|
* +----------+ +---+---+ +----+----+
|
|
* ^ |
|
|
* | tool_result |
|
|
* +---------------+
|
|
* |
|
|
* +-----------+-----------+
|
|
* | TodoManager state |
|
|
* | [ ] task A |
|
|
* | [>] task B |
|
|
* | [x] task C |
|
|
* +-----------------------+
|
|
* </pre>
|
|
* <p>
|
|
* TIP: 对应 Python {@code agents/s03_todo_write.py}。
|
|
* Python 版在工具循环内追踪 {@code rounds_since_todo}，
|
|
* 连续 3 轮未调用 todo 时注入 {@code <reminder>Update your todos.</reminder>} 文本。
|
|
* Spring AI 的 ChatClient 自动管理工具循环，无法在循环内注入文本，
|
|
* 因此改用系统提示注入当前 todo 状态 + 强调更新指令的方式实现同等效果。
|
|
* 如需精确的轮次追踪，可实现自定义 {@code Advisor}。
|
|
*/
|
|
@SpringBootApplication(scanBasePackages = "io.mybatis.learn.core")
|
|
public class S03TodoWrite implements CommandLineRunner {
|
|
|
|
private final ChatModel chatModel;
|
|
private final TodoManager todoManager = new TodoManager();
|
|
|
|
public S03TodoWrite(ChatModel chatModel) {
|
|
this.chatModel = chatModel;
|
|
}
|
|
|
|
@Override
|
|
public void run(String... args) {
|
|
String workDir = System.getProperty("user.dir");
|
|
|
|
/*
|
|
* TIP: 每次用户输入时重建 ChatClient，将最新的 todo 状态注入系统提示。
|
|
* 这对应 Python 版在 agent_loop() 中每轮都能看到 TodoManager 状态的效果。
|
|
*/
|
|
AgentRunner.interactive("s03", userMessage -> {
|
|
// 动态系统提示：包含当前 todo 状态
|
|
String system = "You are a coding agent at " + workDir + ".\n"
|
|
+ "Use the todo tool to plan multi-step tasks. "
|
|
+ "Mark in_progress before starting, completed when done.\n"
|
|
+ "Prefer tools over prose.\n"
|
|
+ "IMPORTANT: You MUST call updateTodos regularly to track your progress.\n\n"
|
|
+ "<current-todos>\n" + todoManager.render() + "\n</current-todos>";
|
|
|
|
ChatClient chatClient = ChatClient.builder(chatModel)
|
|
.defaultSystem(system)
|
|
.defaultTools(
|
|
new BashTool(),
|
|
new ReadFileTool(),
|
|
new WriteFileTool(),
|
|
new EditFileTool(),
|
|
todoManager
|
|
)
|
|
.build();
|
|
|
|
String response = chatClient.prompt()
|
|
.user(userMessage)
|
|
.call()
|
|
.content();
|
|
|
|
return response;
|
|
});
|
|
}
|
|
|
|
public static void main(String[] args) {
|
|
SpringApplication app = new SpringApplication(S03TodoWrite.class);
|
|
app.setWebApplicationType(WebApplicationType.NONE);
|
|
app.run(args);
|
|
}
|
|
}
|
|
|