Update index.js

index.js

@@ -29,7 +29,7 @@ const CONFIG = {
         SIGNATURE_COOKIE: null,
         TEMP_COOKIE: null,
         PICGO_KEY: process.env.PICGO_KEY || null, //想要流式生图的话需要填入这个PICGO图床的key
-        PICUI_KEY: process.env.PICUI_KEY || null //想要流式生图的话需要填入这个PICUI图床的key 两个图床二选一，默认使用PICGO
+        TUMY_KEY: process.env.TUMY_KEY || null //想要流式生图的话需要填入这个TUMY图床的key 两个图床二选一，默认使用PICGO
     },
     SERVER: {
         PORT: process.env.PORT || 3000,
@@ -367,7 +367,7 @@ class GrokApiClient {
     }
 
     async prepareChatRequest(request) {
-        if ((request.model === 'grok-2-imageGen' || request.model === 'grok-3-imageGen') && !CONFIG.API.PICGO_KEY && !CONFIG.API.PICUI_KEY && request.stream) {
+        if ((request.model === 'grok-2-imageGen' || request.model === 'grok-3-imageGen') && !CONFIG.API.PICGO_KEY && !CONFIG.API.TUMY_KEY && request.stream) {
             throw new Error(`该模型流式输出需要配置PICGO或者PICUI图床密钥!`);
         }
 
@@ -712,7 +712,7 @@ async function handleImageResponse(imageUrl) {
     const arrayBuffer = await imageBase64Response.arrayBuffer();
     const imageBuffer = Buffer.from(arrayBuffer);
 
-    if (!CONFIG.API.PICGO_KEY && !CONFIG.API.PICUI_KEY) {
+    if (!CONFIG.API.PICGO_KEY && !CONFIG.API.TUMY_KEY) {
         const base64Image = imageBuffer.toString('base64');
         const imageContentType = imageBase64Response.headers.get('content-type');
         return `![image](data:${imageContentType};base64,${base64Image})`
@@ -736,13 +736,13 @@ async function handleImageResponse(imageUrl) {
         body: formData
         });
         if (!responseURL.ok) {
-            return "生图失败，请查看图床密钥是否设置正确"
+            return "生图失败，请查看PICGO图床密钥是否设置正确"
         } else {
             Logger.info("生图成功", 'Server');
             const result = await responseURL.json();
             return `![image](${result.image.url})`
         }
-    }else if(CONFIG.API.PICUI_KEY){
+    }else if(CONFIG.API.TUMY_KEY){
         const formData = new FormData();
         formData.append('file', imageBuffer, {
             filename: `image-${Date.now()}.jpg`,
@@ -754,16 +754,21 @@ async function handleImageResponse(imageUrl) {
             headers: {
                 ...formDataHeaders,
                 "Accept": "application/json",
-                'Authorization': `Bearer ${CONFIG.API.PICUI_KEY}`
+                'Authorization': `Bearer ${CONFIG.API.TUMY_KEY}`
             },
             body: formData
         });
         if (!responseURL.ok) {
-            return "生图失败，请查看图床密钥是否设置正确"
+            return "生图失败，请查看TUMY图床密钥是否设置正确"
         } else {
             Logger.info("生图成功", 'Server');
             const result = await responseURL.json();
-            return `![image](${result.data.links.url})`
+            try {
+              var url = `![image](${result.data.links.url})`
+              return url;
+            } catch (error) {
+              return "生图失败，请查看TUMY图床密钥是否设置正确"
+            }
         }
     }
 }