From 9fcfb98b0fa740c7a00d69acb32395d3d4511d00 Mon Sep 17 00:00:00 2001 From: mohammadrezapourreza Date: Thu, 16 Nov 2023 16:40:33 -0500 Subject: [PATCH] DH-4992/reformat with black --- dataherald/sql_generator/dataherald_finetuning_agent.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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,