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

fix: clouse issue #2469 by fixing the type (#2533)

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong 2025-03-28 05:46:44 +08:00 committed by GitHub
parent 10619aaf96
commit 7c05bc1784
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
packages/dbgpt-app/src/dbgpt_app/scene/chat_db/professional_qa/chat.py
View File

@ -38,9 +38,7 @@ class ChatWithDbQA(BaseChat):
if self.database.is_graph_type(): if self.database.is_graph_type():
# When the current graph database retrieves source data from ChatDB, the # When the current graph database retrieves source data from ChatDB, the
# topk uses the sum of node table and edge table. # topk uses the sum of node table and edge table.
self.top_k = len(self.tables["vertex_tables"]) + len( self.top_k = len(list(self.tables))
self.tables["edge_tables"]
)
else: else:
logger.info(f"Dialect: {self.database.db_type}") logger.info(f"Dialect: {self.database.db_type}")
self.top_k = self.curr_config.schema_retrieve_top_k self.top_k = self.curr_config.schema_retrieve_top_k