Examples with CompletionSuggestionBuilder org.elasticsearch.search.suggest.completion.CompletionSuggestionBuilder used on opensource projects

Example 26 with CompletionSuggestionBuilder

use of org.elasticsearch.search.suggest.completion.CompletionSuggestionBuilder in project elasticsearch by elastic.

the class CompletionSuggestSearchIT method testEarlyTermination.

public void testEarlyTermination() throws Exception {
    final CompletionMappingBuilder mapping = new CompletionMappingBuilder();
    createIndexAndMapping(mapping);
    int numDocs = atLeast(100);
    List<IndexRequestBuilder> indexRequestBuilders = new ArrayList<>();
    for (int i = 0; i < numDocs; i++) {
        indexRequestBuilders.add(client().prepareIndex(INDEX, TYPE, "" + i).setSource(jsonBuilder().startObject().startObject(FIELD).field("input", "suggestion" + (numDocs - i)).field("weight", numDocs - i).endObject().endObject()));
    }
    indexRandom(true, indexRequestBuilders);
    int size = randomIntBetween(3, 10);
    String[] outputs = new String[size];
    for (int i = 0; i < size; i++) {
        outputs[i] = "suggestion" + (numDocs - i);
    }
    CompletionSuggestionBuilder prefix = SuggestBuilders.completionSuggestion(FIELD).prefix("sug").size(size);
    assertSuggestions("foo", prefix, outputs);
    CompletionSuggestionBuilder regex = SuggestBuilders.completionSuggestion(FIELD).regex("su[g|s]g").size(size);
    assertSuggestions("foo", regex, outputs);
    CompletionSuggestionBuilder fuzzyPrefix = SuggestBuilders.completionSuggestion(FIELD).prefix("sugg", Fuzziness.ONE).size(size);
    assertSuggestions("foo", fuzzyPrefix, outputs);
}

Example 27 with CompletionSuggestionBuilder

use of org.elasticsearch.search.suggest.completion.CompletionSuggestionBuilder in project elasticsearch by elastic.

the class CompletionSuggestSearchIT method testThatFuzzySuggesterIsUnicodeAware.

public void testThatFuzzySuggesterIsUnicodeAware() throws Exception {
    createIndexAndMapping(completionMappingBuilder);
    client().prepareIndex(INDEX, TYPE, "1").setSource(jsonBuilder().startObject().startObject(FIELD).startArray("input").value("ööööö").endArray().endObject().endObject()).get();
    refresh();
    // suggestion with a character, which needs unicode awareness
    org.elasticsearch.search.suggest.completion.CompletionSuggestionBuilder completionSuggestionBuilder = SuggestBuilders.completionSuggestion(FIELD).prefix("öööи", FuzzyOptions.builder().setUnicodeAware(true).build()).size(10);
    SearchResponse searchResponse = client().prepareSearch(INDEX).suggest(new SuggestBuilder().addSuggestion("foo", completionSuggestionBuilder)).execute().actionGet();
    assertSuggestions(searchResponse, false, "foo", "ööööö");
    // removing unicode awareness leads to no result
    completionSuggestionBuilder = SuggestBuilders.completionSuggestion(FIELD).prefix("öööи", FuzzyOptions.builder().setUnicodeAware(false).build()).size(10);
    searchResponse = client().prepareSearch(INDEX).suggest(new SuggestBuilder().addSuggestion("foo", completionSuggestionBuilder)).execute().actionGet();
    assertSuggestions(searchResponse, false, "foo");
    // increasing edit distance instead of unicode awareness works again, as this is only a single character
    completionSuggestionBuilder = SuggestBuilders.completionSuggestion(FIELD).prefix("öööи", FuzzyOptions.builder().setUnicodeAware(false).setFuzziness(Fuzziness.TWO).build()).size(10);
    searchResponse = client().prepareSearch(INDEX).suggest(new SuggestBuilder().addSuggestion("foo", completionSuggestionBuilder)).execute().actionGet();
    assertSuggestions(searchResponse, false, "foo", "ööööö");
}

Example 28 with CompletionSuggestionBuilder

use of org.elasticsearch.search.suggest.completion.CompletionSuggestionBuilder in project elasticsearch by elastic.

the class CompletionSuggestSearchIT method assertSuggestions.

public void assertSuggestions(String suggestion, String... suggestions) {
    String suggestionName = RandomStrings.randomAsciiOfLength(random(), 10);
    CompletionSuggestionBuilder suggestionBuilder = SuggestBuilders.completionSuggestion(FIELD).text(suggestion).size(10);
    assertSuggestions(suggestionName, suggestionBuilder, suggestions);
}