diff --git a/google/generativeai/client.py b/google/generativeai/client.py index 01d0a003b..ad4acb794 100644 --- a/google/generativeai/client.py +++ b/google/generativeai/client.py @@ -220,9 +220,13 @@ def make_client(self, name): ) raise e + return _add_metadata_wrapper(client) + + + def _add_metadata_wrapper(self, client): if not self.default_metadata: return client - + def keep(name, f): if name.startswith("_"): return False @@ -269,6 +273,7 @@ def get_default_operations_client(self) -> operations_v1.OperationsClient: if client is None: model_client = self.get_default_client("Model") client = model_client._transport.operations_client + client = self._add_metadata_wrapper(client) self.clients["operations"] = client return client