Merge branch 'main' of https://github.com/singlestore-labs/singlestoredb-python into users/kaushik/interactive-udfs-changes

Author: kkampli-singlestore

coverage-mysql.cov

@@ -256,11 +256,15 @@ create new ones.
    WorkspaceManager
    WorkspaceManager.organization
    WorkspaceManager.workspace_groups
+   WorkspaceManager.starter_workspaces
    WorkspaceManager.regions
+   WorkspaceManager.shared_tier_regions
    WorkspaceManager.create_workspace_group
    WorkspaceManager.create_workspace
+   WorkspaceManager.create_starter_workspace
    WorkspaceManager.get_workspace_group
    WorkspaceManager.get_workspace
+   WorkspaceManager.get_starter_workspace
 
 
 WorkspaceGroup

docs/src/api.rst

@@ -25,7 +25,7 @@
     DataError, ManagementError,
 )
 from .management import (
-    manage_cluster, manage_workspaces, manage_files,
+    manage_cluster, manage_workspaces, manage_files, manage_regions,
 )
 from .types import (
     Date, Time, Timestamp, DateFromTicks, TimeFromTicks, TimestampFromTicks,

singlestoredb/__init__.py

@@ -438,6 +438,11 @@
     environ=['SINGLESTOREDB_EXT_FUNC_PORT'],
 )
 
+register_option(
+    'external_function.timeout', 'int', check_int, 24*60*60,
+    'Specifies the timeout in seconds for processing a batch of rows.',
+    environ=['SINGLESTOREDB_EXT_FUNC_TIMEOUT'],
+)
 
 #
 # Debugging options

singlestoredb/config.py

@@ -111,6 +111,7 @@ def _func(
             name=name,
             args=expand_types(args),
             returns=expand_types(returns),
+            timeout=timeout,
         ).items() if v is not None
     }
 
@@ -181,6 +182,9 @@ def udf(
         Specifies the return data type of the function. This parameter
         works the same way as `args`. If the function is a table-valued
         function, the return type should be a `Table` object.
+    timeout : int, optional
+        The timeout in seconds for the UDF execution. If not specified,
+        the global default timeout is used.
 
     Returns
     -------
@@ -192,4 +196,5 @@ def udf(
         name=name,
         args=args,
         returns=returns,
+        timeout=timeout,
     )