File size: 1,600 Bytes
755dd12 |
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 |
import Koa from 'koa';
import Router from '@koa/router';
import cors from '@koa/cors';
import { bodyParser } from '@koa/bodyparser';
import serve from 'koa-static';
import path from 'path';
import { whiteListMiddleware } from './middleware';
import history from 'koa2-connect-history-api-fallback';
import { chatStreamController, localChatStreamController, localModelsController, modelsController, searchController, sogouSearchController } from './controller';
// eslint-disable-next-line @typescript-eslint/ban-ts-comment
// @ts-ignore
import dotenvx from '@dotenvx/dotenvx';
dotenvx.config();
const app = new Koa();
const router = new Router();
const port = process.env.PORT || 3000;
app.use(history({
index: '/index.html',
whiteList: ['/api']
}));
// static path
const staticPath = path.join(__dirname, '../web/build');
app.use(serve(staticPath, {
gzip: true,
index: 'index.html'
}));
app.use(cors({
origin: '*'
}));
app.use(bodyParser());
// Error handler
app.use(async (ctx, next) => {
try {
await next();
} catch(err) {
console.error('[server error]', err);
ctx.res.statusCode = 422;
ctx.body = err;
}
});
// router
app.use(router.routes()).use(router.allowedMethods());
// controller
router.post('/api/search', whiteListMiddleware(), searchController);
router.post('/api/sogou/search', sogouSearchController);
router.post('/api/chat', chatStreamController);
router.get('/api/models', modelsController);
// local llm
router.get('/api/local/models', localModelsController);
router.post('/api/local/chat', localChatStreamController);
app.listen(port);
|