pyobvector 0.2.8__py3-none-any.whl → 0.2.10__py3-none-any.whl

pyobvector/client/ob_vec_json_table_client.py

@@ -144,15 +144,16 @@ class ObVecJsonTableClient(ObVecClient):
         if self.user_id is None:
             return True
         
-        session = self.session()
-        distinct_admin_ids = (
-            session.query(ObVecJsonTableClient.JsonTableDataTBL.admin_id)
-            .filter_by(user_id = self.user_id)
-            .distinct()
-            .all()
-        )
-        admin_ids = [admin_id[0] for admin_id in distinct_admin_ids]
-        return (len(admin_ids) == 0) or (len(admin_ids) == 1 and admin_ids[0] == self.admin_id)
+        with self.session() as session:
+            with session.begin():
+                distinct_admin_ids = (
+                    session.query(ObVecJsonTableClient.JsonTableDataTBL.admin_id)
+                    .filter_by(user_id = self.user_id)
+                    .distinct()
+                    .all()
+                )
+                admin_ids = [admin_id[0] for admin_id in distinct_admin_ids]
+                return (len(admin_ids) == 0) or (len(admin_ids) == 1 and admin_ids[0] == self.admin_id)
 
     def _reset(self):
         # Only for test
@@ -845,6 +846,7 @@ class ObVecJsonTableClient(ObVecClient):
             new_select_exprs.append(data_id_col_expr)
             need_replace_select_exprs = True
 
+        alias_names = set()
         for select_expr in ast.args['expressions']:
             if isinstance(select_expr, exp.Star):
                 need_replace_select_exprs = True
@@ -856,6 +858,8 @@ class ObVecJsonTableClient(ObVecClient):
                     col_expr.args['this'] = identifier
                     new_select_exprs.append(col_expr)
             else:
+                if isinstance(select_expr, exp.Alias):
+                    alias_names.add(select_expr.args['alias'].args['this'])
                 new_select_exprs.append(select_expr)
         if need_replace_select_exprs:
             ast.args['expressions'] = new_select_exprs
@@ -864,6 +868,9 @@ class ObVecJsonTableClient(ObVecClient):
         json_table_str = f"json_table({JSON_TABLE_DATA_TABLE_NAME}.jdata, '$' COLUMNS ({', '.join(json_table_meta_str)})) {tmp_table_name}"
 
         for col in ast.find_all(exp.Column):
+            col_name = col.args["this"].args["this"]
+            if col_name in alias_names:
+                continue
             if 'table' in col.args.keys():
                 if col.args['table'].args['this'] != JSON_TABLE_DATA_TABLE_NAME:
                     col.args['table'].args['this'] = tmp_table_name
@@ -893,6 +900,6 @@ class ObVecJsonTableClient(ObVecClient):
             where_clause.args['this'] = parse_one(extra_filter_str)
             ast.args['where'] = where_clause
 
-        select_sql = str(ast)
+        select_sql = ast.sql(dialect="mysql", identify=True)
         logger.debug(f"===================== do select: {select_sql}")
         return self.perform_raw_text_sql(select_sql)

{pyobvector-0.2.8.dist-info → pyobvector-0.2.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pyobvector
-Version: 0.2.8
+Version: 0.2.10
 Summary: A python SDK for OceanBase Vector Store, based on SQLAlchemy, compatible with Milvus API.
 Author: shanhaikang.shk
 Author-email: shanhaikang.shk@oceanbase.com
@@ -15,7 +15,7 @@ Requires-Dist: aiomysql (>=0.2.0,<0.3.0)
 Requires-Dist: numpy (>=1.17.0,<2.0.0)
 Requires-Dist: pydantic (>=2.7.0,<3)
 Requires-Dist: pymysql (>=1.1.1,<2.0.0)
-Requires-Dist: sqlalchemy (>=1.4,<2.0.36)
+Requires-Dist: sqlalchemy (>=1.4,<=3)
 Requires-Dist: sqlglot (>=26.0.1,<27.0.0)
 Description-Content-Type: text/markdown
 
@@ -36,7 +36,7 @@ poetry install
 - install with pip:
 
 ```shell
-pip install pyobvector==0.2.8
+pip install pyobvector==0.2.10
 ```
 
 ## Build Doc

{pyobvector-0.2.8.dist-info → pyobvector-0.2.10.dist-info}/RECORD

@@ -7,7 +7,7 @@ pyobvector/client/fts_index_param.py,sha256=hMCjA3Aecnt0uQQT6UQGTIIqdPk1M4gX4-zR
 pyobvector/client/index_param.py,sha256=3gXi66Ey1PO9x5_61CrH7DmPb496kviBQI5NT7nfbGc,6309
 pyobvector/client/milvus_like_client.py,sha256=CpPo6mkGE8iNFpKGBFof3h7E1VTzy1DAPGlFM9F_s8g,26373
 pyobvector/client/ob_vec_client.py,sha256=Yt2nG0w4268hg7DE0tqkGaytGsY-jqojX8hGTQjmsKg,29390
-pyobvector/client/ob_vec_json_table_client.py,sha256=l9zlryiyuk7wubksWJjmbUuNWErQZwM69K2qu4K8yVE,38621
+pyobvector/client/ob_vec_json_table_client.py,sha256=6xLQaU8HwhR4H9z26nyzNwDYXtvdCpfwz0ZK8NbPVkg,39034
 pyobvector/client/partitions.py,sha256=Bxwr5yVNlXwZc7SXBC03NeqL9giy4Fe6S2qZdHD8xGw,15621
 pyobvector/client/schema_type.py,sha256=ICCSriOhk-P7Q1PhK0D0XQMTd3ZDFenCrJMXp6hRQdw,1579
 pyobvector/json_table/__init__.py,sha256=X5MmK3f10oyJleUUFZJFeunMEfzmf6P1f_7094b-FZc,554
@@ -29,7 +29,7 @@ pyobvector/schema/vector_index.py,sha256=aNtrEBUclc4s6QuqCZpu3Hj3OdjyhbWgtLiJzo6
 pyobvector/util/__init__.py,sha256=D9EgRDlcMSDhY3uI__vnCl45Or75dOXMWSval5P5fqs,251
 pyobvector/util/ob_version.py,sha256=ZIySam8q_MCiwctAiAHPB4GdAzGQiXEo1wVkc9IOTDU,1539
 pyobvector/util/vector.py,sha256=xyM-NuOyd78K7P3kinqyWvLIzEbf9c-4TKn_QVF7qgw,2265
-pyobvector-0.2.8.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyobvector-0.2.8.dist-info/METADATA,sha256=PVTCENc27h0nh_UwNAF4M1ewnyHZlRGijs9wR3Gkv1o,6661
-pyobvector-0.2.8.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-pyobvector-0.2.8.dist-info/RECORD,,
+pyobvector-0.2.10.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyobvector-0.2.10.dist-info/METADATA,sha256=Wgm6wNdD8Nkbd_NW4AxoLG5kT0mqIGve6ehjLMJ4UWM,6659
+pyobvector-0.2.10.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+pyobvector-0.2.10.dist-info/RECORD,,