rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-02-12 10:00:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add error handling to AI21 tokenization

Browse Source
This commit is contained in:
Cohee 2023-09-01 15:44:41 +03:00
parent 66bc15edc0
commit 52ae038297
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server.js
View File

@ -3751,7 +3751,7 @@ async function sendAI21Request(request, response) {
}
app.post("/tokenize_ai21", jsonParser, function (request, response_tokenize_ai21 = response) {
app.post("/tokenize_ai21", jsonParser, async function (request, response_tokenize_ai21) {
if (!request.body) return response_tokenize_ai21.sendStatus(400);
const options = {
method: 'POST',
@ -3763,10 +3763,14 @@ app.post("/tokenize_ai21", jsonParser, function (request, response_tokenize_ai21
body: JSON.stringify({ text: request.body[0].content })
};
fetch('https://api.ai21.com/studio/v1/tokenize', options)
.then(response => response.json())
.then(response => response_tokenize_ai21.send({ "token_count": response.tokens.length }))
.catch(err => console.error(err));
try {
const response = await fetch('https://api.ai21.com/studio/v1/tokenize', options);
const data = await response.json();
return response_tokenize_ai21.send({ "token_count": data?.tokens?.length || 0 });
} catch (err) {
console.error(err);
return response_tokenize_ai21.send({ "token_count": 0 });
}
});
app.post("/save_preset", jsonParser, function (request, response) {