SVDB

Swift Vector數據庫（SVDB）

Swift應用程序的新快速本地啟動矢量數據庫。

專為那些建立下一代用戶體驗的人，只有在設備上的智能上才有可能。

本地的設備矢量數據庫僅僅是開始。

安裝

要使用Swift Package Manager安裝它，請使用XCode 11將其直接添加到您的項目中，或將其指定為package.swift文件中的依賴項：

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

用法

1。創建嵌入

 let document = "cat"

chatgpt：

我發現這個Swift Openai包是最好的

 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。添加文檔

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

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

3。搜尋

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

let results = animalCollection.search(query: dogEmedding)

演示

查看演示演示

托多

沒有把握。我想讓添加文檔並為您處理嵌入變得更容易。有什麼建議嗎？