singlestoredb 1.14.0__cp38-abi3-macosx_10_9_universal2.whl → 1.14.2__cp38-abi3-macosx_10_9_universal2.whl

singlestoredb/__init__.py

@@ -13,7 +13,7 @@ Examples
 
 """
 
-__version__ = '1.14.0'
+__version__ = '1.14.2'
 
 from typing import Any
 
@@ -31,14 +31,18 @@ from .types import (
     Date, Time, Timestamp, DateFromTicks, TimeFromTicks, TimestampFromTicks,
     Binary, STRING, BINARY, NUMBER, DATETIME, ROWID,
 )
-from .vectorstore import (
-    vector_db, IndexInterface, IndexList, IndexModel, MatchTypedDict,
-    Metric, IndexStatsTypedDict, NamespaceStatsTypedDict, Vector,
-    VectorDictMetadataValue, VectorMetadataTypedDict, VectorTuple,
-    VectorTupleWithMetadata, DeletionProtection, AndFilter, EqFilter,
-    ExactMatchFilter, FilterTypedDict, GteFilter, GtFilter, InFilter,
-    LteFilter, LtFilter, NeFilter, NinFilter, OrFilter, SimpleFilter,
-)
+# These are only loaded if the singlestore-vectorstore package is available
+try:
+    from .vectorstore import (
+        vector_db, IndexInterface, IndexList, IndexModel, MatchTypedDict,
+        Metric, IndexStatsTypedDict, NamespaceStatsTypedDict, Vector,
+        VectorDictMetadataValue, VectorMetadataTypedDict, VectorTuple,
+        VectorTupleWithMetadata, DeletionProtection, AndFilter, EqFilter,
+        ExactMatchFilter, FilterTypedDict, GteFilter, GtFilter, InFilter,
+        LteFilter, LtFilter, NeFilter, NinFilter, OrFilter, SimpleFilter,
+    )
+except (ImportError, ModuleNotFoundError):
+    pass
 
 
 #

singlestoredb/fusion/handler.py

@@ -33,7 +33,7 @@ CORE_GRAMMAR = r'''
     close_paren = ws* ")" ws*
     open_repeats = ws* ~r"[\(\[\{]" ws*
     close_repeats = ws* ~r"[\)\]\}]" ws*
-    select = ~r"SELECT"i ws+ ~r".+" ws*
+    statement = ~r"[\s\S]*" ws*
     table = ~r"(?:([A-Za-z0-9_\-]+)|`([^\`]+)`)(?:\.(?:([A-Za-z0-9_\-]+)|`([^\`]+)`))?" ws*
     column = ~r"(?:([A-Za-z0-9_\-]+)|`([^\`]+)`)(?:\.(?:([A-Za-z0-9_\-]+)|`([^\`]+)`))?" ws*
     link_name = ~r"(?:([A-Za-z0-9_\-]+)|`([^\`]+)`)(?:\.(?:([A-Za-z0-9_\-]+)|`([^\`]+)`))?" ws*
@@ -77,6 +77,7 @@ BUILTINS = {
     '<file-type>': r'''
     file_type = { FILE | FOLDER }
     ''',
+    '<statement>': '',
 }
 
 BUILTIN_DEFAULTS = {  # type: ignore
@@ -627,6 +628,18 @@ class SQLHandler(NodeVisitor):
         cls.compile()
         registry.register_handler(cls, overwrite=overwrite)
 
+    def create_result(self) -> result.FusionSQLResult:
+        """
+        Create a new result object.
+
+        Returns
+        -------
+        FusionSQLResult
+            A new result object for this handler
+
+        """
+        return result.FusionSQLResult()
+
     def execute(self, sql: str) -> result.FusionSQLResult:
         """
         Parse the SQL and invoke the handler method.
@@ -746,9 +759,9 @@ class SQLHandler(NodeVisitor):
         _, out, *_ = visited_children
         return out
 
-    def visit_select(self, node: Node, visited_children: Iterable[Any]) -> Any:
-        out = ' '.join(flatten(visited_children))
-        return {'select': out}
+    def visit_statement(self, node: Node, visited_children: Iterable[Any]) -> Any:
+        out = ' '.join(flatten(visited_children)).strip()
+        return {'statement': out}
 
     def visit_order_by(self, node: Node, visited_children: Iterable[Any]) -> Any:
         """Handle ORDER BY."""

singlestoredb/fusion/handlers/export.py

@@ -1,10 +1,12 @@
 #!/usr/bin/env python3
+import datetime
 import json
 from typing import Any
 from typing import Dict
 from typing import Optional
 
 from .. import result
+from ...management.export import _get_exports
 from ...management.export import ExportService
 from ...management.export import ExportStatus
 from ..handler import SQLHandler
@@ -104,7 +106,100 @@ class CreateClusterIdentity(SQLHandler):
 CreateClusterIdentity.register(overwrite=True)
 
 
