singlestoredb 1.15.5__cp38-abi3-macosx_10_9_universal2.whl → 1.15.7__cp38-abi3-macosx_10_9_universal2.whl

singlestoredb/__init__.py

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

singlestoredb/ai/__init__.py

@@ -1,2 +1,5 @@
+from .chat import SingleStoreChat  # noqa: F401
+from .chat import SingleStoreChatFactory  # noqa: F401
 from .chat import SingleStoreChatOpenAI  # noqa: F401
 from .embeddings import SingleStoreEmbeddings  # noqa: F401
+from .embeddings import SingleStoreEmbeddingsFactory  # noqa: F401

singlestoredb/ai/chat.py

@@ -1,7 +1,12 @@
 import os
 from typing import Any
+from typing import Callable
+from typing import Optional
+from typing import Union
 
-from singlestoredb.fusion.handlers.utils import get_workspace_manager
+import httpx
+
+from singlestoredb import manage_workspaces
 
 try:
     from langchain_openai import ChatOpenAI
@@ -11,30 +16,144 @@ except ImportError:
         'Please install it with `pip install langchain_openai`.',
     )
 
+try:
+    from langchain_aws import ChatBedrockConverse
+except ImportError:
+    raise ImportError(
+        'Could not import langchain-aws python package. '
+        'Please install it with `pip install langchain-aws`.',
+    )
+
+import boto3
+from botocore import UNSIGNED
+from botocore.config import Config
+
 
 class SingleStoreChatOpenAI(ChatOpenAI):
-    def __init__(self, model_name: str, **kwargs: Any):
+    def __init__(self, model_name: str, api_key: Optional[str] = None, **kwargs: Any):
         inference_api_manger = (
-            get_workspace_manager().organizations.current.inference_apis
+            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=os.environ.get('SINGLESTOREDB_USER_TOKEN'),
+            api_key=token,
             model=model_name,
             **kwargs,
         )
 
 
 class SingleStoreChat(ChatOpenAI):
-    def __init__(self, model_name: str, **kwargs: Any):
+    def __init__(self, model_name: str, api_key: Optional[str] = None, **kwargs: Any):
         inference_api_manger = (
-            get_workspace_manager().organizations.current.inference_apis
+            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=os.environ.get('SINGLESTOREDB_USER_TOKEN'),
+            api_key=token,
             model=model_name,
             **kwargs,
         )
