chromium/components/history_embeddings/ml_embedder.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_HISTORY_EMBEDDINGS_ML_EMBEDDER_H_
#define COMPONENTS_HISTORY_EMBEDDINGS_ML_EMBEDDER_H_

#include "base/memory/raw_ptr.h"
#include "components/history_embeddings/embedder.h"
#include "components/optimization_guide/core/optimization_target_model_observer.h"

namespace optimization_guide {
class OptimizationGuideModelProvider;
}  // namespace optimization_guide

namespace history_embeddings {

class PassageEmbeddingsServiceController;

// An embedder that returns embeddings from a machine learning model.
class MlEmbedder : public Embedder,
                   public optimization_guide::OptimizationTargetModelObserver { … };

}  // namespace history_embeddings

#endif  // COMPONENTS_HISTORY_EMBEDDINGS_ML_EMBEDDER_H_