fix: bug fix for VectorSearchType.SimilarityScoreThreshold (#290)

* bug fix for VectorSearchType.SimilarityScoreThreshold initializing ScoreThreshold value * Update VectorStoreRetriever.cs --------- Co-authored-by: Konstantin S <[email protected]>
tryAGI · May 3, 2024 · a07e22c · a07e22c
1 parent 9e46ac5
commit a07e22c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/Core/src/Retrievers/VectorStoreRetriever.cs b/src/Core/src/Retrievers/VectorStoreRetriever.cs
@@ -36,14 +36,13 @@ public VectorStoreRetriever(
         float? scoreThreshold = null)
     {
         SearchType = searchType;
+        ScoreThreshold = scoreThreshold;
 
         if (SearchType == VectorSearchType.SimilarityScoreThreshold && ScoreThreshold == null)
             throw new ArgumentException($"ScoreThreshold required for {SearchType}");
 
         EmbeddingModel = embeddingModel;
         VectorCollection = vectorCollection;
-        SearchType = searchType;
-        ScoreThreshold = scoreThreshold;
     }
 
     /// <inheritdoc/>
@@ -71,4 +70,4 @@ public Task<IReadOnlyCollection<string>> AddDocumentsAsync(IReadOnlyCollection<D
     {
         return VectorCollection.AddDocumentsAsync(EmbeddingModel, documents);
     }
-}
+}