+
+
+def SingleStoreChatFactory(
+    model_name: str,
+    api_key: Optional[str] = None,
+    streaming: bool = True,
+    http_client: Optional[httpx.Client] = None,
+    obo_token_getter: Optional[Callable[[], Optional[str]]] = None,
+    **kwargs: Any,
+) -> Union[ChatOpenAI, ChatBedrockConverse]:
+    """Return a chat model instance (ChatOpenAI or ChatBedrockConverse).
+    """
+    inference_api_manager = (
+        manage_workspaces().organizations.current.inference_apis
+    )
+    info = inference_api_manager.get(model_name=model_name)
+    token_env = os.environ.get('SINGLESTOREDB_USER_TOKEN')
+    token = api_key if api_key is not None else token_env
+
+    if info.hosting_platform == 'Amazon':
+        # Instantiate Bedrock client
+        cfg_kwargs = {
+            '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
+
+        cfg = Config(**cfg_kwargs)
+        client = boto3.client(
+            'bedrock-runtime',
+            endpoint_url=info.connection_url,
+            region_name='us-east-1',
+            aws_access_key_id='placeholder',
+            aws_secret_access_key='placeholder',
+            config=cfg,
+        )
+
+        def _inject_headers(request: Any, **_ignored: Any) -> None:
+            """Inject dynamic auth/OBO headers prior to Bedrock sending."""
+            if obo_token_getter is not None:
+                obo_val = obo_token_getter()
+                if obo_val:
+                    request.headers['X-S2-OBO'] = obo_val
+            if token:
+                request.headers['Authorization'] = f'Bearer {token}'
+            request.headers.pop('X-Amz-Date', None)
+            request.headers.pop('X-Amz-Security-Token', None)
+
+        emitter = client._endpoint._event_emitter
+        emitter.register_first(
+            'before-send.bedrock-runtime.Converse',
+            _inject_headers,
+        )
+        emitter.register_first(
+            'before-send.bedrock-runtime.ConverseStream',
+            _inject_headers,
+        )
+        emitter.register_first(
+            'before-send.bedrock-runtime.InvokeModel',
+            _inject_headers,
+        )
+        emitter.register_first(
+            'before-send.bedrock-runtime.InvokeModelWithResponseStream',
+            _inject_headers,
+        )
+
+        return ChatBedrockConverse(
+            model_id=model_name,
+            endpoint_url=info.connection_url,
+            region_name='us-east-1',
+            aws_access_key_id='placeholder',
+            aws_secret_access_key='placeholder',
+            disable_streaming=not streaming,
+            client=client,
+            **kwargs,
+        )
+
+    # OpenAI / Azure OpenAI path
+    openai_kwargs = dict(
+        base_url=info.connection_url,
+        api_key=token,
+        model=model_name,
+        streaming=streaming,
+    )
+    if http_client is not None:
+        openai_kwargs['http_client'] = http_client
+    return ChatOpenAI(
+        **openai_kwargs,
+        **kwargs,
+    )

singlestoredb/ai/embeddings.py

@@ -1,7 +1,12 @@
 import os
 from typing import Any
+from typing import Callable
+from typing import Optional
+from typing import Union
 
-from singlestoredb.fusion.handlers.utils import get_workspace_manager
+import httpx
+
+from singlestoredb import manage_workspaces
 
 try:
     from langchain_openai import OpenAIEmbeddings
@@ -11,12 +16,24 @@ except ImportError:
         'Please install it with `pip install langchain_openai`.',
     )
 
+try:
+    from langchain_aws import BedrockEmbeddings
+except ImportError:
+    raise ImportError(
+        'Could not import langchain-aws python package. '
+        'Please install it with `pip install langchain-aws`.',
+    )
+
+import boto3
+from botocore import UNSIGNED
+from botocore.config import Config
+
 
 class SingleStoreEmbeddings(OpenAIEmbeddings):
 
     def __init__(self, model_name: str, **kwargs: Any):
         inference_api_manger = (
-            get_workspace_manager().organizations.current.inference_apis
+            manage_workspaces().organizations.current.inference_apis
         )
         info = inference_api_manger.get(model_name=model_name)
         super().__init__(
@@ -25,3 +42,84 @@ class SingleStoreEmbeddings(OpenAIEmbeddings):
             model=model_name,
             **kwargs,
         )
