update kb config

2024-07-31 20:22:05 +08:00 · 2024-07-31 20:22:05 +08:00 · cb00a8f4a9
commit cb00a8f4a9
parent 1906c97782
15 changed files with 285 additions and 57 deletions
--- a/src/config/init.py
+++ b/src/config/init.py
@ -43,15 +43,14 @@ class Config(SimpleConfig):
        self.add_item("save_dir", default="saves", des="保存目录")
        # 功能选项
        self.add_item("enable_reranker", default=False, des="是否开启重排序")
-        self.add_item("enable_knowledge_base", default=True, des="是否开启知识库")
-        self.add_item("enable_knowledge_graph", default=True, des="是否开启知识图谱")
-        self.add_item("enable_search_engine", default=True, des="是否开启搜索引擎")
+        self.add_item("enable_knowledge_base", default=False, des="是否开启知识库")
+        self.add_item("enable_search_engine", default=False, des="是否开启搜索引擎")

        # 模型配置
        ## 注意这里是模型名，而不是具体的模型路径，默认使用 HuggingFace 的路径
        ## 如果需要自定义路径，则在 config/base.yaml 中配置 model_local_paths
-        self.add_item("model_provider", default="qianfan", des="模型提供商", choices=["qianfan", "vllm", "zhipu", "deepseek"])
-        self.add_item("model_name", default=None, des="模型名称，为空则表示使用默认值")
+        self.add_item("model_provider", default="qianfan", des="模型提供商", choices=["qianfan", "vllm", "zhipu", "deepseek", "dashscope"])
+        self.add_item("model_name", default=None, des="模型名称")
        self.add_item("embed_model", default="bge-large-zh-v1.5", des="Embedding 模型", choices=["bge-large-zh-v1.5", "zhipu"])
        self.add_item("reranker", default="bge-reranker-v2-m3", des="Re-Ranker 模型", choices=["bge-reranker-v2-m3"])
        self.add_item("model_local_paths", default={}, des="本地模型路径")
@ -80,6 +79,15 @@ class Config(SimpleConfig):
                if not model_rel_path.startswith("/"):
                    self.model_local_paths[model] = os.path.join(model_root_dir, model_rel_path)

+        self.model_names = MODEL_NAMES
+
+        if self.model_name not in self.model_names[self.model_provider]:
+            logger.warning(f"Model name {self.model_name} not in {self.model_provider}, using default model name")
+            self.model_name = self.model_names[self.model_provider][0]
+
+        default_model_name = self.model_names[self.model_provider][0]
+        self.model_name = self.get("model_name") or default_model_name
+
    def load(self):
        """根据传入的文件覆盖掉默认配置"""
        logger.info(f"Loading config from {self.filename}")
@ -121,4 +129,48 @@ class Config(SimpleConfig):
            with open(self.filename, 'w+') as f:
                json.dump(self, f, indent=4)

