mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-31 22:37:28 +00:00
4e70b8d5d8
批 1，不含 SDK 仓与目录改名。 - go.mod 第 1 行 module 路径 - 366 个 .go 文件的完整前缀 github.com/JianFeeeee/HomeAgent 机械替换 （857 处，仅完整前缀；展示名/注释里的 HomeAgent 字样留给批 5） - 未改：third_party 目录名、SDK 自身 go.mod module、go.mod replace 目标、 homed/waiter 可执行名、.hmap 后缀
69 lines
1.9 KiB
Go
69 lines
1.9 KiB
Go
package core
|
||
|
||
import (
|
||
"encoding/json"
|
||
"os"
|
||
"sync"
|
||
"time"
|
||
|
||
agentAPI "github.com/JianFeeeee/HologramAgent/internal/agent/api"
|
||
)
|
||
|
||
// 提示词抓包（opt-in 诊断）。
|
||
//
|
||
// 为什么需要：系统提示词是**每轮动态拼装**的（身份 + 人格 + 记忆 + 时序 +
|
||
// 工具顺序 + 输出规则 + 技能索引 + 首启门禁…），只看代码永远说不清模型
|
||
// 这一轮到底收到了什么。排查「它怎么又列清单了」「为什么它说有某能力」
|
||
// 这类问题时，唯一可靠的事实是**实际发出去的 payload**。
|
||
//
|
||
// 开关是环境变量而不是配置项：它会把记忆与用户输入原样落盘，属于
|
||
// 临时诊断手段，不该在界面上有一个能被随手打开的开关。
|
||
//
|
||
// HOMED_PROMPT_DUMP=/var/log/homeagent/prompt.jsonl
|
||
//
|
||
// 每行一条 JSON：时间、通道、模型、完整 messages（role + content）。
|
||
// 文件权限 0600 —— 内容是用户隐私。
|
||
const promptDumpEnv = "HOMED_PROMPT_DUMP"
|
||
|
||
var promptDumpMu sync.Mutex
|
||
|
||
func dumpPrompt(channel string, req *agentAPI.CompletionRequest) {
|
||
path := os.Getenv(promptDumpEnv)
|
||
if path == "" || req == nil {
|
||
return
|
||
}
|
||
entry := map[string]interface{}{
|
||
"at": time.Now().Format(time.RFC3339Nano),
|
||
"channel": channel,
|
||
"model": req.Model,
|
||
}
|
||
msgs := make([]map[string]interface{}, 0, len(req.Messages))
|
||
for _, m := range req.Messages {
|
||
item := map[string]interface{}{"role": m.Role, "content": m.Content}
|
||
if len(m.Blocks) > 0 {
|
||
item["blocks"] = len(m.Blocks)
|
||
}
|
||
if len(m.ToolCalls) > 0 {
|
||
item["tool_calls"] = len(m.ToolCalls)
|
||
}
|
||
msgs = append(msgs, item)
|
||
}
|
||
entry["messages"] = msgs
|
||
entry["tool_count"] = len(req.Tools)
|
||
|
||
line, err := json.Marshal(entry)
|
||
if err != nil {
|
||
return
|
||
}
|
||
line = append(line, '\n')
|
||
|
||
promptDumpMu.Lock()
|
||
defer promptDumpMu.Unlock()
|
||
f, err := os.OpenFile(path, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0600)
|
||
if err != nil {
|
||
return
|
||
}
|
||
defer f.Close()
|
||
f.Write(line)
|
||
}
|