pyobvector 0.2.14__tar.gz → 0.2.16__tar.gz

{pyobvector-0.2.14 → pyobvector-0.2.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pyobvector
-Version: 0.2.14
+Version: 0.2.16
 Summary: A python SDK for OceanBase Vector Store, based on SQLAlchemy, compatible with Milvus API.
 Author: shanhaikang.shk
 Author-email: shanhaikang.shk@oceanbase.com
@@ -36,7 +36,7 @@ poetry install
 - install with pip:
 
 ```shell
-pip install pyobvector==0.2.14
+pip install pyobvector==0.2.16
 ```
 
 ## Build Doc

{pyobvector-0.2.14 → pyobvector-0.2.16}/README.md

@@ -15,7 +15,7 @@ poetry install
 - install with pip:
 
 ```shell
-pip install pyobvector==0.2.14
+pip install pyobvector==0.2.16
 ```
 
 ## Build Doc

{pyobvector-0.2.14 → pyobvector-0.2.16}/pyobvector/client/exceptions.py

@@ -108,7 +108,7 @@ class ExceptionsMessage:
     )
     CollectionNotExists = "Collection does not exist."
     MetricTypeParamTypeInvalid = "MetricType param type should be string."
-    MetricTypeValueInvalid = "MetricType should be 'l2'/'ip' in ann search."
+    MetricTypeValueInvalid = "MetricType should be 'l2'/'ip'/'neg_ip'/'cosine' in ann search."
     UsingInIDsWhenMultiPrimaryKey = "Using 'ids' when table has multi primary key."
     ClusterVersionIsLow = (
         "OceanBase Vector Store is not supported because cluster version is below 4.3.3.0."

{pyobvector-0.2.14 → pyobvector-0.2.16}/pyobvector/client/milvus_like_client.py

@@ -379,8 +379,8 @@ class MilvusLikeClient(Client):
                     )
                 lower_metric_type_str = search_params["metric_type"].lower()
                 if lower_metric_type_str not in (
-                    "l2", "neg_ip"
-                ):  # For OceanBase, only support l2/ip distance in ann_search
+                    "l2", "neg_ip", "cosine", "ip"
+                ):
                     raise VectorMetricTypeException(
                         code=ErrorCode.INVALID_ARGUMENT,
                         message=ExceptionsMessage.MetricTypeValueInvalid,

{pyobvector-0.2.14 → pyobvector-0.2.16}/pyobvector/schema/reflection.py

@@ -49,10 +49,12 @@ class OceanBaseTableDefinitionParser(MySQLTableDefinitionParser):
             # r"(?:(?P<type>\S+) )?KEY"
             r"(?: +{iq}(?P<name>(?:{esc_fq}|[^{fq}])+){fq})?"
             r"(?: +USING +(?P<using_pre>\S+))?"
-            r" +\((?P<columns>.+?)\)"
+            r" +\((?P<columns>[^)]+)\)"
             r"(?: +USING +(?P<using_post>\S+))?"
-            r"(?: +(KEY_)?BLOCK_SIZE *[ =]? *(?P<keyblock>\S+) *(LOCAL)?)?"
+            r"(?: +WITH +\((?P<vector_params>[^)]+)\))?"
             r"(?: +WITH PARSER +(?P<parser>\S+))?"
+            r"(?: +PARSER_PROPERTIES=\((?P<parser_properties>[^)]+)\))?"
+            r"(?: +(KEY_)?BLOCK_SIZE *[ =]? *(?P<keyblock>\S+) *(LOCAL)?)?"
             r"(?: +COMMENT +(?P<comment>(\x27\x27|\x27([^\x27])*?\x27)+))?"
             r"(?: +/\*(?P<version_sql>.+)\*/ *)?"
             r",?$".format(iq=quotes["iq"], esc_fq=quotes["esc_fq"], fq=quotes["fq"])
@@ -137,16 +139,15 @@ class OceanBaseTableDefinitionParser(MySQLTableDefinitionParser):
         if ret:
             tp, spec = ret
 
-            if tp is None:
-                return ret
-            if tp == "partition":
-                # do not handle partition
+            if tp is None or tp == "partition" or not isinstance(spec, dict):
                 return ret
+
             if tp == "fk_constraint":
-                if len(spec["table"]) == 2 and spec["table"][0] == self.default_schema:
-                    spec["table"] = spec["table"][1:]
-            if spec.get("onupdate", "").lower() == "restrict":
-                spec["onupdate"] = None
-            if spec.get("ondelete", "").lower() == "restrict":
-                spec["ondelete"] = None
+                table = spec.get("table", [])
+                if isinstance(table, list) and len(table) == 2 and table[0] == self.default_schema:
+                    spec["table"] = table[1:]
+
+            for action in ["onupdate", "ondelete"]:
+                if (spec.get(action) or "").lower() == "restrict":
+                    spec[action] = None
         return ret

{pyobvector-0.2.14 → pyobvector-0.2.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyobvector"
-version = "0.2.14"
+version = "0.2.16"
 description = "A python SDK for OceanBase Vector Store, based on SQLAlchemy, compatible with Milvus API."
 authors = ["shanhaikang.shk <shanhaikang.shk@oceanbase.com>"]
 readme = "README.md"