feat: 优化全文检索和混合检索方式(#47)

This commit is contained in:
wangdan-fit2cloud 2024-04-25 17:57:01 +08:00
parent 84e8ca0f84
commit 8af8a4949c
3 changed files with 98 additions and 51 deletions

View File

@ -527,6 +527,9 @@ h4 {
.active {
border: 1px solid var(--el-color-primary);
}
.el-card__body {
padding: calc(var(--app-base-px) + 4px) calc(var(--app-base-px) * 2);
}
}
// AI模型选择添加模型hover样式

View File

@ -1,11 +1,17 @@
<template>
<el-dialog title="参数设置" class="param-dialog" v-model="dialogVisible" style="width: 550px">
<el-dialog
align-center
title="参数设置"
class="param-dialog"
v-model="dialogVisible"
style="width: 550px"
>
<div class="dialog-max-height">
<el-scrollbar>
<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">
<el-radio-group v-model="form.search_mode" class="card__radio" @change="changeHandle">
<el-card
shadow="never"
class="mb-16"
@ -26,11 +32,7 @@
<el-text type="info">通过关键词检索返回包含关键词最多的文本分段</el-text>
</el-radio>
</el-card>
<el-card
shadow="never"
class="mb-16"
:class="form.search_mode === 'blend' ? 'active' : ''"
>
<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"
@ -40,27 +42,41 @@
</el-card>
</el-radio-group>
</el-form-item>
<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="1"
:precision="3"
:step="0.1"
controls-position="right"
/>
</el-form-item>
<el-form-item label="引用分段数 TOP">
<el-input-number v-model="form.top_n" :min="1" :max="10" controls-position="right" />
</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="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"
@ -215,6 +231,14 @@ const submit = async (formEl: FormInstance | undefined) => {
})
}
function changeHandle(val: string) {
if (val === 'keywords') {
form.value.similarity = 0
} else {
form.value.similarity = 0.6
}
}
defineExpose({ open })
</script>
<style lang="scss" scope>
@ -227,7 +251,7 @@ defineExpose({ open })
padding: 0 !important;
}
.dialog-max-height {
height: calc(100vh - 260px);
height: calc(100vh - 180px);
}
.custom-slider {
.el-input-number.is-without-controls .el-input__wrapper {

View File

@ -84,7 +84,11 @@
</template>
<div class="mb-16">
<div class="title mb-8">检索模式</div>
<el-radio-group v-model="cloneForm.search_mode" class="card__radio">
<el-radio-group
v-model="cloneForm.search_mode"
class="card__radio"
@change="changeHandle"
>
<el-card
shadow="never"
class="mb-16"
@ -119,27 +123,35 @@
</el-card>
</el-radio-group>
</div>
<div class="mb-16">
<div class="title mb-8">相似度高于</div>
<el-input-number
v-model="cloneForm.similarity"
:min="0"
:max="1"
:precision="3"
:step="0.1"
controls-position="right"
/>
</div>
<div class="mb-16">
<div class="title mb-8">返回分段数 TOP</div>
<el-row :gutter="20">
<el-col :span="12">
<div class="mb-16">
<div class="title mb-8">相似度高于</div>
<el-input-number
v-model="cloneForm.similarity"
:min="0"
:max="1"
:precision="3"
:step="0.1"
controls-position="right"
class="w-full"
/>
</div>
</el-col>
<el-col :span="12">
<div class="mb-16">
<div class="title mb-8">返回分段数 TOP</div>
<el-input-number
v-model="cloneForm.top_number"
:min="1"
:max="10"
controls-position="right"
class="w-full"
/>
</div>
</el-col>
</el-row>
<el-input-number
v-model="cloneForm.top_number"
:min="1"
:max="10"
controls-position="right"
/>
</div>
<div class="text-right">
<el-button @click="popoverVisible = false">取消</el-button>
<el-button type="primary" @click="settingChange('close')">确认</el-button>
@ -218,6 +230,14 @@ const isDataset = computed(() => {
return meta?.activeMenu.includes('dataset')
})
function changeHandle(val: string) {
if (val === 'keywords') {
cloneForm.value.similarity = 0
} else {
cloneForm.value.similarity = 0.6
}
}
function settingChange(val: string) {
if (val === 'open') {
popoverVisible.value = true