Dh-4992/update prompt

Dataherald · Nov 16, 2023 · 52f527a · 52f527a
1 parent 8fc94a6
commit 52f527a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dataherald/sql_generator/dataherald_finetuning_agent.py b/dataherald/sql_generator/dataherald_finetuning_agent.py
@@ -373,8 +373,11 @@ def create_sql_agent(
         **kwargs: Dict[str, Any],
     ) -> AgentExecutor:
         tools = toolkit.get_tools()
+        admin_instructions = ""
+        for index, instruction in enumerate(toolkit.instructions):
+            admin_instructions += f"{index+1}) {instruction['instruction']}\n"
         prefix = prefix.format(
-            dialect=toolkit.dialect, admin_instructions=toolkit.instructions
+            dialect=toolkit.dialect, admin_instructions=admin_instructions
         )
         prompt = ZeroShotAgent.create_prompt(
             tools,