yinbangzhong
2024-08-24 2627bea177b22bc803926bba5a29518bfb6ee0eb
src/views/sessionManager/index.vue
@@ -1,5 +1,5 @@
<template>
  <div class="container">
  <div class="container" id="container">
    <!--    <AddSession-->
    <!--      :modalObj="modalObj"-->
    <!--      @addSession="addSession"-->
@@ -23,14 +23,16 @@
                <!--                <span class="title">{{ agentTitle }}</span>-->
                <a-popover position="bottom" trigger="click">
                  <a-button border
                  <a-button border>
                    <span
                      style="
                        width: 100px;
                        overflow: hidden;
                        text-overflow: ellipsis;
                        white-space: nowrap;
                      "
                      >{{ from.name }}</span
                    >
                    <span style="
                       width: 100px;
                      overflow: hidden;
                      text-overflow: ellipsis;
                      white-space: nowrap;
                    ">{{ from.name }}</span>
                    <icon-down style="margin-left: 4px" />
                  </a-button>
                  <template #content>
@@ -55,12 +57,13 @@
            ref="scrollbar"
            id="home"
            class="chat-list"
            style="
              width: 80%;
              overflow: auto;
              height: calc(100vh - 380px);
              margin: 0px auto 20px;
            "
            style="width: 80%; overflow: auto; margin: 0px auto 20px"
            :style="{
              height:
                uploaditemList.length > 0
                  ? 'calc(100vh - 480px)'
                  : 'calc(100vh - 380px)',
            }"
          >
            <div
              class="chat-item"
@@ -89,25 +92,40 @@
                  />
                </template>
                <template #content>
                  <!--                  <a-card :class="{ chatItemAnswer: theme === 'light' }">-->
                  <!--                    <div-->
                  <!--                      :class="{ light: theme === 'light' }"-->
                  <!--                      v-html="sessionDetail.content.replace(/\n/g, '<br/>')"-->
                  <!--                    >-->
                  <!--                    </div>-->
                  <!--                  </a-card>-->
                  <a-textarea
                    readonly
                    auto-size
                    v-model="sessionDetail.content"
                    :class="{ chatItemAnswer: theme === 'light' }"
                    :style="{
                  <a-card v-if="isExistTip(sessionDetail.content)">
                    <div  :class="{ chatItemAnswer: theme === 'light' }"
                      :style="{backgroundColor: theme === 'light' ? '#ffffff' : '#000000'}" style="border: none;display: inline;"  v-for="(item,tipIndex) in tipMatch(sessionDetail)"
                    >
                      <span v-if="tipIndex == 0">
                        {{ sessionDetail.content.substring(0, item.index) }}
                        <a-tooltip background-color="#3491FA" :content="getTipContent(
                          messagenList.reference,
                          index
                        )">
                          <img style="width: 20px;height: 20px" :src="tipImage" />
                        </a-tooltip>
                      </span>
                      <span v-else>
                        {{ sessionDetail.content.substring(item.preIndex+item.item.length, item.index) }}
                        <a-tooltip background-color="#3491FA" :content="getTipContent(
                          messagenList.reference,
                          index
                        )">
                          <img style="width: 20px;height: 20px" :src="tipImage" />
                        </a-tooltip>
                      </span>
                    </div>
                  </a-card>
                  <a-card  v-else>
                    <div :class="{ chatItemAnswer: theme === 'light' }"
                         :style="{
                      backgroundColor:
                        theme === 'light' ? '#ffffff' : '#000000',
                    }"
                    style="border: none"
                  >
                  </a-textarea>
                    }" style="border: none">
                      {{ sessionDetail.content }}
                    </div>
                  </a-card>
                </template>
                <!-- <div>{{ sessionDetail.role === 'assistant' }}</div> -->
                <template #actions>
@@ -219,7 +237,11 @@
                v-model="inputMsg"
                @keydown.shift.enter="handleShiftEnter"
                @keydown.enter="sendMessage"
                placeholder="输入您想了解的内容,Shift+Enter换行,Enter发送"
                :placeholder="
                  uploaditemList.length > 0
                    ? '整理这些文件的核心内容'
                    : '输入您想了解的内容,Shift+Enter换行,Enter发送'
                "
                allow-clear
                show-word-limit
                :disabled="chatDis"