-class CreateExport(SQLHandler):
+def _start_export(params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+    # From table
+    if isinstance(params['from_table'], str):
+        from_database = None
+        from_table = params['from_table']
+    else:
+        from_database, from_table = params['from_table']
+
+    # Catalog
+    catalog_config = json.loads(params['catalog'].get('catalog_config', '{}') or '{}')
+    catalog_creds = json.loads(params['catalog'].get('catalog_creds', '{}') or '{}')
+
+    # Storage
+    storage_config = json.loads(params['storage'].get('link_config', '{}') or '{}')
+    storage_creds = json.loads(params['storage'].get('link_creds', '{}') or '{}')
+
+    storage_config['provider'] = 'S3'
+
+    wsg = get_workspace_group({})
+
+    if from_database is None:
+        raise ValueError('database name must be specified for source table')
+
+    if wsg._manager is None:
+        raise TypeError('no workspace manager is associated with workspace group')
+
+    partition_by = []
+    if params['partition_by']:
+        for key in params['partition_by']:
+            transform = key['partition_key']['transform']['col_transform']
+            part = {}
+            part['transform'] = transform[0].lower()
+            part['name'] = transform[-1]['transform_col']
+            partition_by.append(part)
+
+    order_by = []
+    if params['order_by'] and params['order_by']['by']:
+        for key in params['order_by']['by']:
+            transform = key['transform']['col_transform']
+            order = {}
+            order['transform'] = transform[0].lower()
+            order['name'] = transform[-1]['transform_col']
+            order['direction'] = 'ascending'
+            order['null_order'] = 'nulls_first'
+            if key.get('direction'):
+                if 'desc' in key['direction'].lower():
+                    order['direction'] = 'descending'
+            if key.get('null_order'):
+                if 'last' in key['null_order'].lower():
+                    order['null_order'] = 'nulls_last'
+            order_by.append(order)
+
+    # Refresh interval
+    refresh_interval_delta = None
+    refresh_interval = params.get('refresh_interval', None)
+    if refresh_interval is not None:
+        value = int(refresh_interval['refresh_interval_value'])
+        time_unit = refresh_interval['refresh_interval_time_unit'].upper()
+        if value < 0:
+            raise ValueError('refresh interval must be greater than 0')
+        if time_unit == 'SECONDS':
+            refresh_interval_delta = datetime.timedelta(seconds=int(value))
+        elif time_unit == 'MINUTES':
+            refresh_interval_delta = datetime.timedelta(minutes=int(value))
+        elif time_unit == 'HOURS':
+            refresh_interval_delta = datetime.timedelta(hours=int(value))
+        elif time_unit == 'DAYS':
+            refresh_interval_delta = datetime.timedelta(days=int(value))
+        else:
+            raise ValueError('invalid refresh interval time unit')
+
+    out = ExportService(
+        wsg,
+        from_database,
+        from_table,
+        dict(**catalog_config, **catalog_creds),
+        dict(**storage_config, **storage_creds),
+        columns=None,
+        partition_by=partition_by or None,
+        order_by=order_by or None,
+        properties=json.loads(params['properties']) if params['properties'] else None,
+        incremental=params.get('incremental', False),
+        refresh_interval=int(refresh_interval_delta.total_seconds())
+        if refresh_interval_delta is not None else None,
+    ).start()
+
+    res = FusionSQLResult()
+    res.add_field('ExportID', result.STRING)
+    res.set_rows([(out.export_id,)])
+
+    return res
+
+
+class StartExport(SQLHandler):
     """
     START EXPORT
         from_table
@@ -150,7 +245,7 @@ class CreateExport(SQLHandler):
 
     Description
     -----------
-    Create an export configuration.
+    Start an export.
 
     Arguments
     ---------
@@ -180,7 +275,6 @@ class CreateExport(SQLHandler):
             LINK S3 CONFIG '{
                 "region": "us-east-1",
                 "endpoint_url": "s3://bucket-name"
-
             }'
         ;
 
@@ -189,77 +283,129 @@ class CreateExport(SQLHandler):
     _enabled = False
 
     def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
-        # From table
-        if isinstance(params['from_table'], str):
-            from_database = None
-            from_table = params['from_table']
-        else:
-            from_database, from_table = params['from_table']
+        return _start_export(params)
 
-        # Catalog
-        catalog_config = json.loads(params['catalog'].get('catalog_config', '{}') or '{}')
-        catalog_creds = json.loads(params['catalog'].get('catalog_creds', '{}') or '{}')
 
-        # Storage
-        storage_config = json.loads(params['storage'].get('link_config', '{}') or '{}')
-        storage_creds = json.loads(params['storage'].get('link_creds', '{}') or '{}')
+StartExport.register(overwrite=True)
 
-        storage_config['provider'] = 'S3'
 
-        wsg = get_workspace_group({})
+class StartIncrementalExport(SQLHandler):
+    """
+    START INCREMENTAL EXPORT
+        from_table
+        catalog
+        storage
+        [ partition_by ]
+        [ order_by ]
+        [ properties ]
+        [ refresh_interval ]
+    ;
 
-        if from_database is None:
-            raise ValueError('database name must be specified for source table')
+    # From table
+    from_table = FROM <table>
 
-        if wsg._manager is None:
-            raise TypeError('no workspace manager is associated with workspace group')
+    # Transforms
+    _col_transform = { VOID | IDENTITY | YEAR | MONTH | DAY | HOUR } ( _transform_col )
+    _transform_col = <column>
+    _arg_transform = { BUCKET | TRUNCATE } ( _transform_col <comma> _transform_arg )
+    _transform_arg = <integer>
+    transform = { _col_transform | _arg_transform }
 
-        partition_by = []
-        if params['partition_by']:
-            for key in params['partition_by']:
-                transform = key['partition_key']['transform']['col_transform']
-                part = {}
-                part['transform'] = transform[0].lower()
-                part['name'] = transform[-1]['transform_col']
-                partition_by.append(part)
-
-        order_by = []
-        if params['order_by'] and params['order_by']['by']:
-            for key in params['order_by']['by']:
-                transform = key['transform']['col_transform']
-                order = {}
-                order['transform'] = transform[0].lower()
-                order['name'] = transform[-1]['transform_col']
-                order['direction'] = 'ascending'
-                order['null_order'] = 'nulls_first'
-                if key.get('direction'):
-                    if 'desc' in key['direction'].lower():
-                        order['direction'] = 'descending'
-                if key.get('null_order'):
-                    if 'last' in key['null_order'].lower():
-                        order['null_order'] = 'nulls_last'
-                order_by.append(order)
+    # Partitions
+    partition_by = PARTITION BY partition_key,...
+    partition_key = transform
 
-        out = ExportService(
-            wsg,
-            from_database,
-            from_table,
-            dict(**catalog_config, **catalog_creds),
-            dict(**storage_config, **storage_creds),
-            columns=None,
-            partition_by=partition_by or None,
-            order_by=order_by or None,
-            properties=json.loads(params['properties']) if params['properties'] else None,
-        ).start()
+    # Sort order
+    order_by = ORDER BY sort_key,...
+    sort_key = transform [ direction ] [ null_order ]
+    direction = { ASC | DESC | ASCENDING | DESCENDING }
+    null_order = { NULLS_FIRST | NULLS_LAST }
 
-        res = FusionSQLResult()
-        res.add_field('ExportID', result.STRING)
-        res.set_rows([(out.export_id,)])
+    # Properties
+    properties = PROPERTIES '<json>'
 
-        return res
+    # Catolog
+    catalog = CATALOG [ _catalog_config ] [ _catalog_creds ]
+    _catalog_config = CONFIG '<catalog-config>'
+    _catalog_creds = CREDENTIALS '<catalog-creds>'
+
+    # Storage
+    storage = LINK [ _link_config ] [ _link_creds ]
+    _link_config = S3 CONFIG '<link-config>'
+    _link_creds = CREDENTIALS '<link-creds>'
+
+    # Refresh interval
+    refresh_interval = REFRESH INTERVAL _refresh_interval_value _refresh_interval_time_unit
+    _refresh_interval_value = <integer>
+    _refresh_interval_time_unit = { SECONDS | MINUTES | HOURS | DAYS }
+
+    Description
+    -----------
+    Start an incremental export.
+
+    Arguments
+    ---------
+    * ``<catalog-config>`` and ``<catalog-creds>``: The catalog configuration.
+    * ``<link-config>`` and ``<link-creds>``: The storage link configuration.
+
+    Remarks
+    -------
+    * ``FROM <table>`` specifies the SingleStore table to export. The same name will
+      be used for the exported table.
+    * ``CATALOG`` specifies the details of the catalog to connect to.
+    * ``LINK`` specifies the details of the data storage to connect to.
+    * ``REFRESH INTERVAL`` specifies the interval for refreshing the
+      incremental export. The default is 1 day.
+
+    Examples
+    --------
+    The following statement starts an export operation with the given
+    catalog and link configurations. The source table to export is
+    named "customer_data"::
 
+        START INCREMENTAL EXPORT FROM my_db.customer_data
+            CATALOG CONFIG '{
+                "catalog_type": "GLUE",
+                "table_format": "ICEBERG",
+                "catalog_id": "13983498723498",
+                "catalog_region": "us-east-1"
+            }'
+            LINK S3 CONFIG '{
+                "region": "us-east-1",
+                "endpoint_url": "s3://bucket-name"
+            }'
+            REFRESH INTERVAL 24 HOURS
+        ;
+
+    """  # noqa
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        params['incremental'] = True
+        return _start_export(params)
+
+
+StartIncrementalExport.register(overwrite=True)
+
+
+def _format_status(export_id: str, status: ExportStatus) -> Optional[FusionSQLResult]:
+    """Return the status of an export operation."""
+    info = status._info()
 
-CreateExport.register(overwrite=True)
+    res = FusionSQLResult()
+    res.add_field('ExportID', result.STRING)
+    res.add_field('Status', result.STRING)
+    res.add_field('Message', result.STRING)
+    res.set_rows([
+        (
+            export_id,
+            info.get('status', 'Unknown'),
+            info.get('statusMsg', ''),
+        ),
+    ])
+
+    return res
 
 
 class ShowExport(SQLHandler):
@@ -275,9 +421,29 @@ class ShowExport(SQLHandler):
 
     def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
         wsg = get_workspace_group({})
-        out = ExportStatus(params['export_id'], wsg)
+        return _format_status(
+            params['export_id'], ExportStatus(params['export_id'], wsg),
+        )
+
+
+ShowExport.register(overwrite=True)
+
+
+class ShowExports(SQLHandler):
+    """
+    SHOW EXPORTS [ scope ];
+
+    # Location of the export
+    scope = FOR '<scope>'
+
+    """
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        wsg = get_workspace_group({})
 
-        status = out._info()
+        exports = _get_exports(wsg, params.get('scope', 'all'))
 
         res = FusionSQLResult()
         res.add_field('ExportID', result.STRING)
@@ -285,13 +451,75 @@ class ShowExport(SQLHandler):
         res.add_field('Message', result.STRING)
         res.set_rows([
             (
-                params['export_id'],
-                status.get('status', 'Unknown'),
-                status.get('statusMsg', ''),
-            ),
+                info['egressID'],
+                info.get('status', 'Unknown'),
+                info.get('statusMsg', ''),
+            )
+            for info in [x._info() for x in exports]
         ])
 
         return res
 
 
-ShowExport.register(overwrite=True)
+ShowExports.register(overwrite=True)
+
+
+class SuspendExport(SQLHandler):
+    """
+    SUSPEND EXPORT export_id;
+
+    # ID of export
+    export_id = '<export-id>'
+
+    """
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        wsg = get_workspace_group({})
+        service = ExportService.from_export_id(wsg, params['export_id'])
+        return _format_status(params['export_id'], service.suspend())
+
+
+SuspendExport.register(overwrite=True)
+
+
+class ResumeExport(SQLHandler):
+    """
+    RESUME EXPORT export_id;
+
+    # ID of export
+    export_id = '<export-id>'
+
+    """
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        wsg = get_workspace_group({})
+        service = ExportService.from_export_id(wsg, params['export_id'])
+        return _format_status(params['export_id'], service.resume())
+
+
+ResumeExport.register(overwrite=True)
+
+
+class DropExport(SQLHandler):
+    """
+    DROP EXPORT export_id;
+
+    # ID of export
+    export_id = '<export-id>'
+
+    """
+
+    _enabled = False
+
+    def run(self, params: Dict[str, Any]) -> Optional[FusionSQLResult]:
+        wsg = get_workspace_group({})
+        service = ExportService.from_export_id(wsg, params['export_id'])
+        service.drop()
+        return None
+
+
+DropExport.register(overwrite=True)

singlestoredb/http/connection.py

@@ -569,8 +569,10 @@ class Cursor(connection.Cursor):
 
         if res.status_code >= 400:
             if res.text:
-                if re.match(r'^Error\s+\d+:', res.text):
-                    code, msg = res.text.split(':', 1)
+                m = re.match(r'^Error\s+(\d+).*?:', res.text)
+                if m:
+                    code = m.group(1)
+                    msg = res.text.split(':', 1)[-1]
                     icode = int(code.split()[-1])
                 else:
                     icode = res.status_code

singlestoredb/management/export.py

@@ -27,6 +27,9 @@ class ExportService(object):
     partition_by: Optional[List[Dict[str, str]]]
     order_by: Optional[List[Dict[str, Dict[str, str]]]]
     properties: Optional[Dict[str, Any]]
+    incremental: bool
+    refresh_interval: Optional[int]
+    export_id: Optional[str]
 
     def __init__(
         self,
@@ -38,6 +41,8 @@ class ExportService(object):
         columns: Optional[List[str]] = None,
         partition_by: Optional[List[Dict[str, str]]] = None,
         order_by: Optional[List[Dict[str, Dict[str, str]]]] = None,
+        incremental: bool = False,
+        refresh_interval: Optional[int] = None,
         properties: Optional[Dict[str, Any]] = None,
     ):
         #: Workspace group
@@ -68,8 +73,30 @@ class ExportService(object):
         self.order_by = order_by or None
         self.properties = properties or None
 
+        self.incremental = incremental
+        self.refresh_interval = refresh_interval
+
+        self.export_id = None
+
         self._manager: Optional[WorkspaceManager] = workspace_group._manager
 
+    @classmethod
+    def from_export_id(
+        self,
+        workspace_group: WorkspaceGroup,
+        export_id: str,
+    ) -> ExportService:
+        """Create export service from export ID."""
+        out = ExportService(
+            workspace_group=workspace_group,
+            database='',
+            table='',
+            catalog_info={},
+            storage_info={},
+        )
+        out.export_id = export_id
+        return out
+
     def __str__(self) -> str:
         """Return string representation."""
         return vars_to_str(self)
@@ -98,6 +125,11 @@ class ExportService(object):
 
     def start(self, tags: Optional[List[str]] = None) -> 'ExportStatus':
         """Start the export process."""
+        if not self.table or not self.database:
+            raise ManagementError(
+                msg='Database and table must be set before starting the export.',
+            )
+
         if self._manager is None:
             raise ManagementError(
                 msg='No workspace manager is associated with this object.',
@@ -122,11 +154,87 @@ class ExportService(object):
                     partitionSpec=partition_spec,
                     sortOrderSpec=sort_order_spec,
                     properties=self.properties,
+                    incremental=self.incremental or None,
+                    refreshInterval=self.refresh_interval
+                    if self.refresh_interval is not None else None,
                 ).items() if v is not None
             },
         )
 
-        return ExportStatus(out.json()['egressID'], self.workspace_group)
+        self.export_id = str(out.json()['egressID'])
+
+        return ExportStatus(self.export_id, self.workspace_group)
+
+    def suspend(self) -> 'ExportStatus':
+        """Suspend the export process."""
+        if self._manager is None:
+            raise ManagementError(
+                msg='No workspace manager is associated with this object.',
+            )
+
+        if self.export_id is None:
+            raise ManagementError(
+                msg='Export ID is not set. You must start the export first.',
+            )
+
+        self._manager._post(
+            f'workspaceGroups/{self.workspace_group.id}/egress/suspendTableEgress',
+            json=dict(egressID=self.export_id),
+        )
+
+        return ExportStatus(self.export_id, self.workspace_group)
+
+    def resume(self) -> 'ExportStatus':
+        """Resume the export process."""
+        if self._manager is None:
+            raise ManagementError(
+                msg='No workspace manager is associated with this object.',
+            )
+
+        if self.export_id is None:
+            raise ManagementError(
+                msg='Export ID is not set. You must start the export first.',
+            )
+
+        self._manager._post(
+            f'workspaceGroups/{self.workspace_group.id}/egress/resumeTableEgress',
+            json=dict(egressID=self.export_id),
+        )
+
+        return ExportStatus(self.export_id, self.workspace_group)
+
+    def drop(self) -> None:
+        """Drop the export process."""
+        if self._manager is None:
+            raise ManagementError(
+                msg='No workspace manager is associated with this object.',
+            )
+
+        if self.export_id is None:
+            raise ManagementError(
+                msg='Export ID is not set. You must start the export first.',
+            )
+
+        self._manager._delete(
+            f'workspaceGroups/{self.workspace_group.id}/egress/dropTableEgress',
+            json=dict(egressID=self.export_id),
+        )
+
+        return None
+
+    def status(self) -> ExportStatus:
+        """Get the status of the export process."""
+        if self._manager is None:
+            raise ManagementError(
+                msg='No workspace manager is associated with this object.',
+            )
+
+        if self.export_id is None:
+            raise ManagementError(
+                msg='Export ID is not set. You must start the export first.',
+            )
+
+        return ExportStatus(self.export_id, self.workspace_group)
 
 
 class ExportStatus(object):
@@ -167,3 +275,21 @@ class ExportStatus(object):
 
     def __repr__(self) -> str:
         return self.status
+
+
+def _get_exports(
+    workspace_group: WorkspaceGroup,
+    scope: str = 'all',
+) -> List[ExportStatus]:
+    """Get all exports in the workspace group."""
+    if workspace_group._manager is None:
+        raise ManagementError(
+            msg='No workspace manager is associated with this object.',
+        )
+
+    out = workspace_group._manager._get(
+        f'workspaceGroups/{workspace_group.id}/egress/tableEgressStatus',
+        json=dict(scope=scope),
+    )
+
+    return out.json()

singlestoredb/tests/test_fusion.py

@@ -499,7 +499,10 @@ class TestJobsFusion(unittest.TestCase):
     @classmethod
     def tearDownClass(cls):
         for job_id in cls.job_ids:
-            cls.manager.organizations.current.jobs.delete(job_id)
+            try:
+                cls.manager.organizations.current.jobs.delete(job_id)
+            except Exception:
+                pass
         if cls.workspace_group is not None:
             cls.workspace_group.terminate(force=True)
         cls.manager = None

singlestoredb/tests/test_management.py

@@ -397,13 +397,16 @@ class TestStage(unittest.TestCase):
     def test_upload_file(self):
         st = self.wg.stage
 
+        upload_test_sql = f'upload_test_{id(self)}.sql'
+        upload_test2_sql = f'upload_test2_{id(self)}.sql'
+
         root = st.info('/')
         assert str(root.path) == '/'
         assert root.type == 'directory'
 
         # Upload file
-        f = st.upload_file(TEST_DIR / 'test.sql', 'upload_test.sql')
-        assert str(f.path) == 'upload_test.sql'
+        f = st.upload_file(TEST_DIR / 'test.sql', upload_test_sql)
+        assert str(f.path) == upload_test_sql
         assert f.type == 'file'
 
         # Download and compare to original
@@ -412,15 +415,15 @@ class TestStage(unittest.TestCase):
 
         # Make sure we can't overwrite
         with self.assertRaises(OSError):
-            st.upload_file(TEST_DIR / 'test.sql', 'upload_test.sql')
+            st.upload_file(TEST_DIR / 'test.sql', upload_test_sql)
 
         # Force overwrite with new content; use file object this time
         f = st.upload_file(
             open(TEST_DIR / 'test2.sql', 'r'),
-            'upload_test.sql',
+            upload_test_sql,
             overwrite=True,
         )
-        assert str(f.path) == 'upload_test.sql'
+        assert str(f.path) == upload_test_sql
         assert f.type == 'file'
 
         # Verify new content
@@ -442,63 +445,67 @@ class TestStage(unittest.TestCase):
         # Write file into folder
         f = st.upload_file(
             TEST_DIR / 'test2.sql',
-            os.path.join(lib.path, 'upload_test2.sql'),
+            os.path.join(lib.path, upload_test2_sql),
         )
-        assert str(f.path) == 'lib/upload_test2.sql'
+        assert str(f.path) == 'lib/' + upload_test2_sql
         assert f.type == 'file'
 
     def test_open(self):
         st = self.wg.stage
 
+        open_test_sql = f'open_test_{id(self)}.sql'
+
         # See if error is raised for non-existent file
         with self.assertRaises(s2.ManagementError):
-            st.open('open_test.sql', 'r')
+            st.open(open_test_sql, 'r')
 
         # Load test file
-        st.upload_file(TEST_DIR / 'test.sql', 'open_test.sql')
+        st.upload_file(TEST_DIR / 'test.sql', open_test_sql)
 
         # Read file using `open`
-        with st.open('open_test.sql', 'r') as rfile:
+        with st.open(open_test_sql, 'r') as rfile:
             assert rfile.read() == open(TEST_DIR / 'test.sql').read()
 
         # Read file using `open` with 'rt' mode
-        with st.open('open_test.sql', 'rt') as rfile:
+        with st.open(open_test_sql, 'rt') as rfile:
             assert rfile.read() == open(TEST_DIR / 'test.sql').read()
 
         # Read file using `open` with 'rb' mode
-        with st.open('open_test.sql', 'rb') as rfile:
+        with st.open(open_test_sql, 'rb') as rfile:
             assert rfile.read() == open(TEST_DIR / 'test.sql', 'rb').read()
 
         # Read file using `open` with 'rb' mode
         with self.assertRaises(ValueError):
-            with st.open('open_test.sql', 'b') as rfile:
+            with st.open(open_test_sql, 'b') as rfile:
                 pass
 
         # Attempt overwrite file using `open` with mode 'x'
         with self.assertRaises(OSError):
-            with st.open('open_test.sql', 'x') as wfile:
+            with st.open(open_test_sql, 'x') as wfile:
                 pass
 
         # Attempt overwrite file using `open` with mode 'w'
-        with st.open('open_test.sql', 'w') as wfile:
+        with st.open(open_test_sql, 'w') as wfile:
             wfile.write(open(TEST_DIR / 'test2.sql').read())
 
-        txt = st.download_file('open_test.sql', encoding='utf-8')
+        txt = st.download_file(open_test_sql, encoding='utf-8')
 
         assert txt == open(TEST_DIR / 'test2.sql').read()
 
+        open_raw_test_sql = f'open_raw_test_{id(self)}.sql'
+
         # Test writer without context manager
-        wfile = st.open('open_raw_test.sql', 'w')
+        wfile = st.open(open_raw_test_sql, 'w')
         for line in open(TEST_DIR / 'test.sql'):
             wfile.write(line)
         wfile.close()
 
-        txt = st.download_file('open_raw_test.sql', encoding='utf-8')
+        txt = st.download_file(open_raw_test_sql, encoding='utf-8')
 
         assert txt == open(TEST_DIR / 'test.sql').read()
 
         # Test reader without context manager
-        rfile = st.open('open_raw_test.sql', 'r')
+        rfile = st.open(open_raw_test_sql, 'r')
         txt = ''
         for line in rfile:
             txt += line
@@ -509,15 +516,18 @@ class TestStage(unittest.TestCase):
     def test_obj_open(self):
         st = self.wg.stage
 
+        obj_open_test_sql = f'obj_open_test_{id(self)}.sql'
+        obj_open_dir = f'obj_open_dir_{id(self)}'
+
         # Load test file
-        f = st.upload_file(TEST_DIR / 'test.sql', 'obj_open_test.sql')
+        f = st.upload_file(TEST_DIR / 'test.sql', obj_open_test_sql)
 
         # Read file using `open`
         with f.open() as rfile:
             assert rfile.read() == open(TEST_DIR / 'test.sql').read()
 
         # Make sure directories error out
-        d = st.mkdir('obj_open_dir')
+        d = st.mkdir(obj_open_dir)
         with self.assertRaises(IsADirectoryError):
             d.open()
 
@@ -921,7 +931,10 @@ class TestJob(unittest.TestCase):
     @classmethod
     def tearDownClass(cls):
         for job_id in cls.job_ids:
-            cls.manager.organizations.current.jobs.delete(job_id)
+            try:
+                cls.manager.organizations.current.jobs.delete(job_id)
+            except Exception:
+                pass
         if cls.workspace_group is not None:
             cls.workspace_group.terminate(force=True)
         cls.workspace_group = None
@@ -1054,6 +1067,8 @@ class TestFileSpaces(unittest.TestCase):
         cls.shared_space = None
 
     def test_upload_file(self):
+        upload_test_ipynb = f'upload_test_{id(self)}.ipynb'
+
         for space in [self.personal_space, self.shared_space]:
             root = space.info('/')
             assert str(root.path) == '/'
@@ -1062,9 +1077,9 @@ class TestFileSpaces(unittest.TestCase):
             # Upload files
             f = space.upload_file(
                 TEST_DIR / 'test.ipynb',
-                'upload_test.ipynb',
+                upload_test_ipynb,
             )
-            assert str(f.path) == 'upload_test.ipynb'
+            assert str(f.path) == upload_test_ipynb
             assert f.type == 'notebook'
 
             # Download and compare to original
@@ -1075,15 +1090,15 @@ class TestFileSpaces(unittest.TestCase):
             with self.assertRaises(OSError):
                 space.upload_file(
                     TEST_DIR / 'test.ipynb',
-                    'upload_test.ipynb',
+                    upload_test_ipynb,
                 )
 
             # Force overwrite with new content
             f = space.upload_file(
                 TEST_DIR / 'test2.ipynb',
-                'upload_test.ipynb', overwrite=True,
+                upload_test_ipynb, overwrite=True,
             )
-            assert str(f.path) == 'upload_test.ipynb'
+            assert str(f.path) == upload_test_ipynb
             assert f.type == 'notebook'
 
             # Verify new content
@@ -1095,9 +1110,11 @@ class TestFileSpaces(unittest.TestCase):
                 space.upload_folder(TEST_DIR, 'test')
 
             # Cleanup
-            space.remove('upload_test.ipynb')
+            space.remove(upload_test_ipynb)
 
     def test_upload_file_io(self):
+        upload_test_ipynb = f'upload_test_{id(self)}.ipynb'
+
         for space in [self.personal_space, self.shared_space]:
             root = space.info('/')
             assert str(root.path) == '/'
@@ -1106,9 +1123,9 @@ class TestFileSpaces(unittest.TestCase):
             # Upload files
             f = space.upload_file(
                 open(TEST_DIR / 'test.ipynb', 'r'),
-                'upload_test.ipynb',
+                upload_test_ipynb,
             )
-            assert str(f.path) == 'upload_test.ipynb'
+            assert str(f.path) == upload_test_ipynb
             assert f.type == 'notebook'
 
             # Download and compare to original
@@ -1119,15 +1136,15 @@ class TestFileSpaces(unittest.TestCase):
             with self.assertRaises(OSError):
                 space.upload_file(
                     open(TEST_DIR / 'test.ipynb', 'r'),
-                    'upload_test.ipynb',
+                    upload_test_ipynb,
                 )
 
             # Force overwrite with new content
             f = space.upload_file(
                 open(TEST_DIR / 'test2.ipynb', 'r'),
-                'upload_test.ipynb', overwrite=True,
+                upload_test_ipynb, overwrite=True,
             )
-            assert str(f.path) == 'upload_test.ipynb'
+            assert str(f.path) == upload_test_ipynb
             assert f.type == 'notebook'
 
             # Verify new content
@@ -1139,62 +1156,66 @@ class TestFileSpaces(unittest.TestCase):
                 space.upload_folder(TEST_DIR, 'test')
 
             # Cleanup
-            space.remove('upload_test.ipynb')
+            space.remove(upload_test_ipynb)
 
     def test_open(self):
         for space in [self.personal_space, self.shared_space]:
+            open_test_ipynb = f'open_test_ipynb_{id(self)}.ipynb'
+
             # See if error is raised for non-existent file
             with self.assertRaises(s2.ManagementError):
-                space.open('open_test.ipynb', 'r')
+                space.open(open_test_ipynb, 'r')
 
             # Load test file
-            space.upload_file(TEST_DIR / 'test.ipynb', 'open_test.ipynb')
+            space.upload_file(TEST_DIR / 'test.ipynb', open_test_ipynb)
 
             # Read file using `open`
-            with space.open('open_test.ipynb', 'r') as rfile:
+            with space.open(open_test_ipynb, 'r') as rfile:
                 assert rfile.read() == open(TEST_DIR / 'test.ipynb').read()
 
             # Read file using `open` with 'rt' mode
-            with space.open('open_test.ipynb', 'rt') as rfile:
+            with space.open(open_test_ipynb, 'rt') as rfile:
                 assert rfile.read() == open(TEST_DIR / 'test.ipynb').read()
 
             # Read file using `open` with 'rb' mode
-            with space.open('open_test.ipynb', 'rb') as rfile:
+            with space.open(open_test_ipynb, 'rb') as rfile:
                 assert rfile.read() == open(TEST_DIR / 'test.ipynb', 'rb').read()
 
             # Read file using `open` with 'rb' mode
             with self.assertRaises(ValueError):
-                with space.open('open_test.ipynb', 'b') as rfile:
+                with space.open(open_test_ipynb, 'b') as rfile:
                     pass
 
             # Attempt overwrite file using `open` with mode 'x'
             with self.assertRaises(OSError):
-                with space.open('open_test.ipynb', 'x') as wfile:
+                with space.open(open_test_ipynb, 'x') as wfile:
                     pass
 
             # Attempt overwrite file using `open` with mode 'w'
-            with space.open('open_test.ipynb', 'w') as wfile:
+            with space.open(open_test_ipynb, 'w') as wfile:
                 wfile.write(open(TEST_DIR / 'test2.ipynb').read())
 
-            txt = space.download_file('open_test.ipynb', encoding='utf-8')
+            txt = space.download_file(open_test_ipynb, encoding='utf-8')
 
             assert txt == open(TEST_DIR / 'test2.ipynb').read()
 
+            open_raw_test_ipynb = f'open_raw_test_{id(self)}.ipynb'
+
             # Test writer without context manager
-            wfile = space.open('open_raw_test.ipynb', 'w')
+            wfile = space.open(open_raw_test_ipynb, 'w')
             for line in open(TEST_DIR / 'test.ipynb'):
                 wfile.write(line)
             wfile.close()
 
             txt = space.download_file(
-                'open_raw_test.ipynb',
+                open_raw_test_ipynb,
                 encoding='utf-8',
             )
 
             assert txt == open(TEST_DIR / 'test.ipynb').read()
 
             # Test reader without context manager
-            rfile = space.open('open_raw_test.ipynb', 'r')
+            rfile = space.open(open_raw_test_ipynb, 'r')
             txt = ''
             for line in rfile:
                 txt += line
@@ -1203,15 +1224,18 @@ class TestFileSpaces(unittest.TestCase):
             assert txt == open(TEST_DIR / 'test.ipynb').read()
 
             # Cleanup
-            space.remove('open_test.ipynb')
-            space.remove('open_raw_test.ipynb')
+            space.remove(open_test_ipynb)
+            space.remove(open_raw_test_ipynb)
 
     def test_obj_open(self):
         for space in [self.personal_space, self.shared_space]:
+            obj_open_test_ipynb = f'obj_open_test_{id(self)}.ipynb'
+            obj_open_dir = f'obj_open_dir_{id(self)}'
+
             # Load test file
             f = space.upload_file(
                 TEST_DIR / 'test.ipynb',
-                'obj_open_test.ipynb',
+                obj_open_test_ipynb,
             )
 
             # Read file using `open`
@@ -1220,7 +1244,7 @@ class TestFileSpaces(unittest.TestCase):
 
             # Make sure directories error out
             with self.assertRaises(s2.ManagementError):
-                space.mkdir('obj_open_dir')
+                space.mkdir(obj_open_dir)
 
             # Write file using `open`
             with f.open('w', encoding='utf-8') as wfile:
@@ -1248,7 +1272,7 @@ class TestFileSpaces(unittest.TestCase):
             assert txt == open(TEST_DIR / 'test.ipynb').read()
 
             # Cleanup
-            space.remove('obj_open_test.ipynb')
+            space.remove(obj_open_test_ipynb)
 
     def test_os_directories(self):
         for space in [self.personal_space, self.shared_space]:

singlestoredb/types.py

@@ -173,17 +173,25 @@ class ColumnType(object):
         'DECIMAL', 'DEC', 'FIXED', 'NUMERIC', 0, decimal.Decimal,
     )
     DEC = FIXED = NUMERIC = DECIMAL
-    TINY = TINYINT = BOOL = BOOLEAN = NumberDBAPIType(
-        'TINY', 'TINYINT', 'BOOL', 'BOOLEAN', 1,
+    TINY = TINYINT = BOOL = BOOLEAN = UNSIGNED_TINY = UNSIGNED_TINYINT = NumberDBAPIType(
+        'TINY', 'TINYINT', 'BOOL', 'BOOLEAN', 'UNSIGNED TINY', 'UNSIGNED TINYINT', 1,
+    )
+    SHORT = SMALLINT = UNSIGNED_SHORT = UNSIGNED_SMALLINT = NumberDBAPIType(
+        'SMALLINT', 'SHORT', 'UNSIGNED SHORT', 'UNSIGNED SMALLINT', 2,
+    )
+    LONG = INT = UNSIGNED_LONG = UNSIGNED_INT = NumberDBAPIType(
+        'LONG', 'INT', 'UNSIGNED LONG', 'UNSIGNED INT', 3,
     )
-    SHORT = SMALLINT = NumberDBAPIType('SMALLINT', 'SHORT', 2)
-    LONG = INT = NumberDBAPIType('LONG', 'INT', 3)
     FLOAT = NumberDBAPIType('FLOAT', 4)
     DOUBLE = REAL = NumberDBAPIType('DOUBLE', 5, float)
     NULL = DBAPIType('NULL', 6)
     TIMESTAMP = DatetimeDBAPIType('TIMESTAMP', 7)
-    LONGLONG = BIGINT = NumberDBAPIType('BIGINT', 'LONGLONG', 8, int)
-    MEDIUMINT = INT24 = NumberDBAPIType('MEDIUMINT', 'INT24', 9)
+    LONGLONG = BIGINT = UNSIGNED_LONGLONG = UNSIGNED_BIGINT = NumberDBAPIType(
+        'BIGINT', 'LONGLONG', 'UNSIGNED LONGLONG', 'UNSIGNED BIGINT', 8, int,
+    )
+    MEDIUMINT = INT24 = UNSIGNED_MEDIUMINT = UNSIGNED_INT24 = NumberDBAPIType(
+        'MEDIUMINT', 'INT24', 'UNSIGNED MEDIUMINT', 'UNSIGNED INT24', 9,
+    )
     DATE = DBAPIType('DATE', 10, datetime.date)
     TIME = DBAPIType('TIME', 11, datetime.time)
     DATETIME = DatetimeDBAPIType('DATETIME', 12, datetime.datetime)

{singlestoredb-1.14.0.dist-info → singlestoredb-1.14.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: singlestoredb
-Version: 1.14.0
+Version: 1.14.2
 Summary: Interface to the SingleStoreDB database and workspace management APIs
 Home-page: https://github.com/singlestore-labs/singlestoredb-python
 Author: SingleStore
@@ -19,7 +19,6 @@ Requires-Dist: build
 Requires-Dist: parsimonious
 Requires-Dist: requests
 Requires-Dist: setuptools
-Requires-Dist: singlestore-vectorstore >=0.1.2
 Requires-Dist: sqlparams
 Requires-Dist: wheel
 Requires-Dist: tomli >=1.1.0 ; python_version < "3.11"
@@ -44,6 +43,8 @@ Provides-Extra: rsa
 Requires-Dist: cryptography ; extra == 'rsa'
 Provides-Extra: sqlalchemy
 Requires-Dist: sqlalchemy-singlestoredb >=1.0.0 ; extra == 'sqlalchemy'
+Provides-Extra: vectorstore
+Requires-Dist: singlestore-vectorstore >=0.1.2 ; extra == 'vectorstore'
 
 # <img src="https://github.com/singlestore-labs/singlestoredb-python/blob/main/resources/singlestore-logo.png" height="60" valign="middle"/> SingleStoreDB Python SDK
 

{singlestoredb-1.14.0.dist-info → singlestoredb-1.14.2.dist-info}/RECORD

@@ -1,24 +1,24 @@
-_singlestoredb_accel.abi3.so,sha256=Xxm-lpuGz75wOE4dR6ZzwdAsUFVZd0D5uBf1-2QqJ_o,207216
+_singlestoredb_accel.abi3.so,sha256=KFB7Tj4Gf86RMqkWtxyU_QxE8if2oQkBMkTyk304zmk,207216
+singlestoredb-1.14.2.dist-info/RECORD,,
+singlestoredb-1.14.2.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
+singlestoredb-1.14.2.dist-info/WHEEL,sha256=_VEguvlLpUd-c8RbFMA4yMIVNMBv2LhpxYLCEQ-Bogk,113
+singlestoredb-1.14.2.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
+singlestoredb-1.14.2.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
+singlestoredb-1.14.2.dist-info/METADATA,sha256=oSEkI_zMB3TgGwyd6a3fR6OGXdvPZFr0U5zVXoOmq7E,5804
 sqlx/magic.py,sha256=JsS9_9aBFaOt91Torm1JPN0c8qB2QmYJmNSKtbSQIY0,3509
 sqlx/__init__.py,sha256=aBYiU8DZXCogvWu3yWafOz7bZS5WWwLZXj7oL0dXGyU,85
-singlestoredb-1.14.0.dist-info/RECORD,,
-singlestoredb-1.14.0.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
-singlestoredb-1.14.0.dist-info/WHEEL,sha256=_VEguvlLpUd-c8RbFMA4yMIVNMBv2LhpxYLCEQ-Bogk,113
-singlestoredb-1.14.0.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
-singlestoredb-1.14.0.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
-singlestoredb-1.14.0.dist-info/METADATA,sha256=VuCeWXgdykhi4lW_WxMRT3qUsLRZAR3GmLLLhBs5YAw,5751
 singlestoredb/auth.py,sha256=u8D9tpKzrqa4ssaHjyZnGDX1q8XBpGtuoOkTkSv7B28,7599
 singlestoredb/config.py,sha256=dayUWwSy2YdgmhF8tzH-7FwFpwon5bgX_VeX-Yu5ia4,12969
 singlestoredb/vectorstore.py,sha256=BZb8e7m02_XVHqOyu8tA94R6kHb3n-BC8F08JyJwDzY,8408
-singlestoredb/__init__.py,sha256=ztn02QsJ2xuKyXGfbfR6nBSJPkhLlSRWcOT2X3myhrU,2091
-singlestoredb/types.py,sha256=FIqO1A7e0Gkk7ITmIysBy-P5S--ItbMSlYvblzqGS30,9969
+singlestoredb/__init__.py,sha256=LXawJyI2-AilQxUSQ8IJnDUiwZk5nrB-FGGpHWcToGI,2256
+singlestoredb/types.py,sha256=Qp_PWYjSYG6PRnmXAZZ7K2QehUqfoG4KSllI3O1stPE,10397
 singlestoredb/connection.py,sha256=ELk3-UpM6RaB993aIt08MydKiiDnejHQ1s8EFiacrAI,46055
 singlestoredb/pytest.py,sha256=OyF3BO9mgxenifYhOihnzGk8WzCJ_zN5_mxe8XyFPOc,9074
 singlestoredb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 singlestoredb/exceptions.py,sha256=HuoA6sMRL5qiCiee-_5ddTGmFbYC9Euk8TYUsh5GvTw,3234
 singlestoredb/converters.py,sha256=Ui-AqdW3pRAQ8A_YcK9EqVYyM4Pt1_Q-tjlotbpK6Cw,20686
 singlestoredb/fusion/graphql.py,sha256=ZA3HcDq5rER-dCEavwTqnF7KM0D2LCYIY7nLQk7lSso,5207
-singlestoredb/fusion/handler.py,sha256=HEW83De1zj94hvG7rbqlOszIIgBKiag0UGO5I0WoJ6A,27400
+singlestoredb/fusion/handler.py,sha256=M5iyNP4zOaGqUqnZg_b5xhRE-8tHgfZSHDH0zKTiJmE,27692
 singlestoredb/fusion/registry.py,sha256=jjdRTYZ3ylhy6gAoW5xBj0tkxGFBT-2yLQ0tztTgDIY,6112
 singlestoredb/fusion/__init__.py,sha256=Qo7SuqGw-l-vE8-EI2jhm6hXJkYfOLUKIws9c7LFNX0,356
 singlestoredb/fusion/result.py,sha256=p5I65C-Dhhl1yeZwetXXZabwritr8Ph2mFvJJ3ovcBM,11790
@@ -26,7 +26,7 @@ singlestoredb/fusion/handlers/files.py,sha256=McoRacihcQn0-qILujBi0HCNyuFcrBoIUG
 singlestoredb/fusion/handlers/models.py,sha256=xJPIG0_GgF-VrmPoIsU2U4AsS7ytDz8JMRaqchglAR0,6236
 singlestoredb/fusion/handlers/job.py,sha256=r0KdOD55VUDw-SymC__5Mn-fzJTZE_xcBgH-O8DYVHc,21095
 singlestoredb/fusion/handlers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-singlestoredb/fusion/handlers/export.py,sha256=3moTJeqsHkDDpitUUAE6x95JYH2rmb28MALbO4x0dcc,8981
+singlestoredb/fusion/handlers/export.py,sha256=Af4eIMPGKEOpmf4LXnvQsgmnvx5F8B5FkRI20RvEa7o,15309
 singlestoredb/fusion/handlers/utils.py,sha256=ozHOWUraoN8XGTK9JZdhv5HV8AQR8zfUd1yh1kLvUXY,10685
 singlestoredb/fusion/handlers/stage.py,sha256=kYVjbPys83kf3jX6jWwN8Ju0oEocKVZ3TIOt2HiC5Ew,14287
 singlestoredb/fusion/handlers/workspace.py,sha256=4xN2TFO4yF7KZB2Fcht7IuvoDdAT6fDfDLjixiHZN8w,27506
@@ -34,7 +34,7 @@ singlestoredb/tests/test.sql,sha256=dfMehVCQ9wObSVTQKyQi-fRFDZeqRxV4Cj8doBCPEFM,
 singlestoredb/tests/test_xdict.py,sha256=fqHspoi39nbX3fIDVkkRXcd5H50xdOsSvK0bxAMQnaE,10408
 singlestoredb/tests/test_results.py,sha256=wg93sujwt-R9_eJCgSCElgAZhLDkIiAo3qPkPydOv78,6582
 singlestoredb/tests/test_vectorstore.py,sha256=anHfp5gQrQy8Iw3Ub4mxFEkaZWahs566OXuKqjpkozM,1554
-singlestoredb/tests/test_fusion.py,sha256=EH1mRwdX2Fajsq6x2l0gBhH1YhcxtvDGIKC9HJ4sDbQ,50521
+singlestoredb/tests/test_fusion.py,sha256=7YQ_nOQoV_7yD4OEpJz2Ov-zok-cBFK9IOJ3FgZ0xo0,50593
 singlestoredb/tests/test_plugin.py,sha256=qpO9wmWc62VaijN1sJ97YSYIX7I7Y5C6sY-WzwrutDQ,812
 singlestoredb/tests/test_basics.py,sha256=Dw1irrtf3gWN7tqGruSH6uhWi5zkmCpJl6ZMQxMqlf4,48446
 singlestoredb/tests/test_ext_func.py,sha256=s1k1cBxQ7vIS1zSrKGkKTgLZE1DT_Rqj-3VNSCSv68I,43261
@@ -43,7 +43,7 @@ singlestoredb/tests/test2.ipynb,sha256=yd1PE1VK-DwiRd6mYS4_0cPBtuVkvcDtycvTwD-Yn
 singlestoredb/tests/test_ext_func_data.py,sha256=yTADD93nPxX6_rZXXLZaOWEI_yPvYyir9psn5PK9ctU,47695
 singlestoredb/tests/test_exceptions.py,sha256=tfr_8X2w1UmG4nkSBzWGB0C7ehrf1GAVgj6_ODaG-TM,1131
 singlestoredb/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-singlestoredb/tests/test_management.py,sha256=6evsyQWA-lOKMehJi8xvjp0udm85EHBHuZDHwQEzxPg,45149
+singlestoredb/tests/test_management.py,sha256=rzM5xZllXys6OzyFBxOieYSn4kDGCwWuyIkJWsPXvAY,45847
 singlestoredb/tests/test_udf.py,sha256=Kb7-oJpnN6MTT3aE5V5dry_r5ze0EwaAIJeh_zR3l0I,28844
 singlestoredb/tests/test_http.py,sha256=RXasTqBWRn__omj0eLFTJYIbZjd0PPdIV2d4Cqz0MC8,8580
 singlestoredb/tests/utils.py,sha256=2A2tEdD3t8aXWUnHtAIcFlWrflsz2MlMcCbUDaAG29c,4995
@@ -64,7 +64,7 @@ singlestoredb/management/organization.py,sha256=_JvW0Znu5emR5uYGVEcZvakQqftNb_vR
 singlestoredb/management/job.py,sha256=4-xLWzbE8odQogVVaFer80UEoTAZY1T28VZ9Ug4rbmM,24611
 singlestoredb/management/region.py,sha256=HnLcWUh7r_aLECliplCDHak4a_F3B7LOSXEYMW66qD0,1611
 singlestoredb/management/__init__.py,sha256=ofNTPCdkZ1dS_aX2aUujd8aMHQi8Lle5Ced0aaO3RH4,269
-singlestoredb/management/export.py,sha256=jJCe25ecH_LzKSDc7vS1-5DQaWFrZipeawLPpArByJE,5108
+singlestoredb/management/export.py,sha256=yR-yZUE9USFrP5OR_5iLFqEc8GLiKDQypSEp08CmT5k,9083
 singlestoredb/management/utils.py,sha256=QIhZCZSRaDbAG35xu1_n7ihmRXON8swc-gEK2FGYutI,13203
 singlestoredb/management/cluster.py,sha256=h75grXSxq4Anr4RxwKxcZW4TkWJ4bFg_ql5iRWCNLdQ,14405
 singlestoredb/management/inference_api.py,sha256=L6eFqaUaPugF_cmrZ4xlArj8CIv25vWqQs1vwgKPEF4,2583
@@ -84,7 +84,7 @@ singlestoredb/utils/xdict.py,sha256=S9HKgrPrnu_6b7iOwa2KrW8CmU1Uqx0BWdEyogFzWbE,
 singlestoredb/utils/debug.py,sha256=0JiLA37u_9CKiDGiN9BK_PtFMUku3vIcNjERWaTNRSU,349
 singlestoredb/utils/mogrify.py,sha256=-a56IF70U6CkfadeaZgfjRSVsAD3PuqRrzPpjZlgbwY,4050
 singlestoredb/http/__init__.py,sha256=A_2ZUCCpvRYIA6YDpPy57wL5R1eZ5SfP6I1To5nfJ2s,912
-singlestoredb/http/connection.py,sha256=EgE2m_nxisGPo6YV3AJd-RRafdT0f70HRbIo1ONQ668,39893
+singlestoredb/http/connection.py,sha256=X5GEPPOE-rMm17d0-TPhcdxUHibcYl-MZAnPhut8xyo,39956
 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