Sfoglia il codice sorgente

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 mesi fa
parent
commit
3ccaa06031
Nessun account collegato all'indirizzo email del committer
1 ha cambiato i file con 2 aggiunte e 1 eliminazioni
  1. 2
    1
      agent/tools/exesql.py

+ 2
- 1
agent/tools/exesql.py Vedi File

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

Loading…
Annulla
Salva