mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-22 01:48:11 +00:00
feat: 完整实现 NLP 三元组提取系统 + token budget 上下文分配
- 重写 extractor.go: 分句、17条 POS 模板、依存模板 + COO 链、ATT合并 - parser.go: 分句循环 + TransE 向量验证(h+r≈t) - fallback.go: jieba POS 降级解析器 - bridge.go: nlp.Triple ↔ memory.Triple 转换 - pipeline.go: extractKeyTriples 改用 NLP 提取器, 删除5条旧前缀规则 - distill.go: docToTriples 改用 NLP 提取器 - reorgGraph: 语义相似度增强检测, 保持纯 LLM 决断 - Provider 接口加 MaxContextTokens() + 模型窗口映射表 - tokenbudget.go: 中文 token 估算器 + budget 分配(80%利用率) - process.go/buildSystemPrompt: 按 token 预算截断 memory+timeline
This commit is contained in:
@ -118,11 +118,11 @@ func TestRecallWithDepth(t *testing.T) {
|
||||
defer g.Close()
|
||||
|
||||
g.Commit([]Triple{
|
||||
{Subject: "甲", Relation: "认识", Object: "乙"},
|
||||
{Subject: "乙", Relation: "认识", Object: "丙"},
|
||||
{Subject: "小明", Relation: "认识", Object: "小红"},
|
||||
{Subject: "小红", Relation: "认识", Object: "小刚"},
|
||||
}, "session3", 0)
|
||||
|
||||
result, err := g.Recall(nil, []string{"甲"}, 2, "")
|
||||
result, err := g.Recall(nil, []string{"小明"}, 2, "")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user