DH-5611 add newline on final answer from sql generation stream endpoi…

…nt for correct parsing of SQL markdown -- added markdown bold text (#450)
Dataherald · Apr 2, 2024 · 5f2c7a4 · 5f2c7a4
1 parent 8efb00f
commit 5f2c7a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dataherald/sql_generator/__init__.py b/dataherald/sql_generator/__init__.py
@@ -173,9 +173,9 @@ def stream_agent_steps(  # noqa: C901
                             queue.put(message.content + "\n")
                     elif "steps" in chunk:
                         for step in chunk["steps"]:
-                            queue.put(f"Observation: `{step.observation}`\n")
+                            queue.put(f"**Observation:**\n `{step.observation}`\n")
                     elif "output" in chunk:
-                        queue.put(f'Final Answer: {chunk["output"]}')
+                        queue.put(f'**Final Answer:**\n {chunk["output"]}')
                         if "```sql" in chunk["output"]:
                             response.sql = replace_unprocessable_characters(
                                 self.remove_markdown(chunk["output"])