|
36 | 36 | import org.elasticsearch.index.mapper.vectors.DenseVectorFieldMapper; |
37 | 37 | import org.elasticsearch.index.shard.ShardId; |
38 | 38 | import org.elasticsearch.inference.ChunkedInference; |
| 39 | +import org.elasticsearch.inference.ChunkingSettings; |
39 | 40 | import org.elasticsearch.inference.InferenceService; |
40 | 41 | import org.elasticsearch.inference.InferenceServiceRegistry; |
41 | 42 | import org.elasticsearch.inference.Model; |
|
77 | 78 | import static org.elasticsearch.xpack.inference.action.filter.ShardBulkInferenceActionFilter.getIndexRequestOrNull; |
78 | 79 | import static org.elasticsearch.xpack.inference.mapper.SemanticTextField.getChunksFieldName; |
79 | 80 | import static org.elasticsearch.xpack.inference.mapper.SemanticTextField.getOriginalTextFieldName; |
| 81 | +import static org.elasticsearch.xpack.inference.mapper.SemanticTextFieldTests.generateRandomChunkingSettings; |
80 | 82 | import static org.elasticsearch.xpack.inference.mapper.SemanticTextFieldTests.randomChunkedInferenceEmbeddingSparse; |
81 | 83 | import static org.elasticsearch.xpack.inference.mapper.SemanticTextFieldTests.randomSemanticText; |
82 | 84 | import static org.elasticsearch.xpack.inference.mapper.SemanticTextFieldTests.randomSemanticTextInput; |
@@ -380,7 +382,10 @@ public void testManyRandomDocs() throws Exception { |
380 | 382 | for (int i = 0; i < numInferenceFields; i++) { |
381 | 383 | String field = randomAlphaOfLengthBetween(5, 10); |
382 | 384 | 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)); |
384 | 389 | } |
385 | 390 |
|
386 | 391 | int numRequests = atLeast(100); |
|
0 commit comments