Przeglądaj źródła

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>
tags/v0.20.2
goodboyasia 2 miesięcy temu
rodzic
commit
3ccaa06031
No account linked to committer's email address
1 zmienionych plików z 2 dodań i 1 usunięć
  1. 2
    1
      agent/tools/exesql.py

+ 2
- 1
agent/tools/exesql.py Wyświetl plik

@@ -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!"})

Ładowanie…
Anuluj
Zapisz