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