chromium/components/segmentation_platform/embedder/default_model/contextual_page_actions_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/contextual_page_actions_model.h"

#include "base/feature_list.h"
#include "base/task/sequenced_task_runner.h"
#include "components/segmentation_platform/internal/metadata/metadata_writer.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"

namespace segmentation_platform {

namespace {
SegmentId;

// Label input size
constexpr int kLabelInputSize = …;
// Default parameters for contextual page actions model.
constexpr SegmentId kSegmentId = …;
constexpr int64_t kOneDayInSeconds = …;
// Parameters for share action model.
constexpr std::array<MetadataWriter::UMAFeature, 6> kShareUMAFeatures = …;

constexpr std::array<const char*, kLabelInputSize>
    kContextualPageActionModelLabels = …;

}  // namespace

ContextualPageActionsModel::ContextualPageActionsModel()
    : … { … }

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

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

}  // namespace segmentation_platform