fix bugs

Canner · Jul 1, 2024 · de803c7 · de803c7
1 parent f991782
commit de803c7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/wren-ai-service/src/web/v1/services/sql_explanation.py b/wren-ai-service/src/web/v1/services/sql_explanation.py
@@ -68,7 +68,7 @@ async def sql_explanation(self, sql_explanation_request: SQLExplanationRequest):
                 full_sql=sql_explanation_request.full_sql,
             )
 
-            sql_explanation_result = generation_result["post_processor"]["results"]
+            sql_explanation_result = generation_result["post_process"]["results"]
 
             logger.debug(f"sql explanation results: {sql_explanation_result}")
 

diff --git a/wren-ai-service/src/web/v1/services/sql_regeneration.py b/wren-ai-service/src/web/v1/services/sql_regeneration.py
@@ -102,7 +102,7 @@ async def sql_regeneration(
             )
 
             sql_regeneration_result = generation_result[
-                "description_regeneration_post_processor"
+                "description_regeneration_post_process"
             ]["results"]
 
             logger.debug(f"sql regeneration results: {sql_regeneration_result}")