jianf/HomeAgent
1
0
Fork 0
mirror of https://gitcode.com/JianFeeeee/HomeAgent.git synced 2026-10-15 12:15:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

蒸馏嵌入接线：Distiller/Agent 注入共享 embedder，修复配置缺失时蒸馏零产出

Browse Source
This commit is contained in:
JianFeeeee
2026-08-05 09:59:33 +08:00
parent adaf37909d
commit 954c9dafcb
7 changed files with 42 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
internal/memory/pipeline/pipeline_test.go
View File

@ -7,8 +7,16 @@ import (
"time"
"gitcode.com/JianFeeeee/HomeAgent/internal/memory"
"gitcode.com/JianFeeeee/HomeAgent/internal/memory/vector"
)
// constVectorizer 恒等向量器：所有候选统一过融合阈值，验证嵌入接线是否生效。
type constVectorizer struct{}
func (constVectorizer) Vectorize(text string) vector.Vector {
return vector.Vector{"0": 1.0, "1": 0.5}
}
func TestNewDistiller(t *testing.T) {
d := NewDistiller(nil, t.TempDir(), DistillerConfig{
Interval: 10 * time.Minute,
@ -147,7 +155,7 @@ func TestExtractKeyTriples(t *testing.T) {
}
for _, tt := range tests {
triples := extractKeyTriples(tt.user, tt.assistant)
triples := extractKeyTriples(tt.user, tt.assistant, constVectorizer{})
if tt.check != nil && !tt.check(triples) {
t.Errorf("extractKeyTriples(%q) = %v, check failed", tt.user, triples)
}