vastdb 2.0.2__py3-none-any.whl → 2.0.5__py3-none-any.whl

vastdb/tests/test_vector_search.py

@@ -0,0 +1,211 @@
+from collections import defaultdict
+
+import numpy as np
+import pyarrow as pa
+import pytest
+
+from vastdb._adbc import _ibis_to_qe_predicates
+from vastdb._internal import VectorIndex, VectorIndexSpec
+from vastdb._table_interface import IbisPredicate
+from vastdb.conftest import SessionFactory
+from vastdb.table_metadata import TableMetadata, TableRef, TableType
+
+DIM = 8
+TEST_DISTANCE_FUNC = 'array_distance'
+TEST_DISTANCE_METRIC = 'l2sq'
+
+VectorColumnArrowType = pa.list_(
+    pa.field(name='item', type=pa.float32(), nullable=False), DIM)
+
+query_vector: np.ndarray = np.ones(DIM)
+first_closest_vector = (query_vector * 1.1).tolist()
+second_closest_vector = (query_vector * 1.2).tolist()
+third_closest_vector = (query_vector * 1.3).tolist()
+fourth_closest_vector = (query_vector * 1.4).tolist()
+fifth_closest_vector = (query_vector * 1.5).tolist()
+furthest_vector = (np.ones(DIM) * 5).tolist()
+
+vector_column_name = 'vector_col'
+
+data = [
+    {
+        "id": 0,
+        "n1": 1,
+        "n2": 100,
+        vector_column_name: first_closest_vector,
+    },
+    {
+        "id": 1,
+        "n1": 2,
+        "n2": 100,
+        vector_column_name: second_closest_vector,
+    },
+    {
+        "id": 2,
+        "n1": 3,
+        "n2": 200,
+        vector_column_name: third_closest_vector,
+    },
+    {
+        "id": 3,
+        "n1": 4,
+        "n2": 200,
+        vector_column_name: fourth_closest_vector,
+    },
+    {
+        "id": 4,
+        "n1": 5,
+        "n2": 200,
+        vector_column_name: fifth_closest_vector,
+    },
+    {
+        "id": 5,
+        "n1": 6,
+        "n2": 200,
+        vector_column_name: furthest_vector,
+    },
+
+]
+
+
+def into_arrow_arrays(data: list[dict]) -> list[list]:
+    agg = defaultdict(list)
+    for d in data:
+        for k, v in d.items():
+            agg[k].append(v)
+
+    return [v for v in agg.values()]
+
+
+def test_sanity(session_factory: SessionFactory, clean_bucket_name: str):
+    session = session_factory(with_adbc=True)
+
+    arrow_schema = pa.schema([('id', pa.int32()), ('n1', pa.int32(
+    )), ('n2', pa.int32()), (vector_column_name, VectorColumnArrowType),])
+
+    limit = 3
+
+    ref = TableRef(clean_bucket_name, 's', 't')
+    table_md = TableMetadata(ref,
+                             arrow_schema,
+                             TableType.Regular,
+                             vector_index=VectorIndex(vector_column_name,
+                                                      TEST_DISTANCE_METRIC,
+                                                      TEST_DISTANCE_FUNC))
+    data_table = pa.table(schema=arrow_schema, data=into_arrow_arrays(data))
+
+    with session.transaction() as tx:
+        table = tx.bucket(clean_bucket_name).create_schema(
+            's').create_table('t', arrow_schema)
+        table.insert(data_table)
+
+    # TODO merge in same tx
+    with session.transaction() as tx:
+        table = tx.table_from_metadata(table_md)
+
+        reader = table.vector_search(vec=query_vector.tolist(),
+                                     columns=['id', 'n1', 'n2'],
+                                     limit=limit)
+
+        result_table = reader.read_all()
+
+        assert set([v.as_py() for v in result_table['n1']]) == {1, 2, 3}
+
+
+def test_with_predicates(session_factory: SessionFactory, clean_bucket_name: str):
+    session = session_factory(with_adbc=True)
+
+    vector_column_name = 'vector_column'
+    arrow_schema = pa.schema([('id', pa.int32()), ('n1', pa.int32(
+    )), ('n2', pa.int32()), (vector_column_name, VectorColumnArrowType),])
+    limit = 3
+
+    ref = TableRef(clean_bucket_name, 's', 't')
+    table_md = TableMetadata(ref, arrow_schema, TableType.Regular,
+                             vector_index=VectorIndex(vector_column_name,
+                                                      TEST_DISTANCE_METRIC,
+                                                      TEST_DISTANCE_FUNC))
+    data_table = pa.table(schema=arrow_schema, data=into_arrow_arrays(data))
+
+    with session.transaction() as tx:
+        table = tx.bucket(clean_bucket_name).create_schema(
+            's').create_table('t', arrow_schema)
+        table.insert(data_table)
+
+    with session.transaction() as tx:
+        table = tx.table_from_metadata(table_md)
+
+        pred = table_md.ibis_table['n2'] == 200
+        reader = table.vector_search(vec=query_vector.tolist(),
+                                     columns=['id', 'n1', 'n2'],
+                                     limit=limit,
+                                     predicate=pred)
+
+        result_table = reader.read_all()
+
+        assert set([v.as_py() for v in result_table['n1']]) == {3, 4, 5}
+
+
+arrow_schema = pa.schema([('id', pa.int32()), ('n1', pa.int32(
+)), ('n2', pa.int32()), (vector_column_name, VectorColumnArrowType),])
+
+ref = TableRef('b', 's', 't')
+table_md = TableMetadata(ref, arrow_schema, TableType.Regular)
+
+
+@pytest.mark.parametrize('ibis_predicate, expected', [
+    ((table_md.ibis_table['n1'] == 1) & (table_md.ibis_table['n2'] == 2),
+     '("n1" = 1) AND ("n2" = 2)'),
+    ((table_md.ibis_table['n1'] == 1),
+     '"n1" = 1')
+])
+def test_ibis_to_query_engine_predicates(ibis_predicate: IbisPredicate, expected: str):
+    assert _ibis_to_qe_predicates(ibis_predicate) == expected
+
+
+@pytest.mark.skip(reason="see https://vastdata.atlassian.net/browse/ORION-307908")
+def test_with_predicates_get_vector_index_properties_from_server(
+    session_factory: SessionFactory,
+    clean_bucket_name: str
+    ):
+    session = session_factory(with_adbc=True)
+
+    vector_column_name = 'vector_column'
+    vector_index_distance_metric = 'l2sq'
+    vector_index_sql_distance_function = "array_distance"
+
+    arrow_schema = pa.schema([('id', pa.int32()), ('n1', pa.int32(
+    )), ('n2', pa.int32()), (vector_column_name, VectorColumnArrowType),])
+    limit = 3
+
+    ref = TableRef(clean_bucket_name, 's', 't')
+    table_md = TableMetadata(ref, arrow_schema, TableType.Regular)
+    data_table = pa.table(schema=arrow_schema, data=into_arrow_arrays(data))
+
+    with session.transaction() as tx:
+        table = (tx.bucket(clean_bucket_name)
+                 .create_schema('s')
+                 .create_table('t',
+                               arrow_schema,
+                               vector_index=VectorIndexSpec(vector_column_name,
+                                                            vector_index_distance_metric)))
+        table.insert(data_table)
+
+    with session.transaction() as tx:
+        table = tx.table_from_metadata(table_md)
+
+        table.reload_stats()
+        assert table.vector_index == VectorIndex(
+            column=vector_column_name,
+            distance_metric=vector_index_distance_metric,
+            sql_distance_function=vector_index_sql_distance_function)
+
+        pred = table_md.ibis_table['n2'] == 200
+        reader = table.vector_search(vec=query_vector.tolist(),
+                                     columns=['id', 'n1', 'n2'],
+                                     limit=limit,
+                                     predicate=pred)
+
+        result_table = reader.read_all()
+
+        assert set([v.as_py() for v in result_table['n1']]) == {3, 4, 5}

