feat: Phase5C+5D /compact AI摘要压缩 + 多行输入支持

/compact 增强:
- 用AI生成对话历史摘要替代简单清空
- 压缩后保留:原始系统提示 + AI摘要 + 最后一轮对话
- AI摘要保留关键决策、代码变更、文件路径等技术细节
- 摘要生成失败时静默降级(仅保留最近对话)

多行输入:
- 支持反斜杠(\\)续行:行末输入\\后回车继续下一行
- 续行提示符: '  ... '
- JLine SECONDARY_PROMPT_PATTERN 配置

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
pull/1/head
liuzh 1 month ago
parent 399cea4478
commit 094057f0d2
  1. 108
      src/main/java/com/claudecode/command/impl/CompactCommand.java
  2. 18
      src/main/java/com/claudecode/repl/ReplSession.java

@ -4,15 +4,34 @@ import com.claudecode.command.CommandContext;
import com.claudecode.command.SlashCommand; import com.claudecode.command.SlashCommand;
import com.claudecode.console.AnsiStyle; import com.claudecode.console.AnsiStyle;
import com.claudecode.core.TokenTracker; import com.claudecode.core.TokenTracker;
import org.springframework.ai.chat.messages.*;
import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.chat.model.ChatResponse;
import org.springframework.ai.chat.prompt.Prompt;
import java.util.ArrayList;
import java.util.List;
/** /**
* /compact 命令 压缩当前对话上下文 * /compact 命令 AI 生成摘要来压缩上下文
* <p> * <p>
* 对应 claude-code/src/commands/compact.ts * 对应 claude-code/src/commands/compact.ts
* 保留系统提示词用摘要替换详细的对话历史 * 将详细的对话历史替换为 AI 生成的摘要大幅减少 token 消耗
* 保留系统提示词和最近一轮对话
*/ */
public class CompactCommand implements SlashCommand { public class CompactCommand implements SlashCommand {
private static final String COMPACT_PROMPT = """
请将以下对话历史压缩为一段简洁的摘要要求
1. 保留所有关键决策代码变更和技术细节
2. 保留文件路径函数名等具体信息
3. 保留用户的偏好和要求
4. 省略重复的讨论和无关的细节
5. 用中文输出控制在500字以内
对话历史
""";
@Override @Override
public String name() { public String name() {
return "compact"; return "compact";
@ -20,38 +39,97 @@ public class CompactCommand implements SlashCommand {
@Override @Override
public String description() { public String description() {
return "Compact conversation context"; return "用AI摘要压缩对话上下文";
} }
@Override @Override
public String execute(String args, CommandContext context) { public String execute(String args, CommandContext context) {
if (context.agentLoop() == null) { if (context.agentLoop() == null) {
return AnsiStyle.yellow(" ⚠ No active conversation to compact."); return AnsiStyle.yellow(" ⚠ 没有活跃的对话可压缩。");
} }
int before = context.agentLoop().getMessageHistory().size(); List<Message> history = context.agentLoop().getMessageHistory();
int before = history.size();
if (before <= 2) { if (before <= 3) {
return AnsiStyle.dim(" Context is already minimal (" + before + " messages). Nothing to compact."); return AnsiStyle.dim(" 上下文已经很小(" + before + " 条消息),无需压缩。");
} }
// 记录压缩前的 token 使用
TokenTracker tracker = context.agentLoop().getTokenTracker(); TokenTracker tracker = context.agentLoop().getTokenTracker();
long tokensBefore = tracker.getInputTokens() + tracker.getOutputTokens(); long tokensBefore = tracker.getInputTokens() + tracker.getOutputTokens();
// 重置历史(保留系统提示词) // 尝试用 AI 生成摘要
context.agentLoop().reset(); String summary = generateSummary(context, history);
// 构建压缩后的历史:系统提示 + 摘要作为系统消息 + 保留最后一轮对话
List<Message> compacted = new ArrayList<>();
compacted.add(history.getFirst()); // 原始系统提示词
if (summary != null && !summary.isBlank()) {
compacted.add(new SystemMessage("[对话历史摘要] " + summary));
}
// 保留最后一轮用户消息和助手回复(如果有)
for (int i = Math.max(1, before - 2); i < before; i++) {
compacted.add(history.get(i));
}
int after = context.agentLoop().getMessageHistory().size(); context.agentLoop().replaceHistory(compacted);
int after = compacted.size();
StringBuilder sb = new StringBuilder(); StringBuilder sb = new StringBuilder();
sb.append(AnsiStyle.green(" ✅ Context compacted")).append("\n"); sb.append(AnsiStyle.green(" ✅ 上下文已压缩")).append("\n");
sb.append(" Messages: ").append(before).append(" → ").append(after).append("\n"); sb.append(" 消息数: ").append(before).append(" → ").append(after).append("\n");
if (tokensBefore > 0) { if (tokensBefore > 0) {
sb.append(" Tokens used before compact: ").append(TokenTracker.formatTokens(tokensBefore)).append("\n"); sb.append(" 压缩前累计 Token: ").append(TokenTracker.formatTokens(tokensBefore)).append("\n");
}
if (summary != null) {
sb.append(AnsiStyle.dim(" 📝 AI 摘要已生成并注入上下文"));
} else {
sb.append(AnsiStyle.dim(" ⚠ AI 摘要生成失败,仅保留最近对话"));
} }
sb.append(AnsiStyle.dim(" Conversation history cleared. System prompt retained."));
return sb.toString(); return sb.toString();
} }
/** 调用 AI 生成对话摘要 */
private String generateSummary(CommandContext context, List<Message> history) {
try {
ChatModel chatModel = context.agentLoop().getChatModel();
// 构建摘要请求的消息列史
StringBuilder dialogText = new StringBuilder();
for (Message msg : history) {
switch (msg) {
case UserMessage um -> dialogText.append("[用户] ").append(um.getText()).append("\n");
case AssistantMessage am -> {
if (am.getText() != null && !am.getText().isBlank()) {
// 截断过长的助手回复
String text = am.getText();
if (text.length() > 500) text = text.substring(0, 500) + "...";
dialogText.append("[助手] ").append(text).append("\n");
}
if (am.hasToolCalls()) {
for (var tc : am.getToolCalls()) {
dialogText.append("[工具调用] ").append(tc.name()).append("\n");
}
}
}
default -> {} // 跳过系统消息和工具响应
}
}
if (dialogText.isEmpty()) return null;
Prompt summaryPrompt = new Prompt(List.of(
new UserMessage(COMPACT_PROMPT + dialogText)
));
ChatResponse response = chatModel.call(summaryPrompt);
return response.getResult().getOutput().getText();
} catch (Exception e) {
// 摘要生成失败不影响压缩操作
return null;
}
}
} }

@ -114,35 +114,33 @@ public class ReplSession {
.streams(System.in, System.out) .streams(System.in, System.out)
.build()) { .build()) {
// 检测是否为 dumb 终端并提示
boolean isDumb = "dumb".equals(terminal.getType()); boolean isDumb = "dumb".equals(terminal.getType());
if (isDumb) { if (isDumb) {
log.info("当前为 dumb 终端模式,建议使用 Windows Terminal / PowerShell / cmd 获得完整体验"); log.info("当前为 dumb 终端模式,建议使用 Windows Terminal / PowerShell / cmd 获得完整体验");
} }
// 配置 Parser:支持反斜杠续行 (\) 和 三引号块 (""")
DefaultParser parser = new DefaultParser();
parser.setEscapeChars(new char[]{'\\'}); // 反斜杠续行
LineReader reader = LineReaderBuilder.builder() LineReader reader = LineReaderBuilder.builder()
.terminal(terminal) .terminal(terminal)
.parser(new DefaultParser()) .parser(parser)
.completer(new ClaudeCodeCompleter(commandRegistry, toolRegistry)) .completer(new ClaudeCodeCompleter(commandRegistry, toolRegistry))
.variable(LineReader.HISTORY_FILE, historyDir.resolve("history")) .variable(LineReader.HISTORY_FILE, historyDir.resolve("history"))
.variable(LineReader.HISTORY_SIZE, 1000) .variable(LineReader.HISTORY_SIZE, 1000)
.variable(LineReader.SECONDARY_PROMPT_PATTERN, "%P ... ")
.option(LineReader.Option.CASE_INSENSITIVE, true) .option(LineReader.Option.CASE_INSENSITIVE, true)
.option(LineReader.Option.AUTO_LIST, true) .option(LineReader.Option.AUTO_LIST, true)
.build(); .build();
// 构建彩色提示符 // 提示符
String prompt = new AttributedStringBuilder() String prompt = new AttributedStringBuilder()
.style(AttributedStyle.BOLD.foreground(AttributedStyle.CYAN)) .style(AttributedStyle.BOLD.foreground(AttributedStyle.CYAN))
.append("❯ ") .append("❯ ")
.style(AttributedStyle.DEFAULT) .style(AttributedStyle.DEFAULT)
.toAnsi(terminal); .toAnsi(terminal);
// 续行提示符(多行输入时显示)
String rightPrompt = new AttributedStringBuilder()
.style(AttributedStyle.DEFAULT.foreground(AttributedStyle.BRIGHT))
.append("")
.toAnsi(terminal);
printBanner(terminal); printBanner(terminal);
CommandContext cmdContext = new CommandContext(agentLoop, toolRegistry, out, () -> running = false); CommandContext cmdContext = new CommandContext(agentLoop, toolRegistry, out, () -> running = false);
@ -152,12 +150,10 @@ public class ReplSession {
try { try {
input = reader.readLine(prompt).strip(); input = reader.readLine(prompt).strip();
} catch (UserInterruptException e) { } catch (UserInterruptException e) {
// Ctrl+C —— 取消当前输入,继续等待
spinner.stop(); spinner.stop();
out.println(AnsiStyle.dim(" ^C")); out.println(AnsiStyle.dim(" ^C"));
continue; continue;
} catch (EndOfFileException e) { } catch (EndOfFileException e) {
// Ctrl+D —— 退出
break; break;
} }

Loading…
Cancel
Save