+
+
+def SingleStoreEmbeddingsFactory(
+    model_name: str,
+    api_key: Optional[str] = None,
+    http_client: Optional[httpx.Client] = None,
+    obo_token_getter: Optional[Callable[[], Optional[str]]] = None,
+    **kwargs: Any,
+) -> Union[OpenAIEmbeddings, BedrockEmbeddings]:
+    """Return an embeddings model instance (OpenAIEmbeddings or BedrockEmbeddings).
+    """
+    inference_api_manager = (
+        manage_workspaces().organizations.current.inference_apis
+    )
+    info = inference_api_manager.get(model_name=model_name)
+    token_env = os.environ.get('SINGLESTOREDB_USER_TOKEN')
+    token = api_key if api_key is not None else token_env
+
+    if info.hosting_platform == 'Amazon':
+        # Instantiate Bedrock client
+        cfg_kwargs = {
+            '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
+
+        cfg = Config(**cfg_kwargs)
+        client = boto3.client(
+            'bedrock-runtime',
+            endpoint_url=info.connection_url,
+            region_name='us-east-1',
+            aws_access_key_id='placeholder',
+            aws_secret_access_key='placeholder',
+            config=cfg,
+        )
+
+        def _inject_headers(request: Any, **_ignored: Any) -> None:
+            """Inject dynamic auth/OBO headers prior to Bedrock sending."""
+            if obo_token_getter is not None:
+                obo_val = obo_token_getter()
+                if obo_val:
+                    request.headers['X-S2-OBO'] = obo_val
+            if token:
+                request.headers['Authorization'] = f'Bearer {token}'
+            request.headers.pop('X-Amz-Date', None)
+            request.headers.pop('X-Amz-Security-Token', None)
+
+        emitter = client._endpoint._event_emitter
+        emitter.register_first(
+            'before-send.bedrock-runtime.InvokeModel',
+            _inject_headers,
+        )
+        emitter.register_first(
+            'before-send.bedrock-runtime.InvokeModelWithResponseStream',
+            _inject_headers,
+        )
+
+        return BedrockEmbeddings(
+            model_id=model_name,
+            endpoint_url=info.connection_url,
+            region_name='us-east-1',
+            aws_access_key_id='placeholder',
+            aws_secret_access_key='placeholder',
+            client=client,
+            **kwargs,
+        )
+
+    # OpenAI / Azure OpenAI path
+    openai_kwargs = dict(
+        base_url=info.connection_url,
+        api_key=token,
+        model=model_name,
+    )
+    if http_client is not None:
+        openai_kwargs['http_client'] = http_client
+    return OpenAIEmbeddings(
+        **openai_kwargs,
+        **kwargs,
+    )

singlestoredb/functions/ext/asgi.py

@@ -42,6 +42,7 @@ import tempfile
 import textwrap
 import threading
 import time
+import traceback
 import typing
 import urllib
 import uuid
@@ -709,21 +710,21 @@ class Application(object):
     error_response_dict: Dict[str, Any] = dict(
         type='http.response.start',
         status=500,
-        headers=[(b'content-type', b'application/json')],
+        headers=[(b'content-type', b'text/plain')],
     )
 
     # Timeout response start
     timeout_response_dict: Dict[str, Any] = dict(
         type='http.response.start',
         status=504,
-        headers=[(b'content-type', b'application/json')],
+        headers=[(b'content-type', b'text/plain')],
     )
 
     # Cancel response start
     cancel_response_dict: Dict[str, Any] = dict(
         type='http.response.start',
         status=503,
-        headers=[(b'content-type', b'application/json')],
+        headers=[(b'content-type', b'text/plain')],
     )
 
     # JSON response start
@@ -1247,12 +1248,10 @@ class Application(object):
                         'timeout': func_info['timeout'],
                     },
                 )
-                body = json.dumps(
-                    dict(
-                        error='[TimeoutError] Function call timed out after ' +
-                        str(func_info['timeout']) +
-                        ' seconds',
-                    ),
+                body = (
+                    'TimeoutError: Function call timed out after ' +
+                    str(func_info['timeout']) +
+                    ' seconds'
                 ).encode('utf-8')
                 await send(self.timeout_response_dict)
 
@@ -1265,11 +1264,7 @@ class Application(object):
                         'function_name': func_name.decode('utf-8'),
                     },
                 )
-                body = json.dumps(
-                    dict(
-                        error='[CancelledError] Function call was cancelled',
-                    ),
-                ).encode('utf-8')
+                body = b'CancelledError: Function call was cancelled'
                 await send(self.cancel_response_dict)
 
             except Exception as e:
@@ -1282,11 +1277,12 @@ class Application(object):
                         'exception_type': type(e).__name__,
                     },
                 )
-                body = json.dumps(
-                    dict(
-                        error=f'[{type(e).__name__}] {str(e).strip()}',
-                    ),
-                ).encode('utf-8')
+                msg = traceback.format_exc().strip().split(' File ')[-1]
+                if msg.startswith('"/tmp/ipykernel_'):
+                    msg = 'Line ' + msg.split(', line ')[-1]
+                else:
+                    msg = 'File ' + msg
+                body = msg.encode('utf-8')
                 await send(self.error_response_dict)
 
             finally:

