import cancel udf changes

Author: kkampli-singlestore

singlestoredb/functions/ext/asgi.py

@@ -307,6 +307,7 @@ def build_vector_udf_endpoint(
     """
     masks = get_masked_params(func)
     array_cls = get_array_class(returns_data_format)
+    is_async = asyncio.iscoroutinefunction(func)
 
     async def do_func(
         row_ids: Sequence[int],
@@ -319,15 +320,16 @@ async def do_func(
         row_ids = array_cls(row_ids)
 
         # Call the function with `cols` as the function parameters
-        is_async = inspect.iscoroutinefunction(func) or inspect.iscoroutinefunction(getattr(func, "__wrapped__", None))
-        if is_async:
-            out = await func(*[x if m else x[0] for x, m in zip(cols, masks)])
+        if cols and cols[0]:
+            if is_async:
+                out = await func(*[x if m else x[0] for x, m in zip(cols, masks)])
+            else:
+                out = func(*[x if m else x[0] for x, m in zip(cols, masks)])
         else:
-            out = await asyncio.to_thread(func, *[x if m else x[0] for x, m in zip(cols, masks)])
-            
-        # Single masked value
-        if isinstance(out, Masked):
-            return row_ids, [tuple(out)]
+            if is_async:
+                out = await func()
+            else:
+                out = func()
 
         # Multiple return values
         if isinstance(out, tuple):