update go demos

This commit is contained in:
Alex Garcia 2024-07-16 23:11:26 -07:00
parent e507bc0230
commit 6eb2397537
6 changed files with 14 additions and 25 deletions

View file

@ -1,26 +1,13 @@
package main
import (
"bytes"
_ "embed"
"encoding/binary"
"log"
_ "github.com/asg017/sqlite-vec-ncruces-bindings"
sqlite_vec "github.com/asg017/sqlite-vec-go-bindings/ncruces"
"github.com/ncruces/go-sqlite3"
)
func serializeFloat32(vector []float32) ([]byte, error) {
buf := new(bytes.Buffer)
err := binary.Write(buf, binary.LittleEndian, vector)
if err != nil {
return nil, err
}
return buf.Bytes(), nil
}
func main() {
db, err := sqlite3.Open(":memory:")
if err != nil {
@ -56,7 +43,7 @@ func main() {
}
for id, values := range items {
v, err := serializeFloat32(values)
v, err := sqlite_vec.SerializeFloat32(values)
if err != nil {
log.Fatal(err)
}
@ -86,7 +73,7 @@ func main() {
log.Fatal(err)
}
query, err := serializeFloat32(q)
query, err := sqlite_vec.SerializeFloat32(q)
if err != nil {
log.Fatal(err)
}