singlestoredb/management/inference_api.py

@@ -23,6 +23,7 @@ class InferenceAPIInfo(object):
     name: str
     connection_url: str
     project_id: str
+    hosting_platform: str
 
     def __init__(
         self,
@@ -31,12 +32,14 @@ class InferenceAPIInfo(object):
         name: str,
         connection_url: str,
         project_id: str,
+        hosting_platform: str,
     ):
         self.service_id = service_id
         self.connection_url = connection_url
         self.model_name = model_name
         self.name = name
         self.project_id = project_id
+        self.hosting_platform = hosting_platform
 
     @classmethod
     def from_dict(
@@ -62,6 +65,7 @@ class InferenceAPIInfo(object):
             model_name=obj['modelName'],
             name=obj['name'],
             connection_url=obj['connectionURL'],
+            hosting_platform=obj['hostingPlatform'],
         )
         return out
 

singlestoredb/tests/test_management.py

@@ -873,12 +873,20 @@ class TestStage(unittest.TestCase):
     def test_file_object(self):
         st = self.wg.stage
 
-        st.mkdir('obj_test')
-        st.mkdir('obj_test/nest_1')
+        obj_test_dir = f'obj_test_{id(self)}'
 
-        f1 = st.upload_file(TEST_DIR / 'test.sql', 'obj_test.sql')
-        f2 = st.upload_file(TEST_DIR / 'test.sql', 'obj_test/nest_1/obj_test.sql')
-        d2 = st.info('obj_test/nest_1/')
+        st.mkdir(obj_test_dir)
+        st.mkdir(f'{obj_test_dir}/nest_1')
+
+        obj_test_sql = f'obj_test_{id(self)}.sql'
+        obj_test_2_sql = f'obj_test_2_{id(self)}.sql'
+
+        f1 = st.upload_file(TEST_DIR / 'test.sql', obj_test_sql)
+        f2 = st.upload_file(
+            TEST_DIR / 'test.sql',
+            f'{obj_test_dir}/nest_1/{obj_test_sql}',
+        )
+        d2 = st.info(f'{obj_test_dir}/nest_1/')
 
         # is_file / is_dir
         assert not f1.is_dir()
@@ -889,17 +897,17 @@ class TestStage(unittest.TestCase):
         assert not d2.is_file()
 
         # abspath / basename / dirname / exists
-        assert f1.abspath() == 'obj_test.sql'
-        assert f1.basename() == 'obj_test.sql'
+        assert f1.abspath() == obj_test_sql
+        assert f1.basename() == obj_test_sql
         assert f1.dirname() == '/'
         assert f1.exists()
-        assert f2.abspath() == 'obj_test/nest_1/obj_test.sql'
-        assert f2.basename() == 'obj_test.sql'
-        assert f2.dirname() == 'obj_test/nest_1/'
+        assert f2.abspath() == f'{obj_test_dir}/nest_1/{obj_test_sql}'
+        assert f2.basename() == obj_test_sql
+        assert f2.dirname() == f'{obj_test_dir}/nest_1/'
         assert f2.exists()
-        assert d2.abspath() == 'obj_test/nest_1/'
+        assert d2.abspath() == f'{obj_test_dir}/nest_1/'
         assert d2.basename() == 'nest_1'
-        assert d2.dirname() == 'obj_test/'
+        assert d2.dirname() == f'{obj_test_dir}/'
         assert d2.exists()
 
         # download
@@ -910,9 +918,9 @@ class TestStage(unittest.TestCase):
         with self.assertRaises(IsADirectoryError):
             d2.remove()
 
-        assert st.is_file('obj_test.sql')
+        assert st.is_file(obj_test_sql)
         f1.remove()
-        assert not st.is_file('obj_test.sql')
+        assert not st.is_file(obj_test_sql)
 
         # removedirs
         with self.assertRaises(NotADirectoryError):
