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
|
// 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.
#ifndef THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_METRICS_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_METRICS_H_
#include <optional>
#include <string>
#include "third_party/blink/public/mojom/ai/ai_language_model.mojom-blink.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_language_model_message_content.h"
namespace blink {
class AIMetrics {
public:
// This class contains all the supported session types.
// LINT.IfChange(AISessionType)
enum class AISessionType {
kLanguageModel = 0,
kWriter = 1,
kRewriter = 2,
kSummarizer = 3,
kTranslator = 4,
kLanguageDetector = 5,
kProofreader = 6,
kMaxValue = kProofreader,
};
// LINT.ThenChange(//tools/metrics/histograms/metadata/ai/histograms.xml:SessionType)
// This class contains all the model execution API supported.
// These values are persisted to logs. Entries should not be renumbered and
// numeric values should never be reused.
// TODO(crbug.com/355967885): update the enums when adding metrics for
// language model API.
// LINT.IfChange(AIAPI)
enum class AIAPI {
kCanCreateSession = 0,
kCreateSession = 1,
kSessionPrompt = 2,
kSessionPromptStreaming = 3,
kDefaultTextSessionOptions = 4,
kSessionDestroy = 5,
kSessionClone = 6,
kTextModelInfo = 7,
kSessionSummarize = 8,
kSessionSummarizeStreaming = 9,
kWriterWrite = 10,
kWriterWriteStreaming = 11,
kRewriterRewrite = 12,
kRewriterRewriteStreaming = 13,
kSummarizerSummarize = 14,
kSummarizerSummarizeStreaming = 15,
kSummarizerCreate = 16,
kSummarizerDestroy = 17,
kSessionCountPromptTokens = 18,
kProofreaderProofread = 19,
kProofreaderCreate = 20,
kProofreaderDestroy = 21,
kMaxValue = kProofreaderDestroy,
};
// LINT.ThenChange(//tools/metrics/histograms/metadata/ai/enums.xml:AIAPI)
// LINT.IfChange(LanguageModelInputType)
enum class LanguageModelInputType {
kText = 0,
kImage = 1,
kAudio = 2,
kMaxValue = kAudio,
};
// LINT.ThenChange(//tools/metrics/histograms/metadata/ai/enums.xml:LanguageModelInputType)
// LINT.IfChange(LanguageModelInputRole)
enum class LanguageModelInputRole {
kSystem = 0,
kUser = 1,
kAssistant = 2,
kMaxValue = kAssistant,
};
// LINT.ThenChange(//tools/metrics/histograms/metadata/ai/enums.xml:LanguageModelInputRole)
static std::string GetAIAPIUsageMetricName(AISessionType session_type);
static std::string GetAvailabilityMetricName(AISessionType session_type);
static std::string GetAISessionRequestSizeMetricName(
AISessionType session_type);
static std::string GetAISessionResponseStatusMetricName(
AISessionType session_type);
static std::string GetAISessionResponseSizeMetricName(
AISessionType session_type);
static std::string GetAISessionResponseCallbackCountMetricName(
AISessionType session_type);
// Enum mappings from mojo/V8 enums to metric enums. Returns nullopt if the
// enum is not mapped.
static LanguageModelInputType ToLanguageModelInputType(
mojom::blink::AILanguageModelPromptContent::Tag type);
static LanguageModelInputType ToLanguageModelInputType(
V8LanguageModelMessageType::Enum type);
static LanguageModelInputRole ToLanguageModelInputRole(
mojom::blink::AILanguageModelPromptRole role);
};
} // namespace blink
#endif // THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_METRICS_H_
|
