matrixone-python-sdk 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

matrixone/async_client.py

@@ -1488,11 +1488,12 @@ class AsyncClient(BaseMatrixOneClient):
 
     async def connect(
         self,
-        host: str,
-        port: int,
-        user: str,
-        password: str,
-        database: str = None,
+        *,
+        host: str = "localhost",
+        port: int = 6001,
+        user: str = "root",
+        password: str = "111",
+        database: str,
         account: Optional[str] = None,
         role: Optional[str] = None,
         charset: str = "utf8mb4",
@@ -1573,8 +1574,12 @@ class AsyncClient(BaseMatrixOneClient):
 
             self.logger.log_connection(host, port, final_user, database or "default", success=True)
 
-            # Setup connection hook if provided
-            if on_connect:
+            # Setup connection hook (default to ENABLE_ALL if not provided)
+            # Allow empty list [] to explicitly disable hooks
+            if on_connect is None:
+                on_connect = [ConnectionAction.ENABLE_ALL]
+
+            if on_connect:  # Only setup if not empty list
                 self._setup_connection_hook(on_connect)
                 # Execute the hook once immediately for the initial connection
                 await self._execute_connection_hook_immediately(on_connect)
@@ -1582,7 +1587,18 @@ class AsyncClient(BaseMatrixOneClient):
         except Exception as e:
             self.logger.log_connection(host, port, final_user, database or "default", success=False)
             self.logger.log_error(e, context="Async connection")
-            raise ConnectionError(f"Failed to connect to MatrixOne: {e}")
+
+            # Provide user-friendly error messages for common issues
+            error_msg = str(e)
+            if 'Unknown database' in error_msg or '1049' in error_msg:
+                db_name = database or "default"
+                raise ConnectionError(
+                    f"Database '{db_name}' does not exist. Please create it first:\n"
+                    f"  mysql -h{host} -P{port} -u{user.split('#')[0] if '#' in user else user} -p{password} "
+                    f"-e \"CREATE DATABASE {db_name}\""
+                ) from e
+            else:
+                raise ConnectionError(f"Failed to connect to MatrixOne: {e}") from e
 
     def _setup_connection_hook(
         self, on_connect: Union[ConnectionHook, List[Union[ConnectionAction, str]], Callable]

matrixone/async_vector_index_manager.py

@@ -395,7 +395,9 @@ class AsyncVectorManager:
 
         # Convert distance type to enum
         if distance_type == "l2":
-            distance_func = DistanceFunction.L2_SQ
+            distance_func = DistanceFunction.L2  # Use L2 for consistency with ORM l2_distance()
+        elif distance_type == "l2_sq":
+            distance_func = DistanceFunction.L2_SQ  # Explicitly use squared distance if needed
         elif distance_type == "cosine":
             distance_func = DistanceFunction.COSINE
         elif distance_type == "inner_product":

matrixone/client.py

@@ -244,10 +244,11 @@ class Client(BaseMatrixOneClient):
 
     def connect(
         self,
-        host: str,
-        port: int,
-        user: str,
-        password: str,
+        *,
+        host: str = "localhost",
+        port: int = 6001,
+        user: str = "root",
+        password: str = "111",
         database: str,
         ssl_mode: str = "preferred",
         ssl_ca: Optional[str] = None,
@@ -345,8 +346,12 @@ class Client(BaseMatrixOneClient):
             except Exception as e:
                 self.logger.warning(f"Failed to detect backend version: {e}")
 
-            # Setup connection hook if provided
-            if on_connect:
+            # Setup connection hook (default to ENABLE_ALL if not provided)
+            # Allow empty list [] to explicitly disable hooks
+            if on_connect is None:
+                on_connect = [ConnectionAction.ENABLE_ALL]
+
+            if on_connect:  # Only setup if not empty list
                 self._setup_connection_hook(on_connect)
                 # Execute the hook once immediately for the initial connection
                 self._execute_connection_hook_immediately(on_connect)
@@ -354,7 +359,17 @@ class Client(BaseMatrixOneClient):
         except Exception as e:
             self.logger.log_connection(host, port, final_user, database, success=False)
             self.logger.log_error(e, context="Connection")
-            raise ConnectionError(f"Failed to connect to MatrixOne: {e}")
+
+            # Provide user-friendly error messages for common issues
+            error_msg = str(e)
+            if 'Unknown database' in error_msg or '1049' in error_msg:
+                raise ConnectionError(
+                    f"Database '{database}' does not exist. Please create it first:\n"
+                    f"  mysql -h{host} -P{port} -u{user.split('#')[0] if '#' in user else user} -p{password} "
+                    f"-e \"CREATE DATABASE {database}\""
+                ) from e
+            else:
+                raise ConnectionError(f"Failed to connect to MatrixOne: {e}") from e
 
     def _setup_connection_hook(
         self, on_connect: Union[ConnectionHook, List[Union[ConnectionAction, str]], Callable]
@@ -1686,7 +1701,25 @@ class Client(BaseMatrixOneClient):
             from .sqlalchemy_ext import FulltextIndex, VectorIndex
             from sqlalchemy.schema import CreateTable, CreateIndex
 
-            with self.get_sqlalchemy_engine().begin() as conn:
+            try:
+                engine_context = self.get_sqlalchemy_engine().begin()
+            except Exception as e:
+                # Handle database connection errors with user-friendly messages
+                error_msg = str(e)
+                if 'Unknown database' in error_msg or '1049' in error_msg:
+                    db_name = self._connection_params.get('database', 'unknown')
+                    raise ConnectionError(
+                        f"Database '{db_name}' does not exist. Please create it first:\n"
+                        f"  mysql -h{self._connection_params.get('host', 'localhost')} "
+                        f"-P{self._connection_params.get('port', 6001)} "
+                        f"-u{self._connection_params.get('user', 'root')} "
+                        f"-p{self._connection_params.get('password', '***')} "
+                        f"-e \"CREATE DATABASE {db_name}\""
+                    ) from e
+                else:
+                    raise ConnectionError(f"Failed to connect to database: {error_msg}") from e
+
+            with engine_context as conn:
                 # Create table without indexes first
                 # Build CREATE TABLE statement without indexes
                 create_table_sql = str(CreateTable(table).compile(dialect=conn.dialect))
@@ -3599,6 +3632,12 @@ class VectorManager:
         if not success:
             raise Exception(f"Failed to create IVFFLAT vector index {name} on table {table_name}")
 
+        # Add summary log
+        op_type_str = op_type.value if hasattr(op_type, 'value') else op_type
+        self.client.logger.info(
+            f"✓ Created IVF index '{name}' on {table_name}.{column} | " f"lists={lists} | op_type={op_type_str}"
+        )
+
         return self
 
     def create_hnsw(
@@ -3652,6 +3691,7 @@ class VectorManager:
         if op_type is None:
             op_type = VectorOpType.VECTOR_L2_OPS
 
+        # Build index creation SQL
         success = HnswVectorIndex.create_index(
             engine=self.client.get_sqlalchemy_engine(),
             table_name=table_name,
@@ -3666,6 +3706,13 @@ class VectorManager:
         if not success:
             raise Exception(f"Failed to create HNSW vector index {name} on table {table_name}")
 
+        # Add summary log
+        op_type_str = op_type.value if hasattr(op_type, 'value') else op_type
+        self.client.logger.info(
+            f"✓ Created HNSW index '{name}' on {table_name}.{column} | "
+            f"m={m} | ef_construction={ef_construction} | ef_search={ef_search} | op_type={op_type_str}"
+        )
+
         return self
 
     def create_ivf_in_transaction(
@@ -3805,6 +3852,9 @@ class VectorManager:
         if not success:
             raise Exception(f"Failed to drop vector index {name} from table {table_name}")
 
+        # Add summary log
+        self.client.logger.info(f"✓ Dropped vector index '{name}' from {table_name}")
+
         return self
 
     def enable_ivf(self, probe_limit: int = 1) -> "VectorManager":
@@ -4018,7 +4068,9 @@ class VectorManager:
 
         # Convert distance type to enum
         if distance_type == "l2":
-            distance_func = DistanceFunction.L2_SQ
+            distance_func = DistanceFunction.L2  # Use L2 for consistency with ORM l2_distance()
+        elif distance_type == "l2_sq":
+            distance_func = DistanceFunction.L2_SQ  # Explicitly use squared distance if needed
         elif distance_type == "cosine":
             distance_func = DistanceFunction.COSINE
         elif distance_type == "inner_product":

matrixone/restore.py

@@ -204,7 +204,9 @@ class RestoreManager:
             result = self._client.execute(sql)
             return result is not None
         except Exception as e:
-            raise RestoreError(f"Failed to restore database '{database_name}' from snapshot '{snapshot_name}': {e}") from None
+            raise RestoreError(
+                f"Failed to restore database '{database_name}' from snapshot '{snapshot_name}': {e}"
+            ) from None
 
     def restore_table(
         self,
@@ -323,7 +325,9 @@ class RestoreManager:
                     )
             elif restore_type == "table":
                 if not all([account_name, database_name, table_name]):
-                    raise RestoreError("Account name, database name, and table name are required for table restore") from None
+                    raise RestoreError(
+                        "Account name, database name, and table name are required for table restore"
+                    ) from None
                 if to_account:
                     sql = (
                         f"RESTORE ACCOUNT {self._client._escape_identifier(account_name)} "

matrixone/sqlalchemy_ext/vector_index.py

@@ -247,7 +247,21 @@ class IVFVectorIndex(Index):
                 # Enable IVF indexing
                 _exec_sql_safe(conn, "SET experimental_ivf_index = 1")
                 _exec_sql_safe(conn, "SET probe_limit = 1")
+
+                # Execute CREATE INDEX
                 _exec_sql_safe(conn, sql)
+
+                # Try to log using the connection's info attribute if available
+                try:
+                    from ..client import logger
+
+                    logger.info(
+                        f"✓ | CREATE INDEX {name} USING ivfflat ON {table_name}({column}) "
+                        f"LISTS {lists} op_type '{op_type}'"
+                    )
+                except Exception:
+                    pass  # Silently skip logging if not available (for tests)
+
             return True
         except Exception as e:
             print(f"Failed to create IVFFLAT vector index: {e}")
@@ -648,7 +662,21 @@ class HnswVectorIndex(Index):
             with engine.begin() as conn:
                 # Enable HNSW indexing
                 _exec_sql_safe(conn, "SET experimental_hnsw_index = 1")
+
+                # Execute CREATE INDEX
                 _exec_sql_safe(conn, sql)
+
+                # Try to log using the connection's info attribute if available
+                try:
+                    from ..client import logger
+
+                    logger.info(
+                        f"✓ | CREATE INDEX {name} USING hnsw ON {table_name}({column}) "
+                        f"M {m} EF_CONSTRUCTION {ef_construction} EF_SEARCH {ef_search} op_type '{op_type}'"
+                    )
+                except Exception:
+                    pass  # Silently skip logging if not available (for tests)
+
             return True
         except Exception as e:
             print(f"Failed to create HNSW vector index: {e}")

{matrixone_python_sdk-0.1.0.dist-info → matrixone_python_sdk-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrixone-python-sdk
-Version: 0.1.0
+Version: 0.1.2
 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
@@ -185,6 +185,22 @@ print(f"MatrixOne version: {version}")
 client.disconnect()
 ```
 
+> **📝 Connection Parameters**
+> 
+> The `connect()` method requires **keyword arguments** (not positional):
+> - `database` - **Required**, no default value
+> - `host` - Default: `'localhost'`
+> - `port` - Default: `6001`
+> - `user` - Default: `'root'`
+> - `password` - Default: `'111'`
+> 
+> **Minimal connection** (uses all defaults):
+> ```python
+> client.connect(database='test')
+> ```
+> 
+> By default, all features (IVF, HNSW, fulltext) are automatically enabled via `on_connect=[ConnectionAction.ENABLE_ALL]`.
+
 ### Async Usage
 
 ```python

{matrixone_python_sdk-0.1.0.dist-info → matrixone_python_sdk-0.1.2.dist-info}/RECORD

@@ -1,11 +1,11 @@
 matrixone/__init__.py,sha256=7bePzwzerZTiZIlbUT9zI_u4fp49DvQNcIqyiIvEVVs,3966
 matrixone/account.py,sha256=0r9xLNTiUfXa3xWZQUhEJ_uUcNp_gja-YulOR5iYDU4,24712
-matrixone/async_client.py,sha256=HeuUDJbABLi9O-4-2OF8oYW_fSvobcD8z-bKsQl0DO4,150693
+matrixone/async_client.py,sha256=krF9xtn_aGS45VcShV5eYy89FHk0mCI9Ig-8RufBJmY,151518
 matrixone/async_metadata_manager.py,sha256=7k5qtQbKO-52IXzhDN0qhvYdp5wars5M6ZjdSLfD8RM,10230
 matrixone/async_orm.py,sha256=O4Rf85MnZg0_fuLvbHGfOPrHI2OLznYwLk9eam5J5ik,4864
-matrixone/async_vector_index_manager.py,sha256=i881732IV4shT4NqqLBbEW9_z7XYCNlaJgPitCyIkhM,22953
+matrixone/async_vector_index_manager.py,sha256=fVnS00zgbUXw4YmZR-wRCzm541n30TLlmXXmXV34ELU,23134
 matrixone/base_client.py,sha256=PUhq0c7Si9OIk83B4ZZ73ozwaHDRZ1zi0Q94YixseqY,6952
-matrixone/client.py,sha256=XkDphVnmoOyhISmOy0zz6ZhX7QVyW3Yq3Xp6qxTqHWU,180877
+matrixone/client.py,sha256=XWEXEq8FlMfAJ4O4qbnriy2S7qmYGmpLeWe37HHrzg0,183615
 matrixone/config.py,sha256=jrm1rLBhLWrX3dVdPx67sFaZ_9M8Vj4hImVrAmFaYbw,13917
 matrixone/connection_hooks.py,sha256=kjv0HMCUZkma-9nTndNBV66NVcp4cTWs-8u5b6KCj2E,13530
 matrixone/exceptions.py,sha256=oUQUY_TD8UTSzZzbwGdUgQ0kFJIYZq5Fx-vFsMaOfvw,1787
@@ -15,7 +15,7 @@ matrixone/moctl.py,sha256=rLt69ZlhIr_pCbyu8cImKgdR_YOGf85XvqCHcW12sc0,6714
 matrixone/orm.py,sha256=QWNdfJ067KcgWU-eB__vXwJuANBaDUeVVG7thFH5kiU,90473
 matrixone/pitr.py,sha256=NPsexPsJXIAiJTH3oMyFOwiXWjvfohPzWQyFHhQKh5k,21632
 matrixone/pubsub.py,sha256=ckaG6Z3IuEtcdwp11qvB88PrCDHYymocSOJ1v4TftzY,26893
-matrixone/restore.py,sha256=V6NbMcCQb5h2YONVNL_CCXtoqI-Vyl5Hp_lTzYAPVCY,15715
+matrixone/restore.py,sha256=xAbsRUL-2hCk6YpuouVLbiwc3GwUNzCoIVEFtcmghkg,15791
 matrixone/search_vector_index.py,sha256=NjQpluaPcja66herm-OYq3nNeNAqY_-FGGL4SOBOWNA,46075
 matrixone/snapshot.py,sha256=cOatk26Z6Jt6aAdasOMejxxiv73S7Z2o2if_va20Ihg,18103
 matrixone/sql_builder.py,sha256=ZR9TpRkV4cN_RiDJQXMLwSzodpESyxAqV-TTIr--ZbY,29009
@@ -28,14 +28,14 @@ matrixone/sqlalchemy_ext/fulltext_search.py,sha256=1pCMxw9GyTjRhjsJ9KbKq_YsMEPuC
 matrixone/sqlalchemy_ext/hnsw_config.py,sha256=5NYZXK1jsIKlSTrtqpv8n2V_yhbMsmOhItFrbmS5Vs4,5498
 matrixone/sqlalchemy_ext/ivf_config.py,sha256=-esAijWMfyF1GUJivpZh-F4lvyKRmY8LYIzVgmMVtVo,6367
 matrixone/sqlalchemy_ext/table_builder.py,sha256=JVritBPnCXZt0eJUivDrmGdlpaYO-uL7i2n24HiarEE,12870
-matrixone/sqlalchemy_ext/vector_index.py,sha256=UIWuU30HjWypyfq30UiqUJ078DoD7Q5HtD_7iWwjnb0,54832
+matrixone/sqlalchemy_ext/vector_index.py,sha256=dasW3kT4f69io9y6DLQMarvf1FTbzh0UN-HALs0kBVs,55907
 matrixone/sqlalchemy_ext/vector_type.py,sha256=HaOJ9dRdW_yrecD9qGUucW9bMfM3zCxbDC-0Ca32Kmk,30669
-matrixone_python_sdk-0.1.0.dist-info/licenses/LICENSE,sha256=-PpUMwDyMyFlH9H7cnzkTh0Uo42tRvz43k7hnxe7G_I,11252
+matrixone_python_sdk-0.1.2.dist-info/licenses/LICENSE,sha256=-PpUMwDyMyFlH9H7cnzkTh0Uo42tRvz43k7hnxe7G_I,11252
 tests/__init__.py,sha256=odB22tIaJIHSwRhumhDlQYD6Fug_C0opWa07dSKkeQs,694
 tests/offline/__init__.py,sha256=M13mz7gtVDS0_dJUW1EFyyiAGhEj282k3ia7eWA3dPs,703
 tests/offline/conftest.py,sha256=Mz_NT6GBOxqSZsSCR2SXe1pkSpLGKT2-ssWNHhh9xOg,2494
 tests/offline/test_account.py,sha256=uCHKPOBGiJrpEwXdyF2CoNoS5yqxuyKJgLTPMvKZQbk,26723
-tests/offline/test_async_client_query_comprehensive.py,sha256=ZS-HcaQmQFlF9_QjVYn4cgAS6Mg3ZDwguFxqDXpc_Ak,45356
+tests/offline/test_async_client_query_comprehensive.py,sha256=HmLg8iYlD6l5K4lJN5VSXhJDtKGg17vym-9KaX6WDzY,45416
 tests/offline/test_basic.py,sha256=pVyPnidD5jIoC8_40U1iXND29399G-0O5e4itm-2uhc,1710
 tests/offline/test_case_sensitivity.py,sha256=aYlhA5F8RSlSy6wl3HWIrPylE3Z3eAUf4RWQohwD0eA,8715
 tests/offline/test_connection_hooks_offline.py,sha256=pMbkjiEdn5ShljuX8QZRVnsMQLbJxI43tfb0saohBRw,11678
@@ -75,12 +75,12 @@ tests/online/__init__.py,sha256=91rCeFIUZe6Ld-d4PonsoBc2rSHxPGlt6pYGSNsjyzY,692
 tests/online/conftest.py,sha256=1xOqAAFYsx9Io0e3m7DUQNX60B3sqvY98l8JhXY1CAA,7504
 tests/online/test_account_management.py,sha256=mWIF3kPaegcb-JEhHBM9LrSae2xZ3ppjkRefWEzzwH4,6929
 tests/online/test_advanced_features.py,sha256=f0oOe1bB6dT4bXZqx4LzrWd4ljwwcr9iIZLDHnTdPhU,13641
-tests/online/test_async_client_interfaces.py,sha256=jWVIoVGpzhziYzdohThtnio5Zv8DeAwrNppPHn5idBo,12262
-tests/online/test_async_client_online.py,sha256=g_BCdRlLFWF7XNFp0_EE0cTc_LpVAyfiDzA6oi9Z2oI,10873
+tests/online/test_async_client_interfaces.py,sha256=ZT2kWm-ZT_O-_ZkGE7mVLE5c52_eRsbCC_wEksrUis0,12295
+tests/online/test_async_client_online.py,sha256=LZp0QXk4l4FfuQOHNNg__degRTwh9K61StFtEmeUmZ8,10906
 tests/online/test_async_model_insert_online.py,sha256=YPlcNjuaeimvG355JjjgyZUPfa9J0HtC5FVJr9uIp_k,10857
-tests/online/test_async_orm_online.py,sha256=ed2y3PyvnWURC-O-lNnWXPVSrVZQu9VmZtNVSwlU_UY,10918
+tests/online/test_async_orm_online.py,sha256=8wjQb9oXMyNh4HQ1YkwiFYfHo73_YVPicOR_aiXrbEg,11039
 tests/online/test_async_simple_query_online.py,sha256=nBMsCv-GaEiSRxh1juLNCiXNNsfUwWRApQvStzLeGr4,31021
-tests/online/test_async_transaction_simple_query.py,sha256=25z0GN2MiAFGEN3Q4xTEUeOgOkDoGswhgqqkE4FVDmU,11792
+tests/online/test_async_transaction_simple_query.py,sha256=cLOH4C1fWW6vaCmBK33XjswQafxOMitMU0S4r-PAUxU,11825
 tests/online/test_basic_connection.py,sha256=rDukB2MN3Gy_yLCG9_lHtUCcfFoSdkABCTpLsdwJYWo,5010
 tests/online/test_client_online.py,sha256=w9ainOy0VnjSFOkGsSzrdB0jE-0wc87Hp4IqQTJS0OM,8716
 tests/online/test_config.py,sha256=YwghdNuw5VBXP99kxKwxO47-fn5Hv6tXmtRxCCUZQMs,3323
@@ -109,14 +109,14 @@ tests/online/test_simple_fulltext_online.py,sha256=gNHETrSZ9sXZENmI0YkJ1fTJm2cK9
 tests/online/test_snapshot_comprehensive.py,sha256=A8_Qvzxg5f6hkFF0tkESTyDQJSPsMX7nnBEsyY5Vmcc,40603
 tests/online/test_sqlalchemy_engine_integration.py,sha256=XmdgvkJc9TOzkCP4O_GBCTRBdjQ5RncYfuKkcuHqick,12096
 tests/online/test_sqlalchemy_integration.py,sha256=Za2s5Z55SxSAGdwTAfee2I7xoDuH6KLGSBGd-rsF_Wo,15439
-tests/online/test_transaction_contexts.py,sha256=adliTmahWdGuXgbrqZuH7yPGXaHMzh4txoTCpiCiwus,49663
-tests/online/test_transaction_insert_methods.py,sha256=lxBcNKTlpc_m_bJFu12bgiju-8Z_tydAufJp7skXSiA,14009
-tests/online/test_transaction_query_methods.py,sha256=esBk2Ru6CbAbHNcsU3WtpUm41vbsNAvLnaC-MmDt3D8,10909
-tests/online/test_unified_filter_online.py,sha256=A_rxj3lnY8LrQNy-FXSpmxXFLPle2OznAQxAptFl6cM,21181
+tests/online/test_transaction_contexts.py,sha256=ZPsP5vtN1dspCmgZvZ5AUVmX171pymdaIMj_JJ-5_14,49993
+tests/online/test_transaction_insert_methods.py,sha256=TgonknmyidAk_86K8QVowqOG6nLlJ056msGR0tykhxU,14075
+tests/online/test_transaction_query_methods.py,sha256=VaPvHOYskwJVyM7lvVvBJ6HqhdPsXAibFfDyIv1Ox1A,10975
+tests/online/test_unified_filter_online.py,sha256=86JwEKUaBLNuLGJlR4E0zjzdWL5xVFyYLvyxbzkdEVI,21214
 tests/online/test_vector_comprehensive.py,sha256=vmnwwVXLA7lUI_zSK3fJcf1HKx2AvrVDcUT0_d-gQwg,26980
 tests/online/test_version_management.py,sha256=BU8Vc1fDKNCwhRliZi6XmEnd0HYdHuki9Xxi09vnriA,11416
-matrixone_python_sdk-0.1.0.dist-info/METADATA,sha256=hXnXd7Pf1V_VyRVY4VFSSA4hr_0bVJIAJpYm3bPnjvc,20299
-matrixone_python_sdk-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrixone_python_sdk-0.1.0.dist-info/entry_points.txt,sha256=4wUGPC_7_f5ZDR33JRo1LZmmTuyfkYAv41_5H5Qy-Ik,138
-matrixone_python_sdk-0.1.0.dist-info/top_level.txt,sha256=LQZabpBx_dtQk8JbKeH3MbjmC8HYDLE8UQeEf6NfQJA,16
-matrixone_python_sdk-0.1.0.dist-info/RECORD,,
+matrixone_python_sdk-0.1.2.dist-info/METADATA,sha256=eZci5V6XBGdSXpvmVs7SNdkFcLWvRCx2Lh9F-xBiG2c,20815
+matrixone_python_sdk-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrixone_python_sdk-0.1.2.dist-info/entry_points.txt,sha256=4wUGPC_7_f5ZDR33JRo1LZmmTuyfkYAv41_5H5Qy-Ik,138
+matrixone_python_sdk-0.1.2.dist-info/top_level.txt,sha256=LQZabpBx_dtQk8JbKeH3MbjmC8HYDLE8UQeEf6NfQJA,16
+matrixone_python_sdk-0.1.2.dist-info/RECORD,,

tests/offline/test_async_client_query_comprehensive.py

@@ -263,7 +263,7 @@ class TestAsyncClientConnection(unittest.IsolatedAsyncioTestCase):
         # Mock create_async_engine to return our mock engine
         mock_create_async_engine.return_value = mock_engine
 
-        await self.client.connect(host="localhost", port=6001, user="root", password="111", database="test")
+        await self.client.connect(host="localhost", port=6001, user="root", password="111", database="test", on_connect=[])
 
         # Verify create_async_engine was called
         mock_create_async_engine.assert_called_once()
@@ -280,7 +280,9 @@ class TestAsyncClientConnection(unittest.IsolatedAsyncioTestCase):
         mock_create_async_engine.side_effect = Exception("Connection failed")
 
         with self.assertRaises(Exception):
-            await self.client.connect(host="localhost", port=6001, user="root", password="111", database="test")
+            await self.client.connect(
+                host="localhost", port=6001, user="root", password="111", database="test", on_connect=[]
+            )
 
     async def test_disconnect(self):
         """Test async disconnection"""

tests/online/test_async_client_interfaces.py

@@ -38,7 +38,7 @@ class TestAsyncClientMissingInterfaces:
         """Create and connect AsyncClient for testing"""
         host, port, user, password, database = get_connection_params()
         client = AsyncClient(logger=create_default_logger())
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
         try:
             yield client
         finally:

tests/online/test_async_client_online.py

@@ -40,7 +40,7 @@ class TestAsyncClientOnline:
         """Create and connect AsyncClient for testing"""
         host, port, user, password, database = online_config.get_connection_params()
         client = AsyncClient()
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
         try:
             yield client
         finally:

tests/online/test_async_orm_online.py

@@ -100,13 +100,15 @@ class TestAsyncORMOnline:
 
         # First create the database using default connection
         temp_client = AsyncClient()
-        await temp_client.connect(host, port, user, password, "test")  # Connect to default database
+        await temp_client.connect(
+            host=host, port=port, user=user, password=password, database="test"
+        )  # Connect to default database
         await temp_client.execute(f"CREATE DATABASE IF NOT EXISTS {test_database}")
         await temp_client.disconnect()
 
         # Now connect to the test database
         client = AsyncClient()
-        await client.connect(host, port, user, password, test_database)
+        await client.connect(host=host, port=port, user=user, password=password, database=test_database)
 
         try:
             # Create test tables
@@ -158,7 +160,7 @@ class TestAsyncORMOnline:
                 await client.disconnect()
                 # Clean up database
                 cleanup_client = AsyncClient()
-                await cleanup_client.connect(host, port, user, password, "test")
+                await cleanup_client.connect(host=host, port=port, user=user, password=password, database="test")
                 await cleanup_client.execute(f"DROP DATABASE IF EXISTS {test_database}")
                 await cleanup_client.disconnect()
             except Exception as e:

tests/online/test_async_transaction_simple_query.py

@@ -29,7 +29,7 @@ class TestAsyncTransactionSimpleQuery:
     async def async_client_setup(self):
         """Setup async client for testing"""
         client = AsyncClient()
-        await client.connect("127.0.0.1", 6001, "root", "111", "test")
+        await client.connect(host="127.0.0.1", port=6001, user="root", password="111", database="test")
 
         # Enable fulltext indexing
         await client.execute("SET experimental_fulltext_index=1")

tests/online/test_transaction_contexts.py

@@ -34,7 +34,7 @@ class TestSyncTransactionContexts:
         """Setup sync client for testing"""
         client = Client()
         host, port, user, password, database = online_config.get_connection_params()
-        client.connect(host, port, user, password, database)
+        client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Enable fulltext indexing
         client.execute("SET experimental_fulltext_index=1")
@@ -154,7 +154,7 @@ class TestAsyncTransactionContexts:
         """Setup async client for testing"""
         client = AsyncClient()
         host, port, user, password, database = online_config.get_connection_params()
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Enable fulltext indexing
         await client.execute("SET experimental_fulltext_index=1")
@@ -295,7 +295,7 @@ class TestTransactionContextCompatibility:
         """Setup sync client for testing"""
         client = Client()
         host, port, user, password, database = online_config.get_connection_params()
-        client.connect(host, port, user, password, database)
+        client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Enable fulltext indexing
         client.execute("SET experimental_fulltext_index=1")
@@ -364,7 +364,7 @@ class TestTransactionContextCompatibility:
         """Test that async transaction wrapper has all necessary managers"""
         client = AsyncClient()
         host, port, user, password, database = online_config.get_connection_params()
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
 
         try:
             # Enable fulltext indexing
@@ -432,7 +432,7 @@ class TestTransactionContextFeatures:
         """Setup sync client for testing"""
         client = Client()
         host, port, user, password, database = online_config.get_connection_params()
-        client.connect(host, port, user, password, database)
+        client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Enable fulltext indexing
         client.execute("SET experimental_fulltext_index=1")
@@ -537,7 +537,7 @@ class TestTransactionContextFeatures:
         """Test fulltext features in async transaction context"""
         client = AsyncClient()
         host, port, user, password, database = online_config.get_connection_params()
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
 
         try:
             # Enable fulltext indexing
@@ -648,7 +648,7 @@ class TestGetConnectionInterface:
         """Setup sync client for testing"""
         client = Client()
         host, port, user, password, database = online_config.get_connection_params()
-        client.connect(host, port, user, password, database)
+        client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Create test database and table
         test_db = "get_conn_test"
@@ -758,7 +758,7 @@ class TestGetConnectionInterface:
         """Test basic usage of get_connection in async transaction"""
         client = AsyncClient()
         host, port, user, password, database = online_config.get_connection_params()
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
 
         try:
             # Create test database and table
@@ -869,7 +869,7 @@ class TestAsyncTransactionManagerConsistency:
         """Setup sync client for testing"""
         client = Client()
         host, port, user, password, database = online_config.get_connection_params()
-        client.connect(host, port, user, password, database)
+        client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Create test database and table
         test_db = "sync_async_consistency_test"
@@ -920,7 +920,7 @@ class TestAsyncTransactionManagerConsistency:
         """Setup async client for testing"""
         client = AsyncClient()
         host, port, user, password, database = online_config.get_connection_params()
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Create test database and table
         test_db = "async_consistency_test"

tests/online/test_transaction_insert_methods.py

@@ -30,7 +30,7 @@ class TestTransactionInsertMethods:
         """Setup sync client for testing"""
         client = Client()
         host, port, user, password, database = online_config.get_connection_params()
-        client.connect(host, port, user, password, database)
+        client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Create test database
         test_db = "sync_insert_test"
@@ -64,7 +64,7 @@ class TestTransactionInsertMethods:
         """Setup async client for testing"""
         client = AsyncClient()
         host, port, user, password, database = online_config.get_connection_params()
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Create test database
         test_db = "async_insert_test"

tests/online/test_transaction_query_methods.py

@@ -30,7 +30,7 @@ class TestTransactionQueryMethods:
         """Setup sync client for testing"""
         client = Client()
         host, port, user, password, database = online_config.get_connection_params()
-        client.connect(host, port, user, password, database)
+        client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Create test database
         test_db = "sync_query_test"
@@ -74,7 +74,7 @@ class TestTransactionQueryMethods:
         """Setup async client for testing"""
         client = AsyncClient()
         host, port, user, password, database = online_config.get_connection_params()
-        await client.connect(host, port, user, password, database)
+        await client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Create test database
         test_db = "async_query_test"

tests/online/test_unified_filter_online.py

@@ -71,7 +71,7 @@ class TestUnifiedFilterOnline(unittest.TestCase):
 
         # Create client
         cls.client = Client()
-        cls.client.connect(host, port, user, password, database)
+        cls.client.connect(host=host, port=port, user=user, password=password, database=database)
 
         # Create test database
         cls.test_db = "unified_filter_test"