// 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/passage_embeddings/core/passage_embeddings_features.h"

#include "base/feature_list.h"
#include "base/metrics/field_trial_params.h"
#include "base/time/time.h"

namespace passage_embeddings {

namespace {

constexpr auto enabled_by_default_desktop_only =
#if BUILDFLAG(IS_ANDROID) || BUILDFLAG(IS_IOS)
    base::FEATURE_DISABLED_BY_DEFAULT;
#else
    base::FEATURE_ENABLED_BY_DEFAULT;
#endif

}  // namespace

BASE_FEATURE(kPassageEmbedder, enabled_by_default_desktop_only);

BASE_FEATURE(kPDFEmbeddingsGeneration, enabled_by_default_desktop_only);

#if BUILDFLAG(IS_ANDROID)
const base::FeatureParam<int> kPassageEmbedderMinRequiredRamMb(
    &kPassageEmbedder,
    "PassageEmbedderMinRequiredRamMb",
    0);
#endif  // BUILDFLAG(IS_ANDROID)

const base::FeatureParam<int> kUserInitiatedPriorityNumThreads(
    &kPassageEmbedder,
    "UserInitiatedPriorityNumThreads",
    4);

const base::FeatureParam<int>
    kUrgentPriorityNumThreads(&kPassageEmbedder, "UrgentPriorityNumThreads", 4);

const base::FeatureParam<int> kPassivePriorityNumThreads(
    &kPassageEmbedder,
    "PassivePriorityNumThreads",
    1);

const base::FeatureParam<int> kEmbedderCacheSize(&kPassageEmbedder,
                                                 "EmbedderCacheSize",
                                                 1000);

const base::FeatureParam<base::TimeDelta> kEmbedderTimeout(&kPassageEmbedder,
                                                           "EmbedderTimeout",
                                                           base::Seconds(60));

const base::FeatureParam<base::TimeDelta> kEmbeddingsServiceTimeout(
    &kPassageEmbedder,
    "EmbeddingsServiceTimeout",
    base::Seconds(60));

const base::FeatureParam<base::TimeDelta> kPassageExtractionDelay(
    &kPassageEmbedder,
    "PassageExtractionDelay",
    base::Seconds(5));

const base::FeatureParam<size_t> kMaxWordsPerAggregatePassage(
    &kPassageEmbedder,
    "MaxWordsPerAggregatePassage",
    100);

const base::FeatureParam<size_t> kMaxPassagesPerPage(&kPassageEmbedder,
                                                     "MaxPassagesPerPage",
                                                     10);

const base::FeatureParam<size_t> kMinWordsPerPassage(&kPassageEmbedder,
                                                     "MinWordsPerPassage",
                                                     5);

const base::FeatureParam<size_t> kMaxPassagesFromPDF(&kPDFEmbeddingsGeneration,
                                                     "MaxPassagesFromPDF",
                                                     10);

const base::FeatureParam<bool> kAllowGpuExecution(&kPassageEmbedder,
                                                  "AllowGpuExecution",
                                                  false);

const base::FeatureParam<int> kSchedulerMaxJobs(&kPassageEmbedder,
                                                "SchedulerMaxJobs",
                                                64);

const base::FeatureParam<int> kSchedulerMaxBatchSize(&kPassageEmbedder,
                                                     "SchedulerMaxBatchSize",
                                                     1);

const base::FeatureParam<bool> kUsePerformanceScenario(&kPassageEmbedder,
                                                       "UsePerformanceScenario",
                                                       false);

const base::FeatureParam<bool> kUseBackgroundPassageEmbedder(
    &kPassageEmbedder,
    "UseBackgroundPassageEmbedder",
    false);

}  // namespace passage_embeddings