singlestoredb 1.12.4__cp38-abi3-win_amd64.whl → 1.13.1__cp38-abi3-win_amd64.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,34 +1,38 @@
-_singlestoredb_accel.pyd,sha256=BjwcTKJQb9p7VofVO-Lrg_BXA6nM4nvVUNyKF9WkU7c,59392
-singlestoredb/__init__.py,sha256=Y3oYr5E9O4QiB6TxEq0HF58H-agjoOb93fc2WyLXPBg,1712
+_singlestoredb_accel.pyd,sha256=n4oYggBtLgFHDfEEd8c01AHMlcvUrv2eTga5XAF978g,61440
+singlestoredb/__init__.py,sha256=M6sWPnnrYVitraV7HD5cUx71dFDnwnRDUaDyoOAJPEU,1712
 singlestoredb/auth.py,sha256=RmYiH0Wlc2RXc4pTlRMysxtBI445ggCIwojWKC_eDLE,7844
-singlestoredb/config.py,sha256=7-M2c7IAv3B1AmAKizgeNn880TDBbM4gbElLfStejIU,13035
+singlestoredb/config.py,sha256=t3aiWi1i3kT5VhEgXca0gwT6591YkZUed-wzvVEBMs0,13424
 singlestoredb/connection.py,sha256=Ty_idVYH50Qx-j8WXy7NeB-DYLAcpdjGYTrTHkKzG9U,47309
 singlestoredb/converters.py,sha256=6gN3_RzSbw0Aimd5cGgBNPNq1yiHb1a_NK8qC9DmOQ0,21636
 singlestoredb/exceptions.py,sha256=WCCJrNSsU-hD-621Jpd6bwmvGftQ7byXkk-XKXlaxpg,3354
 singlestoredb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/pytest.py,sha256=TH364xRCN7_QaN0oRQDHixrEcDx_ZBgu3bmY0tvKrYU,9357
 singlestoredb/types.py,sha256=Lv0BEQl6aSZBiAe0OSI07FEJhcHZ9HX45iT9NU_mxHQ,10334
-singlestoredb/ai/__init__.py,sha256=nT048t90xqjaNhz7KJ10KfSVW4RcZRoujyC6po6Nmb8,61
-singlestoredb/ai/embeddings.py,sha256=KVvQY3viyYWXDBobFpj0xqiGRijt36zcHHlPNAfFAxA,770
+singlestoredb/ai/__init__.py,sha256=5vlx0XpzxalMKySnVF7y40gfuCgaz7COUKqN4KfNKF8,116
+singlestoredb/ai/chat.py,sha256=oDig8C8QdPEHL-JmmpdFvt_Ct7-K_D0pG_UJ00WCZ7Y,828
+singlestoredb/ai/embeddings.py,sha256=2F6ALJ5Z0PKih3mt3DIFepABcC3u0lshtWu2OUXWZgg,842
 singlestoredb/alchemy/__init__.py,sha256=bUmCl1xUn2v36RMbXLIrvgKzZSqx71mp1ReUw9JeVA8,2613
-singlestoredb/apps/__init__.py,sha256=IKQrPKb1d_LQvmr7jXQvgPRrB5Ja_1kGAXwYvdjp6ok,120
+singlestoredb/apps/__init__.py,sha256=7l4d4hCtm1ykDNf7UBi3Qnqg9N0qPs5jbQ0Al5tS5aM,173
 singlestoredb/apps/_cloud_functions.py,sha256=DMRC-4z3Q52hsKb_WlolfNcYV-5XmQGiJWbbaUxFZ0s,2794
-singlestoredb/apps/_config.py,sha256=kMNpkc-RtPrk_XW0kLAXeF444WIlBXzI8dLnFgIW5Jw,2184
-singlestoredb/apps/_connection_info.py,sha256=t8hFOSRALXt5tqvDX0fholKoT148xktE5r__SYdH6Dk,185
+singlestoredb/apps/_config.py,sha256=b_Op6KjSJdPwym-AlHcy0dXLHiks1cL8Q2ea1W8r3NA,2513
+singlestoredb/apps/_connection_info.py,sha256=P9rW4t2k3QFk3A34dIg9DbCWBqrcHDcZPi2Q9r2-o3A,321
 singlestoredb/apps/_dashboards.py,sha256=qEdDivjwS68Uukay0Qw-3awHZFpkcqapzd3vLaVUzWo,1521
 singlestoredb/apps/_process.py,sha256=eMiBO4piaRX1S6zdnMx0X0E4J7E1XrXndnVW0GRYq1Y,976
