matrixone-python-sdk 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

matrixone/async_client.py

@@ -2556,7 +2556,7 @@ class AsyncClient(BaseMatrixOneClient):
 
     async def get_secondary_index_tables(self, table_name: str) -> List[str]:
         """
-        Get all secondary index table names for a given table (async version).
+        Get all secondary index table names for a given table in the current database (async version).
 
         Args:
             table_name: Name of the table to get secondary indexes for
@@ -2573,13 +2573,16 @@ class AsyncClient(BaseMatrixOneClient):
         """
         from .index_utils import build_get_index_tables_sql
 
-        sql, params = build_get_index_tables_sql(table_name)
+        # Get current database from connection params
+        database = self._connection_params.get('database') if hasattr(self, '_connection_params') else None
+
+        sql, params = build_get_index_tables_sql(table_name, database)
         result = await self.execute(sql, params)
         return [row[0] for row in result.fetchall()]
 
     async def get_secondary_index_table_by_name(self, table_name: str, index_name: str) -> Optional[str]:
         """
-        Get the physical table name of a secondary index by its index name (async version).
+        Get the physical table name of a secondary index by its index name in the current database (async version).
 
         Args:
             table_name: Name of the table
@@ -2597,7 +2600,10 @@ class AsyncClient(BaseMatrixOneClient):
         """
         from .index_utils import build_get_index_table_by_name_sql
 
-        sql, params = build_get_index_table_by_name_sql(table_name, index_name)
+        # Get current database from connection params
+        database = self._connection_params.get('database') if hasattr(self, '_connection_params') else None
+
+        sql, params = build_get_index_table_by_name_sql(table_name, index_name, database)
         result = await self.execute(sql, params)
         row = result.fetchone()
         return row[0] if row else None

matrixone/client.py

@@ -2615,7 +2615,7 @@ class Client(BaseMatrixOneClient):
 
     def get_secondary_index_tables(self, table_name: str) -> List[str]:
         """
-        Get all secondary index table names for a given table.
+        Get all secondary index table names for a given table in the current database.
 
         Args:
             table_name: Name of the table to get secondary indexes for
@@ -2633,13 +2633,16 @@ class Client(BaseMatrixOneClient):
         """
         from .index_utils import build_get_index_tables_sql
 
-        sql, params = build_get_index_tables_sql(table_name)
+        # Get current database from connection params
+        database = self._connection_params.get('database') if hasattr(self, '_connection_params') else None
+
+        sql, params = build_get_index_tables_sql(table_name, database)
         result = self.execute(sql, params)
         return [row[0] for row in result.fetchall()]
 
     def get_secondary_index_table_by_name(self, table_name: str, index_name: str) -> Optional[str]:
         """
-        Get the physical table name of a secondary index by its index name.
+        Get the physical table name of a secondary index by its index name in the current database.
 
         Args:
             table_name: Name of the table
@@ -2658,7 +2661,10 @@ class Client(BaseMatrixOneClient):
         """
         from .index_utils import build_get_index_table_by_name_sql
 
-        sql, params = build_get_index_table_by_name_sql(table_name, index_name)
+        # Get current database from connection params
+        database = self._connection_params.get('database') if hasattr(self, '_connection_params') else None
+
+        sql, params = build_get_index_table_by_name_sql(table_name, index_name, database)
         result = self.execute(sql, params)
         row = result.fetchone()
         return row[0] if row else None

matrixone/connection_hooks.py

@@ -69,13 +69,13 @@ class ConnectionHook:
             event.listen(engine.sync_engine, "connect", self._on_connect_sync)
             event.listen(engine.sync_engine, "before_cursor_execute", self._on_before_cursor_execute)
             if hasattr(self._client_ref, 'logger'):
-                self._client_ref.logger.info("Attached connection hook to async engine")
+                self._client_ref.logger.debug("Attached connection hook to async engine")
         else:
             # For sync engines, listen to both connect and before_cursor_execute events
             event.listen(engine, "connect", self._on_connect_sync)
             event.listen(engine, "before_cursor_execute", self._on_before_cursor_execute)
             if hasattr(self._client_ref, 'logger'):
