save
This commit is contained in:
parent
f4ea3bf2ff
commit
11f98b4bc6
@ -1,63 +1,59 @@
|
|||||||
<template>
|
<template>
|
||||||
<div class="chat-container">
|
<div class="chat-container" ref="chatContainerRef">
|
||||||
<el-container class="full-height">
|
<el-scrollbar :height="scrollHeight">
|
||||||
<el-header class="chat-header">
|
<div class="message-list">
|
||||||
<h2>AI 对话</h2>
|
<div v-for="(message, index) in messages" :key="index" :class="['message-item', message.role]">
|
||||||
</el-header>
|
<div class="message-avatar">
|
||||||
|
<el-avatar :icon="message.role === 'user' ? User : Comment" />
|
||||||
|
</div>
|
||||||
|
<div class="message-content">
|
||||||
|
<div class="message-role">{{ message.role === 'user' ? '你' : 'AI' }}</div>
|
||||||
|
<div class="message-text">{{ message.content }}</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
<el-main class="chat-main">
|
<div v-if="isLoading" class="message-item assistant">
|
||||||
<div class="message-list">
|
<div class="message-avatar">
|
||||||
<div v-for="(message, index) in messages" :key="index" :class="['message-item', message.role]">
|
<el-avatar :icon="Comment" />
|
||||||
<div class="message-avatar">
|
</div>
|
||||||
<el-avatar :icon="message.role === 'user' ? User : Comment" />
|
<div class="message-content">
|
||||||
</div>
|
<div class="message-role">AI</div>
|
||||||
<div class="message-content">
|
<div class="message-text">{{ streamingContent }}<span class="typing-cursor">|</span></div>
|
||||||
<div class="message-role">{{ message.role === 'user' ? '你' : 'AI' }}</div>
|
</div>
|
||||||
<div class="message-text">{{ message.content }}</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</el-scrollbar>
|
||||||
|
|
||||||
<div v-if="isLoading" class="message-item assistant">
|
<el-footer class="chat-footer" ref="footerRef">
|
||||||
<div class="message-avatar">
|
<div class="input-area">
|
||||||
<el-avatar :icon="Comment" />
|
<el-input v-model="userInput" type="textarea" :rows="3" placeholder="输入消息..." :disabled="isLoading"
|
||||||
</div>
|
@keydown.enter.prevent="handleSend" />
|
||||||
<div class="message-content">
|
<el-button type="primary" :loading="isLoading" @click="handleSend" class="send-button">
|
||||||
<div class="message-role">AI</div>
|
发送
|
||||||
<div class="message-text">{{ streamingContent }}<span class="typing-cursor">|</span></div>
|
</el-button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</el-footer>
|
||||||
</div>
|
</div>
|
||||||
</el-main>
|
|
||||||
|
|
||||||
<el-footer class="chat-footer">
|
|
||||||
<div class="input-area">
|
|
||||||
<el-input v-model="userInput" type="textarea" :rows="3" placeholder="输入消息..." :disabled="isLoading"
|
|
||||||
@keydown.enter.prevent="handleSend" />
|
|
||||||
<el-button type="primary" :loading="isLoading" @click="handleSend" class="send-button">
|
|
||||||
发送
|
|
||||||
</el-button>
|
|
||||||
</div>
|
|
||||||
</el-footer>
|
|
||||||
</el-container>
|
|
||||||
</div>
|
|
||||||
</template>
|
</template>
|
||||||
|
|
||||||
<script setup lang="ts">
|
<script setup lang="ts">
|
||||||
import { ref, onMounted, defineComponent, defineProps } from 'vue';
|
import { ref, onMounted, defineComponent, defineProps, onUnmounted } from 'vue';
|
||||||
import { User, Comment } from '@element-plus/icons-vue';
|
import { User, Comment } from '@element-plus/icons-vue';
|
||||||
import { useMessageBridge } from "@/api/message-bridge";
|
import { useMessageBridge } from "@/api/message-bridge";
|
||||||
|
import { ElMessage } from 'element-plus';
|
||||||
|
|
||||||
defineComponent({ name: 'chat' });
|
defineComponent({ name: 'chat' });
|
||||||
|
|
||||||
const props = defineProps({
|
const props = defineProps({
|
||||||
tabId: {
|
tabId: {
|
||||||
type: Number,
|
type: Number,
|
||||||
required: true
|
required: true
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
interface ChatMessage {
|
interface ChatMessage {
|
||||||
role: 'user' | 'assistant';
|
role: 'user' | 'assistant';
|
||||||
content: string;
|
content: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
const bridge = useMessageBridge();
|
const bridge = useMessageBridge();
|
||||||
@ -65,155 +61,170 @@ const userInput = ref('');
|
|||||||
const messages = ref<ChatMessage[]>([]);
|
const messages = ref<ChatMessage[]>([]);
|
||||||
const isLoading = ref(false);
|
const isLoading = ref(false);
|
||||||
const streamingContent = ref('');
|
const streamingContent = ref('');
|
||||||
|
const chatContainerRef = ref<HTMLElement>();
|
||||||
|
const footerRef = ref<HTMLElement>();
|
||||||
|
const scrollHeight = ref('500px');
|
||||||
|
|
||||||
|
const updateScrollHeight = () => {
|
||||||
|
if (chatContainerRef.value && footerRef.value) {
|
||||||
|
const containerHeight = chatContainerRef.value.clientHeight;
|
||||||
|
const footerHeight = footerRef.value.clientHeight;
|
||||||
|
scrollHeight.value = `${containerHeight - footerHeight}px`;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
const handleSend = () => {
|
const handleSend = () => {
|
||||||
if (!userInput.value.trim() || isLoading.value) return;
|
if (!userInput.value.trim() || isLoading.value) return;
|
||||||
|
|
||||||
const userMessage = userInput.value.trim();
|
const userMessage = userInput.value.trim();
|
||||||
messages.value.push({ role: 'user', content: userMessage });
|
messages.value.push({ role: 'user', content: userMessage });
|
||||||
|
|
||||||
const chatData = {
|
const chatData = {
|
||||||
messages: [
|
messages: [
|
||||||
...messages.value.filter(msg => msg.role === 'user').map(msg => ({
|
...messages.value.filter(msg => msg.role === 'user').map(msg => ({
|
||||||
role: msg.role,
|
role: msg.role,
|
||||||
content: msg.content
|
content: msg.content
|
||||||
})),
|
})),
|
||||||
{ role: 'assistant', content: streamingContent.value }
|
{ role: 'assistant', content: streamingContent.value }
|
||||||
]
|
]
|
||||||
};
|
};
|
||||||
|
|
||||||
isLoading.value = true;
|
isLoading.value = true;
|
||||||
streamingContent.value = '';
|
streamingContent.value = '';
|
||||||
|
|
||||||
bridge.postMessage({
|
bridge.postMessage({
|
||||||
command: 'llm/chat/completions',
|
command: 'llm/chat/completions',
|
||||||
data: chatData
|
data: chatData
|
||||||
});
|
});
|
||||||
|
|
||||||
userInput.value = '';
|
userInput.value = '';
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleError = (errorMsg: string) => {
|
||||||
|
ElMessage.error(errorMsg);
|
||||||
|
messages.value.push({
|
||||||
|
role: 'assistant',
|
||||||
|
content: `错误: ${errorMsg}`
|
||||||
|
});
|
||||||
|
streamingContent.value = '';
|
||||||
|
isLoading.value = false;
|
||||||
};
|
};
|
||||||
|
|
||||||
onMounted(() => {
|
onMounted(() => {
|
||||||
bridge.addCommandListener('llm/chat/completions/chunk', data => {
|
updateScrollHeight();
|
||||||
const { content } = data.msg;
|
window.addEventListener('resize', updateScrollHeight);
|
||||||
if (content) {
|
|
||||||
streamingContent.value += content;
|
|
||||||
}
|
|
||||||
}, { once: false });
|
|
||||||
|
|
||||||
bridge.addCommandListener('llm/chat/completions/done', () => {
|
bridge.addCommandListener('llm/chat/completions/chunk', data => {
|
||||||
if (streamingContent.value) {
|
if (data.code !== 200) {
|
||||||
messages.value.push({
|
handleError(data.msg || '请求模型服务时发生错误');
|
||||||
role: 'assistant',
|
return;
|
||||||
content: streamingContent.value
|
}
|
||||||
});
|
const { content } = data.msg;
|
||||||
streamingContent.value = '';
|
if (content) {
|
||||||
}
|
streamingContent.value += content;
|
||||||
isLoading.value = false;
|
}
|
||||||
}, { once: false });
|
}, { once: false });
|
||||||
|
|
||||||
|
bridge.addCommandListener('llm/chat/completions/done', data => {
|
||||||
|
if (data.code !== 200) {
|
||||||
|
handleError(data.msg || '模型服务处理完成但返回错误');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (streamingContent.value) {
|
||||||
|
messages.value.push({
|
||||||
|
role: 'assistant',
|
||||||
|
content: streamingContent.value
|
||||||
|
});
|
||||||
|
streamingContent.value = '';
|
||||||
|
}
|
||||||
|
isLoading.value = false;
|
||||||
|
}, { once: false });
|
||||||
|
});
|
||||||
|
|
||||||
|
onUnmounted(() => {
|
||||||
|
window.removeEventListener('resize', updateScrollHeight);
|
||||||
});
|
});
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<style scoped>
|
<style scoped>
|
||||||
.chat-container {
|
.chat-container {
|
||||||
height: fit-content;
|
height: 100%;
|
||||||
display: flex;
|
display: flex;
|
||||||
flex-direction: column;
|
flex-direction: column;
|
||||||
}
|
|
||||||
|
|
||||||
.full-height {
|
|
||||||
height: 100%;
|
|
||||||
}
|
|
||||||
|
|
||||||
.chat-header {
|
|
||||||
border-bottom: 1px solid var(--el-border-color);
|
|
||||||
display: flex;
|
|
||||||
align-items: center;
|
|
||||||
}
|
|
||||||
|
|
||||||
.chat-main {
|
|
||||||
flex: 1;
|
|
||||||
overflow-y: auto;
|
|
||||||
padding: 16px;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
.message-list {
|
.message-list {
|
||||||
max-width: 800px;
|
max-width: 800px;
|
||||||
margin: 0 auto;
|
margin: 0 auto;
|
||||||
|
padding: 16px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.message-item {
|
.message-item {
|
||||||
display: flex;
|
display: flex;
|
||||||
margin-bottom: 16px;
|
margin-bottom: 16px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.message-avatar {
|
.message-avatar {
|
||||||
margin-right: 12px;
|
margin-right: 12px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.message-content {
|
.message-content {
|
||||||
flex: 1;
|
flex: 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
.message-role {
|
.message-role {
|
||||||
font-weight: bold;
|
font-weight: bold;
|
||||||
margin-bottom: 4px;
|
margin-bottom: 4px;
|
||||||
color: var(--el-text-color-regular);
|
color: var(--el-text-color-regular);
|
||||||
}
|
}
|
||||||
|
|
||||||
.message-text {
|
.message-text {
|
||||||
line-height: 1.6;
|
line-height: 1.6;
|
||||||
white-space: pre-wrap;
|
white-space: pre-wrap;
|
||||||
}
|
}
|
||||||
|
|
||||||
.user {
|
.user {
|
||||||
flex-direction: row-reverse;
|
flex-direction: row-reverse;
|
||||||
text-align: right;
|
text-align: right;
|
||||||
}
|
}
|
||||||
|
|
||||||
.user .message-avatar {
|
.user .message-avatar {
|
||||||
margin-right: 0;
|
margin-right: 0;
|
||||||
margin-left: 12px;
|
margin-left: 12px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.user .message-content {
|
.user .message-content {
|
||||||
align-items: flex-end;
|
align-items: flex-end;
|
||||||
}
|
}
|
||||||
|
|
||||||
.assistant {
|
.assistant {
|
||||||
text-align: left;
|
text-align: left;
|
||||||
}
|
}
|
||||||
|
|
||||||
.chat-footer {
|
.chat-footer {
|
||||||
padding: 16px;
|
padding: 16px;
|
||||||
border-top: 1px solid var(--el-border-color);
|
border-top: 1px solid var(--el-border-color);
|
||||||
|
flex-shrink: 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
.input-area {
|
.input-area {
|
||||||
max-width: 800px;
|
max-width: 800px;
|
||||||
margin: 0 auto;
|
margin: 0 auto;
|
||||||
display: flex;
|
display: flex;
|
||||||
gap: 12px;
|
gap: 12px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.send-button {
|
.send-button {
|
||||||
align-self: flex-end;
|
align-self: flex-end;
|
||||||
height: 72px;
|
height: 72px;
|
||||||
}
|
}
|
||||||
|
|
||||||
.typing-cursor {
|
.typing-cursor {
|
||||||
animation: blink 1s infinite;
|
animation: blink 1s infinite;
|
||||||
}
|
}
|
||||||
|
|
||||||
@keyframes blink {
|
@keyframes blink {
|
||||||
|
0%, 100% { opacity: 1; }
|
||||||
0%,
|
50% { opacity: 0; }
|
||||||
100% {
|
|
||||||
opacity: 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
50% {
|
|
||||||
opacity: 0;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
</style>
|
</style>
|
||||||
@ -58,7 +58,7 @@ export async function chatCompletionHandler(client: MCPClient | undefined, data:
|
|||||||
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
webview.postMessage({
|
webview.postMessage({
|
||||||
command: 'llm/chat/completions/error',
|
command: 'llm/chat/completions/chunk',
|
||||||
data: {
|
data: {
|
||||||
code: 500,
|
code: 500,
|
||||||
msg: `OpenAI API error: ${(error as Error).message}`
|
msg: `OpenAI API error: ${(error as Error).message}`
|
||||||
|
|||||||
@ -7,6 +7,15 @@
|
|||||||
"type": "blank",
|
"type": "blank",
|
||||||
"componentIndex": 3,
|
"componentIndex": 3,
|
||||||
"storage": {}
|
"storage": {}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "资源",
|
||||||
|
"icon": "icon-file",
|
||||||
|
"type": "blank",
|
||||||
|
"componentIndex": 0,
|
||||||
|
"storage": {
|
||||||
|
"currentResourceName": "greeting"
|
||||||
|
}
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
Loading…
x
Reference in New Issue
Block a user