+singlestoredb/apps/_python_udfs.py,sha256=PW-Lai3dFINsNtqrLlyUafWgs1j86qGqaNZpUD9OxVI,2811
 singlestoredb/apps/_stdout_supress.py,sha256=QRV-IHQQMvWMeJfqORuVE2-Il6ohO2Ti4IokFoTCJWE,689
 singlestoredb/apps/_uvicorn_util.py,sha256=Petkmq5keBPfXZsHBrnZfY3O2rUHvb3Cw6o-BRz5MP0,994
-singlestoredb/functions/__init__.py,sha256=Ehyp1pa40cvizzSYNGZ4UP4tiEcyfaxf_LI-oyt-Lro,84
-singlestoredb/functions/decorator.py,sha256=6c0uIknQNZrN5LAj-mWUw25Bcnl-_dRel6RMDzn2vK8,11816
-singlestoredb/functions/dtypes.py,sha256=5IwMSaSzxtSowxXrm5hZXW1lpNm6QILxiU4mAUEkBO0,32854
-singlestoredb/functions/signature.py,sha256=0d1mqH1J0cV7W1NBAqM2C98DWIZomF5qYSS5-T6OW_g,23324
+singlestoredb/functions/__init__.py,sha256=KPBjRaiVipCQwTSsryHvBE_qrwy7Kj74lKnutqplcao,487
+singlestoredb/functions/decorator.py,sha256=Zw4mTSzpTxFTdqfrFzdt_8hFptg0PXcrB3TbUMNynyU,5614
+singlestoredb/functions/dtypes.py,sha256=7w_atIL5jAvDNtu6RDCvY440Y9U-p19_Nf7R5ki46Co,41607
+singlestoredb/functions/signature.py,sha256=K1WftlfVei2xmizCxHi91z7mK8vW5VJxJAM7FiCnpEY,43926
+singlestoredb/functions/typing.py,sha256=5AJG4nx-HKCeemNxL0qc1VunYPJ5lHRzpYAK_qMybNw,1380
+singlestoredb/functions/utils.py,sha256=lZPxdYfHxrSfxGWCoF0YZyakVy2iYlozJ1lPSaPKRlo,11190
 singlestoredb/functions/ext/__init__.py,sha256=5ppI8IZN_zOwoJFdu_Oq9ipxtyHw9n6OMVAa_s9T_yY,24
 singlestoredb/functions/ext/arrow.py,sha256=mQhwaMpvCH_dP92WIhP_j-stu272n4UAHsFUOBTgnq0,9436
-singlestoredb/functions/ext/asgi.py,sha256=TNQlzosguNaM1-Rs1XXO78phav25XdKsbLTFpIvXgRA,45362
-singlestoredb/functions/ext/json.py,sha256=CROdj37cuJhAZ-CM93EI-SoLb4kxFcMGudsJ5QGyCoI,10831
-singlestoredb/functions/ext/mmap.py,sha256=zo6eweOFCZp0KIzAeL1vvuSjqvQhE8ybVhHbU0ZICt4,14124
-singlestoredb/functions/ext/rowdat_1.py,sha256=KYj_y5JWm3_B2-QC47HK-CNOrzujBqGUwLJfE49jwRg,23050
+singlestoredb/functions/ext/asgi.py,sha256=vFX-7P7aVItK76AxBw-DjLGy0nDV9KzcJYyLsLRCsi0,53455
+singlestoredb/functions/ext/json.py,sha256=j9133xOpyuSqb8smBmi_bPvv6OYCbNfpbLbEicyGqmQ,10522
+singlestoredb/functions/ext/mmap.py,sha256=0BN9OyEONZ174qdZWe2m3Xykt3-QcxyLYBt2iCG772Q,14123
+singlestoredb/functions/ext/rowdat_1.py,sha256=UNMMUA8mb6iIRfJV2FsdA20Sw6s-LEdHQ_tC4K4g70Q,21836
 singlestoredb/functions/ext/utils.py,sha256=OPMFD-tTCx2Kk9jguQkrTr7e4AgNkt15YsvaT1YSmN8,5480
 singlestoredb/fusion/__init__.py,sha256=FHWtrg6OJFTf6Ye197V5sU6ssryr2h6FBcDIgXP7-H4,367
 singlestoredb/fusion/graphql.py,sha256=SHqsPe4xgawdsTPHEtJGQlybYGWqPrGMmyK-v20RLac,5420
@@ -44,7 +48,7 @@ singlestoredb/fusion/handlers/stage.py,sha256=PP-SSP204lwpmnycSXXSmFPzoN535JVuwg
 singlestoredb/fusion/handlers/utils.py,sha256=nV2lSzKhv7CzM7I_uIh5kmDV0Ec6VeeKoHczx5pVNcw,11009
 singlestoredb/fusion/handlers/workspace.py,sha256=NxoEY5xd5lCQmXiim4nhAYCL0agHo1H_rGPpqa31hiw,28397
 singlestoredb/http/__init__.py,sha256=4cEDvLloGc3LSpU-PnIwacyu0n5oIIIE6xk2SPyWD_w,939
