测试

src/service/article-service.ts

@@ -12,7 +12,6 @@ async function getImageBufferByText(text: string, apiKey: string) {
   console.log(`translatedText：${translatedText}`);
   const summaryText = await retryAsync(() => getSummaryText(translatedText, apiKey));
   console.log(`summaryText:${summaryText}`);
-  // 使用axios以async/await形式调用API接口生成图片
   const buffer = await retryAsync(() => getImageBuffer(summaryText, apiKey));
   console.log('------------- 图片buffer获取成功 ---------------');
   return buffer;
@@ -20,7 +19,7 @@ async function getImageBufferByText(text: string, apiKey: string) {
 
 async function preheat(apiKey: string) {
   try {
-    await retryAsync(() => getImageBufferByText('预热', apiKey));
+    await getImageBufferByText('一个苹果', apiKey);
   } catch (error) {}
 }
 

src/utils/common.ts

@@ -6,14 +6,15 @@ export async function delay(time: number) {
   console.log(`延迟${time}毫秒后执行`);
 }
 
-export async function retryAsync(func: () => Promise<any>, x = 3, timeout = 10 * 1000) {
+export async function retryAsync(func: () => Promise<any>, x = 3, timeout = 5 * 1000) {
   let attempts = 0;
   while (attempts < x) {
     try {
       return await func();
     } catch (error) {
+      console.log(`第${attempts + 1}次尝试失败：${error}`);
       attempts++;
-      delay(timeout);
+      await delay(timeout);
     }
   }
   throw new Error(`Async function failed after ${x} attempts.`);

src/utils/file-utils.ts

@@ -69,22 +69,26 @@ export async function downloadDocx(url: string, fileName: string) {
 }
 
 export async function getImageBuffer(text: string, apiKey: string) {
-  const response = await axios.post(
-    'https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev',
-    {
-      inputs: text,
-      options: {
-        use_cache: false,
+  try {
+    const response = await axios.post(
+      'https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev',
+      {
+        inputs: text,
+        options: {
+          use_cache: true,
+        },
       },
-    },
-    {
-      headers: {
-        'Content-Type': 'application/json',
-        Authorization: `Bearer ${apiKey}`,
+      {
+        headers: {
+          'Content-Type': 'application/json',
+          Authorization: `Bearer ${apiKey}`,
+        },
+        responseType: 'arraybuffer', // 设置响应类型为 blob
       },
-      responseType: 'arraybuffer', // 设置响应类型为 blob
-    },
-  );
-  const arrayBuffer = response.data as ArrayBuffer;
-  return Buffer.from(arrayBuffer);
+    );
+    const arrayBuffer = response.data as ArrayBuffer;
+    return Buffer.from(arrayBuffer);
+  } catch (error) {
+    throw new Error(`生成图片失败，错误信息：${error}`);
+  }
 }

src/utils/hugging-face.ts

@@ -1,36 +1,44 @@
 import axios from 'axios';
 
 export async function getTranslatedText(text: string, apiKey: string) {
-  const translatedData = await axios.post(
-    'https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-zh-en',
-    { inputs: text, src_lang: 'ch', target_lang: 'en' },
-    {
-      headers: {
-        'Content-Type': 'application/json',
-        // 如果需要认证
-        Authorization: `Bearer ${apiKey}`,
+  try {
+    const translatedData = await axios.post(
+      'https://api-inference.huggingface.co/models/Helsinki-NLP/opus-mt-zh-en',
+      { inputs: text, src_lang: 'ch', target_lang: 'en' },
+      {
+        headers: {
+          'Content-Type': 'application/json',
+          // 如果需要认证
+          Authorization: `Bearer ${apiKey}`,
+        },
       },
-    },
-  );
-  const translatedText = translatedData.data[0].translation_text as string;
-
-  return translatedText;
+    );
+    const translatedText = translatedData.data[0].translation_text as string;
+  
+    return translatedText;
+  } catch (error) {
+    throw new Error(`翻译失败，错误信息：${error}`);
+  }
 }
 
 export async function getSummaryText(text: string, apiKey: string) {
-  const summaryData = await axios.post(
-    'https://api-inference.huggingface.co/models/csebuetnlp/mT5_multilingual_XLSum',
-    { inputs: text },
-    {
-      headers: {
-        'Content-Type': 'application/json',
-        // 如果需要认证
-        Authorization: `Bearer ${apiKey}`,
+  try {
+    const summaryData = await axios.post(
+      'https://api-inference.huggingface.co/models/csebuetnlp/mT5_multilingual_XLSum',
+      { inputs: text },
+      {
+        headers: {
+          'Content-Type': 'application/json',
+          // 如果需要认证
+          Authorization: `Bearer ${apiKey}`,
+        },
       },
-    },
-  );
-
-  const summaryText = summaryData.data[0].summary_text as string;
-
-  return summaryText;
+    );
+  
+    const summaryText = summaryData.data[0].summary_text as string;
+  
+    return summaryText;
+  } catch (error) {
+    throw new Error(`总结失败，错误信息：${error}`);
+  }
 }