zhangxiao
2024-08-26 d75c2769026f982ac561f74e739a52ed5f041298
fix: 修改bug
4个文件已修改
76 ■■■■ 已修改文件
src/api/kbList.ts 7 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/components/navbar/index.vue 16 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/views/dmx/knowledgeLib/config.vue 29 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/views/dmx/knowledgeLib/index.vue 24 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/api/kbList.ts
@@ -117,6 +117,13 @@
  });
}
// 获取知识库管理配置嵌入模型
export function getKnowledgeBaseModel(params) {
  return axios.get('/api/v1/llm/my_llms', {
    params,
  });
}
// 测试接口
export function kbretrievalTest(params) {
  return axios.post('/api/v1/chunk/retrieval_test', params);
src/components/navbar/index.vue
@@ -185,7 +185,21 @@
          </template>
        </a-dropdown>
      </li>
      <li style="padding: 0"> {{ name }} </li>
      <li style="padding: 0">
        <a-dropdown trigger="click" @select="changeLocale as any">
          <div> {{ name }}</div>
          <template #content>
            <a-doption>
              <a-space @click="handleLogout">
                <icon-export />
                <span>
                  {{ $t('messageBox.logout') }}
                </span>
              </a-space>
            </a-doption>
          </template>
        </a-dropdown>
      </li>
    </ul>
  </div>
</template>
src/views/dmx/knowledgeLib/config.vue
@@ -35,7 +35,7 @@
<!--          </a-form-item>-->
          <a-form-item field="embd_id" label="嵌入模型">
            <a-select :size="'large'" v-model="form.embd_id" placeholder="请选择 ..." :disabled="kbObj.chunk_num && kbObj.token_num">
              <a-optgroup  :label="index" v-for="(item,index) in modelList" :key=index>
              <!-- <a-optgroup  :label="index" v-for="(item,index) in mobileData" :key=index>
                <a-option
                  v-for="(obj) in item"
                  :key='obj.fid'
@@ -43,7 +43,14 @@
                >
                  {{obj.llm_name}}
                </a-option>
              </a-optgroup>
              </a-optgroup> -->
              <a-option
                  v-for="(item,index) in mobileData"
                  :key='item.name'
                  :value="item.name"
                >
                  {{item.name}}
                </a-option>
            </a-select>
          </a-form-item>
          <a-form-item field="parser_id" label="解析方法">
@@ -152,7 +159,7 @@
<script lang="ts" setup>
import { onMounted, onBeforeMount, reactive, ref, nextTick } from "vue";
import { Message } from "@arco-design/web-vue";
import { queryModelList,kbUpdate } from "@/api/kbList";
import { queryModelList,kbUpdate,getKnowledgeBaseModel } from "@/api/kbList";
import message from "@arco-design/web-vue/es/message";
import useLoading from "@/hooks/loading";
const { loading,setLoading } = useLoading(true);
@@ -187,7 +194,7 @@
  description: "",
  language: "English",
  permission: "",
  embd_id: "BAAI/bge-large-zh-v1.5",
  embd_id: "",
  parser_id: '',
  layout_recognize: true
});
@@ -221,6 +228,7 @@
let visible = ref(false);
const formRef = ref(null);
const modelList = ref({});
const mobileData= ref<any>([])
const file = ref();
const uploadAction = '/api/v1/llm/upload'; // 替换为你的上传API
@@ -319,9 +327,20 @@
const queryModel = async (params) => {
  try {
    const data = await queryModelList(params);
    const data = await getKnowledgeBaseModel(params);
    // const res =await queryModelList(params)
    // console.log(data.data, '大模型列表');
    modelList.value = data.data
    for(let key in data.data) {
      if (data.data.hasOwnProperty(key)) { // 确保是对象自有属性,不是原型链上的属性
        if(data.data[key].llm.length > 0){
          mobileData.value= data.data[key].llm.filter(item => item.type == "embedding")
        }
      }
    }
  } catch (err) {
    // you can report use errorHandler or other
  } finally {
src/views/dmx/knowledgeLib/index.vue
@@ -34,6 +34,7 @@
                  <span
                    style="cursor: pointer; color: #4977ba; font-size: 12px"
                    @click="deleteKnowledge(tab.id)"
                    v-hasPermi="'/kb/delete'"
                    ><icon-delete style="font-size: 14px" />删除</span
                  >
                </template>
@@ -88,6 +89,7 @@
                                  type="text"
                                  class="button"
                                  style="color: #2a2a2b"
                                  v-hasPermi="'/kb/data_status'"
                                  @click="enableDisable(1)"
                                >
                                  <template #icon>
@@ -101,6 +103,7 @@
                                  type="text"
                                  class="button"
                                  style="color: #2a2a2b"
                                  v-hasPermi="'/kb/data_status'"
                                  @click="enableDisable(0)"
                                >
                                  <template #icon>
@@ -114,6 +117,7 @@
                                  type="text"
                                  class="button"
                                  style="color: #2a2a2b"
                                  v-hasPermi="'/kb/data_analysis'"
                                  @click="addBlock(1)"
                                >
                                  <template #icon>
@@ -127,6 +131,7 @@
                                  type="text"
                                  class="button"
                                  style="color: #2a2a2b"
                                  v-hasPermi="'/kb/data_analysis'"
                                  @click="addBlock(2)"
                                >
                                  <template #icon>
@@ -140,6 +145,7 @@
                                <a-button
                                  type="text"
                                  class="button"
                                  v-hasPermi="'/kb/data_delete'"
                                  style="color: #2a2a2b"
                                  @click="addBlock(3)"
                                >
@@ -469,7 +475,7 @@
                        type="text"
                        :disabled="record.run == '1'"
                        @click="handleClick(record)"
                        v-hasPermi="'/kb/data_update'"
                        查
                        size="small"
                      >
                        <template #icon>
@@ -513,8 +519,8 @@
                        </template>
                      </a-button>
                    </a-tooltip>
                    <!-- <a-button
                    <!--
                    <a-button
                      type="text"
                      size="small"
                      :disabled="record.run == '1'"
@@ -599,12 +605,12 @@
        <!--        </div>-->
      </div>
    </div>
<!--    <a-modal title=" " v-model:visible="visible" :footer="false" fullscreen>-->
<!--      &lt;!&ndash;      <docx  previewSrc="http://192.168.20.116:1080/v1/document/get/405c3efa4d8c11ef97560242ac120006"></docx>&ndash;&gt;-->
<!--      <docx v-if="documenttype == 'docx'" :previewSrc="previewSrc"></docx>-->
<!--      <excel v-if="documenttype == 'excel'" :previewSrc="previewSrc"></excel>-->
<!--      <txtPdf v-if="documenttype == 'txtPdf'" :previewSrc="previewSrc"></txtPdf>-->
<!--    </a-modal>-->
    <!--    <a-modal title=" " v-model:visible="visible" :footer="false" fullscreen>-->
    <!--      &lt;!&ndash;      <docx  previewSrc="http://192.168.20.116:1080/v1/document/get/405c3efa4d8c11ef97560242ac120006"></docx>&ndash;&gt;-->
    <!--      <docx v-if="documenttype == 'docx'" :previewSrc="previewSrc"></docx>-->
    <!--      <excel v-if="documenttype == 'excel'" :previewSrc="previewSrc"></excel>-->
    <!--      <txtPdf v-if="documenttype == 'txtPdf'" :previewSrc="previewSrc"></txtPdf>-->
    <!--    </a-modal>-->
    <tool
      ref="toolForm"
      :kbtenantInfo="kbtenantInfo"