-                self._client_ref.logger.info("Attached connection hook to sync engine")
+                self._client_ref.logger.debug("Attached connection hook to sync engine")
 
     def _on_connect_sync(self, dbapi_connection, connection_record):
         """SQLAlchemy event handler for new connections (sync)"""
@@ -86,7 +86,7 @@ class ConnectionHook:
                 try:
                     # Log that the hook is being executed
                     if hasattr(self._client_ref, 'logger'):
-                        self._client_ref.logger.info(f"Executing connection hook on new connection {conn_id}")
+                        self._client_ref.logger.debug(f"Executing connection hook on new connection {conn_id}")
                     # Pass the connection to avoid creating new connections
                     self.execute_sync_with_connection(self._client_ref, dbapi_connection)
                     self._executed_connections.add(conn_id)
@@ -104,7 +104,7 @@ class ConnectionHook:
                 try:
                     # Log that the hook is being executed
                     if hasattr(self._client_ref, 'logger'):
-                        self._client_ref.logger.info(f"Executing connection hook on connection {conn_id}")
+                        self._client_ref.logger.debug(f"Executing connection hook on connection {conn_id}")
                     # Use the connection to avoid creating new connections
                     self.execute_sync_with_connection(self._client_ref, conn.connection)
                     self._executed_connections.add(conn_id)
@@ -212,7 +212,7 @@ class ConnectionHook:
             cursor = dbapi_connection.cursor()
             cursor.execute("SET experimental_ivf_index = 1")
             cursor.close()
-            client.logger.info("✓ Enabled IVF vector operations")
+            client.logger.debug("✓ Enabled IVF vector operations")
         except Exception as e:
             client.logger.warning(f"Failed to enable IVF: {e}")
 
@@ -223,7 +223,7 @@ class ConnectionHook:
             cursor = dbapi_connection.cursor()
             cursor.execute("SET experimental_hnsw_index = 1")
             cursor.close()
-            client.logger.info("✓ Enabled HNSW vector operations")
+            client.logger.debug("✓ Enabled HNSW vector operations")
         except Exception as e:
             client.logger.warning(f"Failed to enable HNSW: {e}")
 
@@ -234,7 +234,7 @@ class ConnectionHook:
             cursor = dbapi_connection.cursor()
             cursor.execute("SET experimental_fulltext_index = 1")
             cursor.close()
-            client.logger.info("✓ Enabled fulltext search operations")
+            client.logger.debug("✓ Enabled fulltext search operations")
         except Exception as e:
             client.logger.warning(f"Failed to enable fulltext: {e}")
 

matrixone/index_utils.py

@@ -19,43 +19,65 @@ Index utilities - Shared logic for secondary index operations
 from typing import List, Tuple
 
 
-def build_get_index_tables_sql(table_name: str) -> Tuple[str, Tuple]:
+def build_get_index_tables_sql(table_name: str, database: str = None) -> Tuple[str, Tuple]:
     """
     Build SQL to get all secondary index table names for a given table.
 
     Args:
         table_name: Name of the table
+        database: Name of the database (optional, but recommended to avoid cross-database conflicts)
 
     Returns:
         Tuple of (sql, params)
     """
