mirror of
https://github.com/LLM-Red-Team/kimi-free-api.git
synced 2025-10-19 08:43:51 +00:00
Compare commits
6 Commits
Author | SHA1 | Date | |
---|---|---|---|
![]() |
a5beade70a | ||
![]() |
1395278a6e | ||
![]() |
08a4b2e720 | ||
![]() |
2f26d29de1 | ||
![]() |
3bb24b26d3 | ||
![]() |
dd8bb923a8 |
46
.github/workflows/docker-image.yml
vendored
Normal file
46
.github/workflows/docker-image.yml
vendored
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
name: Build and Push Docker Image
|
||||||
|
|
||||||
|
on:
|
||||||
|
release:
|
||||||
|
types: [created]
|
||||||
|
workflow_dispatch:
|
||||||
|
inputs:
|
||||||
|
tag:
|
||||||
|
description: 'Tag Name'
|
||||||
|
required: true
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build-and-push:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v2
|
||||||
|
|
||||||
|
- name: Set up Docker Buildx
|
||||||
|
uses: docker/setup-buildx-action@v1
|
||||||
|
|
||||||
|
- name: Login to Docker Hub
|
||||||
|
uses: docker/login-action@v1
|
||||||
|
with:
|
||||||
|
username: ${{ secrets.DOCKERHUB_USERNAME }}
|
||||||
|
password: ${{ secrets.DOCKERHUB_PASSWORD }}
|
||||||
|
|
||||||
|
- name: Set tag name
|
||||||
|
id: tag_name
|
||||||
|
run: |
|
||||||
|
if [ "${{ github.event_name }}" = "release" ]; then
|
||||||
|
echo "::set-output name=tag::${GITHUB_REF#refs/tags/}"
|
||||||
|
elif [ "${{ github.event_name }}" = "workflow_dispatch" ]; then
|
||||||
|
echo "::set-output name=tag::${{ github.event.inputs.tag }}"
|
||||||
|
fi
|
||||||
|
|
||||||
|
- name: Build and push Docker image with Release tag
|
||||||
|
uses: docker/build-push-action@v2
|
||||||
|
with:
|
||||||
|
context: .
|
||||||
|
file: ./Dockerfile
|
||||||
|
push: true
|
||||||
|
tags: |
|
||||||
|
vinlic/kimi-free-api:${{ steps.tag_name.outputs.tag }}
|
||||||
|
vinlic/kimi-free-api:latest
|
||||||
|
platforms: linux/amd64,linux/arm64
|
||||||
|
build-args: TARGETPLATFORM=${{ matrix.platform }}
|
@@ -66,6 +66,10 @@ https://udify.app/chat/Po0F6BMJ15q5vu2P
|
|||||||
|
|
||||||

|

