From 56df51386e7a7ffe3f9912e6c7df3dae63859576 Mon Sep 17 00:00:00 2001 From: mohammadrezapourreza Date: Sat, 16 Mar 2024 11:41:06 -0600 Subject: [PATCH] DH5610/fix the intermediate steps --- dataherald/sql_generator/dataherald_finetuning_agent.py | 2 +- dataherald/sql_generator/dataherald_sqlagent.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dataherald/sql_generator/dataherald_finetuning_agent.py b/dataherald/sql_generator/dataherald_finetuning_agent.py index eb177268..53651031 100644 --- a/dataherald/sql_generator/dataherald_finetuning_agent.py +++ b/dataherald/sql_generator/dataherald_finetuning_agent.py @@ -585,7 +585,7 @@ def generate_response( sql_query = self.remove_markdown(result["output"]) else: sql_query = self.extract_query_from_intermediate_steps( - result["intermediate"] + result["intermediate_steps"] ) logger.info(f"cost: {str(cb.total_cost)} tokens: {str(cb.total_tokens)}") response.sql = replace_unprocessable_characters(sql_query) diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py index 4af5b3e9..09dccef7 100644 --- a/dataherald/sql_generator/dataherald_sqlagent.py +++ b/dataherald/sql_generator/dataherald_sqlagent.py @@ -730,7 +730,7 @@ def generate_response( sql_query = self.remove_markdown(result["output"]) else: sql_query = self.extract_query_from_intermediate_steps( - result["intermediate"] + result["intermediate_steps"] ) logger.info(f"cost: {str(cb.total_cost)} tokens: {str(cb.total_tokens)}") response.sql = replace_unprocessable_characters(sql_query)