diff --git a/pkgs/development/python-modules/google-ai-generativelanguage/default.nix b/pkgs/development/python-modules/google-ai-generativelanguage/default.nix
index 600828a9000f..5d88dedcd2b5 100644
--- a/pkgs/development/python-modules/google-ai-generativelanguage/default.nix
+++ b/pkgs/development/python-modules/google-ai-generativelanguage/default.nix
@@ -3,9 +3,9 @@
   buildPythonPackage,
   fetchPypi,
   google-api-core,
+  google-auth,
   google-cloud-testutils,
   grpcio,
-  grpcio-status,
   mock,
   proto-plus,
   protobuf,
@@ -29,11 +29,12 @@ buildPythonPackage rec {
 
   dependencies = [
     google-api-core
+    google-auth
     grpcio
-    grpcio-status
     proto-plus
     protobuf
-  ];
+  ]
+  ++ google-api-core.optional-dependencies.grpc;
 
   nativeCheckInputs = [
     google-cloud-testutils