基于Agent的金融问答系统:前后端流程打通 原创
前言
在上一章《【项目实战】基于Agent的金融问答系统:Agent框架的构建》中,我们已经完成了基于agent的问答系统主流程构建,本章将介绍如何将Agent通过langserve部署为后端服务,同时基于Vue.js+Vite构建前端页面,完成问答系统前后端联调。
目标
- 将Agent部署为后端服务
- 构建前端页面,完成问答系统前后端联调
后端实现
代码实现
基于《大模型应用开发之Prompt初步了解》中langserve的所学内容,我们需要做的第一步为: 1、安装langserve依赖包:
pip install "langserve[all]"
2、创建一个server.py文件,具体如下;
代码文件:app/server.py
from fastapi importFastAPI,HTTPException
from fastapi.middleware.cors importCORSMiddleware
from finance_bot_ex importFinanceBotEx
finance_bot_ex =FinanceBotEx()
# 创建 FastAPI 应用
app =FastAPI(
title="Qwen API",
version="0.1",
description="Qwen API",
)
# 添加 CORS 中间件
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],# 允许所有的来源
allow_credentials=True,
allow_methods=["*"],# 允许的HTTP方法
allow_headers=["*"],# 允许的请求头
)
@app.post("/queryex", response_model=dict)
asyncdefquery(query: dict):# 使用字典类型代替Query模型
try:
# 从字典中获取input
input_data = query.get("input")
result = finance_bot_ex.handle_query(input_data)
# 返回字典格式的响应
return{"output": result}
exceptExceptionas e:
raiseHTTPException(status_code=500, detail=str(e))
# 运行Uvicorn服务器
if __name__ =="__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=8082)
代码说明:
- 首先,依赖包引入langserve的必要包:
from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware - 其次,将我们之前封装好的FinanceBotEx类引入进来:
from finance_bot_ex import FinanceBotEx
- 然后,初始化FinanceBotEx类,并且创建FastAPI应用:
# 实例化FinanceBotEx类
finance_bot_ex =FinanceBotEx()
# 创建 FastAPI 应用
app =FastAPI(
title="Qwen API",
version="0.1",
description="Qwen API",
)
# 添加 CORS 中间件(允许跨域访问)
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],# 允许所有的来源
allow_credentials=True,
allow_methods=["*"],# 允许的HTTP方法
allow_headers=["*"],# 允许的请求头
) - 最后,给定一个接口,接收用户输入,调用FinanceBotEx类的handle_query方法,返回结果: ```python @app.post("/queryex", response_model=dict) async def query(query: dict): # 使用字典类型代替Query模型 try: # 从字典中获取input input_data = query.get("input") result = finance_bot_ex.handle_query(input_data) # 返回字典格式的响应 return {"output": result} except Exception as e: raise HTTPException(status_code=500, detail=str(e))
### 服务部署
目前,整个工程目录结构如下:
```bash
smart-finance-bot \
|- app \ # 该目录用于服务端代码
|- conf
|- .qwen
|- utils \
|- util.py # 实现连接大模型的方法
|- rag \
|- pdf_processor.py
|- chroma_conn.py
|- test_framework.py
|- finance_bot_ex.py # 封装好的FinanceBotEx类
|- server.py # 新增的服务代码
启动后端服务方法
# 切换至项目根目录
cd smart-finance-bot
# 启动服务
python app/server.py
运行结果:
接口测试
服务正常启动后,我们通过curl命令测试后端接口正常
curl -X POST http://localhost:8082/queryex \
-H "Content-Type: application/json" \
-d '{
"config": {},
"input": "在20190211,按照中信行业分类的标准,一级行业中的\"A股公司数量\"最多的是\"机械\"行业,共有549家公司。"
}'
运行结果:
后端接口可以接收请求,服务一切正常,接下来我们开始构建前端页面,完成问答系统前后端联调。
前端实现
前端页面的实现方案有很多种,可以写个简单的HTML页面,也可以使用Vue.js框架来实现一个前端。 因为Vue.js是目前主流的前端框架,它有丰富的生态和API说明,在github上也有非常多的开源项目供参考,所以本例中我们将使用Vue.js框架来实现前端页面。
背景知识
Vue.js简介 Vue.js 是一个用于构建用户界面的渐进式 JavaScript 框架。它的核心库专注于视图层,易于上手,并与其他库或现有项目进行整合。
Vue.js的特点
- 响应式数据绑定:Vue.js 提供了双向数据绑定,能够实时更新视图与数据之间的变化。
- 指令:Vue.js 提供了一些内置指令(如 v-bind、v-model、v-for 等),使得数据绑定和 DOM 操作变得简单直观。
- 生态系统:Vue.js 拥有丰富的生态系统,包括路由管理(Vue Router)、状态管理(Vuex)等工具,支持构建复杂的应用。
Vue.js的简单示例
<div id="app">
<h1>{{ message }}</h1>
<input v-model="message" placeholder="编辑我">
</div>
<script src="https://cdn.jsdelivr.net/npm/vue@2"></script>
<script>
newVue({
el:'#app',
data:{
message:'Hello Vue!'
}
});
</script>
说明:
- 在上面的示例中,{{ message }} 用于显示数据,v-model 用于实现双向数据绑定,允许用户通过输入框修改 message 的内容。
- 更为详细的Vue.js使用方法,请查看官方文档地址:https://cn.vuejs.org/
代码实现
在本例中,我们参考Github上的一个开源项目进行二次开发,该项目git clone之后目录如下:
chatweb \
|- index.html # 前端入口文件
|- node_modules # 依赖包目录
|- package-lock.json # 锁定依赖版本的文件
|- package.json # 项目描述和依赖配置文件
|- postcss.config.js # PostCSS 配置文件
|- public # 公共资源目录
|- src # 源代码目录
|-App.vue # 主应用组件
|- assets # 资源文件目录
|- components # 组件目录
|-ChatComponents.vue # 聊天组件
|-Home.vue # 首页组件
|-Login.vue # 登录组件
|-NotFound.vue # 404 页面组件
|- main.js # 应用入口文件
|- routes # 路由配置目录
|- tailwind.config.js # Tailwind CSS 配置文件
|- tsconfig.json # TypeScript 配置文件
|- vite.config.js # Vite 配置文件
该项目是仿chatGPT的问答系统,仓库地址为:SinMu-L/chatweb
修改index.html
编辑index.html,修改如下:
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<link rel="icon" href="./src/assets/icon.jpg">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>金融千问机器人</title>
</head>
<body>
<div id="app"></div>
<script type="module" src="/src/main.js"></script>
</body>
</html>
修改路由配置文件
编辑src/routes/routers.js,修改如下:
import ContainerTemplatefrom'../components/ChatComponents.vue'
importNotFoundfrom'../components/NotFound.vue'
importHomefrom'../components/Home.vue'
var routes =[
{path:"/",name:'root',component:Home},
{path:"/chat/:uuid?",component:ContainerTemplate,name:'chat'},
{path:'/404',name:'not_found',component:NotFound}// 捕获所有未匹配的路径
];
exportdefault routes;
说明:
- 配置文件用于控制页面请求的路由信息
- 当用户访问的路径为/时,将显示Home组件;
- 用户访问的路径为/chat时,将显示ChatComponents组件;
- 当用户访问的路径为其他路径,将显示404页面。
由上也可以看到,后续修改的对话页面在 ChatComponents.vue
中。
修改聊天组件页面
由于原始的 ChatComponents.vue
代码不够规范,显示布局、显示样式以及逻辑代码没有分离,所以我们对其进行重构,重构内容分为三部分:
重构页面的显示布局
<template>
<div>
<div v-if="centerLodding" class="loading-container">
<n-spin size="large" />
</div>
<Login class="border border-red-400" />
<div class="flex flex-row h-min" :class="hasLogin('main')">
<!-- 非移动端模式下侧边栏的样式 -->
<div class="sidebar" :class="controlSidebarHidden ? 'w-0' : ''">
<div class="hidden sm:flex sm:flex-col sm:h-screen sm:border">
<!-- 新建按钮 -->
<div class="basis-1/12 flex justify-center items-center">
<n-button class="w-4/5" @click="addLeftListEle">新的会话</n-button>
</div>
<!-- 列表 -->
<div class="basis-10/12 overflow-auto border">
<div v-for="item in left_data.left_list" :key="item.uuid">
<router-link :to="`/chat/${item.uuid}`" class="sidebar-item">
<div class="w-4/5 flex items-center">
<n-icon size="medium">
<game-controller-outline />
</n-icon>
<div class="truncate mx-2" style="color: #000;">
<p v-if="!item.enable_edit" class="truncate h-full">{{ item.title }}</p>
<n-input v-else type="text" size="small" class="h-full" v-model:value="item.title" @keyup.enter="submit(item.uuid)" />
</div>
</div>
<div class="w-1/5 flex justify-center items-center" :class="route.params.uuid != item.uuid ? 'hidden' : ''">
<n-button-group size="small">
<n-button text @click="editLeftListEle(item.uuid)">
<n-icon><Edit /></n-icon>
</n-button>
<n-button text @click="delLeftListEle(item.uuid)">
<n-icon><Delete /></n-icon>
</n-button>
</n-button-group>
</div>
</router-link>
</div>
</div>
<!-- 设置页面 -->
<div class="footer flex flex-col items-center justify-between p-4 h-15">
<div class="user-info font-bold text-lg">金融千问机器人</div>
<div class="logo-container relative"><!-- 添加一个容器 -->
<img src="../assets/background.png" alt="Logo" class="logo" />
</div>
<div class="robot-description text-sm text-gray-600 mt-2">
金融千问机器人,通过RAG对既有的PDF招股书建立了知识库,同时借助大模型+Agent对金融SQL数据库进行动态查询,旨在为用户提供快速、准确的金融信息和咨询服务。
</div>
<div class="settings-icon">
<n-icon @click="showSettingFunc()">
<SettingsOutline />
</n-icon>
</div>
</div>
</div>
</div>
<!-- 移动端模式下侧边栏的样式 -->
<div class="sm:hidden absolute top-1 left-1 h-full w-full flex flex-col">
<div>
<n-button text style="font-size:32px" @click="controlSidebarHidden = !controlSidebarHidden">
<n-icon class="text-black">
<Menu />
</n-icon>
</n-button>
</div>
<div v-if="!controlSidebarHidden" class="mobile-sidebar">
<div class="w-full flex flex-col h-screen border">
<div class="basis-1/12 flex justify-center items-center">
<n-button class="w-4/5" @click="addLeftListEle">新的对话</n-button>
</div>
<div class="basis-10/12 overflow-auto border">
<div v-for="item in left_data.left_list" :key="item.uuid">
<router-link :to="`/chat/${item.uuid}`" class="sidebar-item">
<div class="w-4/5 flex items-center">
<n-icon size="medium">
<game-controller-outline />
</n-icon>
<div class="truncate mx-2">
<p v-if="!item.enable_edit" class="truncate h-full">{{ item.title }}</p>
<n-input v-else type="text" size="small" class="h-full" v-model:value="item.title" @keyup.enter="submit(item.uuid)" />
</div>
</div>
<div class="w-1/5 flex justify-center items-center" :class="route.params.uuid != item.uuid ? 'hidden' : ''">
<n-button-group size="small">
<n-button text @click="editLeftListEle(item.uuid)">
<n-icon><Edit /></n-icon>
</n-button>
<n-button text @click="delLeftListEle(item.uuid)">
<n-icon><Delete /></n-icon>
</n-button>
</n-button-group>
</div>
</router-link>
</div>
</div>
<div class="footer flex items-center justify-between p-4 h-15">
<div class="user-info font-bold">金融千问机器人</div>
<div class="settings-icon">
<n-icon @click="showSettingFunc()">
<SettingsOutline />
</n-icon>
</div>
</div>
</div>
</div>
</div>
<div class="w-full sm:w-4/5 h-full">
<div class="flex flex-col h-screen">
<div v-if="left_list_is_empty" class="empty-chat-message">
Hi
</div>
<div v-else class="chat-area" id="msgArea">
<div v-for="(msglist, index) in getMsgList(route.params.uuid)" :key="index" class="flex flex-col mt-1 msgItem">
<div :class="msglist.reversion ? 'flex-row-reverse' : 'flex-row'" class="flex justify-start items-center h-10">
<img
class="rounded-full avatar"
:src="msglist.reversion ? userAvatar : robotAvatar"
alt="头像"
/>
<span class="ml-4 text-sm">{{ msglist.create_time }}</span>
</div>
<div class="flex" :class="msglist.reversion ? 'flex-row-reverse' : 'flex-row'">
<div class="message-container">
<n-spin v-if="msglist.msgload" size="small" stroke="red" />
<Markdown v-else :source="msglist.content"></Markdown>
</div>
</div>
</div>
</div>
<div v-if="!left_list_is_empty" class="input-area">
<div class="p-2">
<n-input-group>
<n-tooltip trigger="hover">
<template #trigger>
<n-button text size="large" class="px-2" @click="deleteChatItemHistory(route.params.uuid)">
<n-icon class="text-black">
<Delete />
</n-icon>
</n-button>
</template>
删除当前会话记录
</n-tooltip>
<n-tooltip trigger="hover">
<template #trigger>
<n-button text size="large" class="pr-4" @click="dom2img()">
<n-icon class="text-black">
<Download />
</n-icon>
</n-button>
</template>
下载当前会话为图片
</n-tooltip>
<a href="" id="link" class="hidden"></a>
<n-input show-count @keyup.ctrl.enter="addMessageListItem(route.params.uuid)" placeholder="Ctrl+Enter 发送消息,发送消息长度需要大于2个字" v-model:value="input_area_value" type="textarea" size="tiny" :autosize="{ minRows: 2, maxRows: 5 }" />
<n-button ghost class="h-auto" @click="addMessageListItem(route.params.uuid)">
发送
</n-button>
</n-input-group>
</div>
</div>
</div>
</div>
</div>
<!-- 设置模态框 -->
<n-modal v-model:show="showSetting" style="width: 600px" class="custom-card" preset="card" title="设置">
<n-tabs type="line" animated>
<n-tab-pane name="about" tab="关于">
<div>这是一个demo项目,仅用于学习。</div>
<div class="my-4">技术栈:Vue3 + Vite + tailwindCss3 + NaiveUi</div>
</n-tab-pane>
<n-tab-pane name="settings" tab="设置">
<div class="grid grid-rows-3 gap-4">
<div>
<span class="mr-4">Server URL: </span>
<n-input v-model:value="setting.server_url" placeholder="请输入服务器 URL" />
</div>
<div>
<span class="mr-4">type: </span>
<n-select :style="{ width: '80%' }" :options="selectOptions" v-model:value="setting.type" />
</div>
<div class="flex justify-end">
<n-button
@click="confirmSettings"
class="custom-button">
确认
</n-button>
</div>
</div>
</n-tab-pane>
<n-tab-pane name="other" tab="其他">
其他
</n-tab-pane>
</n-tabs>
</n-modal>
</div>
</template>
重构页面的显示样式
<style scoped>
.router-link-active{
border-color:#18a058;
color:#18a058;
}
.loading-container{
position: absolute;
top:33%;
left:33%;
background-color: gray;
width:33%;
height:33%;
display: flex;
justify-content: center;
align-items: center;
}
.sidebar{
width:20%;
height:100%;
}
.sidebar-item{
margin:0.5rem;
display: flex;
justify-content: space-between;
align-items: center;
border:1px solid gray;
border-radius:0.375rem;/* rounded-md */
padding:0.5rem;
color: white;
}
.settings-container{
display: flex;
justify-content: flex-start;
align-items: center;
height:50%;
}
.user-info{
width:50%;
height:100%;
display: grid;
grid-template-rows:repeat(2,1fr);
}
.settings-icon{
width:25%;
height:100%;
display: flex;
justify-content: center;
align-items: center;
}
.mobile-sidebar{
width:80%;
background-color: white;
dark:bg-black;
}
.empty-chat-message{
flex-basis:91%;
width:100%;
padding:3rem;
overflow:auto;
display: flex;
justify-content: center;
align-items: flex-start;
color: gray;
font-style: italic;
}
.chat-area{
flex-basis:91%;
width:100%;
padding:3rem;
overflow:auto;
}
.message-container{
background-color:#bfdbfe; /* bg-blue-200 */
color: black;/* dark:text-black */
width:auto;
max-width:80%;
min-width:1%;
break-words:break-word;
overflow: hidden;
border-radius:0.375rem;/* rounded-sm */
padding:0.5rem;
margin:0.25rem0;
}
.input-area{
flex-basis:8%;
width:100%;
}
.avatar{
width:50px;
height:50px;
}
.footer{
text-align: center;/* 居中显示 */
}
.robot-description{
margin-top:8px;/* 描述与 Logo 之间的间距 */
text-align: left;/* 左对齐 */
width:100%;/* 使描述占满可用宽度 */
padding-left:16px;/* 可选:增加左侧内边距 */
}
.logo-container{
position: relative;/* 设置容器为相对定位 */
}
.background-logo{
position: absolute;/* 设置背景图为绝对定位 */
top:0;/* 顶部对齐 */
left:0;/* 左侧对齐 */
width:240px;/* 设置背景图宽度为240px */
height:auto;/* 保持比例 */
z-index:1;/* 确保背景图在logo下方 */
}
.logo{
position: relative;/* 设置logo为相对定位 */
z-index:2;/* 确保logo在背景图上方 */
width:200px;/* 根据需要调整 Logo 大小 */
height:auto;/* 保持比例 */
top:30%;/* 垂直居中 */
left:50%;/* 水平居中 */
transform:translate(-50%,-50%);/* 使 logo 在背景图中心 */
}
.custom-button{
background-color:#007bff; /* 按钮背景色 */
color: white;/* 字体颜色 */
border: none;/* 去掉边框 */
padding:10px20px;/* 内边距 */
border-radius:5px;/* 圆角 */
cursor: pointer;/* 鼠标样式 */
transition: background-color 0.3s;/* 添加过渡效果 */
}
.custom-button:hover{
background-color:#0056b3; /* 悬停时的背景颜色 */
}
</style>
重构页面的JS代码
<script setup>
import{
NButton,NInput,NIcon,NButtonGroup,NSpin,
NInputGroup,NCard,NModal,NTabs,NTabPane,NInputNumber,NSelect,
NTooltip,
useMessage
}from'naive-ui'
import{GameControllerOutline,GameController}from'@vicons/ionicons5'
import{LogInOutlineasLogInIcon,SettingsOutline,Menu}from'@vicons/ionicons5'
import userAvatar from'@/assets/user-avatar.png';
import robotAvatar from'@/assets/robot-avatar.png';
import{Edit,Delete,Download}from'@vicons/carbon'
importMarkdownfrom'vue3-markdown-it';
importLoginfrom'./Login.vue'
import{ reactive, ref, getCurrentInstance, watch, watchEffect, nextTick }from'vue';
import{ useRouter, useRoute }from'vue-router'
import html2canvas from"html2canvas";
const router =useRouter()
const route =useRoute()
const instaceV =getCurrentInstance()
const message =useMessage()
import{ onMounted }from'vue';
// 在组件的 setup 函数中添加
onMounted(() =>{
// 检查是否有会话记录
if(left_data.chat.length>0){
// 获取最后一个会话的 UUID
const lastChatUuid = left_data.chat[left_data.chat.length-1].uuid;
// 路由跳转到最后一个会话
router.push({name:'chat',params:{uuid: lastChatUuid }});
}else{
// 如果没有会话,设置左侧列表为空
left_list_is_empty.value=true;
}
});
// 控制侧边栏显示隐藏
var controlSidebarHidden =ref(true)
// 移动端下侧边栏显影
const showSetting =ref(false)
varLLM_APIKEY=ref("sk-cwtdeSy4Ownmy6Uh5e9b6a67Fe4c4454A3Dc524876348eB1")
var setting =reactive({
server_url:'http://localhost:8082/query',// 默认值
type:'query',
})
let left_list_is_empty =ref(false)
var is404 =ref(false)
var segmented ={
content:'soft',
footer:'soft'
}
var selectOptions =ref([
{
label:'Query式访问',
value:'query'
},
{
label:'流式访问',
value:'stream_log'
}
])
var centerLodding =ref(false)
var input_area_value =ref('')
var left_data =reactive({
left_list:[
// { uuid: 1, title: 'New Chat1', enable_edit: false },
// { uuid: 2, title: 'New Chat2', enable_edit: false },
],
chat:[
// {
// uuid: 1, msg_list: [
// { content: 'hello1', create_time: '2023-11-09 11:50:23', reversion: false, msgload: false },
// ]
// },
// {
// uuid: 2, msg_list: [
// { content: 'xxx', create_time: '2023-11-09 11:50:23', reversion: false, msgload: false },
// ]
// },
],
})
// 监听响应式数据
watch(left_data,(newValue, oldValue) =>{
if(newValue.chat.length>0){
left_list_is_empty.value=false
}else{
left_list_is_empty.value=true
}
localStorage.setItem('chatweb',JSON.stringify(newValue))
})
// 创建响应式变量后只执行一次输出的需求
watchEffect(() =>{
// 读取 localstorage
const data =localStorage.getItem('chatweb')
if(data){
const history =JSON.parse(data)
left_data.left_list= history.left_list
left_data.chat= history.chat
}else{
left_list_is_empty.value=true
}
})
// 添加侧壁栏item
functionaddLeftListEle(){
const uuid =randomUuid()
left_data.left_list.push({
uuid: uuid,
title:`新的对话${uuid}`,
enable_edit:false
})
left_data.chat.push({
uuid: uuid,
msg_list:[]
})
// 路由跳转到最新的item
router.push({name:'chat',params:{uuid: uuid }})
}
// 点击侧边栏某个item的编辑按钮
functioneditLeftListEle(uuid){
const index = left_data.left_list.findIndex(v => v.uuid== uuid)
left_data.left_list[index].enable_edit=!left_data.left_list[index].enable_edit
// all_data.left_list[index].enable_edit = !all_data.left_list[index].enable_edit
// ls.updateLeftListItemEnableEditButton(uuid)
}
// 点击侧边栏某个item的删除按钮
asyncfunctiondelLeftListEle(uuid){
var index = left_data.left_list.findIndex(v => v.uuid== uuid)
left_data.left_list.splice(index,1)
const chat_index = left_data.chat.findIndex(v => v.uuid== uuid)
left_data.chat.splice(chat_index,1)
if(left_data.chat.length==0){
left_list_is_empty =true
}else{
awaitnextTick()
// 默认跳转到最新的 chat
const last_chat_uuid = left_data.chat[left_data.chat.length-1].uuid
router.push({name:'chat',params:{"uuid": last_chat_uuid }})
}
}
// 获取每个 chat 的msg list
functiongetMsgList(uuid){
if(uuid && uuid !=undefined){
var index = left_data.chat.findIndex(v => v.uuid== uuid)
if(index ==-1){
return[]
}
return left_data.chat[index].msg_list
}
return[]
}
functionrandomUuid(){
var len =9
var uuid ='';
for(let i =0; i < len; i++){
uuid +=Math.floor(Math.random()*10)
}
returnNumber(uuid,10)
}
// 监听侧边栏item的回车事件
functionsubmit(index){
editLeftListEle(index)
}
// 发送消息
functionaddMessageListItem(uuid){
if(input_area_value.value.length<=2){
message.info('内容长度不得小于2')
returnfalse
}
var index = left_data.chat.findIndex(v => v.uuid== uuid)
constnow_t=(newDate()).toLocaleString('sv-SE',{"timeZone":"PRC"})
left_data.chat[index].msg_list.push({
content: input_area_value.value,
create_time:now_t,
reversion:true,
msgload:false
})
const body ={
config:{},
input: input_area_value.value// 按要求的格式拼接
};
// 清空编辑框
input_area_value.value=''
var ele =document.getElementById("msgArea")
ele.scrollTop= ele.scrollHeight+ ele.offsetHeight
// 添加一个占位消息,表示正在加载
left_data.chat[index].msg_list.push({
content:'',
create_time:now_t,
reversion:false,
msgload:true
})
console.info("开始发送消息...")
// 使用编辑框中的 URL 进行请求
const url = setting.server_url;// 获取 URL
if(setting.type==='stream_log'){
startStream(index, body, url);// 传递 URL
}else{
startRequest(index, body, url);// 传递 URL
}
}
functionbuildMessagePromt(index){
const res =[]
left_data.chat[index].msg_list.forEach(v =>{
let role = v.reversion?'user':'assistant'
res.push({
role: role,
content: v.content
})
})
res.pop()
return res
}
asyncfunctionstartRequest(index, body, url){
const key =LLM_APIKEY.value;
let response =null;
try{
response =awaitfetch(url,{
method:"POST",
headers:{
"Content-Type":"application/json",
"Accept":"application/json",// 使用 JSON 作为接受格式
"Authorization":`Bearer ${key}`,// 如果需要 API 密钥
},
mode:"cors",
body:JSON.stringify(body),
});
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].msgload=false;
}catch(error){
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content+=`发生了一些错误:${error.message}`;
returnfalse;
}
if(response.status!==200){
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content+=`发生了一些错误:${response.status}-${response.statusText}`;
returnfalse;
}
try{
const data =await response.json();
if(data && data.output){// 检查数据是否存在并包含 output
const outputText = data.output;
const currentContent = left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content;
if(!currentContent.includes(outputText)){
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content+= outputText;
}
}else{
thrownewError('返回的数据结构不符合预期');
}
}catch(e){
console.error('Error parsing JSON:', e);
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content+='解析响应时发生错误';
}
returntrue;// 返回成功状态
}
asyncfunctionstartStream(index, body, url){
let response =null;
try{
response =awaitfetch(url,{
method:"POST",
headers:{
"Content-Type":"application/json",
"Accept":"text/event-stream",
"Accept-Encoding":"gzip, deflate, br, zstd",
"Accept-Language":"zh-CN,zh;q=0.9"
},
mode:"cors",
body:JSON.stringify(body),
});
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].msgload=false;
}catch(error){
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content+=`发生了一些错误:${error.message}`;
returnfalse;
}
if(response.status!==200){
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content+=`发生了一些错误:${response.status}-${response.statusText}`;
returnfalse;
}
const reader = response.body.getReader();
const decoder =newTextDecoder('utf-8');
let buffer ='';
constreadStream=async()=>{
const{ done, value }=await reader.read();
if(done){
console.log('Stream reading complete');
return;
}
buffer += decoder.decode(value,{stream:true});
let completeData ='';
let separatorIndex;
while((separatorIndex = buffer.indexOf('\n'))!==-1){
completeData = buffer.slice(0, separatorIndex).trim();
buffer = buffer.slice(separatorIndex +1);
if(completeData.startsWith('data: ')){
const jsonString = completeData.slice(6);// 去掉 'data: '
try{
const data =JSON.parse(jsonString);
data.ops.forEach(op =>{
if(op.op==='add'&& op.path.includes('/streamed_output')){
const outputText = op.value;
if(outputText){
const currentContent = left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content;
if(!currentContent.includes(outputText)){
left_data.chat[index].msg_list[left_data.chat[index].msg_list.length-1].content+= outputText;
}
}
}
});
}catch(e){
console.error('Error parsing JSON:', e,'Complete Data:', completeData);
}
}else{
console.log('No valid JSON found in:', completeData);
}
}
returnreadStream();
};
returnreadStream();
}
functionhasLogin(compomentName = 'Login'){
if(compomentName =='login')return instaceV.proxy.hasLogin?'hidden':'';
if(compomentName =='main')return instaceV.proxy.hasLogin?'':'hidden';
}
functionshowSettingFunc(){
showSetting.value=!showSetting.value
}
// 删除当前会话记录
functiondeleteChatItemHistory(uuid){
const index = left_data.chat.findIndex(v => v.uuid== uuid);
left_data.chat[index].msg_list=[]
message.success('当前会话记录已清理')
}
functionconfirmSettings(){
// 这里可以添加保存设置的逻辑
console.log("设置已确认:", setting);
// 例如,你可以关闭模态框
showSetting.value=false;
}
// 当前会话下载为图片
asyncfunctiondom2img(){
centerLodding.value=true
var ele =document.querySelectorAll(".msgItem")
var msgAreaDom =document.getElementById("msgArea")
const width = msgAreaDom.offsetWidth*2
const height = msgAreaDom.scrollHeight*1.5
let canvas1 =document.createElement('canvas');
let context = canvas1.getContext('2d');
canvas1.width= width;
canvas1.height= height;
// 绘制矩形添加白色背景色
context.rect(0,0, width, height);
context.fillStyle="#fff";
context.fill();
let beforeHeight =0
for(let i =0; i < ele.length; i++){
const dom_canvas =awaithtml2canvas(ele[i],{
scrollX:0,
scrollY:0,
height: ele[i].scrollHeight,
width: ele[i].scrollWidth,
})
// var image = dom_canvas.toDataURL("image/png");
context.drawImage(dom_canvas,0, beforeHeight, dom_canvas.width, dom_canvas.height)
beforeHeight = beforeHeight + dom_canvas.height;
}
var image = canvas1.toDataURL("image/png").replace("image/png","image/octet-stream");
var link =document.getElementById("link");
link.setAttribute("download",`chatweb-${(new Date()).getTime()}.png`);
link.setAttribute("href", image);
link.click();
centerLodding.value=false
message.success('图片下载完成')
}
</script>
服务部署
# 切换目录到chatweb目录
cd chatweb
# 安装依赖的组件
npm install
# 启动服务
npm run dev
运行效果:
接口联调
接下来,通过浏览器的开发者工具进行接口调试
- 在浏览器中打开 http://localhost:5173/
- 点击开始按钮切换页面至聊天页面,打开浏览器的开发者工具
- 输入问题后点击发送,观察右侧开发者工具的Console面板,观察请求发送情况,以及返回的响应数据。
具体调试遇到的问题不尽相同,所以本章不再赘述具体调试方式,请读者根据问题自行调试。
待整体调试通过之后,问答系统的前后端即完成主流程。
本文转载自公众号一起AI技术 作者:Dongming