store self

Author: ohmayr

gapic/utils/cache.py

@@ -81,14 +81,14 @@ def wrapper(self, *, collisions, **kwargs):
 
         # 3. Check Cache
         if key in context_cache:
-            return context_cache[key]
+            return context_cache[key][0]
 
         # 4. Execute the actual function
         # We ensure context_cache is passed down to the recursive calls
         result = func(self, collisions=collisions, **kwargs)
 
         # 5. Update Cache
-        context_cache[key] = result
+        context_cache[key] = (result, self)
         return result
 
     return wrapper