From 0efaffc031f0c7f1a6b2f39fe14ac64176f7bcc8 Mon Sep 17 00:00:00 2001 From: aries_ckt <916701291@qq.com> Date: Tue, 29 Aug 2023 19:37:48 +0800 Subject: [PATCH] fix:chat data summary --- pilot/scene/chat_db/auto_execute/chat.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/pilot/scene/chat_db/auto_execute/chat.py b/pilot/scene/chat_db/auto_execute/chat.py index 6d047e61e..d6c2b0536 100644 --- a/pilot/scene/chat_db/auto_execute/chat.py +++ b/pilot/scene/chat_db/auto_execute/chat.py @@ -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,