singlestoredb 1.15.7__tar.gz → 1.16.0__tar.gz

{singlestoredb-1.15.7/singlestoredb.egg-info → singlestoredb-1.16.0}/PKG-INFO

@@ -1,13 +1,11 @@
 Metadata-Version: 2.1
 Name: singlestoredb
-Version: 1.15.7
+Version: 1.16.0
 Summary: Interface to the SingleStoreDB database and workspace management APIs
-Home-page: https://github.com/singlestore-labs/singlestoredb-python
-Author: SingleStore
-Author-email: support@singlestore.com
+Author-email: SingleStore <support@singlestore.com>
 License: Apache-2.0
+Project-URL: Homepage, https://github.com/singlestore-labs/singlestoredb-python
 Classifier: Development Status :: 5 - Production/Stable
-Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Database
@@ -24,6 +22,10 @@ Provides-Extra: pytest
 Provides-Extra: rsa
 Provides-Extra: sqlalchemy
 Provides-Extra: vectorstore
+Provides-Extra: test
+Provides-Extra: docs
+Provides-Extra: build
+Provides-Extra: dev
 License-File: LICENSE
 
 # <img src="https://github.com/singlestore-labs/singlestoredb-python/blob/main/resources/singlestore-logo.png" height="60" valign="middle"/> SingleStoreDB Python SDK

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/accel.c

@@ -3079,11 +3079,11 @@ static PyObject *dump_rowdat_1_numpy(PyObject *self, PyObject *args, PyObject *k
         PyErr_SetString(PyExc_ValueError, "value is outside the valid range for UNSIGNED INT"); \
         goto error; \
     }
-#define CHECK_BIGINT(x, unsigned_input) if ((x) < ((unsigned_input) ? 0 : -9223372036854775808) || (x) > 9223372036854775807) { \
+#define CHECK_BIGINT(x, unsigned_input) if ((x) < ((unsigned_input) ? 0 : INT64_MIN) || (x) > INT64_MAX) { \
         PyErr_SetString(PyExc_ValueError, "value is outside the valid range for BIGINT"); \
         goto error; \
     }
-#define CHECK_UNSIGNED_BIGINT(x, unsigned_input) if ((x) < 0 || (x) > 18446744073709551615) { \
+#define CHECK_UNSIGNED_BIGINT(x, unsigned_input) if ((x) < 0) { \
         PyErr_SetString(PyExc_ValueError, "value is outside the valid range for UNSIGNED BIGINT"); \
         goto error; \
     }

singlestoredb-1.16.0/pyproject.toml

