visual_query_suggestions_service.cc

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/companion/visual_query/visual_query_suggestions_service.h"

#include "base/files/file.h"
#include "base/files/file_util.h"
#include "base/metrics/histogram_macros_local.h"
#include "base/task/sequenced_task_runner.h"
#include "components/optimization_guide/core/optimization_guide_model_provider.h"
#include "components/optimization_guide/core/optimization_guide_util.h"
#include "components/optimization_guide/proto/models.pb.h"
#include "mojo/public/mojom/base/proto_wrapper.mojom.h"

namespace companion::visual_query {

namespace {

// Load the model file at the provided file path.
base::File LoadModelFile(const base::FilePath& model_file_path) { … }

// Close the provided model file.
void CloseModelFile(base::File model_file) { … }

// Extracts the model string value from the model metadata.
// The model string is expected to be a serialized string of the
// |EligibilitySpec| proto.
std::optional<mojo_base::ProtoWrapper> GetModelSpec(ModelMetadata& metadata) { … }

}  // namespace

VisualQuerySuggestionsService::VisualQuerySuggestionsService(
    optimization_guide::OptimizationGuideModelProvider* model_provider,
    const scoped_refptr<base::SequencedTaskRunner>& background_task_runner)
    : … { … }

VisualQuerySuggestionsService::~VisualQuerySuggestionsService() { … }

void VisualQuerySuggestionsService::BindModelReceiver(
    mojo::PendingReceiver<mojom::VisualSuggestionsModelProvider> receiver) { … }

void VisualQuerySuggestionsService::Shutdown() { … }

void VisualQuerySuggestionsService::UnloadModelFile() { … }

void VisualQuerySuggestionsService::NotifyModelUpdatesAndClear() { … }

void VisualQuerySuggestionsService::OnModelFileLoaded(base::File model_file) { … }

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

void VisualQuerySuggestionsService::RegisterModelUpdateCallback(
    ModelUpdateCallback callback) { … }

void VisualQuerySuggestionsService::GetModelWithMetadata(
    GetModelWithMetadataCallback callback) { … }

}  // namespace companion::visual_query
chromium/chrome/browser/companion/visual_query/visual_query_suggestions_service.cc
