Spaces:
Runtime error
Runtime error
File size: 5,408 Bytes
d70d08b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 |
/*
* https://gitgud.io/ahsk/clewd
* https://github.com/h-a-s-k/clewd
*/
'use strict';
const {randomInt, randomBytes} = require('node:crypto'), {version: Version} = require('../package.json'), Encoder = (new TextDecoder,
new TextEncoder), Main = 'clewd v' + Version + '修改版 by tera', Replacements = {
user: 'Human',
assistant: 'Assistant',
system: '',
example_user: 'H',
example_assistant: 'A'
}, DangerChars = [ ...new Set([ ...Object.values(Replacements).join(''), ...'\n', ...':', ...'\\n' ]) ].filter((char => ' ' !== char)).sort(), AI = {
end: () => Buffer.from([ 104, 116, 116, 112, 115, 58, 47, 47, 99, 108, 97, 117, 100, 101, 46, 97, 105 ]).toString(),
mdl: () => Buffer.from([ 99, 108, 97, 117, 100, 101, 45, 50 ]).toString(),
zone: () => Buffer.from([ 65, 109, 101, 114, 105, 99, 97, 47, 78, 101, 119, 95, 89, 111, 114, 107 ]).toString(),
agent: () => Buffer.from([ 77, 111, 122, 105, 108, 108, 97, 47, 53, 46, 48, 32, 40, 77, 97, 99, 105, 110, 116, 111, 115, 104, 59, 32, 73, 110, 116, 101, 108, 32, 77, 97, 99, 32, 79, 83, 32, 88, 32, 49, 48, 95, 49, 53, 95, 55, 41, 32, 65, 112, 112, 108, 101, 87, 101, 98, 75, 105, 116, 47, 53, 51, 55, 46, 51, 54, 32, 40, 75, 72, 84, 77, 76, 44, 32, 108, 105, 107, 101, 32, 71, 101, 99, 107, 111, 41, 32, 67, 104, 114, 111, 109, 101, 47, 49, 49, 52, 46, 48, 46, 48, 46, 48, 32, 83, 97, 102, 97, 114, 105, 47, 53, 51, 55, 46, 51, 54, 32, 69, 100, 103, 47, 49, 49, 52, 46, 48, 46, 49, 56, 50, 51, 46, 55, 57 ]).toString(),
cp: () => Buffer.from([ 55, 55, 49, 44, 52, 56, 54, 53, 45, 52, 56, 54, 54, 45, 52, 56, 54, 55, 45, 52, 57, 49, 57, 53, 45, 52, 57, 49, 57, 57, 45, 52, 57, 49, 57, 54, 45, 52, 57, 50, 48, 48, 45, 53, 50, 51, 57, 51, 45, 53, 50, 51, 57, 50, 45, 52, 57, 49, 55, 49, 45, 52, 57, 49, 55, 50, 45, 49, 53, 54, 45, 49, 53, 55, 45, 52, 55, 45, 53, 51, 44, 48, 45, 50, 51, 45, 54, 53, 50, 56, 49, 45, 49, 48, 45, 49, 49, 45, 51, 53, 45, 49, 54, 45, 53, 45, 49, 51, 45, 49, 56, 45, 53, 49, 45, 52, 53, 45, 52, 51, 45, 50, 55, 45, 49, 55, 53, 49, 51, 45, 50, 49, 44, 50, 57, 45, 50, 51, 45, 50, 52, 44, 48 ]).toString(),
hdr: refPath => ({
'Content-Type': 'application/json',
Referer: `${AI.end()}/${refPath ? 'chat/' + refPath : ''}`,
Origin: '' + AI.end()
})
}, indexOfH = (text, last = false) => {
let location = -1;
const matchesH = text.match(/(?:(?:\\n)|\n){2}((?:Human|H): ?)/gm);
matchesH?.length > 0 && (location = last ? text.lastIndexOf(matchesH[matchesH.length - 1]) : text.indexOf(matchesH[0]));
return location;
}, indexOfA = (text, last = false) => {
let location = -1;
const matchesA = text.match(/(?:(?:\\n)|\n){2}((?:Assistant|A): ?)/gm);
matchesA?.length > 0 && (location = last ? text.lastIndexOf(matchesA[matchesA.length - 1]) : text.indexOf(matchesA[0]));
return location;
};
module.exports.encodeDataJSON = completion => Encoder.encode(`data: ${JSON.stringify(completion)}\n\n`);
module.exports.genericFixes = text => text.replace(/(\r\n|\r|\\n)/gm, '\n');
module.exports.Replacements = Replacements;
module.exports.DangerChars = DangerChars;
module.exports.checkResErr = async (res, throwIt = true) => {
let err, json, errAPI;
if ('string' == typeof res) {
json = JSON.parse(res);
errAPI = json.error;
err = Error(errAPI.message);
} else if (res.status < 200 || res.status >= 300) {
err = Error('Unexpected response code: ' + (res.status || json.status));
json = await res.json();
errAPI = json.error;
}
if (errAPI) {
err.status = res.status || json.status;
err.planned = true;
errAPI.message && (err.message = errAPI.message);
errAPI.type && (err.type = errAPI.type);
if ((429 === res.status || 429 === json.status) && errAPI.resets_at) {
const hours = ((new Date(1e3 * errAPI.resets_at).getTime() - Date.now()) / 1e3 / 60 / 60).toFixed(1);
err.message += `, expires in ${hours} hours`;
}
if (throwIt) {
throw err;
}
}
return err;
};
module.exports.bytesToSize = (bytes = 0) => {
const b = [ 'B', 'KB', 'MB', 'GB', 'TB' ];
if (0 === bytes) {
return '0 B';
}
const c = Math.min(Math.floor(Math.log(bytes) / Math.log(1024)), 4);
return 0 === c ? `${bytes} ${b[c]}` : `${(bytes / 1024 ** c).toFixed(1)} ${b[c]}`;
};
module.exports.indexOfAny = (text, last = false) => {
let location = -1;
const fakes = [ indexOfH(text, last), indexOfA(text, last) ].filter((idx => idx > -1)).sort();
location = last ? fakes.reverse()[0] : fakes[0];
return isNaN(location) ? -1 : location;
};
module.exports.cleanJSON = json => json.replace(/^data: {/gim, '{').replace(/\s+$/gim, '');
module.exports.fileName = () => {
const len = randomInt(5, 15);
let name = randomBytes(len).toString('hex');
for (let i = 0; i < name.length; i++) {
const char = name.charAt(i);
isNaN(char) && randomInt(1, 5) % 2 == 0 && ' ' !== name.charAt(i - 1) && (name = name.slice(0, i) + ' ' + name.slice(i));
}
return name + '.txt';
};
module.exports.indexOfA = indexOfA;
module.exports.indexOfH = indexOfH;
module.exports.setTitle = title => {
title = `${Main} - ${title}`;
process.title !== title && (process.title = title);
};
module.exports.Main = Main;
module.exports.AI = AI; |