vastdb/tests/util.py

@@ -25,11 +25,12 @@ def assert_row_ids_ascending_on_first_insertion_to_table(row_ids, expected_num_r
 @contextmanager
 def prepare_data(session: Session,
                  clean_bucket_name: str, schema_name: str, table_name: str,
-                 arrow_table: pa.Table, sorting_key: List[str] = []) -> Iterator[Table]:
+                 arrow_table: pa.Table, sorting_key: List[str] = [],
+                 insert_by_columns: bool = True) -> Iterator[Table]:
     with session.transaction() as tx:
         s = tx.bucket(clean_bucket_name).create_schema(schema_name)
         t = s.create_table(table_name, arrow_table.schema, sorting_key=sorting_key)
-        row_ids_array = t.insert(arrow_table)
+        row_ids_array = t.insert(arrow_table, by_columns=insert_by_columns)
         row_ids = row_ids_array.to_pylist()
         assert_row_ids_ascending_on_first_insertion_to_table(row_ids, arrow_table.num_rows, t.sorted_table)
         yield t

vastdb/transaction.py

@@ -11,6 +11,7 @@ from copy import deepcopy
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Iterable, Optional
 
+from vastdb._adbc import AdbcConnection, AdbcDriver
 from vastdb._table_interface import ITable
 from vastdb.table import TableInTransaction
 from vastdb.table_metadata import TableMetadata
