chromium/components/optimization_guide/core/model_execution/test/feature_config_builder.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/optimization_guide/core/model_execution/test/feature_config_builder.h"

#include <initializer_list>

#include "components/optimization_guide/core/model_execution/feature_keys.h"
#include "components/optimization_guide/proto/descriptors.pb.h"
#include "components/optimization_guide/proto/features/compose.pb.h"
#include "components/optimization_guide/proto/model_execution.pb.h"
#include "components/optimization_guide/proto/on_device_model_execution_config.pb.h"
#include "components/optimization_guide/proto/text_safety_model_metadata.pb.h"

namespace optimization_guide {

proto::SafetyCategoryThreshold ForbidUnsafe() { … }

proto::SafetyCategoryThreshold RequireReasonable() { … }

proto::ProtoField ProtoField(std::initializer_list<int32_t> tags) { … }

proto::ProtoField PageUrlField() { … }

proto::ProtoField UserInputField() { … }

proto::ProtoField PreviousResponseField() { … }

proto::ProtoField OutputField() { … }

proto::ProtoField StringValueField() { … }

proto::SubstitutedString FieldSubstitution(const std::string& tmpl,
                                           proto::ProtoField&& field) { … }

proto::SubstitutedString PageUrlSubstitution() { … }

proto::RedactRules SimpleRedactRule(const std::string& regex,
                                    proto::RedactBehavior behavior,
                                    std::optional<std::string> replacement) { … }

proto::OnDeviceModelExecutionFeatureConfig SimpleComposeConfig() { … }

proto::FeatureTextSafetyConfiguration ComposeSafetyConfig() { … }

proto::TextSafetyModelMetadata SafetyMetadata(
    std::initializer_list<proto::FeatureTextSafetyConfiguration> configs) { … }

}  // namespace optimization_guide