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)

演示

查看演示演示

托多

没有把握。我想让添加文档并为您处理嵌入变得更容易。有什么建议吗？