/dream 完成后存档会话并开启新对话,新增/forge直接清空

This commit is contained in:
2026-08-14 19:33:06 +08:00
parent fc7dd39c54
commit 1973b9d516
4 changed files with 45 additions and 2 deletions
+5
View File
@@ -138,6 +138,11 @@ func (b *Bot) Tools() []string {
return b.toolRegistry.List() return b.toolRegistry.List()
} }
// ClearHistory 清空内存中的会话历史,开启新对话(系统提示词保留)。
func (b *Bot) ClearHistory() {
b.history = nil
}
func (b *Bot) ContextWindow() int64 { func (b *Bot) ContextWindow() int64 {
if m := config.FindModel(b.provider, b.model); m != nil { if m := config.FindModel(b.provider, b.model); m != nil {
return m.ContextWindow return m.ContextWindow
+12
View File
@@ -85,3 +85,15 @@ func TestRestoreTrimsHistory(t *testing.T) {
t.Errorf("history 应截断到 %d, got %d", maxHistory, len(b.history)) t.Errorf("history 应截断到 %d, got %d", maxHistory, len(b.history))
} }
} }
func TestClearHistory(t *testing.T) {
b := newTestBot(t)
b.history = append(b.history, openai.UserMessage("hi"))
b.ClearHistory()
if len(b.history) != 0 {
t.Errorf("history 应清空, got %d", len(b.history))
}
if b.systemPrompt != "测试系统提示" {
t.Errorf("systemPrompt 应保留: %q", b.systemPrompt)
}
}
+27 -1
View File
@@ -20,6 +20,26 @@ func New(b *bot.Bot, db *sql.DB) *Handler {
return &Handler{bot: b, db: db} return &Handler{bot: b, db: db}
} }
// saveSessionAndReset 将当前内存会话存档为新的会话行,并清空内存开启新对话。
func (h *Handler) saveSessionAndReset() {
msgs := h.bot.SessionMessages()
if len(msgs) == 0 {
return
}
provider, model := h.bot.Current()
sess := &store.Session{
Provider: provider,
Model: model,
Messages: msgs,
}
if _, err := store.SaveSession(h.db, sess); err != nil {
fmt.Printf("⚠️ 保存会话失败: %v\n", err)
return
}
h.bot.ClearHistory()
fmt.Printf("💾 会话已存档 (%d 条消息),已开启新对话\n", len(msgs))
}
func formatWindow(n int64) string { func formatWindow(n int64) string {
switch { switch {
case n <= 0: case n <= 0:
@@ -63,7 +83,8 @@ func (h *Handler) Handle(input string) bool {
fmt.Println(" /effort <low|high|max> 设置思考强度") fmt.Println(" /effort <low|high|max> 设置思考强度")
fmt.Println(" /context 打印当前聊天上下文") fmt.Println(" /context 打印当前聊天上下文")
fmt.Println(" /tools 列出可用工具") fmt.Println(" /tools 列出可用工具")
fmt.Println(" /dream 从对话中提取长期记忆") fmt.Println(" /dream 从对话中提取长期记忆并开启新对话")
fmt.Println(" /forge 直接清空对话,不提取记忆不存档")
fmt.Println(" /memories 列出已提取的记忆") fmt.Println(" /memories 列出已提取的记忆")
fmt.Println(" /sessions 列出历史会话") fmt.Println(" /sessions 列出历史会话")
fmt.Println(" /session <id> 切换到历史会话,如 /session 3") fmt.Println(" /session <id> 切换到历史会话,如 /session 3")
@@ -141,6 +162,7 @@ func (h *Handler) Handle(input string) bool {
} }
if len(ms) == 0 { if len(ms) == 0 {
fmt.Println("🧠 没有新的记忆") fmt.Println("🧠 没有新的记忆")
h.saveSessionAndReset()
return true return true
} }
if _, err := store.SaveMemories(h.db, ms); err != nil { if _, err := store.SaveMemories(h.db, ms); err != nil {
@@ -151,6 +173,10 @@ func (h *Handler) Handle(input string) bool {
for _, m := range ms { for _, m := range ms {
fmt.Printf(" [%s %d] %s\n", m.Category, m.Importance, m.Content) fmt.Printf(" [%s %d] %s\n", m.Category, m.Importance, m.Content)
} }
h.saveSessionAndReset()
case "/forge":
h.bot.ClearHistory()
fmt.Println("💬 会话已清空,已开启新对话")
case "/memories": case "/memories":
list, err := store.ListMemories(h.db) list, err := store.ListMemories(h.db)
if err != nil { if err != nil {
+1 -1
View File
@@ -2,7 +2,7 @@ package cli
import "strings" import "strings"
var commands = []string{"/exit", "/quit", "/help", "/models", "/use", "/think", "/effort", "/context", "/tools", "/dream", "/memories", "/sessions", "/session", "/info"} var commands = []string{"/exit", "/quit", "/help", "/models", "/use", "/think", "/effort", "/context", "/tools", "/dream", "/forge", "/memories", "/sessions", "/session", "/info"}
func Complete(line string, models []string) []string { func Complete(line string, models []string) []string {
fields := strings.Fields(line) fields := strings.Fields(line)