From d24c160cd8aa421d6aaf18e9736058511d831c0a Mon Sep 17 00:00:00 2001 From: xuyuan23 <643854343@qq.com> Date: Sun, 28 May 2023 21:06:03 +0800 Subject: [PATCH] modify file conversation using config language, and fix 'sql_schema_info' format --- pilot/conversation.py | 13 +++++++------ pilot/language/lang_content_mapping.py | 8 +++++--- 2 files changed, 12 insertions(+), 9 deletions(-) diff --git a/pilot/conversation.py b/pilot/conversation.py index ba5ab2701..4d2673c1c 100644 --- a/pilot/conversation.py +++ b/pilot/conversation.py @@ -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 = { diff --git a/pilot/language/lang_content_mapping.py b/pilot/language/lang_content_mapping.py index 88cc92ffd..c6126e671 100644 --- a/pilot/language/lang_content_mapping.py +++ b/pilot/language/lang_content_mapping.py @@ -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",