apunkt/sqlite-vec
1
0
Fork 0
mirror of https://github.com/asg017/sqlite-vec.git synced 2026-04-25 00:36:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

cleanup example

Browse source
This commit is contained in:
Alex Garcia 2024-07-12 16:59:55 -07:00
parent d300718726
commit b57a05e2e8
2 changed files with 17 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
examples/simple-go-ncruces/Makefile
View file

@ -1,6 +1,13 @@
demo: demo.go VERSION=0.0.1-alpha.27
demo: demo.go sqlite3.vec.wasm
go build -o $@ $< go build -o $@ $<
sqlite3.vec.wasm: sqlite3.vec.wasm:
exit 1 curl -L -o $@ https://github.com/asg017/sqlite-vec/releases/download/v$(VERSION)/sqlite-vec-$(VERSION)-ncruces-go.wasm
clean:
rm demo sqlite3.vec.wasm
.PHONY: clean

13
examples/simple-go-ncruces/demo.go
View file

@ -10,6 +10,9 @@ import (
"github.com/ncruces/go-sqlite3" "github.com/ncruces/go-sqlite3"
) )
//go:embed sqlite3.vec.wasm
var sqliteWithVecWasm []byte
func serializeFloat32(vector []float32) ([]byte, error) { func serializeFloat32(vector []float32) ([]byte, error) {
buf := new(bytes.Buffer) buf := new(bytes.Buffer)
err := binary.Write(buf, binary.LittleEndian, vector) err := binary.Write(buf, binary.LittleEndian, vector)
@ -19,15 +22,11 @@ func serializeFloat32(vector []float32) ([]byte, error) {
return buf.Bytes(), nil return buf.Bytes(), nil
} }
const memory = ":memory:"
//go:embed sqlite3.vec.wasm
var sqliteWithVecWasm []byte
func main() { func main() {
sqlite3.Binary = sqliteWithVecWasm sqlite3.Binary = sqliteWithVecWasm
db, err := sqlite3.Open(memory) db, err := sqlite3.Open(":memory:")
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
@ -37,6 +36,10 @@ func main() {
log.Fatal(err) log.Fatal(err)
} }
stmt.Step()
fmt.Printf("sqlite_version=%s, vec_version=%s\n", stmt.ColumnText(0), stmt.ColumnText(1))
err = db.Exec("CREATE VIRTUAL TABLE vec_items USING vec0(embedding float[4])") err = db.Exec("CREATE VIRTUAL TABLE vec_items USING vec0(embedding float[4])")
if err != nil { if err != nil {