MaxKB/ui/src/views/document/component/ImportDocumentDialog.vue

<template>
  <el-dialog
    title="导入文档"
    v-model="dialogVisible"
    :close-on-click-modal="false"
    :close-on-press-escape="false"
    :destroy-on-close="true"
  >
    <el-form
      label-position="top"
      ref="webFormRef"
      :rules="rules"
      :model="form"
      require-asterisk-position="right"
    >
      <el-form-item label="文档地址" prop="source_url" v-if="isImport">
        <el-input
          v-model="form.source_url"
          placeholder="请输入文档地址，一行一个，地址不正确文档会导入失败。"
          :rows="10"
          type="textarea"
        />
      </el-form-item>
      <el-form-item
        v-else-if="!isImport && documentType === '1'"
        label="文档地址"
        prop="source_url"
      >
        <el-input v-model="form.source_url" placeholder="请输入文档地址" />
      </el-form-item>
      <el-form-item label="选择器" v-if="documentType === '1'">
        <el-input
          v-model="form.selector"
          placeholder="默认为 body，可输入 .classname/#idname/tagname"
        />
      </el-form-item>
      <el-form-item v-if="!isImport">
        <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-radio-group v-model="form.hit_handling_method" class="radio-block mt-4">
          <template v-for="(value, key) of hitHandlingMethod" :key="key">
            <el-radio :value="key">{{ value }} </el-radio>
          </template>
        </el-radio-group>
      </el-form-item>
      <el-form-item
        prop="directly_return_similarity"
        v-if="!isImport && form.hit_handling_method === 'directly_return'"
      >
        <div class="lighter w-full" style="margin-top: -20px">
          <span>相似度高于</span>
          <el-input-number
            v-model="form.directly_return_similarity"
            :min="0"
            :max="1"
            :precision="3"
            :step="0.1"
            controls-position="right"
            size="small"
            class="ml-4 mr-4"
          /><span>直接返回分段内容</span>
        </div>
      </el-form-item>
    </el-form>
    <template #footer>
      <span class="dialog-footer">
        <el-button @click.prevent="dialogVisible = false"> 取消 </el-button>
        <el-button type="primary" @click="submit(webFormRef)" :loading="loading"> 确定 </el-button>
      </span>
    </template>
  </el-dialog>
</template>
<script setup lang="ts">
import { ref, reactive, watch } from 'vue'
import { useRoute } from 'vue-router'
import type { FormInstance } from 'element-plus'
import documentApi from '@/api/document'
import { MsgSuccess } from '@/utils/message'
import { hitHandlingMethod } from '../utils'

const route = useRoute()
const {
  params: { id }
} = route as any

const emit = defineEmits(['refresh'])
const webFormRef = ref()
const loading = ref<boolean>(false)
const isImport = ref<boolean>(false)
const form = ref<any>({
  source_url: '',
  selector: '',
  hit_handling_method: 'optimization',
  directly_return_similarity: 0.9
})

// 文档设置
const documentId = ref('')
const documentType = ref<string | number>('') //文档类型：1: web文档；0:普通文档

// 批量设置
const documentList = ref<Array<string>>([])

const rules = reactive({
  source_url: [{ required: true, message: '请输入文档地址', trigger: 'blur' }],
  directly_return_similarity: [{ required: true, message: '请输入相似度', trigger: 'blur' }]
})

const dialogVisible = ref<boolean>(false)

watch(dialogVisible, (bool) => {
  if (!bool) {
    form.value = {
      source_url: '',
      selector: '',
      hit_handling_method: 'optimization',
      directly_return_similarity: 0.9
    }
    isImport.value = false
    documentType.value = ''
    documentId.value = ''
    documentList.value = []
  }
})

const open = (row: any, list: Array<string>) => {
  if (row) {
    documentType.value = row.type
    documentId.value = row.id
    form.value = {
      hit_handling_method: row.hit_handling_method,
      directly_return_similarity: row.directly_return_similarity,
      ...row.meta
    }
    isImport.value = false
  } else if (list) {
    // 批量设置
    documentList.value = list
  } else {
    // 导入 只有web文档类型
    documentType.value = '1'
    isImport.value = true
  }
  dialogVisible.value = true
}

const submit = async (formEl: FormInstance | undefined) => {
  if (!formEl) return
  await formEl.validate((valid) => {
    if (valid) {
      if (isImport.value) {
        const obj = {
          source_url_list: form.value.source_url.split('\n'),
          selector: form.value.selector
        }
        documentApi.postWebDocument(id, obj, loading).then(() => {
          MsgSuccess('导入成功')
          emit('refresh')
          dialogVisible.value = false
        })
      } else {
        if (documentId.value) {
          const obj = {
            hit_handling_method: form.value.hit_handling_method,
            directly_return_similarity: form.value.directly_return_similarity || 0.9,
            meta: {
              source_url: form.value.source_url,
              selector: form.value.selector
            }
          }
          documentApi.putDocument(id, documentId.value, obj, loading).then(() => {
            MsgSuccess('设置成功')
            emit('refresh')
            dialogVisible.value = false
          })
        } else if (documentList.value.length > 0) {
          // 批量设置
          const obj = {
            hit_handling_method: form.value.hit_handling_method,
            directly_return_similarity: form.value.directly_return_similarity || 0.9,
            id_list: documentList.value
          }
          documentApi.batchEditHitHandling(id, obj, loading).then(() => {
            MsgSuccess('设置成功')
            emit('refresh')
            dialogVisible.value = false
          })
        }
      }
    }
  })
}

defineExpose({ open })
</script>
<style lang="scss" scoped></style>