Skip to content

Commit b112c2b

Browse files
committed
Revert "Add WithFieldName implementation to QueryBuilders (opensearch-project#1285)"
This reverts commit d42efb1.
1 parent 6f3aabb commit b112c2b

File tree

3 files changed

+3
-30
lines changed

3 files changed

+3
-30
lines changed

src/main/java/org/opensearch/neuralsearch/query/NeuralKNNQueryBuilder.java

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
import org.opensearch.core.xcontent.XContentBuilder;
1212
import org.opensearch.index.query.AbstractQueryBuilder;
1313
import org.opensearch.index.query.QueryBuilder;
14-
import org.opensearch.index.query.WithFieldName;
1514
import org.opensearch.index.query.QueryRewriteContext;
1615
import org.opensearch.index.query.QueryShardContext;
1716
import org.opensearch.knn.index.query.KNNQueryBuilder;
@@ -49,7 +48,7 @@
4948
*/
5049

5150
@Getter
52-
public class NeuralKNNQueryBuilder extends AbstractQueryBuilder<NeuralKNNQueryBuilder> implements WithFieldName {
51+
public class NeuralKNNQueryBuilder extends AbstractQueryBuilder<NeuralKNNQueryBuilder> {
5352
/**
5453
* The name of the query
5554
*/
@@ -84,7 +83,6 @@ public static Builder builder() {
8483
*
8584
* @return The field name used in the KNN query
8685
*/
87-
@Override
8886
public String fieldName() {
8987
return knnQueryBuilder.fieldName();
9088
}

src/main/java/org/opensearch/neuralsearch/query/NeuralQueryBuilder.java

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@
3838
import org.opensearch.index.mapper.NumberFieldMapper;
3939
import org.opensearch.index.query.AbstractQueryBuilder;
4040
import org.opensearch.index.query.QueryBuilder;
41-
import org.opensearch.index.query.WithFieldName;
4241
import org.opensearch.index.query.QueryRewriteContext;
4342
import org.opensearch.index.query.QueryShardContext;
4443
import org.opensearch.knn.index.query.parser.MethodParametersParser;
@@ -70,7 +69,7 @@
7069
@Accessors(chain = true, fluent = true)
7170
@NoArgsConstructor(access = AccessLevel.PRIVATE)
7271
@AllArgsConstructor(access = AccessLevel.PRIVATE)
73-
public class NeuralQueryBuilder extends AbstractQueryBuilder<NeuralQueryBuilder> implements ModelInferenceQueryBuilder, WithFieldName {
72+
public class NeuralQueryBuilder extends AbstractQueryBuilder<NeuralQueryBuilder> implements ModelInferenceQueryBuilder {
7473

7574
public static final String NAME = "neural";
7675

@@ -611,14 +610,4 @@ private static boolean validateKNNQueryType(Integer k, Float maxDistance, Float
611610
}
612611
return queryCount == 1;
613612
}
614-
615-
/**
616-
* Gets the field name that this query is searching against.
617-
*
618-
* @return The field name used in the Neural query
619-
*/
620-
@Override
621-
public String fieldName() {
622-
return this.fieldName;
623-
}
624613
}

src/main/java/org/opensearch/neuralsearch/query/NeuralSparseQueryBuilder.java

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@
4141
import org.opensearch.index.mapper.MappedFieldType;
4242
import org.opensearch.index.query.AbstractQueryBuilder;
4343
import org.opensearch.index.query.QueryBuilder;
44-
import org.opensearch.index.query.WithFieldName;
4544
import org.opensearch.index.query.QueryRewriteContext;
4645
import org.opensearch.index.query.QueryShardContext;
4746
import org.opensearch.neuralsearch.ml.MLCommonsClientAccessor;
@@ -70,10 +69,7 @@
7069
@Accessors(chain = true, fluent = true)
7170
@NoArgsConstructor
7271
@AllArgsConstructor
73-
public class NeuralSparseQueryBuilder extends AbstractQueryBuilder<NeuralSparseQueryBuilder>
74-
implements
75-
ModelInferenceQueryBuilder,
76-
WithFieldName {
72+
public class NeuralSparseQueryBuilder extends AbstractQueryBuilder<NeuralSparseQueryBuilder> implements ModelInferenceQueryBuilder {
7773
public static final String NAME = "neural_sparse";
7874
@VisibleForTesting
7975
static final ParseField QUERY_TEXT_FIELD = new ParseField("query_text");
@@ -544,14 +540,4 @@ private boolean shouldUseModelId() {
544540
public static float bytesToFloat(byte[] bytes) {
545541
return ByteBuffer.wrap(bytes).getFloat();
546542
}
547-
548-
/**
549-
* Gets the field name that this query is searching against.
550-
*
551-
* @return The field name used in the Neural Sparse query
552-
*/
553-
@Override
554-
public String fieldName() {
555-
return this.fieldName;
556-
}
557543
}

0 commit comments

Comments
 (0)