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,80 @@ 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();
+            for (const [token, expiredTime] of this.expiredTokens.entries()) {
+                if (now - expiredTime >= 2 * 60 * 60 * 1000) {
+                    this.activeTokens.push(token);
+                    this.expiredTokens.delete(token);
+                    console.log(`Token ${token} recovered`);
+                }
+            }
+        }, 2 * 60 * 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 +141,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)}`
         };
     }
 
@@ -202,14 +283,45 @@ class TwitterGrokApiClient {
     }
 
     async transformMessages(messages) {
+        if (messages[0].role === 'assistant') {
+            throw new Error('ai不能是第一个消息');
+        }
+
         const processedMessages = [];
-        for (let i = 0; i < messages.length; i++) {
-            const isLastTwoMessages = i >= messages.length - 2;
-            const content = await this.processMessageContent(messages[i], isLastTwoMessages);
-            if (content) {
-                processedMessages.push(content);
+        let currentMessage = null;
+
+        for (const msg of messages) {
+            const normalizedMsg = msg.role === 'system' ? { ...msg, role: 'user' } : msg;
+
+            if (!currentMessage || currentMessage.role !== normalizedMsg.role) {
+                if (currentMessage) {
+                    const processedContent = await this.processMessageContent(
+                        currentMessage,
+                        processedMessages.length >= messages.length - 2
+                    );
+                    if (processedContent) {
+                        processedMessages.push(processedContent);
+                    }
+                }
+                currentMessage = normalizedMsg;
+            } else {
+                currentMessage.content = typeof currentMessage.content === 'string' && typeof normalizedMsg.content === 'string'
+                    ? `${currentMessage.content}\n${normalizedMsg.content}`
+                    : normalizedMsg.content;
             }
         }
+
+        // 处理最后一个消息
+        if (currentMessage) {
+            const processedContent = await this.processMessageContent(
+                currentMessage,
+                true
+            );
+            if (processedContent) {
+                processedMessages.push(processedContent);
+            }
+        }
+
         return processedMessages;
     }
 
@@ -228,7 +340,7 @@ class TwitterGrokApiClient {
 
         return {
             message,
-            sender: role === 'user' ? 1 : 2,
+            sender: role === 'assistant' ? 2 : 1,
             ...(role === 'user' && { fileAttachments })
         };
     }
@@ -386,7 +498,6 @@ class ResponseHandler {
         let buffer = '';
         let fullResponse = '';
         let imageUrl = null;
-
         try {
             for await (const chunk of reader) {
                 const lines = (buffer + chunk.toString()).split('\n');
@@ -404,6 +515,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);
             }
@@ -495,7 +609,6 @@ app.get('/hf/v1/models', (req, res) => {
         }))
     });
 });
-
 app.post('/hf/v1/chat/completions', async (req, res) => {
     try {
         const authToken = req.headers.authorization?.replace('Bearer ', '');
@@ -503,22 +616,54 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
             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');
@@ -530,10 +675,6 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
                 code: error.code || null
             }
         });
-    } finally {
-        if (req.body.conversationId) {
-            await ConversationManager.deleteConversation(req.body.conversationId);
-        }
     }
 });
 
@@ -545,4 +686,5 @@ app.use((req, res) => {
 // 启动服务器
 app.listen(CONFIG.SERVER.PORT, () => {
     Logger.info(`服务器运行在端口 ${CONFIG.SERVER.PORT}`, 'Server');
-});
+});
+