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

feature:db_summary topk

Browse Source
This commit is contained in:
aries-ckt 2023-06-05 23:51:47 +08:00
parent 4b41842277
commit c752c9de08
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pilot/scene/chat_db/professional_qa/chat.py
View File

@ -35,7 +35,13 @@ class ChatWithDbQA(BaseChat):
self.database = CFG.local_db self.database = CFG.local_db
# 准备DB信息(拿到指定库的链接) # 准备DB信息(拿到指定库的链接)
self.db_connect = self.database.get_session(self.db_name) self.db_connect = self.database.get_session(self.db_name)
self.top_k: int = 5 self.tables = self.database.get_table_names()
self.top_k = (
CFG.KNOWLEDGE_SEARCH_TOP_SIZE
if len(self.tables) > CFG.KNOWLEDGE_SEARCH_TOP_SIZE
else len(self.tables)
)
def generate_input_values(self): def generate_input_values(self):
table_info = "" table_info = ""