Upload 2 files

src/index.js

@@ -19,6 +19,71 @@ app.use(bodyParser({
   textLimit: '30mb',  // text 数据大小限制
 }))
 
+// 添加 CORS 中间件
+app.use(async (ctx, next) => {
+  ctx.set('Access-Control-Allow-Origin', '*');
+  ctx.set('Access-Control-Allow-Headers', 'Origin, X-Requested-With, Content-Type, Accept, Authorization');
+  ctx.set('Access-Control-Allow-Methods', 'POST, GET, PUT, DELETE, OPTIONS');
+  ctx.set('Access-Control-Max-Age', '3600');
+  
+  // 处理 OPTIONS 请求
+  if (ctx.method === 'OPTIONS') {
+    ctx.status = 204;
+    return;
+  }
+  
+  await next();
+});
+
+// 添加请求日志中间件
+app.use(async (ctx, next) => {
+  const start = Date.now()
+  console.log(`[${new Date().toISOString()}] ${ctx.method} ${ctx.url} - Request started`)
+  try {
+    await next()
+  } catch (err) {
+    throw err
+  } finally {
+    const ms = Date.now() - start
+    console.log(`[${new Date().toISOString()}] ${ctx.method} ${ctx.url} - Response ${ctx.status} - ${ms}ms`)
+  }
+})
+
+// 错误处理中间件
+app.use(async (ctx, next) => {
+  try {
+    await next()
+    // 处理 404 错误
+    if (ctx.status === 404) {
+      ctx.status = 404
+      ctx.body = {
+        error: 'Not Found',
+        message: `The requested path ${ctx.path} was not found`,
+        availableEndpoints: [
+          '/v1/chat/completions',
+          '/hf/v1/chat/completions',
+          '/v1/models',
+          '/hf/v1/models'
+        ]
+      }
+    }
+  } catch (err) {
+    console.error('Error details:', {
+      path: ctx.path,
+      method: ctx.method,
+      error: err.message,
+      stack: err.stack
+    })
+    ctx.status = err.status || 500
+    ctx.body = {
+      error: err.name || 'Internal Server Error',
+      message: err.message,
+      path: ctx.path
+    }
+    ctx.app.emit('error', err, ctx)
+  }
+})
+
 const makeRequest = async (session_id, requestModel, messages) => {
   console.log('开始请求 genspark.ai:', { session_id, requestModel })
   try {
@@ -217,7 +282,7 @@ const handleChatCompletions = async (ctx) => {
   }
 }
 
-// 创建获取模型列表��处理函数
+// 创建获取模型列表处理函数
 const handleModels = async (ctx) => {
   ctx.body = {
     object: "list",
@@ -251,53 +316,9 @@ router.get('/', async (ctx) => {
   }
 })
 
-// 注册路由中间件
+// 注册路由��间件 (移到错误处理之后)
 app.use(router.routes()).use(router.allowedMethods())
 
-// 错误处理中间件之前添加请求日志中间件
-app.use(async (ctx, next) => {
-  const start = Date.now()
-  console.log(`[${new Date().toISOString()}] ${ctx.method} ${ctx.url} - Request started`)
-  await next()
-  const ms = Date.now() - start
-  console.log(`[${new Date().toISOString()}] ${ctx.method} ${ctx.url} - Response ${ctx.status} - ${ms}ms`)
-})
-
-// 修改错误处理中间件
-app.use(async (ctx, next) => {
-  try {
-    await next()
-    // 处理 404 错误
-    if (ctx.status === 404) {
-      ctx.status = 404
-      ctx.body = {
-        error: 'Not Found',
-        message: `The requested path ${ctx.path} was not found`,
-        availableEndpoints: [
-          '/v1/chat/completions',
-          '/hf/v1/chat/completions',
-          '/v1/models',
-          '/hf/v1/models'
-        ]
-      }
-    }
-  } catch (err) {
-    console.error('Error details:', {
-      path: ctx.path,
-      method: ctx.method,
-      error: err.message,
-      stack: err.stack
-    })
-    ctx.status = err.status || 500
-    ctx.body = {
-      error: err.name || 'Internal Server Error',
-      message: err.message,
-      path: ctx.path
-    }
-    ctx.app.emit('error', err, ctx)
-  }
-})
-
 // 启动服务器
 const PORT = process.env.PORT || 3000
 app.listen(PORT, '0.0.0.0', () => {