-    sql = """
-        SELECT DISTINCT index_table_name
-        FROM mo_catalog.mo_indexes
-        JOIN mo_catalog.mo_tables ON mo_indexes.table_id = mo_tables.rel_id
-        WHERE relname = ? AND type = 'MULTIPLE'
-    """
-    return sql, (table_name,)
-
-
-def build_get_index_table_by_name_sql(table_name: str, index_name: str) -> Tuple[str, Tuple]:
+    if database:
+        sql = """
+            SELECT DISTINCT index_table_name
+            FROM mo_catalog.mo_indexes
+            JOIN mo_catalog.mo_tables ON mo_indexes.table_id = mo_tables.rel_id
+            WHERE relname = ? AND reldatabase = ? AND type = 'MULTIPLE'
+        """
+        return sql, (table_name, database)
+    else:
+        # Fallback to old behavior if database is not provided
+        sql = """
+            SELECT DISTINCT index_table_name
+            FROM mo_catalog.mo_indexes
+            JOIN mo_catalog.mo_tables ON mo_indexes.table_id = mo_tables.rel_id
+            WHERE relname = ? AND type = 'MULTIPLE'
+        """
+        return sql, (table_name,)
+
+
+def build_get_index_table_by_name_sql(table_name: str, index_name: str, database: str = None) -> Tuple[str, Tuple]:
     """
     Build SQL to get the physical table name of a secondary index by its index name.
 
     Args:
         table_name: Name of the table
         index_name: Name of the secondary index
+        database: Name of the database (optional, but recommended to avoid cross-database conflicts)
 
     Returns:
         Tuple of (sql, params)
     """
-    sql = """
-        SELECT DISTINCT index_table_name
-        FROM mo_catalog.mo_indexes
-        JOIN mo_catalog.mo_tables ON mo_indexes.table_id = mo_tables.rel_id
-        WHERE relname = ? AND name = ?
-    """
-    return sql, (table_name, index_name)
+    if database:
+        sql = """
+            SELECT DISTINCT index_table_name
+            FROM mo_catalog.mo_indexes
+            JOIN mo_catalog.mo_tables ON mo_indexes.table_id = mo_tables.rel_id
+            WHERE relname = ? AND name = ? AND reldatabase = ?
+        """
+        return sql, (table_name, index_name, database)
+    else:
+        # Fallback to old behavior if database is not provided
+        sql = """
+            SELECT DISTINCT index_table_name
+            FROM mo_catalog.mo_indexes
+            JOIN mo_catalog.mo_tables ON mo_indexes.table_id = mo_tables.rel_id
+            WHERE relname = ? AND name = ?
+        """
+        return sql, (table_name, index_name)
 
 
 def build_verify_counts_sql(table_name: str, index_tables: List[str]) -> str:

{matrixone_python_sdk-0.1.3.dist-info → matrixone_python_sdk-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrixone-python-sdk
-Version: 0.1.3
+Version: 0.1.4
 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.3.dist-info → matrixone_python_sdk-0.1.4.dist-info}/RECORD

@@ -1,15 +1,15 @@
 matrixone/__init__.py,sha256=7bePzwzerZTiZIlbUT9zI_u4fp49DvQNcIqyiIvEVVs,3966
 matrixone/account.py,sha256=0r9xLNTiUfXa3xWZQUhEJ_uUcNp_gja-YulOR5iYDU4,24712
-matrixone/async_client.py,sha256=KKvWIzcV2PtwaFT31jgawRLt1CJhI_rePQdAs8xLzEc,155275
+matrixone/async_client.py,sha256=yluma410CLUy3FhxMxeOm-5uhRwhD-ONzMIJd-dEuvg,155669
 matrixone/async_metadata_manager.py,sha256=W3TJBCy56diDPBzimpBImFrmf9DBc7iiS5CdBHLAh_Q,10232
 matrixone/async_orm.py,sha256=O4Rf85MnZg0_fuLvbHGfOPrHI2OLznYwLk9eam5J5ik,4864
 matrixone/async_vector_index_manager.py,sha256=fVnS00zgbUXw4YmZR-wRCzm541n30TLlmXXmXV34ELU,23134
 matrixone/base_client.py,sha256=PUhq0c7Si9OIk83B4ZZ73ozwaHDRZ1zi0Q94YixseqY,6952
-matrixone/client.py,sha256=bxdKUVTzo18yMDFmgcUdSIfoKozC-pfUOlKuWCwwGwI,187341
+matrixone/client.py,sha256=NONid82vkq2GYWVfJqjHgFjeTBkXauEEvcoYtSoeuYs,187735
 matrixone/config.py,sha256=jrm1rLBhLWrX3dVdPx67sFaZ_9M8Vj4hImVrAmFaYbw,13917
