feat: 使用qw

src/service/article-service.ts

@@ -2,26 +2,33 @@ import fs from 'fs';
 import htmlDocx from 'html-docx-js';
 import path from 'path';
 
-import { delay, getImageBuffer, getSummaryText, getTranslatedText, retryAsync } from '@/utils';
+import { delay, getImageBuffer, requestQw } from '@/utils';
 
 const uploadDir = path.join(__dirname, '../../uploads');
 
 async function getImageBufferByText(text: string, apiKey: string) {
   console.log(`------------ 开始获取图片，原文案：${text} ---------------`);
-  const translatedText = await retryAsync(() => getTranslatedText(text, apiKey));
-  console.log(`translatedText：${translatedText}`);
-  const summaryText = await retryAsync(() => getSummaryText(translatedText, apiKey));
-  console.log(`summaryText:${summaryText}`);
-  const buffer = await retryAsync(() => getImageBuffer(summaryText, apiKey));
+
+  const summaryTranslatedText = await requestQw({
+    "messages": [
+        {
+            "role": "user",
+            "content": `将以下内容简化内容提炼核心并翻译为英语，控制在八个单词内：${text}`
+        }
+    ],
+    "model": "qwen-plus"
+})
+  console.log(`summaryTranslatedText:${summaryTranslatedText}`);
+  const buffer = await getImageBuffer(summaryTranslatedText, apiKey)
   console.log('------------- 图片buffer获取成功 ---------------');
   return buffer;
 }
 
-async function preheat(apiKey: string) {
-  try {
-    await getImageBufferByText('一个苹果', apiKey);
-  } catch (error) {}
-}
+// async function preheat(apiKey: string) {
+//   try {
+//     await getImageBufferByText('一个苹果', apiKey);
+//   } catch (error) {}
+// }
 
 export const processArticleServe = async (data: { title: string; content: string; config: { output: string; hfApiKey: string } }) => {
   const { title, content, config } = data;
@@ -31,11 +38,11 @@ export const processArticleServe = async (data: { title: string; content: string
 
   let htmlWithImages = '';
 
-  await preheat(config.hfApiKey);
+  // await preheat(config.hfApiKey);
 
   // 遍历段落，根据段落内容调用接口生成图片并插入到HTML中
   for (let i = 0; i < paragraphs.length; i++) {
-    // await delay(10 * 1000);
+    await delay(10 * 1000);
 
     // 确定用于生成图片的描述内容，除了第一段，其余为上一段的内容
     const description = i === 0 ? paragraphs[i] : paragraphs[i - 1];

src/utils/common.ts

@@ -1,3 +1,4 @@
+import axios from 'axios';
 import util from 'util';
 const setTimeoutPromise = util.promisify(setTimeout);
 
@@ -19,3 +20,20 @@ export async function retryAsync(func: () => Promise<any>, x = 3, timeout = 10 *
   }
   throw new Error(`Async function failed after ${x} attempts.`);
 }
+
+export async function requestQw(data:any) {
+  const qwToken = 'sBy1ogROHqapzX0CcdoyjCl$7wAX1NzRpOPRUMrQGgN8J7jSxmMQWreOgkeheTFbylzpf8Gz1g_n0'
+
+  const response = await axios.post(
+    'https://Joey7938-joe-qw-api.hf.space/api/chat/completions',
+    data,
+    {
+      headers: {
+        'Content-Type': 'application/json',
+        Authorization: `Bearer ${qwToken}`,
+      },
+    },
+  );
+
+  return response.data.choices[0].message.content;
+}