chromium/components/segmentation_platform/embedder/default_model/search_user_model.cc

// Copyright 2022 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/segmentation_platform/embedder/default_model/search_user_model.h"

#include <array>

#include "base/feature_list.h"
#include "base/metrics/field_trial_params.h"
#include "base/task/sequenced_task_runner.h"
#include "components/segmentation_platform/internal/database/ukm_types.h"
#include "components/segmentation_platform/internal/metadata/metadata_writer.h"
#include "components/segmentation_platform/public/config.h"
#include "components/segmentation_platform/public/constants.h"
#include "components/segmentation_platform/public/features.h"
#include "components/segmentation_platform/public/model_provider.h"
#include "components/segmentation_platform/public/proto/model_metadata.pb.h"
#include "services/metrics/public/cpp/ukm_builders.h"

namespace segmentation_platform {

namespace {
SegmentId;

// Default parameters for search user model.
constexpr uint64_t kSearchUserModelVersion = …;
constexpr SegmentId kSearchUserSegmentId = …;
constexpr int64_t kSearchUserSignalStorageLength = …;
constexpr int64_t kSearchUserMinSignalCollectionLength = …;

// Reference to the UMA ClientSummarizedResultType enum for Search.
constexpr std::array<int32_t, 1> kOnlySearch{ … };

// InputFeatures.
constexpr std::array<MetadataWriter::UMAFeature, 1> kSearchUserUMAFeatures = …;

#if BUILDFLAG(IS_IOS)
constexpr UkmEventHash kPageLoadHash = UkmEventHash::FromUnsafeValue(
    ukm::builders::MainFrameNavigation::kEntryNameHash);
constexpr UkmMetricHash kNavMetricHash = UkmMetricHash::FromUnsafeValue(
    ukm::builders::MainFrameNavigation::kDidCommitNameHash);
#elif !BUILDFLAG(IS_CHROMEOS)
constexpr UkmEventHash kPageLoadHash = …;
constexpr UkmMetricHash kNavMetricHash = …;
#endif

std::unique_ptr<DefaultModelProvider> GetSearchUserDefaultModel() { … }

}  // namespace

// static
std::unique_ptr<Config> SearchUserModel::GetConfig() { … }

SearchUserModel::SearchUserModel()
    : … { … }

std::unique_ptr<DefaultModelProvider::ModelConfig>
SearchUserModel::GetModelConfig() { … }

void SearchUserModel::ExecuteModelWithInput(
    const ModelProvider::Request& inputs,
    ExecutionCallback callback) { … }

}  // namespace segmentation_platform