完成知识库管理增删改查，交互界面未完成
This commit is contained in:
junleea
parent
fafaf3205b
commit
b7efda9f2d
|
|
@ -0,0 +1,49 @@
|
||||||
|
import request from '@/utils/request2';
|
||||||
|
|
||||||
|
export const FindKnowledgeBaseService = (Data) => {
|
||||||
|
const params = new URLSearchParams();
|
||||||
|
for (let key in Data) {
|
||||||
|
params.append(key, Data[key])
|
||||||
|
}
|
||||||
|
return request.post('/knowledgeBase/list', params,{
|
||||||
|
headers: {
|
||||||
|
'token': Data.token, //token
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
export const AddKnowledgeBaseService = (Data) => {
|
||||||
|
const params = new URLSearchParams();
|
||||||
|
for (let key in Data) {
|
||||||
|
params.append(key, Data[key])
|
||||||
|
}
|
||||||
|
return request.post('/knowledgeBase/create', params,{
|
||||||
|
headers: {
|
||||||
|
'token': Data.token, //token
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
export const UpdateKnowledgeBaseService = (Data) => {
|
||||||
|
const params = new URLSearchParams();
|
||||||
|
for (let key in Data) {
|
||||||
|
params.append(key, Data[key])
|
||||||
|
}
|
||||||
|
return request.post('/knowledgeBase/update', params,{
|
||||||
|
headers: {
|
||||||
|
'token': Data.token, //token
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
export const DelKnowledgeBaseService = (Data) => {
|
||||||
|
const params = new URLSearchParams();
|
||||||
|
for (let key in Data) {
|
||||||
|
params.append(key, Data[key])
|
||||||
|
}
|
||||||
|
return request.post('/knowledgeBase/delete', params,{
|
||||||
|
headers: {
|
||||||
|
'token': Data.token, //token
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
@ -53,7 +53,12 @@ export const menuData: Menus[] = [
|
||||||
index: '/manage-file',
|
index: '/manage-file',
|
||||||
title: '文件管理',
|
title: '文件管理',
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
id: '58',
|
||||||
|
pid: '1',
|
||||||
|
index: '/manage-kbase',
|
||||||
|
title: '知识库管理',
|
||||||
|
}
|
||||||
],
|
],
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
|
|
@ -206,8 +211,8 @@ export const menuData: Menus[] = [
|
||||||
{
|
{
|
||||||
id: '9',
|
id: '9',
|
||||||
icon: 'Files',
|
icon: 'Files',
|
||||||
index: '/reco-topic',
|
index: '/kbase-chat',
|
||||||
title: '智能选题推荐',
|
title: '知识库交互',
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
id: '53',
|
id: '53',
|
||||||
|
|
|
||||||
|
|
@ -28,7 +28,7 @@ import { ElMessage } from 'element-plus';
|
||||||
const handle = (rawFile: any) => {
|
const handle = (rawFile: any) => {
|
||||||
console.log(rawFile);
|
console.log(rawFile);
|
||||||
};
|
};
|
||||||
const allowedTypes = ['doc', 'docx', 'pdf', 'txt', 'png', 'jpg', 'jpeg','md'];
|
const allowedTypes = ['doc', 'docx', 'pdf', 'txt', 'png', 'jpg', 'jpeg','md', "epub"];
|
||||||
|
|
||||||
interface UploadData {
|
interface UploadData {
|
||||||
upload_type: string;
|
upload_type: string;
|
||||||
|
|
@ -65,7 +65,7 @@ const headers = {
|
||||||
const fileExtension = file.name.split('.').pop().toLowerCase();
|
const fileExtension = file.name.split('.').pop().toLowerCase();
|
||||||
const isAllowedType = allowedTypes.includes(fileExtension);
|
const isAllowedType = allowedTypes.includes(fileExtension);
|
||||||
if (!isAllowedType) {
|
if (!isAllowedType) {
|
||||||
ElMessage.error('不允许的文件类型，仅支持 doc, docx, pdf, txt, png, jpg, jpeg, md 格式');
|
ElMessage.error('不允许的文件类型，仅支持 doc, docx, pdf, txt, png, jpg, jpeg, md, epub 格式');
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
// 可以在这里进行文件验证等操作
|
// 可以在这里进行文件验证等操作
|
||||||
|
|
|
||||||
|
|
@ -169,12 +169,20 @@ const routes: RouteRecordRaw[] = [
|
||||||
component: () => import(/* webpackChunkName: "ucenter" */ '../views/pages/ucenter.vue'),
|
component: () => import(/* webpackChunkName: "ucenter" */ '../views/pages/ucenter.vue'),
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
path: '/reco-topic',
|
path: '/kbase-chat',
|
||||||
name: 'reco-topic',
|
name: 'kbase-chat',
|
||||||
meta: {
|
meta: {
|
||||||
title: '智能选题推荐',
|
title: '智能选题推荐',
|
||||||
},
|
},
|
||||||
component: () => import(/* webpackChunkName: "reco-topic" */ '../views/system/reco-topic.vue'),
|
component: () => import(/* webpackChunkName: "reco-topic" */ '../views/system/kbase-chat.vue'),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
path: '/manage-kbase',
|
||||||
|
name: 'manage-kbase',
|
||||||
|
meta: {
|
||||||
|
title: '知识库管理',
|
||||||
|
},
|
||||||
|
component: () => import(/* webpackChunkName: "manage-kbase" */ '../views/system/manage-kbase.vue'),
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
path: '/editor',
|
path: '/editor',
|
||||||
|
|
|
||||||
|
|
@ -54,9 +54,10 @@ export const usePermissStore = defineStore("permiss", {
|
||||||
"55", //提示词生成
|
"55", //提示词生成
|
||||||
"56", //文件管理
|
"56", //文件管理
|
||||||
"57", //智能ppt生成
|
"57", //智能ppt生成
|
||||||
|
"58", //知识库管理
|
||||||
"71", //用户功能管理
|
"71", //用户功能管理
|
||||||
],
|
],
|
||||||
user: ["0", "8", "7", "9", "51" ,"53","55" ,"56", "57", "61", "71"],
|
user: ["0", "8", "7", "9", "51" ,"53","55" ,"56", "57", "58", "61", "71"],
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
},
|
},
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,13 @@
|
||||||
|
export interface KBase {
|
||||||
|
ID: number;
|
||||||
|
Name: string;
|
||||||
|
Description: string;
|
||||||
|
UUID: string;
|
||||||
|
SessionID: number;
|
||||||
|
CreatedAt: string;
|
||||||
|
UpdatedAt: string;
|
||||||
|
DeletedAt: string;
|
||||||
|
AuthID: number;
|
||||||
|
FileNameList: string;
|
||||||
|
FileIDs: string | Object | Array<any> | any;
|
||||||
|
}
|
||||||
|
|
@ -354,7 +354,7 @@ const doButtonD = () => {
|
||||||
}
|
}
|
||||||
|
|
||||||
// 获取代码类型
|
// 获取代码类型
|
||||||
const codeType = codeBlock.className.replace("hljs ", "");
|
const codeType = codeBlock.className.replace("hljs", "");
|
||||||
// 创建代码类型显示元素
|
// 创建代码类型显示元素
|
||||||
const codeTypeElement = document.createElement("span");
|
const codeTypeElement = document.createElement("span");
|
||||||
codeTypeElement.textContent = codeType.split("-")[1];
|
codeTypeElement.textContent = codeType.split("-")[1];
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,988 @@
|
||||||
|
<template>
|
||||||
|
<div class="chat-app">
|
||||||
|
<!-- 历史会话侧边栏 -->
|
||||||
|
<div class="history-sessions" v-if="sessionIsShow">
|
||||||
|
<!-- <div>
|
||||||
|
<el-button type="primary" @click="clearCurrent">新会话</el-button>
|
||||||
|
</div> -->
|
||||||
|
<el-card class="session-card">
|
||||||
|
<template #header>
|
||||||
|
<h3>当前知识库</h3>
|
||||||
|
</template>
|
||||||
|
<ul>
|
||||||
|
<li>{{ getShortenedName(sessionName) }}</li>
|
||||||
|
</ul>
|
||||||
|
</el-card>
|
||||||
|
<el-card class="session-card2">
|
||||||
|
<template #header>
|
||||||
|
<h3>知识库</h3>
|
||||||
|
</template>
|
||||||
|
<el-scrollbar style="height: 400px; scrollbar-width: none">
|
||||||
|
<ul>
|
||||||
|
<li
|
||||||
|
v-for="(session, index) in historySessions"
|
||||||
|
:key="index"
|
||||||
|
@click="loadSession(session.ID)"
|
||||||
|
>
|
||||||
|
{{ getShortenedName(session.Name) }}
|
||||||
|
</li>
|
||||||
|
</ul>
|
||||||
|
</el-scrollbar>
|
||||||
|
</el-card>
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<div @click="showSession" style="cursor: pointer">
|
||||||
|
<el-icon v-if="sessionIsShow">
|
||||||
|
<Expand />
|
||||||
|
</el-icon>
|
||||||
|
<el-icon v-else>
|
||||||
|
<Fold />
|
||||||
|
</el-icon>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 原有的聊天区域 -->
|
||||||
|
<div class="chat-container">
|
||||||
|
<!-- 消息列表 -->
|
||||||
|
<el-card
|
||||||
|
class="chat-messages"
|
||||||
|
shadow="never"
|
||||||
|
ref="messagesContainer"
|
||||||
|
v-if="messages.length > 0"
|
||||||
|
>
|
||||||
|
<div
|
||||||
|
v-for="(message, index) in messages"
|
||||||
|
:key="index"
|
||||||
|
:class="['message', message.role]"
|
||||||
|
>
|
||||||
|
<div class="message-avatar">
|
||||||
|
<span v-if="message.role === 'assistant'">💬</span>
|
||||||
|
<span v-else>👤</span>
|
||||||
|
</div>
|
||||||
|
<div class="message-content">
|
||||||
|
<div v-html="renderMarkdown(message.content)"></div>
|
||||||
|
<!-- 添加复制 -->
|
||||||
|
<div>
|
||||||
|
<el-button
|
||||||
|
type="text"
|
||||||
|
:icon="DocumentCopy"
|
||||||
|
@click="copyMessage(message.content)"
|
||||||
|
></el-button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div v-if="loading" class="loading-indicator">Loading...</div>
|
||||||
|
</el-card>
|
||||||
|
|
||||||
|
<!-- 输入区域 -->
|
||||||
|
<el-card class="chat-input" shadow="never">
|
||||||
|
<el-row :gutter="10">
|
||||||
|
<el-col :span="20">
|
||||||
|
<el-input
|
||||||
|
v-model="inputMessage"
|
||||||
|
type="textarea"
|
||||||
|
style="border: 0"
|
||||||
|
:rows="5"
|
||||||
|
placeholder="输入消息..."
|
||||||
|
@keyup.enter="sendMessage"
|
||||||
|
/>
|
||||||
|
<!-- <el-text
|
||||||
|
v-model="inputMessage"
|
||||||
|
aria-placeholder="输入信息...."
|
||||||
|
></el-text> -->
|
||||||
|
</el-col>
|
||||||
|
<el-col :span="4" style="text-align: center">
|
||||||
|
<el-button
|
||||||
|
@click="sendMessage"
|
||||||
|
type="success"
|
||||||
|
:icon="Check"
|
||||||
|
round
|
||||||
|
:disabled="loading"
|
||||||
|
>发送</el-button
|
||||||
|
>
|
||||||
|
<div>
|
||||||
|
<p>模型参数</p>
|
||||||
|
<el-slider v-model="temperature" :min="0" :max="1" :step="0.1"
|
||||||
|
>temperature</el-slider
|
||||||
|
>
|
||||||
|
<el-slider
|
||||||
|
v-model="topP"
|
||||||
|
:min="0"
|
||||||
|
:max="1"
|
||||||
|
:step="0.1"
|
||||||
|
></el-slider>
|
||||||
|
</div>
|
||||||
|
</el-col>
|
||||||
|
<el-col :span="3" style="text-align: center">
|
||||||
|
<el-select v-model="selectModel" placeholder="选择模型">
|
||||||
|
<el-option
|
||||||
|
v-for="item in ModelList"
|
||||||
|
:key="item.ID"
|
||||||
|
:label="item.Type + ':' + item.Description"
|
||||||
|
:value="item.ID"
|
||||||
|
></el-option>
|
||||||
|
</el-select>
|
||||||
|
</el-col>
|
||||||
|
<el-col :span="1" style="text-align: center" v-if="false">
|
||||||
|
<el-button @click="handleSelectFileVisible"
|
||||||
|
><el-icon><Files /></el-icon
|
||||||
|
></el-button>
|
||||||
|
</el-col>
|
||||||
|
<!-- <el-col :span="1" style="text-align: center">
|
||||||
|
<el-button @click="handleUploadPicture"><el-icon><Picture /></el-icon></el-button>
|
||||||
|
</el-col>
|
||||||
|
<el-col :span="1" style="text-align: center">
|
||||||
|
<el-button><el-icon><VideoCamera /></el-icon></el-button>
|
||||||
|
</el-col> -->
|
||||||
|
<!-- 已选文件一行显示 -->
|
||||||
|
<el-col :span="12" style="text-align: center">
|
||||||
|
<el-tag
|
||||||
|
v-for="(file, index) in selectedFiles"
|
||||||
|
:key="index"
|
||||||
|
closable
|
||||||
|
@close="removeFile(index)"
|
||||||
|
>{{ file.UserFileName }}</el-tag
|
||||||
|
>
|
||||||
|
</el-col>
|
||||||
|
</el-row>
|
||||||
|
</el-card>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div>
|
||||||
|
<!-- 文件对话框 -->
|
||||||
|
<el-dialog
|
||||||
|
v-model="selectFileVisible"
|
||||||
|
title="从上传文件中选择"
|
||||||
|
width="50%"
|
||||||
|
>
|
||||||
|
<el-input
|
||||||
|
placeholder="搜索文件"
|
||||||
|
v-model="searchFileQuery"
|
||||||
|
prefix-icon="el-icon-search"
|
||||||
|
/>
|
||||||
|
<el-button @click="uploadMessageFile">上传文件</el-button>
|
||||||
|
<!-- 文件列表 -->
|
||||||
|
<div class="file-list">
|
||||||
|
<el-checkbox-group v-model="selectedFiles">
|
||||||
|
<el-checkbox
|
||||||
|
v-for="(item, index) in filteredFiles"
|
||||||
|
:key="index"
|
||||||
|
:label="item"
|
||||||
|
>
|
||||||
|
<span class="file-icon">
|
||||||
|
<!-- 根据文件类型展示不同图标 -->
|
||||||
|
<i
|
||||||
|
v-if="item.UploadType === 'image'"
|
||||||
|
class="el-icon-picture"
|
||||||
|
></i>
|
||||||
|
<i
|
||||||
|
v-else-if="item.UploadType === 'file'"
|
||||||
|
class="el-icon-document"
|
||||||
|
></i>
|
||||||
|
<!-- 可继续补充其他文件类型图标 -->
|
||||||
|
</span>
|
||||||
|
{{ item.UserFileName }}
|
||||||
|
<!-- <span class="file-time">{{ item.CreatedAt }}</span> -->
|
||||||
|
</el-checkbox>
|
||||||
|
</el-checkbox-group>
|
||||||
|
</div>
|
||||||
|
<!-- 底部状态栏和按钮 -->
|
||||||
|
<div class="footer-bar">
|
||||||
|
<span class="selected-count"
|
||||||
|
>已选 {{ selectedFiles.length }} 个文件</span
|
||||||
|
>
|
||||||
|
<el-button @click="selectFileVisible = false">取消</el-button>
|
||||||
|
<el-button type="primary" @click="handleSelectFileConfirm"
|
||||||
|
>确认添加({{ selectedFiles.length }})</el-button
|
||||||
|
>
|
||||||
|
</div>
|
||||||
|
</el-dialog>
|
||||||
|
</div>
|
||||||
|
<!-- 上传文件对话框 -->
|
||||||
|
<div>
|
||||||
|
<el-dialog
|
||||||
|
title="上传文件"
|
||||||
|
v-model="uploadFileVisible"
|
||||||
|
width="50%"
|
||||||
|
:before-close="handleUploadFileClose"
|
||||||
|
>
|
||||||
|
<UploadFile></UploadFile>
|
||||||
|
</el-dialog>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script setup lang="ts">
|
||||||
|
import { ref, onMounted, onUnmounted, reactive, nextTick } from "vue";
|
||||||
|
import { ElCard, ElInput, ElButton, ElDialog } from "element-plus";
|
||||||
|
import { WSMessage, GenMessage } from "@/types/im";
|
||||||
|
import { GetMessageService } from "@/api/im";
|
||||||
|
import { FindUserFileService } from "@/api/file";
|
||||||
|
import { Model } from "@/types/model";
|
||||||
|
import { File, fileUrl } from "@/types/file";
|
||||||
|
import { Session } from "@/types/session";
|
||||||
|
import { FindSessionService } from "@/api/session";
|
||||||
|
import { ElMessage } from "element-plus";
|
||||||
|
import { Check, DocumentCopy } from "@element-plus/icons-vue";
|
||||||
|
import MarkdownIt from "markdown-it";
|
||||||
|
import hljs from "highlight.js";
|
||||||
|
import UploadFile from "@/components/upload-file.vue";
|
||||||
|
import { FindModelListByFunctionName } from "@/api/function";
|
||||||
|
import markdownItHighlightjs from "markdown-it-highlightjs";
|
||||||
|
import markdownItKatex from "markdown-it-katex";
|
||||||
|
import mermaidPlugin from "@agoose77/markdown-it-mermaid";
|
||||||
|
import "katex/dist/katex.min.css";
|
||||||
|
interface Message {
|
||||||
|
role: "user" | "assistant";
|
||||||
|
content: string;
|
||||||
|
finished?: boolean;
|
||||||
|
}
|
||||||
|
interface SendImageMessage {
|
||||||
|
img_name : string;
|
||||||
|
img_url : string;
|
||||||
|
}
|
||||||
|
interface ImageMessage {
|
||||||
|
image_content: SendImageMessage[];
|
||||||
|
text: string;
|
||||||
|
}
|
||||||
|
interface FileMessage {
|
||||||
|
file_content: File; //文件内容，文件内容的base64编码
|
||||||
|
file_type: string | null; //文件类型，文本类型，图片类型等，text_file,img_file等
|
||||||
|
}
|
||||||
|
|
||||||
|
interface GenerationMessage {
|
||||||
|
text: string;
|
||||||
|
file_content: FileMessage[];
|
||||||
|
}
|
||||||
|
|
||||||
|
const md = new MarkdownIt();
|
||||||
|
md.use(markdownItHighlightjs, {
|
||||||
|
hljs,
|
||||||
|
auto: true,
|
||||||
|
code: true,
|
||||||
|
});
|
||||||
|
md.renderer.rules.image = function (tokens, idx, options, env, self) {
|
||||||
|
const token = tokens[idx];
|
||||||
|
// 设置图片的宽度和高度属性
|
||||||
|
token.attrSet('width', '400');
|
||||||
|
token.attrSet('height', '300');
|
||||||
|
return self.renderToken(tokens, idx, options, env, self);
|
||||||
|
};
|
||||||
|
md.use(markdownItKatex);
|
||||||
|
md.use(mermaidPlugin);
|
||||||
|
|
||||||
|
//md.use(markdownItMermaid);
|
||||||
|
|
||||||
|
const historySessions = ref<Session[]>([]);
|
||||||
|
const loading = ref(false);
|
||||||
|
const isUserScrolling = ref<boolean>(false);
|
||||||
|
const messages = reactive([]);
|
||||||
|
const inputMessage = ref("");
|
||||||
|
const socket = ref(null);
|
||||||
|
const currentAIMessage = ref("");
|
||||||
|
const sessionID = ref(0);
|
||||||
|
const messagesContainer = ref<HTMLDivElement | null>(null);
|
||||||
|
const sessionIsShow = ref(false);
|
||||||
|
const sessionName = ref("");
|
||||||
|
const ModelList = ref<Model[]>([]);
|
||||||
|
const selectModel = ref(0);
|
||||||
|
const temperature = ref(0.5);
|
||||||
|
const topP = ref(0.9);
|
||||||
|
const selectedFiles = ref<File[]>([]); // 用于存储已选文件
|
||||||
|
const selectFileVisible = ref(false); // 控制文件选择对话框的显示与隐藏
|
||||||
|
const searchFileQuery = ref(""); // 用于搜索文件的查询条件
|
||||||
|
const filteredFiles = ref<File[]>([]); // 用于存储过滤后的文件列表
|
||||||
|
const uploadFileVisible = ref(false); // 控制上传文件对话框的显示与隐藏
|
||||||
|
|
||||||
|
const renderMarkdown = (content: string) => {
|
||||||
|
return md.render(content);
|
||||||
|
};
|
||||||
|
|
||||||
|
const scrollToBottom = () => {
|
||||||
|
let x = document.getElementsByClassName("chat-messages")[0];
|
||||||
|
if (!x) return;
|
||||||
|
x.scrollTop = x.scrollHeight; //将滚轮置底
|
||||||
|
};
|
||||||
|
|
||||||
|
const copyCode = (code: string) => {
|
||||||
|
navigator.clipboard.writeText(code).then(() => {
|
||||||
|
ElMessage.success("代码已复制到剪贴板");
|
||||||
|
});
|
||||||
|
};
|
||||||
|
const removeFile = (index: number) => {
|
||||||
|
selectedFiles.value.splice(index, 1);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleSelectFileVisible = async () => {
|
||||||
|
await getFileListData(); // 获取文件列表
|
||||||
|
console.log("selectedFiles:", selectedFiles.value);
|
||||||
|
selectFileVisible.value = true; // 显示对话框
|
||||||
|
console.log("handleSelectFileVisible:", selectFileVisible.value);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleUploadFileClose = async () => {
|
||||||
|
uploadFileVisible.value = false; // 关闭上传文件对话框
|
||||||
|
await getFileListData(); // 获取文件列表
|
||||||
|
console.log("handleUploadFileClose:", uploadFileVisible.value);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleUploadPicture = () => {
|
||||||
|
// 处理上传图片的逻辑
|
||||||
|
//选择图片并上传
|
||||||
|
};
|
||||||
|
|
||||||
|
const uploadMessageFile = () => {
|
||||||
|
// 处理上传文件的逻辑
|
||||||
|
// 这里可以调用上传文件的API
|
||||||
|
uploadFileVisible.value = true; // 显示上传文件对话框
|
||||||
|
console.log("上传文件:", selectedFiles.value);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleSelectFileConfirm = () => {
|
||||||
|
// 处理选中的文件
|
||||||
|
console.log("选中的文件:", selectedFiles.value);
|
||||||
|
// 在这里可以进行文件上传或其他操作
|
||||||
|
selectFileVisible.value = false; // 关闭对话框
|
||||||
|
};
|
||||||
|
|
||||||
|
const doButtonD = () => {
|
||||||
|
const codeBlocks = document.querySelectorAll("pre code");
|
||||||
|
codeBlocks.forEach((codeBlock) => {
|
||||||
|
//先查看是否已经添加了复制按钮
|
||||||
|
if (codeBlock.parentNode.querySelector(".code-controls")) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取代码类型
|
||||||
|
const codeType = codeBlock.className.replace("hljs", "");
|
||||||
|
// 创建代码类型显示元素
|
||||||
|
const codeTypeElement = document.createElement("span");
|
||||||
|
codeTypeElement.textContent = codeType.split("-")[1];
|
||||||
|
codeTypeElement.setAttribute("background-color", "rgba(0, 0, 0, 0.1)");
|
||||||
|
codeTypeElement.setAttribute("padding", "3px 6px");
|
||||||
|
codeTypeElement.setAttribute("border-radius", "4px");
|
||||||
|
codeTypeElement.setAttribute("font-size", "0.9em");
|
||||||
|
|
||||||
|
// 创建复制按钮
|
||||||
|
const copyButton = document.createElement("button");
|
||||||
|
copyButton.setAttribute("background-color", "dodgerblue");
|
||||||
|
copyButton.setAttribute("display", "flex");
|
||||||
|
copyButton.setAttribute("align-items", "center");
|
||||||
|
copyButton.setAttribute("padding", "5px 10px");
|
||||||
|
copyButton.setAttribute("cursor", "pointer");
|
||||||
|
copyButton.setAttribute("border-radius", "4px");
|
||||||
|
copyButton.textContent = "复制";
|
||||||
|
copyButton.classList.add();
|
||||||
|
copyButton.addEventListener("click", () => {
|
||||||
|
copyCode(codeBlock.textContent);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 设置代码块父元素的定位，以便按钮定位
|
||||||
|
const pre = codeBlock.parentNode;
|
||||||
|
// pre.style.position = 'relative';
|
||||||
|
|
||||||
|
// 创建一个容器用于放置代码类型和复制按钮
|
||||||
|
const controlsContainer = document.createElement("div");
|
||||||
|
controlsContainer.classList.add("code-controls");
|
||||||
|
controlsContainer.appendChild(codeTypeElement);
|
||||||
|
controlsContainer.appendChild(copyButton);
|
||||||
|
|
||||||
|
// 将容器添加到代码块父元素中
|
||||||
|
pre.insertBefore(controlsContainer, codeBlock);
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
onMounted(() => {
|
||||||
|
// if (typeof window !== 'undefined') {
|
||||||
|
// // 浏览器环境
|
||||||
|
// md.use(markdownItMermaid);
|
||||||
|
// }
|
||||||
|
let url =
|
||||||
|
"wss://pm.ljsea.top/im/ai_chat_ws?" +
|
||||||
|
"token=" +
|
||||||
|
localStorage.getItem("token");
|
||||||
|
//获取模型列表
|
||||||
|
let test_url = "ws://127.0.0.1:8084/im/ai_chat_ws?" + "token=" + localStorage.getItem("token");
|
||||||
|
//url =test_url;
|
||||||
|
socket.value = new WebSocket(url);
|
||||||
|
socket.value.onopen = () => {
|
||||||
|
console.log("WebSocket 连接已建立");
|
||||||
|
ElMessage.success("连接成功");
|
||||||
|
};
|
||||||
|
//getMessage(125);
|
||||||
|
messagesContainer.value = document.querySelector(".chat-messages");
|
||||||
|
|
||||||
|
socket.value.onmessage = (event) => {
|
||||||
|
let msg: WSMessage = JSON.parse(event.data);
|
||||||
|
const existingMessage = messages.find(
|
||||||
|
(msg) => msg.role === "assistant" && !msg.finished
|
||||||
|
);
|
||||||
|
if (existingMessage) {
|
||||||
|
// 追加内容
|
||||||
|
existingMessage.content += msg.msg.msg.response;
|
||||||
|
} else {
|
||||||
|
// 新消息
|
||||||
|
messages.push({
|
||||||
|
role: "assistant",
|
||||||
|
content: msg.msg.msg.response,
|
||||||
|
finished: false,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
//console.log("resp:", msg);
|
||||||
|
sessionID.value = msg.session_id;
|
||||||
|
currentAIMessage.value += msg.msg.msg.response;
|
||||||
|
if (msg.msg.msg.done) {
|
||||||
|
const assistantMessage = messages[messages.length - 1];
|
||||||
|
assistantMessage.finished = true;
|
||||||
|
loading.value = false;
|
||||||
|
doButtonD();
|
||||||
|
}
|
||||||
|
nextTick(() => {
|
||||||
|
scrollToBottom(); // 新增滚动调用
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
socket.value.onclose = () => {
|
||||||
|
console.log("WebSocket 连接已关闭");
|
||||||
|
ElMessage.error("连接已关闭");
|
||||||
|
//重新连接
|
||||||
|
socket.value = new WebSocket(url);
|
||||||
|
};
|
||||||
|
|
||||||
|
socket.value.onerror = (error) => {
|
||||||
|
socket.value = new WebSocket(url);
|
||||||
|
console.error("WebSocket 发生错误:", error);
|
||||||
|
};
|
||||||
|
});
|
||||||
|
|
||||||
|
onUnmounted(() => {
|
||||||
|
if (socket.value) {
|
||||||
|
socket.value.close();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
const sendMessage = () => {
|
||||||
|
sendMessageWithFile()
|
||||||
|
return;
|
||||||
|
if (inputMessage.value.trim() === "") {
|
||||||
|
ElMessage.warning("消息不能为空");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
let msg = {
|
||||||
|
msg: inputMessage.value,
|
||||||
|
type: "ollama",
|
||||||
|
function: "gen-ai-chat",
|
||||||
|
session_id: sessionID.value,
|
||||||
|
model_id: selectModel.value,
|
||||||
|
temperature: temperature.value,
|
||||||
|
top_p: topP.value,
|
||||||
|
};
|
||||||
|
if (selectedFiles.value.length > 0) {
|
||||||
|
// 处理选中的文件
|
||||||
|
console.log("选中的文件:", selectedFiles.value);
|
||||||
|
let img_file: File = selectedFiles.value[0];
|
||||||
|
let img_content = []
|
||||||
|
for (let i = 0; i < selectedFiles.value.length; i++) {
|
||||||
|
img_content.push({"img_name": selectedFiles.value[i].UserFileName, "img_url": fileUrl + selectedFiles.value[i].file_store_name});
|
||||||
|
}
|
||||||
|
let img_msg: ImageMessage = {
|
||||||
|
image_content: img_content,
|
||||||
|
text: inputMessage.value,
|
||||||
|
};
|
||||||
|
let img_msg_str = JSON.stringify(img_msg);
|
||||||
|
msg["msg"] = img_msg_str;
|
||||||
|
msg["is_image"] = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
socket.value.send(JSON.stringify(msg));
|
||||||
|
} catch (e) {
|
||||||
|
ElMessage.error("发送失败!连接已断开！");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (sessionID.value == 0) {
|
||||||
|
sessionName.value = inputMessage.value;
|
||||||
|
}
|
||||||
|
let pMsgContent ="";
|
||||||
|
if (msg["is_image"]) {
|
||||||
|
let img_msg: ImageMessage = JSON.parse(msg["msg"]);
|
||||||
|
//解析成md格式
|
||||||
|
let img_content= img_msg.image_content
|
||||||
|
for (let i = 0; i < img_content.length; i++) {
|
||||||
|
pMsgContent += `![${img_content[i].img_name}](${img_content[i].img_url})` + "\n";
|
||||||
|
}
|
||||||
|
pMsgContent = pMsgContent + img_msg.text;
|
||||||
|
} else {
|
||||||
|
pMsgContent = msg.msg;
|
||||||
|
}
|
||||||
|
messages.push({ role: "user", content: pMsgContent, finished: true });
|
||||||
|
inputMessage.value = "";
|
||||||
|
nextTick(() => {
|
||||||
|
scrollToBottom(); // 新增滚动调用
|
||||||
|
});
|
||||||
|
loading.value = true;
|
||||||
|
if (sessionID.value == 0) {
|
||||||
|
sessionName.value = msg.msg;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
const sendMessageWithFile = () => {
|
||||||
|
if (inputMessage.value.trim() === "") {
|
||||||
|
ElMessage.warning("消息不能为空");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
let end_msg = {
|
||||||
|
msg: inputMessage.value,
|
||||||
|
type: "ollama",
|
||||||
|
function: "gen-ai-chat",
|
||||||
|
session_id: sessionID.value,
|
||||||
|
model_id: selectModel.value,
|
||||||
|
temperature: temperature.value,
|
||||||
|
top_p: topP.value,
|
||||||
|
};
|
||||||
|
if (selectedFiles.value.length > 0) {
|
||||||
|
// 处理选中的文件
|
||||||
|
console.log("选中的文件:", selectedFiles.value);
|
||||||
|
let file_contents = []
|
||||||
|
let file_type= ""
|
||||||
|
for (let i = 0; i < selectedFiles.value.length; i++) {
|
||||||
|
let file: File = selectedFiles.value[i];
|
||||||
|
//图片文件：jpg、png、gif、bmp
|
||||||
|
if (file.UserFileName.endsWith(".jpg") || file.UserFileName.endsWith(".png") || file.UserFileName.endsWith(".gif") || file.UserFileName.endsWith(".bmp")) {
|
||||||
|
file_type = "image_file"
|
||||||
|
}else{
|
||||||
|
file_type = "text_file"
|
||||||
|
}
|
||||||
|
let file_msg: FileMessage = {
|
||||||
|
file_content: file,
|
||||||
|
file_type: file_type,
|
||||||
|
};
|
||||||
|
file_contents.push(file_msg);
|
||||||
|
}
|
||||||
|
let msg : GenerationMessage = {
|
||||||
|
text: inputMessage.value,
|
||||||
|
file_content: file_contents,
|
||||||
|
};
|
||||||
|
let msg_str = JSON.stringify(msg);
|
||||||
|
end_msg["msg"] = msg_str;
|
||||||
|
end_msg["is_file"] = true;
|
||||||
|
}
|
||||||
|
console.log("end_msg:", end_msg);
|
||||||
|
|
||||||
|
try {
|
||||||
|
socket.value.send(JSON.stringify(end_msg));
|
||||||
|
console.log("send msg:", end_msg);
|
||||||
|
} catch (e) {
|
||||||
|
ElMessage.error("发送失败!连接已断开！");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (sessionID.value == 0) {
|
||||||
|
sessionName.value = inputMessage.value;
|
||||||
|
}
|
||||||
|
let pMsgContent ="";
|
||||||
|
if (end_msg["is_file"]) {
|
||||||
|
let file_msg: GenerationMessage = JSON.parse(end_msg["msg"]);
|
||||||
|
//解析成md格式
|
||||||
|
let file_content= file_msg.file_content
|
||||||
|
for (let i = 0; i < file_content.length; i++) {
|
||||||
|
if(file_content[i].file_type == "image_file"){
|
||||||
|
pMsgContent += `![${file_content[i].file_content.UserFileName}](${ fileUrl + file_content[i].file_content.file_store_name})` + "\n\n";
|
||||||
|
}else{
|
||||||
|
pMsgContent += `文件：[${file_content[i].file_content.UserFileName}](${ fileUrl + file_content[i].file_content.file_store_name})` + "\n\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
// let img_content= img_msg.image_content
|
||||||
|
// for (let i = 0; i < img_content.length; i++) {
|
||||||
|
// pMsgContent += `![${img_content[i].img_name}](${img_content[i].img_url})` + "\n";
|
||||||
|
// }
|
||||||
|
pMsgContent = pMsgContent + "输入：" + file_msg.text;
|
||||||
|
} else {
|
||||||
|
pMsgContent = end_msg.msg;
|
||||||
|
}
|
||||||
|
messages.push({ role: "user", content: pMsgContent, finished: true });
|
||||||
|
inputMessage.value = "";
|
||||||
|
nextTick(() => {
|
||||||
|
scrollToBottom(); // 新增滚动调用
|
||||||
|
});
|
||||||
|
loading.value = true;
|
||||||
|
if (sessionID.value == 0) {
|
||||||
|
sessionName.value = end_msg.msg;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const loadSession = async (session_id: number) => {
|
||||||
|
sessionID.value = session_id;
|
||||||
|
messages.length = 0; // 清空消息
|
||||||
|
sessionName.value = historySessions.value.find(
|
||||||
|
(session) => session.ID == session_id
|
||||||
|
)?.Name;
|
||||||
|
await getMessage(session_id);
|
||||||
|
scrollToBottom();
|
||||||
|
doButtonD();
|
||||||
|
};
|
||||||
|
|
||||||
|
const clearCurrent = () => {
|
||||||
|
sessionID.value = 0;
|
||||||
|
messages.length = 0; // 清空消息
|
||||||
|
sessionName.value = "新会话";
|
||||||
|
ElMessage.success("新会话已创建!可以开始聊天了");
|
||||||
|
};
|
||||||
|
|
||||||
|
const showSession = async () => {
|
||||||
|
//获取历史会话
|
||||||
|
let req = {
|
||||||
|
token: localStorage.getItem("token"),
|
||||||
|
type: "UserID",
|
||||||
|
session_type: 4, //知识库会话
|
||||||
|
};
|
||||||
|
let result = await FindSessionService(req);
|
||||||
|
historySessions.value = result.data;
|
||||||
|
sessionIsShow.value = !sessionIsShow.value;
|
||||||
|
};
|
||||||
|
const getShortenedName = (name: string) => {
|
||||||
|
if (name.length > 10) {
|
||||||
|
return name.slice(0, 10) + "...";
|
||||||
|
}
|
||||||
|
return name;
|
||||||
|
};
|
||||||
|
|
||||||
|
const getMessage = async (session_id: number) => {
|
||||||
|
let result = {};
|
||||||
|
try {
|
||||||
|
let req = {
|
||||||
|
token: localStorage.getItem("token"),
|
||||||
|
session_id: session_id,
|
||||||
|
};
|
||||||
|
result = await GetMessageService(req);
|
||||||
|
if (result["code"] === 0) {
|
||||||
|
console.log(result["data"]);
|
||||||
|
let data = result["data"];
|
||||||
|
for (let i = 0; i < data.length; i++) {
|
||||||
|
if (data[i]["Type"] === 3) {
|
||||||
|
let msg: GenMessage = data[i];
|
||||||
|
let pMsgContent="";
|
||||||
|
if (msg.Status == 3) {
|
||||||
|
let file_msg: GenerationMessage = JSON.parse(msg.Msg);
|
||||||
|
let file_content= file_msg.file_content
|
||||||
|
for (let i = 0; i < file_content.length; i++) {
|
||||||
|
if(file_content[i].file_type == "image_file"){
|
||||||
|
pMsgContent += `![${file_content[i].file_content.UserFileName}](${ fileUrl + file_content[i].file_content.file_store_name})` + "\n\n";
|
||||||
|
}else{
|
||||||
|
pMsgContent += `文件：[${file_content[i].file_content.UserFileName}](${ fileUrl + file_content[i].file_content.file_store_name})` + "\n\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
pMsgContent = pMsgContent + file_msg.text;
|
||||||
|
} else {
|
||||||
|
pMsgContent = msg.Msg;
|
||||||
|
}
|
||||||
|
messages.push({
|
||||||
|
role: "user",
|
||||||
|
content: pMsgContent,
|
||||||
|
finished: true,
|
||||||
|
});
|
||||||
|
} else if (data[i]["Type"] === 4) {
|
||||||
|
messages.push({
|
||||||
|
role: "assistant",
|
||||||
|
content: data[i]["Msg"],
|
||||||
|
finished: true,
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
console.log("未知消息类型");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.log(e);
|
||||||
|
}
|
||||||
|
return {};
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
const getMessageWithFile = async (session_id: number) => {
|
||||||
|
let result = {};
|
||||||
|
try {
|
||||||
|
let req = {
|
||||||
|
token: localStorage.getItem("token"),
|
||||||
|
session_id: session_id,
|
||||||
|
};
|
||||||
|
result = await GetMessageService(req);
|
||||||
|
if (result["code"] === 0) {
|
||||||
|
console.log(result["data"]);
|
||||||
|
let data = result["data"];
|
||||||
|
for (let i = 0; i < data.length; i++) {
|
||||||
|
if (data[i]["Type"] === 3) {
|
||||||
|
let msg: GenMessage = data[i];
|
||||||
|
let pMsgContent="";
|
||||||
|
if (msg.Status == 3) {
|
||||||
|
let img_msg: ImageMessage = JSON.parse(msg.Msg);
|
||||||
|
//解析成md格式
|
||||||
|
let img_content= img_msg.image_content
|
||||||
|
for (let i = 0; i < img_content.length; i++) {
|
||||||
|
pMsgContent += `![${img_content[i].img_name}](${img_content[i].img_url})` + "\n";
|
||||||
|
}
|
||||||
|
pMsgContent = pMsgContent + img_msg.text;
|
||||||
|
} else {
|
||||||
|
pMsgContent = msg.Msg;
|
||||||
|
}
|
||||||
|
messages.push({
|
||||||
|
role: "user",
|
||||||
|
content: pMsgContent,
|
||||||
|
finished: true,
|
||||||
|
});
|
||||||
|
} else if (data[i]["Type"] === 4) {
|
||||||
|
messages.push({
|
||||||
|
role: "assistant",
|
||||||
|
content: data[i]["Msg"],
|
||||||
|
finished: true,
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
console.log("未知消息类型");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.log(e);
|
||||||
|
}
|
||||||
|
return {};
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
const copyMessage = (content: string) => {
|
||||||
|
navigator.clipboard
|
||||||
|
.writeText(content)
|
||||||
|
.then(() => {
|
||||||
|
ElMessage.success("复制成功");
|
||||||
|
})
|
||||||
|
.catch((error) => {
|
||||||
|
ElMessage.error("复制失败: " + error);
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
const GetModelListByFunctionName = async () => {
|
||||||
|
let req = {
|
||||||
|
function: "kbase-chat",
|
||||||
|
token: localStorage.getItem("token"),
|
||||||
|
};
|
||||||
|
try {
|
||||||
|
let result = await FindModelListByFunctionName(req);
|
||||||
|
if (result["code"] === 0) {
|
||||||
|
ModelList.value = result["data"];
|
||||||
|
selectModel.value = ModelList.value[0].ID;
|
||||||
|
console.log("model_list:", ModelList.value);
|
||||||
|
} else {
|
||||||
|
ElMessage.error(result["msg"]);
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.log(e);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
GetModelListByFunctionName();
|
||||||
|
|
||||||
|
const getFileListData = async () => {
|
||||||
|
let req = {
|
||||||
|
token: localStorage.getItem("token"),
|
||||||
|
type: "all",
|
||||||
|
};
|
||||||
|
let result = await FindUserFileService(req);
|
||||||
|
if (result["code"] === 0) {
|
||||||
|
filteredFiles.value = result["data"];
|
||||||
|
} else {
|
||||||
|
ElMessage.error(result["msg"]);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
</script>
|
||||||
|
<style scoped>
|
||||||
|
.chat-app {
|
||||||
|
display: flex;
|
||||||
|
height: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.history-sessions {
|
||||||
|
width: 15%;
|
||||||
|
padding: 20px;
|
||||||
|
height: 100%;
|
||||||
|
overflow-y: auto;
|
||||||
|
border-right: 1px solid #e1e1e1;
|
||||||
|
scrollbar-width: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
.history-sessions ul {
|
||||||
|
list-style-type: none;
|
||||||
|
padding: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.history-sessions li {
|
||||||
|
cursor: pointer;
|
||||||
|
padding: 10px;
|
||||||
|
border-bottom: 1px solid #e1e1e1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.history-sessions li:hover {
|
||||||
|
background-color: #f0f0f0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.chat-container {
|
||||||
|
flex: 1;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
padding: 20px;
|
||||||
|
height: 100%;
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
|
||||||
|
.chat-messages {
|
||||||
|
flex: 1;
|
||||||
|
overflow-y: auto; /* 允许垂直滚动 */
|
||||||
|
padding: 10px;
|
||||||
|
margin-bottom: 20px;
|
||||||
|
scrollbar-width: 10px;
|
||||||
|
height: 100%;
|
||||||
|
position: relative;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
|
||||||
|
.messages-wrapper {
|
||||||
|
flex: 1;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
overflow-y: auto; /* 确保内部可以滚动 */
|
||||||
|
height: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message {
|
||||||
|
display: flex;
|
||||||
|
margin-bottom: 15px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.user {
|
||||||
|
justify-content: flex-end;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-avatar {
|
||||||
|
width: 40px;
|
||||||
|
text-align: center;
|
||||||
|
line-height: 40px;
|
||||||
|
font-size: 20px;
|
||||||
|
margin-right: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.assistant .message-avatar {
|
||||||
|
margin-right: 0;
|
||||||
|
margin-left: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message-content {
|
||||||
|
background-color: #f0f9eb;
|
||||||
|
border: 1px solid #e1f3d8;
|
||||||
|
border-radius: 4px;
|
||||||
|
padding: 10px 15px;
|
||||||
|
white-space: pre-wrap;
|
||||||
|
max-width: 70%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.message.assistant .message-content {
|
||||||
|
background-color: #f3f4f6;
|
||||||
|
border: 1px solid #dce4eb;
|
||||||
|
max-width: 70%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.loading-indicator {
|
||||||
|
text-align: center;
|
||||||
|
color: #999;
|
||||||
|
font-size: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.chat-input {
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.el-input {
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.session-card {
|
||||||
|
margin-bottom: 20px;
|
||||||
|
height: 20%;
|
||||||
|
}
|
||||||
|
.session-card2 {
|
||||||
|
margin-bottom: 20px;
|
||||||
|
height: 70%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.session-card ul {
|
||||||
|
list-style-type: none;
|
||||||
|
padding: 0;
|
||||||
|
margin: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.session-card li {
|
||||||
|
cursor: pointer;
|
||||||
|
padding: 10px;
|
||||||
|
border-bottom: 1px solid #e1e1e1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.session-card li:hover {
|
||||||
|
background-color: #f0f0f0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.scroll-wrapper {
|
||||||
|
height: 100%;
|
||||||
|
overflow-y: auto;
|
||||||
|
}
|
||||||
|
|
||||||
|
.code-controls {
|
||||||
|
position: absolute;
|
||||||
|
top: 5px;
|
||||||
|
right: 5px;
|
||||||
|
display: flex;
|
||||||
|
gap: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.copy-code-button {
|
||||||
|
background-color: dodgerblue;
|
||||||
|
color: white;
|
||||||
|
width: 30px;
|
||||||
|
height: 20px;
|
||||||
|
border: 0;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.el-icon-copy {
|
||||||
|
margin-right: 5px;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 对话框 */
|
||||||
|
.file-list {
|
||||||
|
max-height: 400px;
|
||||||
|
overflow-y: auto;
|
||||||
|
border: 1px solid #e4e7ed;
|
||||||
|
border-radius: 4px;
|
||||||
|
margin-bottom: 10px;
|
||||||
|
}
|
||||||
|
.file-icon {
|
||||||
|
margin-right: 5px;
|
||||||
|
color: #606266;
|
||||||
|
}
|
||||||
|
.file-time {
|
||||||
|
float: right;
|
||||||
|
color: #909399;
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
.footer-bar {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
.selected-count {
|
||||||
|
margin-right: 10px;
|
||||||
|
}
|
||||||
|
.el-icon-document {
|
||||||
|
color: #409eff;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
|
||||||
|
|
@ -0,0 +1,314 @@
|
||||||
|
<template>
|
||||||
|
<div>
|
||||||
|
<TableSearch :query="query" :options="searchOpt" :search="handleSearch" :refresh="getData" />
|
||||||
|
<div class="container">
|
||||||
|
<TableCustom :columns="columns" :tableData="tableData" :total="page.total" :viewFunc="handleView"
|
||||||
|
:delFunc="handleDelete" :page-change="changePage" :editFunc="handleEdit" :genOperate="handleGenOperate" :refresh="getData">
|
||||||
|
<template #toolbarBtn>
|
||||||
|
<el-button type="warning" :icon="CirclePlusFilled" @click="visible_add = true">新增</el-button>
|
||||||
|
</template>
|
||||||
|
</TableCustom>
|
||||||
|
|
||||||
|
</div>
|
||||||
|
<el-dialog :title="isEdit ? '编辑': '编辑'" v-model="visible" width="700px" destroy-on-close
|
||||||
|
:close-on-click-modal="false" @close="closeDialog">
|
||||||
|
<TableEdit :form-data="rowData" :options="options_edit" :edit="isEdit" :update="updateData" />
|
||||||
|
</el-dialog>
|
||||||
|
<el-dialog :title="isAdd ? '新增' : '新增'" v-model="visible_add" width="700px" destroy-on-close
|
||||||
|
:close-on-click-modal="false" @close="closeDialog">
|
||||||
|
<TableEdit :form-data="rowData" :options="options" :edit="isAdd" :update="addData" />
|
||||||
|
</el-dialog>
|
||||||
|
<el-dialog title="查看详情" v-model="visible1" width="700px" destroy-on-close>
|
||||||
|
<TableDetail :data="viewData"></TableDetail>
|
||||||
|
</el-dialog>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script setup lang="ts" name="system-user">
|
||||||
|
import { ref, reactive } from 'vue';
|
||||||
|
import { ElMessage } from 'element-plus';
|
||||||
|
import { CirclePlusFilled } from '@element-plus/icons-vue';
|
||||||
|
import { FindUserFileService } from "@/api/file";
|
||||||
|
import { KBase } from '@/types/kbase';
|
||||||
|
import {FindKnowledgeBaseService} from "@/api/kbase";
|
||||||
|
import {UpdateKnowledgeBaseService} from "@/api/kbase";
|
||||||
|
import {AddKnowledgeBaseService} from "@/api/kbase";
|
||||||
|
import {DelKnowledgeBaseService} from "@/api/kbase";
|
||||||
|
import TableCustom from '@/components/table-custom.vue';
|
||||||
|
import TableDetail from '@/components/table-detail.vue';
|
||||||
|
import TableSearch from '@/components/table-search.vue';
|
||||||
|
import { FormOption, FormOptionList } from '@/types/form-option';
|
||||||
|
|
||||||
|
|
||||||
|
// 查询相关
|
||||||
|
const query = reactive({
|
||||||
|
name: '',
|
||||||
|
});
|
||||||
|
const searchOpt = ref<FormOptionList[]>([
|
||||||
|
{ type: 'input', label: '模型ID：', prop: 'name' }
|
||||||
|
])
|
||||||
|
const handleSearch = async () => {
|
||||||
|
// query.name是否是数字
|
||||||
|
if (isNaN(Number(query.name))) {
|
||||||
|
ElMessage.error('请输入数字');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (query.name === '') {
|
||||||
|
ElMessage.error('请输入模型ID');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
let req={
|
||||||
|
token: localStorage.getItem('token'),
|
||||||
|
type: "ID",
|
||||||
|
id: parseInt(query.name)
|
||||||
|
}
|
||||||
|
let result = await FindKnowledgeBaseService(req);
|
||||||
|
tableData.value = result.data;
|
||||||
|
page.total = result.data.length;
|
||||||
|
};
|
||||||
|
|
||||||
|
// 表格相关
|
||||||
|
let columns = ref([
|
||||||
|
// { type: 'index', label: '序号', width: 55, align: 'center' },
|
||||||
|
{ prop: 'ID', label: '知识库ID' },
|
||||||
|
{ prop: "Name", label: "知识库名称" },
|
||||||
|
{prop: 'Description', label: '描述'},
|
||||||
|
{ prop: 'FileNameList', label:"知识库文件", width: 300},
|
||||||
|
{ prop: 'CreatedAt', label: '创建时间',type: 'date' },
|
||||||
|
{ prop: 'UpdatedAt', label: '上次更新时间', type: 'date'},
|
||||||
|
{ prop: 'operator', label: '操作', width: 250 , operate: { view: true, edit: true, delete: true,push: {link: false,label:"继续该会话"},gen: {show: false,label:"下载文件"} }},
|
||||||
|
])
|
||||||
|
const page = reactive({
|
||||||
|
index: 1,
|
||||||
|
size: 10,
|
||||||
|
total: 122,
|
||||||
|
})
|
||||||
|
const tableData = ref<KBase[]>([]);
|
||||||
|
const file_ids_select_opts = ref([]);
|
||||||
|
const getData = async () => {
|
||||||
|
let req={
|
||||||
|
token: localStorage.getItem('token'),
|
||||||
|
type: "UserID"
|
||||||
|
}
|
||||||
|
let result = await FindKnowledgeBaseService(req);
|
||||||
|
let req2 = {
|
||||||
|
token: localStorage.getItem("token"),
|
||||||
|
type: "all",
|
||||||
|
}
|
||||||
|
let fileResult = await FindUserFileService(req2);
|
||||||
|
file_ids_select_opts.value.length = 0; // 清空数组
|
||||||
|
for (let i = 0; i < fileResult.data.length; i++) {
|
||||||
|
let userFile = fileResult.data[i];
|
||||||
|
file_ids_select_opts.value.push({
|
||||||
|
label: userFile.UserFileName,
|
||||||
|
value: userFile.ID
|
||||||
|
})
|
||||||
|
}
|
||||||
|
tableData.value = result.data;
|
||||||
|
//将文件id与对应文件对应的文件名拼接成字符串
|
||||||
|
for (let i = 0; i < tableData.value.length; i++) {
|
||||||
|
let file_ids = JSON.parse(tableData.value[i].FileIDs.toString())
|
||||||
|
let file_id_list = []
|
||||||
|
for (let j = 0; j < file_ids.length; j++) {
|
||||||
|
file_id_list.push(file_ids[j]["file_id"])
|
||||||
|
}
|
||||||
|
let file_name_list = []
|
||||||
|
//fileResult.data是文件列表
|
||||||
|
for (let k = 0; k < fileResult.data.length; k++) {
|
||||||
|
let userFile = fileResult.data[k];
|
||||||
|
for (let l = 0; l < file_id_list.length; l++) {
|
||||||
|
if (userFile.ID == file_id_list[l]) {
|
||||||
|
file_name_list.push(userFile.UserFileName)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
tableData.value[i].FileNameList = file_name_list.join(",")
|
||||||
|
// console.log("file_name_list:", file_name_list)
|
||||||
|
}
|
||||||
|
page.total = result.data.length;
|
||||||
|
};
|
||||||
|
getData();
|
||||||
|
|
||||||
|
const changePage = (val: number) => {
|
||||||
|
page.index = val;
|
||||||
|
getData();
|
||||||
|
};
|
||||||
|
|
||||||
|
// 新增弹窗
|
||||||
|
let options = ref<FormOption>({
|
||||||
|
labelWidth: '100px',
|
||||||
|
span: 12,
|
||||||
|
list: [
|
||||||
|
{ type: 'input', label: '名称', prop: 'Name', required: true },
|
||||||
|
{ type: 'select', label: '文件', prop: 'FileIDs', required: true, opts:file_ids_select_opts.value ,multiple: true},
|
||||||
|
{ type: 'input', label: '描述', prop: 'Description', required: true },
|
||||||
|
]
|
||||||
|
})
|
||||||
|
|
||||||
|
//编辑弹窗
|
||||||
|
let options_edit = ref<FormOption>({
|
||||||
|
labelWidth: '100px',
|
||||||
|
span: 12,
|
||||||
|
list: [
|
||||||
|
{ type: 'input', label: '名称', prop: 'Name', required: true },
|
||||||
|
{ type: 'select', label: '文件', prop: 'FileIDs', required: true, opts:file_ids_select_opts.value ,multiple: true},
|
||||||
|
{ type: 'input', label: '描述', prop: 'Description', required: true },
|
||||||
|
]
|
||||||
|
})
|
||||||
|
|
||||||
|
const visible = ref(false);
|
||||||
|
const visible_add = ref(false);
|
||||||
|
const isEdit = ref(false);
|
||||||
|
const isAdd = ref(false);
|
||||||
|
const rowData = ref({});
|
||||||
|
const handleEdit = async (row: KBase) => {
|
||||||
|
let data = row;
|
||||||
|
rowData.value = data;
|
||||||
|
if(typeof data.FileIDs === "string"){
|
||||||
|
console.log("edit_row_data:", rowData.value);
|
||||||
|
let file_ids = JSON.parse(data.FileIDs.toString())
|
||||||
|
let file_id_list = []
|
||||||
|
for (let i = 0; i < file_ids.length; i++) {
|
||||||
|
file_id_list.push(file_ids[i]["file_id"])
|
||||||
|
}
|
||||||
|
console.log("file_id_list:", file_id_list)
|
||||||
|
rowData.value.FileIDs = file_id_list;
|
||||||
|
}
|
||||||
|
isEdit.value = true;
|
||||||
|
visible.value = true;
|
||||||
|
};
|
||||||
|
const updateData = async (data) => {
|
||||||
|
let file_id =[]
|
||||||
|
console.log("file update:",data)
|
||||||
|
let file_id_list= data["FileIDs"]
|
||||||
|
for (let i = 0; i < file_id_list.length; i++) {
|
||||||
|
file_id.push({"file_id":file_id_list[i]})
|
||||||
|
}
|
||||||
|
let file_ids = JSON.stringify(file_id)
|
||||||
|
let result ={}
|
||||||
|
try{
|
||||||
|
let req={
|
||||||
|
token:localStorage.getItem("token"),
|
||||||
|
id: data.ID,
|
||||||
|
name: data.Name,
|
||||||
|
description: data.Description,
|
||||||
|
file_ids: file_ids,
|
||||||
|
};
|
||||||
|
result = await UpdateKnowledgeBaseService(req)
|
||||||
|
if (result["code"] === 0) {
|
||||||
|
ElMessage.success("更新成功");
|
||||||
|
} else {
|
||||||
|
ElMessage.error("更新失败");
|
||||||
|
}
|
||||||
|
|
||||||
|
}catch(e){
|
||||||
|
console.log(e);
|
||||||
|
}
|
||||||
|
closeDialog();
|
||||||
|
getData();
|
||||||
|
};
|
||||||
|
|
||||||
|
const addData = async (data) => {
|
||||||
|
let file_id =[]
|
||||||
|
console.log("file add:",data)
|
||||||
|
let file_id_list= data["FileIDs"]
|
||||||
|
for (let i = 0; i < file_id_list.length; i++) {
|
||||||
|
file_id.push({"file_id":file_id_list[i]})
|
||||||
|
}
|
||||||
|
let file_ids = JSON.stringify(file_id)
|
||||||
|
let result ={}
|
||||||
|
try{
|
||||||
|
let req={
|
||||||
|
token:localStorage.getItem("token"),
|
||||||
|
name: data.Name,
|
||||||
|
description: data.Description,
|
||||||
|
file_ids: file_ids,
|
||||||
|
};
|
||||||
|
console.log("add row data:", req);
|
||||||
|
result = await AddKnowledgeBaseService(req)
|
||||||
|
if (result["code"] === 0) {
|
||||||
|
ElMessage.success("新增成功");
|
||||||
|
} else {
|
||||||
|
ElMessage.error("新增失败");
|
||||||
|
}
|
||||||
|
|
||||||
|
}catch(e){
|
||||||
|
console.log(e);
|
||||||
|
}
|
||||||
|
closeDialog();
|
||||||
|
getData();
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
const closeDialog = () => {
|
||||||
|
visible.value = false;
|
||||||
|
visible_add.value = false;
|
||||||
|
isEdit.value = false;
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleGenOperate = async (row: File) => {
|
||||||
|
console.log("gen row:", row);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 查看详情弹窗相关
|
||||||
|
const visible1 = ref(false);
|
||||||
|
const viewData = ref({
|
||||||
|
row: {},
|
||||||
|
list: [
|
||||||
|
|
||||||
|
]
|
||||||
|
});
|
||||||
|
const handleView =async (row: KBase) => {
|
||||||
|
viewData.value.row = row;
|
||||||
|
viewData.value.list = [
|
||||||
|
{
|
||||||
|
prop: 'ID',
|
||||||
|
label: '知识库ID',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
prop: 'Name',
|
||||||
|
label: '知识库名称',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
prop: 'FileNameList',
|
||||||
|
label: '知识库文件',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
prop: 'Description',
|
||||||
|
label: '描述',
|
||||||
|
},
|
||||||
|
{
|
||||||
|
prop: 'CreatedAt',
|
||||||
|
label: '创建时间',
|
||||||
|
type: 'date'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
prop: 'UpdatedAt',
|
||||||
|
label: '更新时间',
|
||||||
|
type: 'date'
|
||||||
|
},
|
||||||
|
]
|
||||||
|
visible1.value = true;
|
||||||
|
};
|
||||||
|
|
||||||
|
// 删除相关
|
||||||
|
const handleDelete = async (row: KBase) => {
|
||||||
|
let req={
|
||||||
|
token: localStorage.getItem('token'),
|
||||||
|
id: row.ID,
|
||||||
|
}
|
||||||
|
try{
|
||||||
|
let result = await DelKnowledgeBaseService(req);
|
||||||
|
if(result["code"] === 0){
|
||||||
|
ElMessage.success("删除成功");
|
||||||
|
getData();
|
||||||
|
}else{
|
||||||
|
ElMessage.error("删除失败");
|
||||||
|
}
|
||||||
|
}catch(e){
|
||||||
|
console.log(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<style scoped></style>
|
||||||
|
|
@ -1,8 +0,0 @@
|
||||||
<template>
|
|
||||||
<div>
|
|
||||||
<h2>智能选题推荐</h2>
|
|
||||||
</div>
|
|
||||||
</template>
|
|
||||||
|
|
||||||
<script setup lang="ts">
|
|
||||||
</script>
|
|
||||||
Loading…
Reference in New Issue