Update index.js

index.js

@@ -6,13 +6,14 @@ import cors from 'cors';
 import Logger from './logger.js';
 import dotenv from 'dotenv';
 
+
 // 初始化环境变量
 dotenv.config();
 
 // 配置管理
 const CONFIG = {
     SERVER: {
-        PORT: process.env.PORT || 25526,
+        PORT: process.env.PORT || 7860,
         BODY_LIMIT: '5mb',
         CORS_OPTIONS: {
             origin: '*',
@@ -38,6 +39,7 @@ const CONFIG = {
         "grok-3-reasoning": "grok-3",
         "grok-3-imageGen": "grok-3",
     },
+    SIGNATURE_INDEX: 0,
     IS_IMG_GEN: false,
     IS_THINKING: false
 };
@@ -58,6 +60,82 @@ const DEFAULT_HEADERS = {
     'priority': 'u=1, i'
 };
 
+
+async function initialization() {
+    const auth_tokenArray = CONFIG.API.AUTH_TOKEN.split(',');
+    const ct0Array = CONFIG.API.CT0.split(',');
+    auth_tokenArray.forEach((auth_token, index) => {
+        tokenManager.addToken(`auth_token=${auth_token};ct0=${ct0Array[index]}`);
+    });
+    Logger.info("初始化完成", 'Server');
+}
+class AuthTokenManager {
+    constructor() {
+        this.activeTokens = [];
+        this.expiredTokens = new Map();
+        this.isRecoveryProcess = false;
+    }
+
+    // 添加 token
+    addToken(token) {
+        if (!this.activeTokens.includes(token)) {
+            this.activeTokens.push(token);
+        }
+    }
+
+    // 通过下标获取 token
+    getTokenByIndex(index) {
+        if (index < 0 || index >= this.activeTokens.length) {
+            throw new Error(`无效的索引：${index}`);
+        }
+        return this.activeTokens[index];
+    }
+
+    // 通过下标移除 token
+    removeTokenByIndex(index) {
+        if (index < 0 || index >= this.activeTokens.length) {
+            throw new Error(`无效的索引：${index}`);
+        }
+
+        const token = this.activeTokens[index];
+        this.activeTokens.splice(index, 1);
+
+        // 记录失效时间
+        this.expiredTokens.set(token, Date.now());
+        return token;
+    }
+
+    // 启动定期恢复机制
+    startTokenRecoveryProcess() {
+        setInterval(() => {
+            const now = Date.now();
+
+            // 遍历并恢复过期的 token
+            for (const [token, expiredTime] of this.expiredTokens.entries()) {
+                if (now - expiredTime >= 1 * 60 * 1000) {
+                    this.activeTokens.push(token);
+                    this.expiredTokens.delete(token);
+                    console.log(`Token ${token} recovered`);
+                }
+            }
+        }, 1 * 60 * 1000);
+    }
+
+    // 获取 token 总数
+    getTokenCount() {
+        return this.activeTokens.length;
+    }
+
+    // 获取所有活跃 token
+    getActiveTokens() {
+        return [...this.activeTokens];
+    }
+}
+
+const tokenManager = new AuthTokenManager();
+await initialization();
+
+
 // 工具类
 class Utils {
     static generateRandomString(length, charset = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ') {
@@ -65,12 +143,17 @@ class Utils {
             .map(() => charset[Math.floor(Math.random() * charset.length)])
             .join('');
     }
-
+    static getRandomID(size) {
+        const customDict = '0123456789';
+        return Array(size).fill(null)
+            .map(() => customDict[(Math.random() * customDict.length) | 0])
+            .join('');
+    }
     static createAuthHeaders() {
         return {
             ...DEFAULT_HEADERS,
-            'x-csrf-token': CONFIG.API.CT0,
-            'cookie': `auth_token=${CONFIG.API.AUTH_TOKEN};ct0=${CONFIG.API.CT0}`
+            'x-csrf-token': tokenManager.getTokenByIndex(CONFIG.SIGNATURE_INDEX).split(';')[1].split('=')[1],
+            'cookie': `${tokenManager.getTokenByIndex(CONFIG.SIGNATURE_INDEX)}`
         };
     }
 
@@ -102,6 +185,10 @@ class ConversationManager {
         const data = await Utils.handleApiResponse(response, '创建会话失败!');
         return data.data.create_grok_conversation.conversation_id;
     }
+    // static generateNewId() {
+    //     this.conversationId = `1875${Utils.getRandomID(15)}`;
+    //     return this.conversationId;
+    // }
 
     static async deleteConversation(conversationId) {
         if (!conversationId) return;
@@ -417,7 +504,6 @@ class ResponseHandler {
         let buffer = '';
         let fullResponse = '';
         let imageUrl = null;
-
         try {
             for await (const chunk of reader) {
                 const lines = (buffer + chunk.toString()).split('\n');
@@ -435,6 +521,9 @@ class ResponseHandler {
             if (imageUrl) {
                 await this.sendImageResponse(imageUrl, model, res);
             } else {
+                if (fullResponse.includes("You've reached your limit of 15 Grok")) {
+                    throw new Error('You have reached your limit of 15 Grok');
+                }
                 const responseData = MessageProcessor.createChatResponse(fullResponse, model);
                 res.json(responseData);
             }
@@ -515,7 +604,7 @@ app.use(express.json({ limit: CONFIG.SERVER.BODY_LIMIT }));
 app.use(express.urlencoded({ extended: true, limit: CONFIG.SERVER.BODY_LIMIT }));
 
 // API 路由
-app.get('/hf/v1/models', (req, res) => {
+app.get('/v1/models', (req, res) => {
     res.json({
         object: "list",
         data: Object.keys(CONFIG.MODELS).map(model => ({
@@ -526,30 +615,61 @@ app.get('/hf/v1/models', (req, res) => {
         }))
     });
 });
-
-app.post('/hf/v1/chat/completions', async (req, res) => {
+app.post('/v1/chat/completions', async (req, res) => {
     try {
         const authToken = req.headers.authorization?.replace('Bearer ', '');
         if (authToken !== CONFIG.API.API_KEY) {
             return res.status(401).json({ error: 'Unauthorized' });
         }
 
-        const grokClient = new TwitterGrokApiClient(req.body.model);
-        const requestPayload = await grokClient.prepareChatRequest(req.body);
-
-        const response = await fetch(CONFIG.API.ENDPOINTS.CHAT, {
-            method: 'POST',
-            headers: Utils.createAuthHeaders(),
-            body: JSON.stringify(requestPayload)
-        });
-
-        if (!response.ok) {
-            throw new Error(`上游服务请求失败! status: ${response.status}`);
+        while (tokenManager.getTokenCount() > 0) {
+            const grokClient = new TwitterGrokApiClient(req.body.model);
+            const requestPayload = await grokClient.prepareChatRequest(req.body);
+            Logger.info(`当前令牌索引: ${CONFIG.SIGNATURE_INDEX}`, 'Server');
+            const response = await fetch(CONFIG.API.ENDPOINTS.CHAT, {
+                method: 'POST',
+                headers: Utils.createAuthHeaders(),
+                body: JSON.stringify(requestPayload)
+            });
+
+            if (response.ok) {
+                Logger.info(`请求成功`, 'Server');
+                CONFIG.SIGNATURE_INDEX = (CONFIG.SIGNATURE_INDEX + 1) % tokenManager.getTokenCount();
+                Logger.info(`当前剩余可用令牌数: ${tokenManager.getTokenCount()}`, 'Server');
+                try {
+                    await (req.body.stream
+                        ? ResponseHandler.handleStreamResponse(response, req.body.model, res)
+                        : ResponseHandler.handleNormalResponse(response, req.body.model, res));
+                    return; // 成功后直接返回
+                } catch (error) {
+                    tokenManager.removeTokenByIndex(CONFIG.SIGNATURE_INDEX);
+                    if (!tokenManager.isRecoveryProcess) {
+                        tokenManager.startTokenRecoveryProcess();
+                    }
+                    Logger.warn(`当前令牌失效，已移除令牌，剩余令牌数: ${tokenManager.getTokenCount()}`, 'Server');
+                    // 更新签名索引
+                    CONFIG.SIGNATURE_INDEX = (CONFIG.SIGNATURE_INDEX + 1) % tokenManager.getTokenCount();
+                }
+            } else {
+                // 处理429错误
+                if (response.status === 429) {
+                    tokenManager.removeTokenByIndex(CONFIG.SIGNATURE_INDEX);
+                    if (!tokenManager.isRecoveryProcess) {
+                        tokenManager.startTokenRecoveryProcess();
+                    }
+                    Logger.warn(`当前令牌失效，已移除令牌，剩余令牌数: ${tokenManager.getTokenCount()}`, 'Server');
+                    // 更新签名索引
+                    CONFIG.SIGNATURE_INDEX = (CONFIG.SIGNATURE_INDEX + 1) % tokenManager.getTokenCount();
+                    Logger.warn(`请求被限流，剩余重试次数: ${tokenManager.getTokenCount()}`, 'ChatAPI');
+                } else {
+                    // 非429错误直接抛出
+                    throw new Error(`上游服务请求失败! status: ${response.status}`);
+                }
+            }
         }
 
-        await (req.body.stream
-            ? ResponseHandler.handleStreamResponse(response, req.body.model, res)
-            : ResponseHandler.handleNormalResponse(response, req.body.model, res));
+        // 如果重试次数用完仍然是429
+        throw new Error('所有令牌都已耗尽，请求被限流');
 
     } catch (error) {
         Logger.error('Chat Completions Request Error', error, 'ChatAPI');
@@ -561,16 +681,12 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
                 code: error.code || null
             }
         });
-    } finally {
-        if (req.body.conversationId) {
-            await ConversationManager.deleteConversation(req.body.conversationId);
-        }
     }
 });
 
 // 404 处理
 app.use((req, res) => {
-    res.status(404).json({ message: '服务创建成功，正在运行' });
+    res.status(404).json({ error: '请求路径不存在' });
 });
 
 // 启动服务器