singlestoredb 1.15.1__tar.gz → 1.15.3__tar.gz

{singlestoredb-1.15.1/singlestoredb.egg-info → singlestoredb-1.15.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: singlestoredb
-Version: 1.15.1
+Version: 1.15.3
 Summary: Interface to the SingleStoreDB database and workspace management APIs
 Home-page: https://github.com/singlestore-labs/singlestoredb-python
 Author: SingleStore

{singlestoredb-1.15.1 → singlestoredb-1.15.3}/accel.c

@@ -1950,7 +1950,6 @@ static PyObject *read_row_from_packet(
         case ACCEL_OUT_DICTS:
         case ACCEL_OUT_ARROW:
             PyDict_SetItem(py_result, py_state->py_names[i], py_item);
-            Py_INCREF(py_state->py_names[i]);
             Py_DECREF(py_item);
             break;
         default:
@@ -2678,8 +2677,19 @@ static PyObject *load_rowdat_1_numpy(PyObject *self, PyObject *args, PyObject *k
 
 exit:
     if (ctypes) free(ctypes);
-    if (out_cols) free(out_cols);
-    if (mask_cols) free(mask_cols);
+    if (out_cols) {
+        for (i = 0; i < n_cols; i++) {
+            if (out_cols[i]) free(out_cols[i]);
+        }
+        free(out_cols);
+    }
+    if (mask_cols) {
+        for (i = 0; i < n_cols; i++) {
+            if (mask_cols[i]) free(mask_cols[i]);
+        }
+        free(mask_cols);
+    }
+    if (out_row_ids) free(out_row_ids);
     if (data_formats) free(data_formats);
     if (item_sizes) free(item_sizes);
 
@@ -2943,11 +2953,17 @@ static PyObject *dump_rowdat_1_numpy(PyObject *self, PyObject *args, PyObject *k
     out_l = 256 * n_cols;
     out_idx = 0;
     out = malloc(out_l);
-    if (!out) goto error;
+    if (!out) {
+        PyErr_SetString(PyExc_MemoryError, "failed to allocate output buffer");
+        goto error;
+    }
 
     // Get return types
     returns = malloc(sizeof(int) * n_cols);
-    if (!returns) goto error;
+    if (!returns) {
+        PyErr_SetString(PyExc_MemoryError, "failed to allocate returns array");
+        goto error;
+    }
 
     for (i = 0; i < n_cols; i++) {
         PyObject *py_item = PySequence_GetItem(py_returns, i);
@@ -2959,11 +2975,20 @@ static PyObject *dump_rowdat_1_numpy(PyObject *self, PyObject *args, PyObject *k
 
     // Get column array memory
     cols = calloc(sizeof(char*), n_cols);
-    if (!cols) goto error;
+    if (!cols) {
+        PyErr_SetString(PyExc_MemoryError, "failed to allocate cols array");
+        goto error;
+    }
     col_types = calloc(sizeof(NumpyColType), n_cols);
-    if (!col_types) goto error;
+    if (!col_types) {
+        PyErr_SetString(PyExc_MemoryError, "failed to allocate col_types array");
+        goto error;
+    }
     masks = calloc(sizeof(char*), n_cols);
-    if (!masks) goto error;
+    if (!masks) {
+        PyErr_SetString(PyExc_MemoryError, "failed to allocate masks array");
+        goto error;
+    }
     for (i = 0; i < n_cols; i++) {
         PyObject *py_item = PyList_GetItem(py_cols, i);
         if (!py_item) goto error;
@@ -2996,8 +3021,12 @@ static PyObject *dump_rowdat_1_numpy(PyObject *self, PyObject *args, PyObject *k
 #define CHECKMEM(x) \
     if ((out_idx + x) > out_l) { \
         out_l = out_l * 2 + x; \
-        out = realloc(out, out_l); \
-        if (!out) goto error; \
+        char *new_out = realloc(out, out_l); \
+        if (!new_out) { \
+            PyErr_SetString(PyExc_MemoryError, "failed to reallocate output buffer"); \
+            goto error; \
+        } \
+        out = new_out; \
     }
 
     for (j = 0; j < n_rows; j++) {
@@ -4079,10 +4108,10 @@ static PyObject *dump_rowdat_1_numpy(PyObject *self, PyObject *args, PyObject *k
         }
     }
 
-    py_out = PyMemoryView_FromMemory(out, out_idx, PyBUF_WRITE);
-    if (!py_out) goto error;
+    py_out = PyBytes_FromStringAndSize(out, out_idx);
 
 exit:
+    if (out) free(out);
     if (returns) free(returns);
     if (masks) free(masks);
     if (cols) free(cols);
@@ -4091,7 +4120,6 @@ exit:
     return py_out;
 
 error:
-    if (!py_out && out) free(out);
     Py_XDECREF(py_out);
     py_out = NULL;
 
@@ -4471,8 +4499,12 @@ static PyObject *dump_rowdat_1(PyObject *self, PyObject *args, PyObject *kwargs)
 #define CHECKMEM(x) \
     if ((out_idx + x) > out_l) { \
         out_l = out_l * 2 + x; \
-        out = realloc(out, out_l); \
-        if (!out) goto error; \
+        char *new_out = realloc(out, out_l); \
+        if (!new_out) { \
+            PyErr_SetString(PyExc_MemoryError, "failed to reallocate output buffer"); \
+            goto error; \
+        } \
+        out = new_out; \
     }
 
     py_rows_iter = PyObject_GetIter(py_rows);
@@ -4483,12 +4515,20 @@ static PyObject *dump_rowdat_1(PyObject *self, PyObject *args, PyObject *kwargs)
 
     while ((py_row = PyIter_Next(py_rows_iter))) {
         py_row_iter = PyObject_GetIter(py_row);
-        if (!py_row_iter) goto error;
+        if (!py_row_iter) {
+            Py_DECREF(py_row);
+            goto error;
+        }
 
         // First item is always a row ID
         py_item = PyIter_Next(py_row_ids_iter);
-        if (!py_item) goto error;
+        if (!py_item) {
+            Py_DECREF(py_row_iter);
+            Py_DECREF(py_row);
+            goto error;
+        }
         row_id = (int64_t)PyLong_AsLongLong(py_item);
+        Py_DECREF(py_item);
 
         CHECKMEM(8);
         memcpy(out+out_idx, &row_id, 8);
@@ -4631,12 +4671,16 @@ static PyObject *dump_rowdat_1(PyObject *self, PyObject *args, PyObject *kwargs)
                     out_idx += 8;
                 } else {
                     PyObject *py_bytes = PyUnicode_AsEncodedString(py_item, "utf-8", "strict");
-                    if (!py_bytes) goto error;
+                    if (!py_bytes) {
+                        Py_DECREF(py_item);
+                        goto error;
+                    }
 
                     char *str = NULL;
                     Py_ssize_t str_l = 0;
                     if (PyBytes_AsStringAndSize(py_bytes, &str, &str_l) < 0) {
                         Py_DECREF(py_bytes);
+                        Py_DECREF(py_item);
                         goto error;
                     }
 
@@ -4671,6 +4715,7 @@ static PyObject *dump_rowdat_1(PyObject *self, PyObject *args, PyObject *kwargs)
                     char *str = NULL;
                     Py_ssize_t str_l = 0;
                     if (PyBytes_AsStringAndSize(py_item, &str, &str_l) < 0) {
+                        Py_DECREF(py_item);
                         goto error;
                     }
 
@@ -4684,6 +4729,7 @@ static PyObject *dump_rowdat_1(PyObject *self, PyObject *args, PyObject *kwargs)
                 break;
 
             default:
+                Py_DECREF(py_item);
                 goto error;
             }
 
@@ -4693,14 +4739,17 @@ static PyObject *dump_rowdat_1(PyObject *self, PyObject *args, PyObject *kwargs)
             i++;
         }
 
+        Py_DECREF(py_row_iter);
         Py_DECREF(py_row);
+        py_row_iter = NULL;
         py_row = NULL;
     }
 
-    py_out = PyMemoryView_FromMemory(out, out_idx, PyBUF_WRITE);
-    if (!py_out) goto error;
+    // Convert the output buffer to a Python bytes object and free the buffer
+    py_out = PyBytes_FromStringAndSize(out, out_idx);
 
 exit:
+    if (out) free(out);
     if (returns) free(returns);
 
     Py_XDECREF(py_item);
@@ -4712,7 +4761,6 @@ exit:
     return py_out;
 
 error:
-    if (!py_out && out) free(out);
     Py_XDECREF(py_out);
     py_out = NULL;
 
@@ -4839,7 +4887,7 @@ PyMODINIT_FUNC PyInit__singlestoredb_accel(void) {
 
     PyObj.create_numpy_array_kwargs = PyDict_New();
     if (!PyObj.create_numpy_array_kwargs) goto error;
-    if (PyDict_SetItemString(PyObj.create_numpy_array_kwargs, "copy", Py_False)) {
+    if (PyDict_SetItemString(PyObj.create_numpy_array_kwargs, "copy", Py_True)) {
         goto error;
     }
 

{singlestoredb-1.15.1 → singlestoredb-1.15.3}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = singlestoredb
-version = 1.15.1
+version = 1.15.3
 description = Interface to the SingleStoreDB database and workspace management APIs
 long_description = file: README.md
 long_description_content_type = text/markdown

{singlestoredb-1.15.1 → singlestoredb-1.15.3}/singlestoredb/__init__.py

@@ -13,7 +13,7 @@ Examples
 
 """
 
-__version__ = '1.15.1'
+__version__ = '1.15.3'
 
 from typing import Any
 

{singlestoredb-1.15.1 → singlestoredb-1.15.3}/singlestoredb/apps/_python_udfs.py

@@ -13,6 +13,9 @@ if typing.TYPE_CHECKING:
 # Keep track of currently running server
 _running_server: 'typing.Optional[AwaitableUvicornServer]' = None
 
+# Maximum number of UDFs allowed
+MAX_UDFS_LIMIT = 10
+
 
 async def run_udf_app(
     log_level: str = 'error',
@@ -44,20 +47,32 @@ async def run_udf_app(
     udf_suffix = ''
     if app_config.running_interactively:
         udf_suffix = '_test'
-    app = Application(url=base_url, app_mode='managed', name_suffix=udf_suffix)
+    app = Application(
+        url=base_url,
+        app_mode='managed',
+        name_suffix=udf_suffix,
+        log_level=log_level,
+    )
+
+    if not app.endpoints:
+        raise ValueError('You must define at least one function.')
+    if len(app.endpoints) > MAX_UDFS_LIMIT:
+        raise ValueError(
+            f'You can only define a maximum of {MAX_UDFS_LIMIT} functions.',
+        )
 
     config = uvicorn.Config(
         app,
         host='0.0.0.0',
         port=app_config.listen_port,
-        log_level=log_level,
+        log_config=app.get_uvicorn_log_config(),
     )
-    _running_server = AwaitableUvicornServer(config)
 
     # Register the functions only if the app is running interactively.
     if app_config.running_interactively:
         app.register_functions(replace=True)
 
+    _running_server = AwaitableUvicornServer(config)
     asyncio.create_task(_running_server.serve())
     await _running_server.wait_for_startup()
 

{singlestoredb-1.15.1 → singlestoredb-1.15.3}/singlestoredb/apps/_stdout_supress.py

@@ -9,7 +9,7 @@ class StdoutSuppressor:
     This should not be used for asynchronous or threaded executions.
 
     ```py
-    with Supressor():
+    with StdoutSupressor():
         print("This won't be printed")
     ```
 

{singlestoredb-1.15.1 → singlestoredb-1.15.3}/singlestoredb/apps/_uvicorn_util.py

@@ -30,3 +30,7 @@ class AwaitableUvicornServer(uvicorn.Server):
 
     async def wait_for_startup(self) -> None:
         await self._startup_future
+
+    async def shutdown(self, sockets: Optional[list[socket.socket]] = None) -> None:
+        if self.started:
+            await super().shutdown(sockets)

{singlestoredb-1.15.1 → singlestoredb-1.15.3}/singlestoredb/config.py

@@ -407,6 +407,12 @@ register_option(
     environ=['SINGLESTOREDB_EXT_FUNC_LOG_LEVEL'],
 )
 
+register_option(
+    'external_function.log_file', 'string', check_str, None,
+    'File path to write logs to instead of console.',
+    environ=['SINGLESTOREDB_EXT_FUNC_LOG_FILE'],
+)
+
 register_option(
     'external_function.name_prefix', 'string', check_str, '',
     'Prefix to add to external function names.',
@@ -450,6 +456,18 @@ register_option(
     environ=['SINGLESTOREDB_EXT_FUNC_TIMEOUT'],
 )
 
+register_option(
+    'external_function.disable_metrics', 'bool', check_bool, False,
+    'Disable logging of function call metrics.',
+    environ=['SINGLESTOREDB_EXT_FUNC_DISABLE_METRICS'],
+)
+
+register_option(
+    'external_function.app_name', 'string', check_str, None,
+    'Name for the external function application instance.',
+    environ=['SINGLESTOREDB_EXT_FUNC_APP_NAME'],
+)
+
 #
 # Debugging options
 #

{singlestoredb-1.15.1 → singlestoredb-1.15.3}/singlestoredb/converters.py

@@ -23,7 +23,7 @@ except ImportError:
 try:
     import pygeos
     has_pygeos = True
-except ImportError:
+except (AttributeError, ImportError):
     has_pygeos = False
 
 try: