Update index.js

index.js

@@ -73,9 +73,8 @@ async function initialization() {
         return;
     }
     const ssoArray = process.env.SSO.split(',');
-    const ssorwArray = process.env.SSO_RW.split(',');
-    ssoArray.forEach((sso, index) => {
-        tokenManager.addToken(`sso-rw=${ssorwArray[index]};sso=${sso}`);
+    ssoArray.forEach((sso) => {
+        tokenManager.addToken(`sso-rw=${sso};sso=${sso}`);
     });
     console.log(JSON.stringify(tokenManager.getActiveTokens(), null, 2));
     await Utils.get_signature()
@@ -215,7 +214,6 @@ class Utils {
     static async extractGrokHeaders() {
         Logger.info("开始提取头信息", 'Server');
         try {
-            // 启动浏览器
             const browser = await puppeteer.launch({
                 headless: true,
                 args: [
@@ -236,16 +234,12 @@ class Utils {
             const cookies = await page.cookies();
             const targetHeaders = ['x-anonuserid', 'x-challenge', 'x-signature'];
             const extractedHeaders = {};
-            // 遍历 Cookies
             for (const cookie of cookies) {
-                // 检查是否为目标头信息
                 if (targetHeaders.includes(cookie.name.toLowerCase())) {
                     extractedHeaders[cookie.name.toLowerCase()] = cookie.value;
                 }
             }
-            // 关闭浏览器
             await browser.close();
-            // 打印并返回提取的头信息
             Logger.info('提取的头信息:', JSON.stringify(extractedHeaders, null, 2), 'Server');
             return extractedHeaders;
 
@@ -724,6 +718,7 @@ async function handleImageResponse(imageUrl) {
         return `![image](data:${imageContentType};base64,${base64Image})`
     }
 
+    Logger.info("开始请求图床", 'Server');
     const formData = new FormData();
     if(CONFIG.API.PICGO_KEY){
         formData.append('source', imageBuffer, {
@@ -754,7 +749,7 @@ async function handleImageResponse(imageUrl) {
             contentType: 'image/jpeg'
         });
         const formDataHeaders = formData.getHeaders();
-        const responseURL = await fetch("https://picui.cn/api/v1/upload", {
+        const responseURL = await fetch("https://tu.my/api/v1/upload", {
             method: "POST",
             headers: {
                 ...formDataHeaders,
@@ -804,15 +799,11 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
     try {
         const authToken = req.headers.authorization?.replace('Bearer ', '');
         if (CONFIG.API.IS_CUSTOM_SSO) {
-            if (authToken && authToken.includes(';')) {
-                const parts = authToken.split(';');
-                const result = [
-                    `sso=${parts[0]}`,
-                    `ssp_rw=${parts[1]}`
-                ].join(';');
+            if (authToken) {
+                const result = `sso=${authToken};ssp_rw=${authToken}`;
                 tokenManager.setToken(result);
             } else {
-                return res.status(401).json({ error: '自定义的SSO令牌格式错误' });
+                return res.status(401).json({ error: '自定义的SSO令牌缺失' });
             }
         } else if (authToken !== CONFIG.API.API_KEY) {
             return res.status(401).json({ error: 'Unauthorized' });
@@ -836,7 +827,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
                 CONFIG.API.SIGNATURE_COOKIE = await Utils.createAuthHeaders(req.body.model);
             }
             if (!CONFIG.API.SIGNATURE_COOKIE) {
-                throw new Error('无可用令牌');
+                throw new Error('该模型无可用令牌');
             }
             Logger.info(`当前令牌索引: ${CONFIG.SSO_INDEX}`, 'Server');
             Logger.info(`当前令牌: ${JSON.stringify(CONFIG.API.SIGNATURE_COOKIE,null,2)}`, 'Server');
@@ -859,7 +850,7 @@ app.post('/hf/v1/chat/completions', async (req, res) => {
                 const responseText = await newMessageReq.json();
                 conversationId = responseText.conversationId;
             } else {
-                Logger.error(`创建会话请求: ${responseText2}`, 'Server');
+                Logger.error(`创建会话响应错误: ${responseText2}`, 'Server');
                 throw new Error(`创建会话响应错误: ${responseText2}`);
             }