@@ -39,17 +40,37 @@ class TransactionNotActiveError(Exception):
     pass
 
 
+class NoAdbcConnectionError(Exception):
+    """No Adbc Connection Error."""
+
+    pass
+
+
 @dataclass
 class Transaction:
     """A holder of a single VAST transaction."""
 
     _rpc: "session.Session"
     txid: Optional[int] = None
+    _adbc_driver: Optional[AdbcDriver] = None
+    _adbc_conn: Optional[AdbcConnection] = None
+    _end_user: Optional[str] = None
 
     def __enter__(self):
         """Create a transaction and store its ID."""
         response = self._rpc.api.begin_transaction()
         self.txid = int(response.headers['tabular-txid'])
+
+        if self._adbc_driver is not None:
+            self._adbc_conn = AdbcConnection(
+                self._adbc_driver,
+                self._rpc.endpoint,
+                self._rpc.access,
+                self._rpc.secret,
+                self.txid,
+                self._end_user
+            )
+
         log.debug("opened txid=%016x", self.txid)
         return self
 
@@ -57,6 +78,10 @@ class Transaction:
         """On success, the transaction is committed. Otherwise, it is rolled back."""
         txid = self.txid
         self.txid = None
+        if self._adbc_conn is not None:
+            self.adbc_conn.close()
+            self._adbc_conn = None
+
         if (exc_type, exc_value, exc_traceback) == (None, None, None):
             log.debug("committing txid=%016x", txid)
             self._rpc.api.commit_transaction(txid)
@@ -110,3 +135,10 @@ class Transaction:
     def table_from_metadata(self, metadata: TableMetadata) -> ITable:
         """Create Table from TableMetadata."""
         return TableInTransaction(deepcopy(metadata), tx=self)
+
+    @property
+    def adbc_conn(self) -> AdbcConnection:
+        """ADBC connection in transaction."""
+        if self._adbc_conn is None:
+            raise NoAdbcConnectionError("Adbc Driver may not have been supplied")
+        return self._adbc_conn

vastdb/vast_flatbuf/tabular/GetTableStatsResponse.py

@@ -105,74 +105,66 @@ class GetTableStatsResponse(object):
             return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
         return 0
 
-def GetTableStatsResponseStart(builder):
-    builder.StartObject(9)
-
-def Start(builder):
-    GetTableStatsResponseStart(builder)
+    # GetTableStatsResponse
+    def VectorIndexMetadata(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(22))
+        if o != 0:
+            x = self._tab.Indirect(o + self._tab.Pos)
+            from vastdb.vast_flatbuf.tabular.VectorIndexMetadata import VectorIndexMetadata
+            obj = VectorIndexMetadata()
+            obj.Init(self._tab.Bytes, x)
+            return obj
+        return None
 