|
||||||
|
|
||||||
|
### 100线程并发测试
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
## 接入准备
|
## 接入准备
|
||||||
|
|
||||||
从 [kimi.moonshot.cn](https://kimi.moonshot.cn) 获取refresh_token
|
从 [kimi.moonshot.cn](https://kimi.moonshot.cn) 获取refresh_token
|
||||||
|
BIN
doc/example-7.jpg
Normal file
BIN
doc/example-7.jpg
Normal file
Binary file not shown.
After Width: | Height: | Size: 98 KiB |
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "kimi-free-api",
|
"name": "kimi-free-api",
|
||||||
"version": "0.0.12",
|
"version": "0.0.13",
|
||||||
"description": "Kimi Free API Server",
|
"description": "Kimi Free API Server",
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"main": "dist/index.js",
|
"main": "dist/index.js",
|
||||||
|
@@ -14,6 +14,10 @@ import util from '@/lib/util.ts';
|
|||||||
const MODEL_NAME = 'kimi';
|
const MODEL_NAME = 'kimi';
|
||||||
// access_token有效期
|
// access_token有效期
|
||||||
const ACCESS_TOKEN_EXPIRES = 300;
|
const ACCESS_TOKEN_EXPIRES = 300;
|
||||||
|
// 最大重试次数
|
||||||
|
const MAX_RETRY_COUNT = 3;
|
||||||
|
// 重试延迟
|
||||||
|
const RETRY_DELAY = 5000;
|
||||||
// 伪装headers
|
// 伪装headers
|
||||||
const FAKE_HEADERS = {
|
const FAKE_HEADERS = {
|
||||||
'Accept': '*/*',
|
'Accept': '*/*',
|
||||||
@@ -163,8 +167,10 @@ async function removeConversation(convId: string, refreshToken: string) {
|
|||||||
* @param messages 参考gpt系列消息格式,多轮对话请完整提供上下文
|
* @param messages 参考gpt系列消息格式,多轮对话请完整提供上下文
|
||||||
* @param refreshToken 用于刷新access_token的refresh_token
|
* @param refreshToken 用于刷新access_token的refresh_token
|
||||||
* @param useSearch 是否开启联网搜索
|
* @param useSearch 是否开启联网搜索
|
||||||
|
* @param retryCount 重试次数
|
||||||
*/
|
*/
|
||||||
async function createCompletion(messages: any[], refreshToken: string, useSearch = true) {
|
async function createCompletion(messages: any[], refreshToken: string, useSearch = true, retryCount = 0) {
|
||||||
|
return (async () => {
|
||||||
logger.info(messages);
|
logger.info(messages);
|
||||||
|
|
||||||
// 提取引用文件URL并上传kimi获得引用的文件ID列表
|
// 提取引用文件URL并上传kimi获得引用的文件ID列表
|
||||||
@@ -202,6 +208,18 @@ async function createCompletion(messages: any[], refreshToken: string, useSearch
|
|||||||
.catch(err => console.error(err));
|
.catch(err => console.error(err));
|
||||||
|
|
||||||
return answer;
|
return answer;
|
||||||
|
})()
|
||||||
|
.catch(err => {
|
||||||
|
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);
|
||||||
|
})();
|
||||||
|
}
|
||||||
|
throw err;
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -210,8 +228,10 @@ async function createCompletion(messages: any[], refreshToken: string, useSearch
|
|||||||
* @param messages 参考gpt系列消息格式,多轮对话请完整提供上下文
|
* @param messages 参考gpt系列消息格式,多轮对话请完整提供上下文
|
||||||
* @param refreshToken 用于刷新access_token的refresh_token
|
* @param refreshToken 用于刷新access_token的refresh_token
|
||||||
* @param useSearch 是否开启联网搜索
|
* @param useSearch 是否开启联网搜索
|
||||||
|
* @param retryCount 重试次数
|
||||||
*/
|
*/
|
||||||
async function createCompletionStream(messages: any[], refreshToken: string, useSearch = true) {
|
async function createCompletionStream(messages: any[], refreshToken: string, useSearch = true, retryCount = 0) {
|
||||||
|
return (async () => {
|
||||||
logger.info(messages);
|
logger.info(messages);
|
||||||
|
|
||||||
// 提取引用文件URL并上传kimi获得引用的文件ID列表
|
// 提取引用文件URL并上传kimi获得引用的文件ID列表
|
||||||
@@ -246,6 +266,18 @@ async function createCompletionStream(messages: any[], refreshToken: string, use
|
|||||||
removeConversation(convId, refreshToken)
|
removeConversation(convId, refreshToken)
|
||||||
.catch(err => console.error(err));
|
.catch(err => console.error(err));
|
||||||
});
|
});
|
||||||
|
})()
|
||||||
|
.catch(err => {
|
||||||
|
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);
|
||||||
|
})();
|
||||||
|
}
|
||||||
|
throw err;
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@@ -29,6 +29,11 @@ export default class Exception extends Error {
|
|||||||
this.errmsg = _errmsg || errmsg;
|
this.errmsg = _errmsg || errmsg;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
compare(exception: (string | number)[]) {
|
||||||
|
const [errcode] = exception as [number, string];
|
||||||
|
return this.errcode == errcode;
|
||||||
|
}
|
||||||
|
|
||||||
setHTTPStatusCode(value: number) {
|
setHTTPStatusCode(value: number) {
|
||||||
this.httpStatusCode = value;
|
this.httpStatusCode = value;
|
||||||
return this;
|
return this;
|
||||||
|
Reference in New Issue
Block a user