rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use Express router for openai endpoint

This commit is contained in:
valadaptive 2023-12-04 12:53:17 -05:00
parent 414c9bd5fb
commit 2d19645c4e
2 changed files with 184 additions and 190 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server.js
View File

@ -3576,7 +3576,7 @@ async function fetchJSON(url, args = {}) {
// ** END ** // ** END **
// OpenAI API // OpenAI API
require('./src/endpoints/openai').registerEndpoints(app, jsonParser, urlencodedParser); app.use('/api/openai', require('./src/endpoints/openai').router);
// Tokenizers // Tokenizers
require('./src/endpoints/tokenizers').registerEndpoints(app, jsonParser); require('./src/endpoints/tokenizers').registerEndpoints(app, jsonParser);

24
src/endpoints/openai.js
View File

@ -1,16 +1,13 @@
const { readSecret, SECRET_KEYS } = require('./secrets'); const { readSecret, SECRET_KEYS } = require('./secrets');
const fetch = require('node-fetch').default; const fetch = require('node-fetch').default;
const express = require('express');
const FormData = require('form-data'); const FormData = require('form-data');
const fs = require('fs'); const fs = require('fs');
const { jsonParser, urlencodedParser } = require('../express-common');
/** const router = express.Router();
* Registers the OpenAI endpoints.
* @param {import("express").Express} app Express app router.post('/caption-image', jsonParser, async (request, response) => {
* @param {any} jsonParser JSON parser
* @param {any} urlencodedParser Form data parser
*/
function registerEndpoints(app, jsonParser, urlencodedParser) {
app.post('/api/openai/caption-image', jsonParser, async (request, response) => {
try { try {
let key = ''; let key = '';
@ -88,7 +85,7 @@ function registerEndpoints(app, jsonParser, urlencodedParser) {
} }
}); });
app.post('/api/openai/transcribe-audio', urlencodedParser, async (request, response) => { router.post('/transcribe-audio', urlencodedParser, async (request, response) => {
try { try {
const key = readSecret(SECRET_KEYS.OPENAI); const key = readSecret(SECRET_KEYS.OPENAI);
@ -136,7 +133,7 @@ function registerEndpoints(app, jsonParser, urlencodedParser) {
} }
}); });
app.post('/api/openai/generate-voice', jsonParser, async (request, response) => { router.post('/generate-voice', jsonParser, async (request, response) => {
try { try {
const key = readSecret(SECRET_KEYS.OPENAI); const key = readSecret(SECRET_KEYS.OPENAI);
@ -175,7 +172,7 @@ function registerEndpoints(app, jsonParser, urlencodedParser) {
} }
}); });
app.post('/api/openai/generate-image', jsonParser, async (request, response) => { router.post('/generate-image', jsonParser, async (request, response) => {
try { try {
const key = readSecret(SECRET_KEYS.OPENAI); const key = readSecret(SECRET_KEYS.OPENAI);
@ -209,8 +206,5 @@ function registerEndpoints(app, jsonParser, urlencodedParser) {
response.status(500).send('Internal server error'); response.status(500).send('Internal server error');
} }
}); });
}
module.exports = { module.exports = { router };
registerEndpoints,
};