Release 0.0.15

README.md

@@ -74,10 +74,14 @@ https://udify.app/chat/Po0F6BMJ15q5vu2P
 
 从 [kimi.moonshot.cn](https://kimi.moonshot.cn) 获取refresh_token
 
-进入kimi随便发起一个对话，然后F12打开开发者工具，从Application > Local Storage中找到refresh_token的值，这将作为Authorization的Bearer Token值：`Authorization: Bearer TOKEN`
+进入kimi随便发起一个对话，然后F12打开开发者工具，从Application > Local Storage中找到`refresh_token`的值，这将作为Authorization的Bearer Token值：`Authorization: Bearer TOKEN`
 
 ![example0](./doc/example-0.png)
 
+如果你看到的`refresh_token`是一个数组，请使用`.`拼接起来再使用。
+
+![example8](./doc/example-8.jpg)
+
 ### 多账号接入
 
 目前kimi限制普通账号每3小时内只能进行30轮长文本的问答，你可以通过提供多个账号的refresh_token并使用`,`拼接提供：
@@ -227,9 +231,9 @@ Authorization: Bearer [refresh_token]
         }
     ],
     "usage": {
-        "prompt_tokens": 0,
-        "completion_tokens": 0,
-        "total_tokens": 0
+        "prompt_tokens": 1,
+        "completion_tokens": 1,
+        "total_tokens": 2
     },
     "created": 1710152062
 }
@@ -289,9 +293,9 @@ Authorization: Bearer [refresh_token]
         }
     ],
     "usage": {
-        "prompt_tokens": 0,
-        "completion_tokens": 0,
-        "total_tokens": 0
+        "prompt_tokens": 1,
+        "completion_tokens": 1,
+        "total_tokens": 2
     },
     "created": 100920
 }
@@ -353,9 +357,9 @@ Authorization: Bearer [refresh_token]
         }
     ],
     "usage": {
-        "prompt_tokens": 0,
-        "completion_tokens": 0,
-        "total_tokens": 0
+        "prompt_tokens": 1,
+        "completion_tokens": 1,
+        "total_tokens": 2
     },
     "created": 1710123627
 }
@@ -378,4 +382,8 @@ tcp_nopush on;
 tcp_nodelay on;
 # 设置保持连接的超时时间，这里设置为120秒。如果在这段时间内，客户端和服务器之间没有进一步的通信，连接将被关闭。
 keepalive_timeout 120;
-```
+```
+
+### Token统计
+
+由于推理侧不再kimi-free-api，因此token不可统计，将以固定数字返回。

package.json

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

src/api/controllers/chat.ts

