Upload 2 files

src/index.js

@@ -53,8 +53,8 @@ app.use(async (ctx, next) => {
 app.use(async (ctx, next) => {
   try {
     await next()
-    // 处理 404 错误
-    if (ctx.status === 404) {
+    // 只有当没有任何路由处理请求时才设置 404
+    if (ctx.status === 404 && !ctx.body) {
       ctx.status = 404
       ctx.body = {
         error: 'Not Found',
@@ -176,12 +176,14 @@ const handleChatCompletions = async (ctx) => {
     })
 
     if (stream == "true" || stream == true) {
+      ctx.status = 200  // 明确设置状态码
       ctx.set({
         'Content-Type': 'text/event-stream',
         'Cache-Control': 'no-cache',
         'Connection': 'keep-alive',
       })
     } else {
+      ctx.status = 200  // 明确设置状态码
       ctx.set({
         'Content-Type': 'application/json',
       })
@@ -225,6 +227,7 @@ const handleChatCompletions = async (ctx) => {
         const { done, value } = await reader.read()
         if (done) {
           if (stream == "true" || stream == true) {
+            console.log('流式响应完成')
             ctx.res.write('data: [DONE]\n\n')
           }
           break
@@ -232,6 +235,7 @@ const handleChatCompletions = async (ctx) => {
 
         if (stream == "true" || stream == true) {
           const text = new TextDecoder().decode(value)
+          console.log('处理流式数据块:', text.length, '字节')
           const textContent = [...text.matchAll(/data:.*"}/g)]
 
           textContent.forEach(item => {
@@ -289,12 +293,38 @@ const handleChatCompletions = async (ctx) => {
         }
         ctx.body = resBody
       } else {
+        console.log('正常结束流式响应')
         ctx.res.end()
       }
     } catch (error) {
       console.error('流式响应出错:', error)
+      if (stream == "true" || stream == true) {
+        try {
+          console.log('发送错误信息到流')
+          ctx.res.write(`data: ${JSON.stringify({
+            "id": `chatcmpl-${messageId}`,
+            "choices": [
+              {
+                "index": 0,
+                "delta": {
+                  "content": "\n\n抱歉，处理您的请求时出现错误。"
+                }
+              }
+            ],
+            "created": Math.floor(Date.now() / 1000),
+            "model": model,
+            "object": "chat.completion.chunk"
+          })}\n\n`)
+          ctx.res.write('data: [DONE]\n\n')
+        } catch (e) {
+          console.error('发送错误信息失败:', e)
+        } finally {
+          ctx.res.end()
+        }
+      } else {
       ctx.status = 500
       ctx.body = { error: '流式响应处理失败', details: error.toString() }
+      }
     }
   } catch (error) {
     console.error('请求处理出错:', error)