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

#include "base/task/sequenced_task_runner.h"
#include "components/history_embeddings/vector_database.h"

namespace history_embeddings {

namespace {

constexpr int64_t kModelVersion = …;
constexpr size_t kOutputSize = …;
constexpr size_t kMockPassageWordCount = …;

Embedding ComputeEmbeddingForPassage(const std::string& passage) { … }

std::vector<Embedding> ComputeEmbeddingsForPassages(
    const std::vector<std::string>& passages) { … }

}  // namespace

MockEmbedder::MockEmbedder() = default;
MockEmbedder::~MockEmbedder() = default;

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

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

}  // namespace history_embeddings