From fc9ec60b6908e6bfb5ef38a2a8dbd17165ef5201 Mon Sep 17 00:00:00 2001
From: Xuanwo <github@xuanwo.io>
Date: Fri, 21 Aug 2026 22:33:20 +0800
Subject: [PATCH] package demo udf callable for remote execution

---
 python/python/lancedb/functions.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/python/python/lancedb/functions.py b/python/python/lancedb/functions.py
index 5979ba5fe..88bd53d7b 100644
--- a/python/python/lancedb/functions.py
+++ b/python/python/lancedb/functions.py
@@ -741,7 +741,11 @@ def _package_source(function: Callable[..., Any]) -> bytes:
     if globals_source:
         parts.extend(["", *globals_source])
     parts.extend(["", function_source, ""])
-    return "\n".join(parts).encode("utf-8")
+    try:
+        import cloudpickle
+    except ImportError as error:
+        raise RuntimeError("@udf remote registration requires cloudpickle") from error
+    return cloudpickle.dumps(function)
 
 
 class UdfDefinition: