"\"Update Package.swift, README.md, Distilbert.swift, and SimilarityS…

…earchKit.swift: Fix Distilbert embedding and vector store loading/saving.\""

Package.swift

@@ -42,7 +42,7 @@ let package = Package(
                 .product(name: "SwiftSoup", package: "SwiftSoup"),
                 .product(name: "SwiftFileStore", package: "swift-filestore"),
                 .product(name: "SimilaritySearchKit", package: "similarity-search-kit", condition: .when(platforms: [.macOS, .iOS, .visionOS])),
-//                .product(name: "SimilaritySearchKitDistilbert", package: "similarity-search-kit", condition: .when(platforms: [.macOS, .iOS, .visionOS])),
+                .product(name: "SimilaritySearchKitDistilbert", package: "similarity-search-kit", condition: .when(platforms: [.macOS, .iOS, .visionOS])),
                 .product(name: "GoogleGenerativeAI", package: "generative-ai-swift"),
                 .product(name: "SwiftyNotion", package: "SwiftyNotion"),
                 .product(name: "FeedKit", package: "FeedKit"),

README.md

@@ -402,7 +402,7 @@ Task(priority: .background)  {
     - [x] FileStore
 - Embedding
     - [x] OpenAI
-    - [ ] Distilbert
+    - [x] Distilbert
 - Chain
     - [x] Base
     - [x] LLM

Sources/LangChain/embeddings/Distilbert.swift

@@ -1,21 +1,22 @@
-////
-////  File.swift
-////  
-////
-////  Created by 顾艳华 on 2/11/24.
-////
 //
-//import Foundation
-//import SimilaritySearchKitDistilbert
+//  File.swift
+//  
 //
-//public struct Distilbert: Embeddings {
-//    let n = DistilbertEmbeddings()
-//    public init() {
-//        
-//    }
-//    
-//    
-//    public func embedQuery(text: String) async -> [Float] {
-//        await n.encode(sentence: text)!
-//    }
-//}
+//  Created by 顾艳华 on 2/11/24.
+//
+
+import Foundation
+import SimilaritySearchKitDistilbert
+
+@available(macOS 13.0, *)
+public struct Distilbert: Embeddings {
+    let n = DistilbertEmbeddings()
+    public init() {
+
+    }
+
+
+    public func embedQuery(text: String) async -> [Float] {
+        await n.encode(sentence: text)!
+    }
+}

Sources/LangChain/vectorstores/SimilaritySearchKit.swift

@@ -47,7 +47,7 @@ public class SimilaritySearchKit: VectorStore {
         )
         if #available(macOS 13.0, *) {
             if autoLoad {
-                try? vs.loadIndex()
+                let _ = try? vs.loadIndex()
             }
         } else {
             // Fallback on earlier versions
@@ -64,7 +64,7 @@ public class SimilaritySearchKit: VectorStore {
 
     @available(macOS 13.0, *)
     public func writeToFile() {
-        try? vs.saveIndex()
+        let _ = try? vs.saveIndex()
     }
 }
 #endif