From 628b6a6446b43f44941d4f8ac5a4cc75aa450568 Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Wed, 20 Nov 2024 06:42:44 -0800 Subject: [PATCH] Add metadata wrapper --- google/generativeai/client.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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