Update src/index.js

src/index.js

@@ -1,159 +1,159 @@
-const express = require('express')
-const bodyParser = require('body-parser')
-const axios = require('axios')
-const app = express()
-const uuid = require('uuid')
-const { uploadImage } = require('./image')
-
-app.use(bodyParser.json())
-// 设置上传文件大小限制
-app.use(bodyParser.json({ limit: '50mb' }))
-app.use(bodyParser.urlencoded({ limit: '50mb', extended: true }))
-
-app.get('/v1/models', async (req, res) => {
-  try {
-    const response = await axios.get('https://chat.qwenlm.ai/api/models',
-      {
-        headers: {
-          "Authorization": req.headers.authorization,
-          "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
-        }
-      })
-    res.json(response.data)
-  } catch (error) {
-    res.status(403)
-      .json({
-        error: "请提供正确的 Authorization token"
-      })
-  }
-})
-
-app.post('/v1/chat/completions', async (req, res) => {
-  if (!req.headers.authorization) {
-    return res.status(403)
-      .json({
-        error: "请提供正确的 Authorization token"
-      })
-  }
-  const messages = req.body.messages
-  let imageId = null
-  const isImageMessage = Array.isArray(messages[messages.length - 1].content) === true && messages[messages.length - 1].content[1].image_url.url
-  if (isImageMessage) {
-    imageId = await uploadImage(messages[messages.length - 1].content[1].image_url.url, req.headers.authorization)
-    messages[messages.length - 1].content[1] = {
-      "type": "image",
-      "image": imageId
-    }
-  }
-
-  const stream = req.body.stream
-
-  const notStreamResponse = async (response) => {
-    const bodyTemplate = {
-      "id": `chatcmpl-${uuid.v4()}`,
-      "object": "chat.completion",
-      "created": new Date().getTime(),
-      "model": req.body.model,
-      "choices": [
-        {
-          "index": 0,
-          "message": {
-            "role": "assistant",
-            "content": response.choices[0].message.content
-          },
-          "finish_reason": "stop"
-        }
-      ],
-      "usage": {
-        "prompt_tokens": JSON.stringify(req.body.messages).length,
-        "completion_tokens": response.choices[0].message.content.length,
-        "total_tokens": JSON.stringify(req.body.messages).length + response.choices[0].message.content.length
-      }
-    }
-    res.json(bodyTemplate)
-  }
-
-  const streamResponse = async (response) => {
-    const id = uuid.v4()
-    const decoder = new TextDecoder('utf-8')
-    let backContent = null
-    response.on('data', (chunk) => {
-      const decodeText = decoder.decode(chunk)
-
-      const lists = decodeText.split('\n').filter(item => item.trim() !== '')
-      for (const item of lists) {
-        const decodeJson = JSON.parse(item.replace(/^data: /, ''))
-        let content = decodeJson.choices[0].delta.content
-
-        if (backContent === null) {
-          backContent = content
-        } else {
-          const temp = content
-          content = content.replace(backContent, '')
-          backContent = temp
-        }
-
-        const StreamTemplate = {
-          "id": `chatcmpl-${id}`,
-          "object": "chat.completion.chunk",
-          "created": new Date().getTime(),
-          "choices": [
-            {
-              "index": 0,
-              "delta": {
-                "content": content
-              },
-              "finish_reason": null
-            }
-          ]
-        }
-        res.write(`data: ${JSON.stringify(StreamTemplate)}\n\n`)
-      }
-    })
-
-    response.on('end', () => {
-      res.write(`data: [DONE]\n\n`)
-      res.end()
-    })
-
-  }
-
-  try {
-    const response = await axios.post('https://chat.qwenlm.ai/api/chat/completions',
-      req.body,
-      {
-        headers: {
-          "Authorization": req.headers.authorization,
-          "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
-        },
-        responseType: stream ? 'stream' : 'json'
-      }
-    )
-    if (stream) {
-      res.set({
-        'Content-Type': 'text/event-stream',
-        'Cache-Control': 'no-cache',
-        'Connection': 'keep-alive',
-      })
-      streamResponse(response.data)
-    } else {
-      res.set({
-        'Content-Type': 'application/json',
-      })
-      notStreamResponse(response.data)
-    }
-
-  } catch (error) {
-    console.log(error)
-    res.status(500)
-      .json({
-        error: "罢工了，不干了!!!"
-      })
-  }
-
-})
-
-app.listen(3000, () => {
-  console.log('Server is running on port 3000')
-})
-
+const express = require('express')
+const bodyParser = require('body-parser')
+const axios = require('axios')
+const app = express()
+const uuid = require('uuid')
+const { uploadImage } = require('./image')
+
+app.use(bodyParser.json())
+// 设置上传文件大小限制
+app.use(bodyParser.json({ limit: '50mb' }))
+app.use(bodyParser.urlencoded({ limit: '50mb', extended: true }))
+
+app.get('/api/v1/models', async (req, res) => {
+  try {
+    const response = await axios.get('https://chat.qwenlm.ai/api/models',
+      {
+        headers: {
+          "Authorization": req.headers.authorization,
+          "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+        }
+      })
+    res.json(response.data)
+  } catch (error) {
+    res.status(403)
+      .json({
+        error: "请提供正确的 Authorization token"
+      })
+  }
+})
+
+app.post('/api/v1/chat/completions', async (req, res) => {
+  if (!req.headers.authorization) {
+    return res.status(403)
+      .json({
+        error: "请提供正确的 Authorization token"
+      })
+  }
+  const messages = req.body.messages
+  let imageId = null
+  const isImageMessage = Array.isArray(messages[messages.length - 1].content) === true && messages[messages.length - 1].content[1].image_url.url
+  if (isImageMessage) {
+    imageId = await uploadImage(messages[messages.length - 1].content[1].image_url.url, req.headers.authorization)
+    messages[messages.length - 1].content[1] = {
+      "type": "image",
+      "image": imageId
+    }
+  }
+
+  const stream = req.body.stream
+
+  const notStreamResponse = async (response) => {
+    const bodyTemplate = {
+      "id": `chatcmpl-${uuid.v4()}`,
+      "object": "chat.completion",
+      "created": new Date().getTime(),
+      "model": req.body.model,
+      "choices": [
+        {
+          "index": 0,
+          "message": {
+            "role": "assistant",
+            "content": response.choices[0].message.content
+          },
+          "finish_reason": "stop"
+        }
+      ],
+      "usage": {
+        "prompt_tokens": JSON.stringify(req.body.messages).length,
+        "completion_tokens": response.choices[0].message.content.length,
+        "total_tokens": JSON.stringify(req.body.messages).length + response.choices[0].message.content.length
+      }
+    }
+    res.json(bodyTemplate)
+  }
+
+  const streamResponse = async (response) => {
+    const id = uuid.v4()
+    const decoder = new TextDecoder('utf-8')
+    let backContent = null
+    response.on('data', (chunk) => {
+      const decodeText = decoder.decode(chunk)
+
+      const lists = decodeText.split('\n').filter(item => item.trim() !== '')
+      for (const item of lists) {
+        const decodeJson = JSON.parse(item.replace(/^data: /, ''))
+        let content = decodeJson.choices[0].delta.content
+
+        if (backContent === null) {
+          backContent = content
+        } else {
+          const temp = content
+          content = content.replace(backContent, '')
+          backContent = temp
+        }
+
+        const StreamTemplate = {
+          "id": `chatcmpl-${id}`,
+          "object": "chat.completion.chunk",
+          "created": new Date().getTime(),
+          "choices": [
+            {
+              "index": 0,
+              "delta": {
+                "content": content
+              },
+              "finish_reason": null
+            }
+          ]
+        }
+        res.write(`data: ${JSON.stringify(StreamTemplate)}\n\n`)
+      }
+    })
+
+    response.on('end', () => {
+      res.write(`data: [DONE]\n\n`)
+      res.end()
+    })
+
+  }
+
+  try {
+    const response = await axios.post('https://chat.qwenlm.ai/api/chat/completions',
+      req.body,
+      {
+        headers: {
+          "Authorization": req.headers.authorization,
+          "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+        },
+        responseType: stream ? 'stream' : 'json'
+      }
+    )
+    if (stream) {
+      res.set({
+        'Content-Type': 'text/event-stream',
+        'Cache-Control': 'no-cache',
+        'Connection': 'keep-alive',
+      })
+      streamResponse(response.data)
+    } else {
+      res.set({
+        'Content-Type': 'application/json',
+      })
+      notStreamResponse(response.data)
+    }
+
+  } catch (error) {
+    console.log(error)
+    res.status(500)
+      .json({
+        error: "罢工了，不干了!!!"
+      })
+  }
+
+})
+
+app.listen(3000, () => {
+  console.log('Server is running on port 3000')
+})
+