fix index.cpp

microsoft · Jan 19, 2024 · 70264f9 · 70264f9
1 parent a820f62
commit 70264f9
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/index.cpp b/src/index.cpp
@@ -2393,17 +2393,20 @@ std::pair<uint32_t, uint32_t> Index<T, TagT, LabelT>::search_with_filters(const
         if (pos == K)
             break;
     }
-    if (pos <= K && context.GetAllowLessThanKResults())
+    if (pos < K && context.GetAllowLessThanKResults())
     {
         context.SetState(State::Success);
+        context.UpdateResultReturned(pos);
     }
     else if(pos < K)
     {
         context.SetState(State::Failure);
+        context.UpdateResultReturned(pos);
     }
     else
     {
-        context.SetState(State::Failure);
+        context.SetState(State::Success);
+        context.UpdateResultReturned(K);
     }
 
     return retval;