farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 19:00:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

add rateLimiter to prediction

Browse Source
This commit is contained in:
chungyau97
2023-08-30 21:43:42 +08:00
parent 1c1d6d2fd5
commit a58adcadf1
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/server/src/index.ts
+7 -3
View File
@@ -660,9 +660,14 @@ export class App {
// ---------------------------------------- // ----------------------------------------
// Send input message and get prediction result (External) // Send input message and get prediction result (External)
this.app.post('/api/v1/prediction/:id', upload.array('files'), async (req: Request, res: Response) => { this.app.post(
'/api/v1/prediction/:id',
upload.array('files'),
(req: Request, res: Response, next: NextFunction) => getRateLimiter(req, res, next),
async (req: Request, res: Response) => {
await this.processPrediction(req, res, socketIO) await this.processPrediction(req, res, socketIO)
}) }
)
// Send input message and get prediction result (Internal) // Send input message and get prediction result (Internal)
this.app.post('/api/v1/internal-prediction/:id', async (req: Request, res: Response) => { this.app.post('/api/v1/internal-prediction/:id', async (req: Request, res: Response) => {
@@ -765,7 +770,6 @@ export class App {
'/api/v1/rate-limit/:id', '/api/v1/rate-limit/:id',
upload.array('files'), upload.array('files'),
(req: Request, res: Response, next: NextFunction) => getRateLimiter(req, res, next), (req: Request, res: Response, next: NextFunction) => getRateLimiter(req, res, next),
// specificRouteLimiter,
async (req: Request, res: Response) => { async (req: Request, res: Response) => {
res.send("you're fine") res.send("you're fine")
} }