fix style

Author: volodymyr-memsql

docs/src/whatsnew.rst

@@ -11,7 +11,7 @@ This document outlines features and improvements from each release.
 v1.14.0 - May 28, 2025
 ------------------------
 * Add ``vector_db`` property to a ``Connection`` object to get the vectorstor API to the current database
-* Add ``vector_db(...)`` function to connect to a database and optain a vectorstor API 
+* Add ``vector_db(...)`` function to connect to a database and optain a vectorstor API
 
 v1.13.0 - April 16, 2025
 ------------------------

requirements.txt

@@ -3,8 +3,8 @@ parsimonious
 PyJWT
 requests
 setuptools
+singlestore-vecorstore
 sqlparams
 tomli>=1.1.0; python_version < '3.11'
 typing_extensions<=4.13.2
 wheel
-singlestore-vecorstore>=0.1.1

setup.cfg

@@ -24,9 +24,9 @@ install_requires =
     parsimonious
     requests
     setuptools
+    singlestore-vecorstore
     sqlparams
     wheel
-    singlestore-vecorstore>=0.1.1
     tomli>=1.1.0;python_version < '3.11'
     typing-extensions<=4.13.2;python_version < '3.11'
 python_requires = >=3.8

singlestoredb/__init__.py

@@ -31,7 +31,15 @@
     Date, Time, Timestamp, DateFromTicks, TimeFromTicks, TimestampFromTicks,
     Binary, STRING, BINARY, NUMBER, DATETIME, ROWID,
 )
-from .vectorstore import vector_db
+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,
+)
+
 
 #
 # This function is defined here to prevent the side-effect of

singlestoredb/connection.py

@@ -1,6 +1,5 @@
 #!/usr/bin/env python
 """SingleStoreDB connections and cursors."""
-
 import abc
 import inspect
 import io
@@ -24,7 +23,6 @@
 from urllib.parse import unquote_plus
 from urllib.parse import urlparse
 
-
 import sqlparams
 try:
     from pandas import DataFrame
@@ -1293,13 +1291,14 @@ def show(self) -> ShowAccessor:
     @property
     def vector_db(self) -> Any:
         """
-        Get vectorstore API accessor 
+        Get vectorstore API accessor
         """
         from vectorstore import VectorDB
         if not hasattr(self, '_vector_db'):
             self._vector_db = VectorDB(connection=self)
         return self._vector_db
 
+
 #
 # NOTE: When adding parameters to this function, you should always
 #       make the value optional with a default of None. The options

singlestoredb/tests/test_vectorstore.py

@@ -1,10 +1,11 @@
 import os
 import unittest
 
+from vectorstore import VectorDB
+
 import singlestoredb as s2
 from . import utils
 
-from vectorstore import VectorDB
 
 class TestVectorDB(unittest.TestCase):
 
@@ -14,28 +15,34 @@ class TestVectorDB(unittest.TestCase):
     dbexisted: bool = False
 
     @classmethod
-    def setUpClass(cls):
+    def setUpClass(cls) -> None:
         sql_file = os.path.join(os.path.dirname(__file__), 'empty.sql')
-        cls.dbname, cls.dbexisted = utils.load_sql(sql_file)
+        cls.dbname, cls.dbexisted = utils.load_sql(sql_file)  # type: ignore
 
     @classmethod
-    def tearDownClass(cls):
+    def tearDownClass(cls) -> None:
         if not cls.dbexisted:
-            utils.drop_database(cls.dbname)
+            utils.drop_database(cls.dbname)  # type: ignore
 
-    def test_vectordb_from_params(self):
+    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"
+        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")
+        assert index.tags == {'name': 'test_tag'}
+        assert db.has_index('test_index')
 
-    def test_vectordb_from_connection(self):
+    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"
+            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")
+            assert index.tags == {'name': 'test_tag'}
+            assert db.has_index('test_index_1')

singlestoredb/vectorstore.py

@@ -1,34 +1,35 @@
-from __future__ import annotations
-from typing import Optional, Dict, Callable, Any
+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
 
-from vectorstore import (
-    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
-)
 
 def vector_db(
     host: Optional[str] = None, user: Optional[str] = None,
@@ -62,9 +63,9 @@ def vector_db(
     pool_size: Optional[int] = 5,
     max_overflow: Optional[int] = 10,
     timeout: Optional[float] = 30,
-) -> "VectorDB":
+) -> Any:
     """
-    Return a vectorstore API connection. 
+    Return a vectorstore API connection.
     Database should be specified in the URL or as a keyword.
 
     Parameters
@@ -87,7 +88,7 @@ def vector_db(
         Database port. This defaults to 3306 for non-HTTP connections, 80
         for HTTP connections, and 443 for HTTPS connections.
     database : str, optional
-        Database name. 
+        Database name.
     pure_python : bool, optional
         Use the connector in pure Python mode
     local_infile : bool, optional
@@ -167,23 +168,25 @@ def vector_db(
 
     """
     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)
+    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,
+    )