ChatGPT/kimi-free-api
1
0
Fork 0
mirror of https://github.com/LLM-Red-Team/kimi-free-api.git synced 2025-10-18 16:23:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

支持多路refresh_token

Browse Source
This commit is contained in:
Vinlic
2024-03-14 23:21:22 +08:00
parent 17809274a5
commit 4949b16091
3 changed files with 28 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/api/routes/chat.ts
View File

@@ -15,17 +15,19 @@ export default {
request
.validate('body.messages', _.isArray)
.validate('headers.authorization', _.isString)
const token = request.headers.authorization;
const refreshToken = token.replace('Bearer ', '');
// refresh_token切分
const tokens = chat.tokenSplit(request.headers.authorization);
// 随机挑选一个refresh_token
const token = _.sample(tokens);
const messages = request.body.messages;
if (request.body.stream) {
const stream = await chat.createCompletionStream(request.body.messages, refreshToken, request.body.use_search);
const stream = await chat.createCompletionStream(request.body.messages, token, request.body.use_search);
return new Response(stream, {
type: "text/event-stream"
});
}
else
return await chat.createCompletion(messages, refreshToken, request.body.use_search);
return await chat.createCompletion(messages, token, request.body.use_search);
}
}