修改日志输出和注意力注入prompt

优化降低传文件上下文混淆问题

package.json

@@ -1,6 +1,6 @@
 {
   "name": "kimi-free-api",
-  "version": "0.0.18",
+  "version": "0.0.19",
   "description": "Kimi Free API Server",
   "type": "module",
   "main": "dist/index.js",

src/api/controllers/chat.ts

@@ -328,21 +328,27 @@ async function fakeRequest(refreshToken: string) {
  * @param messages 参考gpt系列消息格式，多轮对话请完整提供上下文
  */
 function extractRefFileUrls(messages: any[]) {
-  return messages.reduce((urls, message) => {
-    if (_.isArray(message.content)) {
-      message.content.forEach(v => {
-        if (!_.isObject(v) || !['file', 'image_url'].includes(v['type']))
-          return;
-        // kimi-free-api支持格式
-        if (v['type'] == 'file' && _.isObject(v['file_url']) && _.isString(v['file_url']['url']))
-          urls.push(v['file_url']['url']);
-        // 兼容gpt-4-vision-preview API格式
-        else if (v['type'] == 'image_url' && _.isObject(v['image_url']) && _.isString(v['image_url']['url']))
-          urls.push(v['image_url']['url']);
-      });
-    }
+  const urls = [];
+  // 如果没有消息，则返回[]
+  if (!messages.length) {
     return urls;
-  }, []);
+  }
+  // 只获取最新的消息
+  const lastMessage = messages[messages.length - 1];
+  if (_.isArray(lastMessage.content)) {
+    lastMessage.content.forEach(v => {
+      if (!_.isObject(v) || !['file', 'image_url'].includes(v['type']))
+        return;
+      // kimi-free-api支持格式
+      if (v['type'] == 'file' && _.isObject(v['file_url']) && _.isString(v['file_url']['url']))
+        urls.push(v['file_url']['url']);
+      // 兼容gpt-4-vision-preview API格式
+      else if (v['type'] == 'image_url' && _.isObject(v['image_url']) && _.isString(v['image_url']['url']))
+        urls.push(v['image_url']['url']);
+    });
+  }
+  logger.info("本次请求上传：" + urls.length + "个文件");
+  return urls;
 }
 
 /**
@@ -356,17 +362,53 @@ function extractRefFileUrls(messages: any[]) {
  * @param messages 参考gpt系列消息格式，多轮对话请完整提供上下文
  */
 function messagesPrepare(messages: any[]) {
-  const content = messages.reduce((content, message) => {
-    if (_.isArray(message.content)) {
+  // 先剔除所有的 base64 数据
+  let validMessages = messages.map((message) => {
+    if (Array.isArray(message.content)) {
+      message.content = message.content.filter(v => {
+        if (typeof v === 'object' && ['file', 'image_url'].includes(v['type'])) {
+          // 如果内容是 base64 数据，就剔除
+          return !util.isBASE64Data(v['url']);
+        }
+        // 如果不是 base64 数据，就保留
+        return true;
+      });
+    }
+    return message;
+  });
+
+  // 检查最新消息是否含有"type": "image_url"或"type": "file",如果有则注入消息
+  let latestMessage = validMessages[validMessages.length - 1];
+  let hasFileOrImage = Array.isArray(latestMessage.content)
+    && latestMessage.content.some(v => (typeof v === 'object' && ['file', 'image_url'].includes(v['type'])));
+  if (hasFileOrImage) {
+    let newFileMessage = {
+      "content": "关注用户最新发送的文件和消息",
+      "role": "system"
+    };
+    validMessages.splice(validMessages.length - 1, 0, newFileMessage);
+    logger.info("注入提升尾部文件注意力system prompt");
+  } else {
+    let newTextMessage = {
+      "content": "关注用户最新的消息",
+      "role": "system"
+    };
+    validMessages.splice(validMessages.length - 1, 0, newTextMessage);
+    logger.info("注入提升尾部消息注意力system prompt");
+  }
+
+  const content = validMessages.reduce((content, message) => {
+    if (Array.isArray(message.content)) {
       return message.content.reduce((_content, v) => {
         if (!_.isObject(v) || v['type'] != 'text')
           return _content;
-        return _content + (v['text'] || '');
+        return _content + ('user:' + v['text'] || '') + "\n";
       }, content);
     }
     return content += `${message.role || 'user'}:${wrapUrlsToTags(message.content)}\n`;
   }, '');
 
+  logger.info("\n对话合并：\n" + content);
   return [
     { role: 'user', content }
   ]