matrixone-python-sdk 0.1.14__tar.gz → 0.1.15__tar.gz

{matrixone_python_sdk-0.1.14/matrixone_python_sdk.egg-info → matrixone_python_sdk-0.1.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrixone-python-sdk
-Version: 0.1.14
+Version: 0.1.15
 Summary: A comprehensive Python SDK for MatrixOne database operations with vector search, fulltext search, and advanced features
 Home-page: https://github.com/matrixorigin/matrixone
 Author: MatrixOne Team

{matrixone_python_sdk-0.1.14 → matrixone_python_sdk-0.1.15}/matrixone/async_client.py

@@ -23,11 +23,13 @@ except ImportError:
 
 try:
     from sqlalchemy import text
+    from sqlalchemy.engine import URL
     from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
 except ImportError:
     create_async_engine = None
     AsyncEngine = None
     text = None
+    URL = None
 
 from contextlib import asynccontextmanager
 from datetime import datetime
@@ -862,19 +864,21 @@ class AsyncClient(BaseMatrixOneClient):
         if not create_async_engine:
             raise ConnectionError("SQLAlchemy async engine not available. Please install sqlalchemy[asyncio]")
 
-        # Build connection string for async engine
-        connection_string = (
-            f"mysql+aiomysql://{self._connection_params['user']}:"
-            f"{self._connection_params['password']}@"
-            f"{self._connection_params['host']}:"
-            f"{self._connection_params['port']}/"
-            f"{self._connection_params['database'] or ''}"
-            f"?charset={self._connection_params['charset']}"
+        # Build connection URL using SQLAlchemy's URL.create() for proper escaping
+        # of special characters in user/password (e.g., @, #, %, etc.)
+        url = URL.create(
+            drivername="mysql+aiomysql",
+            username=self._connection_params["user"],
+            password=self._connection_params["password"],
+            host=self._connection_params["host"],
+            port=self._connection_params["port"],
+            database=self._connection_params["database"] or None,
+            query={"charset": self._connection_params["charset"]},
         )
 
         # Create async engine with connection pooling
         engine = create_async_engine(
-            connection_string,
+            url,
             pool_size=5,  # Smaller pool size for testing
             max_overflow=10,  # Smaller max overflow
             pool_timeout=30,  # Default pool timeout

{matrixone_python_sdk-0.1.14 → matrixone_python_sdk-0.1.15}/matrixone/client.py

@@ -25,7 +25,7 @@ if TYPE_CHECKING:
     from sqlalchemy.engine import Connection
 
 from sqlalchemy import create_engine
-from sqlalchemy.engine import Engine
+from sqlalchemy.engine import Engine, URL
 from sqlalchemy.orm import Session as SQLAlchemySession
 
 from .account import AccountManager
@@ -548,27 +548,34 @@ class Client(BaseMatrixOneClient):
 
     def _create_engine(self) -> Engine:
         """Create SQLAlchemy engine with connection pooling"""
-        # Build connection string
-        connection_string = (
-            f"mysql+pymysql://{self._connection_params['user']}:"
-            f"{self._connection_params['password']}@"
-            f"{self._connection_params['host']}:"
-            f"{self._connection_params['port']}/"
-            f"{self._connection_params['database']}"
+        # Build connection URL using SQLAlchemy's URL.create() for proper escaping
+        # of special characters in user/password (e.g., @, #, %, etc.)
+        connect_args = {}
+        if self._connection_params.get("ssl_ca"):
+            connect_args["ssl"] = {"ca": self._connection_params["ssl_ca"]}
+            if self._connection_params.get("ssl_cert"):
+                connect_args["ssl"]["cert"] = self._connection_params["ssl_cert"]
+            if self._connection_params.get("ssl_key"):
+                connect_args["ssl"]["key"] = self._connection_params["ssl_key"]
+
+        url = URL.create(
+            drivername="mysql+pymysql",
+            username=self._connection_params["user"],
+            password=self._connection_params["password"],
+            host=self._connection_params["host"],
+            port=self._connection_params["port"],
+            database=self._connection_params["database"] or None,
         )
 
-        # Add SSL parameters if needed
-        if "ssl_ca" in self._connection_params:
-            connection_string += f"?ssl_ca={self._connection_params['ssl_ca']}"
-
         # Create engine with connection pooling
         engine = create_engine(
-            connection_string,
+            url,
             pool_size=self.pool_size,
             max_overflow=self.max_overflow,
             pool_timeout=self.pool_timeout,
             pool_recycle=self.pool_recycle,
             pool_pre_ping=True,  # Enable connection health checks
+            connect_args=connect_args,
         )
 
         # Replace the dialect with MatrixOne dialect for proper vector type support

{matrixone_python_sdk-0.1.14 → matrixone_python_sdk-0.1.15}/matrixone/sqlalchemy_ext/dialect.py

@@ -17,7 +17,30 @@ MatrixOne SQLAlchemy dialect support.
 """
 
 import sqlalchemy
-from sqlalchemy.dialects.mysql.base import MySQLDialect
+from sqlalchemy.dialects.mysql.base import MySQLDialect, ischema_names as mysql_ischema_names
+
+from .vector_type import VectorType, VectorPrecision
+
+
+class _Vecf32Type(VectorType):
+    """VectorType with f32 precision for ischema_names registration."""
+
+    def __init__(self, dimension=None):
+        super().__init__(dimension=dimension, precision=VectorPrecision.F32)
+
+
+class _Vecf64Type(VectorType):
+    """VectorType with f64 precision for ischema_names registration."""
+
+    def __init__(self, dimension=None):
+        super().__init__(dimension=dimension, precision=VectorPrecision.F64)
+
+
+# Extend MySQL's ischema_names with MatrixOne-specific types
+_mo_ischema_names = mysql_ischema_names.copy()
+_mo_ischema_names["vecf32"] = _Vecf32Type
+_mo_ischema_names["vecf64"] = _Vecf64Type
+_mo_ischema_names["bool"] = mysql_ischema_names["boolean"]
 
 
 class MatrixOneDialect(MySQLDialect):
@@ -62,6 +85,8 @@ class MatrixOneDialect(MySQLDialect):
     name = "matrixone"
     driver = "pymysql"
 
+    ischema_names = _mo_ischema_names
+
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         # Initialize missing MySQL dialect attributes

{matrixone_python_sdk-0.1.14 → matrixone_python_sdk-0.1.15/matrixone_python_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrixone-python-sdk
-Version: 0.1.14
+Version: 0.1.15
 Summary: A comprehensive Python SDK for MatrixOne database operations with vector search, fulltext search, and advanced features
 Home-page: https://github.com/matrixorigin/matrixone
 Author: MatrixOne Team

{matrixone_python_sdk-0.1.14 → matrixone_python_sdk-0.1.15}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "matrixone-python-sdk"
-version = "0.1.14"
+version = "0.1.15"
 description = "A comprehensive Python SDK for MatrixOne database operations with vector search, fulltext search, and advanced features"
 readme = "README_USER.md"
 license = {text = "Apache-2.0"}