pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 12:32:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
1 changed files with 2 additions and 1 deletions
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!"})