singlestoredb 1.12.4__py3-none-any.whl → 1.13.1__py3-none-any.whl

singlestoredb/tests/test_udf_returns.py

@@ -0,0 +1,459 @@
+# mypy: disable-error-code="type-arg"
+# from __future__ import annotations
+import unittest
+from typing import Any
+from typing import Callable
+from typing import List
+from typing import NamedTuple
+from typing import Optional
+from typing import TypedDict
+
+import numpy as np
+import numpy.typing as npt
+import pandas as pd
+import polars as pl
+import pyarrow as pa
+from pydantic import BaseModel
+
+from singlestoredb.functions import Table
+from singlestoredb.functions import udf
+from singlestoredb.functions.signature import get_signature
+from singlestoredb.functions.signature import signature_to_sql
+
+
+def to_sql(func: Callable[..., Any]) -> str:
+    """Convert a function signature to SQL."""
+    out = signature_to_sql(get_signature(func))
+    return out.split('EXTERNAL FUNCTION ')[1].split('AS REMOTE')[0].strip()
+
+
+class Parameters(NamedTuple):
+    x: Optional[str] = ''
+
+
+class UDFTuple(NamedTuple):
+    value: str
+
+
+class TVFTuple(NamedTuple):
+    idx: int
+    value: Optional[str]
+
+
+class TVFDict(TypedDict):
+    idx: int
+    value: Optional[str]
+
+
+class TVFBaseModel(BaseModel):
+    idx: int
+    value: Optional[str]
+
+
+class UDFResultsTest(unittest.TestCase):
+
+    def test_udf_returns(self) -> None:
+        # Plain UDF
+        @udf
+        def foo_a(x: str) -> str:
+            return f'0: {x}'
+
+        foo_a_out = foo_a('cat')
+
+        assert type(foo_a_out) is str
+        assert foo_a_out == '0: cat'
+        assert to_sql(foo_a) == '`foo_a`(`x` TEXT NOT NULL) RETURNS TEXT NOT NULL'
+
+        # Vectorized UDF using lists
+        @udf
+        def foo_b(x: List[str]) -> List[str]:
+            return [f'{i}: {y}' for i, y in enumerate(x)]
+
+        foo_b_out = foo_b(['cat', 'dog', 'monkey'])
+
+        assert type(foo_b_out) is list
+        assert foo_b_out == ['0: cat', '1: dog', '2: monkey']
+        assert to_sql(foo_b) == '`foo_b`(`x` TEXT NOT NULL) RETURNS TEXT NOT NULL'
+
+        # Illegal return type for UDF
+        @udf
+        def foo_c(x: List[str]) -> List[UDFTuple]:
+            return [UDFTuple(value='invalid')]
+
+        # Vectorized UDF using pandas Series
+        @udf(args=Parameters, returns=UDFTuple)
+        def foo_d(x: pd.Series) -> pd.Series:
+            return pd.Series([f'{i}: {y}' for i, y in enumerate(x)])
+
+        foo_d_out = foo_d(pd.Series(['cat', 'dog', 'monkey']))
+
+        assert type(foo_d_out) is pd.Series
+        assert list(foo_d_out) == ['0: cat', '1: dog', '2: monkey']
+        assert to_sql(foo_d) == "`foo_d`(`x` TEXT NULL DEFAULT '') RETURNS TEXT NOT NULL"
+
+        # Vectorized UDF using polars Series
+        @udf(args=Parameters, returns=UDFTuple)
+        def foo_e(x: pl.Series) -> pl.Series:
+            return pl.Series([f'{i}: {y}' for i, y in enumerate(x)])
+
+        foo_e_out = foo_e(pl.Series(['cat', 'dog', 'monkey']))
+
+        assert type(foo_e_out) is pl.Series
+        assert list(foo_e_out) == ['0: cat', '1: dog', '2: monkey']
+        assert to_sql(foo_e) == "`foo_e`(`x` TEXT NULL DEFAULT '') RETURNS TEXT NOT NULL"
+
+        # Vectorized UDF using numpy arrays
+        @udf(args=Parameters, returns=UDFTuple)
+        def foo_f(x: np.ndarray) -> np.ndarray:
+            return np.array([f'{i}: {y}' for i, y in enumerate(x)])
+
+        foo_f_out = foo_f(np.array(['cat', 'dog', 'monkey']))
+
+        assert type(foo_f_out) is np.ndarray
+        assert list(foo_f_out) == ['0: cat', '1: dog', '2: monkey']
+        assert to_sql(foo_f) == "`foo_f`(`x` TEXT NULL DEFAULT '') RETURNS TEXT NOT NULL"
+
+        # Vectorized UDF using typed numpy arrays
+        @udf
+        def foo_g(x: npt.NDArray[np.str_]) -> npt.NDArray[np.str_]:
+            return np.array([f'{i}: {y}' for i, y in enumerate(x)])
+
+        foo_g_out = foo_g(np.array(['cat', 'dog', 'monkey']))
+
+        assert type(foo_g_out) is np.ndarray
+        assert list(foo_g_out) == ['0: cat', '1: dog', '2: monkey']
+        assert to_sql(foo_g) == '`foo_g`(`x` TEXT NOT NULL) RETURNS TEXT NOT NULL'
+
+        # Plain TVF using one list
+        @udf
+        def foo_h_(x: str) -> Table[List[str]]:
+            return Table([x] * 3)
+
+        foo_h__out = foo_h_('cat')
+
+        assert type(foo_h__out) is Table
+        assert foo_h__out == Table(['cat', 'cat', 'cat'])
+
+        assert to_sql(foo_h_) == \
+            '`foo_h_`(`x` TEXT NOT NULL) RETURNS TABLE(`a` TEXT NOT NULL)'
+
+        # Plain TVF using multiple lists -- Illegal!
+        @udf
+        def foo_h(x: str) -> Table[List[int], List[str]]:
+            return Table(list(range(3)), [x] * 3)
+
+        foo_h_out = foo_h('cat')
+
+        assert type(foo_h_out) is Table
+        assert foo_h_out == Table([0, 1, 2], ['cat', 'cat', 'cat'])
+
+        with self.assertRaises(TypeError):
+            to_sql(foo_h)
+
+        # Plain TVF using lists of NamedTuples
+        @udf
+        def foo_i(x: str) -> Table[List[TVFTuple]]:
+            return Table([
+                TVFTuple(idx=0, value=x),
+                TVFTuple(idx=1, value=x),
+                TVFTuple(idx=2, value=x),
+            ])
+
+        foo_i_out = foo_i('cat')
+
+        assert type(foo_i_out) is Table
+        assert foo_i_out == Table([
+            TVFTuple(idx=0, value='cat'),
+            TVFTuple(idx=1, value='cat'),
+            TVFTuple(idx=2, value='cat'),
+        ])
+        assert to_sql(foo_i) == (
+            '`foo_i`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using lists of TypedDicts
+        @udf
+        def foo_j(x: str) -> Table[List[TVFDict]]:
+            return Table([
+                dict(idx=0, value=x),
+                dict(idx=1, value=x),
+                dict(idx=2, value=x),
+            ])
+
+        foo_j_out = foo_j('cat')
+
+        assert type(foo_j_out) is Table
+        assert foo_j_out == Table([
+            dict(idx=0, value='cat'),
+            dict(idx=1, value='cat'),
+            dict(idx=2, value='cat'),
+        ])
+        assert to_sql(foo_j) == (
+            '`foo_j`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using lists of pydantic BaseModels
+        @udf
+        def foo_k(x: str) -> Table[List[TVFBaseModel]]:
+            return Table([
+                TVFBaseModel(idx=0, value=x),
+                TVFBaseModel(idx=1, value=x),
+                TVFBaseModel(idx=2, value=x),
+            ])
+
+        foo_k_out = foo_k('cat')
+
+        assert type(foo_k_out) is Table
+        assert foo_k_out == Table([
+            TVFBaseModel(idx=0, value='cat'),
+            TVFBaseModel(idx=1, value='cat'),
+            TVFBaseModel(idx=2, value='cat'),
+        ])
+        assert to_sql(foo_k) == (
+            '`foo_k`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using pandas Series
+        @udf(returns=TVFTuple)
+        def foo_l(x: str) -> Table[pd.Series, pd.Series]:
+            return Table(pd.Series(range(3)), pd.Series([x] * 3))
+
+        foo_l_out = foo_l('cat')
+
+        assert type(foo_l_out) is Table
+        assert len(foo_l_out) == 2
+        assert type(foo_l_out[0]) is pd.Series
+        assert list(foo_l_out[0]) == [0, 1, 2]
+        assert type(foo_l_out[1]) is pd.Series
+        assert list(foo_l_out[1]) == ['cat', 'cat', 'cat']
+        assert to_sql(foo_l) == (
+            '`foo_l`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using polars Series
+        @udf(returns=TVFTuple)
+        def foo_m(x: str) -> Table[pl.Series, pl.Series]:
+            return Table(pl.Series(range(3)), pl.Series([x] * 3))
+
+        foo_m_out = foo_m('cat')
+
+        assert type(foo_m_out) is Table
+        assert len(foo_m_out) == 2
+        assert type(foo_m_out[0]) is pl.Series
+        assert list(foo_m_out[0]) == [0, 1, 2]
+        assert type(foo_m_out[1]) is pl.Series
+        assert list(foo_m_out[1]) == ['cat', 'cat', 'cat']
+        assert to_sql(foo_m) == (
+            '`foo_m`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using pyarrow Array
+        @udf(returns=TVFTuple)
+        def foo_n(x: str) -> Table[pa.Array, pa.Array]:
+            return Table(pa.array(range(3)), pa.array([x] * 3))
+
+        foo_n_out = foo_n('cat')
+
+        assert type(foo_n_out) is Table
+        assert foo_n_out == Table(pa.array([0, 1, 2]), pa.array(['cat', 'cat', 'cat']))
+        assert to_sql(foo_n) == (
+            '`foo_n`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using numpy arrays
+        @udf(returns=TVFTuple)
+        def foo_o(x: str) -> Table[np.ndarray, np.ndarray]:
+            return Table(np.array(range(3)), np.array([x] * 3))
+
+        foo_o_out = foo_o('cat')
+
+        assert type(foo_o_out) is Table
+        assert len(foo_o_out) == 2
+        assert type(foo_o_out[0]) is np.ndarray
+        assert list(foo_o_out[0]) == [0, 1, 2]
+        assert type(foo_o_out[1]) is np.ndarray
+        assert list(foo_o_out[1]) == ['cat', 'cat', 'cat']
+        assert to_sql(foo_o) == (
+            '`foo_o`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using typed numpy arrays
+        @udf
+        def foo_p(x: str) -> Table[npt.NDArray[np.int_], npt.NDArray[np.str_]]:
+            return Table(np.array(range(3)), np.array([x] * 3))
+
+        foo_p_out = foo_p('cat')
+
+        assert type(foo_p_out) is Table
+        assert len(foo_p_out) == 2
+        assert type(foo_p_out[0]) is np.ndarray
+        assert list(foo_p_out[0]) == [0, 1, 2]
+        assert type(foo_p_out[1]) is np.ndarray
+        assert list(foo_p_out[1]) == ['cat', 'cat', 'cat']
+        assert to_sql(foo_p) == (
+            '`foo_p`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`a` BIGINT NOT NULL, `b` TEXT NOT NULL)'
+        )
+
+        # Plain TVF using pandas DataFrame
+        @udf(returns=TVFTuple)
+        def foo_q(x: str) -> Table[pd.DataFrame]:
+            return Table(pd.DataFrame([[0, x], [1, x], [2, x]]))  # columns???
+
+        foo_q_out = foo_q('cat')
+
+        assert type(foo_q_out) is Table
+        assert len(foo_q_out) == 1
+        assert list(foo_q_out[0].iloc[:, 0]) == [0, 1, 2]
+        assert list(foo_q_out[0].iloc[:, 1]) == ['cat', 'cat', 'cat']
+        assert to_sql(foo_q) == (
+            '`foo_q`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using polars DataFrame
+        @udf(returns=TVFTuple)
+        def foo_r(x: str) -> Table[pl.DataFrame]:
+            return Table(pl.DataFrame([[0, 1, 2], [x] * 3]))  # columns???
+
+        foo_r_out = foo_r('cat')
+
+        assert type(foo_r_out) is Table
+        assert len(foo_r_out) == 1
+        assert list(foo_r_out[0][:, 0]) == [0, 1, 2]
+        assert list(foo_r_out[0][:, 1]) == ['cat', 'cat', 'cat']
+        assert to_sql(foo_r) == (
+            '`foo_r`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Plain TVF using pyarrow Table
+        @udf(returns=TVFTuple)
+        def foo_s(x: str) -> Table[pa.Table]:
+            return Table(
+                pa.Table.from_pylist([
+                    dict(idx=0, value='cat'),
+                    dict(idx=1, value='cat'),
+                    dict(idx=2, value='cat'),
+                ]),
+            )  # columns???
+
+        foo_s_out = foo_s('cat')
+
+        assert type(foo_s_out) is Table
+        assert foo_s_out == Table(
+            pa.Table.from_pylist([
+                dict(idx=0, value='cat'),
+                dict(idx=1, value='cat'),
+                dict(idx=2, value='cat'),
+            ]),
+        )
+        assert to_sql(foo_s) == (
+            '`foo_s`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Vectorized TVF using lists -- Illegal!
+        @udf
+        def foo_t(x: List[str]) -> Table[List[int], List[str]]:
+            return Table(list(range(len(x))), x)
+
+        foo_t_out = foo_t(['cat', 'dog', 'monkey'])
+
+        assert type(foo_t_out) is Table
+        assert foo_t_out == Table([0, 1, 2], ['cat', 'dog', 'monkey'])
+        with self.assertRaises(TypeError):
+            to_sql(foo_t)
+
+        # Vectorized TVF using pandas Series
+        @udf(args=Parameters, returns=TVFTuple)
+        def foo_u(x: pd.Series) -> Table[pd.Series, pd.Series]:
+            return Table(pd.Series(range(len(x))), pd.Series(x))
+
+        foo_u_out = foo_u(pd.Series(['cat', 'dog', 'monkey']))
+
+        assert type(foo_u_out) is Table
+        assert len(foo_u_out) == 2
+        assert list(foo_u_out[0]) == [0, 1, 2]
+        assert list(foo_u_out[1]) == ['cat', 'dog', 'monkey']
+        assert to_sql(foo_u) == (
+            "`foo_u`(`x` TEXT NULL DEFAULT '') "
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Vectorized TVF using polars Series
+        @udf(args=Parameters, returns=TVFTuple)
+        def foo_v(x: pl.Series) -> Table[pl.Series, pl.Series]:
+            return Table(pl.Series(range(len(x))), pl.Series(x))
+
+        foo_v_out = foo_v(pl.Series(['cat', 'dog', 'monkey']))
+
+        assert type(foo_v_out) is Table
+        assert len(foo_v_out) == 2
+        assert list(foo_v_out[0]) == [0, 1, 2]
+        assert list(foo_v_out[1]) == ['cat', 'dog', 'monkey']
+        assert to_sql(foo_v) == (
+            "`foo_v`(`x` TEXT NULL DEFAULT '') "
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Vectorized TVF using pyarrow Array
+        @udf(args=Parameters, returns=TVFTuple)
+        def foo_w(x: pa.Array) -> Table[pa.Array, pa.Array]:
+            return Table(pa.array(range(len(x))), pa.array(x))
+
+        foo_w_out = foo_w(pa.array(['cat', 'dog', 'monkey']))
+
+        assert type(foo_w_out) is Table
+        assert foo_w_out == Table(
+            pa.array([0, 1, 2]), pa.array(['cat', 'dog', 'monkey']),
+        )
+        assert to_sql(foo_w) == (
+            "`foo_w`(`x` TEXT NULL DEFAULT '') "
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Vectorized TVF using numpy arrays
+        @udf(args=Parameters, returns=TVFTuple)
+        def foo_x(x: np.ndarray) -> Table[np.ndarray, np.ndarray]:
+            return Table(np.array(range(len(x))), np.array(x))
+
+        foo_x_out = foo_x(np.array(['cat', 'dog', 'monkey']))
+
+        assert type(foo_x_out) is Table
+        assert len(foo_x_out) == 2
+        assert list(foo_x_out[0]) == [0, 1, 2]
+        assert list(foo_x_out[1]) == ['cat', 'dog', 'monkey']
+        assert to_sql(foo_x) == (
+            "`foo_x`(`x` TEXT NULL DEFAULT '') "
+            'RETURNS TABLE(`idx` BIGINT NOT NULL, `value` TEXT NULL)'
+        )
+
+        # Vectorized TVF using typed numpy arrays
+        @udf
+        def foo_y(
+            x: npt.NDArray[np.str_],
+        ) -> Table[npt.NDArray[np.int_], npt.NDArray[np.str_]]:
+            return Table(np.array(range(len(x))), np.array(x))
+
+        foo_y_out = foo_y(np.array(['cat', 'dog', 'monkey']))
+
+        assert type(foo_y_out) is Table
+        assert len(foo_y_out) == 2
+        assert list(foo_y_out[0]) == [0, 1, 2]
+        assert list(foo_y_out[1]) == ['cat', 'dog', 'monkey']
+        assert to_sql(foo_y) == (
+            '`foo_y`(`x` TEXT NOT NULL) '
+            'RETURNS TABLE(`a` BIGINT NOT NULL, `b` TEXT NOT NULL)'
+        )
+
+
+if __name__ == '__main__':
+    unittest.main()

{singlestoredb-1.12.4.dist-info → singlestoredb-1.13.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: singlestoredb
-Version: 1.12.4
+Version: 1.13.1
 Summary: Interface to the SingleStoreDB database and workspace management APIs
 Home-page: https://github.com/singlestore-labs/singlestoredb-python
 Author: SingleStore
@@ -22,6 +22,7 @@ Requires-Dist: setuptools
 Requires-Dist: sqlparams
 Requires-Dist: wheel
 Requires-Dist: tomli>=1.1.0; python_version < "3.11"
+Requires-Dist: typing-extensions<=4.13.2; python_version < "3.11"
 Provides-Extra: dataframe
 Requires-Dist: ibis-singlestoredb; extra == "dataframe"
 Provides-Extra: dbt

{singlestoredb-1.12.4.dist-info → singlestoredb-1.13.1.dist-info}/RECORD

@@ -1,33 +1,37 @@
-singlestoredb/__init__.py,sha256=k2RhlWhHq2KGERDJKr9bjgAV5KCrLJl2vcEyfWJriGw,1649
+singlestoredb/__init__.py,sha256=TlTiqjO5Ea2FP3Iq8Kmk0zAA2vR6oOj-HjURaUcQKcM,1649
 singlestoredb/auth.py,sha256=u8D9tpKzrqa4ssaHjyZnGDX1q8XBpGtuoOkTkSv7B28,7599
-singlestoredb/config.py,sha256=0XooNOf1dUee26C0io7unfBbdXUyfDIYc_mSpxsVsTI,12592
+singlestoredb/config.py,sha256=dayUWwSy2YdgmhF8tzH-7FwFpwon5bgX_VeX-Yu5ia4,12969
 singlestoredb/connection.py,sha256=0HEpjBZXLqQwOTEfveMkgej1H3Kyof47prIHvJJZtoo,45831
 singlestoredb/converters.py,sha256=Ui-AqdW3pRAQ8A_YcK9EqVYyM4Pt1_Q-tjlotbpK6Cw,20686
 singlestoredb/exceptions.py,sha256=HuoA6sMRL5qiCiee-_5ddTGmFbYC9Euk8TYUsh5GvTw,3234
 singlestoredb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/pytest.py,sha256=OyF3BO9mgxenifYhOihnzGk8WzCJ_zN5_mxe8XyFPOc,9074
 singlestoredb/types.py,sha256=FIqO1A7e0Gkk7ITmIysBy-P5S--ItbMSlYvblzqGS30,9969
-singlestoredb/ai/__init__.py,sha256=7Pubobzx5OlyepNo5DOOxWev1DUW9WFc9P6Qver2xpY,60
-singlestoredb/ai/embeddings.py,sha256=3jghE4WMf7vy8RobhrMOLvMLnDNGbkPCF48B3fGM38U,746
+singlestoredb/ai/__init__.py,sha256=-uNcq-bY-AiWhZ5Plq2ZXtfIVL4PaifMJsJf58rdN8I,114
+singlestoredb/ai/chat.py,sha256=8OSBZJ3J2zOlVXzJ_sHSAAyu5E6sy7jqqiNeFhtmjOI,802
+singlestoredb/ai/embeddings.py,sha256=X3g0sJNDVOzXzZwoXz3M3ch-IERQXNkHxuH4cj125I8,815
 singlestoredb/alchemy/__init__.py,sha256=dXRThusYrs_9GjrhPOw0-vw94in_T8yY9jE7SGCqiQk,2523
-singlestoredb/apps/__init__.py,sha256=uuEH2WZ1ROpmkMBBdz1tSkQSdYR9blXXU2nn7E5P4qQ,118
+singlestoredb/apps/__init__.py,sha256=dfN97AZz7Np6JML3i9GJrv22ZbNCUletXmsJpQnKhKg,170
 singlestoredb/apps/_cloud_functions.py,sha256=NJJu0uJsK9TjY3yZjgftpFPR-ga-FrOyaiDD4jWFCtE,2704
-singlestoredb/apps/_config.py,sha256=w21kH0jMJ0_cP_VgAxHhKiLW5Iyrr7xzUIxRmfe_fqs,2118
-singlestoredb/apps/_connection_info.py,sha256=gQPYzJrBQUEH76zVTkxJ7FAypNoN2T7GYHVOSgJ7Q8Q,175
+singlestoredb/apps/_config.py,sha256=FlV0ABP7qlBJoKo9NOme6Fpp4yUFm5QEpHEHbl1A24o,2441
+singlestoredb/apps/_connection_info.py,sha256=QOr-wcQJn6oCZw2kLEP0Uwzo85CGolGz0QIvlem3gug,303
 singlestoredb/apps/_dashboards.py,sha256=_03fI-GJannamA5lxLvIoC6Mim-H1jTRuI8-dw_P--k,1474
 singlestoredb/apps/_process.py,sha256=G37fk6bzIxzhfEqp2aJBk3JCij-T2HFtTd078k5Xq9I,944
+singlestoredb/apps/_python_udfs.py,sha256=-CT0hzfHGnbxWkOKS8gQBJa3Kq6ixT_gn18TGtLtK_k,2726
 singlestoredb/apps/_stdout_supress.py,sha256=8s9zMIIRPpeu44yluJFc_0VueAxZDmr9QVGT6TGiFeY,659
 singlestoredb/apps/_uvicorn_util.py,sha256=rEK4nEmq5hbpRgsmK16UVlxe2DyQSq7C5w5WZSp0kX8,962
-singlestoredb/functions/__init__.py,sha256=nPaLVgtb5XDxbRucDFFwjePPh4n40_6jcbxE8HPebkQ,82
-singlestoredb/functions/decorator.py,sha256=0gfYVCi_GvxKcuDmAcpfIFTTElLD0CBT8RYN_dUrEvU,11485
-singlestoredb/functions/dtypes.py,sha256=a2vevIug8NhiUCFiSOKwRPpdWU69Gn13ZoQ6Aovskhc,31408
-singlestoredb/functions/signature.py,sha256=qjdCBxtJvNnDWCAruX9kFNuYuGYVCWGEJ-heKRu4MXk,22551
+singlestoredb/functions/__init__.py,sha256=I2GnxOhLb4_7xhgOxdIwmwD5NiK7QYPYaE3PUIX-7xk,471
+singlestoredb/functions/decorator.py,sha256=gylwivCwpNMCUmgBEUEYf2ogIpInNhu6IFeTU82W7Ko,5433
+singlestoredb/functions/dtypes.py,sha256=DgJaNXouJ2t-qIqDiQlUYU9IhkXXUTigWeE_MAcmvHM,39814
+singlestoredb/functions/signature.py,sha256=avErza5t3p0vy94p4yjw7Hy2cCDvjolwCyYjEI0PKXM,42481
+singlestoredb/functions/typing.py,sha256=gT_Sz5YH-L-9WeIHwWYMEx-hUCZqis7ec5Ipk3JXpnM,1339
+singlestoredb/functions/utils.py,sha256=1L0Phgzq0XdWK3ecfOOydq4zV955yCwpDoAaCYRGldk,10769
 singlestoredb/functions/ext/__init__.py,sha256=1oLL20yLB1GL9IbFiZD8OReDqiCpFr-yetIR6x1cNkI,23
 singlestoredb/functions/ext/arrow.py,sha256=WB7n1ACslyd8nlbFzUvlbxn1BVuEjA9-BGBEqCWlSOo,9061
-singlestoredb/functions/ext/asgi.py,sha256=AK7dxe_m8TnFh7IuKKzKO1lCyfJVzs4hRJKW1vTO130,44073
-singlestoredb/functions/ext/json.py,sha256=XkI8jirxi1T9F-M0p9NpLezph0MRAhYmDiPuU2Id0Uo,10404
-singlestoredb/functions/ext/mmap.py,sha256=OB6CIYoLe_AYuJM10lE0I6QhZJ5kMhLNbQo2Sp1wiZA,13711
-singlestoredb/functions/ext/rowdat_1.py,sha256=JgKRsVSQYczFD6cmo2xLilbNPYpyLL2tPOWO1Gh25ow,22306
+singlestoredb/functions/ext/asgi.py,sha256=CSjGB8YnBYf0Ca4qMjl25AG1ExBmmnzxKTFAogi_mDc,51874
+singlestoredb/functions/ext/json.py,sha256=RIuZdDybEdHuC-f2p6BdjhFjM3iGb3a1PRQ4k11P6N8,10102
+singlestoredb/functions/ext/mmap.py,sha256=RzyNSLRpI5ZJ8YN6k-AvZlRTLjj80j52byHLtW8c3ps,13710
+singlestoredb/functions/ext/rowdat_1.py,sha256=SlXbJ2042jEoaXw81y5llw1625w0aU2nZ8vI_O3qA-M,21112
 singlestoredb/functions/ext/utils.py,sha256=2-B8YU_Iekv8JcpI-ochs9TIeuyatLaLAH-AyYyUUIg,5311
 singlestoredb/fusion/__init__.py,sha256=Qo7SuqGw-l-vE8-EI2jhm6hXJkYfOLUKIws9c7LFNX0,356
 singlestoredb/fusion/graphql.py,sha256=ZA3HcDq5rER-dCEavwTqnF7KM0D2LCYIY7nLQk7lSso,5207
@@ -43,7 +47,7 @@ singlestoredb/fusion/handlers/stage.py,sha256=kYVjbPys83kf3jX6jWwN8Ju0oEocKVZ3TI
 singlestoredb/fusion/handlers/utils.py,sha256=ozHOWUraoN8XGTK9JZdhv5HV8AQR8zfUd1yh1kLvUXY,10685
 singlestoredb/fusion/handlers/workspace.py,sha256=4xN2TFO4yF7KZB2Fcht7IuvoDdAT6fDfDLjixiHZN8w,27506
 singlestoredb/http/__init__.py,sha256=A_2ZUCCpvRYIA6YDpPy57wL5R1eZ5SfP6I1To5nfJ2s,912
-singlestoredb/http/connection.py,sha256=dNn6OAjCSueR_CIv1T28oDr_TkqXUV05yv1t-XoPrYE,39814
+singlestoredb/http/connection.py,sha256=EgE2m_nxisGPo6YV3AJd-RRafdT0f70HRbIo1ONQ668,39893
 singlestoredb/magics/__init__.py,sha256=lZjkT3Webo9c1EQAzlRCRh6B2pckQH8uvNrrB__abcI,1210
 singlestoredb/magics/run_personal.py,sha256=2f7u1T7iblxGzZurHNgNXLrPBvsvPADZKo_RD_IjYuE,1844
 singlestoredb/magics/run_shared.py,sha256=SI8dCBRMaGn-xZU7dto4jsAqKBi-Ll14htUsMUSBpJM,1752
@@ -52,11 +56,12 @@ singlestoredb/management/billing_usage.py,sha256=9ighjIpcopgIyJOktBYQ6pahBZmWGHO
 singlestoredb/management/cluster.py,sha256=h75grXSxq4Anr4RxwKxcZW4TkWJ4bFg_ql5iRWCNLdQ,14405
 singlestoredb/management/export.py,sha256=jJCe25ecH_LzKSDc7vS1-5DQaWFrZipeawLPpArByJE,5108
 singlestoredb/management/files.py,sha256=89IhpGw9WdwxVeksavHEDMVn9wb_jxb-utZuIDqkLHw,30477
+singlestoredb/management/inference_api.py,sha256=L6eFqaUaPugF_cmrZ4xlArj8CIv25vWqQs1vwgKPEF4,2583
 singlestoredb/management/job.py,sha256=4-xLWzbE8odQogVVaFer80UEoTAZY1T28VZ9Ug4rbmM,24611
-singlestoredb/management/manager.py,sha256=X29VEHlUEzmWvGo_bQMzo8a6d4nYMLE1CewlNBjrD7M,8851
-singlestoredb/management/organization.py,sha256=hqMaM7H-naMjNbxDl_f7G_2o5TkiGKyzPhxuzDveJAw,5402
+singlestoredb/management/manager.py,sha256=V9_PVMpUOj8laKwNFtp4Nd2Taww2Y65TeSRK5ZWzOo0,8922
+singlestoredb/management/organization.py,sha256=_JvW0Znu5emR5uYGVEcZvakQqftNb_vRhzmkOoPRPfc,5869
 singlestoredb/management/region.py,sha256=HnLcWUh7r_aLECliplCDHak4a_F3B7LOSXEYMW66qD0,1611
-singlestoredb/management/utils.py,sha256=P4fp8a7EwaYiag_hvpILcgwXtdFNYKKO70dsKjmxn1A,13171
+singlestoredb/management/utils.py,sha256=QIhZCZSRaDbAG35xu1_n7ihmRXON8swc-gEK2FGYutI,13203
 singlestoredb/management/workspace.py,sha256=ze-eE-cO3JCrR3uttVFaBOndDbEE8_qWR2kzOjzbKaY,56234
 singlestoredb/mysql/__init__.py,sha256=olUTAvkiERhDW41JXQMawkg-i0tvBEkoTkII1tt6lxU,4492
 singlestoredb/mysql/_auth.py,sha256=AugRitoUwgRIDFuJxuAH4MWIAmckY7Ji2pP6r_Ng9dY,8043
@@ -117,18 +122,19 @@ singlestoredb/tests/test_config.py,sha256=63lyIQ2KrvGE6C9403B_4Mc90mX4tp42ys5Bih
 singlestoredb/tests/test_connection.py,sha256=fvn-kPdeIMI9RGNz0dNk5ZmTCep1amwWQDHYfPdqO60,119699
 singlestoredb/tests/test_dbapi.py,sha256=IKq5Hcwx8WikASP8_AB5fo3TXv7ryWPCVGonoly00gI,652
 singlestoredb/tests/test_exceptions.py,sha256=tfr_8X2w1UmG4nkSBzWGB0C7ehrf1GAVgj6_ODaG-TM,1131
-singlestoredb/tests/test_ext_func.py,sha256=OWd-CJ1Owhx72nikSWWEF2EQFCJk7vEXZM2Oy9EbYQo,37357
+singlestoredb/tests/test_ext_func.py,sha256=s1k1cBxQ7vIS1zSrKGkKTgLZE1DT_Rqj-3VNSCSv68I,43261
 singlestoredb/tests/test_ext_func_data.py,sha256=yTADD93nPxX6_rZXXLZaOWEI_yPvYyir9psn5PK9ctU,47695
 singlestoredb/tests/test_fusion.py,sha256=EH1mRwdX2Fajsq6x2l0gBhH1YhcxtvDGIKC9HJ4sDbQ,50521
 singlestoredb/tests/test_http.py,sha256=RXasTqBWRn__omj0eLFTJYIbZjd0PPdIV2d4Cqz0MC8,8580
-singlestoredb/tests/test_management.py,sha256=9WKaFChAWkKMm0kDhwTgjaJIAcga7mQZiF0x9BNI1Tc,45112
+singlestoredb/tests/test_management.py,sha256=6evsyQWA-lOKMehJi8xvjp0udm85EHBHuZDHwQEzxPg,45149
 singlestoredb/tests/test_plugin.py,sha256=qpO9wmWc62VaijN1sJ97YSYIX7I7Y5C6sY-WzwrutDQ,812
 singlestoredb/tests/test_results.py,sha256=wg93sujwt-R9_eJCgSCElgAZhLDkIiAo3qPkPydOv78,6582
 singlestoredb/tests/test_types.py,sha256=jqoAaSjhbgwB3vt0KsTcl7XBWoMMIa0mPFKhEi5bBjo,4500
-singlestoredb/tests/test_udf.py,sha256=G6MgAzu5ZiMHmtHaGbWYXRZ-naEvwYzT5MRB900BU3I,30029
+singlestoredb/tests/test_udf.py,sha256=Kb7-oJpnN6MTT3aE5V5dry_r5ze0EwaAIJeh_zR3l0I,28844
+singlestoredb/tests/test_udf_returns.py,sha256=k31L6Ir2Xw8MEZ18upuu0p_D_OpbrPAzWhDQXVFDS7I,15541
 singlestoredb/tests/test_xdict.py,sha256=fqHspoi39nbX3fIDVkkRXcd5H50xdOsSvK0bxAMQnaE,10408
 singlestoredb/tests/utils.py,sha256=2A2tEdD3t8aXWUnHtAIcFlWrflsz2MlMcCbUDaAG29c,4995
-singlestoredb/tests/ext_funcs/__init__.py,sha256=qZLnDI_Ck0tguVi-K-BKXDHAcC0jui3dsm93Djj4x08,9290
+singlestoredb/tests/ext_funcs/__init__.py,sha256=gtyhykoEk8_-il5ukTwvqDu-4D1LgwxMFseYg1wgOHo,14103
 singlestoredb/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/utils/config.py,sha256=m3Xn6hsbdKyLufSnbokhFJ9Vfaz9Qpkj1IEnIiH9oJQ,24503
 singlestoredb/utils/convert_rows.py,sha256=A6up7a8Bq-eV2BXdGCotQviqp1Q7XdJ2MA9339hLYVQ,1816
@@ -140,9 +146,9 @@ singlestoredb/utils/results.py,sha256=bJtaUaDiFq26IsPAKZ2FHGB7csMn94EAxLKrP4HaEE
 singlestoredb/utils/xdict.py,sha256=S9HKgrPrnu_6b7iOwa2KrW8CmU1Uqx0BWdEyogFzWbE,12896
 sqlx/__init__.py,sha256=aBYiU8DZXCogvWu3yWafOz7bZS5WWwLZXj7oL0dXGyU,85
 sqlx/magic.py,sha256=JsS9_9aBFaOt91Torm1JPN0c8qB2QmYJmNSKtbSQIY0,3509
-singlestoredb-1.12.4.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
-singlestoredb-1.12.4.dist-info/METADATA,sha256=YmOcqvh_bJwEzfcaVM-lG6ane49lQwXy2TJdl0t42Rc,5622
-singlestoredb-1.12.4.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-singlestoredb-1.12.4.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
-singlestoredb-1.12.4.dist-info/top_level.txt,sha256=DfFGz7bM4XrshloiCeTABgylT3BUnS8T5pJam3ewT6Q,19
-singlestoredb-1.12.4.dist-info/RECORD,,
+singlestoredb-1.13.1.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
+singlestoredb-1.13.1.dist-info/METADATA,sha256=rybhqtR5TzE7p-keWvO6V21MlYNZZtIULT9atDRgXpw,5688
+singlestoredb-1.13.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+singlestoredb-1.13.1.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
+singlestoredb-1.13.1.dist-info/top_level.txt,sha256=DfFGz7bM4XrshloiCeTABgylT3BUnS8T5pJam3ewT6Q,19
+singlestoredb-1.13.1.dist-info/RECORD,,