From 9f191e739e7927f43ab0f002824e032ec2a57304 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Fri, 13 Sep 2024 21:05:46 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E7=9F=A5=E8=AF=86=E5=BA=93=E7=9A=84?=
 =?UTF-8?q?=E6=A3=80=E7=B4=A2=E8=AE=BE=E7=BD=AE=E9=87=8C=E9=9D=A2=E6=B2=A1?=
 =?UTF-8?q?=E6=9C=89=E8=AE=BE=E8=AE=A1=E9=9C=80=E8=A6=81=E5=90=91=E9=87=8F?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E5=8C=B9=E9=85=8D=E6=89=8D=E8=83=BD=E4=BD=BF?=
 =?UTF-8?q?=E7=94=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 web/src/views/DataBaseInfoView.vue | 34 +++++++++++++++++++-----------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/web/src/views/DataBaseInfoView.vue b/web/src/views/DataBaseInfoView.vue
index 553da669..b68a94e7 100644
--- a/web/src/views/DataBaseInfoView.vue
+++ b/web/src/views/DataBaseInfoView.vue
@@ -61,6 +61,7 @@
     <a-tab-pane key="add">
       <template #tab><span><CloudUploadOutlined />添加文件</span></template>
       <div class="db-info-container">
+        <a-alert v-if="database.embed_model != configStore.config.embed_model" message="向量模型不匹配，请重新选择" type="warning" />
         <!-- <h3>向知识库中添加文件</h3> -->
         <div class="upload">
           <a-upload-dragger
@@ -79,16 +80,18 @@
             </p>
           </a-upload-dragger>
         </div>
-        <a-button
-          type="primary"
-          @click="addDocumentByFile"
-          :loading="state.loading"
-          :disabled="fileList.length === 0"
-          style="margin: 0px 20px 20px 0;"
-        >
-          添加到知识库
-        </a-button>
-        <a-button @click="handleRefresh" :loading="state.refrashing">刷新状态</a-button>
+        <div class="actions">
+          <a-button
+            type="primary"
+            @click="addDocumentByFile"
+            :loading="state.loading"
+            :disabled="fileList.length === 0"
+            style="margin: 0px 20px 20px 0;"
+          >
+            添加到知识库
+          </a-button>
+          <a-button @click="handleRefresh" :loading="state.refrashing">刷新状态</a-button>
+        </div>
         <a-table :columns="columns" :data-source="database.files" row-key="filename" class="my-table">
           <template #bodyCell="{ column, text, record }">
             <template v-if="column.key === 'filename'">
@@ -136,6 +139,7 @@
     <a-tab-pane key="query-test" force-render>
       <template #tab><span><SearchOutlined />检索测试</span></template>
       <div class="db-info-container">
+        <a-alert v-if="database.embed_model != configStore.config.embed_model" message="向量模型不匹配，请重新选择" type="warning" />
         <!-- <h3>检索测试</h3> -->
         <div class="query-action">
           <a-textarea
@@ -254,6 +258,11 @@ const filterQueryResults = () => {
 }
 
 const onQuery = () => {
+  if (database.value.embed_model != configStore.config.embed_model) {
+    message.error('向量模型不匹配，请重新选择')
+    return
+  }
+
   console.log(queryText.value)
   state.searchLoading = true
   if (!queryText.value.trim()) {
@@ -614,6 +623,9 @@ onMounted(() => {
 
 .db-info-container {
   flex: 1 1 auto;
+  display: flex;
+  flex-direction: column;
+  gap: 12px;
 
   .query-action {
     display: flex;
@@ -714,8 +726,6 @@ onMounted(() => {
 }
 
 .upload {
-  margin-bottom: 20px;
-
   .upload-dragger {
     margin: 0px;
   }