@@ -233,27 +255,18 @@
                  maxRows: 5,
                }"
              />
              <div style="width: 100%;display: flex;justify-content: space-between">
                <a-button
                  :disabled="onFileSelectedLoading"
                  @click="selectFile"
                  type="text"
                  style="border-radius: 24px"
                >
                  <icon-attachment
                    size="28"
                    style="color: #0960bd"
                  />
                  <input
                    ref="fileInput"
                    type="file"
                    style="display: none"
                    @change="onFileSelected"
                  />
                </a-button>
                  <span
                    style="cursor: pointer;margin-left: 20px;">
                </span>
              <div
                style="
                  width: 100%;
                  display: flex;
                  justify-content: space-between;
                "
              >
                <updataFile
                  ref="fileInput"
                  :sessionId="activeSessionId"
                  @selectFileCallback="selectFileCallback"
                ></updataFile>
                <a-button
                  :disabled="chatDis"
                  @click="sentClick"
@@ -264,62 +277,22 @@
                  <icon-send size="32" style="color: #0960bd" />
                </a-button>
              </div>
<!--              <div class="btn-send">-->
<!--                <a-button-->
<!--                  :disabled="chatDis"-->
<!--                  @click="sentClick"-->
<!--                  type="text"-->
<!--                  style="border-radius: 24px"-->
<!--                  :loading="loading"-->
<!--                >-->
<!--                  <icon-send size="32" style="color: #0960bd" />-->
<!--                </a-button>-->
<!--              </div>-->
            </div>
            <!--            <div style="margin-top: 0px">-->
            <!--              <a-upload-->
            <!--                ref="uploadRef"-->
            <!--                :file-list="uploadList"-->
            <!--                :limit="1"-->
            <!--                multiple-->
            <!--                :custom-request="customRequest"-->
            <!--                style="font-size: 24px;margin-bottom: 10px;position: relative;width: 200px">-->
            <!--                <template #upload-button>-->
            <!--                  <icon-attachment style="color: #0960bd;position: absolute;top:-50px;left: 20px;z-index: 10000"/>-->
            <!--                </template>-->
            <!--              </a-upload>-->
            <!--            </div>-->
<!--            <span-->
<!--              style="-->
<!--                position: absolute;-->
<!--                top: 94px;-->
<!--                left: 20px;-->
<!--                z-index: 999;-->
<!--                cursor: pointer;-->
<!--              "-->
<!--            >-->
<!--              <icon-attachment-->
<!--                size="28"-->
<!--                @click="selectFile"-->
<!--                style="color: #0960bd"-->
<!--              />-->
<!--              <input-->
<!--                ref="fileInput"-->
<!--                type="file"-->
<!--                style="display: none"-->
<!--                @change="onFileSelected"-->
<!--              />-->
<!--            </span>-->
            <div class="uploadFileList">
            <div class="uploadFileList" v-if="uploaditemList.length > 0">
              <div
                class="files"
                v-for="(item, index) in uploaditemList"
                :key="index"
                style="position: relative; width: 200px; margin-top: 10px"
                style="
                  position: relative;
                  width: 200px;
                  margin-top: 10px;
                  margin-right: 20px;
                "
              >
                <a-comment
                  :author="item.name"
                  :content="(item.size/1024).toFixed(2) + 'K'"
                  :content="item.size"
                  style="
                    background: var(--color-bg-2);
                    padding: 10px;
@@ -337,9 +310,10 @@
                      </template>
                    </a-spin>
                    <!--                    <a-button type="text" :loading="onFileSelectedLoading" v-if="onFileSelectedLoading"></a-button>-->
                    <a-avatar v-if="!onFileSelectedLoading">
                      <icon-file style="color: #0960bd" />
                    </a-avatar>
                    <!--                    <a-avatar v-if="!onFileSelectedLoading">-->
                    <!--                      <icon-file style="color: #0960bd" />-->
                    <!--                    </a-avatar>-->
                    <img :src="getIconByExtension(item.name)" alt="" />
                  </template>
                </a-comment>
                <icon-close-circle-fill
@@ -389,12 +363,7 @@
      :footer="false"
      title-align="start"
    >
      <a-form
        ref="formRef"
        :rules="rules"
        :model="from"
        @submit="handleSubmit"
      >
      <a-form ref="formRef" :rules="rules" :model="from" @submit="handleSubmit">
        <a-form-item field="name" label="名称">
          <a-input v-model="from.name" placeholder="请输入名称" />
        </a-form-item>
@@ -414,11 +383,6 @@
  </div>
