jianf/HomeAgent
1
0
Fork 0
mirror of https://gitcode.com/JianFeeeee/HomeAgent.git synced 2026-09-22 01:48:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

memory 测试自包含：合成 word2vec 模型替代 /tmp 大模型依赖，测试不再联网下载 fastText

Browse Source
This commit is contained in:
root
2026-08-05 09:59:30 +08:00
parent 93efb1773f
commit c6ebbfca34
5 changed files with 171 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
internal/memory/static_embedder_test.go
View File

@ -32,7 +32,7 @@ func cosineSim(a, b vector.Vector) float64 {
}
func TestStaticEmbedderLoad(t *testing.T) {
e := NewStaticEmbedder("/tmp/cc.zh.sample.vec")
e := newSynthEmbedder(t, 300)
if !e.Loaded() {
t.Fatal("embedder should be loaded")
}
@ -42,7 +42,7 @@ func TestStaticEmbedderLoad(t *testing.T) {
}
func TestStaticEmbedderConsistency(t *testing.T) {
e := NewStaticEmbedder("/tmp/cc.zh.sample.vec")
e := newSynthEmbedder(t, 300)
v1 := e.Vectorize("今天天气怎么样")
v2 := e.Vectorize("今天天气怎么样")
@ -81,7 +81,7 @@ func TestStaticEmbedderFallback(t *testing.T) {
}
func TestStaticEmbedderAllInDenseSpace(t *testing.T) {
e := NewStaticEmbedder("/tmp/cc.zh.sample.vec")
e := newSynthEmbedder(t, 300)
texts := []string{
"今天天气怎么样",
@ -105,7 +105,7 @@ func TestStaticEmbedderAllInDenseSpace(t *testing.T) {
}
func TestStaticEmbedderSemanticSimilarity(t *testing.T) {
e := NewStaticEmbedder("/tmp/cc.zh.sample.vec")
e := newSynthEmbedder(t, 300)
pairs := []struct {
a, b string
@ -134,7 +134,7 @@ func TestStaticEmbedderSemanticSimilarity(t *testing.T) {
}
func TestContextPruneWithRealEmbedding(t *testing.T) {
e := NewStaticEmbedder("/tmp/cc.zh.sample.vec")
e := newSynthEmbedder(t, 300)
type event struct {
input string