1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_UTILS_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_UTILS_H_
#include "base/types/expected.h"
#include "third_party/blink/public/mojom/ai/ai_common.mojom-blink.h"
#include "third_party/blink/public/mojom/ai/ai_language_model.mojom-blink.h"
#include "third_party/blink/public/mojom/ai/ai_proofreader.mojom-blink.h"
#include "third_party/blink/public/mojom/ai/ai_rewriter.mojom-blink.h"
#include "third_party/blink/public/mojom/ai/ai_summarizer.mojom-blink.h"
#include "third_party/blink/public/mojom/ai/ai_writer.mojom-blink.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_language_model_expected.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_proofreader_create_options.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_rewriter_create_options.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_summarizer_create_options.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_writer_create_options.h"
#include "third_party/blink/renderer/modules/ai/availability.h"
#include "third_party/blink/renderer/platform/heap/collection_support/heap_vector.h"
namespace blink {
class LanguageModelCreateCoreOptions;
static constexpr uint64_t kNormalizedDownloadProgressMax = 0x10000;
// Converts string language codes to AILanguageCode mojo struct.
Vector<mojom::blink::AILanguageCodePtr> ToMojoLanguageCodes(
const Vector<String>& language_codes);
enum class SamplingParamsOptionError {
kOnlyOneOfTopKAndTemperatureIsProvided,
kInvalidTopK,
kInvalidTemperature,
};
// Performs check on the sampling params option and return the constructed
// AILanguageModelSamplingParamsPtr if the option is valid, or a DOMException
// otherwise.
MODULES_EXPORT base::expected<mojom::blink::AILanguageModelSamplingParamsPtr,
SamplingParamsOptionError>
ResolveSamplingParamsOption(const LanguageModelCreateCoreOptions* options);
mojom::blink::AISummarizerCreateOptionsPtr ToMojoSummarizerCreateOptions(
const SummarizerCreateOptions* options);
mojom::blink::AISummarizerCreateOptionsPtr ToMojoSummarizerCreateOptions(
const SummarizerCreateCoreOptions* core_options);
mojom::blink::AIWriterCreateOptionsPtr ToMojoWriterCreateOptions(
const WriterCreateOptions* options);
mojom::blink::AIWriterCreateOptionsPtr ToMojoWriterCreateOptions(
const WriterCreateCoreOptions* core_options);
mojom::blink::AIRewriterCreateOptionsPtr ToMojoRewriterCreateOptions(
const RewriterCreateOptions* options);
mojom::blink::AIRewriterCreateOptionsPtr ToMojoRewriterCreateOptions(
const RewriterCreateCoreOptions* core_options);
mojom::blink::AIProofreaderCreateOptionsPtr ToMojoProofreaderCreateOptions(
const ProofreaderCreateOptions* options);
mojom::blink::AIProofreaderCreateOptionsPtr ToMojoProofreaderCreateOptions(
const ProofreaderCreateCoreOptions* core_options);
// Convert language model expected inputs or outputs to the matching mojo type.
Vector<mojom::blink::AILanguageModelExpectedPtr> ToMojoExpectations(
const HeapVector<Member<LanguageModelExpected>>& expected);
// Implementation of LookupMatchingLocaleByBestFit
// (https://tc39.es/ecma402/#sec-lookupmatchinglocalebybestfit) as
// LookupMatchingLocaleByPrefix
// (https://tc39.es/ecma402/#sec-lookupmatchinglocalebyprefix) assuming
// `available_languages` contains no extension.
template <typename SetType>
std::optional<String> LookupMatchingLocaleByBestFit(
const SetType& available_languages,
const String& requested_language) {
String prefix = requested_language;
while (prefix != "") {
if (available_languages.contains(prefix.Ascii())) {
return prefix;
}
int pos = prefix.ReverseFind('-');
if (pos == -1) {
pos = 0;
}
prefix = prefix.Substring(0, pos);
}
return std::nullopt;
}
// Returns a set of language codes that best fit the `requested_languages` given
// `available_languages`
template <typename SetType>
std::optional<Vector<String>> GetBestFitLanguages(
const SetType& available_languages,
const Vector<String>& requested_languages) {
Vector<String> languages;
for (const String& language : requested_languages) {
std::optional<String> best_match =
LookupMatchingLocaleByBestFit(available_languages, language);
if (!best_match) {
return std::nullopt;
}
// Insert if there's no duplicate.
if (!languages.Contains(*best_match)) {
languages.push_back(*std::move(best_match));
}
}
return languages;
}
// Validates and canonicalizes a list of BCP47 formatted language strings.
// Returns nullopt if any of the languages are invalid, otherwise returns the
// canonicalized languages.
std::optional<Vector<String>> ValidateAndCanonicalizeBCP47Languages(
v8::Isolate* isolate,
const Vector<String>& languages);
// Returns whether model availability status requires user activation for
// creating a client.
bool RequiresUserActivation(Availability availability);
// Runs `callback` on destruction unless `Reset` is called.
class RunOnDestruction {
public:
explicit RunOnDestruction(base::OnceClosure callback);
~RunOnDestruction();
RunOnDestruction(const RunOnDestruction&) = delete;
RunOnDestruction& operator=(const RunOnDestruction&) = delete;
RunOnDestruction(RunOnDestruction&& other) = default;
RunOnDestruction& operator=(RunOnDestruction&& other) = default;
void Reset();
private:
base::OnceClosure callback_;
};
} // namespace blink
#endif // THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_UTILS_H_
|
