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
|
// 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.
#include "third_party/blink/renderer/modules/ai/ai_utils.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_language_model_create_core_options.h"
namespace blink {
TEST(ResolveSamplingParamsOptionTest, NoOptions) {
auto result = ResolveSamplingParamsOption(nullptr);
ASSERT_TRUE(result.has_value());
ASSERT_FALSE(result.value());
}
TEST(ResolveSamplingParamsOptionTest, NoSamplingParams) {
auto* options = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
auto result = ResolveSamplingParamsOption(options);
ASSERT_TRUE(result.has_value());
ASSERT_FALSE(result.value());
}
TEST(ResolveSamplingParamsOptionTest, ValidOptions) {
auto* options = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
options->setTopK(1);
options->setTemperature(0.5);
auto result = ResolveSamplingParamsOption(options);
ASSERT_TRUE(result.has_value());
ASSERT_EQ(result.value()->top_k, 1u);
ASSERT_EQ(result.value()->temperature, 0.5);
auto* options2 = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
options2->setTopK(10);
options2->setTemperature(0.5);
auto result2 = ResolveSamplingParamsOption(options2);
ASSERT_TRUE(result2.has_value());
ASSERT_EQ(result2.value()->top_k, 10u);
ASSERT_EQ(result2.value()->temperature, 0.5);
auto* options3 = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
options3->setTopK(10);
options3->setTemperature(0);
auto result3 = ResolveSamplingParamsOption(options3);
ASSERT_TRUE(result3.has_value());
ASSERT_EQ(result3.value()->top_k, 10u);
ASSERT_EQ(result3.value()->temperature, 0);
}
TEST(ResolveSamplingParamsOptionTest, OnlyOneOfTopKAndTemperatureIsProvided) {
auto* options = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
options->setTopK(10);
auto result = ResolveSamplingParamsOption(options);
ASSERT_FALSE(result.has_value());
ASSERT_EQ(result.error(),
SamplingParamsOptionError::kOnlyOneOfTopKAndTemperatureIsProvided);
auto* options2 = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
options2->setTemperature(0.5);
auto result2 = ResolveSamplingParamsOption(options2);
ASSERT_FALSE(result2.has_value());
ASSERT_EQ(result2.error(),
SamplingParamsOptionError::kOnlyOneOfTopKAndTemperatureIsProvided);
}
TEST(ResolveSamplingParamsOptionTest, InvalidTopK) {
auto* options = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
options->setTopK(0.5);
options->setTemperature(0.5);
auto result = ResolveSamplingParamsOption(options);
ASSERT_FALSE(result.has_value());
ASSERT_EQ(result.error(), SamplingParamsOptionError::kInvalidTopK);
auto* options2 = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
options2->setTopK(-1);
options2->setTemperature(0.5);
auto result2 = ResolveSamplingParamsOption(options2);
ASSERT_FALSE(result2.has_value());
ASSERT_EQ(result2.error(), SamplingParamsOptionError::kInvalidTopK);
}
TEST(ResolveSamplingParamsOptionTest, InvalidTemperature) {
auto* options = MakeGarbageCollected<LanguageModelCreateCoreOptions>();
options->setTopK(10);
options->setTemperature(-0.5);
auto result = ResolveSamplingParamsOption(options);
ASSERT_FALSE(result.has_value());
ASSERT_EQ(result.error(), SamplingParamsOptionError::kInvalidTemperature);
}
} // namespace blink
|