@@ -923,8 +931,8 @@ class TestStage(unittest.TestCase):
         assert not st.exists(d2.path)
 
         # rmdir
-        f1 = st.upload_file(TEST_DIR / 'test.sql', 'obj_test.sql')
-        d2 = st.mkdir('obj_test/nest_1')
+        f1 = st.upload_file(TEST_DIR / 'test.sql', obj_test_sql)
+        d2 = st.mkdir(f'{obj_test_dir}/nest_1')
 
         assert st.exists(f1.path)
         assert st.exists(d2.path)
@@ -937,20 +945,20 @@ class TestStage(unittest.TestCase):
 
         d2.rmdir()
 
-        assert not st.exists('obj_test/nest_1/')
-        assert not st.exists('obj_test')
+        assert not st.exists(f'{obj_test_dir}/nest_1/')
+        assert not st.exists(obj_test_dir)
 
         # mtime / ctime
         assert f1.getmtime() > 0
         assert f1.getctime() > 0
 
         # rename
-        assert st.exists('obj_test.sql')
-        assert not st.exists('obj_test_2.sql')
-        f1.rename('obj_test_2.sql')
-        assert not st.exists('obj_test.sql')
-        assert st.exists('obj_test_2.sql')
-        assert f1.abspath() == 'obj_test_2.sql'
+        assert st.exists(obj_test_sql)
+        assert not st.exists(obj_test_2_sql)
+        f1.rename(obj_test_2_sql)
+        assert not st.exists(obj_test_sql)
+        assert st.exists(obj_test_2_sql)
+        assert f1.abspath() == obj_test_2_sql
 
 
 @pytest.mark.management

