github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-31 15:47:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:chat data summary

Browse Source
This commit is contained in:
aries_ckt 2023-08-29 19:37:48 +08:00
parent 54ce53326c
commit 0efaffc031
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pilot/scene/chat_db/auto_execute/chat.py
View File

@ -45,15 +45,15 @@ class ChatWithDbAutoExecute(BaseChat):
except ImportError:
raise ValueError("Could not import DBSummaryClient. ")
client = DBSummaryClient()
# try:
# table_infos = client.get_db_summary(
# dbname=self.db_name, query=self.current_user_input, topk=CFG.KNOWLEDGE_SEARCH_TOP_SIZE
# )
# except Exception as e:
# print("db summary find error!" + str(e))
# table_infos = self.database.table_simple_info()
#
table_infos = self.database.table_simple_info()
try:
table_infos = client.get_db_summary(
dbname=self.db_name, query=self.current_user_input, topk=CFG.KNOWLEDGE_SEARCH_TOP_SIZE
)
except Exception as e:
print("db summary find error!" + str(e))
table_infos = self.database.table_simple_info()
# table_infos = self.database.table_simple_info()
input_values = {
"input": self.current_user_input,