singlestoredb 1.3.0__tar.gz → 1.4.0__tar.gz

{singlestoredb-1.3.0/singlestoredb.egg-info → singlestoredb-1.4.0}/PKG-INFO

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

{singlestoredb-1.3.0 → singlestoredb-1.4.0}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = singlestoredb
-version = 1.3.0
+version = 1.4.0
 description = Interface to the SingleStoreDB database and workspace management APIs
 long_description = file: README.md
 long_description_content_type = text/markdown

{singlestoredb-1.3.0 → singlestoredb-1.4.0}/singlestoredb/__init__.py

@@ -13,7 +13,7 @@ Examples
 
 """
 
-__version__ = '1.3.0'
+__version__ = '1.4.0'
 
 from typing import Any
 

singlestoredb-1.4.0/singlestoredb/ai/__init__.py

@@ -0,0 +1 @@
+from .embeddings import SingleStoreEmbeddings  # noqa: F401

singlestoredb-1.4.0/singlestoredb/ai/embeddings.py

@@ -0,0 +1,24 @@
+import os as _os
+from typing import Any
+
+try:
+    from langchain_community.embeddings.ollama import OllamaEmbeddings
+except ImportError:
+    raise ImportError(
+        'Could not import langchain_community python package. '
+        'Please install it with `pip install langchain_community`.',
+    )
+
+
+class SingleStoreEmbeddings(OllamaEmbeddings):
+
+    def __init__(self, **kwargs: Any):
+        url = _os.getenv('SINGLESTORE_AI_EXPERIMENTAL_URL')
+        if not url:
+            raise ValueError(
+                "Environment variable 'SINGLESTORE_AI_EXPERIMENTAL_URL' must be set",
+            )
+
+        base_url = url.strip('/v1')
+        kwargs = {'model': 'nomic-embed-text', **kwargs}
+        super().__init__(base_url=base_url, **kwargs)

{singlestoredb-1.3.0 → singlestoredb-1.4.0}/singlestoredb/functions/ext/asgi.py

@@ -562,7 +562,6 @@ class Application(object):
         elif method == 'GET' and path == self.show_create_function_path:
             host = headers.get(b'host', b'localhost:80')
             reflected_url = f'{scope["scheme"]}://{host.decode("utf-8")}/invoke'
-            data_format = 'json' if b'json' in content_type else 'rowdat_1'
 
             syntax = []
             for key, (endpoint, endpoint_info) in self.endpoints.items():
@@ -571,7 +570,7 @@ class Application(object):
                         signature_to_sql(
                             endpoint_info['signature'],
                             url=self.url or reflected_url,
-                            data_format=data_format,
+                            data_format=self.data_format,
                         ),
                     )
             body = '\n'.join(syntax).encode('utf-8')

{singlestoredb-1.3.0 → singlestoredb-1.4.0}/singlestoredb/management/organization.py

@@ -35,11 +35,11 @@ class Secret(object):
         self,
         id: str,
         name: str,
-        value: str,
         created_by: str,
         created_at: Union[str, datetime.datetime],
         last_updated_by: str,
         last_updated_at: Union[str, datetime.datetime],
+        value: Optional[str] = None,
         deleted_by: Optional[str] = None,
         deleted_at: Optional[Union[str, datetime.datetime]] = None,
     ):
@@ -88,11 +88,11 @@ class Secret(object):
         out = cls(
             id=obj['secretID'],
             name=obj['name'],
-            value=obj['value'],
             created_by=obj['createdBy'],
             created_at=obj['createdAt'],
             last_updated_by=obj['lastUpdatedBy'],
             last_updated_at=obj['lastUpdatedAt'],
+            value=obj.get('value'),
             deleted_by=obj.get('deletedBy'),
             deleted_at=obj.get('deletedAt'),
         )

{singlestoredb-1.3.0 → singlestoredb-1.4.0}/singlestoredb/management/workspace.py

@@ -39,7 +39,7 @@ def get_organization() -> Organization:
     return manage_workspaces().organization
 
 
-def get_secret(name: str) -> str:
+def get_secret(name: str) -> Optional[str]:
     """Get a secret from the organization."""
     return get_organization().get_secret(name).value
 

{singlestoredb-1.3.0 → singlestoredb-1.4.0}/singlestoredb/mysql/connection.py

@@ -1858,6 +1858,7 @@ class MySQLResultSV(MySQLResult):
                 parse_json=connection.parse_json,
                 invalid_values=connection.invalid_values,
                 unbuffered=unbuffered,
+                encoding_errors=connection.encoding_errors,
             ).items() if v is not UNSET
         }
         self._read_rowdata_packet = functools.partial(

{singlestoredb-1.3.0 → singlestoredb-1.4.0}/singlestoredb/notebook/_objects.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 import functools
 from typing import Any
+from typing import Optional
 
 from ..management import workspace as _ws
 
@@ -8,12 +9,12 @@ from ..management import workspace as _ws
 class Secrets(object):
     """Wrapper for accessing secrets as object attributes."""
 
-    def __getattr__(self, name: str) -> str:
+    def __getattr__(self, name: str) -> Optional[str]:
         if name.startswith('_ipython') or name.startswith('_repr_'):
             raise AttributeError(name)
         return _ws.get_secret(name)
 
-    def __getitem__(self, name: str) -> str:
+    def __getitem__(self, name: str) -> Optional[str]:
         return _ws.get_secret(name)
 
 

{singlestoredb-1.3.0 → singlestoredb-1.4.0/singlestoredb.egg-info}/PKG-INFO

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

{singlestoredb-1.3.0 → singlestoredb-1.4.0}/singlestoredb.egg-info/SOURCES.txt

@@ -17,6 +17,8 @@ singlestoredb.egg-info/dependency_links.txt
 singlestoredb.egg-info/entry_points.txt
 singlestoredb.egg-info/requires.txt
 singlestoredb.egg-info/top_level.txt
+singlestoredb/ai/__init__.py
+singlestoredb/ai/embeddings.py
 singlestoredb/alchemy/__init__.py
 singlestoredb/functions/__init__.py
 singlestoredb/functions/decorator.py