-        logger.info(f"Config file {self.filename} saved")
+        logger.info(f"Config file {self.filename} saved")
+
+MODEL_NAMES = {
+    # https://platform.deepseek.com/api-docs/zh-cn/pricing
+    "deepseek": [
+        "deepseek-chat",
+        "deepseek-coder"
+    ],
+
+    # https://open.bigmodel.cn/dev/api  glm-4-0520、glm-4 、glm-4-air、glm-4-airx、 glm-4-flash
+    "zhipu": [
+        "glm-4",
+        "glm-4-0520",
+        "glm-4-air",
+        "glm-4-airx",
+        "glm-4-flash"
+    ],
+
+    # {'ERNIE-4.0-8K-0104', 'ERNIE-Lite-8K-0308', 'ERNIE-Speed-128K', 'ERNIE-3.5-128K（预览版）', 'Yi-34B-Chat', 'ERNIE-4.0-8K-Preview-0518', 'ERNIE-Bot-4', 'ERNIE-3.5-128K', 'ChatGLM2-6B-32K', 'ERNIE-3.5-8K', 'EB-turbo-AppBuilder', 'ERNIE-Lite-AppBuilder-8K', 'ERNIE-4.0-8K-0329', 'AquilaChat-7B', 'Gemma-7B-it', 'Qianfan-Chinese-Llama-2-70B', 'Mixtral-8x7B-Instruct', 'Gemma-7B-It', 'ERNIE Speed-AppBuilder', 'ERNIE-Function-8K', 'ERNIE-4.0-8K-preview', 'ERNIE-Bot', 'Qianfan-BLOOMZ-7B-compressed', 'ERNIE-4.0-8K', 'BLOOMZ-7B', 'ERNIE-Character-8K', 'ERNIE-3.5-8K-0205', 'ERNIE-4.0-8K-0613', 'Llama-2-70B-Chat', 'ERNIE-Character-Fiction-8K', 'ERNIE-4.0-8K-Preview', 'ERNIE-3.5-8K-Preview', 'ERNIE-Speed', 'ERNIE-Tiny-8K', 'ERNIE-4.0-Turbo-8K-Preview', 'Meta-Llama-3-8B', 'ERNIE-4.0-8K-Latest', 'ERNIE 3.5', 'XuanYuan-70B-Chat-4bit', 'Llama-2-13B-Chat', 'ERNIE-Bot-turbo', 'ERNIE-3.5-8K-0613', 'ERNIE-Lite-AppBuilder-8K-0614', 'ERNIE-4.0-preview', 'Llama-2-7B-Chat', 'Qianfan-Chinese-Llama-2-13B', 'ERNIE-Bot-turbo-AI', 'Meta-Llama-3-70B', 'ERNIE-Functions-8K', 'ERNIE-Lite-8K-0922（原ERNIE-Bot-turbo-0922）', 'ERNIE Speed', 'ERNIE-3.5-preview', 'Qianfan-Chinese-Llama-2-7B', 'ERNIE-Speed-8K', 'ERNIE-Lite-8K-0922', 'ChatLaw', 'ERNIE-3.5-8K-0329', 'ERNIE-4.0-Turbo-8K', 'ERNIE-3.5-8K-preview', 'ERNIE-Lite-8K'}
+    "qianfan": [
+        "ERNIE-Speed",
+        "ERNIE-Speed-8K",
+        "ERNIE-Speed-128K",
+        "ERNIE-Tiny-8K",
+        "ERNIE-Lite-8K",
+        "ERNIE-4.0-8K-Latest"
+        "Yi-34B-Chat",
+    ],
+
+    "vllm": [
+        "vllm",
+    ],
+
+    # https://bailian.console.aliyun.com/?switchAgent=10226727&productCode=p_efm#/model-market
+    "dashscope": [
+        "qwen-long",
+        "qwen2-7b-instruct",
+        "qwen2-1.5b-instruct",
+        "llama3.1-8b-instruct",
+        "llama3-8b-instruct",
+        "llama3.1-405b-instruct",
+        "baichuan2-7b-chat-v1",
+        "qwen2-0.5b-instruct"
+    ]
+}
--- a/src/core/database.py
+++ b/src/core/database.py
@ -55,13 +55,14 @@ class DataBaseManager:
        self.config = config
        self.database_path = os.path.join(config.save_dir, "data", "database.json")
        self.embed_model = get_embedding_model(config)
-        self.knowledge_base = KnowledgeBase(config, self.embed_model)
-        self.graph_base = GraphDatabase(self.config, self.embed_model)
-        self.data = {"databases": [], "graph": {}}

-        if self.config.enable_knowledge_graph:
+        if self.config.enable_knowledge_base:
+            self.knowledge_base = KnowledgeBase(config, self.embed_model)
+            self.graph_base = GraphDatabase(self.config, self.embed_model)
            self.graph_base.start()

+        self.data = {"databases": [], "graph": {}}
+
        self._load_databases()
        self._update_database()