@@ -24,7 +24,7 @@ const FAKE_HEADERS = {
   'Accept-Encoding': 'gzip, deflate, br, zstd',
   'Accept-Language': 'zh-CN,zh;q=0.9',
   'Origin': 'https://kimi.moonshot.cn',
-  'Cookie': util.generateCookie(),
+  // 'Cookie': util.generateCookie(),
   'R-Timezone': 'Asia/Shanghai',
   'Sec-Ch-Ua': '"Chromium";v="122", "Not(A:Brand";v="24", "Google Chrome";v="122"',
   'Sec-Ch-Ua-Mobile': '?0',
@@ -57,7 +57,7 @@ async function requestToken(refreshToken: string) {
     const result = await axios.get('https://kimi.moonshot.cn/api/auth/token/refresh', {
       headers: {
         Authorization: `Bearer ${refreshToken}`,
-        Referer: 'https://kimi.moonshot.cn',
+        Referer: 'https://kimi.moonshot.cn/',
         ...FAKE_HEADERS
       },
       timeout: 15000,
@@ -74,7 +74,7 @@ async function requestToken(refreshToken: string) {
     }
   })()
     .then(result => {
-      if(accessTokenRequestQueueMap[refreshToken]) {
+      if (accessTokenRequestQueueMap[refreshToken]) {
         accessTokenRequestQueueMap[refreshToken].forEach(resolve => resolve(result));
         delete accessTokenRequestQueueMap[refreshToken];
       }
@@ -82,13 +82,13 @@ async function requestToken(refreshToken: string) {
       return result;
     })
     .catch(err => {
-      if(accessTokenRequestQueueMap[refreshToken]) {
+      if (accessTokenRequestQueueMap[refreshToken]) {
         accessTokenRequestQueueMap[refreshToken].forEach(resolve => resolve(err));
         delete accessTokenRequestQueueMap[refreshToken];
       }
       return err;
     });
-  if(_.isError(result))
+  if (_.isError(result))
     throw result;
   return result;
 }
@@ -128,7 +128,7 @@ async function createConversation(name: string, refreshToken: string) {
   }, {
     headers: {
       Authorization: `Bearer ${token}`,
-      Referer: 'https://kimi.moonshot.cn',
+      Referer: 'https://kimi.moonshot.cn/',
       ...FAKE_HEADERS
     },
     timeout: 15000,
@@ -177,6 +177,10 @@ async function createCompletion(messages: any[], refreshToken: string, useSearch
     const refFileUrls = extractRefFileUrls(messages);
     const refs = refFileUrls.length ? await Promise.all(refFileUrls.map(fileUrl => uploadFile(fileUrl, refreshToken))) : [];
 
+    // 伪装调用获取用户信息
+    fakeRequest(refreshToken)
+      .catch(err => logger.error(err));
+
     // 创建会话
     const convId = await createConversation(`cmpl-${util.uuid(false)}`, refreshToken);
 
@@ -210,7 +214,7 @@ async function createCompletion(messages: any[], refreshToken: string, useSearch
     return answer;
   })()
     .catch(err => {
-      if(retryCount < MAX_RETRY_COUNT) {
+      if (retryCount < MAX_RETRY_COUNT) {
         logger.error(`Stream response error: ${err.message}`);
         logger.warn(`Try again after ${RETRY_DELAY / 1000}s...`);
         return (async () => {
@@ -238,6 +242,10 @@ async function createCompletionStream(messages: any[], refreshToken: string, use
     const refFileUrls = extractRefFileUrls(messages);
     const refs = refFileUrls.length ? await Promise.all(refFileUrls.map(fileUrl => uploadFile(fileUrl, refreshToken))) : [];
 
+    // 伪装调用获取用户信息
+    fakeRequest(refreshToken)
+      .catch(err => logger.error(err));
+
     // 创建会话
     const convId = await createConversation(`cmpl-${util.uuid(false)}`, refreshToken);
 
@@ -268,7 +276,7 @@ async function createCompletionStream(messages: any[], refreshToken: string, use
     });
   })()
     .catch(err => {
-      if(retryCount < MAX_RETRY_COUNT) {
+      if (retryCount < MAX_RETRY_COUNT) {
         logger.error(`Stream response error: ${err.message}`);
         logger.warn(`Try again after ${RETRY_DELAY / 1000}s...`);
         return (async () => {
@@ -280,6 +288,38 @@ async function createCompletionStream(messages: any[], refreshToken: string, use
     });
 }
 
+/**
+ * 调用一些接口伪装访问
+ * 
+ * 随机挑一个
+ * 
+ * @param refreshToken 用于刷新access_token的refresh_token
+ */
+async function fakeRequest(refreshToken: string) {
+  const token = await acquireToken(refreshToken);
+  const options =  {
+    headers: {
+      Authorization: `Bearer ${token}`,
+      Referer: `https://kimi.moonshot.cn/`,
+      ...FAKE_HEADERS
+    }
+  };
+  await [
+    () => axios.get('https://kimi.moonshot.cn/api/user', options),
+    () => axios.get('https://kimi.moonshot.cn/api/chat_1m/user/status', options),
+    () => axios.post('https://kimi.moonshot.cn/api/chat/list', {
+      offset: 0,
+      size: 50
+    }, options),
+    () => axios.post('https://kimi.moonshot.cn/api/show_case/list', {
+      offset: 0,
+      size: 4,
+      enable_cache: true,
+      order: "asc"
+    }, options)
+  ][Math.floor(Math.random() * 4)]();
+}
+
 /**
  * 提取消息中引用的文件URL
  * 
@@ -356,7 +396,7 @@ async function preSignUrl(filename: string, refreshToken: string) {
     timeout: 15000,
     headers: {
       Authorization: `Bearer ${token}`,
-      Referer: `https://kimi.moonshot.cn`,
+      Referer: `https://kimi.moonshot.cn/`,
       ...FAKE_HEADERS
     },
     validateStatus: () => true
@@ -437,7 +477,7 @@ async function uploadFile(fileUrl: string, refreshToken: string) {
     headers: {
       'Content-Type': mimeType,
       Authorization: `Bearer ${token}`,
-      Referer: `https://kimi.moonshot.cn`,
+      Referer: `https://kimi.moonshot.cn/`,
       ...FAKE_HEADERS
     },
     validateStatus: () => true
@@ -453,7 +493,7 @@ async function uploadFile(fileUrl: string, refreshToken: string) {
   }, {
     headers: {
       Authorization: `Bearer ${token}`,
-      Referer: `https://kimi.moonshot.cn`,
+      Referer: `https://kimi.moonshot.cn/`,
       ...FAKE_HEADERS
     }
   });
@@ -466,7 +506,7 @@ async function uploadFile(fileUrl: string, refreshToken: string) {
   }, {
     headers: {
       Authorization: `Bearer ${token}`,
-      Referer: `https://kimi.moonshot.cn`,
+      Referer: `https://kimi.moonshot.cn/`,
       ...FAKE_HEADERS
     }
   });
@@ -506,7 +546,7 @@ function checkResult(result: AxiosResponse, refreshToken: string) {
  */
 async function receiveStream(convId: string, stream: any) {
   return new Promise((resolve, reject) => {
-    // 第一条消息初始化
+    // 消息初始化
     const data = {
       id: convId,
       model: MODEL_NAME,
@@ -514,7 +554,7 @@ async function receiveStream(convId: string, stream: any) {
       choices: [
         { index: 0, message: { role: 'assistant', content: '' }, finish_reason: 'stop' }
       ],
-      usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+      usage: { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 },
       created: util.unixTimestamp()
     };
     let refContent = '';
@@ -612,7 +652,7 @@ function createTransStream(convId: string, stream: any, endCallback?: Function)
               } : {}, finish_reason: 'stop'
             }
           ],
-          usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+          usage: { prompt_tokens: 1, completion_tokens: 1, total_tokens: 2 },
           created
         })}\n\n`;
         !transStream.closed && transStream.write(data);