zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-06-29 15:31:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refact: update exesql notification (#14027)

Browse Source 
### What problem does this PR solve?

Refact: update exesql notification

### Type of change


- [x] Refactoring
This commit is contained in:
Magicbook1108
2026-04-10 13:42:57 +08:00
committed by GitHub
parent 87a87a7122
commit 9ce293a736
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
agent/tools/exesql.py
View File

@@ -255,6 +255,10 @@ class ExeSQL(ToolBase, ABC):
if not single_sql:
continue
single_sql = re.sub(r"\[ID:[0-9]+\]", "", single_sql)
if re.match(r"^(insert|update|delete)\b", single_sql, flags=re.IGNORECASE):
sql_res.append({"content": "For security reasons, INSERT, UPDATE, and DELETE statements are not supported."})
formalized_content.append("For security reasons, INSERT, UPDATE, and DELETE statements are not supported.")
continue
cursor.execute(single_sql)
if cursor.rowcount == 0:
sql_res.append({"content": "No record in the database!"})