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

Use Express router for caption endpoint

Browse Source
This commit is contained in:
valadaptive 2023-12-04 12:43:37 -05:00
parent 414c9bd5fb
commit d2e1577acb
2 changed files with 26 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server.js
View File

@ -3621,7 +3621,7 @@ require('./src/endpoints/translate').registerEndpoints(app, jsonParser);
require('./src/endpoints/classify').registerEndpoints(app, jsonParser); require('./src/endpoints/classify').registerEndpoints(app, jsonParser);
// Image captioning // Image captioning
require('./src/endpoints/caption').registerEndpoints(app, jsonParser); app.use('/api/extra/caption', require('./src/endpoints/caption').router);
// Web search extension // Web search extension
require('./src/endpoints/serpapi').registerEndpoints(app, jsonParser); require('./src/endpoints/serpapi').registerEndpoints(app, jsonParser);

19
src/endpoints/caption.js
View File

@ -1,11 +1,11 @@
const express = require('express');
const { jsonParser } = require('../express-common');
const TASK = 'image-to-text'; const TASK = 'image-to-text';
/** const router = express.Router();
* @param {import("express").Express} app
* @param {any} jsonParser router.post('/', jsonParser, async (req, res) => {
*/
function registerEndpoints(app, jsonParser) {
app.post('/api/extra/caption', jsonParser, async (req, res) => {
try { try {
const { image } = req.body; const { image } = req.body;
@ -27,9 +27,6 @@ function registerEndpoints(app, jsonParser) {
console.error(error); console.error(error);
return res.sendStatus(500); return res.sendStatus(500);
} }
}); });
}
module.exports = { module.exports = { router };
registerEndpoints,
};