@ -103,11 +104,11 @@ class DataBaseManager:
        return {"databases": [db.to_dict() for db in self.data["databases"]]}

    def get_graph(self):
-        if self.config.enable_knowledge_graph:
+        if self.config.enable_graph_base:
            self.data["graph"].update(self.graph_base.get_database_info("neo4j"))
            return {"graph": self.data["graph"]}
        else:
-            return {"graph": {}, "message": "Graph database is not enabled"}
+            return {"message": "Graph base not enabled", "graph": {}}

    def create_database(self, database_name, description, db_type):
        new_database = DataBaseLite(database_name, description, db_type, embed_model=self.config.embed_model)
--- a/src/core/retriever.py
+++ b/src/core/retriever.py
@ -59,7 +59,7 @@ class Retriever:
        # res = model.predict("qiansdgsa, dasdh ashdsakjdk ak ").content

        results = []
-        if refs["meta"].get("use_graph"):
+        if refs["meta"].get("use_graph") and self.config.enable_knowledge_base:
            for entity in refs["entities"]:
                result = self.dbm.graph_base.query_by_vector(entity)
                if result != []:
@ -71,7 +71,8 @@ class Retriever:
        query = refs.get("rewritten_query", query)

        kb_res = []
-        if refs["meta"].get("db_name"):
+        final_res = []
+        if refs["meta"].get("db_name") and self.config.enable_knowledge_base:
            db_name = refs["meta"]["db_name"]
            kb = self.dbm.metaname2db[refs["meta"]["db_name"]]
            limit = refs["meta"].get("queryCount", 10)
--- a/src/core/startup.py
+++ b/src/core/startup.py
@ -9,6 +9,9 @@ logger = setup_logger("Startup")

 class Startup:
    def __init__(self):
+        self.start()
+
+    def start(self):
        self.config = Config()
        self.model = select_model(self.config)
        self.dbm = DataBaseManager(self.config)
@ -16,9 +19,7 @@ class Startup:

    def restart(self):
        logger.info("Restarting...")
-        self.model = select_model(self.config)
-        self.dbm = DataBaseManager(self.config)
-        self.retriever = Retriever(self.config, self.dbm, self.model)
+        self.start()
        logger.info("Restarted")


--- a/src/models/init.py
+++ b/src/models/init.py
@ -6,7 +6,7 @@ def select_model(config):
    model_provider = config.model_provider
    model_name = config.model_name

-    logger.info(f"Selecting model from {model_provider} with {model_name or 'default'}")
+    logger.info(f"Selecting model from {model_provider} with {model_name}")

    if model_provider == "deepseek":
        from src.models.chat_model import DeepSeek
@ -24,6 +24,10 @@ def select_model(config):
        from src.models.chat_model import VLLM
        return VLLM(model_name)

+    elif model_provider == "dashscope":
+        from src.models.chat_model import DashScope
+        return DashScope(model_name)
+
    elif model_provider is None:
        raise ValueError("Model provider not specified, please modify `model_provider` in `src/config/base.yaml`")
    else:
--- a/src/models/chat_model.py
+++ b/src/models/chat_model.py
@ -67,9 +67,10 @@ class VLLM(OpenAIBase):
 import qianfan


-class QianfanResponse:
+class GeneralResponse:
    def __init__(self, content):
        self.content = content
+        self.is_full = False


 class Qianfan:
@ -98,7 +99,7 @@ class Qianfan:
            stream=True,
        )
        for chunk in response:
-            yield QianfanResponse(chunk["body"]["result"])
+            yield GeneralResponse(chunk["body"]["result"])

    def _get_response(self, messages):
        response = self.client.do(
@ -106,6 +107,55 @@ class Qianfan:
            messages=messages,
            stream=False,
        )
-        return QianfanResponse(response["body"]["result"])
+        return GeneralResponse(response["body"]["result"])


