vector-db

A simple and efficient vector database written in Go.

Features

• Fast vector indexing and search
• Embedding generation using fastembed
• Flat-index search with cosine similarity

Getting Started

Prerequisites

• Go 1.20+
• ONNX Runtime (see below)

Installation

Clone the repository:

git clone https://github.com/sharath-09/vector-db.git
cd vector-db

Install dependencies:

go run main.go

Embeddings generated using BGEBaseENV15.

Setting up ONNX for Embedding Generation

Download a ONNX runtime file from onnxruntime_go_examples, and set the ONNX_PATH environment variable to the .dylib file for mac, or .so for linux

Testing

Run all tests:

go test ./...

References

• fastembed-go
• vector-db distanceMeasure.go

License

MIT License. See LICENSE for