-matrixone/connection_hooks.py,sha256=FrCGWipg5KUrdBt-cJMORaOUfoIqgSZNZCtDnn9la50,13531
+matrixone/connection_hooks.py,sha256=5o7heheRQ8pLguoUQxEoo023519rT2-OhloPW-vORpE,13538
 matrixone/exceptions.py,sha256=VsGgESLl4vm5-9QbigyE5RTwFujyoYXhQuGhGKoujFw,1667
-matrixone/index_utils.py,sha256=ND_Md-mj1Yh8Qq4RGW5tYxwqkgwItkWzWDLL_SQ9AMI,3759
+matrixone/index_utils.py,sha256=ofEm0PHKE4VIGuq4ymeJPtA5NvV1afwZfHPLN2bd6Dw,4862
 matrixone/logger.py,sha256=aZmetje_AqtyxFxsqZvp-3R68n8I_ldplRGfV2MHg48,26105
 matrixone/metadata.py,sha256=Yf0eJywe3BUqg6aEywSWh9neRcVsrgf6g-uaZFvqLOA,31707
 matrixone/moctl.py,sha256=b_qQPSK52JiyyUH4PYzAuhJPk8dAD6LRfT_sFGsf23o,6704
@@ -31,7 +31,7 @@ matrixone/sqlalchemy_ext/ivf_config.py,sha256=-esAijWMfyF1GUJivpZh-F4lvyKRmY8LYI
 matrixone/sqlalchemy_ext/table_builder.py,sha256=JVritBPnCXZt0eJUivDrmGdlpaYO-uL7i2n24HiarEE,12870
 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.3.dist-info/licenses/LICENSE,sha256=-PpUMwDyMyFlH9H7cnzkTh0Uo42tRvz43k7hnxe7G_I,11252
+matrixone_python_sdk-0.1.4.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
@@ -39,7 +39,7 @@ tests/offline/test_account.py,sha256=uCHKPOBGiJrpEwXdyF2CoNoS5yqxuyKJgLTPMvKZQbk
 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
+tests/offline/test_connection_hooks_offline.py,sha256=mqnummeGGEvqeevG2R7-0VexnBquo4CJyTtmLhMa7Ps,11686
 tests/offline/test_dialect_schema_handling.py,sha256=WK22Cu3cI1uZ2RtmXmkwnWrYXrwMdcXkhqHTwp0Bqqo,24380
 tests/offline/test_explain_methods.py,sha256=QGSwqGGulZ-8e0pnshQ9A0JwC-CQZ6GcE7mkp_LIYIw,12683
 tests/offline/test_filter_logical_in.py,sha256=Qb-bJiy5lhQaVlTeVNVZ75v6Xyx4Qd2IS_faaxC4VUg,8967
@@ -118,8 +118,8 @@ tests/online/test_transaction_query_methods.py,sha256=VaPvHOYskwJVyM7lvVvBJ6Hqhd
 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.3.dist-info/METADATA,sha256=lNJxaTy1NjfSwoOdr8YOVZ-E9ISSjL2U-t9ZDtZsMnI,20815
-matrixone_python_sdk-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrixone_python_sdk-0.1.3.dist-info/entry_points.txt,sha256=4wUGPC_7_f5ZDR33JRo1LZmmTuyfkYAv41_5H5Qy-Ik,138
-matrixone_python_sdk-0.1.3.dist-info/top_level.txt,sha256=LQZabpBx_dtQk8JbKeH3MbjmC8HYDLE8UQeEf6NfQJA,16
-matrixone_python_sdk-0.1.3.dist-info/RECORD,,
+matrixone_python_sdk-0.1.4.dist-info/METADATA,sha256=EOKIuyVQVsFc316C84HjVLJYxD-7fMtq7a307ypoJ54,20815
+matrixone_python_sdk-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrixone_python_sdk-0.1.4.dist-info/entry_points.txt,sha256=4wUGPC_7_f5ZDR33JRo1LZmmTuyfkYAv41_5H5Qy-Ik,138
+matrixone_python_sdk-0.1.4.dist-info/top_level.txt,sha256=LQZabpBx_dtQk8JbKeH3MbjmC8HYDLE8UQeEf6NfQJA,16
+matrixone_python_sdk-0.1.4.dist-info/RECORD,,

