mirror of
https://gitcode.com/JianFeeeee/homeagent-sdk.git
synced 2026-10-31 14:27:30 +00:00
c2474f57a2
## 是什么
agent 统筹开发流程的插件：登记项目 → 开工程线（每条线一个隔离工作目录 +
一个 sidecar 子进程 + 一对通道）→ 投递任务 → 同会话续投 → 评审/合入。
38 个工具：project_*/branch_*/workdir_*/job_*/review_*/test_*/
endpoint_*/agent_*/pi_*/conflict_check。
## sidecar 后端抽象（本轮核心）
传输层（local/ssh）与后端（pi/zcode/自定义）**正交**，四个自由组合：
driver.json 描述一个后端：script / cliPath / env / args / llmConfig
三级查找：插件数据目录 → **随包 rawfile** → 内置 pi 描述
占位符：{workdir} {dataDir} {pluginDir} {rawfile} {sessionId}
★ 内置 pi 回退**与默认后端解耦**（2026-10-17 修）：原实现写
`if cli == defaultAgentCLI { return builtinPiDriver() }`，把"给谁兜底"
和"默认选谁"耦合到同一常量 —— 一旦有人改默认，pi 当场失去兜底，
显式 cli="pi" 开始报「驱动不存在」。现在判断字面 "pi"。
默认后端保持 **pi**（我们本地主用）；交付方用 zcode 走两种局部开关：
· 单线：commondev_branch_open(cli="zcode")
· 按机器：commondev_endpoint_set(host=..., cli="zcode")
## zcode 接入（三条真机判据，全部要求真改工程文件）
· TestZcodeFullPipeline（29-34s）：随包 driver + 真实 ZCode CLI
· TestZcodeOverSSH（39s）：真 ssh + scp，远端 node 跑 zcode
· TestZcodeOverSSHWithPerHostEndpoints（68s）：两个 host 别名配不同
cli_path，走端点表而不是全局 settings
★ 会话库必须隔离到工作区（sidecar.mjs 的 ensureWorkspaceSessionStore）：
实测 ZCode 的会话库路径**硬编码 homedir()**，`ZCODE_DATA_BASE_DIR` 对它
完全无效（26 个会话全落全局库）。但 `<workdir>/.zcode/config.json` 的
`storage.sessionDbPath` **确实生效**。所以启动前写一份工作区配置，
一举解决：多线不共用历史 / 不污染全局 ~/.zcode / 删线不留孤儿会话。
只在不存在时写 —— 用户手工放的要尊重。
## 其他
· rawfile：项目根 rawfile/ 默认全收，解到 <插件目录>/rawfile/，绝对路径拒绝
· 凭证：命名引用 + ${ENV_VAR}，独立 0600 文件，绝不进 config.db(0644)
· secretshell：给不受信 agent 用的安全 shell（ackey + safeshell）
· 冲突检查 / 评审判据 / 镜像 / 每条线独立 workdir（hash 命名，合并后删除）
判据 30+ 条，含 4 条「判据本身的判据」（防止判据假绿）。
448 lines
14 KiB
Go
448 lines
14 KiB
Go
package main
|
||
|
||
// LLM 源导入：把 HomeAgent 内核已配的 LLM 源写进内置 pi 的配置文件。
|
||
//
|
||
// 用户要求（2026-10-14）：
|
||
//
|
||
// 「插件内置 pi 默认使用 homeagent 内部配置的 llms 源，支持添加其他源，
|
||
// 且支持在插件中配置默认使用哪个源」
|
||
// 「pi 配置的源直接写入配置文件」← 不另建存储（pi 本身就有完整配置系统）
|
||
//
|
||
// ## 落点（都是 pi 自己的配置文件）
|
||
//
|
||
// <agentdir>/models.json providers.<name>.{baseUrl,api,apiKey,models[]}
|
||
// <agentdir>/auth.json provider 凭据
|
||
// <agentdir>/settings.json defaultProvider / defaultModel
|
||
//
|
||
// ## 内核侧源的真实位置 ［已核］
|
||
//
|
||
// 存在 **config 表**（不是 config_<plugin>），键前缀 core.llm.：
|
||
//
|
||
// core.llm.provider ← ★ 默认源就是这一项
|
||
// core.llm.sources.<n>.base_url
|
||
// core.llm.sources.<n>.api_key
|
||
// core.llm.sources.<n>.model
|
||
// core.llm.sources.<n>.adapter (“openai” → pi 的 "openai-completions")
|
||
// core.llm.sources.<n>.context_window (可选)
|
||
// core.llm.sources.<n>.vision / .audio (可选)
|
||
//
|
||
// ## 两条安全红线（设计 §3.6.5）
|
||
//
|
||
// 1. models.json / auth.json 含**明文** api key ⇒ 目录 0700、文件 0600。
|
||
// 内核那套是明文存 config.db（该库 644 全局可读）—— **不要把这套惯例带进来**。
|
||
// 2. 导入失败要**报错**，不得静默回退到宿主凭据（~/.pi/agent/auth.json）。
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"sort"
|
||
"strings"
|
||
)
|
||
|
||
// piModelsFile 是 pi 的 models.json 结构（只取我们用到的字段）。
|
||
type piModelsFile struct {
|
||
Providers map[string]*piProvider `json:"providers"`
|
||
}
|
||
|
||
type piProvider struct {
|
||
BaseURL string `json:"baseUrl"`
|
||
API string `json:"api"`
|
||
APIKey string `json:"apiKey,omitempty"`
|
||
Models []piModel `json:"models"`
|
||
}
|
||
|
||
type piModel struct {
|
||
ID string `json:"id"`
|
||
Name string `json:"name,omitempty"`
|
||
Reasoning bool `json:"reasoning"`
|
||
Input []string `json:"input,omitempty"`
|
||
ContextWindow int `json:"contextWindow,omitempty"`
|
||
MaxTokens int `json:"maxTokens,omitempty"`
|
||
}
|
||
|
||
// adapterToAPI 把内核的 adapter 名映射成 pi 的 api 取值。
|
||
//
|
||
// 只映射确定已知的；未知 adapter 返回空串 ⇒ 调用方必须报错，
|
||
// 不能猜一个默认值（猜错会让 pi 用错的协议说话，表现成"模型不回复"）。
|
||
func adapterToAPI(adapter string) string {
|
||
switch strings.ToLower(strings.TrimSpace(adapter)) {
|
||
case "", "openai", "openai-completions", "openai_completions":
|
||
return "openai-completions"
|
||
case "anthropic", "anthropic-messages", "claude":
|
||
return "anthropic-messages"
|
||
default:
|
||
return ""
|
||
}
|
||
}
|
||
|
||
// importKernelSources 把内核已配的 LLM 源导入内置 pi 的配置文件。
|
||
//
|
||
// 返回：导入了几个源、默认源名、错误。
|
||
func (p *Plugin) importKernelSources(agentDir string) (int, string, error) {
|
||
sett := p.sdk.Settings()
|
||
|
||
// ① 源清单：内核的配置是散键（core.llm.sources.<n>.<field>），
|
||
// 所以要先扫出所有 <n>。
|
||
keys, err := sett.ListCore("core.llm.sources.")
|
||
if err != nil {
|
||
return 0, "", fmt.Errorf("读取内核 LLM 源清单失败: %w", err)
|
||
}
|
||
names := map[string]bool{}
|
||
for _, k := range keys {
|
||
rest := strings.TrimPrefix(k, "core.llm.sources.")
|
||
if i := strings.IndexByte(rest, '.'); i > 0 {
|
||
names[rest[:i]] = true
|
||
}
|
||
}
|
||
if len(names) == 0 {
|
||
// 没有任何 sources.* —— 退回单源形态（core.llm.base_url/api_key/model）
|
||
if n, def, err := p.importSingleKernelSource(agentDir); err == nil && n > 0 {
|
||
return n, def, nil
|
||
}
|
||
return 0, "", fmt.Errorf("内核没有任何已配置的 LLM 源（core.llm.sources.* 为空）")
|
||
}
|
||
|
||
ordered := make([]string, 0, len(names))
|
||
for n := range names {
|
||
ordered = append(ordered, n)
|
||
}
|
||
sort.Strings(ordered)
|
||
|
||
models := &piModelsFile{Providers: map[string]*piProvider{}}
|
||
auth := map[string]interface{}{}
|
||
imported := 0
|
||
|
||
for _, n := range ordered {
|
||
prefix := "core.llm.sources." + n + "."
|
||
baseURL := coreStr(sett, prefix+"base_url")
|
||
apiKey := coreStr(sett, prefix+"api_key")
|
||
model := coreStr(sett, prefix+"model")
|
||
adapter := coreStr(sett, prefix+"adapter")
|
||
if baseURL == "" {
|
||
continue // 源不完整：跳过（不静默造一个坏的）
|
||
}
|
||
api := adapterToAPI(adapter)
|
||
if api == "" {
|
||
return imported, "", fmt.Errorf(
|
||
"源 %s 的 adapter=%q 无法映射到 pi 的 api（已知：openai/anthropic）——"+
|
||
"不猜默认值，请显式指定", n, adapter)
|
||
}
|
||
if model == "" {
|
||
model = "AUTO"
|
||
}
|
||
m := piModel{
|
||
ID: model,
|
||
Name: model,
|
||
Input: []string{"text"},
|
||
}
|
||
if cw := coreInt(sett, prefix+"context_window"); cw > 0 {
|
||
m.ContextWindow = cw
|
||
}
|
||
if coreStr(sett, prefix+"vision") == "true" {
|
||
m.Input = []string{"text", "image"}
|
||
}
|
||
models.Providers[n] = &piProvider{
|
||
BaseURL: baseURL, API: api, APIKey: apiKey,
|
||
Models: []piModel{m},
|
||
}
|
||
if apiKey != "" {
|
||
auth[n] = newPiAuthEntry(apiKey)
|
||
}
|
||
imported++
|
||
}
|
||
if imported == 0 {
|
||
return 0, "", fmt.Errorf("内核的 LLM 源都缺 base_url，无可导入")
|
||
}
|
||
|
||
if err := p.writePiConfigs(agentDir, models, auth); err != nil {
|
||
return imported, "", err
|
||
}
|
||
|
||
// ② 默认源 = 内核的 core.llm.provider（用户要求：默认用内核的）
|
||
def := coreStr(sett, "core.llm.provider")
|
||
if def == "" || models.Providers[def] == nil {
|
||
def = ordered[0] // 内核没指定或指到了不存在的源：用第一个（并如实返回）
|
||
}
|
||
if err := p.writePiDefaults(agentDir, def, coreStr(sett, "core.llm.model")); err != nil {
|
||
return imported, def, err
|
||
}
|
||
return imported, def, nil
|
||
}
|
||
|
||
// importSingleKernelSource 处理「内核只有单源、无 sources.* 子表」的形态。
|
||
func (p *Plugin) importSingleKernelSource(agentDir string) (int, string, error) {
|
||
sett := p.sdk.Settings()
|
||
baseURL := coreStr(sett, "core.llm.base_url")
|
||
apiKey := coreStr(sett, "core.llm.api_key")
|
||
model := coreStr(sett, "core.llm.model")
|
||
adapter := coreStr(sett, "core.llm.adapter")
|
||
if baseURL == "" {
|
||
return 0, "", fmt.Errorf("内核未配 core.llm.base_url")
|
||
}
|
||
api := adapterToAPI(adapter)
|
||
if api == "" {
|
||
return 0, "", fmt.Errorf("adapter=%q 无法映射", adapter)
|
||
}
|
||
if model == "" {
|
||
model = "AUTO"
|
||
}
|
||
name := "kernel"
|
||
if n := coreStr(sett, "core.llm.provider"); n != "" {
|
||
name = n
|
||
}
|
||
models := &piModelsFile{Providers: map[string]*piProvider{
|
||
name: {BaseURL: baseURL, API: api, APIKey: apiKey,
|
||
Models: []piModel{{ID: model, Name: model, Input: []string{"text"}}}},
|
||
}}
|
||
auth := map[string]interface{}{}
|
||
if apiKey != "" {
|
||
auth[name] = newPiAuthEntry(apiKey)
|
||
}
|
||
if err := p.writePiConfigs(agentDir, models, auth); err != nil {
|
||
return 0, "", err
|
||
}
|
||
if err := p.writePiDefaults(agentDir, name, model); err != nil {
|
||
return 0, name, err
|
||
}
|
||
return 1, name, nil
|
||
}
|
||
|
||
// writePiConfigs 写 models.json 与 auth.json（★ 0600），并回读校验。
|
||
//
|
||
// 为何要回读校验：json 写坏会让内层 pi 直接起不来，而那种失败看起来
|
||
// 像是"pi 有问题"，排查成本很高。写完必须能解析回来。
|
||
func (p *Plugin) writePiConfigs(agentDir string, models *piModelsFile,
|
||
auth map[string]interface{}) error {
|
||
if err := os.MkdirAll(agentDir, 0o700); err != nil {
|
||
return err
|
||
}
|
||
mb, err := json.MarshalIndent(models, "", " ")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if err := writeFile0600(filepath.Join(agentDir, "models.json"), mb); err != nil {
|
||
return err
|
||
}
|
||
var back piModelsFile
|
||
if err := json.Unmarshal(mb, &back); err != nil {
|
||
return fmt.Errorf("models.json 回读校验失败: %w", err)
|
||
}
|
||
if len(back.Providers) != len(models.Providers) {
|
||
return fmt.Errorf("models.json 回读不一致: 写入 %d 源、读回 %d 源",
|
||
len(models.Providers), len(back.Providers))
|
||
}
|
||
if len(auth) > 0 {
|
||
ab, err := json.MarshalIndent(auth, "", " ")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if err := writeFile0600(filepath.Join(agentDir, "auth.json"), ab); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// writePiDefaults 把默认源写进 settings.json（pi 原生字段）。
|
||
func (p *Plugin) writePiDefaults(agentDir, provider, model string) error {
|
||
path := filepath.Join(agentDir, "settings.json")
|
||
cur := map[string]interface{}{}
|
||
if b, err := os.ReadFile(path); err == nil {
|
||
_ = json.Unmarshal(b, &cur) // 读坏就当空（下面会整体重写）
|
||
}
|
||
if provider != "" {
|
||
cur["defaultProvider"] = provider
|
||
}
|
||
if model != "" {
|
||
cur["defaultModel"] = model
|
||
}
|
||
b, err := json.MarshalIndent(cur, "", " ")
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return writeFile0600(path, b)
|
||
}
|
||
|
||
// writeFile0600 写入并确权 —— 这些文件含明文 api key。
|
||
func writeFile0600(path string, data []byte) error {
|
||
tmp := path + ".tmp"
|
||
if err := os.WriteFile(tmp, data, 0o600); err != nil {
|
||
return err
|
||
}
|
||
if err := os.Chmod(tmp, 0o600); err != nil {
|
||
return err
|
||
}
|
||
return os.Rename(tmp, path)
|
||
}
|
||
|
||
// coreStr / coreInt 读内核配置（类型不匹配一律当空，不 panic）。
|
||
func coreStr(sett interface {
|
||
GetCore(string) (interface{}, error)
|
||
}, key string) string {
|
||
v, err := sett.GetCore(key)
|
||
if err != nil || v == nil {
|
||
return ""
|
||
}
|
||
switch t := v.(type) {
|
||
case string:
|
||
return strings.TrimSpace(t)
|
||
case bool:
|
||
if t {
|
||
return "true"
|
||
}
|
||
return "false"
|
||
case float64:
|
||
return fmt.Sprintf("%v", t)
|
||
default:
|
||
return strings.TrimSpace(fmt.Sprint(v))
|
||
}
|
||
}
|
||
|
||
func coreInt(sett interface {
|
||
GetCore(string) (interface{}, error)
|
||
}, key string) int {
|
||
v, err := sett.GetCore(key)
|
||
if err != nil || v == nil {
|
||
return 0
|
||
}
|
||
switch t := v.(type) {
|
||
case float64:
|
||
return int(t)
|
||
case int:
|
||
return t
|
||
case string:
|
||
var n int
|
||
_, _ = fmt.Sscanf(t, "%d", &n)
|
||
return n
|
||
default:
|
||
return 0
|
||
}
|
||
}
|
||
|
||
// ---------------- 工具实现 ----------------
|
||
|
||
func (p *Plugin) piAgentDir() string {
|
||
return filepath.Join(filepath.Dir(p.path), "runtime", "agentdir")
|
||
}
|
||
|
||
// handlePiSourceImport 把内核的 LLM 源导入内置 pi 的配置文件。
|
||
func (p *Plugin) handlePiSourceImport(args map[string]interface{}) (interface{}, error) {
|
||
n, def, err := p.importKernelSources(p.piAgentDir())
|
||
if err != nil {
|
||
return map[string]interface{}{"error": err.Error()}, nil
|
||
}
|
||
return map[string]interface{}{
|
||
"status": "ok",
|
||
"imported": n,
|
||
"default": def,
|
||
"agentdir": p.piAgentDir(),
|
||
"config_files": []string{"models.json", "auth.json", "settings.json"},
|
||
}, nil
|
||
}
|
||
|
||
// handlePiSourceList 列出已配的源（★ 不返回 key）。
|
||
func (p *Plugin) handlePiSourceList(args map[string]interface{}) (interface{}, error) {
|
||
dir := p.piAgentDir()
|
||
b, err := os.ReadFile(filepath.Join(dir, "models.json"))
|
||
if err != nil {
|
||
if os.IsNotExist(err) {
|
||
return map[string]interface{}{
|
||
"providers": []string{}, "default": "",
|
||
"hint": "内置 pi 尚无 LLM 配置；调 commondev_pi_source_import 导入内核的源",
|
||
}, nil
|
||
}
|
||
return map[string]interface{}{"error": err.Error()}, nil
|
||
}
|
||
var f piModelsFile
|
||
if err := json.Unmarshal(b, &f); err != nil {
|
||
return map[string]interface{}{"error": "models.json 损坏: " + err.Error()}, nil
|
||
}
|
||
|
||
// 默认源从 settings.json 读
|
||
def := ""
|
||
if sb, err := os.ReadFile(filepath.Join(dir, "settings.json")); err == nil {
|
||
var s map[string]interface{}
|
||
if json.Unmarshal(sb, &s) == nil {
|
||
def, _ = s["defaultProvider"].(string)
|
||
}
|
||
}
|
||
// ★ 只暴露名字与端点，绝不返回 apiKey
|
||
type view struct {
|
||
Name string `json:"name"`
|
||
BaseURL string `json:"base_url"`
|
||
API string `json:"api"`
|
||
Model string `json:"model,omitempty"`
|
||
Default bool `json:"default"`
|
||
}
|
||
out := []view{}
|
||
for name, pr := range f.Providers {
|
||
m := ""
|
||
if len(pr.Models) > 0 {
|
||
m = pr.Models[0].ID
|
||
}
|
||
out = append(out, view{Name: name, BaseURL: pr.BaseURL, API: pr.API,
|
||
Model: m, Default: name == def})
|
||
}
|
||
sort.Slice(out, func(i, j int) bool { return out[i].Name < out[j].Name })
|
||
return map[string]interface{}{"providers": out, "default": def, "count": len(out)}, nil
|
||
}
|
||
|
||
// handlePiSourceDefault 切换内置 pi 的默认源。
|
||
func (p *Plugin) handlePiSourceDefault(args map[string]interface{}) (interface{}, error) {
|
||
name := strings.TrimSpace(str(args, "name"))
|
||
if name == "" {
|
||
return map[string]interface{}{"error": "name 必填"}, nil
|
||
}
|
||
dir := p.piAgentDir()
|
||
b, err := os.ReadFile(filepath.Join(dir, "models.json"))
|
||
if err != nil {
|
||
return map[string]interface{}{"error": "读 models.json 失败: " + err.Error()}, nil
|
||
}
|
||
var f piModelsFile
|
||
if err := json.Unmarshal(b, &f); err != nil {
|
||
return map[string]interface{}{"error": "models.json 损坏: " + err.Error()}, nil
|
||
}
|
||
pr, ok := f.Providers[name]
|
||
if !ok {
|
||
names := make([]string, 0, len(f.Providers))
|
||
for n := range f.Providers {
|
||
names = append(names, n)
|
||
}
|
||
sort.Strings(names)
|
||
return map[string]interface{}{
|
||
"error": fmt.Sprintf("源 %q 不存在（可用：%s）", name, strings.Join(names, ", ")),
|
||
}, nil
|
||
}
|
||
model := strings.TrimSpace(str(args, "model"))
|
||
if model == "" && len(pr.Models) > 0 {
|
||
model = pr.Models[0].ID
|
||
}
|
||
if err := p.writePiDefaults(dir, name, model); err != nil {
|
||
return map[string]interface{}{"error": err.Error()}, nil
|
||
}
|
||
return map[string]interface{}{"status": "ok", "default": name, "model": model}, nil
|
||
}
|
||
|
||
// newPiAuthEntry 构造 pi 认的凭据 entry。
|
||
//
|
||
// ★ schema 是 `{"type":"api_key","key":"..."}`，不是 `{"apiKey":"..."}`。
|
||
//
|
||
// 这是**实测踩出来的**（2026-10-15）：内嵌 pi 用旧格式时，
|
||
// auth 读取静默返回空，报 `Provider is not configured: llmsproxy` ——
|
||
// 而 models.json / auth.json / settings.json 三个文件看起来都完全正常。
|
||
// 定位过程：同一个 pi API，只切 PI_CODING_AGENT_DIR 就一边跑得通
|
||
// 一边跑不通，逐字段对比才定位到这个键名。
|
||
//
|
||
// 权威定义（pi 源码）：
|
||
//
|
||
// dist/core/auth-storage.js:185 if (value.type === "api_key") { ... }
|
||
// dist/core/model-runtime.js:423 { type: "api_key", key: apiKey }
|
||
func newPiAuthEntry(apiKey string) map[string]interface{} {
|
||
return map[string]interface{}{
|
||
"type": "api_key",
|
||
"key": apiKey,
|
||
}
|
||
}
|