kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

图的显示和上传优化

Browse Source
This commit is contained in:
Wenjie Zhang 2025-02-28 02:39:47 +08:00
parent 99c7ac659b
commit ce783a7e9b
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/config/__init__.py
View File

@ -87,6 +87,9 @@ class Config(SimpleConfig):
def handle_self(self): def handle_self(self):
self.model_names = MODEL_NAMES self.model_names = MODEL_NAMES
self.embed_model_names = EMBED_MODEL_INFO
self.reranker_names = RERANKER_LIST
model_provider_info = self.model_names.get(self.model_provider, {}) model_provider_info = self.model_names.get(self.model_provider, {})
self.model_dir = os.environ.get("MODEL_DIR", "") self.model_dir = os.environ.get("MODEL_DIR", "")

13
web/src/views/GraphView.vue
View File

@ -50,12 +50,13 @@
@ok="addDocumentByFile" @ok="addDocumentByFile"
@cancel="() => state.showModal = false" @cancel="() => state.showModal = false"
ok-text="添加到图数据库" cancel-text="取消" ok-text="添加到图数据库" cancel-text="取消"
:ok-button-props="{ disabled: disabled }"
:confirm-loading="state.precessing"> :confirm-loading="state.precessing">
<div v-if="graphInfo?.embed_model_name"> <div v-if="graphInfo?.embed_model_name">
<p>当前图数据库向量模型：{{ graphInfo?.embed_model_name }}</p> <p>当前图数据库向量模型：{{ graphInfo?.embed_model_name }}</p>
<p>当前所选择的向量模型是 {{ configStore.config.embed_model }}</p> <p>当前所选择的向量模型是 {{ cur_embed_model }}</p>
</div> </div>
<p v-else>第一次创建之后将无法修改向量模型，当前向量模型 {{ configStore.config.embed_model }}</p> <p v-else>第一次创建之后将无法修改向量模型，当前向量模型 {{ cur_embed_model }}</p>
<div class="upload"> <div class="upload">
<a-upload-dragger <a-upload-dragger
class="upload-dragger" class="upload-dragger"
@ -63,7 +64,7 @@
name="file" name="file"
:fileList="fileList" :fileList="fileList"
:max-count="1" :max-count="1"
:disabled="state.precessing || (graphInfo?.embed_model_name && graphInfo?.embed_model_name !== configStore.config.embed_model)" :disabled="disabled"
action="/api/data/upload" action="/api/data/upload"
@change="handleFileUpload" @change="handleFileUpload"
@drop="handleDrop" @drop="handleDrop"
@ -87,6 +88,12 @@ import { UploadOutlined } from '@ant-design/icons-vue';
import HeaderComponent from '@/components/HeaderComponent.vue'; import HeaderComponent from '@/components/HeaderComponent.vue';
const configStore = useConfigStore() const configStore = useConfigStore()
const cur_embed_model = computed(() => configStore.config.embed_model_names[configStore.config.embed_model].name)
const disabled = computed(() => {
if (state.precessing) return true
if (graphInfo?.value?.embed_model_name !== cur_embed_model.value) return true
return false
})
let graphInstance let graphInstance
const graphInfo = ref(null) const graphInfo = ref(null)