|
|
|
@@ -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 |