64 lines
2.3 KiB
Java
64 lines
2.3 KiB
Java
package io.mybatis.learn.s07;
|
|
|
|
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;
|
|
|
|
import java.nio.file.Path;
|
|
|
|
/**
|
|
* S07 - 任务系统：大目标拆成小任务，排好序，记在磁盘上。
|
|
* <p>
|
|
* 格言: "大目标要拆成小任务, 排好序, 记在磁盘上"
|
|
* <p>
|
|
* TIP: 对应 Python {@code agents/s07_task_system.py}。
|
|
* 核心洞察: 任务状态持久化在磁盘（{@code .tasks/task_*.json}），
|
|
* 不受上下文压缩影响 —— 因为它在对话之外。
|
|
*/
|
|
@SpringBootApplication(scanBasePackages = "io.mybatis.learn.core")
|
|
public class S07TaskSystem implements CommandLineRunner {
|
|
|
|
private final ChatClient chatClient;
|
|
|
|
public S07TaskSystem(ChatModel chatModel) {
|
|
Path tasksDir = Path.of(System.getProperty("user.dir"), ".tasks");
|
|
TaskManager taskManager = new TaskManager(tasksDir);
|
|
|
|
this.chatClient = ChatClient.builder(chatModel)
|
|
.defaultSystem("You are a coding agent at " + System.getProperty("user.dir")
|
|
+ ". Use task tools to plan and track work.")
|
|
.defaultTools(
|
|
new BashTool(),
|
|
new ReadFileTool(),
|
|
new WriteFileTool(),
|
|
new EditFileTool(),
|
|
taskManager
|
|
)
|
|
.build();
|
|
}
|
|
|
|
@Override
|
|
public void run(String... args) {
|
|
AgentRunner.interactive("s07", userMessage ->
|
|
chatClient.prompt()
|
|
.user(userMessage)
|
|
.call()
|
|
.content()
|
|
);
|
|
}
|
|
|
|
public static void main(String[] args) {
|
|
SpringApplication app = new SpringApplication(S07TaskSystem.class);
|
|
app.setWebApplicationType(WebApplicationType.NONE);
|
|
app.run(args);
|
|
}
|
|
}
|