diff --git a/dataherald/sql_generator/dataherald_finetuning_agent.py b/dataherald/sql_generator/dataherald_finetuning_agent.py index 6d7b3486..5fe0ed4d 100644 --- a/dataherald/sql_generator/dataherald_finetuning_agent.py +++ b/dataherald/sql_generator/dataherald_finetuning_agent.py @@ -251,6 +251,7 @@ class QuerySQLDataBaseTool(BaseSQLDatabaseTool, BaseTool): Input: A SQL query between ```sql and ``` tags. Output: Result from the database or an error message if the query is incorrect. Use this tool to execute the SQL query on the database, and return the results. + Add newline after both ```sql and ``` tags. """ args_schema: Type[BaseModel] = SQLInput diff --git a/dataherald/sql_generator/dataherald_sqlagent.py b/dataherald/sql_generator/dataherald_sqlagent.py index cc2a9e9b..3a1ecb28 100644 --- a/dataherald/sql_generator/dataherald_sqlagent.py +++ b/dataherald/sql_generator/dataherald_sqlagent.py @@ -154,6 +154,7 @@ class QuerySQLDataBaseTool(BaseSQLDatabaseTool, BaseTool): Output: Result from the database or an error message if the query is incorrect. If an error occurs, rewrite the query and retry. Use this tool to execute SQL queries. + Add newline after both ```sql and ``` tags. """ @catch_exceptions()