Skip to content

Commit 7aaaee8

Browse files
committed
Update tests
1 parent 51d9aae commit 7aaaee8

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/action/filter/ShardBulkInferenceActionFilterTests.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
import org.elasticsearch.index.mapper.vectors.DenseVectorFieldMapper;
3737
import org.elasticsearch.index.shard.ShardId;
3838
import org.elasticsearch.inference.ChunkedInference;
39+
import org.elasticsearch.inference.ChunkingSettings;
3940
import org.elasticsearch.inference.InferenceService;
4041
import org.elasticsearch.inference.InferenceServiceRegistry;
4142
import org.elasticsearch.inference.Model;
@@ -77,6 +78,7 @@
7778
import static org.elasticsearch.xpack.inference.action.filter.ShardBulkInferenceActionFilter.getIndexRequestOrNull;
7879
import static org.elasticsearch.xpack.inference.mapper.SemanticTextField.getChunksFieldName;
7980
import static org.elasticsearch.xpack.inference.mapper.SemanticTextField.getOriginalTextFieldName;
81+
import static org.elasticsearch.xpack.inference.mapper.SemanticTextFieldTests.generateRandomChunkingSettings;
8082
import static org.elasticsearch.xpack.inference.mapper.SemanticTextFieldTests.randomChunkedInferenceEmbeddingSparse;
8183
import static org.elasticsearch.xpack.inference.mapper.SemanticTextFieldTests.randomSemanticText;
8284
import static org.elasticsearch.xpack.inference.mapper.SemanticTextFieldTests.randomSemanticTextInput;
@@ -380,7 +382,10 @@ public void testManyRandomDocs() throws Exception {
380382
for (int i = 0; i < numInferenceFields; i++) {
381383
String field = randomAlphaOfLengthBetween(5, 10);
382384
String inferenceId = randomFrom(inferenceModelMap.keySet());
383-
inferenceFieldMap.put(field, new InferenceFieldMetadata(field, inferenceId, new String[] { field }, null));
385+
Map<String, Object> chunkingSettingsMap = Optional.ofNullable(generateRandomChunkingSettings())
386+
.map(ChunkingSettings::asMap)
387+
.orElse(null);
388+
inferenceFieldMap.put(field, new InferenceFieldMetadata(field, inferenceId, new String[] { field }, chunkingSettingsMap));
384389
}
385390

386391
int numRequests = atLeast(100);

0 commit comments

Comments
 (0)