pyobvector 0.2.24__tar.gz → 0.2.26__tar.gz

{pyobvector-0.2.24 → pyobvector-0.2.26}/.github/workflows/ci.yml

@@ -13,7 +13,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]
+        python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
         image_tag: ["4.4.1.0-100000032025101610"]
         init_sql: ["ALTER SYSTEM ob_vector_memory_limit_percentage = 30; SET GLOBAL ob_query_timeout=100000000;"]
         test_filter: ["tests/test_hybrid_search.py::HybridSearchTest"]
@@ -65,3 +65,25 @@ jobs:
       - name: Run tests
         run: |
           make test TEST_FILTER='${{ matrix.test_filter }}'
+
+  test-embedded-seekdb:
+    name: Test embedded SeekDB
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out code
+        uses: actions/checkout@v6
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
+        with:
+          python-version: "3.12"
+
+      - name: Install dependencies
+        run: uv sync --dev
+
+      - name: Install pyseekdb (optional dependency for embedded SeekDB)
+        run: uv pip install pyseekdb
+
+      - name: Run embedded SeekDB tests
+        run: |
+          uv run python -m pytest tests/test_seekdb_embedded.py -v

pyobvector-0.2.24/README.md → pyobvector-0.2.26/PKG-INFO

@@ -1,3 +1,24 @@
+Metadata-Version: 2.4
+Name: pyobvector
+Version: 0.2.26
+Summary: A python SDK for OceanBase Vector Store, based on SQLAlchemy, compatible with Milvus API.
+Project-URL: Homepage, https://github.com/oceanbase/pyobvector
+Project-URL: Repository, https://github.com/oceanbase/pyobvector.git
+Author-email: "shanhaikang.shk" <shanhaikang.shk@oceanbase.com>
+License-Expression: Apache-2.0
+License-File: LICENSE
+Keywords: obvector,oceanbase,vector store
+Requires-Python: >=3.10
+Requires-Dist: aiomysql>=0.3.2
+Requires-Dist: numpy>=1.17.0
+Requires-Dist: pydantic<3,>=2.7.0
+Requires-Dist: pymysql>=1.1.1
+Requires-Dist: sqlalchemy<=3,>=1.4
+Requires-Dist: sqlglot>=26.0.1
+Provides-Extra: pyseekdb
+Requires-Dist: pyseekdb>=0.1.0; (python_version >= '3.11') and extra == 'pyseekdb'
+Description-Content-Type: text/markdown
+
 # pyobvector
 
 A python SDK for OceanBase Multimodal Store (Vector Store / Full Text Search / JSON Table), based on SQLAlchemy, compatible with Milvus API.
@@ -15,7 +36,13 @@ uv sync
 - install with pip:
 
 ```shell
-pip install pyobvector==0.2.24
+pip install pyobvector==0.2.26
+```
+
+- for **embedded SeekDB** support (local SeekDB without server):
+
+```shell
+pip install pyobvector[pyseekdb]
 ```
 
 ## Build Doc
