Pass the current builder to the async put thread.

Signed-off-by: Ye Cao <[email protected]>

Author: dashanji

python/vineyard/core/client.py

@@ -42,6 +42,7 @@
 from vineyard._C import VineyardException
 from vineyard._C import _connect
 from vineyard.core.builder import BuilderContext
+from vineyard.core.builder import get_current_builders
 from vineyard.core.builder import put
 from vineyard.core.resolver import ResolverContext
 from vineyard.core.resolver import get
@@ -881,16 +882,19 @@ def put(
         **kwargs,
     ):
         if as_async:
+
             def _default_callback(future):
                 try:
                     result = future.result()
                     print(f"Successfully put object {result}", flush=True)
                 except Exception as e:
                     print(f"Failed to put object: {e}", flush=True)
 
+            current_builder = builder or get_current_builders()
+
             thread_pool = self.put_thread_pool
             result = thread_pool.submit(
-                self._put_internal, value, builder, persist, name, **kwargs
+                self._put_internal, value, current_builder, persist, name, **kwargs
             )
             result.add_done_callback(_default_callback)
             return result