mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-15 12:15:16 +00:00
6d38a164eb
## 背景 仓库早已迁移到 GitHub（`github.com/JianFeeeee/HomeAgent`，CI/CD 都在那边跑）， gitcode 保留为**国内只读镜像**（`ci.yml` 的 sync-gitcode-main 快进推送）。 但 go.mod 的 module 路径一直是 `gitcode.com/JianFeeeee/...` —— 既与权威源 不符，`meta.SourceURL`（AGPL §13 的源码提供入口）也指向镜像而非实际运行的代码。 ## 关键区分（这是本次改动安全的前提） 两个 module 是**不同**的，影响面差一个数量级： | module | 谁依赖 | 改它的后果 | |---|---|---| | `.../HomeAgent`（主仓） | **只有主仓自己** | 内部重构，零外部影响 | | `.../homeagent-sdk`（SDK 仓） | 主仓 49 文件 + 所有外部插件 | 需重编插件 | ★ 实测确认：**插件不 import 主仓 module**（`strings qq/plugin.bin | grep HomeAgent` = 0）， 只依赖 SDK 仓。而插件与内核的边界是 **RPC 方法名字符串** （`plugin.init` / `tool.invoke`…，见 proc/protocol.go），**与 Go module 名无关**。 ⇒ module 改名**不改变任何 wire 协议**，现网已装的插件照常运行。 SDK 的 GitHub 仓名是 `homeagentsdk`（无连字符），module 路径必须与之精确匹配， 否则 `go get` 会因路径不符失败，故取 `github.com/JianFeeeee/homeagentsdk`。 ## 改动 - `go.mod` / `go.sum`：两个 module 路径 + require/replace - 277 个 Go 文件 + Makefile + 打包脚本的 import 与 ldflags 路径 - SDK 仓 `meta.CoreModule`、`tools/hmapdev` 的 module 声明与工程模板 - 对外文档/链接的仓库地址（README、PLUGIN_DEV、OVERVIEW 等） ## 明确**保留** gitcode 的地方（镜像机制，不能动） - `GITCODE_TOKEN`、`GC_URL="…@gitcode.com"`（CI 同步 job） - `gitcode.com/api/v5/...`（release 附件上传） - git remote `gitcode`（镜像拉取/推送） - `docs/zh/ci-cd-runbook.md`（整篇讲镜像运维） ## 验证 - `go build ./...` / `go vet ./...` 绿 - `go test ./... -count=1` → 48 包 ok / 0 FAIL - hmapdev 工具链编译通过（`third_party/homeagent-sdk/tools/hmapdev`） - 全仓残留检查：module 形态的 `gitcode.com/JianFeeeee` = 0 - 镜像机制抽验：GC_URL / @gitcode / api/v5 / remote 均完好 ## 顺带修正 `hmapdev/templates.go` 的注释此前写「gitcode 的模块不在 proxy.golang.org 上」—— module 已迁 GitHub，该论断不再成立。改为如实说明**为何仍默认 replace 到本机 SDK**： 离线可构建、避免版本漂移、内网可用。理由与托管地无关，故 replace 保留。
371 lines
11 KiB
Go
371 lines
11 KiB
Go
package proc
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"github.com/JianFeeeee/HomeAgent/internal/knowledge"
|
||
|
||
pubsdk "github.com/JianFeeeee/homeagentsdk/sdk"
|
||
)
|
||
|
||
// handleGraphMemory 处理图记忆（实体/关系）：recall / commit / introspect / merge / purge。
|
||
//
|
||
// 本函数体是 corehandler.go 里 Handle 那一个大 switch 的**整块平移**：
|
||
// case 标签与 case 体逐字保留，只换了宿主函数（§3.2 的平移原则）。
|
||
func (h *coreHandler) handleGraphMemory(method string, params json.RawMessage) (interface{}, error) {
|
||
switch method {
|
||
// ---- 图记忆（原 case 9/10/11/12/13）----
|
||
case MethodMemoryRecall:
|
||
mem := h.sdk.Memory()
|
||
if mem == nil {
|
||
return nil, errUnavailable("memory")
|
||
}
|
||
var p struct {
|
||
Query []string `json:"query"`
|
||
Depth int `json:"depth"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
entities, relations, err := mem.Recall(p.Query, p.Depth)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if entities == nil {
|
||
entities = []pubsdk.Entity{}
|
||
}
|
||
if relations == nil {
|
||
relations = []pubsdk.Relation{}
|
||
}
|
||
return map[string]interface{}{"entities": entities, "relations": relations}, nil
|
||
|
||
case MethodMemoryCommit:
|
||
mem := h.sdk.Memory()
|
||
if mem == nil {
|
||
return nil, errUnavailable("memory")
|
||
}
|
||
var p struct {
|
||
Triples []pubsdk.Triple `json:"triples"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
return nil, mem.Commit(p.Triples)
|
||
|
||
case MethodMemoryIntrospect:
|
||
mem := h.sdk.Memory()
|
||
if mem == nil {
|
||
return nil, errUnavailable("memory")
|
||
}
|
||
return mem.Introspect()
|
||
|
||
case MethodMemoryMerge:
|
||
mem := h.sdk.Memory()
|
||
if mem == nil {
|
||
return nil, errUnavailable("memory")
|
||
}
|
||
var p struct {
|
||
Source string `json:"source"`
|
||
Target string `json:"target"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
n, err := mem.MergeEntities(p.Source, p.Target)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return map[string]interface{}{"merged": n}, nil
|
||
|
||
case MethodMemoryPurge:
|
||
mem := h.sdk.Memory()
|
||
if mem == nil {
|
||
return nil, errUnavailable("memory")
|
||
}
|
||
var p struct {
|
||
Criteria map[string]string `json:"criteria"`
|
||
Mode string `json:"mode"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
if p.Mode == "" {
|
||
p.Mode = "soft"
|
||
}
|
||
n, err := mem.Purge(p.Criteria, p.Mode)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return map[string]interface{}{"purged": n}, nil
|
||
|
||
}
|
||
|
||
// 组内不应到达：Handle 的分派表与本函数的 case 标签同源，
|
||
// 出现即说明两处不同步。
|
||
return nil, fmt.Errorf("未知 method: %s", method)
|
||
}
|
||
|
||
// handleDocMemory 处理文档记忆：query / insert / insertWithMedia / remove / stats。
|
||
//
|
||
// 本函数体是 corehandler.go 里 Handle 那一个大 switch 的**整块平移**：
|
||
// case 标签与 case 体逐字保留，只换了宿主函数（§3.2 的平移原则）。
|
||
func (h *coreHandler) handleDocMemory(method string, params json.RawMessage) (interface{}, error) {
|
||
switch method {
|
||
// ---- 文档记忆（原 case 14/32/33/34）----
|
||
case MethodDocQuery:
|
||
dm := h.sdk.DocMemory()
|
||
if dm == nil {
|
||
return nil, errUnavailable("doc memory")
|
||
}
|
||
var p struct {
|
||
Text string `json:"text"`
|
||
TopK int `json:"top_k"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
docs := dm.Query(p.Text, p.TopK)
|
||
if docs == nil {
|
||
docs = []*pubsdk.Doc{}
|
||
}
|
||
return map[string]interface{}{"docs": docs}, nil
|
||
|
||
case MethodDocInsert:
|
||
dm := h.sdk.DocMemory()
|
||
if dm == nil {
|
||
return nil, errUnavailable("doc memory")
|
||
}
|
||
var p struct {
|
||
Doc *pubsdk.Doc `json:"doc,omitempty"`
|
||
DocRef SharedRef `json:"doc_ref,omitempty"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
// 文档全文可达几十 KB～数 MB，优先走共享内存。
|
||
if err := h.resolveJSONRef(p.DocRef, &p.Doc); err != nil {
|
||
return nil, err
|
||
}
|
||
if p.Doc == nil {
|
||
return nil, fmt.Errorf("doc.insert: 缺少 doc 字段")
|
||
}
|
||
return nil, dm.Insert(p.Doc)
|
||
|
||
case MethodDocInsertMedia:
|
||
dm := h.sdk.DocMemory()
|
||
if dm == nil {
|
||
return nil, errUnavailable("doc memory")
|
||
}
|
||
var p struct {
|
||
Doc *pubsdk.Doc `json:"doc,omitempty"`
|
||
Attachments []pubsdk.MediaAttachment `json:"attachments,omitempty"`
|
||
DocRef SharedRef `json:"doc_ref,omitempty"`
|
||
AttachRef SharedRef `json:"attachments_ref,omitempty"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
// 文档正文 + 附件（含媒体二进制/data URL）都优先走共享内存。
|
||
if err := h.resolveJSONRef(p.DocRef, &p.Doc); err != nil {
|
||
return nil, err
|
||
}
|
||
if err := h.resolveJSONRef(p.AttachRef, &p.Attachments); err != nil {
|
||
return nil, err
|
||
}
|
||
if p.Doc == nil {
|
||
return nil, fmt.Errorf("doc.insertWithMedia: 缺少 doc 字段")
|
||
}
|
||
if err := dm.InsertWithMedia(p.Doc, p.Attachments); err != nil {
|
||
return nil, err
|
||
}
|
||
// 回传内核补过的字段：ID 新建时才生成，Content 含内核补的媒体标记，
|
||
// MediaDigests 是附件落盘后的完整 digest——插件靠它们后续引用同一份媒体。
|
||
return map[string]interface{}{"doc": p.Doc}, nil
|
||
|
||
case MethodDocRemove:
|
||
dm := h.sdk.DocMemory()
|
||
if dm == nil {
|
||
return nil, errUnavailable("doc memory")
|
||
}
|
||
var p struct {
|
||
ID string `json:"id"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
dm.Remove(p.ID)
|
||
return nil, nil
|
||
|
||
case MethodDocStats:
|
||
dm := h.sdk.DocMemory()
|
||
if dm == nil {
|
||
return nil, errUnavailable("doc memory")
|
||
}
|
||
return dm.Stats(), nil
|
||
|
||
}
|
||
|
||
// 组内不应到达：Handle 的分派表与本函数的 case 标签同源，
|
||
// 出现即说明两处不同步。
|
||
return nil, fmt.Errorf("未知 method: %s", method)
|
||
}
|
||
|
||
// knowledgeMediaAdder 是 knowledge.addWithMedia 需要的扩展能力。
|
||
//
|
||
// 定义为局部接口而非直接依赖 internal/sdk.KnowledgeAPI：那样会让
|
||
// internal/plugin/proc → internal/sdk，而后者已依赖 internal/plugin 的类型，
|
||
// 形成循环（CoreSDK 的注释已说明这一点）。断言失败时返回"能力不可用"，
|
||
// 而不是静默退化成不写媒体——后者会让调用方以为媒体已入库。
|
||
type knowledgeMediaAdder interface {
|
||
AddWithMedia(name, content string, media []knowledge.KnowledgeMediaRef) error
|
||
}
|
||
|
||
// handleKnowledge 处理知识库：search / add / list。
|
||
//
|
||
// 本函数体是 corehandler.go 里 Handle 那一个大 switch 的**整块平移**：
|
||
// case 标签与 case 体逐字保留，只换了宿主函数（§3.2 的平移原则）。
|
||
func (h *coreHandler) handleKnowledge(method string, params json.RawMessage) (interface{}, error) {
|
||
switch method {
|
||
// ---- 知识库（原 case 15/35/36）----
|
||
case MethodKnowledgeSearch:
|
||
kn := h.sdk.Knowledge()
|
||
if kn == nil {
|
||
return nil, errUnavailable("knowledge")
|
||
}
|
||
var p struct {
|
||
Query string `json:"query"`
|
||
TopK int `json:"top_k"`
|
||
Category string `json:"category,omitempty"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
// 分类限定是内核侧扩展能力（见 internal/sdk/knowledge.go），
|
||
// 用局部接口断言取用；能力缺失时退回全库搜索而不是报错——
|
||
// 那只是"少了个筛选条件"，不是调用失败。
|
||
if p.Category != "" {
|
||
if scoped, ok := kn.(interface {
|
||
SearchIn(query, category string, topK int) ([]*pubsdk.Knowledge, error)
|
||
}); ok {
|
||
results, err := scoped.SearchIn(p.Query, p.Category, p.TopK)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if results == nil {
|
||
results = []*pubsdk.Knowledge{}
|
||
}
|
||
return map[string]interface{}{"results": results}, nil
|
||
}
|
||
}
|
||
results, err := kn.Search(p.Query, p.TopK)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if results == nil {
|
||
results = []*pubsdk.Knowledge{}
|
||
}
|
||
return map[string]interface{}{"results": results}, nil
|
||
|
||
case MethodKnowledgeAdd:
|
||
kn := h.sdk.Knowledge()
|
||
if kn == nil {
|
||
return nil, errUnavailable("knowledge")
|
||
}
|
||
var p struct {
|
||
Name string `json:"name"`
|
||
Content string `json:"content,omitempty"`
|
||
ContentRef SharedRef `json:"content_ref,omitempty"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
// 知识正文可达数十 KB，优先走共享内存。内容是 JSON 字符串，
|
||
// 所以从 ref 读出后需再解一层。
|
||
if err := h.resolveJSONRef(p.ContentRef, &p.Content); err != nil {
|
||
return nil, err
|
||
}
|
||
return nil, kn.Add(p.Name, p.Content)
|
||
|
||
case MethodKnowledgeAddMedia:
|
||
kn := h.sdk.Knowledge()
|
||
if kn == nil {
|
||
return nil, errUnavailable("knowledge")
|
||
}
|
||
var p struct {
|
||
Name string `json:"name"`
|
||
Content string `json:"content,omitempty"`
|
||
Media []knowledge.KnowledgeMediaRef `json:"media,omitempty"`
|
||
ContentRef SharedRef `json:"content_ref,omitempty"`
|
||
MediaRef SharedRef `json:"media_ref,omitempty"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
// 正文与媒体清单都可能很大，优先走共享内存（与 knowledge.add 同形）。
|
||
if err := h.resolveJSONRef(p.ContentRef, &p.Content); err != nil {
|
||
return nil, err
|
||
}
|
||
if err := h.resolveJSONRef(p.MediaRef, &p.Media); err != nil {
|
||
return nil, err
|
||
}
|
||
// 多模态是内核侧扩展能力（公开 SDK 契约不含它，见 internal/sdk/knowledge.go）。
|
||
// 这里用局部接口 + 类型断言取用，而不把 internal/sdk 拉进本包：
|
||
// 后者已依赖 internal/plugin，直接引会成环（见 corehandler.go 顶部注释）。
|
||
adder, ok := kn.(knowledgeMediaAdder)
|
||
if !ok {
|
||
return nil, errUnavailable("knowledge multimodal")
|
||
}
|
||
if err := adder.AddWithMedia(p.Name, p.Content, p.Media); err != nil {
|
||
return nil, err
|
||
}
|
||
// 回传媒体清单：插件后续要按 digest 引用同一份媒体。
|
||
return map[string]interface{}{"media": p.Media}, nil
|
||
|
||
case MethodKnowledgeList:
|
||
kn := h.sdk.Knowledge()
|
||
if kn == nil {
|
||
return nil, errUnavailable("knowledge")
|
||
}
|
||
names, err := kn.List()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if names == nil {
|
||
names = []string{}
|
||
}
|
||
return map[string]interface{}{"names": names}, nil
|
||
|
||
}
|
||
|
||
// 组内不应到达：Handle 的分派表与本函数的 case 标签同源，
|
||
// 出现即说明两处不同步。
|
||
return nil, fmt.Errorf("未知 method: %s", method)
|
||
}
|
||
|
||
// handleTextMemory 处理文本记忆追加。
|
||
//
|
||
// 本函数体是 corehandler.go 里 Handle 那一个大 switch 的**整块平移**：
|
||
// case 标签与 case 体逐字保留，只换了宿主函数（§3.2 的平移原则）。
|
||
func (h *coreHandler) handleTextMemory(method string, params json.RawMessage) (interface{}, error) {
|
||
switch method {
|
||
// ---- 文本记忆（原 case 41）----
|
||
case MethodTextMemoryAppend:
|
||
tm := h.sdk.TextMemory()
|
||
if tm == nil {
|
||
return nil, errUnavailable("text memory")
|
||
}
|
||
var p struct {
|
||
Event pubsdk.TextEvent `json:"event"`
|
||
}
|
||
if err := unmarshal(params, &p); err != nil {
|
||
return nil, err
|
||
}
|
||
return nil, tm.Append(p.Event)
|
||
|
||
}
|
||
|
||
// 组内不应到达：Handle 的分派表与本函数的 case 标签同源，
|
||
// 出现即说明两处不同步。
|
||
return nil, fmt.Errorf("未知 method: %s", method)
|
||
}
|