modify file conversation using config language, and fix 'sql_schema_info' format

This commit is contained in:
xuyuan23 2023-05-28 21:06:03 +08:00
parent b9570e395e
commit d24c160cd8
2 changed files with 12 additions and 9 deletions

View File

@ -5,6 +5,7 @@ import dataclasses
import uuid
from enum import auto, Enum
from typing import List, Any
from pilot.language.translation_handler import get_lang_text
from pilot.configs.config import Config
@ -263,15 +264,15 @@ conv_qa_prompt_template = """ 基于以下已知的信息, 专业、简要的回
default_conversation = conv_one_shot
conversation_sql_mode = {
"auto_execute_ai_response": "直接执行结果",
"dont_execute_ai_response": "不直接执行结果",
"auto_execute_ai_response": get_lang_text("sql_generate_mode_direct"),
"dont_execute_ai_response": get_lang_text("sql_generate_mode_none"),
}
conversation_types = {
"native": "LLM原生对话",
"default_knownledge": "默认知识库对话",
"custome": "新增知识库对话",
"auto_execute_plugin": "对话使用插件",
"native": get_lang_text("knowledge_qa_type_llm_native_dialogue"),
"default_knownledge": get_lang_text("knowledge_qa_type_default_knowledge_base_dialogue"),
"custome": get_lang_text("knowledge_qa_type_add_knowledge_base_dialogue"),
"auto_execute_plugin": get_lang_text("dialogue_use_plugin"),
}
conv_templates = {

View File

@ -6,16 +6,17 @@ lang_dicts = {
"db_gpt_introduction": "[DB-GPT](https://github.com/csunny/DB-GPT) 是一个开源的以数据库为基础的GPT实验项目使用本地化的GPT大模型与您的数据和环境进行交互无数据泄露风险100% 私密100% 安全。",
"learn_more_markdown": "该服务是仅供非商业用途的研究预览。受 Vicuna-13B 模型 [License](https://github.com/facebookresearch/llama/blob/main/MODEL_CARD.md) 的约束",
"model_control_param": "模型参数",
"sql_generate_mode_direct": "直接生成结果",
"sql_generate_mode_none": "执行结果",
"sql_generate_mode_direct": "直接执行结果",
"sql_generate_mode_none": "直接执行结果",
"max_input_token_size": "最大输出Token数",
"please_choose_database": "请选择数据",
"sql_generate_diagnostics": "SQL生成与诊断",
"knowledge_qa_type_llm_native_dialogue": "LLM原生对话",
"knowledge_qa_type_default_knowledge_base_dialogue": "默认知识库对话",
"knowledge_qa_type_add_knowledge_base_dialogue": "新增知识库对话",
"dialogue_use_plugin": "对话使用插件",
"create_knowledge_base": "新建知识库",
"sql_schema_info": "数据库{dbname}的Schema信息如下: {message}\n",
"sql_schema_info": "数据库{}的Schema信息如下: {}\n",
"current_dialogue_mode": "当前对话模式",
"database_smart_assistant": "数据库智能助手",
"sql_vs_setting": "自动执行模式下, DB-GPT可以具备执行SQL、从网络读取知识自动化存储学习的能力",
@ -46,6 +47,7 @@ lang_dicts = {
"knowledge_qa_type_llm_native_dialogue": "LLM native dialogue",
"knowledge_qa_type_default_knowledge_base_dialogue": "Default Knowledge Base",
"knowledge_qa_type_add_knowledge_base_dialogue": "Added Knowledge Base",
"dialogue_use_plugin": "Dialogue Extension",
"create_knowledge_base": "Create Knowledge Base",
"sql_schema_info": "the schema information of database {}: {}\n",
"current_dialogue_mode": "Current dialogue mode",