Skip to content

Commit acbe4da

Browse files
ryanwilsongspencergoog
authored andcommitted
Remove generateContent call from count tokens page (#17613)
This isn't clear in the sample UI why this is happening, and ends up using more tokens than necessary.
1 parent b2dcd1a commit acbe4da

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

packages/firebase_ai/firebase_ai/example/lib/pages/token_count_page.dart

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -87,17 +87,9 @@ class _TokenCountPageState extends State<TokenCountPage> {
8787
const prompt = 'tell a short story';
8888
final content = Content.text(prompt);
8989
final tokenResponse = await widget.model.countTokens([content]);
90-
final tokenResult = 'Count token: ${tokenResponse.totalTokens}';
90+
final tokenResult = 'Token Count: ${tokenResponse.totalTokens}';
9191
_messages.add(MessageData(text: tokenResult, fromUser: false));
9292

93-
final contentResponse = await widget.model.generateContent([content]);
94-
final contentMetaData = 'result metadata, promptTokenCount:'
95-
'${contentResponse.usageMetadata!.promptTokenCount}, '
96-
'candidatesTokenCount:'
97-
'${contentResponse.usageMetadata!.candidatesTokenCount}, '
98-
'totalTokenCount:'
99-
'${contentResponse.usageMetadata!.totalTokenCount}';
100-
_messages.add(MessageData(text: contentMetaData, fromUser: false));
10193
setState(() {
10294
_loading = false;
10395
});

0 commit comments

Comments
 (0)