From 31a48e3a3c7d4e37a05ef43fa4b65c84989cc353 Mon Sep 17 00:00:00 2001
From: yinbangzhong <zhongbangyin@126.com>
Date: 星期四, 29 八月 2024 09:26:25 +0800
Subject: [PATCH] select role
---
src/views/dmx/knowledgeLib/test.vue | 85 +++++++++++++++++++++++++-----------------
1 files changed, 51 insertions(+), 34 deletions(-)
diff --git a/src/views/dmx/knowledgeLib/test.vue b/src/views/dmx/knowledgeLib/test.vue
index 69df71d..0156658 100644
--- a/src/views/dmx/knowledgeLib/test.vue
+++ b/src/views/dmx/knowledgeLib/test.vue
@@ -15,22 +15,22 @@
@submit="handleSubmit"
>
<a-form-item field="similarity_threshold" label="鐩镐技搴﹂槇鍊�">
- <a-slider v-model="form.similarity_threshold" />
+ <a-slider
+ v-model="form.similarity_threshold"
+ :step="0.01"
+ :min="0"
+ :max="1"
+ />
</a-form-item>
- <a-form-item
- field="slider"
- label="鍏抽敭瀛楃浉浼煎害鏉冮噸"
- :rules="[
- { type: 'number', min: 5, message: 'slider is min than 5' },
- ]"
- >
- <a-slider v-model="form.vector_similarity_weight" />
+ <a-form-item field="slider" label="鍏抽敭瀛楃浉浼煎害鏉冮噸">
+ <a-slider
+ v-model="form.vector_similarity_weight"
+ :step="0.01"
+ :min="0"
+ :max="1"
+ />
</a-form-item>
- <a-form-item
- field="section"
- label="Rerank妯″瀷"
- :rules="[{ match: /section one/, message: '璇烽�夋嫨' }]"
- >
+ <a-form-item field="section" label="Rerank妯″瀷">
<a-space direction="vertical" size="large">
<a-select
:size="'large'"
@@ -56,13 +56,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>
@@ -81,7 +75,13 @@
/>
</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>
@@ -125,10 +125,10 @@
page: 1,
question: '',
rerank_id: '',
- similarity_threshold: 20,
+ similarity_threshold: 0.2,
size: 10,
top_k: 1024,
- vector_similarity_weight: 30,
+ vector_similarity_weight: 0.7,
});
const formatter = (value) => {
@@ -143,10 +143,27 @@
let formObj = {
...form,
kb_id: kbid.value,
+ vector_similarity_weight: parseFloat(
+ (1 - form.vector_similarity_weight).toFixed(2)
+ ),
};
- formObj.similarity_threshold = formObj.similarity_threshold / 100;
- formObj.vector_similarity_weight = formObj.vector_similarity_weight / 100;
- 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 = () => {
@@ -162,13 +179,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,
@@ -201,14 +219,13 @@
display: flex;
justify-content: space-between;
height: 100%;
- //background: #626aea;
&-lf {
width: 30%;
height: 100%;
- //border: 1px solid #cccccc;
- background: #ffffff;
+ //border: 1px solid var(--color-fill-3);
+ background: var(--color-bg-2);
border-radius: 10px;
- overflow: hidden;
+ overflow: auto;
&-top {
width: 100%;
height: 60px;
@@ -252,8 +269,8 @@
width: 69%;
height: 100%;
//background: #626aea;
- //border: 1px solid #cccccc;
- background: #ffffff;
+ //border: 1px solid var(--color-fill-3);
+ background: var(--color-bg-2);
border-radius: 10px;
}
}
--
Gitblit v1.8.0