chromium/components/history_embeddings/ml_embedder.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/history_embeddings/ml_embedder.h"

#include "base/task/sequenced_task_runner.h"
#include "components/history_embeddings/passage_embeddings_service_controller.h"
#include "components/optimization_guide/core/optimization_guide_model_provider.h"

namespace history_embeddings {

MlEmbedder::MlEmbedder(
    optimization_guide::OptimizationGuideModelProvider* model_provider,
    PassageEmbeddingsServiceController* service_controller)
    : … { … }

MlEmbedder::~MlEmbedder() { … }

void MlEmbedder::ComputePassagesEmbeddings(
    PassageKind kind,
    std::vector<std::string> passages,
    ComputePassagesEmbeddingsCallback callback) { … }

void MlEmbedder::OnModelUpdated(
    optimization_guide::proto::OptimizationTarget optimization_target,
    base::optional_ref<const optimization_guide::ModelInfo> model_info) { … }

void MlEmbedder::SetOnEmbedderReady(OnEmbedderReadyCallback callback) { … }

}  // namespace history_embeddings