Vectory type insert/update filter

Author: sim-wangyan

builder_insert.go

@@ -62,6 +62,10 @@ func (b *InsertBuilder) Set(k string, v interface{}) *InsertBuilder {
 	case time.Time:
 		ts := v.(time.Time).Format("2006-01-02 15:04:05")
 		v = ts
+	case Vector:
+		// Vector 类型：不做处理，保持原样
+		// 让 database/sql 调用 driver.Valuer 接口
+		// Vector.Value() 会返回正确的数据库格式
 	case interface{}:
 		bytes, _ := json.Marshal(v)
 		v = string(bytes)

builder_update.go

@@ -65,6 +65,10 @@ func (ub *UpdateBuilder) Set(k string, v interface{}) *UpdateBuilder {
 	case time.Time:
 		ts := v.(time.Time).Format("2006-01-02 15:04:05")
 		v = ts
+	case Vector:
+		// Vector 类型：不做处理，保持原样
+		// 让 database/sql 调用 driver.Valuer 接口
+		// Vector.Value() 会返回正确的数据库格式
 	case interface{}:
 		bytes, _ := json.Marshal(v)
 		v = string(bytes)

vector_test.go

@@ -44,6 +44,24 @@ func TestVectorSearch_Basic(t *testing.T) {
 		Build().
 		SqlOfVectorSearch()
 
+	t.Logf("=== SELECT 向量检索测试 ===")
+	t.Logf("SQL: %s", sql)
+	t.Logf("Args count: %d", len(args))
+	if len(args) > 0 {
+		t.Logf("Args[0] type: %T", args[0])
+		t.Logf("Args[0] value: %v", args[0])
+
+		// 检查查询参数类型
+		switch args[0].(type) {
+		case Vector:
+			t.Logf("✅ 查询参数是 Vector 类型，driver.Valuer 会被调用")
+		case string:
+			t.Logf("⚠️ 查询参数是 string 类型")
+		default:
+			t.Logf("❓ 查询参数是未知类型: %T", args[0])
+		}
+	}
+
 	expectedSQL := "SELECT *, embedding <-> ? AS distance FROM code_vectors ORDER BY distance LIMIT 10"
 
 	if sql != expectedSQL {
@@ -66,6 +84,9 @@ func TestVectorSearch_WithScalarFilter(t *testing.T) {
 		Build().
 		SqlOfVectorSearch()
 
+	t.Logf("SQL: %s", sql)
+	t.Logf("Args: %d", len(args))
+
 	// SQL 应该包含 WHERE 条件
 	if !containsString(sql, "WHERE") {
 		t.Errorf("Expected WHERE clause in SQL: %s", sql)
@@ -93,12 +114,16 @@ func TestVectorSearch_WithScalarFilter(t *testing.T) {
 func TestVectorSearch_L2Distance(t *testing.T) {
 	queryVector := Vector{0.1, 0.2, 0.3}
 
-	sql, _ := Of(&CodeVector{}).
+	sql, args := Of(&CodeVector{}).
 		VectorSearch("embedding", queryVector, 10).
 		VectorDistance(L2Distance).
 		Build().
 		SqlOfVectorSearch()
 
+	t.Logf("SQL: %s", sql)
+	t.Logf("Distance Metric: L2Distance (<#>)")
+	t.Logf("Args: %d", len(args))
+
 	// SQL 应该使用 <#> 运算符
 	if !containsString(sql, "<#>") {
 		t.Errorf("Expected <#> (L2 distance) in SQL: %s", sql)
@@ -115,6 +140,10 @@ func TestVectorDistanceFilter(t *testing.T) {
 		Build().
 		SqlOfVectorSearch()
 
+	t.Logf("SQL: %s", sql)
+	t.Logf("Threshold: < 0.3")
+	t.Logf("Args: %d", len(args))
+
 	// SQL 应该包含距离过滤条件
 	if !containsString(sql, "<-> ?") {
 		t.Errorf("Expected distance filter in SQL: %s", sql)
@@ -143,6 +172,10 @@ func TestVectorSearch_AutoIgnoreNil(t *testing.T) {
 		Build().
 		SqlOfVectorSearch()
 
+	t.Logf("SQL: %s", sql)
+	t.Logf("Note: Empty language auto-ignored")
+	t.Logf("Args: %d", len(args))
+
 	// SQL 不应该包含 language（因为是空字符串）
 	// 但应该包含 layer
 	if containsString(sql, "language") {
@@ -166,12 +199,14 @@ func TestVector_Distance(t *testing.T) {
 
 	// 余弦距离
 	cosDist := vec1.Distance(vec2, CosineDistance)
+	t.Logf("Cosine Distance: %.4f", cosDist)
 	if cosDist != 1.0 {
 		t.Errorf("Expected cosine distance 1.0, got %f", cosDist)
 	}
 
 	// L2 距离
 	l2Dist := vec1.Distance(vec2, L2Distance)
+	t.Logf("L2 Distance: %.4f", l2Dist)
 	expected := float32(1.414213) // sqrt(2)
 	if abs(l2Dist-expected) > 0.001 {
 		t.Errorf("Expected L2 distance ~1.414, got %f", l2Dist)
@@ -183,6 +218,9 @@ func TestVector_Normalize(t *testing.T) {
 	vec := Vector{3.0, 4.0} // 长度为 5
 	normalized := vec.Normalize()
 
+	t.Logf("Original: %v", vec)
+	t.Logf("Normalized: %v", normalized)
+
 	// 归一化后长度应该为 1
 	expected := Vector{0.6, 0.8}
 
@@ -195,6 +233,113 @@ func TestVector_Normalize(t *testing.T) {
 	}
 }
 
+// 测试向量插入
+func TestVector_Insert(t *testing.T) {
+	code := &CodeVector{
+		Content:   "func main() { fmt.Println(\"Hello\") }",
+		Embedding: Vector{0.1, 0.2, 0.3, 0.4},
+		Language:  "golang",
+		Layer:     "main",
+	}
+
+	sql, args := Of(code).
+		Insert(func(ib *InsertBuilder) {
+			ib.Set("content", code.Content).
+				Set("embedding", code.Embedding).
+				Set("language", code.Language).
+				Set("layer", code.Layer)
+		}).
+		Build().
+		SqlOfInsert()
+
+	t.Logf("=== INSERT 测试 ===")
+	t.Logf("SQL: %s", sql)
+	t.Logf("Args count: %d", len(args))
+	for i, arg := range args {
+		t.Logf("Args[%d] type: %T", i, arg)
+		t.Logf("Args[%d] value: %v", i, arg)
+	}
+
+	// 验证 SQL 包含所有字段
+	if !containsString(sql, "content") {
+		t.Errorf("Expected content field in SQL: %s", sql)
+	}
+	if !containsString(sql, "embedding") {
+		t.Errorf("Expected embedding field in SQL: %s", sql)
+	}
+	if !containsString(sql, "language") {
+		t.Errorf("Expected language field in SQL: %s", sql)
+	}
+}
+
+// 测试向量更新
+func TestVector_Update(t *testing.T) {
+	newEmbedding := Vector{0.5, 0.6, 0.7, 0.8}
+
+	sql, args := Of(&CodeVector{}).
+		Update(func(ub *UpdateBuilder) {
+			ub.Set("embedding", newEmbedding).
+				Set("language", "golang")
+		}).
+		Eq("id", 123).
+		Build().
+		SqlOfUpdate()
+
+	t.Logf("=== UPDATE 测试 ===")
+	t.Logf("SQL: %s", sql)
+	t.Logf("Args count: %d", len(args))
+	for i, arg := range args {
+		t.Logf("Args[%d] type: %T", i, arg)
+		t.Logf("Args[%d] value: %v", i, arg)
+	}
+
+	// 验证 SQL
+	if !containsString(sql, "UPDATE") {
+		t.Errorf("Expected UPDATE in SQL: %s", sql)
+	}
+	if !containsString(sql, "embedding") {
+		t.Errorf("Expected embedding field in SQL: %s", sql)
+	}
+	if !containsString(sql, "WHERE") {
+		t.Errorf("Expected WHERE clause in SQL: %s", sql)
+	}
+}
+
+// 测试向量类型在 Set() 中的处理
+func TestVector_SetBehavior(t *testing.T) {
+	vec := Vector{1.0, 2.0, 3.0}
+
+	// 测试 InsertBuilder.Set()
+	sql, args := Of(&CodeVector{}).
+		Insert(func(ib *InsertBuilder) {
+			ib.Set("embedding", vec)
+		}).
+		Build().
+		SqlOfInsert()
+
+	t.Logf("=== Vector Set() 行为测试 ===")
+	t.Logf("原始 Vector: %v (类型: %T)", vec, vec)
+	t.Logf("SQL: %s", sql)
+
+	if len(args) > 0 {
+		t.Logf("Set() 后 args[0] 类型: %T", args[0])
+		t.Logf("Set() 后 args[0] 值: %v", args[0])
+
+		// 关键检查：args[0] 是 Vector 还是 string？
+		switch args[0].(type) {
+		case Vector:
+			t.Logf("✅ args[0] 是 Vector 类型，driver.Valuer 会被调用")
+		case string:
+			t.Logf("⚠️ args[0] 是 string 类型，已被 JSON Marshal")
+			t.Logf("⚠️ driver.Valuer 不会被调用")
+		case []float32:
+			t.Logf("✅ args[0] 是 []float32 类型")
+		default:
+			t.Logf("❓ args[0] 是未知类型: %T", args[0])
+		}
+	}
+}
+
 // 辅助函数
 func containsString(s, substr string) bool {
 	return len(s) >= len(substr) && (s == substr || len(s) > len(substr) && findSubstring(s, substr))
@@ -215,4 +360,3 @@ func abs(x float32) float32 {
 	}
 	return x
 }
-