v0.2.10: JSON TABLE: support alias in order by clause

Signed-off-by: shanhaikang.shk <shanhaikang.shk@oceanbase.com>

Author: GITHUBear

README.md

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

pyobvector/client/ob_vec_json_table_client.py

@@ -846,6 +846,7 @@ def _handle_jtable_dml_select(
             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 @@ def _handle_jtable_dml_select(
                     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 @@ def _handle_jtable_dml_select(
         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 @@ def _handle_jtable_dml_select(
             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)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyobvector"
-version = "0.2.9"
+version = "0.2.10"
 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"

tests/test_json_table.py

@@ -426,7 +426,7 @@ def test_online_cases(self):
         )
 
         res = self.client.perform_json_table_sql(
-            "SELECT field0, field1, field2 FROM `table_unit_test` WHERE DATE(field3)='2025-03-09' ORDER BY field2 DESC LIMIT 2"
+            "SELECT field0 AS 消费内容, `table_unit_test`.field1 AS 消费类型, field2 AS 消费金额 FROM `table_unit_test` WHERE DATE(`table_unit_test`.field3)='2025-03-09' ORDER BY 消费金额 DESC LIMIT 2"
         )
         # logger.info(get_all_rows(res))
         self.assertEqual(