infinity-sdk 0.7.0.dev2__tar.gz → 0.7.0.dev4__tar.gz

{infinity_sdk-0.7.0.dev2/python/infinity_sdk/infinity_sdk.egg-info → infinity_sdk-0.7.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: infinity-sdk
-Version: 0.7.0.dev2
+Version: 0.7.0.dev4
 Summary: infinity
 License-Expression: Apache-2.0
 Requires-Python: <3.14,>=3.11
@@ -97,7 +97,7 @@ Infinity supports two working modes, embedded mode and client-server mode. The f
 
 2. Install the `infinity-sdk` package:
    ```bash
-   pip install infinity-sdk==0.7.0.dev2
+   pip install infinity-sdk==0.7.0.dev4
    ```
 
 3. Use Infinity to conduct a dense vector search:

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/README.md

@@ -96,7 +96,7 @@ If you are on Windows 10+, you must enable WSL or WSL2 to deploy Infinity using
 ### Install Infinity client
 
 ```
-pip install infinity-sdk==0.7.0.dev2
+pip install infinity-sdk==0.7.0.dev4
 ```
 
 ### Run a vector search

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "infinity-sdk"
-version = "0.7.0.dev2"
+version = "0.7.0.dev4"
 description = "infinity"
 readme = "python/infinity_sdk/README.md"
 license = "Apache-2.0"

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/README.md

@@ -63,7 +63,7 @@ Infinity supports two working modes, embedded mode and client-server mode. The f
 
 2. Install the `infinity-sdk` package:
    ```bash
-   pip install infinity-sdk==0.7.0.dev2
+   pip install infinity-sdk==0.7.0.dev4
    ```
 
 3. Use Infinity to conduct a dense vector search:

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/infinity/connection_pool.py

@@ -28,7 +28,10 @@ class ConnectionPool(object):
             self._create_conn()
 
     def __del__(self):
-        self.destroy()
+        try:
+            self.destroy()
+        except Exception:
+            pass  # Ignore errors in destructor
 
     def _create_conn(self):
         infinity_coon = infinity.connect(self.uri_)
@@ -54,6 +57,11 @@ class ConnectionPool(object):
             logging.debug("release_conn")
 
     def destroy(self):
+        if not hasattr(self, 'free_pool_'):
+            return
         for conn in iter(self.free_pool_):
-            conn.disconnect()
+            try:
+                conn.disconnect()
+            except Exception:
+                pass  # Ignore errors when disconnecting
         self.free_pool_.clear()

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/infinity/http_utils.py

@@ -101,7 +101,7 @@ def function_return_type(function_name, param_type):
     elif function_name in ["json_extract", "json_extract_string"]:
         return 'string'
     elif function_name == "json_extract_int":
-        return 'Int32'
+        return 'Int64'
     elif function_name == "json_extract_double":
         return 'Float64'
     elif function_name in ["json_extract_bool", "json_extract_isnull",

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/infinity/index.py

@@ -30,6 +30,7 @@ class IndexType(Enum):
     EMVB = 6
     BMP = 7
     DiskAnn = 8
+    PLAID = 9
 
     def to_ttype(self):
         match self:
@@ -49,6 +50,8 @@ class IndexType(Enum):
                 return ttypes.IndexType.BMP
             case IndexType.DiskAnn:
                 return ttypes.IndexType.DiskAnn
+            case IndexType.PLAID:
+                return ttypes.IndexType.PLAID
             case _:
                 raise InfinityException(ErrorCode.INVALID_INDEX_TYPE, "Unknown index type")
 

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/infinity/remote_thrift/client.py

@@ -70,8 +70,11 @@ class ThriftInfinityClient:
             self.logger.addHandler(ch)
 
     def __del__(self):
-        if self._is_connected:
-            self.disconnect()
+        try:
+            if hasattr(self, '_is_connected') and self._is_connected:
+                self.disconnect()
+        except Exception:
+            pass  # Ignore errors in destructor
 
     def _reconnect(self):
         if self.transport is not None:
@@ -120,8 +123,8 @@ class ThriftInfinityClient:
         # version: 0.6.8 and 0.6.9 and 0.6.10, client_version: 33
         # version: 0.6.13, client_version: 34
         # version: 0.6.15, client_version: 35
-        # version: 0.7.0, 0.7.0.dev1, 0.7.0.dev2, client_version: 36
-        res = self.client.Connect(ConnectRequest(client_version=36))  # 0.7.0.dev2
+        # version: 0.7.0, 0.7.0.dev1, 0.7.0.dev2, 0.7.0.dev3 and 0.7.0.dev4, client_version: 36
+        res = self.client.Connect(ConnectRequest(client_version=36))  # 0.7.0.dev4
         if res.error_code != 0:
             raise InfinityException(res.error_code, res.error_msg)
         self.session_id = res.session_id

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/infinity/remote_thrift/infinity.py

@@ -42,8 +42,11 @@ class RemoteThriftInfinityConnection(InfinityConnection, ABC):
         self._is_connected = True
 
     def __del__(self):
-        if self._is_connected is True:
-            self.disconnect()
+        try:
+            if hasattr(self, '_is_connected') and self._is_connected:
+                self.disconnect()
+        except Exception:
+            pass  # Ignore errors in destructor
 
     @name_validity_check("db_name", "DB")
     def create_database(self, db_name: str, conflict_type: ConflictType = ConflictType.Error, comment: str = None):

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/infinity/remote_thrift/infinity_thrift_rpc/ttypes.py

@@ -398,6 +398,7 @@ class IndexType(object):
     SecondaryFunctional = 5
     EMVB = 6
     DiskAnn = 7
+    PLAID = 8
 
     _VALUES_TO_NAMES = {
         0: "IVF",
@@ -408,6 +409,7 @@ class IndexType(object):
         5: "SecondaryFunctional",
         6: "EMVB",
         7: "DiskAnn",
+        8: "PLAID",
     }
 
     _NAMES_TO_VALUES = {
@@ -419,6 +421,7 @@ class IndexType(object):
         "SecondaryFunctional": 5,
         "EMVB": 6,
         "DiskAnn": 7,
+        "PLAID": 8,
     }
 
 

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4/python/infinity_sdk/infinity_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: infinity-sdk
-Version: 0.7.0.dev2
+Version: 0.7.0.dev4
 Summary: infinity
 License-Expression: Apache-2.0
 Requires-Python: <3.14,>=3.11
@@ -97,7 +97,7 @@ Infinity supports two working modes, embedded mode and client-server mode. The f
 
 2. Install the `infinity-sdk` package:
    ```bash
-   pip install infinity-sdk==0.7.0.dev2
+   pip install infinity-sdk==0.7.0.dev4
    ```
 
 3. Use Infinity to conduct a dense vector search:

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/infinity_sdk.egg-info/SOURCES.txt

@@ -2,28 +2,6 @@ LICENSE
 README.md
 pyproject.toml
 python/infinity_sdk/README.md
-python/infinity_sdk/build/lib/infinity/__init__.py
-python/infinity_sdk/build/lib/infinity/common.py
-python/infinity_sdk/build/lib/infinity/connection_pool.py
-python/infinity_sdk/build/lib/infinity/db.py
-python/infinity_sdk/build/lib/infinity/errors.py
-python/infinity_sdk/build/lib/infinity/index.py
-python/infinity_sdk/build/lib/infinity/infinity.py
-python/infinity_sdk/build/lib/infinity/rag_tokenizer.py
-python/infinity_sdk/build/lib/infinity/table.py
-python/infinity_sdk/build/lib/infinity/utils.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/__init__.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/client.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/db.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/infinity.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/query_builder.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/table.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/types.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/utils.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/infinity_thrift_rpc/InfinityService.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/infinity_thrift_rpc/__init__.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/infinity_thrift_rpc/constants.py
-python/infinity_sdk/build/lib/infinity/remote_thrift/infinity_thrift_rpc/ttypes.py
 python/infinity_sdk/infinity/__init__.py
 python/infinity_sdk/infinity/common.py
 python/infinity_sdk/infinity/connection_pool.py

{infinity_sdk-0.7.0.dev2 → infinity_sdk-0.7.0.dev4}/python/infinity_sdk/infinity_sdk.egg-info/top_level.txt

@@ -1,2 +1 @@
-build
 infinity

infinity_sdk-0.7.0.dev2/python/infinity_sdk/build/lib/infinity/__init__.py

@@ -1,34 +0,0 @@
-# Copyright(C) 2023 InfiniFlow, Inc. All rights reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-# import importlib.metadata
-#
-# __version__ = importlib.metadata.version("infinity_sdk")
-
-import logging
-import os
-# import pkg_resources
-# __version__ = pkg_resources.get_distribution("infinity_sdk").version
-
-from infinity.common import URI, NetworkAddress, LOCAL_HOST, LOCAL_INFINITY_PATH, InfinityException
-from infinity.infinity import InfinityConnection
-from infinity.remote_thrift.infinity import RemoteThriftInfinityConnection
-from infinity.errors import ErrorCode
-
-
-def connect(uri=LOCAL_HOST, logger: logging.Logger = None) -> InfinityConnection:
-    if isinstance(uri, NetworkAddress):
-        return RemoteThriftInfinityConnection(uri, logger)
-    else:
-        raise InfinityException(ErrorCode.INVALID_SERVER_ADDRESS, f"Unknown uri: {uri}")

infinity_sdk-0.7.0.dev2/python/infinity_sdk/build/lib/infinity/common.py

@@ -1,124 +0,0 @@
-# Copyright(C) 2023 InfiniFlow, Inc. All rights reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from pathlib import Path
-from typing import Union
-from dataclasses import dataclass, field
-import numpy as np
-
-
-class NetworkAddress:
-    def __init__(self, ip, port):
-        self.ip = ip
-        self.port = port
-
-    def __str__(self):
-        return f'IP: {self.ip}, Port: {self.port}'
-
-
-@dataclass
-class SparseVector:
-    indices: list[int]
-    values: Union[list[float], list[int], None] = None
-
-    def __post_init__(self):
-        assert (self.values is None) or (len(self.indices) == len(self.values))
-
-    def to_dict_old(self):
-        d = {"indices": self.indices}
-        if self.values is not None:
-            d["values"] = self.values
-        return d
-
-    def to_dict(self):
-        if self.values is None:
-            raise ValueError("SparseVector.values is None")
-        result = {}
-        for i, v in zip(self.indices, self.values):
-            result[str(i)] = v
-        return result
-
-    @staticmethod
-    def from_dict(d):
-        return SparseVector(d["indices"], d.get("values"))
-
-    def __str__(self):
-        return f"SparseVector(indices={self.indices}{'' if self.values is None else f', values={self.values}'})"
-
-    def __repr__(self):
-        return str(self)
-
-
-@dataclass
-class Array:
-    elements: list = field(default_factory=list)
-
-    def append(self, element):
-        self.elements.append(element)
-
-    def __init__(self, *args):
-        self.elements = list(args)
-
-    def __str__(self):
-        return f"Array({', '.join(str(e) for e in self.elements)})"
-
-    def __repr__(self):
-        return str(self)
-
-
-@dataclass
-class FDE:
-    """Fixed Dimensional Encoding function call for insert operations."""
-    tensor_data: list  # 2D array of numbers
-    target_dimension: int  # Target embedding dimension
-
-    def __init__(self, tensor_data: list, target_dimension: int):
-        self.tensor_data = tensor_data
-        self.target_dimension = target_dimension
-
-    def __str__(self):
-        return f"FDE(tensor_data={self.tensor_data}, target_dimension={self.target_dimension})"
-
-    def __repr__(self):
-        return str(self)
-
-
-URI = Union[NetworkAddress, Path]
-VEC = Union[list, np.ndarray, FDE]
-INSERT_DATA = dict[str, Union[str, int, float, list[Union[int, float]]], SparseVector, dict, Array, FDE]
-
-LOCAL_HOST = NetworkAddress("127.0.0.1", 23817)
-
-# test embedded_infinity
-LOCAL_INFINITY_PATH = "/var/infinity"
-
-
-class ConflictType(object):
-    Ignore = 0
-    Error = 1
-    Replace = 2
-
-
-class SortType(object):
-    Asc = 0
-    Desc = 1
-
-
-class InfinityException(Exception):
-    def __init__(self, error_code=0, error_msg=None):
-        self.error_code = error_code
-        self.error_msg = error_msg
-
-
-DEFAULT_MATCH_VECTOR_TOPN = 10
-DEFAULT_MATCH_SPARSE_TOPN = 10

infinity_sdk-0.7.0.dev2/python/infinity_sdk/build/lib/infinity/connection_pool.py

@@ -1,59 +0,0 @@
-# Copyright(C) 2024 InfiniFlow, Inc. All rights reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from threading import Lock
-import infinity
-from infinity.common import NetworkAddress
-import logging
-
-
-class ConnectionPool(object):
-    def __init__(self, uri=NetworkAddress("127.0.0.1", 23817), max_size=16):
-        self.uri_ = uri
-        self.max_size_ = max_size
-        self.free_pool_ = []
-        self.lock_ = Lock()
-        for i in range(max_size):
-            self._create_conn()
-
-    def _del__(self):
-        self.destroy()
-
-    def _create_conn(self):
-        infinity_coon = infinity.connect(self.uri_)
-        self.free_pool_.append(infinity_coon)
-
-    def get_conn(self):
-        with self.lock_:
-            if (len(self.free_pool_) == 0):
-                self._create_conn()
-            conn = self.free_pool_.pop()
-            logging.debug("get_conn")
-            return conn
-
-    def release_conn(self, conn):
-        """
-        Note: User is allowed to release a connection not created by ConnectionPool, or not releasing(due to exception or some other reasons) a connection created by ConnectionPool.
-        """
-        with self.lock_:
-            if (self.free_pool_.count(conn)):
-                raise Exception("the connection has been released")
-            if (len(self.free_pool_) < self.max_size_):
-                self.free_pool_.append(conn)
-            logging.debug("release_conn")
-
-    def destroy(self):
-        for conn in iter(self.free_pool_):
-            conn.disconnect()
-        self.free_pool_.clear()

infinity_sdk-0.7.0.dev2/python/infinity_sdk/build/lib/infinity/index.py

@@ -1,103 +0,0 @@
-# Copyright(C) 2023 InfiniFlow, Inc. All rights reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from enum import Enum
-
-import infinity.remote_thrift.infinity_thrift_rpc.ttypes as ttypes
-from infinity.common import InfinityException
-from infinity.errors import ErrorCode
-
-
-class IndexType(Enum):
-    IVF = 1
-    Hnsw = 2
-    FullText = 3
-    Secondary = 4
-    EMVB = 5
-    BMP = 6
-    DiskAnn = 7
-
-    def to_ttype(self):
-        match self:
-            case IndexType.IVF:
-                return ttypes.IndexType.IVF
-            case IndexType.Hnsw:
-                return ttypes.IndexType.Hnsw
-            case IndexType.FullText:
-                return ttypes.IndexType.FullText
-            case IndexType.Secondary:
-                return ttypes.IndexType.Secondary
-            case IndexType.EMVB:
-                return ttypes.IndexType.EMVB
-            case IndexType.BMP:
-                return ttypes.IndexType.BMP
-            case IndexType.DiskAnn:
-                return ttypes.IndexType.DiskAnn
-            case _:
-                raise InfinityException(ErrorCode.INVALID_INDEX_TYPE, "Unknown index type")
-
-
-class InitParameter:
-    def __init__(self, param_name: str, param_value: str):
-        self.param_name = param_name
-        self.param_value = param_value
-
-    def __str__(self):
-        return f"InitParameter({self.param_name}, {self.param_value})"
-
-    def __repr__(self):
-        return self.__str__()
-
-    def to_ttype(self):
-        return ttypes.InitParameter(self.param_name, self.param_value)
-
-
-class IndexInfo:
-    def __init__(self, column_name: str, index_type: IndexType, params: dict = None):
-        self.column_name = column_name
-        self.index_type = index_type
-        if params is not None:
-            if isinstance(params, dict):
-                self.params = params
-            else:
-                raise InfinityException(ErrorCode.INVALID_INDEX_PARAM, f"{params} should be dictionary type")
-        else:
-            self.params = None
-
-    def __str__(self):
-        return f"IndexInfo({self.column_name}, {self.index_type}, {self.params})"
-
-    def __repr__(self):
-        return self.__str__()
-
-    def __eq__(self, other):
-        return self.column_name == other.index_name and self.index_type == other.index_type and self.params == other.params
-
-    def __hash__(self):
-        return hash((self.column_name, self.index_type, self.params))
-
-    def to_ttype(self):
-        init_params_list = []
-        if self.params is not None:
-            for key, value in self.params.items():
-                if isinstance(value, str):
-                    init_params_list.append(ttypes.InitParameter(key, value))
-                else:
-                    raise InfinityException(ErrorCode.INVALID_INDEX_PARAM, f"{value} should be string type")
-
-        return ttypes.IndexInfo(
-            self.column_name.strip(),
-            self.index_type.to_ttype(),
-            init_params_list
-        )

infinity_sdk-0.7.0.dev2/python/infinity_sdk/build/lib/infinity/infinity.py

@@ -1,79 +0,0 @@
-# Copyright(C) 2023 InfiniFlow, Inc. All rights reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#      https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-from abc import ABC, abstractmethod
-from enum import Enum
-
-from infinity import URI
-
-
-# abstract class
-class InfinityConnection(ABC):
-    def __init__(self, uri):
-        self.uri = uri
-
-    @abstractmethod
-    def create_database(self, db_name, options=None, comment: str = None):
-        pass
-
-    @abstractmethod
-    def list_databases(self):
-        pass
-
-    @abstractmethod
-    def show_database(self, db_name):
-        pass
-
-    @abstractmethod
-    def drop_database(self, db_name, options=None):
-        pass
-
-    @abstractmethod
-    def get_database(self, db_name):
-        pass
-
-    @abstractmethod
-    def show_current_node(self):
-        pass
-
-    @abstractmethod
-    def disconnect(self):
-        pass
-
-    @abstractmethod
-    def create_database_snapshot(self, snapshot_name: str, db_name: str):
-        pass
-
-    @abstractmethod
-    def restore_database_snapshot(self, snapshot_name: str):
-        pass
-
-    @abstractmethod
-    def create_system_snapshot(self, snapshot_name: str):
-        pass
-
-    @abstractmethod
-    def restore_system_snapshot(self, snapshot_name: str):
-        pass
-
-    @abstractmethod
-    def list_snapshots(self):
-        pass
-
-    @abstractmethod
-    def show_snapshot(self, snapshot_name: str):
-        pass
-    
-    @abstractmethod
-    def drop_snapshot(self, snapshot_name: str):
-        pass