Skip to content

Commit 258612b

Browse files
committed
Add test for missing token details
1 parent dd54879 commit 258612b

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

packages/firebase_ai/firebase_ai/test/developer_api_test.dart

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,26 @@ void main() {
147147
expect(response.usageMetadata, isNull);
148148
});
149149

150+
test('parses usageMetadata when token details are missing', () {
151+
final jsonResponse = {
152+
'usageMetadata': {
153+
'promptTokenCount': 10,
154+
'candidatesTokenCount': 25,
155+
'totalTokenCount': 35,
156+
}
157+
};
158+
159+
final response =
160+
DeveloperSerialization().parseGenerateContentResponse(jsonResponse);
161+
162+
expect(response.usageMetadata, isNotNull);
163+
expect(response.usageMetadata!.promptTokenCount, 10);
164+
expect(response.usageMetadata!.candidatesTokenCount, 25);
165+
expect(response.usageMetadata!.totalTokenCount, 35);
166+
expect(response.usageMetadata!.promptTokensDetails, isNull);
167+
expect(response.usageMetadata!.candidatesTokensDetails, isNull);
168+
});
169+
150170
test('parses inlineData part correctly', () {
151171
final inlineData = Uint8List.fromList([1, 2, 3, 4]);
152172
final jsonResponse = {

0 commit comments

Comments
 (0)