-
Notifications
You must be signed in to change notification settings - Fork 72
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support k-NN radial search parameters in neural search #697
Support k-NN radial search parameters in neural search #697
Conversation
cf2d2ec
to
2cd766b
Compare
src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java
Outdated
Show resolved
Hide resolved
src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java
Outdated
Show resolved
Hide resolved
src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java
Outdated
Show resolved
Hide resolved
src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java
Outdated
Show resolved
Hide resolved
src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java
Outdated
Show resolved
Hide resolved
9484101
to
ac828d9
Compare
src/test/java/org/opensearch/neuralsearch/query/NeuralQueryIT.java
Outdated
Show resolved
Hide resolved
@junqiu-lei please attach the link of the benchmarks you have done on k-NN for radial search in this PR description. |
Yes, added in PR description. |
src/test/java/org/opensearch/neuralsearch/query/NeuralQueryIT.java
Outdated
Show resolved
Hide resolved
@junqiu-lei where will you be adding bwc tests? In K-nn or neural search? |
I think you can add one bwc test in the PR to ensure this does not break earlier versions. |
@junqiu-lei can we add a BWC test here. |
src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java
Outdated
Show resolved
Hide resolved
src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java
Outdated
Show resolved
Hide resolved
src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java
Outdated
Show resolved
Hide resolved
bwc test added @navneet1v @vibrantvarun |
Looks good to me , Will wait @navneet1v comments to get resolved before approving. |
Hey @junqiu-lei BWC tests are failing because of your changes. I think you have to add a check to skip running versions bwc for this tests for particular versions. neural-search/qa/restart-upgrade/build.gradle Line 136 in f5a47a6
|
7b0a56d
to
545178f
Compare
Signed-off-by: Junqiu Lei <[email protected]>
Signed-off-by: Junqiu Lei <[email protected]>
Signed-off-by: Junqiu Lei <[email protected]>
Signed-off-by: Junqiu Lei <[email protected]>
Signed-off-by: Junqiu Lei <[email protected]>
17c2e68
to
c09945f
Compare
Signed-off-by: Junqiu Lei <[email protected]>
c09945f
to
2dae4cf
Compare
The backport to
To backport manually, run these commands in your terminal: # Fetch latest updates from GitHub
git fetch
# Create a new working tree
git worktree add .worktrees/backport-2.x 2.x
# Navigate to the new working tree
cd .worktrees/backport-2.x
# Create a new branch
git switch --create backport/backport-697-to-2.x
# Cherry-pick the merged commit of this pull request and resolve the conflicts
git cherry-pick -x --mainline 1 68cdbc90c4fabd87ec8fecca4bdd753a55994107
# Push it to GitHub
git push --set-upstream origin backport/backport-697-to-2.x
# Go back to the original working tree
cd ../..
# Delete the working tree
git worktree remove .worktrees/backport-2.x Then, create a pull request where the |
…oject#697) * Support k-NN radial search parameters in neural search Signed-off-by: Junqiu Lei <[email protected]> (cherry picked from commit 68cdbc9) Signed-off-by: Junqiu Lei <[email protected]>
…oject#697) Signed-off-by: Junqiu Lei <[email protected]> (cherry picked from commit 68cdbc9) Signed-off-by: Junqiu Lei <[email protected]>
(cherry picked from commit 68cdbc9) Signed-off-by: Junqiu Lei <[email protected]>
Description
Since radial search is supported in k-NN plugin recently, this PR allows neural search to have the ability to use radial search parameters, so finally it can be allowed to use one of
k
,min_score
,max_distance
for vector search.Issues Resolved
Part of opensearch-project/k-NN#814
Benchmark in k-NN plugin
opensearch-project/k-NN#814 (comment)
Check List
By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.
For more information on following Developer Certificate of Origin and signing off your commits, please check here.