{singlestoredb-1.15.5.dist-info → singlestoredb-1.15.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: singlestoredb
-Version: 1.15.5
+Version: 1.15.7
 Summary: Interface to the SingleStoreDB database and workspace management APIs
 Home-page: https://github.com/singlestore-labs/singlestoredb-python
 Author: SingleStore

{singlestoredb-1.15.5.dist-info → singlestoredb-1.15.7.dist-info}/RECORD

@@ -1,16 +1,16 @@
-_singlestoredb_accel.abi3.so,sha256=x1zaDVImJ4CWAfQPrICls-q50fsaCFB1_bbelHM2Y-Q,207264
-singlestoredb-1.15.5.dist-info/RECORD,,
-singlestoredb-1.15.5.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
-singlestoredb-1.15.5.dist-info/WHEEL,sha256=_VEguvlLpUd-c8RbFMA4yMIVNMBv2LhpxYLCEQ-Bogk,113
-singlestoredb-1.15.5.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
-singlestoredb-1.15.5.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
-singlestoredb-1.15.5.dist-info/METADATA,sha256=deTmpWalhUYHaanjCuaVrQLIld_rfhdAPF6a1Fwa18A,5804
+_singlestoredb_accel.abi3.so,sha256=8pD1GEJZQTnRnteVoJG9U1oJfe6vHgWC2koL52xF9G8,207264
+singlestoredb-1.15.7.dist-info/RECORD,,
+singlestoredb-1.15.7.dist-info/LICENSE,sha256=Mlq78idURT-9G026aMYswwwnnrLcgzTLuXeAs5hjDLM,11341
+singlestoredb-1.15.7.dist-info/WHEEL,sha256=_VEguvlLpUd-c8RbFMA4yMIVNMBv2LhpxYLCEQ-Bogk,113
+singlestoredb-1.15.7.dist-info/entry_points.txt,sha256=bSLaTWB5zGjpVYPAaI46MkkDup0su-eb3uAhCNYuRV0,48
+singlestoredb-1.15.7.dist-info/top_level.txt,sha256=lA65Vf4qAMfg_s1oG3LEO90h4t1Z-SPDbRqkevI3bSY,40
+singlestoredb-1.15.7.dist-info/METADATA,sha256=N42M0mnELo1khbLSdxNkFIQX6QBE0RNV6hgPdTwJjg4,5804
 sqlx/magic.py,sha256=JsS9_9aBFaOt91Torm1JPN0c8qB2QmYJmNSKtbSQIY0,3509
 sqlx/__init__.py,sha256=aBYiU8DZXCogvWu3yWafOz7bZS5WWwLZXj7oL0dXGyU,85
 singlestoredb/auth.py,sha256=u8D9tpKzrqa4ssaHjyZnGDX1q8XBpGtuoOkTkSv7B28,7599
 singlestoredb/config.py,sha256=aBdMrPEaNSH-QLi1AXoQaSJsZ9f6ZXoFPN-74Trr6sQ,13935
 singlestoredb/vectorstore.py,sha256=BZb8e7m02_XVHqOyu8tA94R6kHb3n-BC8F08JyJwDzY,8408
-singlestoredb/__init__.py,sha256=laZYv8Vwx1AbBnqnp8ESXHckpITWCEmLX7WlFArId-Y,2272
+singlestoredb/__init__.py,sha256=UYEMO4CiO41DrAiO6xne5VttZqH7BvSD2Q-skkFRNeU,2272
 singlestoredb/types.py,sha256=Qp_PWYjSYG6PRnmXAZZ7K2QehUqfoG4KSllI3O1stPE,10397
 singlestoredb/connection.py,sha256=ELk3-UpM6RaB993aIt08MydKiiDnejHQ1s8EFiacrAI,46055
 singlestoredb/pytest.py,sha256=OyF3BO9mgxenifYhOihnzGk8WzCJ_zN5_mxe8XyFPOc,9074
@@ -45,7 +45,7 @@ singlestoredb/tests/test2.ipynb,sha256=yd1PE1VK-DwiRd6mYS4_0cPBtuVkvcDtycvTwD-Yn
 singlestoredb/tests/test_ext_func_data.py,sha256=kyNklkX1RxSiahI0LZjpqhg3LGDs0iwv8iHuXW3AcSo,47515
 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=R5g5QHWCSPxCrwMUOD0BL-touGIDvqQfDve-YmZyhX4,51973
+singlestoredb/tests/test_management.py,sha256=BK58UWs6L_9h1ePUNjnIJ1fVAEvBvC8rP7HtnfoANYQ,52205
 singlestoredb/tests/test_udf.py,sha256=ycqSwwwCw3Mq4r0cyIylg63ReSjP4KbMX_7XqoYm-dg,29676
 singlestoredb/tests/test_http.py,sha256=RXasTqBWRn__omj0eLFTJYIbZjd0PPdIV2d4Cqz0MC8,8580
 singlestoredb/tests/utils.py,sha256=2A2tEdD3t8aXWUnHtAIcFlWrflsz2MlMcCbUDaAG29c,4995
@@ -69,7 +69,7 @@ singlestoredb/management/__init__.py,sha256=8q7i6-Cr9x-oZ8-NVAvTo_qtfHEndX4wx2g6
 singlestoredb/management/export.py,sha256=yR-yZUE9USFrP5OR_5iLFqEc8GLiKDQypSEp08CmT5k,9083
 singlestoredb/management/utils.py,sha256=QIhZCZSRaDbAG35xu1_n7ihmRXON8swc-gEK2FGYutI,13203
 singlestoredb/management/cluster.py,sha256=vDefpp2IMZRawvueIqZK2pePWVNnPWb6Szrt8mO8gmg,14419
-singlestoredb/management/inference_api.py,sha256=L6eFqaUaPugF_cmrZ4xlArj8CIv25vWqQs1vwgKPEF4,2583
+singlestoredb/management/inference_api.py,sha256=px4JrziJQMwG73iNnYw_95np9DDrp5zo5mKz8c5EF9o,2742
 singlestoredb/management/workspace.py,sha256=frVNqwfUgth1WWhcf5rZqz9nIcPkMtmCb5hKVx1h5SI,61953
 singlestoredb/management/manager.py,sha256=k8JKPXyJVHI7gFwqIhxhUtQr_KdBCXw7Dw4GCdCtdKg,9352
 singlestoredb/management/billing_usage.py,sha256=9ighjIpcopgIyJOktBYQ6pahBZmWGHOPyyCW4gu9FGs,3735
@@ -106,9 +106,9 @@ singlestoredb/docstring/tests/test_numpydoc.py,sha256=9qr9jHr4feTF1E09c8J6SHWSXU
 singlestoredb/docstring/tests/test_google.py,sha256=2dVyFN-NjiuyUC3ZcGEm33aENAZNh9F7Eq_ErmyneIA,27955
 singlestoredb/http/__init__.py,sha256=A_2ZUCCpvRYIA6YDpPy57wL5R1eZ5SfP6I1To5nfJ2s,912
 singlestoredb/http/connection.py,sha256=fCgh3PdYJNYVPMaUe_NlMb7ducxC06se1XfyqOpPpJw,39727
-singlestoredb/ai/__init__.py,sha256=-uNcq-bY-AiWhZ5Plq2ZXtfIVL4PaifMJsJf58rdN8I,114
-singlestoredb/ai/chat.py,sha256=5vi_jVSH8xJ_WIn_mst8uw5cv9tJJBn3AFYRM4LhYI8,1268
-singlestoredb/ai/embeddings.py,sha256=X3g0sJNDVOzXzZwoXz3M3ch-IERQXNkHxuH4cj125I8,815
+singlestoredb/ai/__init__.py,sha256=zNSUVDzDsyIfW7-obV5iRRhDjPqgPzG_EJJaT0_jtLw,284
+singlestoredb/ai/chat.py,sha256=uI_j1sySFO4jpUA3bcUL0Gki0F3jEbLHAq0hQLxqv3w,5074
+singlestoredb/ai/embeddings.py,sha256=yj9fQzwiocy78cBPfoOB9dok4gnWyE0IghCw7JIlTGI,4020
 singlestoredb/mysql/protocol.py,sha256=PfdILK1oK2hlg1etmYZtrKf1obPlcLC7FyEJHEpIplc,14885
 singlestoredb/mysql/cursors.py,sha256=aOLfHkj83aYZPOVuhJPkZ83CWByszIBRynq0fqSaWvY,27046
 singlestoredb/mysql/__init__.py,sha256=olUTAvkiERhDW41JXQMawkg-i0tvBEkoTkII1tt6lxU,4492
@@ -156,7 +156,7 @@ singlestoredb/functions/dtypes.py,sha256=DgJaNXouJ2t-qIqDiQlUYU9IhkXXUTigWeE_MAc
 singlestoredb/functions/utils.py,sha256=1L0Phgzq0XdWK3ecfOOydq4zV955yCwpDoAaCYRGldk,10769
 singlestoredb/functions/signature.py,sha256=h2vFVNP07d5a3gi7zMiM_sztDUNK_HlJWR-Rl3nMxPA,45545
 singlestoredb/functions/ext/timer.py,sha256=-PR__KbhwAMW4PXJ4fGri2FfrU0jRyz6e6yvmySmjaw,2706
-singlestoredb/functions/ext/asgi.py,sha256=HWbSuYpAOfq-iS7oKDdXCkz5OIaMOGnXYwrTIVzFqOM,72370
+singlestoredb/functions/ext/asgi.py,sha256=C6UJXunHY-dISp4k4CH-poEt8dyzqIjLQpuj06_ttRA,72268
 singlestoredb/functions/ext/arrow.py,sha256=WB7n1ACslyd8nlbFzUvlbxn1BVuEjA9-BGBEqCWlSOo,9061
 singlestoredb/functions/ext/__init__.py,sha256=1oLL20yLB1GL9IbFiZD8OReDqiCpFr-yetIR6x1cNkI,23
 singlestoredb/functions/ext/utils.py,sha256=oU2NVmkjcS0QHLfdB8SBiRylVq-r0VzTy8nxGvAgjow,6938