diff --git a/pkgs/dart_services/lib/src/generative_ai.dart b/pkgs/dart_services/lib/src/generative_ai.dart
index 40b787afb..1383d0602 100644
--- a/pkgs/dart_services/lib/src/generative_ai.dart
+++ b/pkgs/dart_services/lib/src/generative_ai.dart
@@ -6,7 +6,7 @@ import 'dart:async';
 
 import 'package:dartpad_shared/model.dart';
 import 'package:google_cloud_ai_generativelanguage_v1beta/generativelanguage.dart';
-import 'package:google_cloud_gax/gax.dart';
+import 'package:google_cloud_rpc/service_client.dart';
 
 import 'logging.dart';
 import 'project_templates.dart';
@@ -198,7 +198,7 @@ $prompt
 
   static Stream<String> _textOnly(Stream<GenerateContentResponse> stream) {
     return stream.map((response) {
-      final parts = response.candidates?.firstOrNull?.content?.parts ?? [];
+      final parts = response.candidates.firstOrNull?.content?.parts ?? [];
       return parts.where((part) => part.text != null).map((p) => p.text).join();
     });
   }
diff --git a/pkgs/dart_services/pubspec.yaml b/pkgs/dart_services/pubspec.yaml
index 1c8a90766..3cdff3b12 100644
--- a/pkgs/dart_services/pubspec.yaml
+++ b/pkgs/dart_services/pubspec.yaml
@@ -15,8 +15,8 @@ dependencies:
   dartpad_shared:
     path: ../dartpad_shared
   encrypt: ^5.0.3
-  google_cloud_ai_generativelanguage_v1beta: ^0.1.2
-  google_cloud_gax: ^0.1.2
+  google_cloud_ai_generativelanguage_v1beta: ^0.2.0
+  google_cloud_rpc: ^0.2.0
   http: ^1.3.0
   json_annotation: ^4.9.0
   logging: ^1.2.0