From 3ccaa06031fdeda2580023c0706649357c1a9c17 Mon Sep 17 00:00:00 2001 From: goodboyasia <126063803+goodboyasia@users.noreply.github.com> Date: Tue, 12 Aug 2025 12:42:28 +0800 Subject: [PATCH] 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 --- agent/tools/exesql.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/agent/tools/exesql.py b/agent/tools/exesql.py index 63cd3d0be..ca03dfc21 100644 --- a/agent/tools/exesql.py +++ b/agent/tools/exesql.py @@ -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!"})