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

// 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/segmentation_platform/embedder/default_model/url_visit_resumption_ranker.h"

#include <memory>

#include "base/metrics/field_trial_params.h"
#include "base/rand_util.h"
#include "base/task/sequenced_task_runner.h"
#include "components/segmentation_platform/internal/metadata/metadata_writer.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 "components/visited_url_ranking/public/url_visit_schema.h"
#include "services/metrics/public/cpp/ukm_builders.h"

namespace segmentation_platform {

namespace {
SegmentId;

constexpr SegmentId kSegmentId = …;
constexpr uint64_t kRankerVersion = …;

constexpr std::array<MetadataWriter::UMAFeature, 1> kOutput = …;

}  // namespace

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

URLVisitResumptionRanker::URLVisitResumptionRanker()
    : … { … }

URLVisitResumptionRanker::~URLVisitResumptionRanker() = default;

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

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

}  // namespace segmentation_platform