mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-09 12:18:12 +00:00
use tool 'black' to check and reformated files
This commit is contained in:
parent
7fa205aefc
commit
b9570e395e
@ -1,4 +1,3 @@
|
|||||||
|
|
||||||
## 短期内在该文件中配置,长期考虑将会存储在默认的数据库中存储,并可以支持多种语言的配置
|
## 短期内在该文件中配置,长期考虑将会存储在默认的数据库中存储,并可以支持多种语言的配置
|
||||||
|
|
||||||
lang_dicts = {
|
lang_dicts = {
|
||||||
@ -32,7 +31,7 @@ lang_dicts = {
|
|||||||
"add_folder": "添加文件夹",
|
"add_folder": "添加文件夹",
|
||||||
"send": "发送",
|
"send": "发送",
|
||||||
"regenerate": "重新生成",
|
"regenerate": "重新生成",
|
||||||
"clear_box": "清理"
|
"clear_box": "清理",
|
||||||
},
|
},
|
||||||
"en": {
|
"en": {
|
||||||
"unique_id": "English Content",
|
"unique_id": "English Content",
|
||||||
@ -64,10 +63,10 @@ lang_dicts = {
|
|||||||
"add_folder": "Add folder",
|
"add_folder": "Add folder",
|
||||||
"send": "Send",
|
"send": "Send",
|
||||||
"regenerate": "Regenerate",
|
"regenerate": "Regenerate",
|
||||||
"clear_box": "Clear"
|
"clear_box": "Clear",
|
||||||
}
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
def get_lang_content(key, language="zh"):
|
def get_lang_content(key, language="zh"):
|
||||||
return lang_dicts.get(language, {}).get(key, "")
|
return lang_dicts.get(language, {}).get(key, "")
|
||||||
|
|
||||||
|
@ -1,12 +1,8 @@
|
|||||||
|
|
||||||
from pilot.configs.config import Config
|
from pilot.configs.config import Config
|
||||||
from pilot.language.lang_content_mapping import get_lang_content
|
from pilot.language.lang_content_mapping import get_lang_content
|
||||||
|
|
||||||
CFG = Config()
|
CFG = Config()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def get_lang_text(key):
|
def get_lang_text(key):
|
||||||
return get_lang_content(key, CFG.LANGUAGE)
|
return get_lang_content(key, CFG.LANGUAGE)
|
||||||
|
|
||||||
|
|
||||||
|
@ -89,9 +89,18 @@ DB_SETTINGS = {
|
|||||||
|
|
||||||
|
|
||||||
llm_native_dialogue = get_lang_text("knowledge_qa_type_llm_native_dialogue")
|
llm_native_dialogue = get_lang_text("knowledge_qa_type_llm_native_dialogue")
|
||||||
default_knowledge_base_dialogue = get_lang_text("knowledge_qa_type_default_knowledge_base_dialogue")
|
default_knowledge_base_dialogue = get_lang_text(
|
||||||
add_knowledge_base_dialogue = get_lang_text("knowledge_qa_type_add_knowledge_base_dialogue")
|
"knowledge_qa_type_default_knowledge_base_dialogue"
|
||||||
knowledge_qa_type_list = [llm_native_dialogue, default_knowledge_base_dialogue, add_knowledge_base_dialogue]
|
)
|
||||||
|
add_knowledge_base_dialogue = get_lang_text(
|
||||||
|
"knowledge_qa_type_add_knowledge_base_dialogue"
|
||||||
|
)
|
||||||
|
knowledge_qa_type_list = [
|
||||||
|
llm_native_dialogue,
|
||||||
|
default_knowledge_base_dialogue,
|
||||||
|
add_knowledge_base_dialogue,
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
def get_simlar(q):
|
def get_simlar(q):
|
||||||
docsearch = knownledge_tovec_st(os.path.join(DATASETS_DIR, "plan.md"))
|
docsearch = knownledge_tovec_st(os.path.join(DATASETS_DIR, "plan.md"))
|
||||||
@ -522,7 +531,9 @@ def build_single_model_ui():
|
|||||||
state = gr.State()
|
state = gr.State()
|
||||||
gr.Markdown(notice_markdown, elem_id="notice_markdown")
|
gr.Markdown(notice_markdown, elem_id="notice_markdown")
|
||||||
|
|
||||||
with gr.Accordion(get_lang_text("model_control_param"), open=False, visible=False) as parameter_row:
|
with gr.Accordion(
|
||||||
|
get_lang_text("model_control_param"), open=False, visible=False
|
||||||
|
) as parameter_row:
|
||||||
temperature = gr.Slider(
|
temperature = gr.Slider(
|
||||||
minimum=0.0,
|
minimum=0.0,
|
||||||
maximum=1.0,
|
maximum=1.0,
|
||||||
@ -554,19 +565,36 @@ def build_single_model_ui():
|
|||||||
show_label=True,
|
show_label=True,
|
||||||
).style(container=False)
|
).style(container=False)
|
||||||
|
|
||||||
sql_mode = gr.Radio([get_lang_text("sql_generate_mode_direct"), get_lang_text("sql_generate_mode_none")], show_label=False, value=get_lang_text("sql_generate_mode_none"))
|
sql_mode = gr.Radio(
|
||||||
|
[
|
||||||
|
get_lang_text("sql_generate_mode_direct"),
|
||||||
|
get_lang_text("sql_generate_mode_none"),
|
||||||
|
],
|
||||||
|
show_label=False,
|
||||||
|
value=get_lang_text("sql_generate_mode_none"),
|
||||||
|
)
|
||||||
sql_vs_setting = gr.Markdown(get_lang_text("sql_vs_setting"))
|
sql_vs_setting = gr.Markdown(get_lang_text("sql_vs_setting"))
|
||||||
sql_mode.change(fn=change_sql_mode, inputs=sql_mode, outputs=sql_vs_setting)
|
sql_mode.change(fn=change_sql_mode, inputs=sql_mode, outputs=sql_vs_setting)
|
||||||
|
|
||||||
tab_qa = gr.TabItem(get_lang_text("knowledge_qa"), elem_id="QA")
|
tab_qa = gr.TabItem(get_lang_text("knowledge_qa"), elem_id="QA")
|
||||||
with tab_qa:
|
with tab_qa:
|
||||||
mode = gr.Radio(
|
mode = gr.Radio(
|
||||||
[llm_native_dialogue, default_knowledge_base_dialogue, add_knowledge_base_dialogue], show_label=False, value=llm_native_dialogue
|
[
|
||||||
|
llm_native_dialogue,
|
||||||
|
default_knowledge_base_dialogue,
|
||||||
|
add_knowledge_base_dialogue,
|
||||||
|
],
|
||||||
|
show_label=False,
|
||||||
|
value=llm_native_dialogue,
|
||||||
|
)
|
||||||
|
vs_setting = gr.Accordion(
|
||||||
|
get_lang_text("configure_knowledge_base"), open=False
|
||||||
)
|
)
|
||||||
vs_setting = gr.Accordion(get_lang_text("configure_knowledge_base"), open=False)
|
|
||||||
mode.change(fn=change_mode, inputs=mode, outputs=vs_setting)
|
mode.change(fn=change_mode, inputs=mode, outputs=vs_setting)
|
||||||
with vs_setting:
|
with vs_setting:
|
||||||
vs_name = gr.Textbox(label=get_lang_text("new_klg_name"), lines=1, interactive=True)
|
vs_name = gr.Textbox(
|
||||||
|
label=get_lang_text("new_klg_name"), lines=1, interactive=True
|
||||||
|
)
|
||||||
vs_add = gr.Button(get_lang_text("add_as_new_klg"))
|
vs_add = gr.Button(get_lang_text("add_as_new_klg"))
|
||||||
with gr.Column() as doc2vec:
|
with gr.Column() as doc2vec:
|
||||||
gr.Markdown(get_lang_text("add_file_to_klg"))
|
gr.Markdown(get_lang_text("add_file_to_klg"))
|
||||||
@ -579,7 +607,9 @@ def build_single_model_ui():
|
|||||||
show_label=False,
|
show_label=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
load_file_button = gr.Button(get_lang_text("upload_and_load_to_klg"))
|
load_file_button = gr.Button(
|
||||||
|
get_lang_text("upload_and_load_to_klg")
|
||||||
|
)
|
||||||
with gr.Tab(get_lang_text("upload_folder")):
|
with gr.Tab(get_lang_text("upload_folder")):
|
||||||
folder_files = gr.File(
|
folder_files = gr.File(
|
||||||
label=get_lang_text("add_folder"),
|
label=get_lang_text("add_folder"),
|
||||||
@ -587,7 +617,9 @@ def build_single_model_ui():
|
|||||||
file_count="directory",
|
file_count="directory",
|
||||||
show_label=False,
|
show_label=False,
|
||||||
)
|
)
|
||||||
load_folder_button = gr.Button(get_lang_text("upload_and_load_to_klg"))
|
load_folder_button = gr.Button(
|
||||||
|
get_lang_text("upload_and_load_to_klg")
|
||||||
|
)
|
||||||
|
|
||||||
with gr.Blocks():
|
with gr.Blocks():
|
||||||
chatbot = grChatbot(elem_id="chatbot", visible=False).style(height=550)
|
chatbot = grChatbot(elem_id="chatbot", visible=False).style(height=550)
|
||||||
|
Loading…
Reference in New Issue
Block a user