</template>
<script setup lang="ts">
  import {
    IconClose,
    IconSearch,
    IconTiktokColor,
  } from '@arco-design/web-vue/es/icon';
  import { useAppStore, useUserStore } from '@/store';
  import {
    computed,
@@ -439,7 +403,7 @@
  import agentSession from '@/views/sessionManager/components/agentSession.vue';
  import historySession from '@/views/sessionManager/components/historySession.vue';
  import smartAi from '@/views/sessionManager/components/smartAi.vue';
  import setName from '@/views/sessionManager/components/setName.vue';
  import updataFile from '@/views/sessionManager/components/updataFile.vue';
  import EventBus from '@/utils/EventBus';
  import {
    addSessionApi,
@@ -458,10 +422,17 @@
  import pdfImg3 from '@/assets/session/execl.png';
  import pdfImg4 from '@/assets/session/icon-txt.png';
  import pdfImg5 from '@/assets/session/txt.png';
  import tipImage from '@/assets/session/tip.png';
  import { number } from "@intlify/core-base";
  // const url = ref('../../assets/session/PDF.png');
  const sessionDetailList = ref([]); //根据会话id出来的会话详情
  const sessionDetailList = ref([
    {
      content: '你好! 我是你的助理,有什么可以帮到你的吗?',
      role: 'assistant',
    },
  ]); //根据会话id出来的会话详情
  const messagenList = ref({});
  const sessionList = ref([]); //会话列表
  const modalObj = reactive({ add: false });
@@ -472,7 +443,7 @@
  const agentTitle = ref('未命名会话');
  let chatObj = reactive({});
  let from = reactive({
    name:'未命名会话',
    name: '未命名会话',
  });
  const isStopChat = ref(false);
  const currIndex = ref(0);
@@ -522,14 +493,31 @@
    }
  };
  let dataItem = [];
  const isExistTip=(message:string):boolean=>{
    if (/##[0-9]\$\$/.test(message)) {
      return true;
    } else {
      return false;
    }
  }
  const tipMatch = (session: any): any[] => {
    //match session.content中的##[0-9]$$的索引
    let indexs: any[] = [];
    let preTip=0;
    session.content.match(/##([0-9]+)\$\$/g)?.map((item) => {
      let i = session.content.indexOf(item);
      indexs.push({ index: i, item: item, preIndex: preTip });
      preTip = i;
    });
    return indexs;
  }
  const getTxt = (data, role, message, index) => {
    if (/##0\$\$/.test(message)) {
    if (isExistTip(message)) {
      if (role == 'assistant' && index) {
        data.forEach((item) => {
          if (Object.keys(item).length !== 0 && item?.doc_aggs.length > 0) {
            dataItem = item.doc_aggs;
          }
        });
        let i = index / 2 - 1 > 0 ? index / 2 - 1 : 0;
        dataItem = data[i].doc_aggs;
      }
    } else {
      dataItem = [];
@@ -538,9 +526,51 @@
    return dataItem;
  };
  const getTipContent = (data, index): string => {
    let maxSimilarityContent = "";
    let i = index / 2 - 1 > 0 ? index / 2 - 1 : 0;
    let maxSimilarity = 0;
    data[i].chunks.forEach((chunk) => {
        if (chunk.similarity > maxSimilarity) {
          maxSimilarity = chunk.similarity;
          maxSimilarityContent = chunk.content_with_weight;
        }
      }
    );
    return maxSimilarityContent;
  };
  const clickHref = (item) => {
    // return Message.warning('暂无法查看');
    window.open(`/api/v1/document/show/${item.doc_id}`, '_blank');
    // window.open(`/api/v1/document/get/${item.doc_id}`, '_blank');
    downloadFile({
      url: `/api/v1/document/get/${item.doc_id}`,
      filename: item.doc_name,
    });
  };
  const downloadFile = ({
    url,
    filename,
    target,
  }: {
    url: string;
    filename?: string;
    target?: string;
  }) => {
    const downloadElement = document.createElement('a');
    downloadElement.style.display = 'none';
    downloadElement.href = url;
    if (target) {
      downloadElement.target = '_blank';
    }
    downloadElement.rel = 'noopener noreferrer';
    if (filename) {
      downloadElement.download = filename;
    }
    document.body.appendChild(downloadElement);
    downloadElement.click();
    document.body.removeChild(downloadElement);
  };
  const getIconByExtension = computed(() => (extension) => {
@@ -578,74 +608,21 @@
    // formRef.value.resetFields();
  };
  const selectFile = () => {
    fileInput.value.click();
  const selectFileCallback = (data) => {
    console.log(data, 'selectFileCallback');
    uploaditemList.value = data;
  };
  let onFileSelectedLoading = ref(false);
  const onFileSelected = (event) => {
    const file = event.target.files[0];
    uploaditemList.value = [
      {
        name: file.name,
        size: file.size,
      },
    ];
    if (file) {
      onFileSelectedLoading.value = true;
      const formData = new FormData();
      formData.append('file', file);
      formData.append('conversation_id', activeSessionId.value);
      uploadWithoutKb(formData).then((res) => {
        // console.log(res);
        if (res.code == 200) {
          console.log(res);
          console.log(uploaditemList.value);
          onFileSelectedLoading.value = false;
          fileInput.value.value = '';
          uploaditemList.value = [];
          Message.success('上传成功');
        } else {
          Message.error('上传失败');
        }
      });
    }
  };
  const deleteFile = (item) => {
    console.log(uploaditemList.value);
    uploaditemList.value = [];
    uploaditemList.value.splice(item.index, 1);
  };
  const { toClipboard } = useClipboard();
  const copy = async (text) => {
    await toClipboard(text); //参数为要复制的文本
  };
  const onChange = (fileList) => {
    // files.value = fileList;
  };
  // 上传文件
  const customRequest = async (option) => {
    const { onProgress, onError, onSuccess, fileItem, name } = option;
    fileItem.status = 'ready';
    if (fileItem.file) {
      const formData = new FormData();
      formData.append('file', fileItem.file);
      formData.append('conversation_id', activeSessionId.value);
      uploadWithoutKb(formData).then((res) => {
        // console.log(res);
        if (res.code == 200) {
          console.log(res);
          console.log(uploadList.value);
          fileItem.status = 'done';
          // uploadList.value = [];
        } else {
          fileItem.status = 'error';
        }
      });
    }
  };
  const DialogList = async () => {
@@ -714,26 +691,6 @@
    event.preventDefault();
    inputMsg.value += '\n';
  };
  const dialogChange = (val) => {
    // 判断当前是智能体或agent
    // console.log(val, 'val');
    dialogId.value = val;
    dialogs.value.forEach((item) => {
      if (item.id === val) {
        Object.assign(dialogObj, item);
      }
    });
    console.log(dialogObj.type, 'dialogObj');
    if (dialogObj.type == 1) {
      agentType.value = '1';
      querySessionList();
    } else {
      agentType.value = '2';
      queryAgentSessionList();
    }
    // querySessionList();
  };
  // 发送
  const sentClick = () => {
@@ -754,21 +711,34 @@
          event.preventDefault(); // 阻止默认行为,即不换行
        }
        // if (!activeSessionId.value) {
        //   Message.warning("请选择会话");
        //   chatDis.value = false;
        //   loading.value = false;
        //   return;
        // }
        // if (displayedText.value) {
        //   querySessionList();
        // }
        if (inputMsg.value) {
          startChat(inputMsg.value);
          if (!activeSessionId.value) {
            //新建会话
            // 如果有会话id
            console.log(inputMsg.value, '新建会话名称');
          inputMsg.value = '';
            const res = await addSessionApi({
              dialog_id: '',
              conversation_desc: inputMsg.value,
            });
            // console.log(res, "res");
            if (res.code == 200) {
              // console.log(res.data.conversation_id);
              activeSessionId.value = res.data?.conversation_id;
              startChat(inputMsg.value);
              inputMsg.value = '';
              // queryNewSessionDetail(res.data?.conversation_id);
            } else {
              Message.error('创建会话失败,请重试');
            }
          } else {
            startChat(inputMsg.value);
            inputMsg.value = '';
          }
        } else {
          Message.warning('消息不能为空');
        }
@@ -970,14 +940,24 @@
    querySessionList();
  };
  onBeforeMount(() => {
    // DialogList();
    //新建会话
    createSession('');
    activeSessionId.value = '';
  });
  onMounted(() => {
    let container = document.getElementById('container');
    container.addEventListener('click', () => {
      fileInput.value.cancel();
    });
    EventBus.on('newChat', () => {
      agentType.value = '1';
      createSession('');
      // createSession('');
      activeSessionId.value = '';
      sessionDetailList.value = [
        {
          content: '你好! 我是你的助理,有什么可以帮到你的吗?',
          role: 'assistant',
        },
      ];
      from.name = '未命名会话';
    });
  });
  onBeforeUnmount(() => {
@@ -1336,4 +1316,19 @@
  :deep(.arco-upload-list-item-operation) {
    //display: none;
  }
  .uploadFileList {
    width: 100%;
    max-height: 140px;
    overflow-y: auto;
    padding: 10px;
    display: flex;
    flex-wrap: wrap;
    :deep(.arco-comment-author) {
      width: 100px;
      display: inline-block;
      overflow: hidden; /* 隐藏超出的内容 */
      text-overflow: ellipsis; /* 使用省略号来代替被隐藏的文字 */
      white-space: nowrap; /* 不换行,使内容在一行内显示 */
    }
  }
</style>