-singlestoredb/http/connection.py,sha256=kLA-LA4zinbNOSemRbGqbTHd3bNn-ucOwmQZa8HIKPI,41075
+singlestoredb/http/connection.py,sha256=pYewBJsoMLVNw5HMtSECQWHwPQ1qf6onwz4BmQ01_TU,41156
 singlestoredb/magics/__init__.py,sha256=fqCBQ0s8o1CYE4Xo_XiSbkLDzLgMNDgpSkOx66-uDZw,1244
 singlestoredb/magics/run_personal.py,sha256=M11xHi9lWquh_pLSpFI89LGE7PhOPQOGqlSPDl48itE,1900
 singlestoredb/magics/run_shared.py,sha256=rnKpW4d8CJvD6ehK8jG8FlxuqZvjZl4KocPTsk-23O8,1805
@@ -53,11 +57,12 @@ singlestoredb/management/billing_usage.py,sha256=0UHFSPCrN0nyeGFFM-HXS3NP8pYmYo2
 singlestoredb/management/cluster.py,sha256=auBzNYIXvnI6rq3DNpPgJhwWoT6JsyZRikjpON23Pxg,14867
 singlestoredb/management/export.py,sha256=2dCvnTnkxVI-arX3_375DtWzHkI1YNK5zaFYdHKE4cs,5277
 singlestoredb/management/files.py,sha256=Z9GpS2EHf9atE8kJdz1vJtsiT80O6TV00MPhqyXfAAw,31579
+singlestoredb/management/inference_api.py,sha256=9d9-7edoZ6JI3SPvStcVDOSHOY6l38V1MFpyskdLAZY,2684
 singlestoredb/management/job.py,sha256=Npfe1JLYJlggGBrXLniPKwKUKF1i3alvSY1SFtvauSs,25498
-singlestoredb/management/manager.py,sha256=8zU0d7NG83PYMhoAs2JriTqbqh-R2tLX7VZoeZtcogY,9148
-singlestoredb/management/organization.py,sha256=JBsNC4R3boUKdYvyCZyfGoVMC1mD6SPuMI1UssBVoOM,5611
+singlestoredb/management/manager.py,sha256=iB4XcerOPNoFc04TN40vjosMrDhlE7HMUyHsseRgjBQ,9224
+singlestoredb/management/organization.py,sha256=viFG8eLVOs-NeoL6zm8nypFRQ-oiRDD2Sk-bL2b6hvw,6095
 singlestoredb/management/region.py,sha256=oGoLLS88dE1GmY7GCc0BV7X3f7bWwKQyeXOVBFmK9Pk,1678
-singlestoredb/management/utils.py,sha256=BP-Wb8Sg16GbdLI_DeBz-3ttMklz6ZjYyMOz-sxElz8,13594
+singlestoredb/management/utils.py,sha256=RtFhdIIliQ6aulYs99fgAQ0FxL2LfV-5oPRd9s_bBok,13626
 singlestoredb/management/workspace.py,sha256=D9DzpeWU7xFjpj8bBYiXyasjVYVANeYjTzgkz2aKvJQ,57984
 singlestoredb/mysql/__init__.py,sha256=CbpwzNUJPAmKPpIobC0-ugBta_RgHCMq7X7N75QLReY,4669
 singlestoredb/mysql/_auth.py,sha256=YaqqyvAHmeraBv3BM207rNveUVPM-mPnW20ts_ynVWg,8341
@@ -118,18 +123,19 @@ singlestoredb/tests/test_config.py,sha256=Ad0PDmCnJMOyy9f7WTKiRasSR_3mYRByUlSb7k
 singlestoredb/tests/test_connection.py,sha256=UmoNo8erkcifEMbHZl83yAaRsyh6HANRdEtY3aViOK8,122792
 singlestoredb/tests/test_dbapi.py,sha256=cNJoTEZvYG7ckcwT7xqlkJX-2TDEYGTDDU1Igucp48k,679
 singlestoredb/tests/test_exceptions.py,sha256=vscMYmdOJr0JmkTAJrNI2w0Q96Nfugjkrt5_lYnw8i0,1176
-singlestoredb/tests/test_ext_func.py,sha256=gQErR-wAN8BqLNG5U4pNbg4qkQEo6Re8Hd9_Ztqo1RM,38550
+singlestoredb/tests/test_ext_func.py,sha256=LhuPz8o3UF7x2LNod5oZ1tlxeLvGDEUE5FnzdsIbSPs,44643
 singlestoredb/tests/test_ext_func_data.py,sha256=9kn8BWmCjkbnP6hSbFhmhcdW4OmVT-GSvBTIzFBLEys,48796
 singlestoredb/tests/test_fusion.py,sha256=S0Jk2NrcOitqM98r5fosHGbZ1sCZ2uxar5t48v-uOD0,52045
 singlestoredb/tests/test_http.py,sha256=7hwXe61hlUes3nji0MTTZweo94tJAlJ-vA5ct9geXFQ,8868