@@ -33,10 +60,11 @@ For detailed release notes and changelog, see [RELEASE_NOTES.md](RELEASE_NOTES.m
 
 ## Usage
 
-`pyobvector` supports three modes:
+`pyobvector` supports four modes:
 
 - `Milvus compatible mode`: You can use the `MilvusLikeClient` class to use vector storage in a way similar to the Milvus API
 - `SQLAlchemy hybrid mode`: You can use the vector storage function provided by the `ObVecClient` class and execute the relational database statement with the SQLAlchemy library. In this mode, you can regard `pyobvector` as an extension of SQLAlchemy.
+- `Embedded SeekDB mode`: Use `ObVecClient` or `SeekdbRemoteClient` with local embedded SeekDB (no server). Same API as remote: `create_table`, `insert`, `ann_search`, etc. Requires optional dependency: `pip install pyobvector[pyseekdb]`.
 - `Hybrid Search mode`: You can use the `HybridSearch` class to perform hybrid search that combines full-text search and vector similarity search, with Elasticsearch-compatible query syntax.
 
 ### Milvus compatible mode
@@ -264,6 +292,70 @@ engine = create_async_engine(connection_str)
 
 - For further usage in pure `SQLAlchemy` mode, please refer to [SQLAlchemy](https://www.sqlalchemy.org/)
 
+### Embedded SeekDB mode
+
+Use the same ObClient/ObVecClient API with **embedded SeekDB** (local file, no server). Install the optional dependency:
+
+```shell
+pip install pyobvector[pyseekdb]
+```
+
+- connect with path or with an existing `pyseekdb.Client`:
+
+```python
+from pyobvector import SeekdbRemoteClient, ObVecClient
+from pyobvector.client.ob_client import ObClient
+
+# Option 1: path to SeekDB data directory
+client = SeekdbRemoteClient(path="./seekdb_data", database="test")
+
+# Option 2: use an existing pyseekdb.Client
+import pyseekdb
+pyseekdb_client = pyseekdb.Client(path="./seekdb_data", database="test")
+client = SeekdbRemoteClient(pyseekdb_client=pyseekdb_client)
+
+# Option 3: ObVecClient directly
+client = ObVecClient(path="./seekdb_data", db_name="test")
+
+assert isinstance(client, ObVecClient)
+assert isinstance(client, ObClient)
+```
+
+- create table, insert, and ann search (same API as remote):
+
+```python
+from sqlalchemy import Column, Integer, VARCHAR
+from pyobvector import VECTOR, VectorIndex, l2_distance
+
+client.drop_table_if_exist("vec_table")
+client.create_table(
+    table_name="vec_table",
+    columns=[
+        Column("id", Integer, primary_key=True),
+        Column("title", VARCHAR(255)),
+        Column("vec", VECTOR(3)),
+    ],
+    indexes=[VectorIndex("vec_idx", "vec", params="distance=l2, type=hnsw, lib=vsag")],
+    mysql_organization="heap",
+)
+client.insert("vec_table", data=[
+    {"id": 1, "title": "doc A", "vec": [1.0, 1.0, 1.0]},
+    {"id": 2, "title": "doc B", "vec": [1.0, 2.0, 3.0]},
+])
+res = client.ann_search(
+    "vec_table",
+    vec_data=[1.0, 2.0, 3.0],
+    vec_column_name="vec",
+    distance_func=l2_distance,
+    with_dist=True,
+    topk=5,
+    output_column_names=["id", "title"],
+)
+client.drop_table_if_exist("vec_table")
+```
+
+- See `tests/test_seekdb_embedded.py` for more examples.
+
 ### Hybrid Search Mode
 
 `pyobvector` supports hybrid search that combines full-text search and vector similarity search, with query syntax compatible with Elasticsearch. This allows you to perform semantic search with both keyword matching and vector similarity in a single query.

pyobvector-0.2.24/PKG-INFO → pyobvector-0.2.26/README.md

@@ -1,22 +1,3 @@
-Metadata-Version: 2.4
-Name: pyobvector
-Version: 0.2.24
-Summary: A python SDK for OceanBase Vector Store, based on SQLAlchemy, compatible with Milvus API.
-Project-URL: Homepage, https://github.com/oceanbase/pyobvector
-Project-URL: Repository, https://github.com/oceanbase/pyobvector.git
-Author-email: "shanhaikang.shk" <shanhaikang.shk@oceanbase.com>
-License-Expression: Apache-2.0
-License-File: LICENSE
-Keywords: obvector,oceanbase,vector store
-Requires-Python: >=3.9
-Requires-Dist: aiomysql>=0.3.2
-Requires-Dist: numpy>=1.17.0
-Requires-Dist: pydantic<3,>=2.7.0
-Requires-Dist: pymysql>=1.1.1
-Requires-Dist: sqlalchemy<=3,>=1.4
-Requires-Dist: sqlglot>=26.0.1
-Description-Content-Type: text/markdown
-
 # pyobvector
 
 A python SDK for OceanBase Multimodal Store (Vector Store / Full Text Search / JSON Table), based on SQLAlchemy, compatible with Milvus API.
@@ -34,7 +15,13 @@ uv sync
 - install with pip:
 
 ```shell
-pip install pyobvector==0.2.24
+pip install pyobvector==0.2.26
+```
+
+- for **embedded SeekDB** support (local SeekDB without server):
+
+```shell
+pip install pyobvector[pyseekdb]
 ```
 
 ## Build Doc
@@ -52,10 +39,11 @@ For detailed release notes and changelog, see [RELEASE_NOTES.md](RELEASE_NOTES.m
 
 ## Usage
 
-`pyobvector` supports three modes:
+`pyobvector` supports four modes:
 
 - `Milvus compatible mode`: You can use the `MilvusLikeClient` class to use vector storage in a way similar to the Milvus API
 - `SQLAlchemy hybrid mode`: You can use the vector storage function provided by the `ObVecClient` class and execute the relational database statement with the SQLAlchemy library. In this mode, you can regard `pyobvector` as an extension of SQLAlchemy.
+- `Embedded SeekDB mode`: Use `ObVecClient` or `SeekdbRemoteClient` with local embedded SeekDB (no server). Same API as remote: `create_table`, `insert`, `ann_search`, etc. Requires optional dependency: `pip install pyobvector[pyseekdb]`.
 - `Hybrid Search mode`: You can use the `HybridSearch` class to perform hybrid search that combines full-text search and vector similarity search, with Elasticsearch-compatible query syntax.
 
 ### Milvus compatible mode
@@ -283,6 +271,70 @@ engine = create_async_engine(connection_str)
 
 - For further usage in pure `SQLAlchemy` mode, please refer to [SQLAlchemy](https://www.sqlalchemy.org/)
 
+### Embedded SeekDB mode
+
+Use the same ObClient/ObVecClient API with **embedded SeekDB** (local file, no server). Install the optional dependency:
+
+```shell
+pip install pyobvector[pyseekdb]
+```
+
+- connect with path or with an existing `pyseekdb.Client`:
+
+```python
+from pyobvector import SeekdbRemoteClient, ObVecClient
+from pyobvector.client.ob_client import ObClient
+
+# Option 1: path to SeekDB data directory
+client = SeekdbRemoteClient(path="./seekdb_data", database="test")
+
+# Option 2: use an existing pyseekdb.Client
+import pyseekdb
+pyseekdb_client = pyseekdb.Client(path="./seekdb_data", database="test")
+client = SeekdbRemoteClient(pyseekdb_client=pyseekdb_client)
+
+# Option 3: ObVecClient directly
+client = ObVecClient(path="./seekdb_data", db_name="test")
+
+assert isinstance(client, ObVecClient)
+assert isinstance(client, ObClient)
+```
+
+- create table, insert, and ann search (same API as remote):
+
+```python
+from sqlalchemy import Column, Integer, VARCHAR
+from pyobvector import VECTOR, VectorIndex, l2_distance
+
+client.drop_table_if_exist("vec_table")
+client.create_table(
+    table_name="vec_table",
+    columns=[
+        Column("id", Integer, primary_key=True),
+        Column("title", VARCHAR(255)),
+        Column("vec", VECTOR(3)),
+    ],
+    indexes=[VectorIndex("vec_idx", "vec", params="distance=l2, type=hnsw, lib=vsag")],
+    mysql_organization="heap",
+)
+client.insert("vec_table", data=[
+    {"id": 1, "title": "doc A", "vec": [1.0, 1.0, 1.0]},
+    {"id": 2, "title": "doc B", "vec": [1.0, 2.0, 3.0]},
+])
+res = client.ann_search(
+    "vec_table",
+    vec_data=[1.0, 2.0, 3.0],
+    vec_column_name="vec",
+    distance_func=l2_distance,
+    with_dist=True,
+    topk=5,
+    output_column_names=["id", "title"],
+)
+client.drop_table_if_exist("vec_table")
+```
+
+- See `tests/test_seekdb_embedded.py` for more examples.
+
 ### Hybrid Search Mode
 
 `pyobvector` supports hybrid search that combines full-text search and vector similarity search, with query syntax compatible with Elasticsearch. This allows you to perform semantic search with both keyword matching and vector similarity in a single query.

{pyobvector-0.2.24 → pyobvector-0.2.26}/RELEASE_NOTES.md

@@ -2,6 +2,15 @@
 
 This file documents all released versions and their notable changes for the pyobvector project. Changes are grouped by version and categorized as Added (new features), Changed (modifications), Fixed (bug fixes), and Security (security updates).
 
+## [0.2.26](https://github.com/oceanbase/pyobvector/compare/release-v0.2.25...release-v0.2.26) - 2026-04-15
+
+-Fix: Compatible with the latest version of sqlglot
+
+## [0.2.25](https://github.com/oceanbase/pyobvector/compare/release-v0.2.24...release-v0.2.25) - 2026-03-10
+
+- Fix: preserve table options when adding sparse vector indexes
+- Feat: ObClient embedded SeekDB support, docs, tests and CI
+
 ## [0.2.24](https://github.com/oceanbase/pyobvector/compare/release-v0.2.23...release-v0.2.24) - 2026-02-05
 
 - Fix: preserve table options when adding sparse vector indexes

{pyobvector-0.2.24 → pyobvector-0.2.26}/pyobvector/__init__.py

@@ -64,6 +64,7 @@ from .schema import (
 from .json_table import OceanBase
 
 __all__ = [
+    "SeekdbRemoteClient",
     "ObVecClient",
     "MilvusLikeClient",
     "ObVecJsonTableClient",

{pyobvector-0.2.24 → pyobvector-0.2.26}/pyobvector/client/__init__.py

@@ -5,8 +5,11 @@ in a way similar to the Milvus API.
 2. `SQLAlchemy hybrid mode`: You can use the vector storage function provided by the
 `ObVecClient` class and execute the relational database statement with the SQLAlchemy library.
 In this mode, you can regard `pyobvector` as an extension of SQLAlchemy.
+3. `Embedded SeekDB`: ObClient/ObVecClient support path= or pyseekdb_client= for embedded
+SeekDB (pip install pyobvector[pyseekdb]). Same API as remote: create_table, insert, etc.
 
-* ObVecClient           MySQL client in SQLAlchemy hybrid mode
+* SeekdbRemoteClient    Connect to embedded (path= / pyseekdb_client=) or remote; returns ObVecClient
+* ObVecClient           MySQL/SeekDB client in SQLAlchemy hybrid mode (uri, path, or pyseekdb_client)
 * MilvusLikeClient      Milvus compatible client
 * VecIndexType          VecIndexType is used to specify vector index type for MilvusLikeClient
 * IndexParam            Specify vector index parameters for MilvusLikeClient
@@ -31,6 +34,9 @@ In this mode, you can regard `pyobvector` as an extension of SQLAlchemy.
 * FtsIndexParam         Full Text Search index parameter
 """
 
+import os
+from typing import Any
+
 from .ob_vec_client import ObVecClient
 from .milvus_like_client import MilvusLikeClient
 from .ob_vec_json_table_client import ObVecJsonTableClient
@@ -40,7 +46,52 @@ from .collection_schema import FieldSchema, CollectionSchema
 from .partitions import *
 from .fts_index_param import FtsParser, FtsIndexParam
 
+
+def _resolve_password(password: str) -> str:
+    return password or os.environ.get("SEEKDB_PASSWORD", "")
+
+
+def SeekdbRemoteClient(
+    path: str | None = None,
+    uri: str | None = None,
+    host: str | None = None,
+    port: int | None = None,
+    tenant: str = "test",
+    database: str = "test",
+    user: str | None = None,
+    password: str = "",
+    pyseekdb_client: Any | None = None,
+    **kwargs: Any,
+) -> Any:
+    """
+    Connect to embedded SeekDB (path= or pyseekdb_client=) or remote OceanBase/SeekDB (uri/host=).
+    Returns ObVecClient with the same API (create_table, insert, ann_search, etc.).
+    Embedded requires: pip install pyobvector[pyseekdb]
+    """
+    password = _resolve_password(password)
+    if pyseekdb_client is not None:
+        return ObVecClient(pyseekdb_client=pyseekdb_client, **kwargs)
+    if path is not None:
+        return ObVecClient(path=path, db_name=database, **kwargs)
+    if uri is None and host is not None:
+        port = port if port is not None else 2881
+        uri = f"{host}:{port}"
+    if uri is None:
+        uri = "127.0.0.1:2881"
+    ob_user = user if user is not None else "root"
+    if "@" not in ob_user:
+        ob_user = f"{ob_user}@{tenant}"
+    return ObVecClient(
+        uri=uri,
+        user=ob_user,
+        password=password,
+        db_name=database,
+        **kwargs,
+    )
+
+
 __all__ = [
+    "SeekdbRemoteClient",
     "ObVecClient",
     "MilvusLikeClient",
     "ObVecJsonTableClient",

{pyobvector-0.2.24 → pyobvector-0.2.26}/pyobvector/client/collection_schema.py

@@ -1,7 +1,6 @@
 """FieldSchema & CollectionSchema definition module to be compatible with Milvus."""
 
 import copy
-from typing import Optional
 from sqlalchemy import Column
 from .schema_type import DataType, convert_datatype_to_sqltype
 from .exceptions import *
@@ -129,8 +128,8 @@ class CollectionSchema:
 
     def __init__(
         self,
-        fields: Optional[list[FieldSchema]] = None,
-        partitions: Optional[ObPartition] = None,
+        fields: list[FieldSchema] | None = None,
+        partitions: ObPartition | None = None,
         description: str = "",  # ignored in oceanbase
         **kwargs,
     ):

{pyobvector-0.2.24 → pyobvector-0.2.26}/pyobvector/client/fts_index_param.py

@@ -1,7 +1,6 @@
 """A module to specify fts index parameters"""
 
 from enum import Enum
-from typing import Optional, Union
 
 
 class FtsParser(Enum):
@@ -28,13 +27,13 @@ class FtsIndexParam:
         self,
         index_name: str,
         field_names: list[str],
-        parser_type: Optional[Union[FtsParser, str]] = None,
+        parser_type: FtsParser | str | None = None,
     ):
         self.index_name = index_name
         self.field_names = field_names
         self.parser_type = parser_type
 
-    def param_str(self) -> Optional[str]:
+    def param_str(self) -> str | None:
         """Convert parser type to string format for SQL."""
         if self.parser_type is None:
             return None  # Default Space parser, no need to specify

{pyobvector-0.2.24 → pyobvector-0.2.26}/pyobvector/client/index_param.py

@@ -1,7 +1,6 @@
 """A module to specify vector index parameters for MilvusLikeClient"""
 
 from enum import Enum
-from typing import Union
 
 
 class VecIndexType(Enum):
@@ -42,7 +41,7 @@ class IndexParam:
         self,
         index_name: str,
         field_name: str,
-        index_type: Union[VecIndexType, str],
+        index_type: VecIndexType | str,
         **kwargs,
     ):
         self.index_name = index_name

{pyobvector-0.2.24 → pyobvector-0.2.26}/pyobvector/client/milvus_like_client.py

@@ -2,7 +2,6 @@
 
 import logging
 import json
-from typing import Optional, Union
 
 from sqlalchemy.exc import NoSuchTableError
 from sqlalchemy import (
@@ -52,15 +51,15 @@ class MilvusLikeClient(Client):
     def create_collection(
         self,
         collection_name: str,
-        dimension: Optional[int] = None,
+        dimension: int | None = None,
         primary_field_name: str = "id",
-        id_type: Union[DataType, str] = DataType.INT64,
+        id_type: DataType | str = DataType.INT64,
         vector_field_name: str = "vector",
         metric_type: str = "l2",
         auto_id: bool = False,
-        timeout: Optional[float] = None,
-        schema: Optional[CollectionSchema] = None,  # Used for custom setup
-        index_params: Optional[IndexParams] = None,  # Used for custom setup
+        timeout: float | None = None,
+        schema: CollectionSchema | None = None,  # Used for custom setup
+        index_params: IndexParams | None = None,  # Used for custom setup
         max_length: int = 16384,
         **kwargs,
     ):  # pylint: disable=unused-argument
@@ -149,7 +148,7 @@ class MilvusLikeClient(Client):
     def get_collection_stats(
         self,
         collection_name: str,
-        timeout: Optional[float] = None,  # pylint: disable=unused-argument
+        timeout: float | None = None,  # pylint: disable=unused-argument
     ) -> dict:
         """Get collection row count.
 
@@ -171,7 +170,7 @@ class MilvusLikeClient(Client):
     def has_collection(
         self,
         collection_name: str,
-        timeout: Optional[float] = None,  # pylint: disable=unused-argument
+        timeout: float | None = None,  # pylint: disable=unused-argument
     ) -> bool:  # pylint: disable=unused-argument
         """Check if collection exists.
 
@@ -196,7 +195,7 @@ class MilvusLikeClient(Client):
         self,
         old_name: str,
         new_name: str,
-        timeout: Optional[float] = None,  # pylint: disable=unused-argument
+        timeout: float | None = None,  # pylint: disable=unused-argument
     ) -> None:
         """Rename collection.
 
@@ -236,7 +235,7 @@ class MilvusLikeClient(Client):
         self,
         collection_name: str,
         index_params: IndexParams,
-        timeout: Optional[float] = None,
+        timeout: float | None = None,
         **kwargs,
     ):  # pylint: disable=unused-argument
         """Create vector index with index params.
@@ -269,7 +268,7 @@ class MilvusLikeClient(Client):
         self,
         collection_name: str,
         index_name: str,
-        timeout: Optional[float] = None,
+        timeout: float | None = None,
         **kwargs,
     ):  # pylint: disable=unused-argument
         """Drop index on specified collection.
@@ -357,15 +356,15 @@ class MilvusLikeClient(Client):
     def search(
         self,
         collection_name: str,
-        data: Union[list, dict],
+        data: list | dict,
         anns_field: str,
         with_dist: bool = False,
         flter=None,
         limit: int = 10,
-        output_fields: Optional[list[str]] = None,
-        search_params: Optional[dict] = None,
-        timeout: Optional[float] = None,  # pylint: disable=unused-argument
-        partition_names: Optional[list[str]] = None,
+        output_fields: list[str] | None = None,
+        search_params: dict | None = None,
+        timeout: float | None = None,  # pylint: disable=unused-argument
+        partition_names: list[str] | None = None,
         **kwargs,  # pylint: disable=unused-argument
     ) -> list[dict]:
         """Perform ann search.
@@ -482,9 +481,9 @@ class MilvusLikeClient(Client):
         self,
         collection_name: str,
         flter=None,
-        output_fields: Optional[list[str]] = None,
-        timeout: Optional[float] = None,  # pylint: disable=unused-argument
-        partition_names: Optional[list[str]] = None,
+        output_fields: list[str] | None = None,
+        timeout: float | None = None,  # pylint: disable=unused-argument
+        partition_names: list[str] | None = None,
         **kwargs,  # pylint: disable=unused-argument
     ) -> list[dict]:
         """Query records.
@@ -549,10 +548,10 @@ class MilvusLikeClient(Client):
     def get(
         self,
         collection_name: str,
-        ids: Union[list, str, int] = None,
-        output_fields: Optional[list[str]] = None,
-        timeout: Optional[float] = None,  # pylint: disable=unused-argument
-        partition_names: Optional[list[str]] = None,
+        ids: list | str | int = None,
+        output_fields: list[str] | None = None,
+        timeout: float | None = None,  # pylint: disable=unused-argument
+        partition_names: list[str] | None = None,
         **kwargs,  # pylint: disable=unused-argument
     ) -> list[dict]:
         """Get records with specified primary field `ids`.
@@ -592,7 +591,7 @@ class MilvusLikeClient(Client):
             )
         if isinstance(ids, list):
             where_in_clause = table.c[pkey_names[0]].in_(ids)
-        elif isinstance(ids, (str, int)):
+        elif isinstance(ids, str | int):
             where_in_clause = table.c[pkey_names[0]].in_([ids])
         else:
             raise TypeError("'ids' is not a list/str/int")
@@ -629,10 +628,10 @@ class MilvusLikeClient(Client):
     def delete(
         self,
         collection_name: str,
-        ids: Optional[Union[list, str, int]] = None,
-        timeout: Optional[float] = None,  # pylint: disable=unused-argument
+        ids: list | str | int | None = None,
+        timeout: float | None = None,  # pylint: disable=unused-argument
         flter=None,
-        partition_name: Optional[str] = "",
+        partition_name: str | None = "",
         **kwargs,  # pylint: disable=unused-argument
     ) -> dict:
         """Delete data in collection.
@@ -667,7 +666,7 @@ class MilvusLikeClient(Client):
                 )
             if isinstance(ids, list):
                 where_in_clause = table.c[pkey_names[0]].in_(ids)
-            elif isinstance(ids, (str, int)):
+            elif isinstance(ids, str | int):
                 where_in_clause = table.c[pkey_names[0]].in_([ids])
             else:
                 raise TypeError("'ids' is not a list/str/int")
@@ -691,9 +690,9 @@ class MilvusLikeClient(Client):
     def insert(
         self,
         collection_name: str,
-        data: Union[dict, list[dict]],
-        timeout: Optional[float] = None,
-        partition_name: Optional[str] = "",
+        data: dict | list[dict],
+        timeout: float | None = None,
+        partition_name: str | None = "",
     ) -> None:  # pylint: disable=unused-argument
         """Insert data into collection.
 
@@ -717,10 +716,10 @@ class MilvusLikeClient(Client):
     def upsert(
         self,
         collection_name: str,
-        data: Union[dict, list[dict]],
-        timeout: Optional[float] = None,  # pylint: disable=unused-argument
-        partition_name: Optional[str] = "",
-    ) -> list[Union[str, int]]:
+        data: dict | list[dict],
+        timeout: float | None = None,  # pylint: disable=unused-argument
+        partition_name: str | None = "",
+    ) -> list[str | int]:
         """Update data in table. If primary key is duplicated, replace it.
 
         Args: