Upload 2 files

src/index.js

@@ -174,6 +174,19 @@ const handleChatCompletions = async (ctx) => {
 
     const messageId = crypto.randomUUID()
     const reader = response.body.getReader()
+    let resBody = {
+      id: `chatcmpl-${messageId}`,
+      object: 'chat.completion',
+      created: Math.floor(Date.now() / 1000),
+      model,
+      choices: [],
+      usage: {
+        prompt_tokens: 0,
+        completion_tokens: 0,
+        total_tokens: 0,
+      },
+    }
+
     if (stream == "true" || stream == true) {
       ctx.res.write(`data: ${JSON.stringify({
         "id": `chatcmpl-${messageId}`,
@@ -193,8 +206,6 @@ const handleChatCompletions = async (ctx) => {
     }
 
     try {
-      let resBody = {}
-
       while (true) {
         const { done, value } = await reader.read()
         if (done) {
@@ -204,7 +215,7 @@ const handleChatCompletions = async (ctx) => {
           break
         }
 
-        if (stream) {
+        if (stream == "true" || stream == true) {
           const text = new TextDecoder().decode(value)
           const textContent = [...text.matchAll(/data:.*"}/g)]
 
@@ -237,40 +248,38 @@ const handleChatCompletions = async (ctx) => {
             const content = JSON.parse(item[0].replace("data: ", ''))
             if (!content || !content.field_value || content.field_name == 'session_state.answer_is_finished' || content.field_name == 'content' || content.field_name == 'session_state' || content.delta || content.type == 'project_field') return
 
-            resBody = {
-              id: `chatcmpl-${messageId}`,
-              object: 'chat.completion',
-              created: Math.floor(Date.now() / 1000),
-              model,
-              choices: [
-                {
-                  index: 0,
-                  message: {
-                    role: 'assistant',
-                    content: content.field_value,
-                  },
-                  finish_reason: 'stop',
-                },
-              ],
-              usage: {
-                prompt_tokens: 0,
-                completion_tokens: 0,
-                total_tokens: content.field_value.length,
+            resBody.choices.push({
+              index: 0,
+              message: {
+                role: 'assistant',
+                content: content.field_value,
               },
-            }
+              finish_reason: 'stop',
+            })
+            resBody.usage.total_tokens = content.field_value.length
           })
         }
       }
 
       if (stream == "false" || stream == false) {
+        if (resBody.choices.length === 0) {
+          resBody.choices.push({
+            index: 0,
+            message: {
+              role: 'assistant',
+              content: '抱歉，我无法处理您的请求。',
+            },
+            finish_reason: 'stop',
+          })
+        }
         ctx.body = resBody
       } else {
         ctx.res.end()
       }
-      return
     } catch (error) {
       console.error('流式响应出错:', error)
-      ctx.res.end()
+      ctx.status = 500
+      ctx.body = { error: '流式响应处理失败', details: error.toString() }
     }
   } catch (error) {
     console.error('请求处理出错:', error)
@@ -316,7 +325,7 @@ router.get('/', async (ctx) => {
   }
 })
 
-// 注册路由��间件 (移到错误处理之后)
+// 注册路由中间件
 app.use(router.routes()).use(router.allowedMethods())
 
 // 启动服务器