matrixone-python-sdk 0.1.1__tar.gz → 0.1.2__tar.gz

{matrixone_python_sdk-0.1.1/matrixone_python_sdk.egg-info → matrixone_python_sdk-0.1.2}/PKG-INFO

@@ -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 → matrixone_python_sdk-0.1.2}/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_python_sdk-0.1.1 → matrixone_python_sdk-0.1.2}/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_python_sdk-0.1.1 → matrixone_python_sdk-0.1.2}/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 → matrixone_python_sdk-0.1.2/matrixone_python_sdk.egg-info}/PKG-INFO

@@ -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 → matrixone_python_sdk-0.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "matrixone-python-sdk"
-version = "0.1.1"
+version = "0.1.2"
 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"}