1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/omnibox/browser/search_scoring_signals_annotator.h"
#include "components/omnibox/browser/autocomplete_input.h"
#include "components/omnibox/browser/autocomplete_match.h"
#include "components/omnibox/browser/autocomplete_result.h"
void SearchScoringSignalsAnnotator::AnnotateResult(
const AutocompleteInput& input,
AutocompleteResult* result) {
for (auto& match : *result) {
// Skip ineligible matches
if (!match.IsMlSignalLoggingEligible()) {
continue;
}
// Initialize the scoring signals if needed.
if (!match.scoring_signals) {
match.scoring_signals = std::make_optional<ScoringSignals>();
}
if (!match.scoring_signals->has_search_suggest_relevance()) {
match.scoring_signals->set_search_suggest_relevance(0);
}
UpdateMatchTypeScoringSignals(match, input.text());
}
}
// static
void SearchScoringSignalsAnnotator::UpdateMatchTypeScoringSignals(
AutocompleteMatch& match,
const std::u16string& input_text) {
match.scoring_signals->set_is_search_suggest_entity(
match.type == AutocompleteMatchType::SEARCH_SUGGEST_ENTITY);
match.scoring_signals->set_is_verbatim(match.IsVerbatimType() ||
match.contents == input_text);
match.scoring_signals->set_is_navsuggest(
match.type == AutocompleteMatchType::NAVSUGGEST ||
match.type == AutocompleteMatchType::NAVSUGGEST_PERSONALIZED ||
match.type == AutocompleteMatchType::TILE_NAVSUGGEST ||
match.type == AutocompleteMatchType::TILE_MOST_VISITED_SITE);
match.scoring_signals->set_is_search_suggest_tail(
match.type == AutocompleteMatchType::SEARCH_SUGGEST_TAIL);
match.scoring_signals->set_is_answer_suggest(
match.answer_type != omnibox::ANSWER_TYPE_UNSPECIFIED);
match.scoring_signals->set_is_calculator_suggest(
match.type == AutocompleteMatchType::CALCULATOR);
}
|
