Pārlūkot izejas kodu

Add template DB Assistant and exesql sql column description (#2054)

### What problem does this PR solve?

### Type of change

- [x] New Feature (non-breaking change which adds functionality)

---------

Co-authored-by: Kevin Hu <kevinhu.sh@gmail.com>
tags/v0.10.0
H pirms 1 gada
vecāks
revīzija
21f2c5838b
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
2 mainītis faili ar 684 papildinājumiem un 3 dzēšanām
  1. 3
    3
      agent/component/exesql.py
  2. 681
    0
      agent/templates/DB Assistant.json

+ 3
- 3
agent/component/exesql.py Parādīt failu

@@ -82,9 +82,9 @@ class ExeSQL(ComponentBase, ABC):
continue
try:
query = db.execute_sql(single_sql)
sql_res.append(
{"content": "\n##Total: " + str(query.rowcount) + "\n" + pd.DataFrame(
[i for i in query.fetchmany(size=self._param.top_n)]).to_markdown()})
single_res = pd.DataFrame([i for i in query.fetchmany(size=self._param.top_n)])
single_res.columns = [i[0] for i in query.description]
sql_res.append({"content": "\nTotal: " + str(query.rowcount) + "\n" + single_res.to_markdown()})
except Exception as e:
sql_res.append({"content": "**Error**:" + str(e) + "\nError SQL Statement:" + single_sql})
pass

+ 681
- 0
agent/templates/DB Assistant.json
Failā izmaiņas netiks attēlotas, jo tās ir par lielu
Parādīt failu


Notiek ielāde…
Atcelt
Saglabāt