-singlestoredb/tests/test_management.py,sha256=EJY-JDFNS83_pM0KKJlEHoRiEZWHxjK2XWtM0YAy63U,46462
+singlestoredb/tests/test_management.py,sha256=bq75vQw_Uc6qrPk-lxoSbsz6p1AkF9hzQqaXRL6vgGY,46499
 singlestoredb/tests/test_plugin.py,sha256=P1nXLnTafaHkHN-6bVbGryxTu7OWJPU9SYFZ_WQUwq8,845
 singlestoredb/tests/test_results.py,sha256=Zg1ynZFRZqalAMfNLOU5C6BDXaox6JxrKm_XZwVNFcg,6753
 singlestoredb/tests/test_types.py,sha256=YeVE6KPqlqzJke-4hbRmc8ko1E7RLHu5S8qLg04Bl5Y,4632
-singlestoredb/tests/test_udf.py,sha256=Hm4Egcrd3awgGQA9JS7T0yBWyNL44z8cdjvZwNL7re4,30768
+singlestoredb/tests/test_udf.py,sha256=b1zOJVuhHvL_cMHYu5iJ5Cinu8gb2Rq_aaCUTaFlNZA,29553
+singlestoredb/tests/test_udf_returns.py,sha256=lDx26AUKGS4V0Vxf5ePO-VcrcX0QamyWGo-tm1GNc-E,16000
 singlestoredb/tests/test_xdict.py,sha256=5ArRJqd5aNXkPK7Y6sFeRbqZ59MZ1YaGBpSlDAbBrjM,10741
 singlestoredb/tests/utils.py,sha256=WR8GFNiC0lU4tz21Y3rlbbp9Gz9WcSwp2jpUSCj7RFU,5136
-singlestoredb/tests/ext_funcs/__init__.py,sha256=mKa-vOh5D8M03kgKTQyvOB74X-Of0nl-mcmJBFzRW0c,9675
+singlestoredb/tests/ext_funcs/__init__.py,sha256=8kGNrG2qZVcmaAc4KAatrqZYPjBXtKbOqoG7zAwMtM4,14727
 singlestoredb/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/utils/config.py,sha256=WVQ567ZzqzlTGueQH5fEpm5tPZuz8y7qvpEQUB-vPjk,25453
 singlestoredb/utils/convert_rows.py,sha256=gkZeZazeJvimCYEQ1FdAC-AmMDwmFGCuP6mi653bpns,1885
@@ -141,9 +147,9 @@ singlestoredb/utils/results.py,sha256=wR70LhCqlobniZf52r67zYLBOKjWHQm68NAskdRQND
 singlestoredb/utils/xdict.py,sha256=-wi1lSPTnY99fhVMBhPKJ8cCsQhNG4GMUfkEBDKYgCw,13321
 sqlx/__init__.py,sha256=4Sdn8HN-Hf8v0_wCt60DCckCg8BvgM3-9r4YVfZycRE,89
 sqlx/magic.py,sha256=6VBlotgjautjev599tHaTYOfcfOA9m6gV_-P1_Qc4lI,3622
-singlestoredb-1.12.4.dist-info/LICENSE,sha256=Bojenzui8aPNjlF3w4ojguDP7sTf8vFV_9Gc2UAG1sg,11542
-singlestoredb-1.12.4.dist-info/METADATA,sha256=K1tOPCbyAof-bQelQHIkS3dyPZ5xx-2ckWoKxUamUSs,5778
-singlestoredb-1.12.4.dist-info/WHEEL,sha256=UyMHzmWA0xVqVPKfTiLs2eN3OWWZUl-kQemNbpIqlKo,100
-singlestoredb-1.12.4.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
-singlestoredb-1.12.4.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
-singlestoredb-1.12.4.dist-info/RECORD,,
+singlestoredb-1.13.1.dist-info/LICENSE,sha256=Bojenzui8aPNjlF3w4ojguDP7sTf8vFV_9Gc2UAG1sg,11542
+singlestoredb-1.13.1.dist-info/METADATA,sha256=JZVSNAL7m8xkaedZ40JWAZIWLY-yxJ_Q9IbM9jID2T4,5847
+singlestoredb-1.13.1.dist-info/WHEEL,sha256=UyMHzmWA0xVqVPKfTiLs2eN3OWWZUl-kQemNbpIqlKo,100
+singlestoredb-1.13.1.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
+singlestoredb-1.13.1.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
+singlestoredb-1.13.1.dist-info/RECORD,,