Merge pull request #16 from peanut996/master

添加health check api
2025-10-15 14:50:37 +00:00 · 2024-03-20 13:47:06 +08:00 · 2024-03-20 13:46:21 +08:00 · 2024-03-20 01:46:22 +08:00 · 2024-03-20 01:37:25 +08:00 · 2024-03-19 15:56:49 +08:00
6 changed files with 98 additions and 31 deletions
--- a/README.md
+++ b/README.md
@@ -9,6 +9,12 @@

 与ChatGPT接口完全兼容。

+还有以下两个free-api欢迎关注：
+
+ZhipuAI (智谱清言) 接口转API [glm-free-api](https://github.com/LLM-Red-Team/glm-free-api)
+
+聆心智能 (Emohaa) 接口转API [emohaa-free-api](https://github.com/LLM-Red-Team/emohaa-free-api)
+
 ## 目录

 * [声明](#声明)
@@ -201,6 +207,8 @@ Authorization: Bearer [refresh_token]
 请求数据：
 ```json
 {
+    // 模型名称随意填写，如果不希望输出检索过程模型名称请包含silent_search
+    "model": "kimi",
    "messages": [
        {
            "role": "user",
@@ -254,6 +262,8 @@ Authorization: Bearer [refresh_token]
 请求数据：
 ```json
 {
+    // 模型名称随意填写，如果不希望输出检索过程模型名称请包含silent_search
+    "model": "kimi",
    "messages": [
        {
            "role": "user",
@@ -318,6 +328,8 @@ Authorization: Bearer [refresh_token]
 请求数据：
 ```json
 {
+    // 模型名称随意填写，如果不希望输出检索过程模型名称请包含silent_search
+    "model": "kimi",
    "messages": [
        {
            "role": "user",
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
  "name": "kimi-free-api",
-  "version": "0.0.14",
+  "version": "0.0.16",
  "description": "Kimi Free API Server",
  "type": "module",
  "main": "dist/index.js",
--- a/src/api/controllers/chat.ts
+++ b/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,
@@ -164,12 +164,13 @@ async function removeConversation(convId: string, refreshToken: string) {
 /**
 * 同步对话补全
 * 
+ * @param model 模型名称
 * @param messages 参考gpt系列消息格式，多轮对话请完整提供上下文
 * @param refreshToken 用于刷新access_token的refresh_token
 * @param useSearch 是否开启联网搜索
 * @param retryCount 重试次数
 */
-async function createCompletion(messages: any[], refreshToken: string, useSearch = true, retryCount = 0) {
+async function createCompletion(model = MODEL_NAME, messages: any[], refreshToken: string, useSearch = true, retryCount = 0) {
  return (async () => {
    logger.info(messages);

@@ -177,6 +178,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);

@@ -200,7 +205,7 @@ async function createCompletion(messages: any[], refreshToken: string, useSearch

    const streamStartTime = util.timestamp();
    // 接收流为输出文本
-    const answer = await receiveStream(convId, result.data);
+    const answer = await receiveStream(model, convId, result.data);
    logger.success(`Stream has completed transfer ${util.timestamp() - streamStartTime}ms`);

    // 异步移除会话，如果消息不合规，此操作可能会抛出数据库错误异常，请忽略
@@ -210,12 +215,12 @@ 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 () => {
          await new Promise(resolve => setTimeout(resolve, RETRY_DELAY));
-          return createCompletion(messages, refreshToken, useSearch, retryCount + 1);
+          return createCompletion(model, messages, refreshToken, useSearch, retryCount + 1);
        })();
      }
      throw err;
@@ -225,12 +230,13 @@ async function createCompletion(messages: any[], refreshToken: string, useSearch
 /**
 * 流式对话补全
 * 
+ * @param model 模型名称
 * @param messages 参考gpt系列消息格式，多轮对话请完整提供上下文
 * @param refreshToken 用于刷新access_token的refresh_token
 * @param useSearch 是否开启联网搜索
 * @param retryCount 重试次数
 */
-async function createCompletionStream(messages: any[], refreshToken: string, useSearch = true, retryCount = 0) {
+async function createCompletionStream(model = MODEL_NAME, messages: any[], refreshToken: string, useSearch = true, retryCount = 0) {
  return (async () => {
    logger.info(messages);

@@ -238,6 +244,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);

@@ -260,7 +270,7 @@ async function createCompletionStream(messages: any[], refreshToken: string, use
    });
    const streamStartTime = util.timestamp();
    // 创建转换流将消息格式转换为gpt兼容格式
-    return createTransStream(convId, result.data, () => {
+    return createTransStream(model, convId, result.data, () => {
      logger.success(`Stream has completed transfer ${util.timestamp() - streamStartTime}ms`);
      // 流传输结束后异步移除会话，如果消息不合规，此操作可能会抛出数据库错误异常，请忽略
      removeConversation(convId, refreshToken)
@@ -268,18 +278,50 @@ 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 () => {
          await new Promise(resolve => setTimeout(resolve, RETRY_DELAY));
-          return createCompletionStream(messages, refreshToken, useSearch, retryCount + 1);
+          return createCompletionStream(model, messages, refreshToken, useSearch, retryCount + 1);
        })();
      }
      throw err;
    });
 }

+/**
+ * 调用一些接口伪装访问
+ * 
+ * 随机挑一个
+ * 
+ * @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 +398,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 +479,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 +495,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 +508,7 @@ async function uploadFile(fileUrl: string, refreshToken: string) {
  }, {
    headers: {
      Authorization: `Bearer ${token}`,
-      Referer: `https://kimi.moonshot.cn`,
+      Referer: `https://kimi.moonshot.cn/`,
      ...FAKE_HEADERS
    }
  });
@@ -501,15 +543,16 @@ function checkResult(result: AxiosResponse, refreshToken: string) {
 /**
 * 从流接收完整的消息内容
 * 
+ * @param model 模型名称
 * @param convId 会话ID
 * @param stream 消息流
 */
-async function receiveStream(convId: string, stream: any) {
+async function receiveStream(model: string, convId: string, stream: any) {
  return new Promise((resolve, reject) => {
    // 消息初始化
    const data = {
      id: convId,
-      model: MODEL_NAME,
+      model,
      object: 'chat.completion',
      choices: [
        { index: 0, message: { role: 'assistant', content: '' }, finish_reason: 'stop' }
@@ -518,6 +561,7 @@ async function receiveStream(convId: string, stream: any) {
      created: util.unixTimestamp()
    };
    let refContent = '';
+    const silentSearch = model.indexOf('silent_search') != -1;
    const parser = createParser(event => {
      try {
        if (event.type !== "event") return;
@@ -536,7 +580,7 @@ async function receiveStream(convId: string, stream: any) {
          resolve(data);
        }
        // 处理联网搜索
-        else if (result.event == 'search_plus' && result.msg && result.msg.type == 'get_res')
+        else if (!silentSearch && result.event == 'search_plus' && result.msg && result.msg.type == 'get_res')
          refContent += `${result.msg.title}(${result.msg.url})\n`;
        // else
        //   logger.warn(result.event, result);
@@ -558,19 +602,21 @@ async function receiveStream(convId: string, stream: any) {
 * 
 * 将流格式转换为gpt兼容流格式
 * 
+ * @param model 模型名称
 * @param convId 会话ID
 * @param stream 消息流
 * @param endCallback 传输结束回调
 */
-function createTransStream(convId: string, stream: any, endCallback?: Function) {
+function createTransStream(model: string, convId: string, stream: any, endCallback?: Function) {
  // 消息创建时间
  const created = util.unixTimestamp();
  // 创建转换流
  const transStream = new PassThrough();
  let searchFlag = false;
+  const silentSearch = model.indexOf('silent_search') != -1;
  !transStream.closed && transStream.write(`data: ${JSON.stringify({
    id: convId,
-    model: MODEL_NAME,
+    model,
    object: 'chat.completion.chunk',
    choices: [
      { index: 0, delta: { role: 'assistant', content: '' }, finish_reason: null }
@@ -588,7 +634,7 @@ function createTransStream(convId: string, stream: any, endCallback?: Function)
      if (result.event == 'cmpl') {
        const data = `data: ${JSON.stringify({
          id: convId,
-          model: MODEL_NAME,
+          model,
          object: 'chat.completion.chunk',
          choices: [
            { index: 0, delta: { content: (searchFlag ? '\n' : '') + result.text }, finish_reason: null }
@@ -603,7 +649,7 @@ function createTransStream(convId: string, stream: any, endCallback?: Function)
      else if (result.event == 'all_done' || result.event == 'error') {
        const data = `data: ${JSON.stringify({
          id: convId,
-          model: MODEL_NAME,
+          model,
          object: 'chat.completion.chunk',
          choices: [
            {
@@ -620,12 +666,12 @@ function createTransStream(convId: string, stream: any, endCallback?: Function)
        endCallback && endCallback();
      }
      // 处理联网搜索
-      else if (result.event == 'search_plus' && result.msg && result.msg.type == 'get_res') {
+      else if (!silentSearch && result.event == 'search_plus' && result.msg && result.msg.type == 'get_res') {
        if (!searchFlag)
          searchFlag = true;
        const data = `data: ${JSON.stringify({
          id: convId,
-          model: MODEL_NAME,
+          model,
          object: 'chat.completion.chunk',
          choices: [
            {
--- a/src/api/routes/chat.ts
+++ b/src/api/routes/chat.ts
@@ -19,15 +19,16 @@ export default {
            const tokens = chat.tokenSplit(request.headers.authorization);
            // 随机挑选一个refresh_token
            const token = _.sample(tokens);
+            const model = request.body.model;
            const messages =  request.body.messages;
            if (request.body.stream) {
-                const stream = await chat.createCompletionStream(request.body.messages, token, request.body.use_search);
+                const stream = await chat.createCompletionStream(model, messages, token, request.body.use_search);
                return new Response(stream, {
                    type: "text/event-stream"
                });
            }
            else
-                return await chat.createCompletion(messages, token, request.body.use_search);
+                return await chat.createCompletion(model, messages, token, request.body.use_search);
        }

    }
--- a/src/api/routes/index.ts
+++ b/src/api/routes/index.ts
@@ -1,5 +1,7 @@
 import chat from "./chat.ts";
+import ping from "./ping.ts";

 export default [
-    chat
+    chat,
+    ping
 ];
--- a/src/api/routes/ping.ts
+++ b/src/api/routes/ping.ts
@@ -0,0 +1,6 @@
+export default {
+  prefix: '/ping',
+  get: {
+    '': async () => "pong"
+  }
+}
Author	SHA1	Message	Date
Vinlic科技	eccce82ade	Merge pull request #16 from peanut996/master 添加health check api	2024-03-20 13:47:06 +08:00
peanut996	4fe9b654f5	添加health check api	2024-03-20 13:46:21 +08:00
Vinlic	7cbebf780c	update README	2024-03-20 01:46:22 +08:00
Vinlic	909796bd91	支持模型名称包含silent_search来静默搜索不输出搜索过程	2024-03-20 01:37:25 +08:00
Vinlic	b8134a64a5	Release 0.0.15	2024-03-19 15:56:49 +08:00
Vinlic	c9b3574b0b	增加伪装请求，企图降低封号几率	2024-03-19 15:56:27 +08:00