SVDB

Swift Vector Database (SVDB)

Database vektor on-perangkat lokal baru yang cepat untuk aplikasi Swift.

Dibangun untuk mereka yang membangun pengalaman pengguna generasi berikutnya hanya mungkin dengan kecerdasan di perangkat.

Database vektor di perangkat lokal hanyalah permulaan.

Instalasi

Untuk menginstalnya menggunakan Swift Package Manager, baik secara langsung menambahkannya ke proyek Anda menggunakan XCode 11, atau tentukan sebagai ketergantungan dalam file package.swift:

 // ...
dependencies: [
    .package(url: "https://github.com/Dripfarm/SVDB.git", from: "1.0.0"),
],
//...

Penggunaan

1. Buat embeddings

 let document = "cat"

Chatgpt:

Saya menemukan paket Openai Swift ini menjadi yang terbaik

 import OpenAI

func embed(text: String) async -> [Double]? {
	let query = EmbeddingsQuery(model: .textEmbeddingAda, input: text)

	let result = try! await openAI.embeddings(query: query)

	return result.data.first?.embedding
}

let wordEmbedding = embed(text: document)

Nlembeddings

 import NaturalLanguage

let embedding: NLEmbedding? = NLEmbedding.wordEmbedding(for: .english)

let wordEmedding = embedding?.vector(for: document) //returns double array

2. Tambahkan dokumen

 let animalCollection = SVDB.shared.collection("animals")

SVDB.shared.addDocument(text: document, embedding: wordEmbedding)

3. Cari

 let dogEmedding = embedding?.vector(for: "dog")

let results = animalCollection.search(query: dogEmedding)

Demo

Lihat demo demo

Todo

Tidak yakin. Saya ingin memudahkan untuk menambahkan dokumen dan mengurus embeddings untuk Anda. Ada saran?