Dh-5738/fixing the few-shot samples format (#470)

Dataherald · Apr 18, 2024 · 6a3df95 · 6a3df95
1 parent 15c2190
commit 6a3df95
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dataherald/sql_generator/__init__.py b/dataherald/sql_generator/__init__.py
@@ -188,7 +188,9 @@ def stream_agent_steps(  # noqa: C901
                             )
                     elif "steps" in chunk:
                         for step in chunk["steps"]:
-                            queue.put(f"\n**Observation:**\n {step.observation}\n")
+                            queue.put(
+                                f"\n**Observation:**\n {self.format_sql_query_intermediate_steps(step.observation)}\n"
+                            )
                     elif "output" in chunk:
                         queue.put(
                             f'\n**Final Answer:**\n {self.format_sql_query_intermediate_steps(chunk["output"])}'