Update

Author: intitni

Tool/Sources/LangChain/VectorStore/TemporaryUSearch.swift

@@ -18,11 +18,11 @@ public actor TemporaryUSearch: VectorStore {
     let index: USearchIndex
     var documents: [USearchLabel: LabeledDocument] = [:]
 
-    public init(identifier: String) {
+    public init(identifier: String, dimensions: Int = 1536 /* text-embedding-ada-002 */ ) {
         self.identifier = calculateMD5Hash(identifier)
         index = .init(
             metric: .IP,
-            dimensions: 1536, // text-embedding-ada-002
+            dimensions: UInt32(dimensions),
             connectivity: 16,
             quantization: .F32
         )

Tool/Sources/OpenAIService/Configuration/EmbeddingConfiguration.swift

@@ -7,6 +7,7 @@ public protocol EmbeddingConfiguration {
     var model: EmbeddingModel { get }
     var apiKey: String { get }
     var maxToken: Int { get }
+    var dimensions: Int { get }
 }
 
 public extension EmbeddingConfiguration {

Tool/Sources/OpenAIService/Configuration/UserPreferenceEmbeddingConfiguration.swift

@@ -14,6 +14,11 @@ public struct UserPreferenceEmbeddingConfiguration: EmbeddingConfiguration {
         model.info.maxTokens
     }
 
+    #warning("TODO: Support different dimensions.")
+    public var dimensions: Int {
+        1536 // text-embedding-ada-002
+    }
+
     public init() {}
 }
 
@@ -24,15 +29,18 @@ public class OverridingEmbeddingConfiguration<
         public var modelId: String?
         public var model: EmbeddingModel?
         public var maxTokens: Int?
+        public var dimensions: Int?
 
         public init(
             modelId: String? = nil,
             model: EmbeddingModel? = nil,
-            maxTokens: Int? = nil
+            maxTokens: Int? = nil,
+            dimensions: Int? = nil
         ) {
             self.modelId = modelId
             self.model = model
             self.maxTokens = maxTokens
+            self.dimensions = dimensions
         }
     }
 
@@ -56,5 +64,9 @@ public class OverridingEmbeddingConfiguration<
     public var maxToken: Int {
         overriding.maxTokens ?? configuration.maxToken
     }
+    
+    public var dimensions: Int {
+        overriding.dimensions ?? configuration.dimensions
+    }
 }