recorder toolUsePromptTokenCount and improve tests

Author: dlarocque

packages/firebase_ai/firebase_ai/lib/src/api.dart

@@ -187,9 +187,9 @@ final class UsageMetadata {
     this.candidatesTokenCount,
     this.totalTokenCount,
     this.thoughtsTokenCount,
+    this.toolUsePromptTokenCount,
     this.promptTokensDetails,
     this.candidatesTokensDetails,
-    this.toolUsePromptTokenCount,
     this.toolUsePromptTokensDetails,
   });
 

packages/firebase_ai/firebase_ai/test/api_test.dart

@@ -1055,6 +1055,51 @@ void main() {
           expect(urlMetadata.urlRetrievalStatus, UrlRetrievalStatus.success);
         });
 
+        test(
+            'parses valid response with full url context metadata and list of url metadata',
+            () {
+          final jsonResponse = {
+            'candidates': [
+              {
+                'content': {
+                  'parts': [
+                    {'text': 'Some text'}
+                  ]
+                },
+                'finishReason': 'STOP',
+                'urlContextMetadata': {
+                  'urlMetadata': [
+                    {
+                      'retrievedUrl': 'https://example.com',
+                      'urlRetrievalStatus': 'URL_RETRIEVAL_STATUS_SUCCESS'
+                    },
+                    {
+                      'retrievedUrl': 'https://foo.com',
+                      'urlRetrievalStatus': 'URL_RETRIEVAL_STATUS_ERROR'
+                    }
+                  ]
+                }
+              }
+            ]
+          };
+          final response =
+              VertexSerialization().parseGenerateContentResponse(jsonResponse);
+          final urlContextMetadata =
+              response.candidates.first.urlContextMetadata;
+          expect(urlContextMetadata, isNotNull);
+          expect(urlContextMetadata!.urlMetadata, hasLength(2));
+          final firstUrlMetadata = urlContextMetadata.urlMetadata.first;
+          expect(
+              firstUrlMetadata.retrievedUrl, Uri.parse('https://example.com'));
+          expect(
+              firstUrlMetadata.urlRetrievalStatus, UrlRetrievalStatus.success);
+          final secondUrlMetadata = urlContextMetadata.urlMetadata[1];
+          expect(
+              secondUrlMetadata.retrievedUrl, Uri.parse('https://foo.com'));
+          expect(
+              secondUrlMetadata.urlRetrievalStatus, UrlRetrievalStatus.error);
+        });
+
         test('parses response with missing retrievedUrl', () {
           final jsonResponse = {
             'candidates': [

packages/firebase_ai/firebase_ai/test/google_ai_response_parsing_test.dart

@@ -853,14 +853,20 @@ void main() {
           generateContentResponse.candidates.first.urlContextMetadata;
       expect(urlContextMetadata, isNotNull);
       expect(urlContextMetadata!.urlMetadata, hasLength(3));
-      expect(urlContextMetadata.urlMetadata[2].retrievedUrl,
-          Uri.parse('https://a-completely-non-existent-url-for-testing.org'));
-      expect(urlContextMetadata.urlMetadata[2].urlRetrievalStatus,
+      expect(
+          urlContextMetadata.urlMetadata[0].retrievedUrl,
+          Uri.parse(
+              'https://www.nytimes.com/2023/06/25/realestate/barbiecore-home-decor-interior-design.html?action=click&contentCollection=undefined&region=Footer&module=WhatsNext&version=WhatsNext&contentID=WhatsNext&moduleDetail=most-emailed-0&pgtype=undefinedl'));
+      expect(urlContextMetadata.urlMetadata[0].urlRetrievalStatus,
           UrlRetrievalStatus.error);
       expect(urlContextMetadata.urlMetadata[1].retrievedUrl,
           Uri.parse('https://ai.google.dev'));
       expect(urlContextMetadata.urlMetadata[1].urlRetrievalStatus,
           UrlRetrievalStatus.success);
+      expect(urlContextMetadata.urlMetadata[2].retrievedUrl,
+          Uri.parse('https://a-completely-non-existent-url-for-testing.org'));
+      expect(urlContextMetadata.urlMetadata[2].urlRetrievalStatus,
+          UrlRetrievalStatus.error);
     });
 
     test('allows missing content', () async {