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

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

@@ -3632,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(
@@ -3685,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,
@@ -3699,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(
@@ -3838,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":
@@ -4051,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/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.1.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.1
+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

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

@@ -3,9 +3,9 @@ matrixone/account.py,sha256=0r9xLNTiUfXa3xWZQUhEJ_uUcNp_gja-YulOR5iYDU4,24712
 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=uEKeA2BMjEd_n38lxBSJEv3NcJBr4S3lYV7a7fw_V-A,182693
+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
@@ -28,9 +28,9 @@ 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.1.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
@@ -115,8 +115,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.1.dist-info/METADATA,sha256=XudxtXRH_tppDic5p80YsrvQYNLJpUXmCnr8o0kGNJQ,20815
-matrixone_python_sdk-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrixone_python_sdk-0.1.1.dist-info/entry_points.txt,sha256=4wUGPC_7_f5ZDR33JRo1LZmmTuyfkYAv41_5H5Qy-Ik,138
-matrixone_python_sdk-0.1.1.dist-info/top_level.txt,sha256=LQZabpBx_dtQk8JbKeH3MbjmC8HYDLE8UQeEf6NfQJA,16
-matrixone_python_sdk-0.1.1.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,,