+def Start(builder): builder.StartObject(10)
+def GetTableStatsResponseStart(builder):
+    """This method is deprecated. Please switch to Start."""
+    return Start(builder)
+def AddNumRows(builder, numRows): builder.PrependInt64Slot(0, numRows, 0)
 def GetTableStatsResponseAddNumRows(builder, numRows):
-    builder.PrependInt64Slot(0, numRows, 0)
-
-def AddNumRows(builder, numRows):
-    GetTableStatsResponseAddNumRows(builder, numRows)
-
+    """This method is deprecated. Please switch to AddNumRows."""
+    return AddNumRows(builder, numRows)
+def AddSizeInBytes(builder, sizeInBytes): builder.PrependInt64Slot(1, sizeInBytes, 0)
 def GetTableStatsResponseAddSizeInBytes(builder, sizeInBytes):
-    builder.PrependInt64Slot(1, sizeInBytes, 0)
-
-def AddSizeInBytes(builder, sizeInBytes):
-    GetTableStatsResponseAddSizeInBytes(builder, sizeInBytes)
-
+    """This method is deprecated. Please switch to AddSizeInBytes."""
+    return AddSizeInBytes(builder, sizeInBytes)
+def AddIsExternalRowidAlloc(builder, isExternalRowidAlloc): builder.PrependBoolSlot(2, isExternalRowidAlloc, 0)
 def GetTableStatsResponseAddIsExternalRowidAlloc(builder, isExternalRowidAlloc):
-    builder.PrependBoolSlot(2, isExternalRowidAlloc, 0)
-
-def AddIsExternalRowidAlloc(builder, isExternalRowidAlloc):
-    GetTableStatsResponseAddIsExternalRowidAlloc(builder, isExternalRowidAlloc)
-
+    """This method is deprecated. Please switch to AddIsExternalRowidAlloc."""
+    return AddIsExternalRowidAlloc(builder, isExternalRowidAlloc)
+def AddAddressType(builder, addressType): builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(addressType), 0)
 def GetTableStatsResponseAddAddressType(builder, addressType):
-    builder.PrependUOffsetTRelativeSlot(3, flatbuffers.number_types.UOffsetTFlags.py_type(addressType), 0)
-
-def AddAddressType(builder, addressType):
-    GetTableStatsResponseAddAddressType(builder, addressType)
-
+    """This method is deprecated. Please switch to AddAddressType."""
+    return AddAddressType(builder, addressType)
+def AddVips(builder, vips): builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(vips), 0)
 def GetTableStatsResponseAddVips(builder, vips):
-    builder.PrependUOffsetTRelativeSlot(4, flatbuffers.number_types.UOffsetTFlags.py_type(vips), 0)
-
-def AddVips(builder, vips):
-    GetTableStatsResponseAddVips(builder, vips)
-
+    """This method is deprecated. Please switch to AddVips."""
+    return AddVips(builder, vips)
+def StartVipsVector(builder, numElems): return builder.StartVector(4, numElems, 4)
 def GetTableStatsResponseStartVipsVector(builder, numElems):
-    return builder.StartVector(4, numElems, 4)
-
-def StartVipsVector(builder, numElems):
-    return GetTableStatsResponseStartVipsVector(builder, numElems)
-
+    """This method is deprecated. Please switch to Start."""
+    return StartVipsVector(builder, numElems)
+def AddSortingKeyEnabled(builder, sortingKeyEnabled): builder.PrependBoolSlot(5, sortingKeyEnabled, 0)
 def GetTableStatsResponseAddSortingKeyEnabled(builder, sortingKeyEnabled):
-    builder.PrependBoolSlot(5, sortingKeyEnabled, 0)
-
-def AddSortingKeyEnabled(builder, sortingKeyEnabled):
-    GetTableStatsResponseAddSortingKeyEnabled(builder, sortingKeyEnabled)
-
+    """This method is deprecated. Please switch to AddSortingKeyEnabled."""
+    return AddSortingKeyEnabled(builder, sortingKeyEnabled)
+def AddSortingScore(builder, sortingScore): builder.PrependInt64Slot(6, sortingScore, 0)
 def GetTableStatsResponseAddSortingScore(builder, sortingScore):
