Skip to content
Prev Previous commit
Next Next commit
Additional adjustments to FieldBuilder and Search Index Customizations
  • Loading branch information
jairmyree committed Apr 23, 2024
commit 81d24431aeaa00c0f921c1a0f98b27aa24525dcf
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,12 @@
import com.azure.search.documents.indexes.FieldBuilderIgnore;
import com.azure.search.documents.indexes.SearchableField;
import com.azure.search.documents.indexes.SimpleField;
import com.azure.search.documents.indexes.models.*;
import com.azure.search.documents.indexes.models.FieldBuilderOptions;
import com.azure.search.documents.indexes.models.LexicalAnalyzerName;
import com.azure.search.documents.indexes.models.LexicalNormalizerName;
import com.azure.search.documents.indexes.models.SearchField;
import com.azure.search.documents.indexes.models.SearchFieldDataType;
import com.azure.search.documents.indexes.models.VectorEncodingFormat;
import reactor.util.annotation.Nullable;

import java.lang.annotation.Annotation;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@ public VectorizableImageBinaryQuery setKNearestNeighborsCount(Integer kNearestNe
/**
* {@inheritDoc}
*/
@Override
public VectorizableImageBinaryQuery setFields(String... fields) {
super.setFields(fields);
return this;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@ public VectorizableImageUrlQuery setKNearestNeighborsCount(Integer kNearestNeigh
/**
* {@inheritDoc}
*/
@Override
public VectorizableImageUrlQuery setFields(String... fields) {
super.setFields(fields);
return this;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -246,7 +246,6 @@ private void customizeSearchError(ClassCustomization classCustomization) {
}

private void customizeVectorizableImageUrlQuery(ClassCustomization classCustomization) {
classCustomization.getMethod("setFields").removeAnnotation("@Override");
customizeAst(classCustomization, clazz -> clazz.getMethodsByName("setFields").get(0)
.setParameters(new NodeList<>(new Parameter().setType("String").setName("fields").setVarArgs(true)))
.setBody(StaticJavaParser.parseBlock("{\n" +
Expand All @@ -256,7 +255,6 @@ private void customizeVectorizableImageUrlQuery(ClassCustomization classCustomiz
}

private void customizeVectorizableImageBinaryQuery(ClassCustomization classCustomization) {
classCustomization.getMethod("setFields").removeAnnotation("@Override");
customizeAst(classCustomization, clazz -> clazz.getMethodsByName("setFields").get(0)
.setParameters(new NodeList<>(new Parameter().setType("String").setName("fields").setVarArgs(true)))
.setBody(StaticJavaParser.parseBlock("{\n" +
Expand Down