// 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.
#ifndef COMPONENTS_OPTIMIZATION_GUIDE_CORE_MODEL_EXECUTION_SUBSTITUTION_H_
#define COMPONENTS_OPTIMIZATION_GUIDE_CORE_MODEL_EXECUTION_SUBSTITUTION_H_

#include <optional>
#include <string>
#include <vector>

#include "components/optimization_guide/core/model_execution/multimodal_message.h"
#include "components/optimization_guide/proto/substitution.pb.h"
#include "services/on_device_model/public/mojom/on_device_model.mojom.h"

namespace optimization_guide {

struct SubstitutionResult {
  SubstitutionResult();
  ~SubstitutionResult();
  SubstitutionResult(SubstitutionResult&&);
  SubstitutionResult& operator=(SubstitutionResult&&);

  std::string ToString() const;

  on_device_model::mojom::InputPtr input;
  bool should_ignore_input_context;
};

std::optional<SubstitutionResult> CreateSubstitutions(
    MultimodalMessageReadView request,
    const google::protobuf::RepeatedPtrField<proto::SubstitutedString>&
        config_substitutions);

std::string OnDeviceInputToString(const on_device_model::mojom::Input& input);

}  // namespace optimization_guide

#endif  // COMPONENTS_OPTIMIZATION_GUIDE_CORE_MODEL_EXECUTION_SUBSTITUTION_H_