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

<template>
  <el-dialog
    :title="title"
    v-model="dialogVisible"
    :close-on-click-modal="false"
    :close-on-press-escape="false"
    :destroy-on-close="true"
    width="550"
  >
    <el-form
      label-position="top"
      ref="webFormRef"
      :rules="rules"
      :model="form"
      require-asterisk-position="right"
    >
      <el-form-item
        :label="$t('views.document.form.source_url.label')"
        prop="source_url"
        v-if="isImport"
      >
        <el-input
          v-model="form.source_url"
          :placeholder="$t('views.document.form.source_url.placeholder')"
          :rows="10"
          type="textarea"
        />
      </el-form-item>
      <el-form-item
        v-else-if="!isImport && documentType === '1'"
        :label="$t('views.document.form.source_url.label')"
        prop="source_url"
      >
        <el-input
          v-model="form.source_url"
          :placeholder="$t('views.document.form.source_url.requiredMessage')"
        />
      </el-form-item>
      <el-form-item :label="$t('views.document.form.selector.label')" v-if="documentType === '1'">
        <el-input
          v-model="form.selector"
          :placeholder="$t('views.document.form.selector.placeholder')"
        />
      </el-form-item>
      <el-form-item v-if="!isImport">
        <template #label>
          <div class="flex align-center">
            <span class="mr-4">{{ $t('views.document.form.hit_handling_method.label') }}</span>
            <el-tooltip
              effect="dark"
              :content="$t('views.document.form.hit_handling_method.tooltip')"
              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">{{ $t(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>{{ $t('views.document.form.similarity.label') }}</span>
          <el-input-number
            v-model="form.directly_return_similarity"
            :min="0"
            :max="1"
            :precision="3"
            :step="0.1"
            :value-on-clear="0"
            controls-position="right"
            size="small"
            class="ml-4 mr-4"
          /><span>{{ $t('views.document.form.similarity.placeholder') }}</span>
        </div>
      </el-form-item>
    </el-form>
    <template #footer>
      <span class="dialog-footer">
        <el-button @click.prevent="dialogVisible = false"> {{ $t('common.cancel') }} </el-button>
        <el-button type="primary" @click="submit(webFormRef)" :loading="loading">
          {{ $t('common.confirm') }}
        </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/knowledge/document'
import { MsgSuccess } from '@/utils/message'
// import { hitHandlingMethod } from '@/enums/document'
import { t } from '@/locales'
const route = useRoute()
const {
  params: { id }
} = route as any

const props = defineProps({
  title: String
})

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: t('views.document.form.source_url.requiredMessage'),
      trigger: 'blur'
    }
  ],
  directly_return_similarity: [
    {
      required: true,
      message: t('views.document.form.similarity.requiredMessage'),
      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(t('views.document.tip.importMessage'))
          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,
            meta: {
              source_url: form.value.source_url,
              selector: form.value.selector
            }
          }
          documentApi.putDocument(id, documentId.value, obj, loading).then(() => {
            MsgSuccess(t('common.settingSuccess'))
            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,
            id_list: documentList.value
          }
          documentApi.batchEditHitHandling(id, obj, loading).then(() => {
            MsgSuccess(t('common.settingSuccess'))
            emit('refresh')
            dialogVisible.value = false
          })
        }
      }
    }
  })
}

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