Add VectorStore protocol

Author: intitni

Tool/Sources/LangChain/VectorStore/VectorStore.swift

@@ -0,0 +1,20 @@
+import Foundation
+
+protocol VectorStore {
+    func add(_ documents: [EmbeddedDocument]) async throws
+    func set(_ documents: [EmbeddedDocument]) async throws
+    func clear() async throws
+    func searchWithDistance(embeddings: [Float], count: Int) async throws
+        -> [(document: Document, distance: Float)]
+}
+
+extension VectorStore {
+    func search(embeddings: [Float], count: Int) async throws -> [Document] {
+        try await searchWithDistance(embeddings: embeddings, count: count).map { $0.document }
+    }
+
+    func add(_ document: EmbeddedDocument) async throws {
+        try await add([document])
+    }
+}
+