tests/offline/test_connection_hooks_offline.py

@@ -106,7 +106,7 @@ class TestConnectionHook:
         # Verify that cursor.execute was called with the correct SQL
         self.mock_cursor.execute.assert_called_with("SET experimental_ivf_index = 1")
         self.mock_cursor.close.assert_called_once()
-        self.mock_client.logger.info.assert_called_with("✓ Enabled IVF vector operations")
+        self.mock_client.logger.debug.assert_called_with("✓ Enabled IVF vector operations")
 
     def test_execute_sync_with_hnsw_action(self):
         """Test synchronous execution with HNSW action"""
@@ -117,7 +117,7 @@ class TestConnectionHook:
         # Verify that cursor.execute was called with the correct SQL
         self.mock_cursor.execute.assert_called_with("SET experimental_hnsw_index = 1")
         self.mock_cursor.close.assert_called_once()
-        self.mock_client.logger.info.assert_called_with("✓ Enabled HNSW vector operations")
+        self.mock_client.logger.debug.assert_called_with("✓ Enabled HNSW vector operations")
 
     def test_execute_sync_with_fulltext_action(self):
         """Test synchronous execution with fulltext action"""
@@ -128,7 +128,7 @@ class TestConnectionHook:
         # Verify that cursor.execute was called with the correct SQL
         self.mock_cursor.execute.assert_called_with("SET experimental_fulltext_index = 1")
         self.mock_cursor.close.assert_called_once()
-        self.mock_client.logger.info.assert_called_with("✓ Enabled fulltext search operations")
+        self.mock_client.logger.debug.assert_called_with("✓ Enabled fulltext search operations")
 
     def test_execute_sync_with_vector_action(self):
         """Test synchronous execution with vector action (enables both IVF and HNSW)"""
@@ -140,8 +140,8 @@ class TestConnectionHook:
         expected_calls = [call("SET experimental_ivf_index = 1"), call("SET experimental_hnsw_index = 1")]
         self.mock_cursor.execute.assert_has_calls(expected_calls, any_order=True)
         self.mock_cursor.close.assert_called()
-        self.mock_client.logger.info.assert_any_call("✓ Enabled IVF vector operations")
-        self.mock_client.logger.info.assert_any_call("✓ Enabled HNSW vector operations")
+        self.mock_client.logger.debug.assert_any_call("✓ Enabled IVF vector operations")
+        self.mock_client.logger.debug.assert_any_call("✓ Enabled HNSW vector operations")
 
     def test_execute_sync_with_all_action(self):
         """Test synchronous execution with all action"""
@@ -157,9 +157,9 @@ class TestConnectionHook:
         ]
         self.mock_cursor.execute.assert_has_calls(expected_calls, any_order=True)
         self.mock_cursor.close.assert_called()
-        self.mock_client.logger.info.assert_any_call("✓ Enabled IVF vector operations")
-        self.mock_client.logger.info.assert_any_call("✓ Enabled HNSW vector operations")
-        self.mock_client.logger.info.assert_any_call("✓ Enabled fulltext search operations")
+        self.mock_client.logger.debug.assert_any_call("✓ Enabled IVF vector operations")
+        self.mock_client.logger.debug.assert_any_call("✓ Enabled HNSW vector operations")
+        self.mock_client.logger.debug.assert_any_call("✓ Enabled fulltext search operations")
 
     def test_execute_sync_with_custom_hook(self):
         """Test synchronous execution with custom hook"""