@@ -0,0 +1,118 @@
+[build-system]
+requires = ["build", "setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "singlestoredb"
+version = "1.16.0"
+description = "Interface to the SingleStoreDB database and workspace management APIs"
+readme = {file = "README.md", content-type = "text/markdown"}
+license = {text = "Apache-2.0"}
+authors = [
+    {name = "SingleStore", email = "support@singlestore.com"}
+]
+requires-python = ">=3.9"
+classifiers = [
+    "Development Status :: 5 - Production/Stable",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "Topic :: Database",
+]
+dependencies = [
+    "PyJWT",
+    "parsimonious",
+    "requests",
+    "sqlparams",
+    "tomli>=1.1.0; python_version < '3.11'",
+    "typing-extensions<=4.13.2; python_version < '3.11'",
+]
+
+[project.urls]
+Homepage = "https://github.com/singlestore-labs/singlestoredb-python"
+
+[project.optional-dependencies]
+dataframe = ["ibis-singlestoredb"]
+dbt = ["dbt-singlestore"]
+docker = ["docker"]
+ed22519 = ["PyNaCl>=1.4.0"]
+gssapi = ["gssapi"]
+ibis = ["ibis-singlestoredb"]
+kerberos = ["gssapi"]
+pytest = ["pytest"]
+rsa = ["cryptography"]
+sqlalchemy = ["sqlalchemy-singlestoredb>=1.0.0"]
+vectorstore = ["singlestore-vectorstore>=0.1.2"]
+
+# Development dependency groups
+test = [
+    "coverage",
+    "dash",
+    "fastapi",
+    "ipython",
+    "jupysql",
+    "pandas",
+    "parameterized",
+    "polars",
+    "pyarrow",
+    "pydantic",
+    "pytest",
+    "pytest-cov",
+    "singlestore-vectorstore>=0.1.2",
+    "uvicorn",
+]
+docs = [
+    "sphinx",
+    "sphinx_rtd_theme",
+]
+build = [
+    "build",
+    "setuptools>=61.0",
+    "wheel",
+]
+dev = [
+    "singlestoredb[test,docs,build]",
+]
+
+[project.entry-points.pytest11]
+singlestoredb = "singlestoredb.pytest"
+
+[tool.setuptools]
+include-package-data = true
+
+[tool.setuptools.packages.find]
+exclude = ["docs*", "resources*", "examples*", "licenses*"]
+
+[tool.setuptools.package-data]
+"*" = ["*.typed", "*.sql", "*.csv", "*.ipynb"]
+
+[tool.pytest.ini_options]
+markers = [
+    "management",
+]
+
+[tool.mypy]
+check_untyped_defs = true
+disallow_any_generics = true
+disallow_incomplete_defs = true
+disallow_untyped_defs = true
+no_implicit_optional = true
+warn_redundant_casts = true
+warn_unused_ignores = false
+
+[[tool.mypy.overrides]]
+module = "testing.*"
+disallow_untyped_defs = false
+
+[[tool.mypy.overrides]]
+module = "tests.*"
+disallow_untyped_defs = false
+
+[tool.coverage.run]
+source_pkgs = ["singlestoredb"]
+
+[tool.coverage.report]
+omit = [
+    "setup.py",
+    "*/tests/*",
+    "*/docs/*",
+]

singlestoredb-1.16.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/__init__.py

@@ -13,7 +13,7 @@ Examples
 
 """
 
-__version__ = '1.15.7'
+__version__ = '1.16.0'
 
 from typing import Any
 

singlestoredb-1.16.0/singlestoredb/ai/__init__.py

@@ -0,0 +1,2 @@
+from .chat import SingleStoreChatFactory  # noqa: F401
+from .embeddings import SingleStoreEmbeddingsFactory  # noqa: F401

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/ai/chat.py

@@ -29,44 +29,6 @@ from botocore import UNSIGNED
 from botocore.config import Config
 
 
-class SingleStoreChatOpenAI(ChatOpenAI):
-    def __init__(self, model_name: str, api_key: Optional[str] = None, **kwargs: Any):
-        inference_api_manger = (
-            manage_workspaces().organizations.current.inference_apis
-        )
-        info = inference_api_manger.get(model_name=model_name)
-        token = (
-            api_key
-            if api_key is not None
-            else os.environ.get('SINGLESTOREDB_USER_TOKEN')
-        )
-        super().__init__(
-            base_url=info.connection_url,
-            api_key=token,
-            model=model_name,
-            **kwargs,
-        )
-
-
-class SingleStoreChat(ChatOpenAI):
-    def __init__(self, model_name: str, api_key: Optional[str] = None, **kwargs: Any):
-        inference_api_manger = (
-            manage_workspaces().organizations.current.inference_apis
-        )
-        info = inference_api_manger.get(model_name=model_name)
-        token = (
-            api_key
-            if api_key is not None
-            else os.environ.get('SINGLESTOREDB_USER_TOKEN')
-        )
-        super().__init__(
-            base_url=info.connection_url,
-            api_key=token,
-            model=model_name,
-            **kwargs,
-        )
-
-
 def SingleStoreChatFactory(
     model_name: str,
     api_key: Optional[str] = None,
@@ -90,9 +52,27 @@ def SingleStoreChatFactory(
             'signature_version': UNSIGNED,
             'retries': {'max_attempts': 1, 'mode': 'standard'},
         }
-        if http_client is not None and http_client.timeout is not None:
-            cfg_kwargs['read_timeout'] = http_client.timeout
-            cfg_kwargs['connect_timeout'] = http_client.timeout
+        # Extract timeouts from http_client if provided
+        t = http_client.timeout if http_client is not None else None
+        connect_timeout = None
+        read_timeout = None
+        if t is not None:
+            if isinstance(t, httpx.Timeout):
+                if t.connect is not None:
+                    connect_timeout = float(t.connect)
+                if t.read is not None:
+                    read_timeout = float(t.read)
+                if connect_timeout is None and read_timeout is not None:
+                    connect_timeout = read_timeout
+                if read_timeout is None and connect_timeout is not None:
+                    read_timeout = connect_timeout
+            elif isinstance(t, (int, float)):
+                connect_timeout = float(t)
+                read_timeout = float(t)
+        if read_timeout is not None:
+            cfg_kwargs['read_timeout'] = read_timeout
+        if connect_timeout is not None:
+            cfg_kwargs['connect_timeout'] = connect_timeout
 
         cfg = Config(**cfg_kwargs)
         client = boto3.client(

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/ai/embeddings.py

@@ -29,21 +29,6 @@ from botocore import UNSIGNED
 from botocore.config import Config
 
 
-class SingleStoreEmbeddings(OpenAIEmbeddings):
-
-    def __init__(self, model_name: str, **kwargs: Any):
-        inference_api_manger = (
-            manage_workspaces().organizations.current.inference_apis
-        )
-        info = inference_api_manger.get(model_name=model_name)
-        super().__init__(
-            base_url=info.connection_url,
-            api_key=os.environ.get('SINGLESTOREDB_USER_TOKEN'),
-            model=model_name,
-            **kwargs,
-        )
-
-
 def SingleStoreEmbeddingsFactory(
     model_name: str,
     api_key: Optional[str] = None,
@@ -66,9 +51,27 @@ def SingleStoreEmbeddingsFactory(
             'signature_version': UNSIGNED,
             'retries': {'max_attempts': 1, 'mode': 'standard'},
         }
-        if http_client is not None and http_client.timeout is not None:
-            cfg_kwargs['read_timeout'] = http_client.timeout
-            cfg_kwargs['connect_timeout'] = http_client.timeout
+        # Extract timeouts from http_client if provided
+        t = http_client.timeout if http_client is not None else None
+        connect_timeout = None
+        read_timeout = None
+        if t is not None:
+            if isinstance(t, httpx.Timeout):
+                if t.connect is not None:
+                    connect_timeout = float(t.connect)
+                if t.read is not None:
+                    read_timeout = float(t.read)
+                if connect_timeout is None and read_timeout is not None:
+                    connect_timeout = read_timeout
+                if read_timeout is None and connect_timeout is not None:
+                    read_timeout = connect_timeout
+            elif isinstance(t, (int, float)):
+                connect_timeout = float(t)
+                read_timeout = float(t)
+        if read_timeout is not None:
+            cfg_kwargs['read_timeout'] = read_timeout
+        if connect_timeout is not None:
+            cfg_kwargs['connect_timeout'] = connect_timeout
 
         cfg = Config(**cfg_kwargs)
         client = boto3.client(

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/connection.py

@@ -9,15 +9,15 @@ import re
 import sys
 import warnings
 import weakref
+from collections.abc import Iterator
 from collections.abc import Mapping
 from collections.abc import MutableMapping
+from collections.abc import Sequence
 from typing import Any
 from typing import Callable
 from typing import Dict
-from typing import Iterator
 from typing import List
 from typing import Optional
-from typing import Sequence
 from typing import Tuple
 from typing import Union
 from urllib.parse import parse_qs

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/functions/ext/asgi.py

@@ -48,15 +48,15 @@ import urllib
 import uuid
 import zipfile
 import zipimport
+from collections.abc import Awaitable
+from collections.abc import Iterable
+from collections.abc import Sequence
 from types import ModuleType
 from typing import Any
-from typing import Awaitable
 from typing import Callable
 from typing import Dict
-from typing import Iterable
 from typing import List
 from typing import Optional
-from typing import Sequence
 from typing import Set
 from typing import Tuple
 from typing import Union

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/functions/ext/rowdat_1.py

@@ -1,11 +1,11 @@
 #!/usr/bin/env python3
 import struct
 import warnings
+from collections.abc import Sequence
 from io import BytesIO
 from typing import Any
 from typing import List
 from typing import Optional
-from typing import Sequence
 from typing import Tuple
 from typing import TYPE_CHECKING
 

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/functions/signature.py

@@ -9,12 +9,12 @@ import string
 import sys
 import types
 import typing
+from collections.abc import Sequence
 from typing import Any
 from typing import Callable
 from typing import Dict
 from typing import List
 from typing import Optional
-from typing import Sequence
 from typing import Tuple
 from typing import TypeVar
 from typing import Union

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/functions/typing/__init__.py

@@ -1,5 +1,5 @@
+from collections.abc import Iterable
 from typing import Any
-from typing import Iterable
 from typing import Tuple
 from typing import TypeVar
 

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/functions/utils.py

@@ -4,10 +4,10 @@ import struct
 import sys
 import types
 import typing
+from collections.abc import Iterable
 from enum import Enum
 from typing import Any
 from typing import Dict
-from typing import Iterable
 from typing import Tuple
 from typing import Union
 

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/fusion/handler.py

@@ -5,10 +5,10 @@ import os
 import re
 import sys
 import textwrap
+from collections.abc import Iterable
 from typing import Any
 from typing import Callable
 from typing import Dict
-from typing import Iterable
 from typing import List
 from typing import Optional
 from typing import Set

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/fusion/result.py

@@ -2,8 +2,8 @@
 from __future__ import annotations
 
 import re
+from collections.abc import Iterable
 from typing import Any
-from typing import Iterable
 from typing import List
 from typing import Optional
 from typing import Tuple

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/http/connection.py

@@ -10,13 +10,13 @@ import os
 import re
 import time
 from base64 import b64decode
+from collections.abc import Iterable
+from collections.abc import Sequence
 from typing import Any
 from typing import Callable
 from typing import Dict
-from typing import Iterable
 from typing import List
 from typing import Optional
-from typing import Sequence
 from typing import Tuple
 from typing import Union
 from urllib.parse import urljoin

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/management/manager.py

@@ -15,6 +15,7 @@ import requests
 
 from .. import config
 from ..exceptions import ManagementError
+from ..exceptions import OperationalError
 from .utils import get_token
 
 
@@ -310,3 +311,63 @@ class Manager(object):
             out = getattr(self, f'get_{self.obj_type}')(out.id)
 
         return out
+
+    def _wait_on_endpoint(
+        self,
+        out: Any,
+        interval: int = 10,
+        timeout: int = 300,
+    ) -> Any:
+        """
+        Wait for the endpoint to be ready by attempting to connect.
+
+        Parameters
+        ----------
+        out : Any
+            Workspace object with a connect method
+        interval : int, optional
+            Interval between each connection attempt (default: 10 seconds)
+        timeout : int, optional
+            Maximum time to wait before raising an exception (default: 300 seconds)
+
+        Raises
+        ------
+        ManagementError
+            If timeout is reached or endpoint is not available
+
+        Returns
+        -------
+        Same object type as `out`
+
+        """
+        # Only wait if workload type is set which means we are in the
+        # notebook environment. Outside of the environment, the endpoint
+        # may not be reachable directly.
+        if not os.environ.get('SINGLESTOREDB_WORKLOAD_TYPE', ''):
+            return out
+
+        if not hasattr(out, 'connect') or not out.connect:
+            raise ManagementError(
+                msg=f'{type(out).__name__} object does not have a valid endpoint',
+            )
+
+        while True:
+            try:
+                # Try to establish a connection to the endpoint using context manager
+                with out.connect(connect_timeout=5):
+                    pass
+            except Exception as exc:
+                # If we get an 'access denied' error, that means that the server is
+                # up and we just aren't authenticating.
+                if isinstance(exc, OperationalError) and exc.errno == 1045:
+                    break
+                # If connection fails, check timeout and retry
+                if timeout <= 0:
+                    raise ManagementError(
+                        msg=f'Exceeded waiting time for {self.obj_type} endpoint '
+                            'to become ready',
+                    )
+                time.sleep(interval)
+                timeout -= interval
+
+        return out

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/management/utils.py

@@ -6,11 +6,11 @@ import itertools
 import os
 import re
 import sys
+from collections.abc import Mapping
 from typing import Any
 from typing import Callable
 from typing import Dict
 from typing import List
-from typing import Mapping
 from typing import Optional
 from typing import SupportsIndex
 from typing import Tuple

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/management/workspace.py

@@ -1794,6 +1794,12 @@ class WorkspaceManager(Manager):
                 interval=wait_interval,
                 timeout=wait_timeout,
             )
+            # After workspace is active, wait for endpoint to be ready
+            out = self._wait_on_endpoint(
+                out,
+                interval=wait_interval,
+                timeout=wait_timeout,
+            )
         return out
 
     def get_workspace_group(self, id: str) -> WorkspaceGroup:

{singlestoredb-1.15.7 → singlestoredb-1.16.0}/singlestoredb/mysql/connection.py

@@ -13,9 +13,9 @@ import struct
 import sys
 import traceback
 import warnings
+from collections.abc import Iterable
 from typing import Any
 from typing import Dict
-from typing import Iterable
 
 try:
     import _singlestoredb_accel