+
+class DashScope:
+
+    def __init__(self, model_name="qwen-long") -> None:
+        self.model_name = model_name
+        self.api_key= os.getenv("DASHSCOPE_API_KEY")
+
+
+    def predict(self, message, stream=False):
+        if isinstance(message, str):
+            messages=[{"role": "user", "content": message}]
+        else:
+            messages = message
+
+        if stream:
+            return self._stream_response(messages)
+        else:
+            return self._get_response(messages)
+
+    def _stream_response(self, messages):
+        import dashscope
+        response = dashscope.Generation.call(
+            api_key=self.api_key,
+            model=self.model_name,
+            messages=messages,
+            result_format='message',
+            stream=True,
+        )
+        for chunk in response:
+            message = chunk.output.choices[0].message
+            message.is_full = True
+            yield chunk.output.choices[0].message
+
+    def _get_response(self, messages):
+        import dashscope
+        response = dashscope.Generation.call(
+            api_key=self.api_key,
+            model=self.model_name,
+            messages=messages,
+            result_format='message',
+            stream=False,
+        )
+        return response.output.choices[0].message
+
+
+if __name__ == "__main__":
+    model = DashScope()
+    for a in model.predict("你好", stream=True):
+        print(a.content)
--- a/src/models/embedding.py
+++ b/src/models/embedding.py
@ -72,6 +72,9 @@ class ZhipuEmbedding:


 def get_embedding_model(config):
+    if not config.enable_knowledge_base:
+        return None
+
    if config.embed_model == "zhipu":
        return ZhipuEmbedding(config)
    else:
--- a/src/views/common_view.py
+++ b/src/views/common_view.py
@ -41,14 +41,20 @@ def chat():
    def generate_response():
        content = ""
        for delta in startup.model.predict(messages, stream=True):
-            if delta.content:
+            if not delta.content:
+                continue
+
+            if hasattr(delta, 'is_full') and delta.is_full:
+                content = delta.content
+            else:
                content += delta.content
-                response_chunk = json.dumps({
-                    "history": history_manager.update_ai(content),
-                    "response": content,
-                    "refs": refs  # TODO: 优化 refs，不需要每次都返回
-                }, ensure_ascii=False).encode('utf8') + b'\n'
-                yield response_chunk
+
+            response_chunk = json.dumps({
+                "history": history_manager.update_ai(content),
+                "response": content,
+                "refs": refs  # TODO: 优化 refs，不需要每次都返回
+            }, ensure_ascii=False).encode('utf8') + b'\n'
+            yield response_chunk

    return Response(generate_response(), content_type='application/json', status=200)

@ -57,7 +63,7 @@ def call():
    request_data = json.loads(request.data)
    query = request_data['query']
    response = startup.model.predict(query)
