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

singlestoredb/__init__.py

@@ -13,7 +13,7 @@ Examples
 
 """
 
-__version__ = '1.13.1'
+__version__ = '1.14.1'
 
 from typing import Any
 
@@ -31,6 +31,14 @@ 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,
+)
 
 
 #

singlestoredb/connection.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 """SingleStoreDB connections and cursors."""
 import abc
+import functools
 import inspect
 import io
 import queue
@@ -1288,6 +1289,14 @@ class Connection(metaclass=abc.ABCMeta):
         """Access server properties managed by the SHOW statement."""
         return ShowAccessor(self)
 
+    @functools.cached_property
+    def vector_db(self) -> Any:
+        """
+        Get vectorstore API accessor
+        """
+        from vectorstore import VectorDB
+        return VectorDB(connection=self)
+
 
 #
 # NOTE: When adding parameters to this function, you should always

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._post(
+            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_management.py

@@ -450,55 +450,59 @@ class TestStage(unittest.TestCase):
     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 +513,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()
 
@@ -1143,58 +1150,62 @@ class TestFileSpaces(unittest.TestCase):
 
     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 +1214,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 +1234,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 +1262,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/tests/test_vectorstore.py

@@ -0,0 +1,51 @@
+import os
+import unittest
+
+from vectorstore import VectorDB
+
+import singlestoredb as s2
+from . import utils
+
+
+class TestVectorDB(unittest.TestCase):
+
+    driver = s2
+
+    dbname: str = ''
+    dbexisted: bool = False
+
+    @classmethod
+    def setUpClass(cls) -> None:
+        sql_file = os.path.join(os.path.dirname(__file__), 'empty.sql')
+        cls.dbname, cls.dbexisted = utils.load_sql(sql_file)  # type: ignore
+
+    @classmethod
+    def tearDownClass(cls) -> None:
+        if not cls.dbexisted:
+            utils.drop_database(cls.dbname)  # type: ignore
+
+    def test_vectordb_from_params(self) -> None:
+        db: VectorDB = s2.vector_db(database=type(self).dbname)
+        index = db.create_index(
+            name='test_index', dimension=3,
+            tags={'name': 'test_tag'},
+        )
+        assert index.name == 'test_index'
+        assert index.dimension == 3
+        assert index.tags == {'name': 'test_tag'}
+        assert db.has_index('test_index')
+
+    def test_vectordb_from_connection(self) -> None:
+        with s2.connect(database=type(self).dbname) as conn:
+            db: VectorDB = conn.vector_db
+            index = db.create_index(
+                name='test_index_1',
+                dimension=4, tags={'name': 'test_tag'},
+            )
+            assert index.name == 'test_index_1'
+            assert index.dimension == 4
+            assert index.tags == {'name': 'test_tag'}
+            assert db.has_index('test_index_1')
+
+            db2: VectorDB = conn.vector_db
+            assert db2.has_index('test_index_1')

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/vectorstore.py

@@ -0,0 +1,192 @@
+from typing import Any
+from typing import Callable
+from typing import Dict
+from typing import Optional
+
+from vectorstore import AndFilter  # noqa: F401
+from vectorstore import DeletionProtection  # noqa: F401
+from vectorstore import EqFilter  # noqa: F401
+from vectorstore import ExactMatchFilter  # noqa: F401
+from vectorstore import FilterTypedDict  # noqa: F401
+from vectorstore import GteFilter  # noqa: F401
+from vectorstore import GtFilter  # noqa: F401
+from vectorstore import IndexInterface  # noqa: F401
+from vectorstore import IndexList  # noqa: F401
+from vectorstore import IndexModel  # noqa: F401
+from vectorstore import IndexStatsTypedDict  # noqa: F401
+from vectorstore import InFilter  # noqa: F401
+from vectorstore import LteFilter  # noqa: F401
+from vectorstore import LtFilter  # noqa: F401
+from vectorstore import MatchTypedDict  # noqa: F401
+from vectorstore import Metric  # noqa: F401
+from vectorstore import NamespaceStatsTypedDict  # noqa: F401
+from vectorstore import NeFilter  # noqa: F401
+from vectorstore import NinFilter  # noqa: F401
+from vectorstore import OrFilter  # noqa: F401
+from vectorstore import SimpleFilter  # noqa: F401
+from vectorstore import Vector  # noqa: F401
+from vectorstore import VectorDictMetadataValue  # noqa: F401
+from vectorstore import VectorMetadataTypedDict  # noqa: F401
+from vectorstore import VectorTuple  # noqa: F401
+from vectorstore import VectorTupleWithMetadata  # noqa: F401
+
+
+def vector_db(
+    host: Optional[str] = None, user: Optional[str] = None,
+    password: Optional[str] = None, port: Optional[int] = None,
+    database: Optional[str] = None, driver: Optional[str] = None,
+    pure_python: Optional[bool] = None, local_infile: Optional[bool] = None,
+    charset: Optional[str] = None,
+    ssl_key: Optional[str] = None, ssl_cert: Optional[str] = None,
+    ssl_ca: Optional[str] = None, ssl_disabled: Optional[bool] = None,
+    ssl_cipher: Optional[str] = None, ssl_verify_cert: Optional[bool] = None,
+    tls_sni_servername: Optional[str] = None,
+    ssl_verify_identity: Optional[bool] = None,
+    conv: Optional[Dict[int, Callable[..., Any]]] = None,
+    credential_type: Optional[str] = None,
+    autocommit: Optional[bool] = None,
+    results_type: Optional[str] = None,
+    buffered: Optional[bool] = None,
+    results_format: Optional[str] = None,
+    program_name: Optional[str] = None,
+    conn_attrs: Optional[Dict[str, str]] = {},
+    multi_statements: Optional[bool] = None,
+    client_found_rows: Optional[bool] = None,
+    connect_timeout: Optional[int] = None,
+    nan_as_null: Optional[bool] = None,
+    inf_as_null: Optional[bool] = None,
+    encoding_errors: Optional[str] = None,
+    track_env: Optional[bool] = None,
+    enable_extended_data_types: Optional[bool] = None,
+    vector_data_format: Optional[str] = None,
+    parse_json: Optional[bool] = None,
+    pool_size: Optional[int] = 5,
+    max_overflow: Optional[int] = 10,
+    timeout: Optional[float] = 30,
+) -> Any:
+    """
+    Return a vectorstore API connection.
+    Database should be specified in the URL or as a keyword.
+
+    Parameters
+    ----------
+    host : str, optional
+        Hostname, IP address, or URL that describes the connection.
+        The scheme or protocol defines which database connector to use.
+        By default, the ``mysql`` scheme is used. To connect to the
+        HTTP API, the scheme can be set to ``http`` or ``https``. The username,
+        password, host, and port are specified as in a standard URL. The path
+        indicates the database name. The overall form of the URL is:
+        ``scheme://user:password@host:port/db_name``.  The scheme can
+        typically be left off (unless you are using the HTTP API):
+        ``user:password@host:port/db_name``.
+    user : str, optional
+        Database user name
+    password : str, optional
+        Database user password
+    port : int, optional
+        Database port. This defaults to 3306 for non-HTTP connections, 80
+        for HTTP connections, and 443 for HTTPS connections.
+    database : str, optional
+        Database name.
+    pure_python : bool, optional
+        Use the connector in pure Python mode
+    local_infile : bool, optional
+        Allow local file uploads
+    charset : str, optional
+        Character set for string values
+    ssl_key : str, optional
+        File containing SSL key
+    ssl_cert : str, optional
+        File containing SSL certificate
+    ssl_ca : str, optional
+        File containing SSL certificate authority
+    ssl_cipher : str, optional
+        Sets the SSL cipher list
+    ssl_disabled : bool, optional
+        Disable SSL usage
+    ssl_verify_cert : bool, optional
+        Verify the server's certificate. This is automatically enabled if
+        ``ssl_ca`` is also specified.
+    ssl_verify_identity : bool, optional
+        Verify the server's identity
+    conv : dict[int, Callable], optional
+        Dictionary of data conversion functions
+    credential_type : str, optional
+        Type of authentication to use: auth.PASSWORD, auth.JWT, or auth.BROWSER_SSO
+    autocommit : bool, optional
+        Enable autocommits
+    results_type : str, optional
+        The form of the query results: tuples, namedtuples, dicts,
+        numpy, polars, pandas, arrow
+    buffered : bool, optional
+        Should the entire query result be buffered in memory? This is the default
+        behavior which allows full cursor control of the result, but does consume
+        more memory.
+    results_format : str, optional
+        Deprecated. This option has been renamed to results_type.
+    program_name : str, optional
+        Name of the program
+    conn_attrs : dict, optional
+        Additional connection attributes for telemetry. Example:
+        {'program_version': "1.0.2", "_connector_name": "dbt connector"}
+    multi_statements: bool, optional
+        Should multiple statements be allowed within a single query?
+    connect_timeout : int, optional
+        The timeout for connecting to the database in seconds.
+        (default: 10, min: 1, max: 31536000)
+    nan_as_null : bool, optional
+        Should NaN values be treated as NULLs when used in parameter
+        substitutions including uploaded data?
+    inf_as_null : bool, optional
+        Should Inf values be treated as NULLs when used in parameter
+        substitutions including uploaded data?
+    encoding_errors : str, optional
+        The error handler name for value decoding errors
+    track_env : bool, optional
+        Should the connection track the SINGLESTOREDB_URL environment variable?
+    enable_extended_data_types : bool, optional
+        Should extended data types (BSON, vector) be enabled?
+    vector_data_format : str, optional
+        Format for vector types: json or binary
+    pool_size : int, optional
+        The number of connections to keep in the connection pool. Default is 5.
+    max_overflow : int, optional
+        The maximum number of connections to allow beyond the pool size.
+        Default is 10.
+    timeout : float, optional
+        The timeout for acquiring a connection from the pool in seconds.
+        Default is 30 seconds.
+
+    See Also
+    --------
+    :class:`Connection`
+
+    Returns
+    -------
+    :class:`VectorDB`
+
+    """
+    from vectorstore import VectorDB
+    return VectorDB(
+        host=host, user=user, password=password, port=port,
+        database=database, driver=driver, pure_python=pure_python,
+        local_infile=local_infile, charset=charset,
+        ssl_key=ssl_key, ssl_cert=ssl_cert, ssl_ca=ssl_ca,
+        ssl_disabled=ssl_disabled, ssl_cipher=ssl_cipher,
+        ssl_verify_cert=ssl_verify_cert,
+        tls_sni_servername=tls_sni_servername,
+        ssl_verify_identity=ssl_verify_identity, conv=conv,
+        credential_type=credential_type, autocommit=autocommit,
+        results_type=results_type, buffered=buffered,
+        results_format=results_format, program_name=program_name,
+        conn_attrs=conn_attrs, multi_statements=multi_statements,
+        client_found_rows=client_found_rows,
+        connect_timeout=connect_timeout, nan_as_null=nan_as_null,
+        inf_as_null=inf_as_null, encoding_errors=encoding_errors,
+        track_env=track_env,
+        enable_extended_data_types=enable_extended_data_types,
+        vector_data_format=vector_data_format,
+        parse_json=parse_json, pool_size=pool_size,
+        max_overflow=max_overflow, timeout=timeout,
+    )

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: singlestoredb
-Version: 1.13.1
+Version: 1.14.1
 Summary: Interface to the SingleStoreDB database and workspace management APIs
 Home-page: https://github.com/singlestore-labs/singlestoredb-python
 Author: SingleStore
@@ -11,7 +11,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Database
-Requires-Python: >=3.8
+Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: PyJWT
@@ -19,6 +19,7 @@ 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"

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

@@ -1,12 +1,13 @@
-singlestoredb/__init__.py,sha256=TlTiqjO5Ea2FP3Iq8Kmk0zAA2vR6oOj-HjURaUcQKcM,1649
+singlestoredb/__init__.py,sha256=EzlSL8GTLh0YsZT6nfBumDkUyKFIwMdORoJ_z5Xx8TI,2091
 singlestoredb/auth.py,sha256=u8D9tpKzrqa4ssaHjyZnGDX1q8XBpGtuoOkTkSv7B28,7599
 singlestoredb/config.py,sha256=dayUWwSy2YdgmhF8tzH-7FwFpwon5bgX_VeX-Yu5ia4,12969
-singlestoredb/connection.py,sha256=0HEpjBZXLqQwOTEfveMkgej1H3Kyof47prIHvJJZtoo,45831
+singlestoredb/connection.py,sha256=ELk3-UpM6RaB993aIt08MydKiiDnejHQ1s8EFiacrAI,46055
 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/types.py,sha256=Qp_PWYjSYG6PRnmXAZZ7K2QehUqfoG4KSllI3O1stPE,10397
+singlestoredb/vectorstore.py,sha256=BZb8e7m02_XVHqOyu8tA94R6kHb3n-BC8F08JyJwDzY,8408
 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
@@ -39,7 +40,7 @@ singlestoredb/fusion/handler.py,sha256=HEW83De1zj94hvG7rbqlOszIIgBKiag0UGO5I0WoJ
 singlestoredb/fusion/registry.py,sha256=jjdRTYZ3ylhy6gAoW5xBj0tkxGFBT-2yLQ0tztTgDIY,6112
 singlestoredb/fusion/result.py,sha256=p5I65C-Dhhl1yeZwetXXZabwritr8Ph2mFvJJ3ovcBM,11790
 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/files.py,sha256=McoRacihcQn0-qILujBi0HCNyuFcrBoIUGlkWlg1cII,18991
 singlestoredb/fusion/handlers/job.py,sha256=r0KdOD55VUDw-SymC__5Mn-fzJTZE_xcBgH-O8DYVHc,21095
 singlestoredb/fusion/handlers/models.py,sha256=xJPIG0_GgF-VrmPoIsU2U4AsS7ytDz8JMRaqchglAR0,6236
@@ -47,14 +48,14 @@ 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=EgE2m_nxisGPo6YV3AJd-RRafdT0f70HRbIo1ONQ668,39893
+singlestoredb/http/connection.py,sha256=X5GEPPOE-rMm17d0-TPhcdxUHibcYl-MZAnPhut8xyo,39956
 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
 singlestoredb/management/__init__.py,sha256=ofNTPCdkZ1dS_aX2aUujd8aMHQi8Lle5Ced0aaO3RH4,269
 singlestoredb/management/billing_usage.py,sha256=9ighjIpcopgIyJOktBYQ6pahBZmWGHOPyyCW4gu9FGs,3735
 singlestoredb/management/cluster.py,sha256=h75grXSxq4Anr4RxwKxcZW4TkWJ4bFg_ql5iRWCNLdQ,14405
-singlestoredb/management/export.py,sha256=jJCe25ecH_LzKSDc7vS1-5DQaWFrZipeawLPpArByJE,5108
+singlestoredb/management/export.py,sha256=VrpGyHVUadtJfprcsi9N2SNS6dunDfA1EV2SfWaPlmA,9081
 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
@@ -126,12 +127,13 @@ singlestoredb/tests/test_ext_func.py,sha256=s1k1cBxQ7vIS1zSrKGkKTgLZE1DT_Rqj-3VN
 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=6evsyQWA-lOKMehJi8xvjp0udm85EHBHuZDHwQEzxPg,45149
+singlestoredb/tests/test_management.py,sha256=IGwBijg5NnSSGVNjbtW2m7wVv2b0SFdvfW4sBXTPLP0,45571
 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=Kb7-oJpnN6MTT3aE5V5dry_r5ze0EwaAIJeh_zR3l0I,28844
 singlestoredb/tests/test_udf_returns.py,sha256=k31L6Ir2Xw8MEZ18upuu0p_D_OpbrPAzWhDQXVFDS7I,15541
+singlestoredb/tests/test_vectorstore.py,sha256=anHfp5gQrQy8Iw3Ub4mxFEkaZWahs566OXuKqjpkozM,1554
 singlestoredb/tests/test_xdict.py,sha256=fqHspoi39nbX3fIDVkkRXcd5H50xdOsSvK0bxAMQnaE,10408
 singlestoredb/tests/utils.py,sha256=2A2tEdD3t8aXWUnHtAIcFlWrflsz2MlMcCbUDaAG29c,4995
 singlestoredb/tests/ext_funcs/__init__.py,sha256=gtyhykoEk8_-il5ukTwvqDu-4D1LgwxMFseYg1wgOHo,14103
@@ -146,9 +148,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.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,,
+singlestoredb-1.14.1.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
+singlestoredb-1.14.1.dist-info/METADATA,sha256=PEUJtCAYwCFmN7FKYnqwkV3o8Ai9_eVYbS3D4dyY1_8,5734
+singlestoredb-1.14.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+singlestoredb-1.14.1.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
+singlestoredb-1.14.1.dist-info/top_level.txt,sha256=DfFGz7bM4XrshloiCeTABgylT3BUnS8T5pJam3ewT6Q,19
+singlestoredb-1.14.1.dist-info/RECORD,,