From 32cd7374d72a53dc177ce5326fc5e8326e0b113b Mon Sep 17 00:00:00 2001
From: liudong <liudong>
Date: 星期五, 09 八月 2024 17:40:31 +0800
Subject: [PATCH] 优化知识库管理
---
src/views/dmx/knowledgeLib/test.vue | 29 ++++++++++++++++++++++++-----
1 files changed, 24 insertions(+), 5 deletions(-)
diff --git a/src/views/dmx/knowledgeLib/test.vue b/src/views/dmx/knowledgeLib/test.vue
index 780a97d..b340e87 100644
--- a/src/views/dmx/knowledgeLib/test.vue
+++ b/src/views/dmx/knowledgeLib/test.vue
@@ -29,7 +29,7 @@
</a-select>
</a-space>
</a-form-item>
- <a-form-item field="slider" label="Top-K" :rules="[{type:'number', min:5,message:'slider is min than 5'}]">
+ <a-form-item field="slider" label="Top-K" v-if="form.rerank_id">
<a-slider v-model="form.top_k" :min="1" :max="2048" />
</a-form-item>
<a-form-item >
@@ -39,7 +39,7 @@
<a-textarea v-model="form.question" style="height: 10rem;border: 1px solid var(--color-fill-3);border-radius: 4px" placeholder="" allow-clear/>
</div>
<div class="main-container-form-item-extra-btn">
- <a-button type="primary" html-type="submit">娴嬭瘯</a-button>
+ <a-button :disabled="!form.question" :loading="loading" type="primary" html-type="submit">娴嬭瘯</a-button>
</div>
</div>
</a-form-item>
@@ -87,7 +87,7 @@
similarity_threshold: 0.2,
size: 10,
top_k: 1024,
- vector_similarity_weight: 0.3,
+ vector_similarity_weight: 0.7,
});
@@ -103,8 +103,26 @@
let formObj = {
...form,
kb_id:kbid.value,
+ vector_similarity_weight:(1 - form.vector_similarity_weight).toFixed(2),
}
- const data = await kbretrievalTest(formObj);
+ if(!formObj.rerank_id){
+ delete formObj.top_k;
+ delete formObj.rerank_id;
+ }
+ try {
+ loading.value = true;
+ const data = await kbretrievalTest(formObj);
+ if(data.code == 0){
+ Message.success('娴嬭瘯鎴愬姛');
+ }else{
+ Message.error('娴嬭瘯澶辫触');
+ }
+ loading.value = false;
+ }catch (err){
+ Message.error(err.message);
+ loading.value = false;
+ }
+
}
const handleClick = () => {
@@ -122,13 +140,14 @@
}
const changekbid = (value) => {
+ console.log(value, 'kbid鐨剉alue');
kbid.value = value;
}
const queryModel = async (params) => {
try {
const data = await queryModelList(params);
- console.log(data.data, '澶фā鍨嬪垪琛�');
+ // console.log(data.data, '澶фā鍨嬪垪琛�');
modelList.value = {
BAAI: [data.data.BAAI[1]],
Jina: data.data.Jina,
--
Gitblit v1.8.0