🦄 优化web搜索输出

LLM-Red-Team · Oct 27, 2024 · bc37f56 · bc37f56
1 parent b2ae055
commit bc37f56
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -723,6 +723,7 @@ function checkResult(result: AxiosResponse, refreshToken: string) {
  * @param stream 消息流
  */
 async function receiveStream(model: string, convId: string, stream: any) {
+  let webSearchCount = 0;
   return new Promise((resolve, reject) => {
     // 消息初始化
     const data = {
@@ -756,8 +757,10 @@ async function receiveStream(model: string, convId: string, stream: any) {
           resolve(data);
         }
         // 处理联网搜索
-        else if (!silentSearch && result.event == 'search_plus' && result.msg && result.msg.type == 'get_res')
-          refContent += `${result.msg.title} - ${result.msg.url}\n`;
+        else if (!silentSearch && result.event == 'search_plus' && result.msg && result.msg.type == 'get_res'){
+          webSearchCount += 1;
+          refContent += `【检索 ${webSearchCount}】 [${result.msg.title}](${result.msg.url})\n\n`;
+        }
         // else
         //   logger.warn(result.event, result);
       }
@@ -788,6 +791,7 @@ function createTransStream(model: string, convId: string, stream: any, endCallba
   const created = util.unixTimestamp();
   // 创建转换流
   const transStream = new PassThrough();
+  let webSearchCount = 0;
   let searchFlag = false;
   const silentSearch = model.indexOf('silent_search') != -1;
   !transStream.closed && transStream.write(`data: ${JSON.stringify({
@@ -847,14 +851,15 @@ function createTransStream(model: string, convId: string, stream: any, endCallba
       else if (!silentSearch && result.event == 'search_plus' && result.msg && result.msg.type == 'get_res') {
         if (!searchFlag)
           searchFlag = true;
+        webSearchCount += 1;
         const data = `data: ${JSON.stringify({
           id: convId,
           model,
           object: 'chat.completion.chunk',
           choices: [
             {
               index: 0, delta: {
-                content: `检索 ${result.msg.title} - ${result.msg.url} ...\n`
+                content: `【检索 ${webSearchCount}】 [${result.msg.title}](${result.msg.url})\n`
               }, finish_reason: null
             }
           ],