From febe85cd4b5180a59a9cb79ad688704fef92e9a5 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Fri, 28 Feb 2025 14:14:00 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=9F=A5=E8=AF=A2=E6=8F=90?=
 =?UTF-8?q?=E9=86=92?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 web/src/views/GraphView.vue | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/web/src/views/GraphView.vue b/web/src/views/GraphView.vue
index 2387d540..cde4a869 100644
--- a/web/src/views/GraphView.vue
+++ b/web/src/views/GraphView.vue
@@ -11,7 +11,7 @@
   <div class="graph-container layout-container" v-else>
     <HeaderComponent
       title="图数据库"
-      :description="`${graphInfo?.database_name || ''} - 共 ${graphInfo?.entity_count || 0} 实体，${graphInfo?.relationship_count || 0} 个关系。向量模型：${graphInfo?.embed_model_name || '未上传文件'}`"
+      :description="graphDescription"
     >
       <template #actions>
         <div class="status-wrapper">
@@ -32,6 +32,7 @@
         <a-button
           type="primary"
           :loading="state.searchLoading"
+          :disabled="state.searchLoading"
           @click="onSearch"
         >
           检索实体
@@ -204,6 +205,22 @@ const loadSampleNodes = () => {
 }
 
 const onSearch = () => {
+  if (state.searchLoading) {
+    message.error('请稍后再试')
+    return
+  }
+
+  if (graphInfo?.value?.embed_model_name !== cur_embed_model.value) {
+    if (!graphInfo?.value?.embed_model_name) {
+      message.error('请先上传文件(jsonl)')
+      return
+    }
+
+    if (!confirm(`构建图数据库时向量模型为 ${graphInfo?.value?.embed_model_name}，当前向量模型为 ${cur_embed_model.value}，是否继续查询？`)) {
+      return
+    }
+  }
+
   if (!state.searchInput) {
     message.error('请输入要查询的实体')
     return
@@ -315,6 +332,15 @@ const graphStatusText = computed(() => {
   return graphInfo.value?.status === 'open' ? '已连接' : '已关闭';
 });
 
+const graphDescription = computed(() => {
+  const dbName = graphInfo.value?.database_name || '';
+  const entityCount = graphInfo.value?.entity_count || 0;
+  const relationCount = graphInfo.value?.relationship_count || 0;
+  const modelName = graphInfo.value?.embed_model_name || '未上传文件';
+
+  return `${dbName} - 共 ${entityCount} 实体，${relationCount} 个关系。向量模型：${modelName}`;
+});
+
 </script>
 
 <style lang="less" scoped>