修改对话输出

This commit is contained in:
junleea 2025-03-25 19:51:06 +08:00
parent 3ff429fa40
commit 6117d92034
4 changed files with 580 additions and 205 deletions

306
README.md
View File

@ -1,2 +1,306 @@
{"model":"spark","appid":"d3a4647c","apiSecret":"YjRlNGNjYzM5ZjIxNGM5NzhkNDJjMGM4","apiKey":"3ee5ae2b03a4039e79ba1a490631309e","domain":"pro-128k"}
<template>
<div class="chat-container">
<!-- 消息列表 -->
<el-card class="chat-messages" shadow="never" ref="messageContainer">
<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>
</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"
:rows="2"
placeholder="输入消息..."
@keyup.enter="sendMessage"
/>
</el-col>
<el-col :span="4">
<el-button type="primary" @click="sendMessage">Send</el-button>
</el-col>
</el-row>
</el-card>
</div>
</template>
<script setup lang="ts">
import { ref, onMounted, onUnmounted, reactive, nextTick } from "vue";
import { ElCard, ElInput, ElButton } from "element-plus";
import { WSMessage, AIQMessage, OllamaMessage } from "@/types/im";
import { ElMessage } from "element-plus";
import { GetMessageService } from "@/api/im";
import MarkdownIt from "markdown-it";
import hljs from "highlight.js";
import bash from "highlight.js/lib/languages/bash";
import javascript from "highlight.js/lib/languages/javascript";
import typescript from "highlight.js/lib/languages/typescript";
import java from "highlight.js/lib/languages/java";
import sql from "highlight.js/lib/languages/sql";
import nginx from "highlight.js/lib/languages/nginx";
import json from "highlight.js/lib/languages/json";
import yaml from "highlight.js/lib/languages/yaml";
import xml from "highlight.js/lib/languages/xml";
import shell from "highlight.js/lib/languages/shell";
interface Message {
role: "user" | "assistant";
content: string;
finished?: boolean;
}
hljs.registerLanguage("bash", bash);
hljs.registerLanguage("javascript", javascript);
hljs.registerLanguage("typescript", typescript);
hljs.registerLanguage("java", java);
hljs.registerLanguage("sql", sql);
hljs.registerLanguage("nginx", nginx);
hljs.registerLanguage("json", json);
hljs.registerLanguage("yaml", yaml);
hljs.registerLanguage("xml", xml);
hljs.registerLanguage("shell", shell);
const md = new MarkdownIt({
html: true,
linkify: true,
breaks: true,
xhtmlOut: true,
typographer: true,
highlight: function (str, lang) {
if (lang && hljs.getLanguage(lang)) {
try {
return (
'<pre class="hljs"><code>' +
hljs.highlight(lang, str, true).value +
"</code></pre>"
);
} catch (__) {}
}
return (
'<pre class="hljs"><code>' + md.utils.escapeHtml(str) + "</code></pre>"
);
},
});
const loading = ref(false);
const isUserScrolling = ref<boolean>(false);
const messages = reactive([]);
const inputMessage = ref("");
const socket = ref(null);
const currentAIMessage = ref("");
const messageContainer = ref(null);
const sessionID = ref(0);
const renderMarkdown = (content: string) => {
return md.render(content);
};
onMounted(() => {
let url =
"wss://pm.ljsea.top/im/ai_chat_ws?" +
"token=" +
localStorage.getItem("token");
socket.value = new WebSocket(url);
socket.value.onopen = () => {
console.log("WebSocket 连接已建立");
ElMessage.success("连接成功");
};
const container = document.querySelector('.messages-wrapper') as HTMLElement | null;
if (container) {
container.addEventListener('scroll', handleScroll);
}
socket.value.onmessage = (event) => {
let msg: WSMessage = JSON.parse(event.data);
const existingMessage = messages.find(
(msg) => msg.role === "assistant" && !msg.finished
);
// if (msg.msg.msg.done) {
// messages.push({ role: 'assistant', content: currentAIMessage.value });
// currentAIMessage.value = '';
// scrollToBottom();
// return;
// }
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;
scrollToBottom();
if (msg.msg.msg.done) {
const assistantMessage = messages[messages.length - 1];
assistantMessage.finished = true;
}
};
socket.value.onclose = () => {
console.log("WebSocket 连接已关闭");
ElMessage.error("连接已关闭");
};
socket.value.onerror = (error) => {
console.error("WebSocket 发生错误:", error);
};
});
onUnmounted(() => {
if (socket.value) {
socket.value.close();
}
});
const sendMessage = () => {
let msg = {
msg: inputMessage.value,
type: "ollama",
function: "gen-ai-chat",
session_id: sessionID.value,
};
if (inputMessage.value.trim() === "") return;
messages.push({ role: "user", content: inputMessage.value, finished: true });
socket.value.send(JSON.stringify(msg));
inputMessage.value = "";
//getMessage();
scrollToBottom();
};
const scrollToBottom = (): void => {
nextTick(() => {
if (messageContainer.value && !isUserScrolling.value) {
messageContainer.value.scrollTop = messageContainer.value.scrollHeight;
}
});
};
// 监听用户滚动行为,判断是否在手动滚动历史消息
const handleScroll = (): void => {
const container = document.querySelector(
".messages-wrapper"
) as HTMLElement | null;
if (container) {
const scrollBottom =
container.scrollHeight - container.scrollTop - container.clientHeight;
isUserScrolling.value = scrollBottom > 50; // 如果距离底部超过50px则认为在查看历史消息
}
};
const getMessage = async () => {
let result = {};
try {
let req = {
token: localStorage.getItem("token"),
session_id: sessionID.value,
};
result = await GetMessageService(req);
if (result["code"] === 0) {
console.log(result["data"]);
}
} catch (e) {
console.log(e);
}
return {};
};
</script>
<style scoped>
.chat-container {
display: flex;
flex-direction: column;
height: 100%; /* 使用视口高度确保容器占满整个屏幕 */
padding: 20px;
box-sizing: border-box;
}
.chat-messages {
flex: 1;
overflow-y: auto; /* 允许垂直滚动 */
padding: 10px;
margin-bottom: 20px;
position: relative;
display: flex;
flex-direction: column;
}
.messages-wrapper {
flex: 1;
display: flex;
flex-direction: column;
justify-content: flex-end; /* 默认将新消息对齐到底部 */
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%;
}
</style>

