mirror of
https://github.com/wanglin2/mind-map.git
synced 2026-02-17 14:04:47 +08:00
82 lines
2.0 KiB
JavaScript
82 lines
2.0 KiB
JavaScript
const express = require('express')
|
|
const http = require('http')
|
|
|
|
const port = 3000
|
|
const baseUrl = 'http://ark.cn-beijing.volces.com'
|
|
|
|
// 起个服务
|
|
const app = express()
|
|
app.use(express.json())
|
|
app.use(express.urlencoded({ extended: true }))
|
|
|
|
// 允许跨域
|
|
app.use((req, res, next) => {
|
|
res.header('Access-Control-Allow-Origin', '*') // 允许所有来源的跨域请求，或者指定一个域名
|
|
res.header('Access-Control-Allow-Methods', '*') // 允许的方法
|
|
res.header('Access-Control-Allow-Headers', '*') // 允许的头部信息
|
|
next()
|
|
})
|
|
|
|
// 监听对话请求
|
|
app.get('/ai/test', (req, res) => {
|
|
res.send('/ai/test').end()
|
|
})
|
|
app.post('/ai/chat', (req, res) => {
|
|
try {
|
|
const { a: apiKey, b: model, messages } = req.body
|
|
|
|
// res.append('Content-Type', 'text/event-stream')
|
|
// res.append('Cache-Control', 'no-cache')
|
|
// res.append('Connection', 'keep-alive')
|
|
res.send(1).end(200)
|
|
} catch (error) {
|
|
console.log(error)
|
|
}
|
|
|
|
// 模拟发送数据
|
|
// const intervalId = setInterval(() => {
|
|
// res.send(`data: ${new Date().toISOString()}\n\n`)
|
|
// }, 1000)
|
|
|
|
// // 监听客户端断开连接
|
|
// req.on('close', () => {
|
|
// console.log('Client disconnected.')
|
|
// clearInterval(intervalId)
|
|
// res.end()
|
|
// })
|
|
|
|
// const aiReq = http.request(
|
|
// baseUrl + '/api/v3/chat/completions',
|
|
// {
|
|
// method: 'POST',
|
|
// headers: {
|
|
// Authorization: 'Bearer ' + apiKey
|
|
// }
|
|
// },
|
|
// aiRes => {
|
|
// aiRes.on('data', chunk => {
|
|
// console.log(`BODY: ${chunk}`)
|
|
// res.send(chunk)
|
|
// })
|
|
// aiRes.on('end', () => {
|
|
// console.log('No more data in response.')
|
|
// res.end()
|
|
// })
|
|
// }
|
|
// )
|
|
// const postData = {
|
|
// model,
|
|
// messages,
|
|
// stream: true
|
|
// }
|
|
// aiReq.write(JSON.stringify(postData))
|
|
// aiReq.end()
|
|
})
|
|
|
|
// res.writeHead(404)
|
|
// res.end()
|
|
|
|
app.listen(port, () => {
|
|
console.log(`app listening on port ${port}`)
|
|
})
|