Fix: Before executing the SQL, remove tags in the format [ID: number] to avoid execution errors. (#9326)
### What problem does this PR solve? Before executing the SQL, remove tags in the format [ID: number] to avoid execution errors. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) Co-authored-by: wangyazhou <wangyazhou@sdibd.cn>
This commit is contained in:
parent
569ab011c4
commit
3ccaa06031
1 changed files with 2 additions and 1 deletions
|
|
@ -14,6 +14,7 @@
|
|||
# limitations under the License.
|
||||
#
|
||||
import os
|
||||
import re
|
||||
from abc import ABC
|
||||
import pandas as pd
|
||||
import pymysql
|
||||
|
|
@ -109,7 +110,7 @@ class ExeSQL(ToolBase, ABC):
|
|||
single_sql = single_sql.replace('```','')
|
||||
if not single_sql:
|
||||
continue
|
||||
|
||||
single_sql = re.sub(r"\[ID:[0-9]+\]", "", single_sql)
|
||||
cursor.execute(single_sql)
|
||||
if cursor.rowcount == 0:
|
||||
sql_res.append({"content": "No record in the database!"})
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue