mirror of
https://github.com/asg017/sqlite-vec.git
synced 2026-04-25 00:36:56 +02:00
cleanup example
This commit is contained in:
parent
d300718726
commit
b57a05e2e8
2 changed files with 17 additions and 7 deletions
|
|
@ -1,6 +1,13 @@
|
|||
|
||||
demo: demo.go
|
||||
VERSION=0.0.1-alpha.27
|
||||
|
||||
demo: demo.go sqlite3.vec.wasm
|
||||
go build -o $@ $<
|
||||
|
||||
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
|
||||
|
|
|
|||
|
|
@ -10,6 +10,9 @@ import (
|
|||
"github.com/ncruces/go-sqlite3"
|
||||
)
|
||||
|
||||
//go:embed sqlite3.vec.wasm
|
||||
var sqliteWithVecWasm []byte
|
||||
|
||||
func serializeFloat32(vector []float32) ([]byte, error) {
|
||||
buf := new(bytes.Buffer)
|
||||
err := binary.Write(buf, binary.LittleEndian, vector)
|
||||
|
|
@ -19,15 +22,11 @@ func serializeFloat32(vector []float32) ([]byte, error) {
|
|||
return buf.Bytes(), nil
|
||||
}
|
||||
|
||||
const memory = ":memory:"
|
||||
|
||||
//go:embed sqlite3.vec.wasm
|
||||
var sqliteWithVecWasm []byte
|
||||
|
||||
func main() {
|
||||
sqlite3.Binary = sqliteWithVecWasm
|
||||
|
||||
db, err := sqlite3.Open(memory)
|
||||
db, err := sqlite3.Open(":memory:")
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
|
@ -37,6 +36,10 @@ func main() {
|
|||
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])")
|
||||
if err != nil {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue