Update index.js

index.js

@@ -1,108 +1,68 @@
-// Import required packages for Hugging Face
+// Import required packages
 import express from 'express';
 import cors from 'cors';
 import fetch from 'node-fetch';
 
 const app = express();
-const PORT = process.env.PORT || 7860; // Hugging Face uses port 7860 by default
+const PORT = process.env.PORT || 7860;
 
-// Target API URLs remain the same
-const TARGET_API_URLS = [
-  "https://gpt4fc.deno.dev/zaiwen",
-  "https://gpt4fc2.deno.dev/zaiwen",
-  "https://gpt4fc3.deno.dev/zaiwen",
-  "https://gpt4fc4.deno.dev/zaiwen",
-];
+// Target server
+const TARGET_DOMAIN = "https://aliyun.zaiwen.top";
 
 // Middleware setup
 app.use(express.json());
 app.use(cors({
   origin: '*',
-  methods: ['POST', 'OPTIONS'],
-  allowedHeaders: ['Content-Type', 'Authorization', 'OpenAI-Organization', 'User-Agent']
+  methods: ['POST'],
+  allowedHeaders: ['Content-Type', 'Authorization']
 }));
 
-// Handler for completions endpoint
-async function handleCompletions(req, res) {
-  try {
-    const requestBody = req.body;
-
-    // Iterate through the target API URLs
-    for (const apiUrl of TARGET_API_URLS) {
-      try {
-        // Forward the request to the target API
-        const headers = {
-          'Content-Type': 'application/json',
-        };
-        
-        // Forward other relevant headers from the original request
-        ['authorization', 'openai-organization', 'user-agent'].forEach(header => {
-          if (req.headers[header]) {
-            headers[header] = req.headers[header];
-          }
-        });
+// Only allow POST requests
+app.all('*', (req, res, next) => {
+  if (req.method !== 'POST') {
+    return res.status(405).send('Method Not Allowed');
+  }
+  next();
+});
 
-        const response = await fetch(`${apiUrl}/v1/chat/completions`, {
-          method: 'POST',
-          headers: headers,
-          body: JSON.stringify(requestBody)
-        });
+// Proxy handler
+app.post('*', async (req, res) => {
+  const targetUrl = `${TARGET_DOMAIN}${req.path}${req.url.includes('?') ? '?' + req.url.split('?')[1] : ''}`;
+  
+  try {
+    const response = await fetch(targetUrl, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        ...req.headers
+      },
+      body: JSON.stringify(req.body)
+    });
 
-        // Check if the target API returned an error
-        if (!response.ok) {
-          console.warn(
-            `Error from target API ${apiUrl}:`,
-            response.status,
-            response.statusText
-          );
-          continue; // Try the next API URL
-        }
+    const body = await response.text(); // 保持原样，前端自己决定如何解析
+    res.status(response.status);
 
-        // Get the response from the target API
-        const responseBody = await response.json();
+    // 把目标服务器返回的 headers 也加回来（注意 CORS 要处理一下）
+    response.headers.forEach((value, key) => {
+      if (key.toLowerCase() === 'content-length') return; // 不转发 content-length
+      res.setHeader(key, value);
+    });
 
-        // Check if the response has content and it's not empty
-        if (
-          responseBody &&
-          responseBody.choices &&
-          responseBody.choices.length > 0 &&
-          responseBody.choices[0].message &&
-          responseBody.choices[0].message.content &&
-          responseBody.choices[0].message.content.trim() !== ""
-        ) {
-          // Return the response to the client
-          return res.status(response.status).json(responseBody);
-        } else {
-          console.warn(`API ${apiUrl} returned empty or invalid content. Trying next API.`);
-          continue; // Try the next API URL
-        }
-      } catch (error) {
-        console.warn(`Error calling API ${apiUrl}:`, error);
-        continue; // Try the next API URL
-      }
-    }
+    res.setHeader('Access-Control-Allow-Origin', '*');
+    res.setHeader('Access-Control-Allow-Methods', 'POST');
+    res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization');
 
-    // If all APIs failed, return an error
-    return res.status(500).send('All APIs failed to return valid content.');
+    return res.send(body);
   } catch (error) {
-    console.error('Error handling /v1/chat/completions:', error);
-    return res.status(500).send('Internal Server Error');
+    console.error('Proxy Error:', error);
+    return res.status(500).send('Proxy Server Error');
   }
-}
-
-// Setup routes
-app.post('/v1/chat/completions', handleCompletions);
-
-// Default route
-app.get('/', (req, res) => {
-  res.send('API is running');
 });
 
-// Start the server
+// Start server
 app.listen(PORT, () => {
-  console.log(`Server is running on port ${PORT}`);
+  console.log(`Server running on port ${PORT}`);
 });
 
-// For Hugging Face deployment
+// For Huggingface deployment
 export default app;
-