This commit is contained in:
锦恢 2025-04-06 14:23:20 +08:00
parent f4ea3bf2ff
commit 11f98b4bc6
3 changed files with 156 additions and 136 deletions

View File

@ -1,11 +1,6 @@
<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">
<h2>AI 对话</h2>
</el-header>
<el-main class="chat-main">
<div class="message-list"> <div class="message-list">
<div v-for="(message, index) in messages" :key="index" :class="['message-item', message.role]"> <div v-for="(message, index) in messages" :key="index" :class="['message-item', message.role]">
<div class="message-avatar"> <div class="message-avatar">
@ -27,9 +22,9 @@
</div> </div>
</div> </div>
</div> </div>
</el-main> </el-scrollbar>
<el-footer class="chat-footer"> <el-footer class="chat-footer" ref="footerRef">
<div class="input-area"> <div class="input-area">
<el-input v-model="userInput" type="textarea" :rows="3" placeholder="输入消息..." :disabled="isLoading" <el-input v-model="userInput" type="textarea" :rows="3" placeholder="输入消息..." :disabled="isLoading"
@keydown.enter.prevent="handleSend" /> @keydown.enter.prevent="handleSend" />
@ -38,14 +33,15 @@
</el-button> </el-button>
</div> </div>
</el-footer> </el-footer>
</el-container>
</div> </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({
@ -65,6 +61,17 @@ 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;
@ -93,15 +100,36 @@ const handleSend = () => {
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(() => {
updateScrollHeight();
window.addEventListener('resize', updateScrollHeight);
bridge.addCommandListener('llm/chat/completions/chunk', data => { bridge.addCommandListener('llm/chat/completions/chunk', data => {
if (data.code !== 200) {
handleError(data.msg || '请求模型服务时发生错误');
return;
}
const { content } = data.msg; const { content } = data.msg;
if (content) { if (content) {
streamingContent.value += content; streamingContent.value += content;
} }
}, { once: false }); }, { once: false });
bridge.addCommandListener('llm/chat/completions/done', () => { bridge.addCommandListener('llm/chat/completions/done', data => {
if (data.code !== 200) {
handleError(data.msg || '模型服务处理完成但返回错误');
return;
}
if (streamingContent.value) { if (streamingContent.value) {
messages.value.push({ messages.value.push({
role: 'assistant', role: 'assistant',
@ -112,34 +140,23 @@ onMounted(() => {
isLoading.value = false; isLoading.value = false;
}, { once: 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 {
@ -187,6 +204,7 @@ onMounted(() => {
.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 {
@ -206,14 +224,7 @@ onMounted(() => {
} }
@keyframes blink { @keyframes blink {
0%, 100% { opacity: 1; }
0%, 50% { opacity: 0; }
100% {
opacity: 1;
}
50% {
opacity: 0;
}
} }
</style> </style>

View File

@ -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}`

View File

@ -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"
}
} }
] ]
} }