fix: 修改bug

zhangxiao

2024-08-30 c02bff2911f720d54388f4abb6456ec1777f8a96

 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的value');
    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;
    }
  }