mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 01:48:11 +00:00
46e014a8ca
WebUI 首启向导只覆盖 WebUI 这一条通道，而「人格该问一次」是所有通道的事： 走 QQ / CLI / ACP / 邮件来的人永远见不到那个向导，人格就永远是没确认过。 - 门禁移到 buildSystemPrompt（每轮重建 → WebUI/QQ/CLI/ACP/邮件全覆盖）， 以 core.internal.persona_initialized 为准：未确认时要求模型主动询问用户 （默认 / 自定义 / 以后再说），确认后该段消失；personaStore 为 nil 时静默关闭。 - 新增内核内置工具 persona_set(mode=default|custom|later[, content])， 落库逻辑与 WebUI 向导**共用 internal/config**（一个实现 + 两个薄入口： ConfigRegistry 直连 / 插件侧 SettingsAPI），避免两套语义各自漂移。 - AgentConfig 增加 PersonaStore 接口，cmd/homed 用 RegistryPersonaStore 实现。 - 非法输入（未知 mode / custom 空内容）在打标记**之前**拒绝：否则标记置位、 向导被跳过，用户再没机会设。 E2E（隔离实例 + 真实 LLM 往返走 /v1/chat/completions，/var/tmp/persona/e2e.sh）9/9 PASS： 未确认时模型主动询问 → 用户答「用默认的」→ 模型调用 persona_set 落库并置位标记 → 之后不再追问；反向对照（清标记 + 清会话上下文 + 重启）重新开始询问， 排除了「同一段对话里已问过」这一混淆。
100 lines
3.2 KiB
Go
100 lines
3.2 KiB
Go
package config
|
||
|
||
import (
|
||
"strings"
|
||
"testing"
|
||
)
|
||
|
||
// fakeKV 是 PersonaKV 的最小实现（模拟插件侧 SettingsAPI）。
|
||
type fakeKV struct{ m map[string]string }
|
||
|
||
func (f *fakeKV) GetCore(k string) (interface{}, error) { return f.m[k], nil }
|
||
func (f *fakeKV) SetCore(k string, v interface{}) error {
|
||
f.m[k] = v.(string)
|
||
return nil
|
||
}
|
||
|
||
// 三选一语义 + 「只问一次」标记：这是首启向导与 persona_set 工具共用的同一份实现。
|
||
func TestSetPersonaKV(t *testing.T) {
|
||
t.Run("custom 写入内容并要求重启", func(t *testing.T) {
|
||
kv := &fakeKV{m: map[string]string{}}
|
||
restart, err := SetPersonaKV(kv, PersonaModeCustom, "你是测试人格")
|
||
if err != nil || !restart {
|
||
t.Fatalf("custom 应成功且需重启: restart=%v err=%v", restart, err)
|
||
}
|
||
if kv.m[PersonaPromptKey] != "你是测试人格" || kv.m[PersonaInitMarkerKey] != "1" {
|
||
t.Fatalf("落库不对: %+v", kv.m)
|
||
}
|
||
if !PersonaInitializedKV(kv) {
|
||
t.Fatal("打过标记应报告已确认")
|
||
}
|
||
})
|
||
|
||
t.Run("default 写默认模板且无需重启", func(t *testing.T) {
|
||
kv := &fakeKV{m: map[string]string{}}
|
||
restart, err := SetPersonaKV(kv, PersonaModeDefault, "")
|
||
if err != nil || restart {
|
||
t.Fatalf("default 不应需重启: restart=%v err=%v", restart, err)
|
||
}
|
||
if kv.m[PersonaPromptKey] != DefaultPersonaPrompt {
|
||
t.Fatal("default 应写入内置默认模板")
|
||
}
|
||
})
|
||
|
||
t.Run("later 保持人格并打标记", func(t *testing.T) {
|
||
kv := &fakeKV{m: map[string]string{PersonaPromptKey: "原有人格"}}
|
||
if _, err := SetPersonaKV(kv, PersonaModeLater, ""); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if kv.m[PersonaPromptKey] != "原有人格" {
|
||
t.Fatal("later 不应改动人格")
|
||
}
|
||
if kv.m[PersonaInitMarkerKey] != "1" {
|
||
t.Fatal("later 也必须打标记（否则每次启动都问）")
|
||
}
|
||
})
|
||
|
||
t.Run("非法输入必须拒绝且不打标记", func(t *testing.T) {
|
||
for _, c := range []struct{ mode, content string }{
|
||
{PersonaModeCustom, " "}, // 空内容
|
||
{"nope", ""}, // 未知 mode
|
||
} {
|
||
kv := &fakeKV{m: map[string]string{}}
|
||
if _, err := SetPersonaKV(kv, c.mode, c.content); err == nil {
|
||
t.Fatalf("mode=%q content=%q 应报错", c.mode, c.content)
|
||
}
|
||
if kv.m[PersonaInitMarkerKey] == "1" {
|
||
t.Fatalf("mode=%q 被拒时不得打标记（否则向导会被跳过）", c.mode)
|
||
}
|
||
}
|
||
})
|
||
|
||
t.Run("CurrentPersonaKV 未设置时回落默认", func(t *testing.T) {
|
||
kv := &fakeKV{m: map[string]string{}}
|
||
if got := CurrentPersonaKV(kv); got != DefaultPersonaPrompt {
|
||
t.Fatal("未设置应回落默认模板")
|
||
}
|
||
})
|
||
}
|
||
|
||
// 内核直连注册表的入口必须与 KV 版行为一致（同一份实现的两个薄入口）。
|
||
func TestSetPersonaRegistryEntry(t *testing.T) {
|
||
dir := t.TempDir()
|
||
reg := NewConfigRegistry(dir + "/config.db")
|
||
reg.SeedDefaults(dir)
|
||
defer reg.Close()
|
||
|
||
if PersonaInitialized(reg) {
|
||
t.Fatal("全新实例不应已确认人格")
|
||
}
|
||
if _, err := SetPersona(reg, PersonaModeCustom, "内核侧人格"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if !PersonaInitialized(reg) {
|
||
t.Fatal("内核侧落库后应报告已确认")
|
||
}
|
||
if got := reg.GetString(PersonaPromptKey, ""); !strings.Contains(got, "内核侧人格") {
|
||
t.Fatalf("注册表里没有人格内容: %q", got)
|
||
}
|
||
}
|