diff --git a/dataherald/sql_generator/dataherald_finetuning_agent.py b/dataherald/sql_generator/dataherald_finetuning_agent.py index 596c3010..59e13679 100644 --- a/dataherald/sql_generator/dataherald_finetuning_agent.py +++ b/dataherald/sql_generator/dataherald_finetuning_agent.py @@ -373,7 +373,9 @@ def create_sql_agent( **kwargs: Dict[str, Any], ) -> AgentExecutor: tools = toolkit.get_tools() - prefix = prefix.format(dialect=toolkit.dialect, admin_instructions=toolkit.instructions) + prefix = prefix.format( + dialect=toolkit.dialect, admin_instructions=toolkit.instructions + ) prompt = ZeroShotAgent.create_prompt( tools, prefix=prefix,