From 307717b045d3e4dadd9934d52ddd78e8ffa356fb Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Wed, 5 Feb 2025 16:23:48 +0800
Subject: [PATCH] Fix exesql re-generate SQL issue. (#4717)

### What problem does this PR solve?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 agent/component/exesql.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/agent/component/exesql.py b/agent/component/exesql.py
index 26525612..6664d913 100644
--- a/agent/component/exesql.py
+++ b/agent/component/exesql.py
@@ -15,6 +15,8 @@
 #
 from abc import ABC
 import re
+from copy import deepcopy
+
 import pandas as pd
 import pymysql
 import psycopg2
@@ -22,6 +24,7 @@ from agent.component import GenerateParam, Generate
 import pyodbc
 import logging
 
+
 class ExeSQLParam(GenerateParam):
     """
     Define the ExeSQL component parameters.
@@ -137,7 +140,9 @@ class ExeSQL(Generate, ABC):
         ## Answer only the modified SQL statement. Please do not give any explanation, just answer the code.
 '''
         self._param.prompt=prompt
-        response = Generate._run(self, [], **kwargs)
+        kwargs_ = deepcopy(kwargs)
+        kwargs_["stream"] = False
+        response = Generate._run(self, [], **kwargs_)
         try:
             regenerated_sql = response.loc[0,"content"]
             return regenerated_sql