View File

@ -5,7 +5,7 @@
<meta charset="utf-8">
<meta http-equiv="X-UA-Compatible" content="IE=edge">
<meta name="viewport" content="width=device-width,initial-scale=1.0">
<title>vue-manage-system后台管理系统</title>
<title>大学生学业作品AI生成工具</title>
<link rel="stylesheet" href="//at.alicdn.com/t/c/font_830376_92o68tc95je.css">
</head>

View File

@ -1,71 +1,76 @@
<template>
<div class="chat-container">
<div class="chat-messages" ref="messageContainer">
<el-card
<!-- 消息列表 -->
<el-card class="chat-messages" shadow="never" ref="messagesContainer">
<div
v-for="(message, index) in messages"
:key="index"
:class="{ 'user-message': message.sender === 'user', 'ai-message': message.sender === 'ai' }"
shadow="never"
:class="['message', message.role]"
>
<!-- <template #header>
<span>{{ message.sender === 'user' ? '你' : 'AI' }}</span>
</template> -->
<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>
</el-card>
<el-card
v-if="currentAIMessage.length > 0"
class="ai-message"
shadow="never"
>
<!-- <template #header>
<span>AI</span>
</template> -->
<div v-html="renderMarkdown(currentAIMessage)"></div>
</el-card>
</div>
<div class="input-container">
</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
type="textarea"
:row = "3"
v-model="inputMessage"
placeholder="输入你的消息..."
type="textarea"
:rows="2"
placeholder="输入消息..."
@keyup.enter="sendMessage"
></el-input>
<el-button @click="sendMessage">发送</el-button>
</div>
/>
</el-col>
<el-col :span="4" style="text-align: center;" >
<el-button type="primary" @click="sendMessage">Send</el-button>
</el-col>
</el-row>
</el-card>
</div>
</template>
<script setup lang="ts">
import { ref, onMounted, onUnmounted } from 'vue';
import { ElCard, ElInput, ElButton } from 'element-plus';
import { WSMessage,AIQMessage,OllamaMessage } from '@/types/im';
import { ElMessage } from 'element-plus';
import { ref, onMounted, onUnmounted, reactive, nextTick } from "vue";
import { ElCard, ElInput, ElButton } from "element-plus";
import { WSMessage, AIQMessage, OllamaMessage } from "@/types/im";
import { ElMessage } from "element-plus";
import { GetMessageService } from "@/api/im";
import MarkdownIt from 'markdown-it';
import hljs from 'highlight.js';
import bash from 'highlight.js/lib/languages/bash'
import javascript from 'highlight.js/lib/languages/javascript';
import typescript from 'highlight.js/lib/languages/typescript';
import java from 'highlight.js/lib/languages/java';
import sql from 'highlight.js/lib/languages/sql';
import nginx from 'highlight.js/lib/languages/nginx';
import json from 'highlight.js/lib/languages/json';
import yaml from 'highlight.js/lib/languages/yaml';
import xml from 'highlight.js/lib/languages/xml';
import shell from 'highlight.js/lib/languages/shell'
hljs.registerLanguage('bash', bash)
hljs.registerLanguage('javascript', javascript);
hljs.registerLanguage('typescript', typescript);
hljs.registerLanguage('java', java);
hljs.registerLanguage('sql', sql);
hljs.registerLanguage('nginx', nginx);
hljs.registerLanguage('json', json);
hljs.registerLanguage('yaml', yaml);
hljs.registerLanguage('xml', xml);
hljs.registerLanguage('shell', shell);
import MarkdownIt from "markdown-it";
import hljs from "highlight.js";
import bash from "highlight.js/lib/languages/bash";
import javascript from "highlight.js/lib/languages/javascript";
import typescript from "highlight.js/lib/languages/typescript";
import java from "highlight.js/lib/languages/java";
import sql from "highlight.js/lib/languages/sql";
import nginx from "highlight.js/lib/languages/nginx";
import json from "highlight.js/lib/languages/json";
import yaml from "highlight.js/lib/languages/yaml";
import xml from "highlight.js/lib/languages/xml";
import shell from "highlight.js/lib/languages/shell";
interface Message {
role: "user" | "assistant";
content: string;
finished?: boolean;
}
hljs.registerLanguage("bash", bash);
hljs.registerLanguage("javascript", javascript);
hljs.registerLanguage("typescript", typescript);
hljs.registerLanguage("java", java);
hljs.registerLanguage("sql", sql);
hljs.registerLanguage("nginx", nginx);
hljs.registerLanguage("json", json);
hljs.registerLanguage("yaml", yaml);
hljs.registerLanguage("xml", xml);
hljs.registerLanguage("shell", shell);
const md = new MarkdownIt({
html: true,
@ -85,56 +90,80 @@ hljs.registerLanguage('shell', shell);
}
return (
'<pre class="hljs"><code>' +
md.utils.escapeHtml(str) +
"</code></pre>"
'<pre class="hljs"><code>' + md.utils.escapeHtml(str) + "</code></pre>"
);
},
});
const messages = ref([]);
const inputMessage = ref('');
const loading = ref(false);
const isUserScrolling = ref<boolean>(false);
const messages = reactive([]);
const inputMessage = ref("");
const socket = ref(null);
const currentAIMessage = ref('');
const messageContainer = ref(null);
const currentAIMessage = ref("");
const sessionID = ref(0);
const AIIsSending =ref(false);
const messagesContainer = ref<HTMLDivElement | null>(null);
const renderMarkdown = (content: string) => {
return md.render(content);
};
const scrollToBottom = () => {
nextTick(() => {
if (messagesContainer.value) {
messagesContainer.value.scrollTop = messagesContainer.value.scrollHeight;
}
});
};
onMounted(() => {
let url ='wss://pm.ljsea.top/im/ai_chat_ws?' + 'token=' + localStorage.getItem('token');
let url =
"wss://pm.ljsea.top/im/ai_chat_ws?" +
"token=" +
localStorage.getItem("token");
socket.value = new WebSocket(url);
socket.value.onopen = () => {
console.log('WebSocket 连接已建立');
ElMessage.success('连接成功');
console.log("WebSocket 连接已建立");
ElMessage.success("连接成功");
};
messagesContainer.value = document.querySelector(".chat-messages");
socket.value.onmessage = (event) => {
let msg: WSMessage = JSON.parse(event.data);
const chunk = event.data;
if (msg.msg.msg.done) {
messages.value.push({ sender: 'ai', content: currentAIMessage.value });
currentAIMessage.value = '';
scrollToBottom();
return;
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;
scrollToBottom();
if (msg.msg.msg.done) {
const assistantMessage = messages[messages.length - 1];
assistantMessage.finished = true;
loading.value = false;
}
nextTick(() => {
scrollToBottom(); //
});
};
socket.value.onclose = () => {
console.log('WebSocket 连接已关闭');
ElMessage.error('连接已关闭');
console.log("WebSocket 连接已关闭");
ElMessage.error("连接已关闭");
};
socket.value.onerror = (error) => {
console.error('WebSocket 发生错误:', error);
console.error("WebSocket 发生错误:", error);
};
});
@ -150,37 +179,33 @@ hljs.registerLanguage('shell', shell);
type: "ollama",
function: "gen-ai-chat",
session_id: sessionID.value,
}
if (inputMessage.value.trim() === '') return;
messages.value.push({ sender: 'user', content: inputMessage.value });
};
if (inputMessage.value.trim() === "") return;
messages.push({ role: "user", content: inputMessage.value, finished: true });
socket.value.send(JSON.stringify(msg));
inputMessage.value = '';
//getMessage();
inputMessage.value = "";
scrollToBottom();
loading.value = true;
//getMessage();
};
const scrollToBottom = () => {
if (messageContainer.value) {
messageContainer.value.scrollTop = messageContainer.value.scrollHeight;
}
};
const getMessage = async () => {
let result = {};
try {
let req = {
token: localStorage.getItem('token'),
token: localStorage.getItem("token"),
session_id: sessionID.value,
};
result = await GetMessageService(req)
if(result['code'] === 0){
console.log(result['data']);
result = await GetMessageService(req);
if (result["code"] === 0) {
console.log(result["data"]);
}
} catch (e) {
console.log(e);
}
return {};
}
};
</script>
<style scoped>
@ -188,31 +213,76 @@ hljs.registerLanguage('shell', shell);
display: flex;
flex-direction: column;
height: 100%;
padding: 20px;
box-sizing: border-box;
}
.chat-messages {
flex: 1;
overflow-y: auto;
overflow-y: auto; /* 允许垂直滚动 */
padding: 10px;
border: 1px solid #ccc;
margin-bottom: 10px;
height: 80%;
}
.user-message {
text-align: right;
background-color: #ccc;
margin-bottom: 10px;
}
.ai-message {
text-align: left;
max-width: 60%;
margin-bottom: 10px;
}
.input-container {
margin-bottom: 20px;
position: relative;
display: flex;
gap: 10px;
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%;
}
</style>

View File

@ -6,6 +6,7 @@
</template>
<script lang="ts">
import { ref, reactive } from 'vue';
</script>
<style lang="scss" scoped>