-    builder.PrependInt64Slot(6, sortingScore, 0)
-
-def AddSortingScore(builder, sortingScore):
-    GetTableStatsResponseAddSortingScore(builder, sortingScore)
-
+    """This method is deprecated. Please switch to AddSortingScore."""
+    return AddSortingScore(builder, sortingScore)
+def AddWriteAmplification(builder, writeAmplification): builder.PrependInt64Slot(7, writeAmplification, 0)
 def GetTableStatsResponseAddWriteAmplification(builder, writeAmplification):
-    builder.PrependInt64Slot(7, writeAmplification, 0)
-
-def AddWriteAmplification(builder, writeAmplification):
-    GetTableStatsResponseAddWriteAmplification(builder, writeAmplification)
-
+    """This method is deprecated. Please switch to AddWriteAmplification."""
+    return AddWriteAmplification(builder, writeAmplification)
+def AddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount): builder.PrependInt64Slot(8, acummulativeRowInseritionCount, 0)
 def GetTableStatsResponseAddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount):
-    builder.PrependInt64Slot(8, acummulativeRowInseritionCount, 0)
-
-def AddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount):
-    GetTableStatsResponseAddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount)
-
+    """This method is deprecated. Please switch to AddAcummulativeRowInseritionCount."""
+    return AddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount)
+def AddVectorIndexMetadata(builder, vectorIndexMetadata): builder.PrependUOffsetTRelativeSlot(9, flatbuffers.number_types.UOffsetTFlags.py_type(vectorIndexMetadata), 0)
+def GetTableStatsResponseAddVectorIndexMetadata(builder, vectorIndexMetadata):
+    """This method is deprecated. Please switch to AddVectorIndexMetadata."""
+    return AddVectorIndexMetadata(builder, vectorIndexMetadata)
+def End(builder): return builder.EndObject()
 def GetTableStatsResponseEnd(builder):
-    return builder.EndObject()
-
-def End(builder):
-    return GetTableStatsResponseEnd(builder)
+    """This method is deprecated. Please switch to End."""
+    return End(builder)

vastdb/vast_flatbuf/tabular/ObjectDetails.py

@@ -95,74 +95,51 @@ class ObjectDetails(object):
             return self._tab.Get(flatbuffers.number_types.Int64Flags, o + self._tab.Pos)
         return 0
 
+def Start(builder): builder.StartObject(10)
 def ObjectDetailsStart(builder):
-    builder.StartObject(10)
-
-def Start(builder):
-    ObjectDetailsStart(builder)
-
+    """This method is deprecated. Please switch to Start."""
+    return Start(builder)
+def AddName(builder, name): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
 def ObjectDetailsAddName(builder, name):
-    builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(name), 0)
-
-def AddName(builder, name):
-    ObjectDetailsAddName(builder, name)
-
+    """This method is deprecated. Please switch to AddName."""
+    return AddName(builder, name)
+def AddProperties(builder, properties): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(properties), 0)
 def ObjectDetailsAddProperties(builder, properties):
-    builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(properties), 0)
-
-def AddProperties(builder, properties):
-    ObjectDetailsAddProperties(builder, properties)
-
+    """This method is deprecated. Please switch to AddProperties."""
+    return AddProperties(builder, properties)
+def AddHandle(builder, handle): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(handle), 0)
 def ObjectDetailsAddHandle(builder, handle):
-    builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(handle), 0)
-
-def AddHandle(builder, handle):
-    ObjectDetailsAddHandle(builder, handle)
-
+    """This method is deprecated. Please switch to AddHandle."""
+    return AddHandle(builder, handle)
+def AddNumRows(builder, numRows): builder.PrependUint64Slot(3, numRows, 0)
 def ObjectDetailsAddNumRows(builder, numRows):
-    builder.PrependUint64Slot(3, numRows, 0)
-
-def AddNumRows(builder, numRows):
-    ObjectDetailsAddNumRows(builder, numRows)
-
+    """This method is deprecated. Please switch to AddNumRows."""
+    return AddNumRows(builder, numRows)
+def AddSizeInBytes(builder, sizeInBytes): builder.PrependUint64Slot(4, sizeInBytes, 0)
 def ObjectDetailsAddSizeInBytes(builder, sizeInBytes):
