Update

Author: BenConstable9

text_2_sql/autogen/src/autogen_text_2_sql/custom_agents/parallel_query_solving_agent.py

@@ -88,7 +88,7 @@ async def on_messages_stream(
         self, messages: Sequence[ChatMessage], cancellation_token: CancellationToken
     ) -> AsyncGenerator[AgentEvent | Response, None]:
         last_response = messages[-1].content
-        parameter_input = messages[0].content
+        parameter_input = messages[-2].content
         try:
             injected_parameters = json.loads(parameter_input)["injected_parameters"]
         except json.JSONDecodeError:
@@ -137,9 +137,7 @@ async def consume_inner_messages_from_agentic_flow(
                                         identifier
                                     ].append(
                                         {
-                                            "sql_query": parsed_message[
-                                                "sql_query"
-                                            ].replace("\n", " "),
+                                            "sql_query": parsed_message["sql_query"],
                                             "sql_rows": parsed_message["sql_rows"],
                                         }
                                     )