github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-03 01:12:15 +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

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

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