-    logger.debug(f"Call query: {query} Response: {response.content}")
+    logger.debug(f"\n\n\nCall query: \n{query} \n\nResponse: \n{response.content}\n\n")

    return jsonify({
        "response": response.content,
--- a/src/views/database_view.py
+++ b/src/views/database_view.py
@ -14,7 +14,11 @@ progress = {} # 只针对单个用户的进度

@db.route('/', methods=['GET'])
 def get_databases():
-    database = startup.dbm.get_databases()
+    try:
+        database = startup.dbm.get_databases()
+    except Exception as e:
+        return jsonify({"message": "获取数据库列表失败", "databases": []})
+
    return jsonify(database)

@db.route('/', methods=['POST'])
--- a/web/src/assets/base.css
+++ b/web/src/assets/base.css
@ -18,7 +18,7 @@
  --c-text-light-1: var(--c-indigo);
  --c-text-light-2: rgba(60, 66, 70, 0.66);
  --c-text-dark-1: var(--c-white);
-  --c-text-dark-2: rgba(235, 235, 235, 0.64);
+  --c-text-dark-2: #b8b8b8;
 }

 /* semantic color variables for this project */
--- a/web/src/components/ChatComponent.vue
+++ b/web/src/components/ChatComponent.vue
@ -43,7 +43,7 @@
          <component :is="opts.showPanel ? FolderOpenOutlined : FolderOutlined" /> <span class="text">选项</span>
        </div>
        <div v-if="opts.showPanel" class="my-panal" ref="panel">
-          <div class="graphbase flex-center">
+          <div class="graphbase flex-center" v-if="configStore.config.enable_knowledge_base">
            知识库
            <div @click.stop>
              <a-dropdown>
@ -65,15 +65,21 @@
              </a-dropdown>
            </div>
          </div>
-          <div class="graphbase flex-center" @click="meta.use_graph = !meta.use_graph">
+          <div class="graphbase flex-center" @click="meta.use_graph = !meta.use_graph" v-if="configStore.config.enable_knowledge_base">
            图数据库 <div @click.stop><a-switch v-model:checked="meta.use_graph" /></div>
          </div>
-          <div class="graphbase flex-center" @click="meta.use_web = !meta.use_web">
+          <div class="graphbase flex-center" @click="meta.use_web = !meta.use_web" v-if="configStore.config.enable_search_engine">
            搜索引擎（Bing） <div @click.stop><a-switch v-model:checked="meta.use_web" /></div>
          </div>
-          <div class="graphbase flex-center" @click="meta.rewrite_query = !meta.rewrite_query">
+          <div class="graphbase flex-center" @click="meta.rewrite_query = !meta.rewrite_query" v-if="configStore.config.enable_reranker">
            重写查询 <div @click.stop><a-switch v-model:checked="meta.rewrite_query" /></div>
          </div>
+          <div class="graphbase flex-center" @click="meta.rewrite_query = !meta.rewrite_query">
+            流式输出 <div @click.stop><a-switch v-model:checked="meta.stream" /></div>
+          </div>
+          <div class="graphbase flex-center" @click="meta.summary_title = !meta.summary_title">
+            总结对话标题 <div @click.stop><a-switch v-model:checked="meta.summary_title" /></div>
+          </div>
        </div>
      </div>
    </div>
@ -150,7 +156,7 @@
          <template #icon> <SendOutlined v-if="!isStreaming" /> <LoadingOutlined v-else/> </template>
        </a-button>
      </div>
-      <p class="note">即便强如雅典娜也可能会出错，请注意辨别内容的可靠性 模型供应商：{{ configStore.config?.model_provider }}</p>
+      <p class="note">即便强如雅典娜也可能会出错，请注意辨别内容的可靠性 模型供应商：{{ configStore.config?.model_provider }}:{{ configStore.config?.model_name }}</p>
    </div>
  </div>
 </template>
@ -208,6 +214,8 @@ const meta = reactive(JSON.parse(localStorage.getItem('meta')) || {
  graph_name: "neo4j",
  rewrite_query: true,
  selectedKB: null,
+  stream: true,
+  summary_title: true,
 })

 // 更多选项可以在 marked 文档中找到：https://marked.js.org/
@ -241,13 +249,17 @@ const handleKeyDown = (e) => {
 }

 const renameTitle = () => {
-  const prompt = '请用一个很短的句子关于下面的对话内容的主题起一个名字，不要带标点符号：'
-  const firstUserMessage = conv.value.messages[0].text
-  const firstAiMessage = conv.value.messages[1].text
-  const context = `${prompt}\n\n问题: ${firstUserMessage}\n\n回复: ${firstAiMessage}，主题是（一句话）：`
-  simpleCall(context).then((data) => {
-    emit('rename-title', data.response.split("：")[0])
-  })
+  if (meta.summary_title) {
+    const prompt = '请用一个很短的句子关于下面的对话内容的主题起一个名字，不要带标点符号：'
+    const firstUserMessage = conv.value.messages[0].text
+    const firstAiMessage = conv.value.messages[1].text
+    const context = `${prompt}\n\n问题: ${firstUserMessage}\n\n回复: ${firstAiMessage}，主题是（一句话）：`
+    simpleCall(context).then((data) => {
+      emit('rename-title', data.response.split("：")[0])
+    })
+  } else {
+    emit('rename-title', conv.value.messages[0].text)
+  }
 }

 const scrollToBottom = () => {
@ -347,7 +359,7 @@ const sendMessage = () => {
  if (user_input) {
    isStreaming.value = true
    appendUserMessage(user_input)
-    appendAiMessage("检索中……", null)
+    appendAiMessage("···", null)
    const cur_res_id = conv.value.messages[conv.value.messages.length - 1].id
    conv.value.inputText = ''
    meta.db_name = opts.databases[meta.selectedKB]?.metaname
@ -441,12 +453,6 @@ watch(
  box-sizing: border-box;
  flex: 5 5 200px;
  overflow-y: scroll;
-  scrollbar-width: none; /* Firefox */
-  -ms-overflow-style: none;  /* IE and Edge */
-
-  &::-webkit-scrollbar {
-    display: none; /* Chrome, Safari, and Opera */
-  }

  .header {
    user-select: none;
@ -503,7 +509,7 @@ watch(
  box-shadow: 0px 0px 10px 1px rgba(0, 0, 0, 0.05);
  border-radius: 12px;
  padding: 12px;
-  z-index: 101;
+  z-index: 11;
  width: 250px;

  .flex-center {
@ -532,7 +538,7 @@ watch(
  position: absolute;
  top: 20%;
  width: 100%;
-  z-index: 100;
+  z-index: 9;

  h1 {
    margin-bottom: 20px;
@ -743,6 +749,34 @@ button:disabled {
  cursor: not-allowed;
 }

+
+
+.chat::-webkit-scrollbar {
+  position: absolute;
+  width: 4px;
+}
+
+.chat::-webkit-scrollbar-track {
+  background: transparent;
+  border-radius: 4px;
+}
+
+.chat::-webkit-scrollbar-thumb {
+  background: var(--c-text-dark-2);
+  border-radius: 4px;
+}
+
+.chat::-webkit-scrollbar-thumb:hover {
+  background: rgb(100, 100, 100);
+  border-radius: 4px;
+}
+
+.chat::-webkit-scrollbar-thumb:active {
+  background: rgb(68, 68, 68);
+  border-radius: 4px;
+}
+
+
@media (max-width: 520px) {
  .chat {
    height: calc(100vh - 60px);
--- a/web/src/layouts/AppLayout.vue
+++ b/web/src/layouts/AppLayout.vue
@ -25,6 +25,9 @@ const getRemoteConfig = () => {
 }

 const getRemoteDatabase = () => {
+  if (!configStore.config.enable_knowledge_base) {
+    return
+  }
  fetch('/api/database').then(res => res.json()).then(data => {
    console.log("database", data)
    databaseStore.setDatabase(data.databases)
--- a/web/src/views/ChatView.vue
+++ b/web/src/views/ChatView.vue
@ -206,6 +206,32 @@ onMounted(() => {
  }
 }

+.conversations::-webkit-scrollbar {
+  position: absolute;
+  width: 4px;
+}
+
+.conversations::-webkit-scrollbar-track {
+  background: transparent;
+  border-radius: 4px;
+}
+
+.conversations::-webkit-scrollbar-thumb {
+  background: var(--c-text-dark-2);
+  border-radius: 4px;
+}
+
+.conversations::-webkit-scrollbar-thumb:hover {
+  background: rgb(100, 100, 100);
+  border-radius: 4px;
+}
+
+.conversations::-webkit-scrollbar-thumb:active {
+  background: rgb(68, 68, 68);
+  border-radius: 4px;
+}
+
+
@media (max-width: 520px) {
  .conversations {
    position: absolute;
--- a/web/src/views/DataBaseView.vue
+++ b/web/src/views/DataBaseView.vue
@ -1,5 +1,5 @@
 <template>
-  <div class="database-container">
+  <div class="database-container" v-if="configStore.config.enable_knowledge_base">
    <h2>文档知识库</h2>
    <p>知识型数据库，主要是非结构化的文本组成，使用向量检索使用。</p>
    <a-modal :open="newDatabase.open" title="新建数据库" @ok="createDatabase">
@ -64,6 +64,15 @@
      </div>
    </div>
  </div>
+  <div class="database-empty" v-else>
+    <a-empty>
+      <template #description>
+        <span>
+          前往 <router-link to="/setting" style="color: var(--main-color); font-weight: bold;">设置</router-link> 页面配置知识库。
+        </span>
+      </template>
+    </a-empty>
+  </div>
 </template>

 <script setup>
@ -71,6 +80,7 @@ import { ref, onMounted, reactive, watch, h } from 'vue'
 import { useRouter, useRoute } from 'vue-router';
 import { message, Button } from 'ant-design-vue'
 import { ReadFilled, PlusOutlined, AppstoreFilled, LoadingOutlined } from '@ant-design/icons-vue'
+import { useConfigStore } from '@/stores/config';

 const route = useRoute()
 const router = useRouter()
@ -79,6 +89,7 @@ const graph = ref(null)
 const graphloading = ref(false)

 const indicator = h(LoadingOutlined, {spin: true});
+const configStore = useConfigStore()

 const newDatabase = reactive({
  name: '',
@ -258,4 +269,13 @@ onMounted(() => {
  filter: blur(2px);
 }

+
+.database-empty {
+  display: flex;
+  justify-content: center;
+  align-items: center;
+  height: 100%;
+  flex-direction: column;
+  color: var(--c-text-light-1);
+}
 </style>
--- a/web/src/views/SettingView.vue
+++ b/web/src/views/SettingView.vue
@ -21,6 +21,25 @@
            </a-select-option>
          </a-select>
        </div>
+        <div class="card">
+          <span class="label">
+            {{ items?.model_name.des }} &nbsp;
+            <a-button small v-if="needRestart.model_name" @click="sendRestart">
+              <ReloadOutlined />需要重启
+            </a-button>
+          </span>
+          <a-select ref="select" style="width: 160px"
+            :value="configStore.config?.model_name"
+            @change="handleChange('model_name', $event)"
+            v-if="configStore.config?.model_names && configStore.config?.model_provider && configStore.config?.model_names[configStore.config?.model_provider]"
+          >
+          <a-select-option
+            v-for="(name, idx) in configStore.config.model_names[configStore.config.model_provider]"
+            :key="idx"
+            :value="name">{{ name }}
+          </a-select-option>
+          </a-select>
+        </div>
        <div class="card">
          <span class="label">{{ items?.embed_model.des }} &nbsp;
            <a-button small v-if="needRestart.embed_model" @click="sendRestart">
@ -58,19 +77,23 @@
      <h3>功能配置</h3>
      <div class="section">
        <div class="card">
-          <span class="label">{{ items?.enable_knowledge_base.des }}</span>
+          <span class="label">{{ items?.enable_knowledge_base.des }}
+            <a-button small v-if="needRestart.enable_knowledge_base" @click="sendRestart">
+              <ReloadOutlined />需要重启
+            </a-button>
+          </span>
          <a-switch
            :checked="configStore.config.enable_knowledge_base"
            @change="handleChange('enable_knowledge_base', !configStore.config.enable_knowledge_base)"
          />
        </div>
-        <div class="card">
+        <!-- <div class="card">
          <span class="label">{{ items?.enable_knowledge_graph.des }}</span>
          <a-switch
            :checked="configStore.config.enable_knowledge_graph"
            @change="handleChange('enable_knowledge_graph', !configStore.config.enable_knowledge_graph)"
          />
-        </div>
+        </div> -->
        <div class="card">
          <span class="label">{{ items?.enable_search_engine.des }}</span>
          <a-switch
@ -126,7 +149,7 @@ const sendRestart = () => {
    message.success({ content: '重新加载完成!', key: "restart", duration: 2 });
    setTimeout(() => {
      window.location.reload()
-    }, 1000)
+    }, 200)
  })
 }
 </script>