@@ -63,9 +63,10 @@ async function main() {
6363 nonInterpolatedMessages : [ userMessage ] ,
6464 } ) ;
6565
66- // Track metrics using trackMetricsOf with VercelProvider.createAIMetrics
67- const result = await aiConfig . tracker . trackMetricsOf ( VercelProvider . createAIMetrics , ( ) =>
68- generateText ( vercelConfig ) ,
66+ // Call the model and track metrics for the ai config
67+ const result = await aiConfig . tracker . trackMetricsOf (
68+ VercelProvider . getAIMetricsFromResponse ,
69+ ( ) => generateText ( vercelConfig ) ,
6970 ) ;
7071
7172 console . log ( 'Response:' , result . text ) ;
@@ -87,11 +88,10 @@ async function main() {
8788 nonInterpolatedMessages : [ userMessage ] ,
8889 } ) ;
8990
90- // Track streaming metrics using trackStreamMetricsOf with provider's extractor
9191 // Stream is returned immediately (synchronously), metrics tracked in background
9292 const streamResult = aiConfig . tracker . trackStreamMetricsOf (
9393 ( ) => streamText ( vercelConfig ) ,
94- VercelProvider . createStreamMetrics ,
94+ VercelProvider . getAIMetricsFromStream ,
9595 ) ;
9696
9797 // Consume the stream immediately - no await needed before this!
0 commit comments