-    builder.PrependUint64Slot(4, sizeInBytes, 0)
-
-def AddSizeInBytes(builder, sizeInBytes):
-    ObjectDetailsAddSizeInBytes(builder, sizeInBytes)
-
+    """This method is deprecated. Please switch to AddSizeInBytes."""
+    return AddSizeInBytes(builder, sizeInBytes)
+def AddNumPartitions(builder, numPartitions): builder.PrependUint64Slot(5, numPartitions, 0)
 def ObjectDetailsAddNumPartitions(builder, numPartitions):
-    builder.PrependUint64Slot(5, numPartitions, 0)
-
-def AddNumPartitions(builder, numPartitions):
-    ObjectDetailsAddNumPartitions(builder, numPartitions)
-
+    """This method is deprecated. Please switch to AddNumPartitions."""
+    return AddNumPartitions(builder, numPartitions)
+def AddSortingKeyEnabled(builder, sortingKeyEnabled): builder.PrependBoolSlot(6, sortingKeyEnabled, 0)
 def ObjectDetailsAddSortingKeyEnabled(builder, sortingKeyEnabled):
-    builder.PrependBoolSlot(6, sortingKeyEnabled, 0)
-
-def AddSortingKeyEnabled(builder, sortingKeyEnabled):
-    ObjectDetailsAddSortingKeyEnabled(builder, sortingKeyEnabled)
-
+    """This method is deprecated. Please switch to AddSortingKeyEnabled."""
+    return AddSortingKeyEnabled(builder, sortingKeyEnabled)
+def AddSortingScore(builder, sortingScore): builder.PrependInt64Slot(7, sortingScore, 0)
 def ObjectDetailsAddSortingScore(builder, sortingScore):
-    builder.PrependInt64Slot(7, sortingScore, 0)
-
-def AddSortingScore(builder, sortingScore):
-    ObjectDetailsAddSortingScore(builder, sortingScore)
-
+    """This method is deprecated. Please switch to AddSortingScore."""
+    return AddSortingScore(builder, sortingScore)
+def AddWriteAmplification(builder, writeAmplification): builder.PrependInt64Slot(8, writeAmplification, 0)
 def ObjectDetailsAddWriteAmplification(builder, writeAmplification):
-    builder.PrependInt64Slot(8, writeAmplification, 0)
-
-def AddWriteAmplification(builder, writeAmplification):
-    ObjectDetailsAddWriteAmplification(builder, writeAmplification)
-
+    """This method is deprecated. Please switch to AddWriteAmplification."""
+    return AddWriteAmplification(builder, writeAmplification)
+def AddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount): builder.PrependInt64Slot(9, acummulativeRowInseritionCount, 0)
 def ObjectDetailsAddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount):
-    builder.PrependInt64Slot(9, acummulativeRowInseritionCount, 0)
-
-def AddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount):
-    ObjectDetailsAddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount)
-
+    """This method is deprecated. Please switch to AddAcummulativeRowInseritionCount."""
+    return AddAcummulativeRowInseritionCount(builder, acummulativeRowInseritionCount)
+def End(builder): return builder.EndObject()
 def ObjectDetailsEnd(builder):
-    return builder.EndObject()
-
-def End(builder):
-    return ObjectDetailsEnd(builder)
+    """This method is deprecated. Please switch to End."""
+    return End(builder)

vastdb/vast_flatbuf/tabular/VectorIndexMetadata.py

