Upload index.js

index.js

@@ -51,6 +51,9 @@ const DEFAULT_HEADERS = {
     'baggage': 'sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c'
 };
 
+// 定义签名文件路径为 /tmp 目录
+const SIGNATURE_FILE_PATH = '/tmp/signature.json';
+
 class Utils {
     static async extractGrokHeaders() {
         try {
@@ -93,8 +96,13 @@ class Utils {
             if (headers) {
                 console.log("获取认证信息成功");
                 CONFIG.API.SIGNATURE_COOKIE = { cookie: `x-anonuserid=${headers["x-anonuserid"]}; x-challenge=${headers["x-challenge"]}; x-signature=${headers["x-signature"]}` };
-                fs.writeFileSync(path.resolve(process.cwd(), 'signature.json'), JSON.stringify(CONFIG.API.SIGNATURE_COOKIE));//保存认证信息
-                return CONFIG.API.SIGNATURE_COOKIE;
+                try {
+                    fs.writeFileSync(SIGNATURE_FILE_PATH, JSON.stringify(CONFIG.API.SIGNATURE_COOKIE));//保存认证信息
+                    return CONFIG.API.SIGNATURE_COOKIE;
+        } catch (error) {
+            console.error('写入签名文件失败:', error);
+                    return CONFIG.API.SIGNATURE_COOKIE;
+                }
             }
             retryCount++;
             if (retryCount >= CONFIG.RETRY.MAX_ATTEMPTS) {
@@ -147,6 +155,72 @@ class Utils {
             }
         });
     }
+
+    static async makeRequest(req) {
+        try {
+            if (!CONFIG.API.SIGNATURE_COOKIE) {
+                await Utils.get_signature();
+                try {
+                    CONFIG.API.SIGNATURE_COOKIE = JSON.parse(fs.readFileSync(SIGNATURE_FILE_PATH));
+                } catch (error) {
+                    console.error('读取签名文件失败:', error);
+                    await Utils.get_signature(); // 如果读取失败，重新获取签名
+                }
+            }
+            const grokClient = new GrokApiClient(req.body.model);
+            const requestPayload = await grokClient.prepareChatRequest(req.body);
+            //创建新对话        
+            const newMessageReq = await fetch(`${CONFIG.API.BASE_URL}/api/rpc`, {
+                method: 'POST',
+                headers: {
+                    ...DEFAULT_HEADERS,
+                    ...CONFIG.API.SIGNATURE_COOKIE
+                },
+                body: JSON.stringify({
+                    rpc: "createConversation",
+                    req: {
+                        temporary: false
+                    }
+                })
+            });
+
+            if (!newMessageReq.ok) {
+                throw new Error(`上游服务请求失败! status: ${newMessageReq.status}`);
+            }
+
+            // 获取响应文本
+            const responseText = await newMessageReq.json();
+            const conversationId = responseText.conversationId;
+            console.log("会话ID:conversationId", conversationId);
+            if (!conversationId) {
+                throw new Error(`创建会话失败! status: ${newMessageReq.status}`);
+            }
+            //发送对话
+            const response = await fetch(`${CONFIG.API.BASE_URL}/api/conversations/${conversationId}/responses`, {
+                method: 'POST',
+                headers: {
+                    "accept": "text/event-stream",
+                    "baggage": "sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c",
+                    "content-type": "text/plain;charset=UTF-8",
+                    "Connection": "keep-alive",
+                    ...CONFIG.API.SIGNATURE_COOKIE
+                },
+                body: JSON.stringify(requestPayload)
+            });
+
+            if (!response.ok) {
+                throw new Error(`上游服务请求失败! status: ${response.status}`);
+            }
+
+            if (req.body.stream) {
+                await handleStreamResponse(response, req.body.model, res);
+            } else {
+                await handleNormalResponse(response, req.body.model, res);
+            }
+        } catch (error) {
+            throw error;
+        }
+    }
 }
 
 
@@ -389,65 +463,10 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
         return res.status(400).json({ error: '该模型不支持流式' });
     }
 
-    const makeRequest = async () => {
-        if (!CONFIG.API.SIGNATURE_COOKIE) {
-            await Utils.get_signature();
-            CONFIG.API.SIGNATURE_COOKIE = JSON.parse(fs.readFileSync(path.resolve(process.cwd(), 'signature.json')));
-        }
-        const grokClient = new GrokApiClient(req.body.model);
-        const requestPayload = await grokClient.prepareChatRequest(req.body);
-        //创建新对话        
-        const newMessageReq = await fetch(`${CONFIG.API.BASE_URL}/api/rpc`, {
-            method: 'POST',
-            headers: {
-                ...DEFAULT_HEADERS,
-                ...CONFIG.API.SIGNATURE_COOKIE
-            },
-            body: JSON.stringify({
-                rpc: "createConversation",
-                req: {
-                    temporary: false
-                }
-            })
-        });
-        if (!newMessageReq.ok) {
-            throw new Error(`上游服务请求失败! status: ${newMessageReq.status}`);
-        }
-
-        // 获取响应文本
-        const responseText = await newMessageReq.json();
-        const conversationId = responseText.conversationId;
-        console.log("会话ID:conversationId", conversationId);
-        if (!conversationId) {
-            throw new Error(`创建会话失败! status: ${newMessageReq.status}`);
-        }
-        //发送对话
-        const response = await fetch(`${CONFIG.API.BASE_URL}/api/conversations/${conversationId}/responses`, {
-            method: 'POST',
-            headers: {
-                "accept": "text/event-stream",
-                "baggage": "sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c",
-                "content-type": "text/plain;charset=UTF-8",
-                "Connection": "keep-alive",
-                ...CONFIG.API.SIGNATURE_COOKIE
-            },
-            body: JSON.stringify(requestPayload)
-        });
-
-        if (!response.ok) {
-            throw new Error(`上游服务请求失败! status: ${response.status}`);
-        }
-
-        if (req.body.stream) {
-            await handleStreamResponse(response, req.body.model, res);
-        } else {
-            await handleNormalResponse(response, req.body.model, res);
-        }
-    }
     try {
-        await makeRequest();
+        await Utils.makeRequest(req);
     } catch (error) {
-        await Utils.handleError(error, res, makeRequest);
+        await Utils.handleError(error, res);
     }
 });