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

#include <memory>
#include <string>
#include <utility>
#include <vector>

#include "base/check_op.h"
#include "base/logging.h"
#include "base/time/time.h"
#include "components/history_embeddings/vector_database.h"

namespace history_embeddings {

SchedulingEmbedder::Job::Job(PassageKind kind,
                             std::vector<std::string> passages,
                             ComputePassagesEmbeddingsCallback callback)
    : … { … }
SchedulingEmbedder::Job::~Job() = default;
SchedulingEmbedder::Job::Job(Job&&) = default;
SchedulingEmbedder::Job& SchedulingEmbedder::Job::operator=(Job&&) = default;

////////////////////////////////////////////////////////////////////////////////

SchedulingEmbedder::SchedulingEmbedder(std::unique_ptr<Embedder> embedder,
                                       size_t scheduled_max)
    : … { … }

SchedulingEmbedder::~SchedulingEmbedder() = default;

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

void SchedulingEmbedder::SubmitWorkToEmbedder() { … }

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

void SchedulingEmbedder::OnEmbedderReady(OnEmbedderReadyCallback callback,
                                         EmbedderMetadata metadata) { … }

void SchedulingEmbedder::OnEmbeddingsComputed(std::vector<std::string> passages,
                                              std::vector<Embedding> embeddings,
                                              ComputeEmbeddingsStatus status) { … }

}  // namespace history_embeddings