diff --git a/package.json b/package.json index a3c4266..5b4b3a1 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "kimi-free-api", - "version": "0.0.3", + "version": "0.0.5", "description": "Kimi Free Server", "type": "module", "main": "dist/index.js", diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts index 5246c3f..4013e01 100644 --- a/src/api/controllers/chat.ts +++ b/src/api/controllers/chat.ts @@ -126,13 +126,17 @@ async function createCompletionStream(messages: any[], refreshToken: string, use function messagesPrepare(messages: any[]) { const content = messages.reduce((content, message) => { - return content += `${message.role || 'user'}:${message.content}\n`; + return content += `${message.role || 'user'}:${wrapUrlsToTags(message.content)}\n`; }, ''); return [ { role: 'user', content } ] } +function wrapUrlsToTags(content: string) { + return content.replace(/https?:\/\/(www\.)?[-a-zA-Z0-9@:%._\+~#=]{2,256}\.[a-z]{2,6}\b([-a-zA-Z0-9@:%_\+.~#?&//=]*)/gi, url => `${url}`); +} + function checkResult(result: AxiosResponse, refreshToken: string) { if(result.status == 401) { accessTokenMap.delete(refreshToken);