@@ -0,0 +1,67 @@
+# automatically generated by the FlatBuffers compiler, do not modify
+
+# namespace: tabular
+
+import flatbuffers
+from flatbuffers.compat import import_numpy
+np = import_numpy()
+
+class VectorIndexMetadata(object):
+    __slots__ = ['_tab']
+
+    @classmethod
+    def GetRootAs(cls, buf, offset=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+        x = VectorIndexMetadata()
+        x.Init(buf, n + offset)
+        return x
+
+    @classmethod
+    def GetRootAsVectorIndexMetadata(cls, buf, offset=0):
+        """This method is deprecated. Please switch to GetRootAs."""
+        return cls.GetRootAs(buf, offset)
+    # VectorIndexMetadata
+    def Init(self, buf, pos):
+        self._tab = flatbuffers.table.Table(buf, pos)
+
+    # VectorIndexMetadata
+    def ColumnName(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        if o != 0:
+            return self._tab.String(o + self._tab.Pos)
+        return None
+
+    # VectorIndexMetadata
+    def DistanceMetric(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
+        if o != 0:
+            return self._tab.String(o + self._tab.Pos)
+        return None
+
+    # VectorIndexMetadata
+    def SqlFunctionName(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
+        if o != 0:
+            return self._tab.String(o + self._tab.Pos)
+        return None
+
+def Start(builder): builder.StartObject(3)
+def VectorIndexMetadataStart(builder):
+    """This method is deprecated. Please switch to Start."""
+    return Start(builder)
+def AddColumnName(builder, columnName): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(columnName), 0)
+def VectorIndexMetadataAddColumnName(builder, columnName):
+    """This method is deprecated. Please switch to AddColumnName."""
+    return AddColumnName(builder, columnName)
+def AddDistanceMetric(builder, distanceMetric): builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(distanceMetric), 0)
+def VectorIndexMetadataAddDistanceMetric(builder, distanceMetric):
+    """This method is deprecated. Please switch to AddDistanceMetric."""
+    return AddDistanceMetric(builder, distanceMetric)
+def AddSqlFunctionName(builder, sqlFunctionName): builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(sqlFunctionName), 0)
+def VectorIndexMetadataAddSqlFunctionName(builder, sqlFunctionName):
+    """This method is deprecated. Please switch to AddSqlFunctionName."""
+    return AddSqlFunctionName(builder, sqlFunctionName)
+def End(builder): return builder.EndObject()
+def VectorIndexMetadataEnd(builder):
+    """This method is deprecated. Please switch to End."""
+    return End(builder)

vastdb/vast_flatbuf/tabular/VipRange.py

@@ -38,19 +38,26 @@ class VipRange(object):
             return self._tab.Get(flatbuffers.number_types.Uint16Flags, o + self._tab.Pos)
         return 0
 
-def Start(builder): builder.StartObject(2)
 def VipRangeStart(builder):
-    """This method is deprecated. Please switch to Start."""
-    return Start(builder)
-def AddStartAddress(builder, startAddress): builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(startAddress), 0)
+    builder.StartObject(2)
+
+def Start(builder):
+    VipRangeStart(builder)
+
 def VipRangeAddStartAddress(builder, startAddress):
-    """This method is deprecated. Please switch to AddStartAddress."""
-    return AddStartAddress(builder, startAddress)
-def AddAddressCount(builder, addressCount): builder.PrependUint16Slot(1, addressCount, 0)
+    builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(startAddress), 0)
+
+def AddStartAddress(builder, startAddress):
+    VipRangeAddStartAddress(builder, startAddress)
+
 def VipRangeAddAddressCount(builder, addressCount):
-    """This method is deprecated. Please switch to AddAddressCount."""
-    return AddAddressCount(builder, addressCount)
-def End(builder): return builder.EndObject()
+    builder.PrependUint16Slot(1, addressCount, 0)
+
+def AddAddressCount(builder, addressCount):
+    VipRangeAddAddressCount(builder, addressCount)
+
 def VipRangeEnd(builder):
-    """This method is deprecated. Please switch to End."""
-    return End(builder)
+    return builder.EndObject()
+
+def End(builder):
+    return VipRangeEnd(builder)

{vastdb-2.0.2.dist-info → vastdb-2.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vastdb
-Version: 2.0.2
+Version: 2.0.5
 Summary: VAST Data SDK
 Home-page: https://github.com/vast-data/vastdb_sdk
 Author: VAST DATA
@@ -29,6 +29,7 @@ Requires-Dist: numpy
 Requires-Dist: requests
 Requires-Dist: xmltodict
 Requires-Dist: backoff==2.2.1
+Requires-Dist: adbc_driver_manager
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier