mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
263 lines
8.7 KiB
Vue
263 lines
8.7 KiB
Vue
<template>
|
|
<el-dialog
|
|
align-center
|
|
title="参数设置"
|
|
class="param-dialog"
|
|
v-model="dialogVisible"
|
|
style="width: 550px"
|
|
>
|
|
<div class="dialog-max-height">
|
|
<el-scrollbar always>
|
|
<div class="p-16">
|
|
<el-form label-position="top" ref="paramFormRef" :model="form">
|
|
<el-form-item label="检索模式">
|
|
<el-radio-group v-model="form.search_mode" class="card__radio" @change="changeHandle">
|
|
<el-card
|
|
shadow="never"
|
|
class="mb-16"
|
|
:class="form.search_mode === 'embedding' ? 'active' : ''"
|
|
>
|
|
<el-radio value="embedding" size="large">
|
|
<p class="mb-4">向量检索</p>
|
|
<el-text type="info">通过向量距离计算与用户问题最相似的文本分段</el-text>
|
|
</el-radio>
|
|
</el-card>
|
|
<el-card
|
|
shadow="never"
|
|
class="mb-16"
|
|
:class="form.search_mode === 'keywords' ? 'active' : ''"
|
|
>
|
|
<el-radio value="keywords" size="large">
|
|
<p class="mb-4">全文检索</p>
|
|
<el-text type="info">通过关键词检索，返回包含关键词最多的文本分段</el-text>
|
|
</el-radio>
|
|
</el-card>
|
|
<el-card shadow="never" :class="form.search_mode === 'blend' ? 'active' : ''">
|
|
<el-radio value="blend" size="large">
|
|
<p class="mb-4">混合检索</p>
|
|
<el-text type="info"
|
|
>同时执行全文检索和向量检索，再进行重排序，从两类查询结果中选择匹配用户问题的最佳结果</el-text
|
|
>
|
|
</el-radio>
|
|
</el-card>
|
|
</el-radio-group>
|
|
</el-form-item>
|
|
<el-row :gutter="20">
|
|
<el-col :span="12">
|
|
<el-form-item>
|
|
<template #label>
|
|
<div class="flex align-center">
|
|
<span class="mr-4">相似度高于</span>
|
|
<el-tooltip effect="dark" content="相似度越高相关性越强。" placement="right">
|
|
<AppIcon iconName="app-warning" class="app-warning-icon"></AppIcon>
|
|
</el-tooltip>
|
|
</div>
|
|
</template>
|
|
<el-input-number
|
|
v-model="form.similarity"
|
|
:min="0"
|
|
:max="form.search_mode === 'blend' ? 2 : 1"
|
|
:precision="3"
|
|
:step="0.1"
|
|
controls-position="right"
|
|
class="w-full"
|
|
/>
|
|
</el-form-item>
|
|
</el-col>
|
|
<el-col :span="12">
|
|
<el-form-item label="引用分段数 TOP">
|
|
<el-input-number
|
|
v-model="form.top_n"
|
|
:min="1"
|
|
:max="10"
|
|
controls-position="right"
|
|
class="w-full"
|
|
/>
|
|
</el-form-item>
|
|
</el-col>
|
|
</el-row>
|
|
|
|
<el-form-item label="最多引用字符数">
|
|
<el-slider
|
|
v-model="form.max_paragraph_char_number"
|
|
show-input
|
|
:show-input-controls="false"
|
|
:min="500"
|
|
:max="10000"
|
|
class="custom-slider"
|
|
/>
|
|
</el-form-item>
|
|
<el-form-item label="无引用知识库分段时">
|
|
<el-form
|
|
label-position="top"
|
|
ref="noReferencesformRef"
|
|
:model="noReferencesform"
|
|
:rules="noReferencesRules"
|
|
class="w-full"
|
|
:hide-required-asterisk="true"
|
|
>
|
|
<el-radio-group
|
|
v-model="form.no_references_setting.status"
|
|
class="radio-block mb-16"
|
|
>
|
|
<div>
|
|
<el-radio value="ai_questioning">
|
|
<p>继续向 AI 模型提问</p>
|
|
<el-form-item
|
|
v-if="form.no_references_setting.status === 'ai_questioning'"
|
|
label="提示词"
|
|
prop="ai_questioning"
|
|
>
|
|
<el-input
|
|
v-model="noReferencesform.ai_questioning"
|
|
:rows="2"
|
|
type="textarea"
|
|
maxlength="2048"
|
|
:placeholder="defaultValue['ai_questioning']"
|
|
/>
|
|
</el-form-item>
|
|
</el-radio>
|
|
</div>
|
|
<div class="mt-8">
|
|
<el-radio value="designated_answer">
|
|
<p>指定回答内容</p>
|
|
<el-form-item
|
|
v-if="form.no_references_setting.status === 'designated_answer'"
|
|
prop="designated_answer"
|
|
>
|
|
<el-input
|
|
v-model="noReferencesform.designated_answer"
|
|
:rows="2"
|
|
type="textarea"
|
|
maxlength="2048"
|
|
:placeholder="defaultValue['designated_answer']"
|
|
/>
|
|
</el-form-item>
|
|
</el-radio>
|
|
</div>
|
|
</el-radio-group>
|
|
</el-form>
|
|
</el-form-item>
|
|
</el-form>
|
|
</div>
|
|
</el-scrollbar>
|
|
</div>
|
|
<template #footer>
|
|
<span class="dialog-footer p-16">
|
|
<el-button @click.prevent="dialogVisible = false"> 取消 </el-button>
|
|
<el-button type="primary" @click="submit(noReferencesformRef)" :loading="loading">
|
|
保存
|
|
</el-button>
|
|
</span>
|
|
</template>
|
|
</el-dialog>
|
|
</template>
|
|
<script setup lang="ts">
|
|
import { ref, watch, reactive } from 'vue'
|
|
import { cloneDeep } from 'lodash'
|
|
import type { FormInstance, FormRules } from 'element-plus'
|
|
|
|
const emit = defineEmits(['refresh'])
|
|
|
|
const paramFormRef = ref()
|
|
const noReferencesformRef = ref()
|
|
|
|
const defaultValue = {
|
|
ai_questioning: '{question}',
|
|
designated_answer:
|
|
'你好，我是 MaxKB 小助手，我的知识库只包含了 MaxKB 产品相关知识，请重新描述您的问题。'
|
|
}
|
|
|
|
const form = ref<any>({
|
|
search_mode: 'embedding',
|
|
top_n: 3,
|
|
similarity: 0.6,
|
|
max_paragraph_char_number: 5000,
|
|
no_references_setting: {
|
|
status: 'ai_questioning',
|
|
value: '{question}'
|
|
}
|
|
})
|
|
|
|
const noReferencesform = ref<any>({
|
|
ai_questioning: defaultValue['ai_questioning'],
|
|
designated_answer: defaultValue['designated_answer']
|
|
})
|
|
|
|
const noReferencesRules = reactive<FormRules<any>>({
|
|
ai_questioning: [{ required: true, message: '请输入提示词', trigger: 'blur' }],
|
|
designated_answer: [{ required: true, message: '请输入内容', trigger: 'blur' }]
|
|
})
|
|
|
|
const dialogVisible = ref<boolean>(false)
|
|
const loading = ref(false)
|
|
|
|
watch(dialogVisible, (bool) => {
|
|
if (!bool) {
|
|
form.value = {
|
|
search_mode: 'embedding',
|
|
top_n: 3,
|
|
similarity: 0.6,
|
|
max_paragraph_char_number: 5000,
|
|
no_references_setting: {
|
|
status: 'ai_questioning',
|
|
value: ''
|
|
}
|
|
}
|
|
noReferencesform.value = {
|
|
ai_questioning: defaultValue['ai_questioning'],
|
|
designated_answer: defaultValue['designated_answer']
|
|
}
|
|
noReferencesformRef.value?.clearValidate()
|
|
}
|
|
})
|
|
|
|
const open = (data: any) => {
|
|
form.value = { ...form.value, ...cloneDeep(data) }
|
|
noReferencesform.value[form.value.no_references_setting.status] =
|
|
form.value.no_references_setting.value
|
|
dialogVisible.value = true
|
|
}
|
|
|
|
const submit = async (formEl: FormInstance | undefined) => {
|
|
if (!formEl) return
|
|
await formEl.validate((valid, fields) => {
|
|
if (valid) {
|
|
form.value.no_references_setting.value =
|
|
noReferencesform.value[form.value.no_references_setting.status]
|
|
emit('refresh', form.value)
|
|
dialogVisible.value = false
|
|
}
|
|
})
|
|
}
|
|
|
|
function changeHandle(val: string) {
|
|
if (val === 'keywords') {
|
|
form.value.similarity = 0
|
|
} else {
|
|
form.value.similarity = 0.6
|
|
}
|
|
}
|
|
|
|
defineExpose({ open })
|
|
</script>
|
|
<style lang="scss" scope>
|
|
.param-dialog {
|
|
padding: 8px 8px 24px 8px;
|
|
.el-dialog__header {
|
|
padding: 16px 16px 0 16px;
|
|
}
|
|
.el-dialog__body {
|
|
padding: 0 !important;
|
|
}
|
|
.dialog-max-height {
|
|
height: 550px;
|
|
}
|
|
.custom-slider {
|
|
.el-input-number.is-without-controls .el-input__wrapper {
|
|
padding: 0 !important;
|
|
}
|
|
}
|
|
}
|
|
</style>
|