mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 01:48:11 +00:00
feat(persona): 首启人格门禁跨通道化 + 内核 persona_set 工具
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 落库并置位标记 → 之后不再追问;反向对照(清标记 + 清会话上下文 + 重启)重新开始询问, 排除了「同一段对话里已问过」这一混淆。
This commit is contained in:
@ -981,12 +981,7 @@ func (h *Handler) handleKernel(w http.ResponseWriter, r *http.Request) {
|
||||
// 为什么需要向导:人格曾经只有 <dataDir>/personal/personal.md 一个来源且无人维护,
|
||||
// 里面写死的旧版本号反过来让实例自述旧版本(v1.2.0 压测发现)。
|
||||
// 现在人格是配置项(默认模板不含任何版本号),首启问一次,之后不再打扰。
|
||||
const (
|
||||
personaPromptKey = "core.agent.personal_prompt"
|
||||
personaInitMarker = "core.internal.persona_initialized"
|
||||
)
|
||||
|
||||
// handlePersona 是首启人格向导的后端。
|
||||
// handlePersona 是首启人格向导的后端(与内核 persona_set 工具共用 internal/config 的实现)。
|
||||
//
|
||||
// GET → {initialized, current_prompt, file_override}
|
||||
// POST → {"mode":"default"|"custom"|"later","content":"..."}
|
||||
@ -995,6 +990,9 @@ const (
|
||||
// 生效时机:人格在 homed 启动时载入(以【人格设定】块拼进系统提示词),
|
||||
// 所以**自定义内容需重启生效**;选「默认」或「稍后」(保持当前默认)无需重启。
|
||||
// 不回答就是「稍后」:保留默认并打标记,不阻塞任何流程。
|
||||
//
|
||||
// 跨通道:这里只是 WebUI 侧的入口;任何通道的消息到来时,内核都会检查同一枚标记,
|
||||
// 未确认则在提示词里要求模型主动询问(见 internal/agent/core 的首启门禁)。
|
||||
func (h *Handler) handlePersona(w http.ResponseWriter, r *http.Request) {
|
||||
if h.settings == nil {
|
||||
writeJSON(w, http.StatusServiceUnavailable, map[string]string{"error": "settings not available"})
|
||||
@ -1002,24 +1000,9 @@ func (h *Handler) handlePersona(w http.ResponseWriter, r *http.Request) {
|
||||
}
|
||||
switch r.Method {
|
||||
case http.MethodGet:
|
||||
initialized := false
|
||||
if v, err := h.settings.GetCore(personaInitMarker); err == nil {
|
||||
if s, ok := v.(string); ok && strings.TrimSpace(s) != "" {
|
||||
initialized = true
|
||||
}
|
||||
}
|
||||
cur := ""
|
||||
if v, err := h.settings.GetCore(personaPromptKey); err == nil {
|
||||
if s, ok := v.(string); ok {
|
||||
cur = s
|
||||
}
|
||||
}
|
||||
if cur == "" {
|
||||
cur = internalConfig.DefaultPersonaPrompt
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{
|
||||
"initialized": initialized,
|
||||
"current_prompt": cur,
|
||||
"initialized": internalConfig.PersonaInitializedKV(h.settings),
|
||||
"current_prompt": internalConfig.CurrentPersonaKV(h.settings),
|
||||
"file_override": h.personaFileExists(),
|
||||
})
|
||||
case http.MethodPost:
|
||||
@ -1031,31 +1014,14 @@ func (h *Handler) handlePersona(w http.ResponseWriter, r *http.Request) {
|
||||
writeJSON(w, http.StatusBadRequest, map[string]string{"error": "invalid request"})
|
||||
return
|
||||
}
|
||||
restart := false
|
||||
switch req.Mode {
|
||||
case "default":
|
||||
if err := h.settings.SetCore(personaPromptKey, internalConfig.DefaultPersonaPrompt); err != nil {
|
||||
writeJSON(w, http.StatusInternalServerError, map[string]string{"error": err.Error()})
|
||||
return
|
||||
restart, err := internalConfig.SetPersonaKV(h.settings, req.Mode, req.Content)
|
||||
if err != nil {
|
||||
// 非法 mode / 空内容 → 400;落库失败 → 500。两者都不打标记。
|
||||
code := http.StatusInternalServerError
|
||||
if strings.Contains(err.Error(), "unknown mode") || strings.Contains(err.Error(), "content required") {
|
||||
code = http.StatusBadRequest
|
||||
}
|
||||
case "custom":
|
||||
if strings.TrimSpace(req.Content) == "" {
|
||||
writeJSON(w, http.StatusBadRequest, map[string]string{"error": "content required for custom mode"})
|
||||
return
|
||||
}
|
||||
if err := h.settings.SetCore(personaPromptKey, req.Content); err != nil {
|
||||
writeJSON(w, http.StatusInternalServerError, map[string]string{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
restart = true // 人格在启动时载入
|
||||
case "later":
|
||||
// 保持当前(默认)人格,只打标记,不再问
|
||||
default:
|
||||
writeJSON(w, http.StatusBadRequest, map[string]string{"error": "unknown mode"})
|
||||
return
|
||||
}
|
||||
if err := h.settings.SetCore(personaInitMarker, "1"); err != nil {
|
||||
writeJSON(w, http.StatusInternalServerError, map[string]string{"error": err.Error()})
|
||||
writeJSON(w, code, map[string]string{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
writeJSON(w, http.StatusOK, map[string]interface{}{
|
||||
|
||||
@ -69,10 +69,10 @@ func TestPersonaWizardFlow(t *testing.T) {
|
||||
if w.Code != http.StatusOK {
|
||||
t.Fatalf("later 状态码 %d: %s", w.Code, w.Body.String())
|
||||
}
|
||||
if v := cfgReg.GetString(personaInitMarker, ""); v == "" {
|
||||
if v := cfgReg.GetString(internalConfig.PersonaInitMarkerKey, ""); v == "" {
|
||||
t.Fatal("later 也必须打一次性标记(否则每次启动都问)")
|
||||
}
|
||||
if v := cfgReg.GetString(personaPromptKey, ""); v != internalConfig.DefaultPersonaPrompt {
|
||||
if v := cfgReg.GetString(internalConfig.PersonaPromptKey, ""); v != internalConfig.DefaultPersonaPrompt {
|
||||
t.Fatalf("later 不应改动人格,实际 %q", v)
|
||||
}
|
||||
|
||||
@ -97,7 +97,7 @@ func TestPersonaWizardFlow(t *testing.T) {
|
||||
if !pr.RestartRequired {
|
||||
t.Fatal("自定义人格应提示需要重启才生效")
|
||||
}
|
||||
if v := cfgReg2.GetString(personaPromptKey, ""); v != "你是测试人格" {
|
||||
if v := cfgReg2.GetString(internalConfig.PersonaPromptKey, ""); v != "你是测试人格" {
|
||||
t.Fatalf("自定义内容未写库: %q", v)
|
||||
}
|
||||
|
||||
@ -111,12 +111,12 @@ func TestPersonaWizardFlow(t *testing.T) {
|
||||
t.Fatalf("未知 mode 应 400,实际 %d", w.Code)
|
||||
}
|
||||
// 7. 被拒的请求不得打标记(否则向导会被跳过)
|
||||
if v := cfgReg2.GetString(personaInitMarker, ""); v == "" {
|
||||
if v := cfgReg2.GetString(internalConfig.PersonaInitMarkerKey, ""); v == "" {
|
||||
t.Fatal("前置条件:第 4 步已打标记")
|
||||
}
|
||||
h4, cfgReg4 := newPersonaHandler(t)
|
||||
_ = doPersona(t, h4, http.MethodPost, `{"mode":"nope"}`)
|
||||
if v := cfgReg4.GetString(personaInitMarker, ""); v != "" {
|
||||
if v := cfgReg4.GetString(internalConfig.PersonaInitMarkerKey, ""); v != "" {
|
||||
t.Fatal("被拒的请求不应打标记")
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user