HSEARCH-5245 Always add index.knn for OpenSearch indexes with vector …

…fields
hibernate · Sep 20, 2024 · 1d6fd4c · 1d6fd4c
1 parent 8fd594c
commit 1d6fd4c
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/...backend/elasticsearch/types/dsl/impl/AbstractElasticsearchVectorFieldTypeOptionsStep.java b/...backend/elasticsearch/types/dsl/impl/AbstractElasticsearchVectorFieldTypeOptionsStep.java
@@ -106,7 +106,6 @@ public ElasticsearchIndexValueFieldType<F> toIndexFieldType() {
 		builder.codec( codec );
 		if ( resolvedSearchable ) {
 			builder.searchable( true );
-			mappingContributor.contribute( builder, this );
 			builder.queryElementFactory( PredicateTypeKeys.EXISTS, new ElasticsearchExistsPredicate.Factory<>() );
 		}
 
@@ -117,6 +116,9 @@ public ElasticsearchIndexValueFieldType<F> toIndexFieldType() {
 
 		builder.multivaluable( false );
 
+		// NOTE: we make additional contribution at the end of the configuration to make sure that
+		//  the context we pass (this) is fully configured:
+		mappingContributor.contribute( builder, this );
 		return builder.build();
 	}
 

diff --git a/...lasticsearch/types/mapping/impl/AbstractOpenSearch2VectorFieldTypeMappingContributor.java b/...lasticsearch/types/mapping/impl/AbstractOpenSearch2VectorFieldTypeMappingContributor.java
@@ -64,9 +64,12 @@ public final <F> void contribute(ElasticsearchIndexValueFieldType.Builder<F> bui
 		SearchQueryElementFactory<? extends KnnPredicateBuilder,
 				ElasticsearchSearchIndexScope<?>,
 				ElasticsearchSearchIndexValueFieldContext<F>> factory = getKnnPredicateFactory( builder );
-		builder.queryElementFactory( PredicateTypeKeys.KNN, factory );
-
 		builder.contributeAdditionalIndexSettings( settings -> settings.addKnn( true ) );
+
+		// only add a predicate if the field on which this contribution happens is searchable:
+		if ( context.searchable() ) {
+			builder.queryElementFactory( PredicateTypeKeys.KNN, factory );
+		}
 	}
 
 	protected abstract <F> SearchQueryElementFactory<? extends KnnPredicateBuilder,

diff --git a/...d/elasticsearch/types/mapping/impl/Elasticsearch812VectorFieldTypeMappingContributor.java b/...d/elasticsearch/types/mapping/impl/Elasticsearch812VectorFieldTypeMappingContributor.java
@@ -43,8 +43,10 @@ protected boolean indexOptionAddCondition(Context context) {
 
 	@Override
 	public <F> void contribute(ElasticsearchIndexValueFieldType.Builder<F> builder, Context context) {
-		builder.queryElementFactory( PredicateTypeKeys.KNN,
-				new ElasticsearchKnnPredicate.Elasticsearch812Factory<>( builder.codec() ) );
+		if ( context.searchable() ) {
+			builder.queryElementFactory( PredicateTypeKeys.KNN,
+					new ElasticsearchKnnPredicate.Elasticsearch812Factory<>( builder.codec() ) );
+		}
 	}
 
 	private static String resolveDefault(VectorSimilarity vectorSimilarity) {