pyobvector 0.2.9__py3-none-any.whl → 0.2.11__py3-none-any.whl

pyobvector/client/ob_vec_json_table_client.py

@@ -26,8 +26,8 @@ from ..json_table import (
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
 
-JSON_TABLE_META_TABLE_NAME = "_meta_json_t"
-JSON_TABLE_DATA_TABLE_NAME = "_data_json_t"
+JSON_TABLE_META_TABLE_NAME = "meta_json_t"
+JSON_TABLE_DATA_TABLE_NAME = "data_json_t"
 
 class ObVecJsonTableClient(ObVecClient):
     """OceanBase Vector Store Client with JSON Table."""
@@ -846,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
@@ -857,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
@@ -865,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
@@ -894,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.9.dist-info → pyobvector-0.2.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pyobvector
-Version: 0.2.9
+Version: 0.2.11
 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.9
+pip install pyobvector==0.2.11
 ```
 
 ## Build Doc

{pyobvector-0.2.9.dist-info → pyobvector-0.2.11.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=V0tljbkbfZI-chnfGTZAarJvOb84KkP6wnyEqLmk2RY,38726
+pyobvector/client/ob_vec_json_table_client.py,sha256=zUNP7PXVuUwohzM9rjtWks8sGVJVJ3GvyRKtoFDnhgI,39032
 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.9.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyobvector-0.2.9.dist-info/METADATA,sha256=5uhcbXDdN1anCq7Hl1aTF4aDtDBmY10UAfZXltZS3s8,6657
-pyobvector-0.2.9.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-pyobvector-0.2.9.dist-info/RECORD,,
+pyobvector-0.2.11.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+pyobvector-0.2.11.dist-info/METADATA,sha256=_wtAgSEx2uHG_ScDgZxKFKcfWs4QFihXOOLmRChHgIM,6659
+pyobvector-0.2.11.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+pyobvector-0.2.11.dist-info/RECORD,,