AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 13:58:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed code error when mssql returns multiple columns (#4420)

Browse Source 
Fixed code error when mssql returns multiple columns
This commit is contained in:
so95 2025-01-09 10:55:18 +07:00 committed by GitHub
parent bc681e2ee9
commit f86d8906e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
agent/component/exesql.py
View File

@ -112,8 +112,11 @@ class ExeSQL(Generate, ABC):
if cursor.rowcount == 0:
sql_res.append({"content": "\nTotal: 0\n No record in the database!"})
break
single_res = pd.DataFrame([i for i in cursor.fetchmany(self._param.top_n)])
single_res.columns = [i[0] for i in cursor.description]
if self._param.db_type == 'mssql':
single_res = pd.DataFrame.from_records(cursor.fetchmany(self._param.top_n),columns = [desc[0] for desc in cursor.description])
else:
single_res = pd.DataFrame([i for i in cursor.fetchmany(self._param.top_n)])
single_res.columns = [i[0] for i in cursor.description]
sql_res.append({"content": "\nTotal: " + str(cursor.rowcount) + "\n" + single_res.to_markdown()})
break
except Exception as e:
@ -143,4 +146,4 @@ class ExeSQL(Generate, ABC):
return None
def debug(self, **kwargs):
return self._run([], **kwargs)
return self._run([], **kwargs)