chromium/components/optimization_guide/core/model_execution/test/fake_model_assets.h

// 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_TEST_FAKE_MODEL_ASSETS_H_
#define COMPONENTS_OPTIMIZATION_GUIDE_CORE_MODEL_EXECUTION_TEST_FAKE_MODEL_ASSETS_H_

#include <cstdint>
#include <memory>

#include "base/containers/flat_set.h"
#include "base/files/file_path.h"
#include "base/files/scoped_temp_dir.h"
#include "components/optimization_guide/core/model_execution/feature_keys.h"
#include "components/optimization_guide/core/model_execution/on_device_model_adaptation_loader.h"
#include "components/optimization_guide/core/model_info.h"
#include "components/optimization_guide/proto/on_device_model_execution_config.pb.h"
#include "components/optimization_guide/proto/text_safety_model_metadata.pb.h"
#include "services/on_device_model/public/cpp/model_assets.h"

namespace optimization_guide {

// Base model files and metadata suitable for a FakeOnDeviceModelService.
class FakeBaseModelAsset { … };

// Adaptation files and metadata suitable for a FakeOnDeviceModelService.
class FakeAdaptationAsset { … };

// Language model files and metadata suitable for a FakeOnDeviceModelService.
class FakeLanguageModelAsset { … };

// Safety model files and metadata suitable for a FakeOnDeviceModelService.
class FakeSafetyModelAsset { … };

}  // namespace optimization_guide

#endif  // COMPONENTS_OPTIMIZATION_GUIDE_CORE_MODEL_EXECUTION_TEST_FAKE_MODEL_ASSETS_H_