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

WEB API independent

Browse Source
This commit is contained in:
yhjun1026 2023-07-13 10:19:33 +08:00
parent 3638f4848f
commit 73b3b7069f
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/scene/chat_dashboard/prompt.py
View File

@ -23,10 +23,11 @@ According to the characteristics of the analyzed data, choose the most suitable
{supported_chat_type} {supported_chat_type}
Pay attention to the length of the output content of the analysis result, do not exceed 4000tokens Pay attention to the length of the output content of the analysis result, do not exceed 4000tokens
Do not use unprovided fields and field value in data analysis SQL, Do not use column pay_status as a query condition in SQL.
According to the characteristics of the analyzed data, choose the best one from the charts provided below to display, use different types of charts as much as possible，chart types: According to the characteristics of the analyzed data, choose the best one from the charts provided below to display, use different types of charts as much as possible，chart types:
{supported_chat_type} {supported_chat_type}
Do not use unprovided fields and field value in data analysis SQL.
Give {dialect} data analysis SQL, analysis title, display method and analytical thinking,respond in the following json format: Give {dialect} data analysis SQL, analysis title, display method and analytical thinking,respond in the following json format:
{response} {response}
Ensure the response is correct json and can be parsed by Python json.loads Ensure the response is correct json and can be parsed by Python json.loads

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

@ -48,8 +48,8 @@ class ChatWithDbAutoExecute(BaseChat):
"input": self.current_user_input, "input": self.current_user_input,
"top_k": str(self.top_k), "top_k": str(self.top_k),
"dialect": self.database.dialect, "dialect": self.database.dialect,
"table_info": self.database.table_simple_info(self.db_connect) # "table_info": self.database.table_simple_info(self.db_connect)
# "table_info": client.get_similar_tables(dbname=self.db_name, query=self.current_user_input, topk=self.top_k) "table_info": client.get_similar_tables(dbname=self.db_name, query=self.current_user_input, topk=self.top_k)
} }
return input_values return input_values