fix bugs

Canner · Jul 4, 2024 · ca81ddb · ca81ddb
1 parent c52f930
commit ca81ddb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/wren-ai-service/src/pipelines/sql_regeneration/generation.py b/wren-ai-service/src/pipelines/sql_regeneration/generation.py
@@ -92,9 +92,7 @@ def preprocess(
     steps: List[SQLExplanationWithUserCorrections],
     sql_regeneration_preprocesser: SQLRegenerationRreprocesser,
 ) -> dict[str, Any]:
-    logger.debug(
-        f"steps: {orjson.dumps(steps.model_dump(mode='json'), option=orjson.OPT_INDENT_2).decode()}"
-    )
+    logger.debug(f"steps: {steps}")
     logger.debug(f"description: {description}")
     return sql_regeneration_preprocesser.run(
         description=description,
@@ -107,6 +105,7 @@ def sql_regeneration_prompt(
     preprocess: Dict[str, Any],
     sql_regeneration_prompt_builder: PromptBuilder,
 ) -> dict:
+    logger.debug(f"preprocess: {preprocess}")
     return sql_regeneration_prompt_builder.run(results=preprocess["results"])