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
|
// 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 "third_party/blink/renderer/modules/ai/ai_metrics.h"
#include <string_view>
#include "base/notreached.h"
#include "base/strings/strcat.h"
namespace blink {
namespace {
std::string_view GetAISessionTypeName(AIMetrics::AISessionType session_type) {
switch (session_type) {
case AIMetrics::AISessionType::kLanguageModel:
return "LanguageModel";
case AIMetrics::AISessionType::kWriter:
return "Writer";
case AIMetrics::AISessionType::kRewriter:
return "Rewriter";
case AIMetrics::AISessionType::kSummarizer:
return "Summarizer";
case AIMetrics::AISessionType::kTranslator:
return "Translator";
case AIMetrics::AISessionType::kLanguageDetector:
return "LanguageDetector";
case AIMetrics::AISessionType::kProofreader:
return "Proofreader";
}
NOTREACHED();
}
} // namespace
// static
std::string AIMetrics::GetAIAPIUsageMetricName(AISessionType session_type) {
return base::StrCat({"AI.", GetAISessionTypeName(session_type), ".APIUsage"});
}
// static
std::string AIMetrics::GetAvailabilityMetricName(AISessionType session_type) {
return base::StrCat(
{"AI.", GetAISessionTypeName(session_type), ".AvailabilityV2"});
}
// static
std::string AIMetrics::GetAISessionRequestSizeMetricName(
AISessionType session_type) {
return base::StrCat({"AI.Session.", GetAISessionTypeName(session_type),
".PromptRequestSize"});
}
// static
std::string AIMetrics::GetAISessionResponseStatusMetricName(
AISessionType session_type) {
return base::StrCat({"AI.Session.", GetAISessionTypeName(session_type),
".PromptResponseStatus"});
}
// static
std::string AIMetrics::GetAISessionResponseSizeMetricName(
AISessionType session_type) {
return base::StrCat({"AI.Session.", GetAISessionTypeName(session_type),
".PromptResponseSize"});
}
// static
std::string AIMetrics::GetAISessionResponseCallbackCountMetricName(
AISessionType session_type) {
return base::StrCat({"AI.Session.", GetAISessionTypeName(session_type),
".PromptResponseCallbackCount"});
}
// static
AIMetrics::LanguageModelInputType AIMetrics::ToLanguageModelInputType(
mojom::blink::AILanguageModelPromptContent::Tag type) {
using MojoEnum = mojom::blink::AILanguageModelPromptContent::Tag;
using MetricsEnum = AIMetrics::LanguageModelInputType;
switch (type) {
case MojoEnum::kText:
return MetricsEnum::kText;
case MojoEnum::kBitmap:
return MetricsEnum::kImage;
case MojoEnum::kAudio:
return MetricsEnum::kAudio;
}
NOTREACHED();
}
// static
AIMetrics::LanguageModelInputType AIMetrics::ToLanguageModelInputType(
V8LanguageModelMessageType::Enum type) {
using V8Enum = V8LanguageModelMessageType::Enum;
using MetricsEnum = AIMetrics::LanguageModelInputType;
switch (type) {
case V8Enum::kText:
return MetricsEnum::kText;
case V8Enum::kImage:
return MetricsEnum::kImage;
case V8Enum::kAudio:
return MetricsEnum::kAudio;
}
NOTREACHED();
}
// static
AIMetrics::LanguageModelInputRole AIMetrics::ToLanguageModelInputRole(
mojom::blink::AILanguageModelPromptRole role) {
using MojoEnum = mojom::blink::AILanguageModelPromptRole;
using MetricsEnum = AIMetrics::LanguageModelInputRole;
switch (role) {
case MojoEnum::kSystem:
return MetricsEnum::kSystem;
case MojoEnum::kUser:
return MetricsEnum::kUser;
case MojoEnum::kAssistant:
return MetricsEnum::kAssistant;
}
NOTREACHED();
}
} // namespace blink
|
