rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'staging' into neo-server

Browse Source
This commit is contained in:
Cohee
2024-04-20 02:56:05 +03:00
parent 78d1d48ea9 33ea3344ff
commit b3bbec83b6
16 changed files with 256 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/constants.js
View File

@ -191,6 +191,7 @@ const CHAT_COMPLETION_SOURCES = {
MISTRALAI: 'mistralai',
CUSTOM: 'custom',
COHERE: 'cohere',
PERPLEXITY: 'perplexity',
};
const UPLOADS_PATH = './uploads';

9
src/endpoints/backends/chat-completions.js
View File

@ -14,6 +14,7 @@ const API_OPENAI = 'https://api.openai.com/v1';
const API_CLAUDE = 'https://api.anthropic.com/v1';
const API_MISTRAL = 'https://api.mistral.ai/v1';
const API_COHERE = 'https://api.cohere.ai/v1';
const API_PERPLEXITY = 'https://api.perplexity.ai';
/**
* Applies a post-processing step to the generated messages.
@ -439,7 +440,7 @@ async function sendAI21Request(request, response) {
} else {
console.log(r.completions[0].data.text);
}
const reply = { choices: [{ 'message': { 'content': r.completions[0].data.text } }] };
const reply = { choices: [{ 'message': { 'content': r.completions?.[0]?.data?.text } }] };
return response.send(reply);
})
.catch(err => {
@ -896,6 +897,12 @@ router.post('/generate', jsonParser, function (request, response) {
request.body.char_name,
request.body.user_name);
}
} else if (request.body.chat_completion_source === CHAT_COMPLETION_SOURCES.PERPLEXITY) {
apiUrl = API_PERPLEXITY;
apiKey = readSecret(SECRET_KEYS.PERPLEXITY);
headers = {};
bodyParams = {};
request.body.messages = postProcessPrompt(request.body.messages, 'claude', request.body.char_name, request.body.user_name);
} else {
console.log('This chat completion source is not supported yet.');
return response.status(400).send({ error: true });

1
src/endpoints/secrets.js
View File

@ -36,6 +36,7 @@ const SECRET_KEYS = {
KOBOLDCPP: 'api_key_koboldcpp',
LLAMACPP: 'api_key_llamacpp',
COHERE: 'api_key_cohere',
PERPLEXITY: 'api_key_perplexity',
};
// These are the keys that are safe to expose, even if allowKeysExposure is false