gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix: Before executing the SQL, remove tags in the format [ID: number] to avoid execution errors. (#9326)

Browse source 
### 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:
goodboyasia 2025-08-12 12:42:28 +08:00 • committed by GitHub
parent 569ab011c4
commit 3ccaa06031
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
agent/tools/exesql.py
View file

@ -14,6 +14,7 @@
# limitations under the License. # limitations under the License.
# #
import os import os
import re
from abc import ABC from abc import ABC
import pandas as pd import pandas as pd
import pymysql import pymysql
@ -109,7 +110,7 @@ class ExeSQL(ToolBase, ABC):
single_sql = single_sql.replace('```','') single_sql = single_sql.replace('```','')
if not single_sql: if not single_sql:
continue continue
single_sql = re.sub(r"\[ID:[0-9]+\]", "", single_sql)
cursor.execute(single_sql) cursor.execute(single_sql)
if cursor.rowcount == 0: if cursor.rowcount == 0:
sql_res.append({"content": "No record in the database!"}) sql_res.append({"content": "No record in the database!"})