[python] vector lumina: python fix code (#7609)

Author: jerry-024

paimon-python/pypaimon/globalindex/lumina/lumina_vector_global_index_reader.py

@@ -23,6 +23,7 @@
 """
 
 import os
+import numpy as np
 
 from pypaimon.globalindex.global_index_reader import GlobalIndexReader
 from pypaimon.globalindex.vector_search_result import DictBasedScoredIndexResult
@@ -57,15 +58,7 @@ def visit_vector_search(self, vector_search):
         self._ensure_loaded()
 
         from lumina_data import MetricType
-
-        query = vector_search.vector
-        # Flatten to a plain list of floats for search_list API
-        if hasattr(query, 'tolist'):
-            query_flat = list(query.flatten()) if hasattr(query, 'flatten') else list(query)
-        else:
-            query_flat = list(query)
-        query_flat = [float(v) for v in query_flat]
-
+        query_flat = [float(v) for v in np.asarray(vector_search.vector).tolist()]
         expected_dim = self._index_meta.dim
         if len(query_flat) != expected_dim:
             raise ValueError(
@@ -152,50 +145,3 @@ def close(self):
         if self._stream is not None:
             self._stream.close()
             self._stream = None
-
-    # =================== unsupported =====================
-
-    def visit_equal(self, field_ref, literal):
-        return None
-
-    def visit_not_equal(self, field_ref, literal):
-        return None
-
-    def visit_less_than(self, field_ref, literal):
-        return None
-
-    def visit_less_or_equal(self, field_ref, literal):
-        return None
-
-    def visit_greater_than(self, field_ref, literal):
-        return None
-
-    def visit_greater_or_equal(self, field_ref, literal):
-        return None
-
-    def visit_is_null(self, field_ref):
-        return None
-
-    def visit_is_not_null(self, field_ref):
-        return None
-
-    def visit_in(self, field_ref, literals):
-        return None
-
-    def visit_not_in(self, field_ref, literals):
-        return None
-
-    def visit_starts_with(self, field_ref, literal):
-        return None
-
-    def visit_ends_with(self, field_ref, literal):
-        return None
-
-    def visit_contains(self, field_ref, literal):
-        return None
-
-    def visit_like(self, field_ref, literal):
-        return None
-
-    def visit_between(self, field_ref, min_v, max_v):
-        return None

paimon-python/pypaimon/table/source/vector_search_read.py

@@ -86,22 +86,17 @@ def _eval(self, row_range_start, row_range_end, vector_index_files):
         file_io = self._table.file_io
         options = self._table.table_schema.options
 
-        reader = _create_vector_reader(
-            index_type, file_io, index_path,
-            index_io_meta_list, options
-        )
-
         vector_search = VectorSearch(
             vector=self._query_vector,
             limit=self._limit,
             field_name=self._vector_column.name
         )
-
-        try:
+        with _create_vector_reader(
+            index_type, file_io, index_path,
+            index_io_meta_list, options
+        ) as reader:
             offset_reader = OffsetGlobalIndexReader(reader, row_range_start, row_range_end)
             return offset_reader.visit_vector_search(vector_search)
-        finally:
-            reader.close()
 
 
 def _create_vector_reader(index_type, file_io, index_path, index_io_meta_list, options=None):