pass more infor to api handler

mindsdb/mindsdb#8721
mindsdb · Feb 1, 2024 · 3358152 · 3358152
1 parent a25debf
commit 3358152
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/mindsdb_sql/planner/query_planner.py b/mindsdb_sql/planner/query_planner.py
@@ -289,9 +289,10 @@ def plan_api_db_select(self, query):
         #     keep only limit and where
         #     the rest goes to outer select
         query2 = Select(
-            targets=[Star()],
+            targets=query.targets,
             from_table=query.from_table,
             where=query.where,
+            order_by=query.order_by,
             limit=query.limit,
         )
         prev_step = self.plan_integration_select(query2)

diff --git a/tests/test_planner/test_integration_select.py b/tests/test_planner/test_integration_select.py
@@ -524,7 +524,7 @@ def test_select_from_table_subselect_api_integration(self):
             steps=[
                 FetchDataframeStep(
                     integration='int1',
-                    query=parse_sql('select * from tab1'),
+                    query=parse_sql('select tab1.`id` AS `id` from tab1'),
                 ),
                 SubSelectStep(
                     dataframe=Result(0),
@@ -534,7 +534,7 @@ def test_select_from_table_subselect_api_integration(self):
                 FetchDataframeStep(
                     integration='int1',
                     query=Select(
-                        targets=[Star()],
+                        targets=[Identifier('tab2.x', alias=Identifier('x'))],
                         from_table=Identifier('tab2'),
                         where=BinaryOperation(
                             op='in',
@@ -597,7 +597,7 @@ def test_delete_from_table_subselect_api_integration(self):
             steps=[
                 FetchDataframeStep(
                     integration='int1',
-                    query=parse_sql('select * from tab1'),
+                    query=parse_sql('select tab1.`id` AS `id` from tab1'),
                 ),
                 SubSelectStep(
                     dataframe=Result(0),